new FD matrix builders

Author: Daniel Ruprecht

examples/acoustic_1d_imex/buildWave1DMatrix.py

@@ -0,0 +1,35 @@
+import sys
+sys.path.append('../')
+import numpy as np
+import scipy.sparse as sp
+from buildFDMatrix import getMatrix, getHorizontalDx, getBCLeft, getBCRight
+
+def getWave1DMatrix(N, dx, bc_left, bc_right):
+  
+  Id = sp.eye(2*N)
+
+  D_u  = getMatrix(N, dx, bc_left[0], bc_right[0])
+  D_p  = getMatrix(N, dx, bc_left[1], bc_right[1])
+  Zero = np.zeros((N,N))
+  M1 = sp.hstack((Zero, D_p), format="csc")
+  M2 = sp.hstack((D_u, Zero), format="csc")
+  M   = sp.vstack((M1, M2),  format="csc")
+  return sp.csc_matrix(Id), sp.csc_matrix(M)
+
+def getWave1DAdvectionMatrix(N, dx, order):
+  Dx   = getHorizontalDx(N, dx, order)
+  Zero = np.zeros((N,N))
+  M1   = sp.hstack((Dx, Zero), format="csc")
+  M2   = sp.hstack((Zero, Dx), format="csc")
+  M    = sp.vstack((M1, M2), format="csc")
+  return sp.csc_matrix(M)
+
+def getWaveBCLeft(value, N, dx, bc_left):
+  bu = getBCLeft(value[0],  N, dx, bc_left[0])
+  bp = getBCLeft(value[1],  N, dx, bc_left[1])
+  return np.concatenate((bp, bu))
+
+def getWaveBCRight(value, N, dx, bc_right):
+  bu = getBCRight(value[0], N, dx, bc_right[0])
+  bp = getBCRight(value[1], N, dx, bc_right[1])
+  return np.concatenate((bp, bu))