Skip to content

Commit 0735b21

Browse files
Fix bug defining spatial grid
1 parent be3c718 commit 0735b21

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

RHS_1D_advection.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,10 @@ def generate_RHS_1D_advection(xlower=0.,xupper=2.,n_elements=40,
5151

5252
#We define our grid (interface points will be repeated)
5353
x_grid = np.zeros(n_elements*n_nodes)
54+
for i in range(n_elements):
55+
j = i*n_nodes
56+
x_grid[j:j+n_nodes] = x_element+x[i]
57+
list_elements.append(x_grid[j:j+n_nodes])
5458

5559
#Local opretaros P and Q
5660
P_LGL, Q_LGL = first_order_P_Q(x_Lagrange_nodes=xi_LGL, x_abcissae=xi_LGL, w_abcissae=w_LGL)
@@ -59,8 +63,8 @@ def generate_RHS_1D_advection(xlower=0.,xupper=2.,n_elements=40,
5963
P = np.kron(np.eye(n_elements), P_LGL)
6064
Q = np.kron(np.eye(n_elements), Q_LGL)
6165

62-
diagP=np.diag(P)
63-
diagPinv=1./diagP
66+
diagP = np.diag(P) #vector with the diagonal of P
67+
diagPinv = 1./diagP #vector with the inverse of the diagonal of P
6468

6569
#Restriction operators R, B
6670
#Local

0 commit comments

Comments
 (0)