Source code for sfepy.postprocess.plot_cmesh

"""
Functions to visualize the CMesh geometry and topology.
"""
from __future__ import absolute_import
from sfepy.postprocess.plot_dofs import _get_axes, _to2d
from six.moves import range

[docs]def plot_wireframe(ax, cmesh, color='k', **kwargs): """ Plot a finite element mesh as a wireframe using edges connectivity. """ coors = cmesh.coors coors = _to2d(coors) dim = cmesh.dim ax = _get_axes(ax, dim) cmesh.setup_connectivity(1, 0) edges = cmesh.get_conn(1, 0) for edge_vertices in edges.indices.reshape((edges.num, 2)): cc = coors[edge_vertices] ax.plot(*cc.T, color=color, **kwargs) return ax
[docs]def plot_entities(ax, cmesh, edim, color='b', size=10, **kwargs): """ Plot mesh topology entities using scatter plot. """ coors = cmesh.get_centroids(edim) coors = _to2d(coors) dim = cmesh.dim ax = _get_axes(ax, dim) ax.scatter(*coors.T, s=size, c=color, **kwargs) return ax
[docs]def label_global_entities(ax, cmesh, edim, color='b', fontsize=10, **kwargs): """ Label mesh topology entities using global ids. """ coors = cmesh.get_centroids(edim) coors = _to2d(coors) dim = cmesh.dim ax = _get_axes(ax, dim) for ii, cc in enumerate(coors): ax.text(*cc.T, s=ii, color=color, fontsize=fontsize, horizontalalignment='center', verticalalignment='center', **kwargs) return ax
[docs]def label_local_entities(ax, cmesh, edim, color='b', fontsize=10, **kwargs): """ Label mesh topology entities using cell-local ids. """ coors = cmesh.get_centroids(edim) coors = _to2d(coors) dim = cmesh.dim centres = cmesh.get_centroids(dim) cmesh.setup_connectivity(dim, edim) conn = cmesh.get_conn(dim, edim) off = conn.offsets ax = _get_axes(ax, dim) eps = 0.015 * fontsize oeps = 1.0 - eps for ii in range(conn.num): for ic, ie in enumerate(conn.indices[off[ii]:off[ii+1]]): # Shift labels towards the cell centre. cc = oeps * coors[ie] + eps * centres[ii] ax.text(*cc.T, s=ic, color=color, fontsize=fontsize, horizontalalignment='center', verticalalignment='center', **kwargs) return ax
[docs]def plot_cmesh(ax, cmesh, wireframe_opts=None, entities_opts=None): """ Convenience function for plotting all entities of a finite element mesh. Pass `plot()` arguments to `wireframe_opts` dict. Pass `'color'`, `'label_global'`, `'label_global'` for `text()` color and font sizes arguments and `'size'` for `scatter()` to each dict for topological entities in `entities_opts` list. Examples -------- >>> # 2D mesh. >>> plot_cmesh(None, cmesh, wireframe_opts = {'color' : 'k', 'linewidth' : 2}, entities_opts=[ {'color' : 'k', 'label_local' : 8, 'size' : 20}, {'color' : 'b', 'label_global' : 12, 'label_local' : 8, 'size' : 10}, {'color' : 'r', 'label_global' : 12, 'size' : 20}, ]) """ dim = cmesh.dim ax = _get_axes(ax, dim) if wireframe_opts is None: wireframe_opts = {} if wireframe_opts.get('color') is not None: ax = plot_wireframe(ax, cmesh, **wireframe_opts) if entities_opts is None: entities_opts = [{}] * dim for ii, entity_opts in enumerate(entities_opts): if entity_opts.get('color') is not None: fsg = entity_opts.pop('label_global', 0) fsl = entity_opts.pop('label_local', 0) size = entity_opts.pop('size', 10) ax = plot_entities(ax, cmesh, ii, size=size, **entity_opts) if fsg: ax = label_global_entities(ax, cmesh, ii, fontsize=fsg, **entity_opts) if fsl: ax = label_local_entities(ax, cmesh, ii, fontsize=fsl, **entity_opts) return ax