fix create_fieldplot_surface (#4191)

Author: gmalinve

_unittest/test_12_1_PostProcessing.py

@@ -78,14 +78,14 @@ def test_01B_Field_Plot(self):
         assert self.aedtapp.post.field_plots[plot1.name].IsoVal == "Tone"
         assert plot1.change_plot_scale(min_value, "30000")
         assert self.aedtapp.post.create_fieldplot_volume("inner", "Vector_E", setup_name, intrinsic)
-
         assert self.aedtapp.post.create_fieldplot_surface(
             self.aedtapp.modeler["outer"].faces[0].id, "Mag_E", setup_name, intrinsic
         )
         assert self.aedtapp.post.create_fieldplot_surface(self.aedtapp.modeler["outer"], "Mag_E", setup_name, intrinsic)
         assert self.aedtapp.post.create_fieldplot_surface(
             self.aedtapp.modeler["outer"].faces, "Mag_E", setup_name, intrinsic
         )
+        assert not self.aedtapp.post.create_fieldplot_surface(123123123, "Mag_E", setup_name, intrinsic)
         assert len(self.aedtapp.setups[0].sweeps[0].frequencies) > 0
         assert isinstance(self.aedtapp.setups[0].sweeps[0].basis_frequencies, list)
         assert len(self.aedtapp.setups[0].sweeps[1].basis_frequencies) == 2

examples/03-Maxwell/Maxwell2D_NissanLeaf.py

@@ -708,25 +708,34 @@ def create_cs_magnets(pm_id, cs_name, point_direction):
 #                            report_category=None, plot_type="Rectangular Plot", context=None, subdesign_id=None,
 #                            polyline_points=1001, plotname=v)
 
+
+##########################################################
+# Analyze and save project
+# ~~~~~~~~~~~~~~~~~~~~~~~~
+# Analyze and save the project.
+
+M2D.save_project()
+M2D.analyze_setup(sName, use_auto_settings=False)
+
 ##########################################################
 # Create flux lines plot on region
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # Create a flux lines plot on a region. The ``object_list`` is
 # formerly created when the section is applied.
 
 faces_reg = mod2D.get_object_faces(object_list[1].name)  # Region
-# Maxwell Transient needs time specified in a dictionary
-# "IntrinsicVar:="	, "Time=\'0\'",
-M2D.post.create_fieldplot_surface(objlist=faces_reg, quantityName='Flux_Lines', intrinsincDict={"Time": "0.000"},
-                                  plot_name="Flux_Lines")
+plot1 = M2D.post.create_fieldplot_surface(objlist=faces_reg,
+                                          quantityName='Flux_Lines',
+                                          intrinsincDict={
+                                              "Time": M2D.variable_manager.variables["StopTime"].evaluated_value},
+                                          plot_name="Flux_Lines")
 
 ##########################################################
-# Analyze and save project
-# ~~~~~~~~~~~~~~~~~~~~~~~~
-# Analyze and save the project.
+# Export a field plot to an image file
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+# Export the flux lines plot to an image file using Python PyVista.
 
-M2D.save_project()
-M2D.analyze_setup(sName, use_auto_settings=False)
+M2D.post.plot_field_from_fieldplot(plot1.name, show=False)
 
 ###############################################
 # Get solution data
@@ -736,15 +745,15 @@ def create_cs_magnets(pm_id, cs_name, point_direction):
 
 solutions = M2D.post.get_solution_data(expressions="Moving1.Torque",
                                        primary_sweep_variable="Time")
-solutions.plot()
+#solutions.plot()
 
 ###############################################
 # Retrieve the data magnitude of an expression
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 # List of shaft torque points and compute average.
 
 mag = solutions.data_magnitude()
-avg = sum(mag)/len(mag)
+avg = sum(mag) / len(mag)
 
 ###############################################
 # Export a report to a file

examples/06-Multiphysics/Hfss_Icepak_Coupling.py

@@ -259,7 +259,7 @@
     view="isometric",
     show=False,
     plot_cad_objs=False,
-    log_scale = False,
+    log_scale=False,
 )
 
 ################################################################################
@@ -329,13 +329,13 @@
 # Generate pdf report
 # ~~~~~~~~~~~~~~~~~~~
 # Generate a pdf report with output of simultion.
-report = AnsysReport(project_name=aedtapp.project_name, design_name=aedtapp.design_name,version=desktopVersion)
+report = AnsysReport(project_name=aedtapp.project_name, design_name=aedtapp.design_name, version=desktopVersion)
 report.create()
 report.add_section()
 report.add_chapter("Hfss Results")
 report.add_sub_chapter("Field Plot")
 report.add_text("This section contains Field plots of Hfss Coaxial.")
-report.add_image(os.path.join(results_folder, plot1.name+".jpg"), "Coaxial Cable")
+report.add_image(os.path.join(results_folder, plot1.name + ".jpg"), "Coaxial Cable")
 report.add_page_break()
 report.add_sub_chapter("S Parameters")
 report.add_chart(my_data.intrinsics["Freq"], my_data.data_db20(), "Freq", trace_names[0], "S-Parameters")
@@ -345,11 +345,9 @@
 report.add_sub_chapter("Temperature Plot")
 report.add_text("This section contains Multiphysics temperature plot.")
 report.add_toc()
-#report.add_image(os.path.join(results_folder, plot5.name+".jpg"), "Coaxial Cable Temperatures")
+# report.add_image(os.path.join(results_folder, plot5.name+".jpg"), "Coaxial Cable Temperatures")
 report.save_pdf(results_folder, "AEDT_Results.pdf")
 
-
-
 ################################################################################
 # Close project and release AEDT
 # ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

pyaedt/modules/PostProcessor.py

@@ -22,6 +22,7 @@
 from pyaedt.generic.general_methods import open_file
 from pyaedt.generic.general_methods import pyaedt_function_handler
 from pyaedt.generic.settings import settings
+from pyaedt.modeler.cad.elements3d import FacePrimitive
 import pyaedt.modules.report_templates as rt
 from pyaedt.modules.solutions import FieldPlot
 from pyaedt.modules.solutions import SolutionData
@@ -3226,11 +3227,11 @@ def create_fieldplot_surface(
         if not isinstance(objlist, (list, tuple)):
             objlist = [objlist]
         new_obj_list = []
-        for objs in objlist:
-            if self._app.modeler[objs]:
-                new_obj_list.extend([i.id for i in self._app.modeler[objs].faces])
-            else:
-                new_obj_list.append(objs)
+        for obj in objlist:
+            if isinstance(obj, (int, FacePrimitive)):
+                new_obj_list.append(obj)
+            elif self._app.modeler[obj]:
+                new_obj_list.extend([face for face in self._app.modeler[obj].faces if face.id not in new_obj_list])
         return self._create_fieldplot(
             new_obj_list, quantityName, setup_name, intrinsincDict, "FacesList", plot_name, field_type=field_type
         )