wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-28 11:26:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

bug fix.

Browse Source
This commit is contained in:
rowanc1
2014-02-11 15:02:30 -08:00
parent aa539462a1
commit 62f00004eb
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
SimPEG/Mesh/TreeMesh.py
+8 -1
View File
@@ -893,9 +893,16 @@ class TreeMesh(InnerProducts, BaseMesh):
self.number()
# TODO: Preallocate!
I, J, V = [], [], []
# kinda a hack for the 2D gradient
# because edges are not stored
edges = self.faces if self.dim == 2 else self.edges
for edge in edges:
I += [edge.num, edge.num]
if self.dim == 3:
I += [edge.num, edge.num]
elif self.dim == 2 and edge.faceType == 'x':
I += [edge.num + self.nFy, edge.num + self.nFy]
elif self.dim == 2 and edge.faceType == 'y':
I += [edge.num - self.nFx, edge.num - self.nFx]
J += [edge.node0.num, edge.node1.num]
V += [-1, 1]
G = sp.csr_matrix((V,(I,J)), shape=(self.nE, self.nN))