Skip to content

Commit 4594606

Browse files
author
Rajesh Meena
committed
add mode number and mode frequency in plotter text
1 parent 6b31ec6 commit 4594606

File tree

2 files changed

+16
-2
lines changed

2 files changed

+16
-2
lines changed

src/ansys/dpf/core/animation.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,18 @@ def animate_mode(
7171
)
7272

7373
# Get fields
74+
available_mode_numbers = fields_container.get_available_ids_for_label("time")
75+
76+
77+
if not mode_number in available_mode_numbers:
78+
raise ValueError(
79+
f"The mode {mode_number} data is not available in field container."
80+
)
7481
fields_mode = fields_container.get_fields({"time": mode_number})
82+
mode_frequencies_field = fields_container.time_freq_support.time_frequencies
83+
mode_frequencies = mode_frequencies_field.data
84+
mode_frequency = mode_frequencies[available_mode_numbers.index(mode_number)]
85+
7586

7687
# Merge fields if needed
7788
if len(fields_mode) > 1:
@@ -84,6 +95,7 @@ def animate_mode(
8495

8596
max_data = float(np.max(field_mode.data))
8697
loop_over = dpf.fields_factory.field_from_array(scale_factor_per_frame)
98+
loop_over.unit = mode_frequencies_field.unit
8799

88100
# Create workflow
89101
wf = dpf.Workflow()
@@ -106,6 +118,7 @@ def animate_mode(
106118
output_name="field",
107119
save_as=save_as,
108120
mode_number=mode_number,
121+
mode_frequency = mode_frequency,
109122
clim=[0, max_data],
110123
scale_factor=deform_scale_factor,
111124
**kwargs,

src/ansys/dpf/core/animator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ def animate_workflow(
3636
input_name="loop_over",
3737
save_as="",
3838
mode_number=None,
39+
mode_frequency = None,
3940
scale_factor=1.0,
4041
**kwargs,
4142
):
@@ -109,9 +110,9 @@ def render_frame(frame):
109110
str_template.format(indices[frame], unit, freq_fmt), **kwargs_in
110111
)
111112
else:
112-
str_template = "frq={0:{2}} {1}"
113+
str_template = "mode={3}\nfrq={0:{2}} {1}"
113114
self._plotter.add_text(
114-
str_template.format(mode_number, unit, freq_fmt), **kwargs_in
115+
str_template.format(mode_frequency, unit, freq_fmt, mode_number), **kwargs_in
115116
)
116117

117118
if cpos:

0 commit comments

Comments
 (0)