In [1]:

import micromagneticmodel as mm
import oommfc as mc

import discretisedfield as df  # df is here chosen to be an alias for discretisedfield

p1 = (0, 0, 0)
p2 = (100e-9, 50e-9, 20e-9)

region = df.Region(p1=p1, p2=p2)

In [2]:

import pyvista as pv

pv.set_jupyter_backend("trame")

In [3]:

pv.__version__

Out[3]:

'0.41.1'

In [4]:

lx, ly, lz = 100e-9, 50e-9, 20e-9

subregions = {
    "bottom_subregion": df.Region(p1=(20e-9, 0, 0), p2=(40e-9, 50e-9, 10e-9)),
    "top_subregion": df.Region(p1=(80e-9, 40e-9, lz / 2), p2=(lx, ly, lz)),
}

cell = (5e-9, 5e-9, 5e-9)

region = df.Region(p1=(0, 0, 0), p2=(lx, ly, lz))
mesh = df.Mesh(region=region, cell=cell, subregions=subregions)

In [5]:

plotter = pv.Plotter()
mesh.pyvista(plotter=plotter, wireframe=True)
plotter.show()

Widget(value="<iframe src='http://localhost:40295/index.html?ui=P_0x7f1c53f13f40_0&reconnect=auto' style='widt…

In [50]:

a, b, c = 1e-9, 1e-9, 1e-9
cell = (0.5e-9, 0.5e-9, 0.5e-9)

mesh = df.Mesh(p1=(-a, -b, -c), p2=(a, b, c), n=(5, 5, 5))  # 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")
field

Out[50]:

Field

Mesh
- Region
  - pmin = [-1e-09, -1e-09, -1e-09]
  - pmax = [1e-09, 1e-09, 1e-09]
  - dims = ['x', 'y', 'z']
  - units = ['m', 'm', 'm']
- n = [5, 5, 5]
nvdim = 3
vdims:
- x
- y
- z

In [44]:

import numpy as np


def test_field():
    mesh = df.Mesh(p1=(-5e-9, -5e-9, -5e-9), p2=(5e-9, 5e-9, 5e-9), n=(5, 5, 5))
    c_array = mesh.coordinate_field().array

    # The norm is defined via numpy for performance reasons;
    # In the simple loop form it would be:
    # x, y, _ = point
    # if x**2 + y**2 <= 5e-9**2:
    #     return 1e5
    # else:
    #     return 0
    def norm(points):
        return np.where(
            (points[..., 0] ** 2 + points[..., 1] ** 2) <= 5e-9**2, 1e5, 0
        )[..., np.newaxis]

    # Values are defined in numpy for performance reasons
    # We define vector fields with vx=0, vy=0, vz=+/-1 for x<0 / x>0
    def value(points):
        res = np.zeros((*mesh.n, 3))
        res[..., 2] = np.where(points[..., 0] <= 0, 1, -1)
        return res

    return df.Field(
        mesh,
        nvdim=3,
        value=value(c_array),
        norm=norm(c_array),
        vdims=["x", "y", "z"],
        valid=True,  #  "norm",
    )

In [46]:

field = test_field()
field

Out[46]:

Field

Mesh
- Region
  - pmin = [-5e-09, -5e-09, -5e-09]
  - pmax = [5e-09, 5e-09, 5e-09]
  - dims = ['x', 'y', 'z']
  - units = ['m', 'm', 'm']
- n = [5, 5, 5]
nvdim = 3
vdims:
- x
- y
- z

In [49]:

field.x

Out[49]:

Field

Mesh
- Region
  - pmin = [-5e-09, -5e-09, -5e-09]
  - pmax = [5e-09, 5e-09, 5e-09]
  - dims = ['x', 'y', 'z']
  - units = ['m', 'm', 'm']
- n = [5, 5, 5]
nvdim = 1

In [51]:

field.x

Out[51]:

Field

Mesh
- Region
  - pmin = [-1e-09, -1e-09, -1e-09]
  - pmax = [1e-09, 1e-09, 1e-09]
  - dims = ['x', 'y', 'z']
  - units = ['m', 'm', 'm']
- n = [5, 5, 5]
nvdim = 1

In [47]:

field.pyvista.valid()

Widget(value="<iframe src='http://localhost:40295/index.html?ui=P_0x7f1b501fc490_21&reconnect=auto' style='wid…

In [42]:

field.x.pyvista.contour(isosurfaces=5)

Widget(value="<iframe src='http://localhost:40295/index.html?ui=P_0x7f1b480f5610_20&reconnect=auto' style='wid…

In [ ]:

In [12]:

field.angle((1, 0, 0))

/home/holtsamu/repos/ubermag-devtools/repos/discretisedfield/discretisedfield/field.py:1242: RuntimeWarning: invalid value encountered in divide
  res_array = function(self.array, other)

Out[12]:

Field

Mesh
- Region
  - pmin = [-1e-09, -1e-09, -1e-09]
  - pmax = [1e-09, 1e-09, 1e-09]
  - dims = ['x', 'y', 'z']
  - units = ['m', 'm', 'm']
- n = [4, 4, 4]
nvdim = 1
unit = rad

In [38]:

field.pyvista.contour(contour_kwargs={"scalars": "x"})

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
/home/holtsamu/repos/ubermag-devtools/repos/discretisedfield/dev/pyvista.ipynb Cell 13 line 1
----> <a href='vscode-notebook-cell:/home/holtsamu/repos/ubermag-devtools/repos/discretisedfield/dev/pyvista.ipynb#X10sZmlsZQ%3D%3D?line=0'>1</a> field.pyvista.contour(contour_kwargs={"scalars": 'x'})

File ~/repos/ubermag-devtools/repos/discretisedfield/discretisedfield/plotting/pyvista_field.py:487, in PyVistaField.contour(self, isosurfaces, contour_scalars, plotter, multiplier, scalars, color_field, filename, contour_kwargs, **kwargs)
    482     scalars = "color_field"
    483 field_pv = field_pv.extract_cells(
    484     field_pv["valid"].astype(bool)
    485 ).cell_data_to_point_data()
--> 487 plot.add_mesh(
    488     field_pv.contour(isosurfaces=isosurfaces, **contour_kwargs),
    489     smooth_shading=True,
    490     scalars=scalars,
    491     **kwargs,
    492 )
    494 self._add_empty_region(plot, multiplier, self.field.mesh.region)
    495 plot.enable_eye_dome_lighting()

File ~/anaconda3/envs/ubermagdev/lib/python3.8/site-packages/pyvista/plotting/plotter.py:3433, in BasePlotter.add_mesh(self, mesh, color, style, scalars, clim, show_edges, edge_color, point_size, line_width, opacity, flip_scalars, lighting, n_colors, interpolate_before_map, cmap, label, reset_camera, scalar_bar_args, show_scalar_bar, multi_colors, name, texture, render_points_as_spheres, render_lines_as_tubes, smooth_shading, split_sharp_edges, ambient, diffuse, specular, specular_power, nan_color, nan_opacity, culling, rgb, categories, silhouette, use_transparency, below_color, above_color, annotations, pickable, preference, log_scale, pbr, metallic, roughness, render, component, emissive, copy_mesh, backface_params, show_vertices, **kwargs)
   3429         raise ValueError('RGB array must be n_points/n_cells by 3/4 in shape.')
   3431 if algo is None and not mesh.n_points:
   3432     # Algorithms may initialize with an empty mesh
-> 3433     raise ValueError('Empty meshes cannot be plotted. Input mesh has zero points.')
   3435 # set main values
   3436 self.mesh = mesh

ValueError: Empty meshes cannot be plotted. Input mesh has zero points.

In [14]:

field.y.pyvista.contour()

Widget(value="<iframe src='http://localhost:40295/index.html?ui=P_0x7f1c327eba30_5&reconnect=auto' style='widt…

In [9]:

field_pv = pv.wrap(field.to_vtk())
field_pv

Out[9]:

Header

Data Arrays

RectilinearGrid	Information
N Cells	1920
N Points	2457
X Bounds	-5.000e+00, 5.000e+00
Y Bounds	-3.000e+00, 3.000e+00
Z Bounds	-2.000e+00, 2.000e+00
Dimensions	21, 13, 9
N Arrays	6

Name	Field	Type	N Comp	Min	Max
norm	Cells	float64	1	0.000e+00	1.000e+06
x	Cells	float64	1	-9.918e+05	9.918e+05
y	Cells	float64	1	-9.972e+05	9.972e+05
z	Cells	float64	1	-9.802e+05	9.802e+05
field	Cells	float64	3	-9.972e+05	9.972e+05
valid	Cells	int64	1	0.000e+00	1.000e+00

In [10]:

pv.wrap(field.x.to_vtk())

Out[10]:

Header

Data Arrays

RectilinearGrid	Information
N Cells	1920
N Points	2457
X Bounds	-5.000e+00, 5.000e+00
Y Bounds	-3.000e+00, 3.000e+00
Z Bounds	-2.000e+00, 2.000e+00
Dimensions	21, 13, 9
N Arrays	3

Name	Field	Type	N Comp	Min	Max
norm	Cells	float64	1	0.000e+00	9.918e+05
field	Cells	float64	1	-9.918e+05	9.918e+05
valid	Cells	int64	1	0.000e+00	1.000e+00

In [11]:

field_pv["color_field"] = pv.wrap(field.x.to_vtk())["field"].copy()
field_pv

Out[11]:

Header

Data Arrays

RectilinearGrid	Information
N Cells	1920
N Points	2457
X Bounds	-5.000e+00, 5.000e+00
Y Bounds	-3.000e+00, 3.000e+00
Z Bounds	-2.000e+00, 2.000e+00
Dimensions	21, 13, 9
N Arrays	7

Name	Field	Type	N Comp	Min	Max
norm	Cells	float64	1	0.000e+00	1.000e+06
x	Cells	float64	1	-9.918e+05	9.918e+05
y	Cells	float64	1	-9.972e+05	9.972e+05
z	Cells	float64	1	-9.802e+05	9.802e+05
field	Cells	float64	3	-9.972e+05	9.972e+05
valid	Cells	int64	1	0.000e+00	1.000e+00
color_field	Cells	float64	1	-9.918e+05	9.918e+05

In [12]:

field.pyvista.vector()

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c6be32df0_2&reconnect=auto' style='widt…

In [13]:

field.pyvista.vector(cmap="viridis", color_field=field.y)

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9cd80d49a0_3&reconnect=auto' style='widt…

In [14]:

field.pyvista.vector(cmap="viridis", scalars="x")

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c6bbd0040_4&reconnect=auto' style='widt…

In [15]:

field.x.pyvista.scalar(opacity=[1, 0, 1])

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9cd80d4190_5&reconnect=auto' style='widt…

In [16]:

field.pyvista.contour(isosurfaces=5)

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c6e48baf0_6&reconnect=auto' style='widt…

In [17]:

field.pyvista.contour(isosurfaces=10, scalars="y")

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c6bbd4c70_7&reconnect=auto' style='widt…

In [21]:

field.pyvista.streamlines()

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c28060250_11&reconnect=auto' style='wid…

In [22]:

field.pyvista.valid(opacity=1, culling="backface")

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c28066790_12&reconnect=auto' style='wid…

In [20]:

plotter = pv.Plotter()
field.pyvista.vector(plotter=plotter, cmap="viridis")
field.pyvista.contour(plotter=plotter, isosurfaces=10, scalars="x")
plotter.show()

Widget(value="<iframe src='http://localhost:39181/index.html?ui=P_0x7f9c280662e0_10&reconnect=auto' style='wid…

In [ ]: