ansys-pyensight-core 0.8.8__py3-none-any.whl → 0.8.10__py3-none-any.whl

ansys/pyensight/core/utils/omniverse_glb_server.py

@@ -0,0 +1,279 @@
+import json
+import logging
+import os
+import sys
+import time
+from typing import Any, Dict, List, Optional
+
+from ansys.api.pyensight.v0 import dynamic_scene_graph_pb2
+import pygltflib
+
+sys.path.insert(0, os.path.dirname(__file__))
+from dsg_server import Part, UpdateHandler  # noqa: E402
+
+
+class GLBSession(object):
+    def __init__(
+        self,
+        verbose: int = 0,
+        normalize_geometry: bool = False,
+        time_scale: float = 1.0,
+        handler: UpdateHandler = UpdateHandler(),
+    ):
+        """
+        Provide an interface to read a GLB file and link it to an UpdateHandler instance
+
+        This class reads GLB files and provides the data to an UpdateHandler instance for
+        further processing.
+
+        Parameters
+        ----------
+        verbose : int
+            The verbosity level.  If set to 1 or higher the class will call logging.info
+            for log output.  The default is ``0``.
+        normalize_geometry : bool
+            If True, the scene coordinates will be remapped into the volume [-1,-1,-1] - [1,1,1]
+            The default is not to remap coordinates.
+        time_scale : float
+            Scale time values by this factor after being read.  The default is ``1.0``.
+        handler : UpdateHandler
+            This is an UpdateHandler subclass that is called back when the state of
+            a scene transfer changes.  For example, methods are called when the
+            transfer begins or ends and when a Part (mesh block) is ready for processing.
+        """
+        super().__init__()
+        self._callback_handler = handler
+        self._verbose = verbose
+        self._normalize_geometry = normalize_geometry
+        self._time_scale = time_scale
+        self._time_limits = [
+            sys.float_info.max,
+            -sys.float_info.max,
+        ]  # Min/max across all time steps
+        self._mesh_block_count = 0
+        self._node_idx: int = -1
+        self._variables: Dict[int, Any] = dict()
+        self._groups: Dict[int, Any] = dict()
+        self._part: Part = Part(self)
+        self._scene_bounds: Optional[List] = None
+        self._cur_timeline: List = [0.0, 0.0]  # Start/End time for current update
+        self._callback_handler.session = self
+        # log any status changes to this file.  external apps will be monitoring
+        self._status_file = os.environ.get("ANSYS_OV_SERVER_STATUS_FILENAME", "")
+        self._status = dict(status="idle", start_time=0.0, processed_buffers=0, total_buffers=0)
+        self._gltf: pygltflib.GLTF2 = pygltflib.GLTF2()
+        self._id_num: int = 0
+
+    def _reset(self):
+        self._variables = dict()
+        self._groups = dict()
+        self._part = Part(self)
+        self._scene_bounds = None
+        self._cur_timeline = [0.0, 0.0]  # Start/End time for current update
+        self._status = dict(status="idle", start_time=0.0, processed_buffers=0, total_buffers=0)
+        self._gltf = pygltflib.GLTF2()
+        self._node_idx = -1
+        self._mesh_block_count = 0
+        self._id_num = 0
+
+    def _next_id(self) -> int:
+        self._id_num += 1
+        return self._id_num
+
+    @property
+    def scene_bounds(self) -> Optional[List]:
+        return self._scene_bounds
+
+    @property
+    def mesh_block_count(self) -> int:
+        return self._mesh_block_count
+
+    @property
+    def normalize_geometry(self) -> bool:
+        return self._normalize_geometry
+
+    @normalize_geometry.setter
+    def normalize_geometry(self, value: bool) -> None:
+        self._normalize_geometry = value
+
+    @property
+    def variables(self) -> dict:
+        return self._variables
+
+    @property
+    def groups(self) -> dict:
+        return self._groups
+
+    @property
+    def part(self) -> Part:
+        return self._part
+
+    @property
+    def time_limits(self) -> List:
+        return self._time_limits
+
+    @property
+    def cur_timeline(self) -> List:
+        return self._cur_timeline
+
+    @cur_timeline.setter
+    def cur_timeline(self, timeline: List) -> None:
+        self._cur_timeline = timeline
+        self._time_limits[0] = min(self._time_limits[0], self._cur_timeline[0])
+        self._time_limits[1] = max(self._time_limits[1], self._cur_timeline[1])
+
+    @property
+    def vrmode(self) -> bool:
+        """No camera support for the present."""
+        return True
+
+    def log(self, s: str, level: int = 0) -> None:
+        """Log a string to the logging system
+
+        If the message level is less than the current verbosity,
+        emit the message.
+        """
+        if level < self._verbose:
+            logging.info(s)
+
+    def _update_status_file(self, timed: bool = False):
+        """
+        Update the status file contents. The status file will contain the
+        following json object, stored as: self._status
+
+        {
+        'status' : "working|idle",
+        'start_time' : timestamp_of_update_begin,
+        'processed_buffers' : number_of_protobuffers_processed,
+        'total_buffers' : number_of_protobuffers_total,
+        }
+
+        Parameters
+        ----------
+        timed : bool, optional:
+            if True, only update every second.
+
+        """
+        if self._status_file:
+            current_time = time.time()
+            if timed:
+                last_time = self._status.get("last_time", 0.0)
+                if current_time - last_time < 1.0:  # type: ignore
+                    return
+            self._status["last_time"] = current_time
+            try:
+                message = json.dumps(self._status)
+                with open(self._status_file, "w") as status_file:
+                    status_file.write(message)
+            except IOError:
+                pass  # Note failure is expected here in some cases
+
+    def _parse_mesh(self, meshid: Any) -> None:
+        mesh = self._gltf.meshes[meshid]
+        logging.warning(f"mesh id: {meshid}, {mesh}")
+        for prim in mesh.primitives:
+            # TODO: GLB Prim -> DSG Part
+            self.log(f"prim {prim}")
+            # TODO: GLB Attributes -> DSG Geom
+
+        # mesh.mode, mesh.indices
+        # mesh.attributes(POSITION, NORMAL, COLOR_0, TEXCOORD_0, TEXCOORD_1)
+        # mesh.material
+        # mesh.images
+
+    def _walk_node(self, nodeid: Any) -> None:
+        node = self._gltf.nodes[nodeid]
+        self.log(f"node id: {nodeid}, {node}")
+        # TODO: GLB node -> DSG Group
+
+        if node.mesh is not None:
+            self._parse_mesh(node.mesh)
+
+        # Handle node.rotation, node.translation, node.scale, node.matrix
+        for child_id in node.children:
+            self._walk_node(child_id)
+
+    def upload_file(self, glb_filename: str) -> bool:
+        """Parse a GLB file and call out to the handler to present the data
+        to another interface (e.g. Omniverse)
+
+        Parameters
+        ----------
+        glb_filename : str
+            The name of the GLB file to parse
+
+        Returns
+        -------
+            bool:
+                returns True on success, False otherwise
+        """
+        try:
+            ok = True
+            self._gltf = pygltflib.GLTF2().load(glb_filename)
+            self.log(f"File: {glb_filename}  Info: {self._gltf.asset}")
+
+            self._callback_handler.begin_update()
+            self._update_status_file()
+
+            # TODO: Variables, Textures
+
+            # TODO: GLB Scene -> DSG View
+
+            # for present, just the default scene
+            for node_id in self._gltf.scenes[self._gltf.scene].nodes:
+                self._walk_node(node_id)
+
+            self._finish_part()
+            self._callback_handler.end_update()
+
+        except Exception as e:
+            self.log(f"Error: Unable to process: {glb_filename} : {e}")
+            ok = False
+
+        self._reset()
+        self._update_status_file()
+        return ok
+
+    def _finish_part(self) -> None:
+        """Complete the current part
+
+        There is always a part being modified.  This method completes the current part, committing
+        it to the handler.
+        """
+        self._callback_handler.finalize_part(self.part)
+        self._mesh_block_count += 1
+
+    def _name(self, node: Any) -> str:
+        if node.name:
+            return node.name
+        self._node_idx += 1
+        if self._node_idx == 0:
+            return "Root"
+        return f"Node_{self._node_idx}"
+
+    def _create_pb(
+        self, cmd_type: str, parent_id: int = -1, name: str = ""
+    ) -> "dynamic_scene_graph_pb2.SceneUpdateCommand":
+        cmd = dynamic_scene_graph_pb2.SceneUpdateCommand()
+        if cmd_type == "PART":
+            cmd.command_type = dynamic_scene_graph_pb2.SceneUpdateCommand.UPDATE_PART
+            subcmd = cmd.update_part
+        elif cmd_type == "GROUP":
+            cmd.command_type = dynamic_scene_graph_pb2.SceneUpdateCommand.UPDATE_GROUP
+            subcmd = cmd.update_group
+        elif cmd_type == "VARIABLE":
+            cmd.command_type = dynamic_scene_graph_pb2.SceneUpdateCommand.UPDATE_VARIABLE
+            subcmd = cmd.update_variable
+        elif cmd_type == "GEOM":
+            cmd.command_type = dynamic_scene_graph_pb2.SceneUpdateCommand.UPDATE_GEOM
+            subcmd = cmd.update_geom
+        elif cmd_type == "VIEW":
+            cmd.command_type = dynamic_scene_graph_pb2.SceneUpdateCommand.UPDATE_VIEW
+            subcmd = cmd.update_view
+        subcmd.id = self._next_id()
+        if parent_id >= 0:
+            subcmd.parent_id = parent_id
+        if cmd_type not in ("GEOM", "VIEW"):
+            if name:
+                subcmd.name = name
+        return cmd, subcmd

ansys/pyensight/core/utils/readers.py

@@ -10,8 +10,6 @@ from types import ModuleType
 from typing import Optional, Tuple, Union
 import uuid
 
-from ansys.pyensight.core import LocalLauncher
-
 try:
     import ensight
 except ImportError:
@@ -27,6 +25,7 @@ class Readers:
 
     @property
     def dvs(self) -> "DVS":
+        """The ensight interface"""
         return self._dvs
 
 
@@ -104,15 +103,20 @@ class DVS:
         """Find the dvs port allocated from the input temporary name"""
         if not tmp_name:
             raise RuntimeError("Temporary name for dvs port file not available")
-        is_local = isinstance(self._ensight._session._launcher, LocalLauncher)
-        if is_local:
-            with open(tmp_name) as dvs_port_file:
-                self._dvs_port = int(dvs_port_file.read().strip())
-        else:
-            log_content = self._ensight._session._launcher.enshell_log_contents()
-            dvs_port_match = re.search(r"\(0.0.0.0\):([0-9]{4,5})\n", log_content)
-            if dvs_port_match:
-                self._dvs_port = int(str(dvs_port_match.groups(1)[0]).strip())
+        try_local = True
+        if self._ensight._session._launcher:
+            if hasattr(self._ensight._session._launcher, "_enshell"):
+                try_local = False
+                log_content = self._ensight._session._launcher.enshell_log_contents()
+                dvs_port_match = re.search(r"\(0.0.0.0\):([0-9]{4,5})\n", log_content)
+                if dvs_port_match:
+                    self._dvs_port = int(str(dvs_port_match.groups(1)[0]).strip())
+        if try_local:
+            try:
+                with open(tmp_name) as dvs_port_file:
+                    self._dvs_port = int(dvs_port_file.read().strip())
+            except Exception:
+                raise RuntimeError("Cannot retrieve DVS Port")
 
     @staticmethod
     def _launch_dvs_callback_in_ensight(

{ansys_pyensight_core-0.8.8.dist-info → ansys_pyensight_core-0.8.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ansys-pyensight-core
-Version: 0.8.8
+Version: 0.8.10
 Summary: A python wrapper for Ansys EnSight
 Author-email: "ANSYS, Inc." <pyansys.core@ansys.com>
 Maintainer-email: "ANSYS, Inc." <pyansys.core@ansys.com>
@@ -16,7 +16,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: importlib-metadata>=4.0; python_version<='3.8'
-Requires-Dist: ansys-api-pyensight==0.4.1
+Requires-Dist: ansys-api-pyensight==0.4.2
 Requires-Dist: requests>=2.28.2
 Requires-Dist: docker>=6.1.0
 Requires-Dist: urllib3<3.0.0
@@ -24,29 +24,33 @@ Requires-Dist: numpy>=1.21.0,<2
 Requires-Dist: Pillow>=9.3.0
 Requires-Dist: pypng>=0.0.20
 Requires-Dist: psutil>=5.9.2
+Requires-Dist: usd-core>=24.8
+Requires-Dist: pygltflib>=1.16.2
 Requires-Dist: build>=0.10.0 ; extra == "dev"
 Requires-Dist: bump2version>=1.0.1 ; extra == "dev"
 Requires-Dist: ipdb>=0.9.4 ; extra == "dev"
 Requires-Dist: dill>=0.3.5.1 ; extra == "dev"
 Requires-Dist: pre-commit>=3.3.3 ; extra == "dev"
-Requires-Dist: Sphinx==7.2.6 ; extra == "doc"
+Requires-Dist: Sphinx==8.0.2 ; extra == "doc"
 Requires-Dist: numpydoc==1.5.0 ; extra == "doc"
 Requires-Dist: ansys-sphinx-theme==0.9.9 ; extra == "doc"
 Requires-Dist: sphinx-copybutton==0.5.2 ; extra == "doc"
 Requires-Dist: sphinx-gallery==0.13.0 ; extra == "doc"
 Requires-Dist: sphinxcontrib-mermaid==0.9.2 ; extra == "doc"
 Requires-Dist: docker>=6.1.0 ; extra == "doc"
-Requires-Dist: matplotlib==3.7.2 ; extra == "doc"
+Requires-Dist: matplotlib==3.9.1.post1 ; extra == "doc"
 Requires-Dist: requests>=2.28.2 ; extra == "doc"
 Requires-Dist: sphinxcontrib.jquery==4.1 ; extra == "doc"
 Requires-Dist: coverage-badge==1.1.0 ; extra == "doc"
 Requires-Dist: sphinxcontrib-openapi==0.8.1 ; extra == "doc"
 Requires-Dist: sphinxcontrib-video==0.2.0 ; extra == "doc"
-Requires-Dist: pytest==7.1.2 ; extra == "tests"
+Requires-Dist: usd-core>=24.8 ; extra == "doc"
+Requires-Dist: pygltflib>=1.16.2 ; extra == "doc"
+Requires-Dist: pytest==8.3.2 ; extra == "tests"
 Requires-Dist: pytest-cov==4.1.0 ; extra == "tests"
 Requires-Dist: dill>=0.3.5.1 ; extra == "tests"
 Requires-Dist: pytest-mock==3.10.0 ; extra == "tests"
-Requires-Dist: urllib3==1.26.10 ; extra == "tests"
+Requires-Dist: urllib3==2.2.2 ; extra == "tests"
 Requires-Dist: requests>=2.28.2 ; extra == "tests"
 Requires-Dist: docker>=6.1.0 ; extra == "tests"
 Project-URL: Changelog, https://github.com/ansys/pyensight/blob/main/CHANGELOG.rst

ansys_pyensight_core-0.8.10.dist-info/RECORD

@@ -0,0 +1,36 @@
+ansys/pyensight/core/__init__.py,sha256=yF8ER4bnq8oPCdC1iyO-0Po8zg1hnmkXIooOtyz2INQ,845
+ansys/pyensight/core/common.py,sha256=BgnRfE5NltIjVkI96U98zWv_u4QcLyeMetD59LKVAI4,7086
+ansys/pyensight/core/deep_pixel_view.html,sha256=oZmcCIS3Dqqsoj8oheYubLAH2ZVKXao3iJM2WXYBEKs,3421
+ansys/pyensight/core/dockerlauncher.py,sha256=LmgEFEjl0l3D_JaeJu5bbGR28aluHYtzb0-BmrC7ta8,28251
+ansys/pyensight/core/enscontext.py,sha256=GSKkjZt1QEPyHEQ59EEBgKGMik9vjCdR9coR4uX7fEw,12141
+ansys/pyensight/core/enshell_grpc.py,sha256=-OxSdFI_p3DmQnqh1jT_a_aSh_w-EUD2IaWGKxrnyjI,17122
+ansys/pyensight/core/ensight_grpc.py,sha256=BJaErleSPrtI8myTIh0m9awFWPaUxrQmHpbuyR3RpM4,16847
+ansys/pyensight/core/ensobj.py,sha256=uDtM2KHcAwd4hu5pcUYWbSD729ApHGIvuqZhEq8PxTI,18558
+ansys/pyensight/core/launch_ensight.py,sha256=iZJM6GdpzGRDLzrv1V2QZ5veIOpNSB5xPpJUFY7rBuo,10254
+ansys/pyensight/core/launcher.py,sha256=ymwfixwoHO7_c4qOetqccQbZjGT1HjeA7jwPi2JxlmE,10585
+ansys/pyensight/core/libuserd.py,sha256=nHHf_iI7eBQ6Z48LqTBDjw0gwz3hrFvqwsKm7kFqJY4,67912
+ansys/pyensight/core/listobj.py,sha256=Trw87IxIMXtmUd1DzywRmMzORU704AG4scX4fqYmO6M,9340
+ansys/pyensight/core/locallauncher.py,sha256=MnSyKVe0JGXN0YJ9f_msw_TBbLGfqDpjWWC4-haAjJ4,16205
+ansys/pyensight/core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ansys/pyensight/core/renderable.py,sha256=Z59F0u1Znot_MU_fT77F4CO7qbDxqDetFGzzgZcekCs,36440
+ansys/pyensight/core/session.py,sha256=jQA1sT7N3Q4iATH0vtzKf3v6ghg461LphZ2nUeyufCU,74587
+ansys/pyensight/core/sgeo_poll.html,sha256=1M4BIc5CZpYA3b40qzk22NcPCLhjFnWdoS2PrS6Rhn4,752
+ansys/pyensight/core/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ansys/pyensight/core/utils/adr.py,sha256=XslZhlwcrSGzOlnhzprOv3ju_ppxxsWBjCnQL5KiNms,3570
+ansys/pyensight/core/utils/dsg_server.py,sha256=zPvIHlgiJxiy0NQm3uIsh2nPY8TyAHaHaXtDMN28jiY,38869
+ansys/pyensight/core/utils/export.py,sha256=ZNpU3earAnRMBHZa6I8nTwMYY54WctXApJfMTkREBNA,23189
+ansys/pyensight/core/utils/omniverse.py,sha256=GWTYQhhVMT9x4kpMPpI51mHNiV1ths2zk8YzBZ2eeD0,14222
+ansys/pyensight/core/utils/omniverse_cli.py,sha256=aePr-2AWla2mMgPx8NlwIApAD1rAdVLmNOBWKjMXNQA,18786
+ansys/pyensight/core/utils/omniverse_dsg_server.py,sha256=SjKY8-q-M26tge13HW6Js71RoQnFgkjWELtHpXk707c,28185
+ansys/pyensight/core/utils/omniverse_glb_server.py,sha256=VYSbIwKMZASAmgYwRGbN5PrCjbSdaL7MaFxMnOJlTFg,9793
+ansys/pyensight/core/utils/parts.py,sha256=zST00r76kjsLLclBaKoOtuYhDUSdQr0vAooOG7AUea0,53372
+ansys/pyensight/core/utils/query.py,sha256=OXKDbf1sOTX0sUvtKcp64LhVl-BcrEsE43w8uMxLOYI,19828
+ansys/pyensight/core/utils/readers.py,sha256=cNNzrE5pjy4wpQKWAzIIJfJCSpY3HU43az02f5I3pVU,11968
+ansys/pyensight/core/utils/support.py,sha256=QI3z9ex7zJxjFbkCPba9DWqWgPFIThORqr0nvRfVjuc,4089
+ansys/pyensight/core/utils/variables.py,sha256=T96aL-qR2FtxH6QafeD9ddcWL9BB6IRSOYs2JauRTlg,95133
+ansys/pyensight/core/utils/views.py,sha256=ZKhJ6vMT7Rdd4bwJ0egMYTV7-D7Q7I19fF2_j_CMQ0o,12489
+ansys/pyensight/core/utils/resources/Materials/000_sky.exr,sha256=xAR1gFd2uxPZDnvgfegdhEhRaqKtZldQDiR_-1rHKO0,8819933
+ansys_pyensight_core-0.8.10.dist-info/LICENSE,sha256=qQWivZ12ETN5l3QxvTARY-QI5eoRRlyHdwLlAj0Bg5I,1089
+ansys_pyensight_core-0.8.10.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+ansys_pyensight_core-0.8.10.dist-info/METADATA,sha256=zQdULyTEURXiRUbAn_kaBYl2bB6NQYwwIhE_9PGZja4,11999
+ansys_pyensight_core-0.8.10.dist-info/RECORD,,

ansys/pyensight/core/exts/ansys.geometry.service/ansys/geometry/service/__init__.py

@@ -1 +0,0 @@
-from .extension import *