plotting¶
In [84]:
import matplotlib.pyplot as plt
import discretisedfield as df
mesh = df.Mesh(
p1=(-10e-9, -10e-9, -10e-9),
p2=(15e-9, 10e-9, 5e-9),
cell=(1e-9, 1e-9, 1e-9),
subregions={
"r1": df.Region(p1=(-10e-9, -10e-9, -10e-9), p2=(0, 10e-9, 5e-9)),
"r2": df.Region(p1=(0e-9, -10e-9, -10e-9), p2=(15e-9, 10e-9, 5e-9)),
},
)
def value_fun(point):
x, y, z = point
c = 1e9
if x < -5e-9:
return (0, y * c, z * c)
else:
return (x * c, y * c, z * c)
field = df.Field(mesh, dim=3, value=value_fun, norm=1e6)
Region¶
In [85]:
field.mesh.region.mpl()
In [86]:
field.mesh.region.mpl(figsize=(5, 5))
In [87]:
field.mesh.region.mpl(figsize=(5, 5), color="#000")
In [88]:
field.mesh.region.mpl(figsize=(5, 5), color="#000", multiplier=1e-6)
In [89]:
field.mesh.region.mpl(
figsize=(5, 5), color="#000", multiplier=1e-6, filename="region.pdf"
)
In [90]:
field.mesh.region.k3d()
Output()
In [91]:
field.mesh.region.k3d(color=0x4F5C6A)
Output()
In [92]:
field.mesh.region.k3d(color=0x4F5C6A, multiplier=1e-6)
Output()
In [93]:
field.mesh.region.k3d(color=0x4F5C6A, multiplier=1e-6, wireframe=True)
Output()
Mesh¶
In [94]:
mesh.mpl()
In [95]:
mesh.mpl(color=("C7", "C2"))
In [96]:
mesh.mpl(figsize=(3, 3))
In [97]:
mesh.mpl(multiplier=1e-6)
In [98]:
mesh.mpl(multiplier=1e-6, linewidth=5)
In [99]:
mesh.mpl(multiplier=1e-6, linewidth=2, filename="mesh.pdf")
In [100]:
mesh.mpl(multiplier=1e-6, linewidth=2, linestyle="dashed")
In [101]:
mesh.mpl_subregions()
In [102]:
mesh.mpl_subregions(figsize=(5, 5))
In [103]:
mesh.mpl_subregions(color=("C2", "C9"))
In [104]:
mesh.mpl_subregions(color=("C2", "C9"), multiplier=1e-6)
In [105]:
mesh.mpl_subregions(color=("C2", "C9"), multiplier=1e-6, linewidth=5)
In [106]:
mesh.mpl_subregions(
color=("C2", "C9"), multiplier=1e-6, linewidth=5, filename="subregions.pdf"
)
In [107]:
mesh.k3d()
Output()
In [108]:
mesh.k3d(color=[34568, 78905])
Output()
In [109]:
mesh.k3d(color=[34568, 78905], multiplier=1e-6)
Output()
In [110]:
mesh.k3d(color=[34568, 78905], multiplier=1e-6, wireframe=True)
Output()
In [111]:
mesh.k3d_subregions()
Output()
In [112]:
mesh.k3d_subregions(multiplier=1e-6)
Output()
In [113]:
mesh.k3d_subregions(multiplier=1e-6, color=(250, 250250))
Output()
In [114]:
mesh.k3d_subregions(multiplier=1e-6, color=(250, 250250), wireframe=True)
Output()
In [115]:
mesh.slider("x")
SelectionSlider(description='x (nm)', index=12, options=((-9.5, -9.5e-09), (-8.5, -8.5e-09), (-7.5, -7.5e-09),…
In [116]:
mesh.slider("y")
SelectionSlider(description='y (nm)', index=10, options=((-9.5, -9.5e-09), (-8.5, -8.5e-09), (-7.5, -7.5e-09),…
In [117]:
mesh.slider("z")
SelectionSlider(description='z (nm)', index=7, options=((-9.5, -9.5e-09), (-8.5, -8.5e-09), (-7.5, -7.5e-09), …
In [118]:
mesh.axis_selector()
Dropdown(description='axis', index=2, options=('x', 'y', 'z'), value='z')
In [119]:
mesh.axis_selector(widget="radiobuttons")
RadioButtons(description='axis', index=2, options=('x', 'y', 'z'), value='z')
In [120]:
mesh.axis_selector(widget="radiobuttons", description="A")
RadioButtons(description='A', index=2, options=('x', 'y', 'z'), value='z')
mpl_scalar¶
Not sliced -> error.
In [36]:
# field.mpl_scalar()
Vector field -> error
In [37]:
# field.plane('z').mpl_scalar()
In [38]:
field.x.plane("z").mpl_scalar()
Filter field must be passed
In [39]:
field.x.plane("z").mpl_scalar(filter_field=field.x)
Turn off colorbar
In [40]:
field.x.plane("z").mpl_scalar(filter_field=field.x, colorbar=False)
Colorbar label
In [41]:
field.x.plane("z").mpl_scalar(filter_field=field.x, colorbar_label="something")
Increase figure size to fit colorbar.
In [42]:
field.x.plane("z").mpl_scalar(
figsize=(12, 8), filter_field=field.x, colorbar_label="something"
)
Colormap limits
In [43]:
field.x.plane("z").mpl_scalar(
figsize=(9, 5), filter_field=field.x, colorbar_label="something", clim=(0, 1e6)
)
Change multiplier
In [44]:
field.x.plane("z").mpl_scalar(
figsize=(9, 5), filter_field=field.x, colorbar_label="something", multiplier=1e-12
)
In [45]:
field.x.plane("z").mpl_scalar(
figsize=(9, 5), filter_field=field.x, colorbar_label="something", multiplier=1
)
Saving field
In [46]:
field.x.plane("z").mpl_scalar(filename="scalar.pdf")
Passing imshow argument
In [47]:
field.x.plane("z").mpl_scalar(interpolation="bilinear")
mpl_vector¶
Not sliced -> error
In [48]:
# field.mpl_vector()
In [49]:
field.plane("z").mpl_vector()
Reduce the number of vectors.
In [50]:
field.plane("z", n=(20, 10)).mpl_vector()
Turn off color
In [51]:
field.plane("z", n=(20, 10)).mpl_vector(color=False)
Use different color field (color is still False).
In [52]:
field.plane("z", n=(20, 10)).mpl_vector(color=False, color_field=field.x)
Color is now on by default
In [53]:
field.plane("z", n=(20, 10)).mpl_vector(color_field=field.x, colorbar=True)
Turn off colorbar
In [54]:
field.plane("z", n=(20, 10)).mpl_vector(color_field=field.x, colorbar=False)
Add colorbar label
In [55]:
field.plane("z", n=(20, 10)).mpl_vector(color_field=field.x, colorbar_label="something")
Colormap
In [56]:
field.plane("z", n=(20, 10)).mpl_vector(
color_field=field.x, colorbar_label="something", cmap="plasma"
)
Colormap limit
In [57]:
field.plane("z", n=(20, 10)).mpl_vector(
color_field=field.x, colorbar_label="something", clim=(0, 1e6)
)
Multiplier
In [58]:
field.plane("z", n=(20, 10)).mpl_vector(
color_field=field.x, colorbar_label="something", multiplier=1
)
Figsize
In [59]:
field.plane("z", n=(20, 10)).mpl_vector(figsize=(5, 3))
Saving plot
In [60]:
field.plane("z", n=(20, 10)).mpl_vector(figsize=(5, 3), filename="vector.pdf")
Quiver argument
In [61]:
field.plane("z", n=(20, 10)).mpl_vector(headwidth=8)
Scale
In [62]:
field.curl.plane("z", n=(20, 10)).mpl_vector(scale=5e15)
mpl¶
Not sliced -> error
In [63]:
# field.mpl()
Default behaviour
In [64]:
field.plane("x").mpl()
Changing figsize to fit colorbar
In [65]:
field.plane("x").mpl(figsize=(10, 6))
Change scalar field
In [66]:
field.plane("x").mpl(figsize=(10, 6), scalar_field=field.y)
Turn off colorbar
In [67]:
field.plane("x").mpl(figsize=(10, 6), scalar_field=field.y, scalar_colorbar=False)
Add scalar colorbar label
In [68]:
field.plane("x").mpl(
figsize=(10, 6), scalar_field=field.y, scalar_colorbar_label="something"
)
Add filter field
In [69]:
field.plane("z").mpl(
figsize=(10, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
)
Scalar colormap
In [70]:
field.plane("z").mpl(
figsize=(10, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
)
In [71]:
field.plane("z").mpl(
figsize=(10, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
)
Vector field
In [72]:
field.plane("z").mpl(
figsize=(10, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field.div,
vector_scale=3e16,
)
Color vector field (vector_color is False)
In [73]:
field.plane("z").mpl(
figsize=(10, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field.div,
vector_color_field=field.y,
vector_scale=3e16,
)
vector_color is now True
In [74]:
field.plane("z").mpl(
figsize=(10, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field.curl,
vector_color_field=field.y,
vector_color=True,
vector_scale=7e15,
)
Add vector colorbar
In [75]:
field.plane("z").mpl(
figsize=(12, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field.div,
vector_color_field=field.y,
vector_color=True,
vector_colorbar=True,
vector_scale=3e16,
)
Label vector colorbar
In [76]:
field.plane("z").mpl(
figsize=(12, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field.div,
vector_color_field=field.y,
vector_color=True,
vector_colorbar=True,
vector_colorbar_label="vector",
vector_scale=3e16,
)
Vector cmap
In [77]:
field.plane("z").mpl(
figsize=(12, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field,
vector_color_field=field.y,
vector_color=True,
vector_colorbar=True,
vector_colorbar_label="vector",
vector_cmap="hsv",
)
Vector clim
In [78]:
field.plane("z").mpl(
figsize=(12, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field,
vector_color_field=field.y,
vector_color=True,
vector_colorbar=True,
vector_colorbar_label="vector",
vector_cmap="hsv",
vector_clim=(0, 1e6),
)
Multiplier
In [79]:
field.plane("z").mpl(
figsize=(12, 6),
scalar_filter_field=field.x,
scalar_field=field.y,
scalar_colorbar_label="something",
scalar_cmap="plasma",
scalar_clim=(0, 1e6),
vector_field=field,
vector_color_field=field.y,
vector_color=True,
vector_colorbar=True,
vector_colorbar_label="vector",
vector_cmap="hsv",
vector_clim=(0, 1e6),
multiplier=1e-12,
)
Save plot
In [80]:
field.plane("x").mpl(
figsize=(12, 6),
scalar_field=field.plane("x").angle,
scalar_colorbar_label="something",
scalar_cmap="twilight",
vector_field=field,
vector_color_field=field.y,
vector_color=True,
vector_colorbar=True,
vector_colorbar_label="vector",
vector_cmap="hsv",
vector_clim=(0, 1e6),
multiplier=1e-12,
filename="mpl.pdf",
)
