New examples for Mesh

2026-08-11 11:26:01 +08:00 · 2015-11-26 13:56:04 -07:00
parent be3667b4ab
commit b2aab4163b
4 changed files with 237 additions and 0 deletions
@@ -0,0 +1,49 @@
+from SimPEG import *
+
+def run(plotIt=True, n=60):
+    """
+        Mesh: QuadTree: FaceDiv
+        =======================
+
+
+
+    """
+
+
+    M = Mesh.TreeMesh([[(1,16)],[(1,16)]], levels=4)
+    M._refineCell([0,0,0])
+    M._refineCell([0,0,1])
+    M._refineCell([4,4,2])
+    M.__dirty__ = True
+    M.number()
+
+
+    if plotIt:
+        import matplotlib.pyplot as plt
+        fig, axes = plt.subplots(2,1,figsize=(10,10))
+
+        M.plotGrid(cells=True, nodes=False, ax=axes[0])
+        axes[0].axis('off')
+        axes[0].set_title('Simple QuadTree Mesh')
+        axes[0].set_xlim([-1,17])
+        axes[0].set_ylim([-1,17])
+
+        for ii, loc in zip(range(M.nC),M.gridCC):
+            axes[0].text(loc[0]+0.2,loc[1],'%d'%ii, color='r')
+
+        axes[0].plot(M.gridFx[:,0],M.gridFx[:,1], 'g>')
+        for ii, loc in zip(range(M.nFx),M.gridFx):
+            axes[0].text(loc[0]+0.2,loc[1],'%d'%ii, color='g')
+
+        axes[0].plot(M.gridFy[:,0],M.gridFy[:,1], 'm^')
+        for ii, loc in zip(range(M.nFy),M.gridFy):
+            axes[0].text(loc[0]+0.2,loc[1]+0.2,'%d'%(ii+M.nFx), color='m')
+
+        axes[1].spy(M.faceDiv)
+        axes[1].set_title('Face Divergence')
+        axes[1].set_ylabel('Cell Number')
+        axes[1].set_xlabel('Face Number')
+        plt.show()
+
+if __name__ == '__main__':
+    run()