mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-30 12:55:59 +08:00
25 lines
743 B
Python
25 lines
743 B
Python
from SimPEG import *
|
|
|
|
def run(plotIt=True):
|
|
sz = [16,16]
|
|
tM = Mesh.TensorMesh(sz)
|
|
qM = Mesh.TreeMesh(sz)
|
|
qM.refine(lambda cell: 4 if np.sqrt(((np.r_[cell.center]-0.5)**2).sum()) < 0.4 else 3)
|
|
rM = Mesh.CurvilinearMesh(Utils.meshutils.exampleLrmGrid(sz,'rotate'))
|
|
|
|
if plotIt:
|
|
import matplotlib.pyplot as plt
|
|
fig, axes = plt.subplots(1,3,figsize=(14,5))
|
|
opts = {}
|
|
tM.plotGrid(ax=axes[0], **opts)
|
|
axes[0].set_title('TensorMesh')
|
|
qM.plotGrid(ax=axes[1], **opts)
|
|
axes[1].set_title('TreeMesh')
|
|
rM.plotGrid(ax=axes[2], **opts)
|
|
axes[2].set_title('CurvilinearMesh')
|
|
plt.show()
|
|
|
|
if __name__ == '__main__':
|
|
Utils._makeExample(__file__)
|
|
run()
|