Issue #29 moved ave and ddx to utils.

This commit is contained in:
Rowan Cockett
2013-11-06 10:58:48 -08:00
parent d67df06ec3
commit 5f75ac9d69
3 changed files with 12 additions and 12 deletions
+10
View File
@@ -20,3 +20,13 @@ def kron3(A, B, C):
def spzeros(n1, n2):
"""spzeros"""
return sp.coo_matrix((n1, n2)).tocsr()
def ddx(n):
"""Define 1D derivatives, inner, this means we go from n+1 to n"""
return sp.spdiags((np.ones((n+1, 1))*[-1, 1]).T, [0, 1], n, n+1, format="csr")
def av(n):
"""Define 1D averaging operator from cell-centres to nodes."""
return sp.spdiags((0.5*np.ones((n+1, 1))*[1, 1]).T, [0, 1], n, n+1, format="csr")