ansys-mechanical-core 0.11.4__py3-none-any.whl → 0.11.6__py3-none-any.whl

ansys/mechanical/core/embedding/app.py

@@ -254,6 +254,10 @@ class App:
         pip install ansys-mechanical-core[viz]
         """
         _plotter = self.plotter()
+
+        if _plotter is None:
+            return
+
         return _plotter.show()
 
     @property

ansys/mechanical/core/embedding/viz/embedding_plotter.py

@@ -72,6 +72,8 @@ def _get_nodes_and_coords(node: "Ansys.Mechanical.Scenegraph.Node"):
 def to_plotter(app: "ansys.mechanical.core.embedding.App"):
     """Convert the app's geometry to an ``ansys.tools.visualization_interface.Plotter`` instance."""
     plotter = Plotter()
+
+    # TODO - use get_scene from utils instead of looping over bodies directly here.
     for body in app.DataModel.GetObjectsByType(
         Ansys.Mechanical.DataModel.Enums.DataModelObjectCategory.Body
     ):

ansys/mechanical/core/embedding/viz/usd_converter.py

@@ -24,15 +24,9 @@
 
 import typing
 
-import clr
 from pxr import Gf, Usd, UsdGeom
 
-clr.AddReference("Ansys.Mechanical.DataModel")
-clr.AddReference("Ansys.ACT.Interfaces")
-
-import Ansys  # isort: skip
-
-from .utils import bgr_to_rgb_tuple, get_nodes_and_coords
+from .utils import bgr_to_rgb_tuple, get_nodes_and_coords, get_scene
 
 
 def _transform_to_rotation_translation(
@@ -94,6 +88,13 @@ def _convert_transform_node(
 
     Currently only supports transforms that contain a single tri tessellation node.
     """
+    import clr
+
+    clr.AddReference("Ansys.Mechanical.DataModel")
+    clr.AddReference("Ansys.ACT.Interfaces")
+
+    import Ansys  # isort: skip
+
     child_node = node.Child
     if isinstance(child_node, Ansys.Mechanical.Scenegraph.TriTessellationNode):
         prim = _create_prim_with_transform(stage, path, node)
@@ -101,19 +102,42 @@ def _convert_transform_node(
         _convert_tri_tessellation_node(child_node, stage, child_path, rgb)
 
 
-def to_usd_stage(app: "ansys.mechanical.core.embedding.App", name: str) -> None:
-    """Convert mechanical scene to usd stage."""
-    stage = Usd.Stage.CreateNew(name)
+def _convert_attribute_node(
+    node: "Ansys.Mechanical.Scenegraph.AttributeNode",
+    stage: Usd.Stage,
+    path: str,
+) -> None:
+    """Add a Usd Prim of the child node with the given attributes node.
 
+    Currently only supports color attributes.
+    """
+    import clr
+
+    clr.AddReference("Ansys.Mechanical.DataModel")
+    clr.AddReference("Ansys.ACT.Interfaces")
+
+    import Ansys  # isort: skip
+
+    child_node = node.Child
+    color = node.Property(Ansys.Mechanical.Scenegraph.ScenegraphIntAttributes.Color)
+    _convert_transform_node(child_node, stage, path, bgr_to_rgb_tuple(color))
+
+
+def load_into_usd_stage(scene: "Ansys.Mechanical.Scenegraph.GroupNode", stage: Usd.Stage) -> None:
+    """Load mechanical scene into usd stage `stage`."""
     root_prim = UsdGeom.Xform.Define(stage, "/root")
 
-    category = Ansys.Mechanical.DataModel.Enums.DataModelObjectCategory.Body
-    bodies = app.DataModel.GetObjectsByType(category)
-    for body in bodies:
-        scenegraph_node = Ansys.ACT.Mechanical.Tools.ScenegraphHelpers.GetScenegraph(body)
-        body_path = root_prim.GetPath().AppendPath(f"body{body.ObjectId}")
-        _convert_transform_node(scenegraph_node, stage, body_path, bgr_to_rgb_tuple(body.Color))
+    for child in scene.Children:
+        child: "Ansys.Mechanical.Scenegraph.AttributeNode" = child
+        child_path = root_prim.GetPath().AppendPath(child.Tag)
+        _convert_attribute_node(child, stage, child_path)
 
+
+def to_usd_stage(app: "ansys.mechanical.core.embedding.App", name: str) -> Usd.Stage:
+    """Convert mechanical scene to new usd stage and return it."""
+    stage = Usd.Stage.CreateNew(name)
+    scene = get_scene(app)
+    load_into_usd_stage(scene, stage)
     return stage
 
 

ansys/mechanical/core/embedding/viz/utils.py

@@ -59,3 +59,32 @@ def get_nodes_and_coords(tri_tessellation: "Ansys.Mechanical.Scenegraph.TriTesse
     )
     np_indices = _reshape_3cols(np.array(tri_tessellation.Indices, dtype=np.int32), "indices")
     return np_coordinates, np_indices
+
+
+def get_scene(
+    app: "ansys.mechanical.core.embedding.App",
+) -> "Ansys.Mechanical.Scenegraph.GroupNode":
+    """Get the scene of the model."""
+    import clr
+
+    clr.AddReference("Ansys.Mechanical.DataModel")
+    clr.AddReference("Ansys.Mechanical.Scenegraph")
+    clr.AddReference("Ansys.ACT.Interfaces")
+
+    import Ansys  # isort: skip
+
+    category = Ansys.Mechanical.DataModel.Enums.DataModelObjectCategory.Body
+    group_node = Ansys.Mechanical.Scenegraph.Builders.GroupNodeBuilder()
+    for body in app.DataModel.GetObjectsByType(category):
+        scenegraph_node = Ansys.ACT.Mechanical.Tools.ScenegraphHelpers.GetScenegraph(body)
+        # wrap the body node in an attribute node using the body color
+        attribute_node_builder = Ansys.Mechanical.Scenegraph.Builders.AttributeNodeBuilder()
+        attribute_node = (
+            attribute_node_builder.Tag(f"body{body.ObjectId}")
+            .Child(scenegraph_node)
+            # set the color, body.Color is a BGR uint bitfield
+            .Property(Ansys.Mechanical.Scenegraph.ScenegraphIntAttributes.Color, body.Color)
+            .Build()
+        )
+        group_node.AddChild(attribute_node)
+    return group_node.Build()

{ansys_mechanical_core-0.11.4.dist-info → ansys_mechanical_core-0.11.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ansys-mechanical-core
-Version: 0.11.4
+Version: 0.11.6
 Summary: A python wrapper for Ansys Mechanical
 Keywords: pymechanical,mechanical,ansys,pyansys
 Author-email: "ANSYS, Inc." <pyansys.core@ansys.com>
@@ -27,18 +27,18 @@ Requires-Dist: clr-loader==0.2.6
 Requires-Dist: grpcio>=1.30.0
 Requires-Dist: protobuf>=3.12.2,<6
 Requires-Dist: tqdm>=4.45.0
-Requires-Dist: sphinx==7.4.7 ; extra == "doc"
-Requires-Dist: ansys-sphinx-theme[autoapi]==0.16.6 ; extra == "doc"
-Requires-Dist: grpcio==1.65.4 ; extra == "doc"
+Requires-Dist: sphinx==8.0.2 ; extra == "doc"
+Requires-Dist: ansys-sphinx-theme[autoapi]==1.0.7 ; extra == "doc"
+Requires-Dist: grpcio==1.66.0 ; extra == "doc"
 Requires-Dist: imageio-ffmpeg==0.5.1 ; extra == "doc"
-Requires-Dist: imageio==2.34.2 ; extra == "doc"
+Requires-Dist: imageio==2.35.1 ; extra == "doc"
 Requires-Dist: jupyter_sphinx==0.5.3 ; extra == "doc"
 Requires-Dist: jupyterlab>=3.2.8 ; extra == "doc"
-Requires-Dist: matplotlib==3.9.1 ; extra == "doc"
-Requires-Dist: numpy==2.0.1 ; extra == "doc"
-Requires-Dist: numpydoc==1.7.0 ; extra == "doc"
+Requires-Dist: matplotlib==3.9.2 ; extra == "doc"
+Requires-Dist: numpy==2.1.0 ; extra == "doc"
+Requires-Dist: numpydoc==1.8.0 ; extra == "doc"
 Requires-Dist: pandas==2.2.2 ; extra == "doc"
-Requires-Dist: panel==1.4.4 ; extra == "doc"
+Requires-Dist: panel==1.4.5 ; extra == "doc"
 Requires-Dist: plotly==5.23.0 ; extra == "doc"
 Requires-Dist: pypandoc==1.13 ; extra == "doc"
 Requires-Dist: pytest-sphinx==0.6.3 ; extra == "doc"
@@ -47,9 +47,9 @@ Requires-Dist: pyvista>=0.39.1 ; extra == "doc"
 Requires-Dist: sphinx-autobuild==2024.4.16 ; extra == "doc"
 Requires-Dist: sphinx-autodoc-typehints==2.2.3 ; extra == "doc"
 Requires-Dist: sphinx-copybutton==0.5.2 ; extra == "doc"
-Requires-Dist: sphinx_design==0.6.0 ; extra == "doc"
-Requires-Dist: sphinx-gallery==0.17.0 ; extra == "doc"
-Requires-Dist: sphinx-notfound-page==1.0.2 ; extra == "doc"
+Requires-Dist: sphinx_design==0.6.1 ; extra == "doc"
+Requires-Dist: sphinx-gallery==0.17.1 ; extra == "doc"
+Requires-Dist: sphinx-notfound-page==1.0.4 ; extra == "doc"
 Requires-Dist: sphinxcontrib-websupport==2.0.0 ; extra == "doc"
 Requires-Dist: sphinxemoji==0.3.1 ; extra == "doc"
 Requires-Dist: pytest==8.3.2 ; extra == "tests"
@@ -66,8 +66,9 @@ Provides-Extra: doc
 Provides-Extra: tests
 Provides-Extra: viz
 
-.. image:: https://raw.githubusercontent.com/ansys/pymechanical/main/doc/source/_static/logo/pymechanical-logo-light.png
+.. image:: https://raw.githubusercontent.com/ansys/pymechanical/main/doc/source/_static/logo/pymechanical-logo.png
    :alt: PyMechanical logo
+   :width: 580px
 
 
 |pyansys| |pypi| |python| |GH-CI| |codecov| |MIT| |black|
@@ -117,7 +118,7 @@ Mechanical within Python's ecosystem. It includes the ability to:
 Install the package
 -------------------
 
-Install PyMechanical using `pip` with::
+Install PyMechanical using ``pip`` with::
 
    pip install ansys-mechanical-core
 

{ansys_mechanical_core-0.11.4.dist-info → ansys_mechanical_core-0.11.6.dist-info}/RECORD

@@ -10,7 +10,7 @@ ansys/mechanical/core/pool.py,sha256=HzU64Ew5QgoKeoZkywdmS_lmuGSCwe-PvucI5y1Hqa0
 ansys/mechanical/core/run.py,sha256=ScB-Ne8KkLuA0EQtquUa8JKFsA8Ert8nUUtxHTlQR4A,9814
 ansys/mechanical/core/embedding/__init__.py,sha256=y0yp3dnBW2oj9Jh_L_qfZstAbpON974EMmpV9w3kT3g,1356
 ansys/mechanical/core/embedding/addins.py,sha256=yRG8CT1f1MgQj0y_mP3ISePVp7t9erw0o5Sbn1tFNkc,2370
-ansys/mechanical/core/embedding/app.py,sha256=KFYs4MEvxrzCP6IRBPeeZyGtVzVVochVfqWT5bS1eWI,16145
+ansys/mechanical/core/embedding/app.py,sha256=hnP-eAFFekfHPuHxHc7wJraDa1DO7s2J65dWtKzRJ9U,16195
 ansys/mechanical/core/embedding/app_libraries.py,sha256=RiTO23AzjssAylIH2DaTa6mcJmxhfrlHW-yYvHpIkt0,2923
 ansys/mechanical/core/embedding/appdata.py,sha256=krcmcgHhraHIlORFr43QvUXlAiXg231g_2iOIxkW_aQ,4223
 ansys/mechanical/core/embedding/background.py,sha256=ICm87gO6CDA4Ot_6Yf3-8YPNUEa3PHWieOECXeUfZmM,3650
@@ -30,13 +30,13 @@ ansys/mechanical/core/embedding/logger/linux_api.py,sha256=VrsO9F4NwHFWYJJE6F7bi
 ansys/mechanical/core/embedding/logger/sinks.py,sha256=FYmV2iWt-7KuGw40E5WcNm4AnDsWqw6aQKSMKlivYoo,1392
 ansys/mechanical/core/embedding/logger/windows_api.py,sha256=OCJ-SJEY7EjigZiW6H5qufQ39N_mL7sXeoivOVl9FHc,5248
 ansys/mechanical/core/embedding/viz/__init__.py,sha256=xgpBdf3yfEq3sn0bNewLwtje-SCH6vVWEmHfCdh6078,1206
-ansys/mechanical/core/embedding/viz/embedding_plotter.py,sha256=JgitPNYanqUImqT1gEtaUIDGmPk99mhuoi0ZTAkuoQ8,3604
-ansys/mechanical/core/embedding/viz/usd_converter.py,sha256=5fjbUq9-nNtKBF5e7Z3Yxgs7jzmTUayWcRANrsIHM-A,4657
-ansys/mechanical/core/embedding/viz/utils.py,sha256=pIROh87BI_OxJtkM8To9KYulpygdQ6I3g1f1BJQ-SBE,2441
+ansys/mechanical/core/embedding/viz/embedding_plotter.py,sha256=ausbFhezwmLCGhu61JZJDM_uxwpRRuM-XWw9mk4i0GQ,3689
+ansys/mechanical/core/embedding/viz/usd_converter.py,sha256=feDq2KrZhYL-RR1miECQL-y0VNDhnZQ9Wke5UOkYmp4,5329
+ansys/mechanical/core/embedding/viz/utils.py,sha256=NWc18HLshIpO7wgZpwfduKG5wReaYtVtJPChOosKXLU,3638
 ansys/mechanical/core/examples/__init__.py,sha256=A1iS8nknTU1ylafHZpYC9LQJ0sY83x8m1cDXsgvFOBo,1267
 ansys/mechanical/core/examples/downloads.py,sha256=lJ2SrX9Qs0t2FSmMwUBB4UgSpk6Qb2-coY1PLlzPzHU,4144
-ansys_mechanical_core-0.11.4.dist-info/entry_points.txt,sha256=wRXPv0eRXSmodweJHjF_Y3x5wmeSmjWcY3Bo6B4tAiM,66
-ansys_mechanical_core-0.11.4.dist-info/LICENSE,sha256=gBJ2GQ6oDJwAWxcxmjx_0uXc-N0P4sHhA7BXsdPTfco,1098
-ansys_mechanical_core-0.11.4.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
-ansys_mechanical_core-0.11.4.dist-info/METADATA,sha256=9LG4h1VMmOL6W9xeLI8eZ4CzeP-7oqbM6lJSlrskS5Q,9771
-ansys_mechanical_core-0.11.4.dist-info/RECORD,,
+ansys_mechanical_core-0.11.6.dist-info/entry_points.txt,sha256=wRXPv0eRXSmodweJHjF_Y3x5wmeSmjWcY3Bo6B4tAiM,66
+ansys_mechanical_core-0.11.6.dist-info/LICENSE,sha256=gBJ2GQ6oDJwAWxcxmjx_0uXc-N0P4sHhA7BXsdPTfco,1098
+ansys_mechanical_core-0.11.6.dist-info/WHEEL,sha256=EZbGkh7Ie4PoZfRQ8I0ZuP9VklN_TvcZ6DSE5Uar4z4,81
+ansys_mechanical_core-0.11.6.dist-info/METADATA,sha256=lhSKf_O6fNcwxZ-WushtsxuXTLExAud-2OLKm04q-L0,9783
+ansys_mechanical_core-0.11.6.dist-info/RECORD,,