pyvista¶Note: If you experience any problems in plotting with pyvista, please make sure you run the Jupyter notebook in Google Chrome.
There are several ways how a field can be visualised, using:
matplotlib-pyvista
holoviewspyvistapyvista provides three-dimensional interactive plots of fields inside Jupyter notebook.
Let us say we have a sample, which is an ellipsoid
$$\frac{x^2}{a^2} + \frac{y^2}{b^2} + \frac{z^2}{c^2} <= 1$$with $a=5\,\text{nm}$, $b=3\,\text{nm}$, and $c=2\,\text{nm}$. The space is discretised into cells with dimensions $(0.5\,\text{nm}, 0.5\,\text{nm}, 0.5\,\text{nm})$. The value of the field at $(x, y, z)$ point is $(-cy, cx, cz)$, with $c=10^{9}$. The norm of the field inside the cylinder is $10^{6}$.
Let us first build that field.
import discretisedfield as df
a, b, c = 5e-9, 3e-9, 2e-9
cell = (0.5e-9, 0.5e-9, 0.5e-9)
mesh = df.Mesh(p1=(-a, -b, -c), p2=(a, b, c), cell=cell)
def norm_fun(pos):
x, y, z = pos
if (x / a) ** 2 + (y / b) ** 2 + (z / c) ** 2 <= 1:
return 1e6
else:
return 0
def value_fun(pos):
x, y, z = pos
c = 1e9
return (-c * y, c * x, c * z)
field = df.Field(mesh, nvdim=3, value=value_fun, norm=norm_fun, valid="norm")
The most basic plot we can show is the plot of all the cells where the value is valid.
field.pyvista.valid()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fce5850e980_0&reconnect=auto' style='widt…
The plot is interactive, so it can be manipulated using a mouse. To change the color of voxels, we can pass the new color via color argument.
field.pyvista.valid(color="red")
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcdbcb850_1&reconnect=auto' style='widt…
Next, we can plot a scalar field. For plotting a scalar field, we are using discretisedfield.Field.pyvista.scalar() method. This produces an interactive slice of the field.
field.pyvista.scalar()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcf1224a0_2&reconnect=auto' style='widt…
By default, this plots the last vdims. However, we can plot any vdims by using the scalars argument.
field.pyvista.scalar(scalars="x")
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcf1227d0_3&reconnect=auto' style='widt…
Alternatively, we can also create a scalar field using discreitsedfield syntax and plot that.
field.pyvista.scalar()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdda83bd30_4&reconnect=auto' style='widt…
To further modify the plot, keyword arguments for pyvista.plotter.add_mesh function are accepted. Please refer to its documentation
Next, we can plot the vector field itself.
field.pyvista.vector()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fce5846dfc0_5&reconnect=auto' style='widt…
By cascading, we can plot its slice at $x=0$.
field.orientation.sel(x=(0, 0)).pyvista.vector()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcdbcbc70_6&reconnect=auto' style='widt…
To improve the understanding of the plot, we can colour the vectors plotted. For that, we need to tell it according to which the vectors will be coloured.
field.pyvista.vector(scalars="x")
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcda93ca0_7&reconnect=auto' style='widt…
We can also colour by any field by using the color_field argument
field.pyvista.vector(color_field=field.x**2)
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcda92380_8&reconnect=auto' style='widt…
Plotting a scalar can also be done using volume. The density and color within the volume represent the scalar value at each point.
field.pyvista.volume()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdd1f7a350_9&reconnect=auto' style='widt…
Isosurfaces of a scalar field can also be plotted using contour
field.pyvista.contour()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcda91e10_10&reconnect=auto' style='wid…
contour_scalars can be used to select the dimension for which the isosurfaces are plotted. If isosurfaces controls the number of isosurfaces plotted.
field.pyvista.contour(isosurfaces=4, contour_scalars="x")
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdcda93520_11&reconnect=auto' style='wid…
If isosurfaces is an iterable, then an isosurface will be plotted at each specific value.
field.pyvista.contour(isosurfaces=[-6e5, 0.0, 2e5], contour_scalars="x")
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdc679ee30_12&reconnect=auto' style='wid…
Streamlines can also be plotted. However, it is common for the default values not to produce desired plots as the optimal values of the arguments can very unique to the specific field being plotted.
field.pyvista.streamlines()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdb7f85330_13&reconnect=auto' style='wid…
Sometimes, it is necessary to show, for example, multiple planes of the sample on the same plot. This can be done by exposing the pyvista.Plotter and passing it to different plotting methods. For instance.
import pyvista
plotter = pyvista.Plotter()
field.pyvista.vector(plotter=plotter, scalars="y", cmap="hsv")
field.pyvista.contour(plotter=plotter, isosurfaces=4, contour_scalars="x")
plotter.show()
Widget(value="<iframe src='http://localhost:45445/index.html?ui=P_0x7fcdb7f87dc0_14&reconnect=auto' style='wid…