ansys-pyensight-core 0.10.9__tar.gz → 0.10.10__tar.gz

{ansys_pyensight_core-0.10.9 → ansys_pyensight_core-0.10.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ansys-pyensight-core
-Version: 0.10.9
+Version: 0.10.10
 Summary: A python wrapper for Ansys EnSight
 Author-email: "ANSYS, Inc." <pyansys.core@ansys.com>
 Maintainer-email: "ANSYS, Inc." <pyansys.core@ansys.com>

{ansys_pyensight_core-0.10.9 → ansys_pyensight_core-0.10.10}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "ansys-pyensight-core"
-version = "0.10.9"
+version = "0.10.10"
 description = "A python wrapper for Ansys EnSight"
 readme = "README.rst"
 requires-python = ">=3.10,<3.14"

{ansys_pyensight_core-0.10.9 → ansys_pyensight_core-0.10.10}/src/ansys/pyensight/core/utils/views.py

@@ -94,8 +94,7 @@ class _Simba:
             self.views.set_view_direction(
                 1, 1, 1, perspective=self.ensight.objs.core.vports[0].PERSPECTIVE
             )
-        self.auto_scale()
-        return self.get_camera()
+        return self.auto_scale()
 
     def get_camera(self):
         """Get EnSight camera settings in VTK format."""
@@ -208,8 +207,49 @@ class _Simba:
         parallel_scale = 1 / data[9]
         return camera_position, focal_point, self.views._normalize_vector(view_up), parallel_scale
 
+    def get_camera_axes(self):
+        """
+        Returns the camera's local axes: right, up, and forward vectors.
+        These are useful for applying transformations in view space.
+
+        Parameters:
+            camera (dict): A dictionary with keys 'position', 'focal_point', and 'view_up'.
+
+        Returns:
+            right (np.ndarray): Right vector (X axis in view space).
+            up (np.ndarray): Up vector (Y axis in view space).
+        forw    ard (np.ndarray): Forward vector (Z axis in view space, pointing from position to focal point).
+        """
+        camera = self.get_camera()
+        position = np.array(camera["position"])
+        focal_point = np.array(camera["focal_point"])
+        view_up = np.array(camera["view_up"])
+
+        # Forward vector: from camera position to focal point
+        forward = focal_point - position
+        forward /= np.linalg.norm(forward)
+
+        # Right vector: cross product of forward and view_up
+        right = np.cross(forward, view_up)
+        right /= np.linalg.norm(right)
+
+        # Recompute up vector to ensure orthogonality
+        up = np.cross(right, forward)
+        up /= np.linalg.norm(up)
+
+        return right, up, forward
+
     def set_camera(
-        self, orthographic, view_up=None, position=None, focal_point=None, view_angle=None, pan=None
+        self,
+        orthographic,
+        view_up=None,
+        position=None,
+        focal_point=None,
+        view_angle=None,
+        pan=None,
+        mousex=None,
+        mousey=None,
+        invert_y=False,
     ):
         """Set the EnSight camera settings from the VTK input."""
         self.ensight.view_transf.function("global")
@@ -219,7 +259,6 @@ class _Simba:
         vport = self.ensight.objs.core.VPORTS[0]
         if view_angle:
             vport.PERSPECTIVEANGLE = view_angle / 2
-
         if view_up and position and focal_point:
             if not pan:
                 q_current = self.normalize(np.array(vport.ROTATION.copy()))
@@ -232,7 +271,17 @@ class _Simba:
                 angles = self.quaternion_to_euler(q_relative)
                 self.ensight.view_transf.rotate(*angles)
             else:
-                self.ensight.view_transf.translate(*pan)
+                if mousex and mousey:
+                    self.screen_to_world(
+                        mousex=mousex, mousey=mousey, invert_y=invert_y, set_center=True
+                    )
+                current_camera = self.get_camera()
+                right, up, _ = self.get_camera_axes()
+                translation_vector = np.array(position) - np.array(current_camera["position"])
+                dx = np.dot(translation_vector, right)
+                dy = np.dot(translation_vector, up)
+                self.ensight.view_transf.translate(-dx, -dy, 0)
+
         self.render()
 
     def set_perspective(self, value):
@@ -265,7 +314,37 @@ class _Simba:
         self.ensight.render()
         self.ensight.refresh(1)
 
-    def drag_allowed(self, mousex, mousey, invert_y=False):
+    def _probe_setup(self, part_obj, get_probe_data=False):
+        self.ensight.query_interact.number_displayed(100)
+        self.ensight.query_interact.query("surface")
+        self.ensight.query_interact.display_id("OFF")
+        self.ensight.query_interact.label_always_on_top("ON")
+        self.ensight.query_interact.marker_size_normalized(2)
+        if get_probe_data:
+            variable_string = """Coordinates 'X' 'Y' 'Z'"""
+            variable_list = [variable_string]
+            variable_name = part_obj.COLORBYPALETTE
+            if variable_name:
+                if isinstance(variable_name, str):
+                    variable_list.append(variable_name)
+                else:
+                    if isinstance(variable_name, list):
+                        if variable_name[0]:
+                            variable_name = variable_name[0].DESCRIPTION
+                            variable_list.append(variable_name)
+                        else:
+                            variable_name = None
+            if isinstance(self.ensight, ModuleType):
+                self.ensight.query_interact.select_varname_begin(*variable_list)
+            else:
+                command = "ensight.query_interact.select_varname_begin("
+                for var in variable_list:
+                    command += var + ","
+                command = command[:-1] + ")"
+                self.ensight._session.cmd(command)
+        self.render()
+
+    def drag_allowed(self, mousex, mousey, invert_y=False, probe=False, get_probe_data=False):
         """Return True if the picked object is allowed dragging in the interactor."""
         mousex = int(mousex)
         mousey = int(mousey)
@@ -277,8 +356,14 @@ class _Simba:
             part_id, tool_id = self.ensight._session.cmd(
                 f"ensight.objs.core.VPORTS[0].simba_what_is_picked({mousex}, {mousey}, {invert_y})"
             )
+        coords = [None, None, None]
+        if probe:
+            screen_to_world = self.screen_to_world(
+                mousex=mousex, mousey=mousey, invert_y=invert_y, set_center=False
+            )
+            coords = screen_to_world["model_point"]
         if tool_id > -1:
-            return True
+            return True, coords[0], coords[1], coords[2], False
         part_types_allowed = [
             self.ensight.objs.enums.PART_CLIP_PLANE,
             self.ensight.objs.enums.PART_ISO_SURFACE,
@@ -286,8 +371,22 @@ class _Simba:
         ]
         if part_id > -1:
             part_obj = self.ensight.objs.core.PARTS.find(part_id, "PARTNUMBER")[0]
-            return part_obj.PARTTYPE in part_types_allowed
-        return False
+            if probe:
+                width, height = tuple(self.ensight.objs.core.WINDOWSIZE)
+                if invert_y:
+                    mousey = height - mousey
+                self.ensight.query_interact.number_displayed(100)
+                self.ensight.query_interact.query("surface")
+                self.ensight.query_interact.display_id("OFF")
+                self.ensight.query_interact.create(mousex / width, mousey / height)
+                self._probe_setup(part_obj, get_probe_data=get_probe_data)
+            return part_obj.PARTTYPE in part_types_allowed, coords[0], coords[1], coords[2], True
+        if (
+            get_probe_data and self.ensight.objs.core.PROBES[0].PROBE_DATA
+        ):  # In case we have picked a probe point
+            for part in self.ensight.objs.core.PARTS:
+                self._probe_setup(part, get_probe_data=get_probe_data)
+        return False, coords[0], coords[1], coords[2], False
 
 
 class Views: