mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-28 17:19:06 +08:00
Brendan Smithyman
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
from __future__ import unicode_literals
|
|
from __future__ import print_function
|
|
from __future__ import division
|
|
from __future__ import absolute_import
|
|
from future import standard_library
|
|
standard_library.install_aliases()
|
|
from builtins import zip
|
|
from builtins import range
|
|
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(list(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(list(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(list(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()
|