Advanced pyvista plotting and annotation options

[lorenzovecchietti]

It is possible to get the pyvista plotter object and modify it further for advanced plotting and annotation.

[lorenzovecchietti]

In this example I am:

1. changing the colors of the objects
2. adding the point monitors with their labels
3. exporting the plot in a pdf file

import pyaedt
import numpy as np
import pyvista as pv

ipk=pyaedt.Icepak("GraphicCard.aedt")

## get the pyvista object with the whole model from pyaedt
pyvista_plot = ipk.plot(show=False)
pyvista_plot.show_legend = False

## change colors of objects
def hex_to_rgb(value): # see: https://stackoverflow.com/questions/29643352/converting-hex-to-rgb-value-in-python
          h=value.lstrip("#")
          return tuple(int(h[i:i+2], 16) for i in (0, 2, 4)))

colors = ("#001ah9, #001219, #005f73, #0a9396, #94d2bd, #e9d8a6, #ee9b00, #ca6702, #bb3e03, #ae2012, #9b2226, #001219, "
          "#005f73").split(", ")
colors = ["white"] + colors
for actor, c in zip(pyvista_plot.objects, colors):
    obj.color = hex_to_rgb(c)
    obj.opacity = 1

pyvista_plot.generate_geometry_mesh()
p=pyvista_plot.pv

## Add the monitors label
points = []
labels = []
for mon_name, mon_value in ipk.monitor.point_monitors.items():
    point_coords = mon_value.location
    labels.append(mon_name)
    points.append(point_coords)
points = np.array(points)
actor = p.add_point_labels(
    points, labels, always_visible=True, point_color="red", font_size=8, point_size=5, margin=5, shape_color="white",
    shape_opacity=0.3
)

# select top view and export
p.camera_position = 'zx'
p.save_graphic("monitors.pdf")

The result looks like this: