main00_mesh.py

# -*- coding: utf-8; -*-

import pathlib

import matplotlib.pyplot as plt

import dolfin

from extrafeathers import meshfunction
from extrafeathers import meshiowrapper
from extrafeathers import plotmagic

from .config import mesh_filename, Domains


print(pathlib.Path.cwd())

meshiowrapper.import_gmsh(src="demo/meshes/cavity_with_obstacle.msh",
                          dst="demo/meshes/cavity_with_obstacle_full.h5")  # for use by the flow solvers
mesh, domain_parts, boundary_parts = meshiowrapper.read_hdf5_mesh("demo/meshes/cavity_with_obstacle_full.h5")

fluid_mesh = dolfin.SubMesh(mesh, domain_parts, Domains.FLUID.value)
fluid_domain_parts = meshfunction.specialize(domain_parts, fluid_mesh)
fluid_boundary_parts = meshfunction.specialize(boundary_parts, fluid_mesh)
meshiowrapper.write_hdf5_mesh(mesh_filename,
                              fluid_mesh, fluid_domain_parts, fluid_boundary_parts)

# Not used in simulation, but let's convert it, too.
structure_mesh = dolfin.SubMesh(mesh, domain_parts, Domains.STRUCTURE.value)
structure_domain_parts = meshfunction.specialize(domain_parts, structure_mesh)
structure_boundary_parts = meshfunction.specialize(boundary_parts, structure_mesh)
meshiowrapper.write_hdf5_mesh("demo/meshes/cavity_with_obstacle_structure.h5",
                              structure_mesh, structure_domain_parts, structure_boundary_parts)

# Visualize the fluid mesh
plt.figure(1)
plt.clf()

# mesh itself
plt.subplot(2, 2, 1)
dolfin.plot(fluid_mesh)
plt.ylabel("Mesh")

# local mesh size
plt.subplot(2, 2, 2)
theplot = dolfin.plot(meshfunction.meshsize(fluid_mesh))
plt.colorbar(theplot)
plt.ylabel("Local mesh size")

# domain parts (subdomains)
plt.subplot(2, 2, 3)
theplot = dolfin.plot(fluid_domain_parts)
plt.colorbar(theplot)
plt.ylabel("Phys. surfaces")

# boundary parts
plt.subplot(2, 2, 4)
plotmagic.plot_facet_meshfunction(fluid_boundary_parts, invalid_values=[2**64 - 1])
plt.axis("scaled")
plt.legend(loc="best")
plt.ylabel("Phys. boundaries")

plt.suptitle("Fluid")

plt.show()