diff --git a/SimPEG/Mesh/NewTreeMesh.py b/SimPEG/Mesh/NewTreeMesh.py index 42848992..7bc904c2 100644 --- a/SimPEG/Mesh/NewTreeMesh.py +++ b/SimPEG/Mesh/NewTreeMesh.py @@ -114,7 +114,10 @@ class TreeMesh(object): def refineFace(self, index): f = self._faces[index,:] - nodeNums = self._edges[f[[FEDGE0, FEDGE1]],:][:,[ENODE0, ENODE1]] + if f[ACTIVE] == 0: + # search for the children up to one level deep + subInds = np.argwhere(self._faces[:,PARENT] == index).flatten() + return subInds, self._faces[subInds,:] self._faces[index, ACTIVE] = 0 @@ -135,6 +138,7 @@ class TreeMesh(object): E2i, E2 = self.refineEdge(f[FEDGE2]) E3i, E3 = self.refineEdge(f[FEDGE3]) + nodeNums = self._edges[f[[FEDGE0, FEDGE1]],:][:,[ENODE0, ENODE1]] newNode, node = self.addNode(nodeNums) # Refine the inner edges @@ -479,11 +483,6 @@ if __name__ == '__main__': plt.spy(tM.faceDiv) tM.plotGrid(ax=plt.subplot(212)) - - print tM.vol - print tM.area - print tM.edge - # plt.figure(2) # plt.plot(SortByX0(tM.gridCC),'b.') plt.show()