def show_mesh(self, p=2, plot=True):
mf = MeshFactory()
mesh = mf.boxmesh2d([0, 1, 0, 1], nx=2, ny=2, meshtype='tri')
node = mesh.entity('node')
cell = mesh.entity('cell')
ltmesh = LagrangeTriangleMesh(node, cell, p=p)
NN = ltmesh.number_of_nodes()
mesh.ds.edge = ltmesh.lds.edge
mesh.ds.edge2cell = ltmesh.lds.edge2cell
node = ltmesh.entity('node')
#ltmesh.print()
if plot:
fig = plt.figure()
axes = fig.gca()
mesh.add_plot(axes)
mesh.find_node(axes, node=node, showindex=True, fontsize=28)
mesh.find_edge(axes, showindex=True)
mesh.find_cell(axes, showindex=True)
plt.show()
from fealpy.mesh import LagrangeTriangleMesh, MeshFactory, LagrangeWedgeMesh
from fealpy.writer import MeshWriter
#p = int(sys.argv[1])
#n = int(sys.argv[2])
#h = sys.argv[3]
#nh = int(sys.argv[4])
fname = 'initial/file1.vtu'
data = meshio.read(fname)
node = data.points
cell = data.cells[0][1]
mesh = LagrangeTriangleMesh(node*500, cell, p=1)
mesh.uniform_refine(n=0)
mesh = LagrangeWedgeMesh(mesh, h=0.005, nh=1, p=1)
edge = mesh.entity('edge')
node = mesh.entity('node')
cell = mesh.entity('cell')
tface, qface = mesh.entity('face')
print(tface.shape)
mesh.celldata['a'] = np.arange(len(cell))
mesh.to_vtk(fname='write.vtu')