Local surface provider.

Author: ajain-work

ansys/fluent/post/matplotlib/matplot_objects.py

@@ -13,7 +13,7 @@ class Plots:
 
     _sessions_state = {}
 
-    def __init__(self, session):
+    def __init__(self, session, local_surfaces_provider=None):
         """Instantiate Plots, container of plot objects."""
         session_state = Plots._sessions_state.get(session.id if session else 1)
         if not session_state:
@@ -23,6 +23,13 @@ def __init__(self, session):
             self._init_module(self, sys.modules[__name__])
         else:
             self.__dict__ = session_state
+        self._local_surfaces_provider = (
+            lambda: local_surfaces_provider
+            if local_surfaces_provider
+            else self.Surfaces
+            if hasattr(self, "Surfaces")
+            else []
+        )
 
     def _init_module(self, obj, mod):
         for name, cls in mod.__dict__.items():
@@ -50,4 +57,6 @@ def plot(self, window_id: Optional[str] = None):
             Window id. If not specified unique id is used.
 
         """
+        self._pre_display()
         matplot_windows_manager.plot(self, window_id)
+        self._post_display()

ansys/fluent/post/matplotlib/matplot_windows_manager.py

@@ -27,9 +27,7 @@ def __init__(
     ):
         self._closed = False
         self.plot_pipe, plotter_pipe = mp.Pipe()
-        self.plotter = ProcessPlotter(
-            window_id, curves, title, xlabel, ylabel
-        )
+        self.plotter = ProcessPlotter(window_id, curves, title, xlabel, ylabel)
         self.plot_process = mp.Process(
             target=self.plotter, args=(plotter_pipe,), daemon=True
         )

ansys/fluent/post/post_object_defns.py

@@ -10,7 +10,41 @@
 )
 
 
-class GraphicsDefn(metaclass=PyLocalNamedObjectMetaAbstract):
+class BasePostObjectDefn:
+    """Base class for post objects."""
+
+    def _pre_display(self):
+        local_surfaces_provider = (
+            self._get_top_most_parent()._local_surfaces_provider()
+        )
+        for surf_name in self.surfaces_list():
+            if surf_name in list(local_surfaces_provider):
+                surf_obj = local_surfaces_provider[surf_name]
+                if surf_obj.surface.type() == "iso-surface":
+                    surf_api = surf_obj._data_extractor.surface_api()
+                    surf_api.iso_surface(
+                        surf_obj.surface.iso_surface.field(),
+                        surf_name,
+                        (),
+                        (),
+                        surf_obj.surface.iso_surface.iso_value(),
+                        (),
+                    )
+
+    def _post_display(self):
+        local_surfaces_provider = (
+            self._get_top_most_parent()._local_surfaces_provider()
+        )
+        for surf_name in self.surfaces_list():
+            if surf_name in list(local_surfaces_provider):
+                surf_obj = local_surfaces_provider[surf_name]
+                surf_api = surf_obj._data_extractor.surface_api()
+                surf_api.delete_surface(surf_name)
+
+
+class GraphicsDefn(
+    BasePostObjectDefn, metaclass=PyLocalNamedObjectMetaAbstract
+):
     """Abstract base class for graphics objects."""
 
     @abstractmethod
@@ -26,7 +60,7 @@ def display(self, plotter_id: Optional[str] = None):
         pass
 
 
-class PlotDefn(metaclass=PyLocalNamedObjectMetaAbstract):
+class PlotDefn(BasePostObjectDefn, metaclass=PyLocalNamedObjectMetaAbstract):
     """Abstract base class for plot objects."""
 
     @abstractmethod
@@ -106,7 +140,7 @@ def allowed_values(self):
             """Surface list allowed values."""
             return list(
                 self._data_extractor.field_info().get_surfaces_info().keys()
-            )
+            ) + list(self._get_top_most_parent()._local_surfaces_provider())
 
 
 class MeshDefn(GraphicsDefn):
@@ -124,7 +158,7 @@ def allowed_values(self):
             """Surface list allowed values."""
             return list(
                 (self._data_extractor.field_info().get_surfaces_info().keys())
-            )
+            ) + list(self._get_top_most_parent()._local_surfaces_provider())
 
     class show_edges(metaclass=PyLocalPropertyMeta):
         """Show edges for mesh."""
@@ -142,17 +176,17 @@ class show_edges(metaclass=PyLocalPropertyMeta):
 
         value: bool = True
 
-    class surface_type(metaclass=PyLocalObjectMeta):
+    class surface(metaclass=PyLocalObjectMeta):
         """Specify surface type."""
 
         def _availability(self, name):
             if name == "plane_surface":
-                return self.surface_type() == "plane-surface"
+                return self.type() == "plane-surface"
             if name == "iso_surface":
-                return self.surface_type() == "iso-surface"
+                return self.type() == "iso-surface"
             return True
 
-        class surface_type(metaclass=PyLocalPropertyMeta):
+        class type(metaclass=PyLocalPropertyMeta):
             """Surface type."""
 
             value: str = "iso-surface"
@@ -251,7 +285,7 @@ def allowed_values(self):
             """Surfaces list allowed values."""
             return list(
                 self._data_extractor.field_info().get_surfaces_info().keys()
-            )
+            ) + list(self._get_top_most_parent()._local_surfaces_provider())
 
     class filled(metaclass=PyLocalPropertyMeta):
         """Show filled contour."""
@@ -278,7 +312,7 @@ class show_edges(metaclass=PyLocalPropertyMeta):
 
         value: bool = False
 
-    class range (metaclass=PyLocalObjectMeta):
+    class range(metaclass=PyLocalObjectMeta):
         """Specify range options."""
 
         def _availability(self, name):
@@ -407,7 +441,7 @@ def allowed_values(self):
             """Surface list allowed values."""
             return list(
                 self._data_extractor.field_info().get_surfaces_info().keys()
-            )
+            ) + list(self._get_top_most_parent()._local_surfaces_provider())
 
     class scale(metaclass=PyLocalPropertyMeta):
         """Vector scale."""

ansys/fluent/post/pyvista/pyvista_objects.py

@@ -21,7 +21,7 @@ class Graphics:
 
     _sessions_state = {}
 
-    def __init__(self, session):
+    def __init__(self, session, local_surfaces_provider=None):
         """Instantiate Graphics, containter of graphics objects."""
         session_state = Graphics._sessions_state.get(
             session.id if session else 1
@@ -35,6 +35,13 @@ def __init__(self, session):
             self._init_module(self, sys.modules[__name__])
         else:
             self.__dict__ = session_state
+        self._local_surfaces_provider = (
+            lambda: local_surfaces_provider
+            if local_surfaces_provider
+            else self.Surfaces
+            if hasattr(self, "Surfaces")
+            else []
+        )
 
     def _init_module(self, obj, mod):
         for name, cls in mod.__dict__.items():
@@ -61,7 +68,9 @@ def display(self, window_id: Optional[str] = None):
         window_id : str, optional
             Window id. If not specified unique id is used.
         """
+        self._pre_display()
         pyvista_windows_manager.plot(self, window_id)
+        self._post_display()
 
 
 class Surface(SurfaceDefn):
@@ -76,7 +85,9 @@ def display(self, window_id: Optional[str] = None):
         window_id : str, optional
             Window id. If not specified unique id is used.
         """
+        self._pre_display()
         pyvista_windows_manager.plot(self, window_id)
+        self._post_display()
 
 
 class Contour(ContourDefn):
@@ -91,7 +102,9 @@ def display(self, window_id: Optional[str] = None):
         window_id : str, optional
             Window id. If not specified unique id is used.
         """
+        self._pre_display()
         pyvista_windows_manager.plot(self, window_id)
+        self._post_display()
 
 
 class Vector(VectorDefn):
@@ -106,4 +119,6 @@ def display(self, window_id: Optional[str] = None):
         window_id : str, optional
             Window id. If not specified unique id is used.
         """
+        self._pre_display()
         pyvista_windows_manager.plot(self, window_id)
+        self._post_display()

tests/test_post.py

@@ -231,8 +231,8 @@ def test_surface_object():
     surf1 = pyvista_graphics.Surfaces["surf-1"]
     field_info = surf1._data_extractor.field_info()
 
-    surf1.surface_type.surface_type = "iso-surface"
-    iso_surf = surf1.surface_type.iso_surface
+    surf1.surface.type = "iso-surface"
+    iso_surf = surf1.surface.iso_surface
 
     assert iso_surf.field.allowed_values == [
         v["solver_name"] for k, v in field_info.get_fields_info().items()
@@ -258,6 +258,19 @@ def test_surface_object():
     range = field_info.get_range(iso_surf.field(), True)
     assert range[0] == pytest.approx(iso_surf.iso_value())
 
+    cont1 = pyvista_graphics.Contours["surf-1"]
+    assert "surf-1" in cont1.surfaces_list.allowed_values
+
+    matplotlib_plots = Plots(session=None)
+    p1 = matplotlib_plots.XYPlots["p-1"]
+    assert "surf-1" not in p1.surfaces_list.allowed_values
+
+    local_surfaces_provider = Graphics(session=None).Surfaces
+    matplotlib_plots = Plots(
+        session=None, local_surfaces_provider=local_surfaces_provider
+    )
+    assert "surf-1" in p1.surfaces_list.allowed_values
+
 
 def test_create_plot_objects():
     matplotlib_plots1 = Plots(session=None)