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:

[lorenzovecchietti]

This script demonstrates how to annotate the point of maximum temperature on a specific object (e.g., "Box2").

1. Initialize AEDT Icepak and Plot Field
2. Retrieve Maximum Temperature Data (method added in FEAT: Method to get max temp location #5840)
3. Annotate the Plot with Maximum Temperature
4. Adjust View and Save Screenshot

from pyvista import Plotter

import ansys.aedt.core as pyaedt
from ansys.aedt.core.visualization.plot.pyvista import ModelPlotter

ipk: pyaedt.Icepak = pyaedt.Icepak(version=2025.1)
plot: ModelPlotter = ipk.post.plot_field(
    "Temperature",
    "Box2",
    show=False,
    keep_plot_after_generation=True,
    plot_cad_objs=False,
)
plot.plot()  # populate pyvista attribute, as `show=False`, nothing will show up

max_temp: dict = ipk.post.get_temperature_extremum(
    assignment="Box2", max_min="Max", location="Surface"
)

plot_pv: Plotter = plot.pv
plot_pv.add_point_labels(
    [max_temp["Position"]],
    [f"{max_temp['Value']:0.2f}"],
    always_visible=False,
    point_color="red",
    font_size=18,
    point_size=25,
    margin=5,
    shape_color="white",
    shape_opacity=0.5,
)

plot_pv.view_yx()  # point on the surface must be visible from the view
plot_pv.screenshot("test.png")

Example output: