Homec4science
Diffusion GooseFEM (neighbour) Browse

rGOOSEFEM/docs/details/figures/MeshQuad4/Map/FineLayer2Regular/map.py
dffe74d45a19neighbour

Last Change
Raw File

map.py
View Options

import GooseFEM as gf
import matplotlib.pyplot as plt
import GooseMPL as gplt
import numpy as np
plt.style.use(['goose'])
# --------------------------------------------------------------------------------------------------
fig, axes = plt.subplots(figsize=(10,10), ncols=2)
# ---
ax = axes[0]
mesh = gf.Mesh.Quad4.FineLayer(6*3, 18*2)
coor = mesh.coor()
conn = mesh.conn()
cindex = np.random.random(conn.shape[0])
im = gplt.patch(coor=coor, conn=conn, cindex=cindex, cmap='jet', axis=ax)
ax.set_aspect('equal')
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
# ---
ax = axes[1]
mapping = gf.Mesh.Quad4.Map.FineLayer2Regular(mesh)
new_mesh = mapping.getRegular()
coor = new_mesh.coor()
conn = new_mesh.conn()
c = mapping.map(cindex)
im = gplt.patch(coor=coor, conn=conn, cindex=c, cmap='jet', axis=ax)
ax.set_aspect('equal')
ax.get_xaxis().set_visible(False)
ax.get_yaxis().set_visible(False)
# ---
plt.savefig('map.svg')
plt.show()
c4science · Help