From 227566151add0e84c6c453d65aea5be688387362 Mon Sep 17 00:00:00 2001 From: Antoine Karcher Date: Wed, 14 Jun 2023 19:43:50 +0200 Subject: [PATCH] Fixed factory regarding time freq support --- .../dpf/core/fields_container_factory.py | 22 +++++++++++-------- 1 file changed, 13 insertions(+), 9 deletions(-) diff --git a/src/ansys/dpf/core/fields_container_factory.py b/src/ansys/dpf/core/fields_container_factory.py index ba76c342d6..b4dd9e8c1b 100644 --- a/src/ansys/dpf/core/fields_container_factory.py +++ b/src/ansys/dpf/core/fields_container_factory.py @@ -49,28 +49,32 @@ def over_time_freq_fields_container(fields, time_freq_unit=None, server=None): if not isinstance(fields, dict) and not isinstance(fields, list): raise dpf_errors.InvalidTypeError("dictionary/list", "fields") fc = FieldsContainer(server=server) + time_freq_support = TimeFreqSupport(server=server) fc.labels = {"time"} i = 0 + time_freq = [] # dict case if isinstance(fields, dict): - time_freq = [] + for field_key in fields: fc.add_field({"time": i + 1}, fields[field_key]) time_freq.append(field_key) i += 1 - time_freq_field = fields_factory.create_scalar_field( - len(fields), location=locations.time_freq, server=server - ) - time_freq_field.append(time_freq, 1) - time_freq_field.unit = time_freq_unit - time_freq_support = TimeFreqSupport(server=server) - time_freq_support.time_frequencies = time_freq_field - fc.time_freq_support = time_freq_support # list case elif isinstance(fields, list): for field in fields: fc.add_field({"time": i + 1}, field) + time_freq.append(i) i += 1 + + time_freq_field = fields_factory.create_scalar_field( + len(fields), location=locations.time_freq, server=server + ) + time_freq_field.append(time_freq, 1) + if time_freq_unit: + time_freq_field.unit = time_freq_unit + time_freq_support.time_frequencies = time_freq_field + fc.time_freq_support = time_freq_support return fc