Element-wise post-processing

[dokken]

I’ve made a local assembler here:

How to enumerate the cells of a mesh with dolfinx v0.6.0?

It is very simply to enumerate cells in DOLFINx. The number of cells on a process can be found by num_cells = msh.topology.index_map(msh.topology.dim).size_local, where the ith local index is i. It is not included in DOLFINx by default. However, here is a code that does what you want import numpy as np import ufl from mpi4py import MPI from petsc4py.PETSc import ScalarType from ufl import ds, dx, grad, inner from dolfinx import fem, io, mesh, plot import cffi class LocalAssembler(): …

You can then invert each local matrix to Get the solution per cell, and cast it to the u.x.array by accessing the cell dofs in the dofmap.