solidipes-solid-mech-plugin 0.0.14__tar.gz → 0.0.16__tar.gz

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: solidipes-solid-mech-plugin
-Version: 0.0.14
+Version: 0.0.16
 Summary: Plugin for Solidipes with solid mechanics components
 Author-email: Son Pham-Ba <son.phamba@epfl.ch>, Guillaume Anciaux <guillaume.anciaux@epfl.ch>
 License: GPL-3.0-or-later
@@ -12,9 +12,12 @@ Requires-Dist: ipython~=8.18.1
 Requires-Dist: mergedeep~=1.3.4
 Requires-Dist: meshio[all]~=5.3.5
 Requires-Dist: streamlit~=1.45.0
-Requires-Dist: trame~=3.5.2
-Requires-Dist: streamlit-pyvista==0.0.20rc0
+Requires-Dist: trame<4,>=3.13.2
+Requires-Dist: trame-pyvista>=0.2.1
+Requires-Dist: trame-vtk>=2.11.15
+Requires-Dist: streamlit-pyvista~=0.0.24
 Requires-Dist: h5py~=3.13.0
+Requires-Dist: netCDF4~=1.7.0
 Provides-Extra: dev
 Requires-Dist: pillow~=10.4.0; extra == "dev"
 Requires-Dist: pixelmatch~=0.3.0; extra == "dev"

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/pyproject.toml

@@ -16,9 +16,12 @@ dependencies = [
 	"mergedeep~=1.3.4",
 	"meshio[all]~=5.3.5",
 	"streamlit~=1.45.0",
-	"trame~=3.5.2",
-	"streamlit-pyvista==0.0.20rc0",
+	"trame>=3.13.2,<4",
+	"trame-pyvista>=0.2.1",
+	"trame-vtk>=2.11.15",
+	"streamlit-pyvista~=0.0.24",
 	"h5py~=3.13.0",
+	"netCDF4~=1.7.0",
 ]
 
 [project.optional-dependencies]

solidipes_solid_mech_plugin-0.0.16/solidipes_solid_mech_plugin/config.py

@@ -0,0 +1,8 @@
+import os
+
+from solidipes.utils import config
+from solidipes.utils.utils import get_read_write_dir
+
+streamlit_pyvista_cache_dir_name = get_read_write_dir(
+    os.path.join(config.solidipes_dirname, ".streamlit-pyvista-cache")
+)

solidipes_solid_mech_plugin-0.0.16/solidipes_solid_mech_plugin/loaders/exodus.py

@@ -0,0 +1,184 @@
+import re
+
+import numpy as np
+from solidipes.loaders.file import File
+from solidipes.loaders.sequence import Sequence
+from solidipes_core_plugin.loaders.netcdf import NetCDF
+
+
+def _decode_exodus_string(raw) -> str:
+    """Decode a character array row from an Exodus II netCDF file."""
+    if isinstance(raw, np.ndarray):
+        raw = raw.tobytes()
+
+    if isinstance(raw, (bytes, np.bytes_)):
+        return raw.decode("utf-8", errors="ignore").strip("\x00").strip()
+
+    return str(raw)
+
+
+class Exodus(Sequence, NetCDF):
+    """Exodus II finite element database file, read with meshio and netCDF4.
+
+    Every Exodus file is treated as a sequence of time steps (a mesh-only
+    database counts as a single frame). Nodal and elemental result variables
+    are exposed as point and cell data for each available time step, and the
+    mesh can be viewed with the PyvistaPlotter.
+
+    The `netcdf` loadable and the netCDF metadata (dimensions, variables,
+    global attributes) are inherited from the NetCDF loader.
+    """
+
+    from ..viewers.pyvista_plotter import PyvistaPlotter
+
+    supported_mime_types = {"application/x-netcdf": ["e", "exo", "gen"]}
+    _compatible_viewers = [PyvistaPlotter]
+
+    @File.loadable
+    def meshio_mesh(self):
+        """Static mesh (connectivity and coordinates) read with meshio"""
+        import meshio
+
+        return meshio.read(self.file_info.path)
+
+    @File.rocrate_metadata
+    def title(self):
+        return self.netcdf.title
+
+    @File.rocrate_metadata
+    def n_time_steps(self):
+        if "time_step" not in self.netcdf.dimensions:
+            return 1
+
+        # A mesh-only database has an empty (or missing) time_step dimension:
+        # the mesh itself counts as a single frame
+        return max(1, len(self.netcdf.dimensions["time_step"]))
+
+    @File.rocrate_metadata
+    def n_nodes(self):
+        return len(self.netcdf.dimensions["num_nodes"])
+
+    @File.rocrate_metadata
+    def n_elements(self):
+        return len(self.netcdf.dimensions["num_elem"])
+
+    @File.rocrate_metadata
+    def nodal_variable_names(self):
+        if "name_nod_var" not in self.netcdf.variables:
+            return []
+
+        return [_decode_exodus_string(raw) for raw in self.netcdf.variables["name_nod_var"][:]]
+
+    @File.rocrate_metadata
+    def element_block_names(self):
+        if "eb_names" not in self.netcdf.variables:
+            return {}
+
+        return {i + 1: _decode_exodus_string(raw) for i, raw in enumerate(self.netcdf.variables["eb_names"][:])}
+
+    @File.rocrate_metadata
+    def elemental_variable_names(self):
+        """List of (element_block_number, variable_number, name) triples"""
+        names = []
+
+        for key, var in self.netcdf.variables.items():
+            if not key.startswith("name_elem_var"):
+                continue
+
+            block = int(key[len("name_elem_var") :]) if len(key) > len("name_elem_var") else 1
+
+            for i, raw in enumerate(var[:]):
+                name = _decode_exodus_string(raw)
+
+                if name:
+                    names.append((block, i + 1, name))
+
+        return names
+
+    @property
+    def _element_count(self):
+        return self.n_time_steps
+
+    def select_frame(self, frame):
+        self.select_element(frame)
+
+    @property
+    def mesh(self):
+        """Pyvista mesh of the currently selected time step"""
+        return self._current_element
+
+    def _load_element(self, n):
+        """Load the mesh with nodal and elemental data at time step n"""
+        import pyvista as pv
+        from meshio import Mesh
+
+        meshio_mesh = self.meshio_mesh
+        point_data = self._point_data_at_step(n)
+        cell_data = self._cell_data_at_step(n)
+        mesh = Mesh(meshio_mesh.points, meshio_mesh.cells, point_data=point_data, cell_data=cell_data)
+        return pv.from_meshio(mesh)
+
+    def _point_data_at_step(self, n):
+        """Nodal variable values at time step n"""
+        nc = self.netcdf
+        names = self.nodal_variable_names
+        data = {}
+
+        if "vals_nod_var" in nc.variables:
+            # Standard layout: single variable with dimensions
+            # (time_step, num_nod_var, num_nodes)
+            all_data = np.asarray(nc.variables["vals_nod_var"][n])
+
+            for i, name in enumerate(names):
+                data[name] = all_data[i]
+
+        else:
+            # Alternative layout: one variable per nodal variable
+            for i, name in enumerate(names):
+                var_name = f"vals_nod_var{i + 1}"
+
+                if var_name in nc.variables:
+                    data[name] = np.asarray(nc.variables[var_name][n])
+
+        return data
+
+    def _cell_data_at_step(self, n):
+        """Elemental variable values at time step n, per meshio cell block"""
+        nc = self.netcdf
+        names = self.elemental_variable_names
+
+        if len(names) == 0:
+            return {}
+
+        # Collect data per (variable index, element block)
+        raw: dict[int, dict[int, np.ndarray]] = {}
+
+        for key, var in nc.variables.items():
+            match = re.fullmatch(r"vals_elem_var(\d+)?(?:eb(\d+))?", key)
+
+            if match is None:
+                continue
+
+            var_index = 0 if match.group(1) is None else int(match.group(1)) - 1
+            block = 0 if match.group(2) is None else int(match.group(2)) - 1
+            raw.setdefault(var_index, {})[block] = np.asarray(var[n])
+
+        cell_sizes = [len(c.data) for c in self.meshio_mesh.cells]
+        offsets = np.cumsum([0, *cell_sizes])
+
+        cell_data = {}
+
+        for block, var_index, name in names:
+            if var_index - 1 not in raw:
+                continue
+
+            by_block = raw[var_index - 1]
+            concatenated = np.concatenate([by_block[b] for b in sorted(by_block)])
+
+            if len(concatenated) != offsets[-1]:
+                # Element blocks without data for this variable: skip
+                continue
+
+            cell_data[name] = [concatenated[start:stop] for start, stop in zip(offsets[:-1], offsets[1:])]
+
+        return cell_data

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/solidipes_solid_mech_plugin/viewers/pyvista_plotter.py

@@ -95,7 +95,7 @@ class PyvistaPlotter(Viewer):
                 method
         """
         from streamlit_pyvista.mesh_viewer_component import MeshViewerComponent  # noqa: E402
-        from streamlit_pyvista.server_managers import ServerManagerProxified  # noqa: E402
+        from streamlit_pyvista.server_managers import ServerManager  # noqa: E402
         from streamlit_pyvista.trame_viewers import get_advanced_viewer_path  # noqa: E402
 
         max_frames = 500
@@ -133,7 +133,7 @@ class PyvistaPlotter(Viewer):
             MeshViewerComponent(
                 paths,
                 trame_viewer_class=get_advanced_viewer_path(),
-                server_manager_class=ServerManagerProxified,
+                server_manager_class=ServerManager,
             ).show()
         elif viewer_backends.current_backend == "python":
             self.shown = True
@@ -143,7 +143,7 @@ class PyvistaPlotter(Viewer):
             MeshViewerComponent(
                 paths,
                 trame_viewer_class=get_advanced_viewer_path(),
-                server_manager_class=ServerManagerProxified,
+                server_manager_class=ServerManager,
             ).show()
 
     def save(self, path, **kwargs):

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/solidipes_solid_mech_plugin.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: solidipes-solid-mech-plugin
-Version: 0.0.14
+Version: 0.0.16
 Summary: Plugin for Solidipes with solid mechanics components
 Author-email: Son Pham-Ba <son.phamba@epfl.ch>, Guillaume Anciaux <guillaume.anciaux@epfl.ch>
 License: GPL-3.0-or-later
@@ -12,9 +12,12 @@ Requires-Dist: ipython~=8.18.1
 Requires-Dist: mergedeep~=1.3.4
 Requires-Dist: meshio[all]~=5.3.5
 Requires-Dist: streamlit~=1.45.0
-Requires-Dist: trame~=3.5.2
-Requires-Dist: streamlit-pyvista==0.0.20rc0
+Requires-Dist: trame<4,>=3.13.2
+Requires-Dist: trame-pyvista>=0.2.1
+Requires-Dist: trame-vtk>=2.11.15
+Requires-Dist: streamlit-pyvista~=0.0.24
 Requires-Dist: h5py~=3.13.0
+Requires-Dist: netCDF4~=1.7.0
 Provides-Extra: dev
 Requires-Dist: pillow~=10.4.0; extra == "dev"
 Requires-Dist: pixelmatch~=0.3.0; extra == "dev"

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/solidipes_solid_mech_plugin.egg-info/SOURCES.txt

@@ -20,6 +20,7 @@ solidipes_solid_mech_plugin.egg-info/top_level.txt
 solidipes_solid_mech_plugin/loaders/__init__.py
 solidipes_solid_mech_plugin/loaders/abaqus.py
 solidipes_solid_mech_plugin/loaders/abaqus_python_code.py
+solidipes_solid_mech_plugin/loaders/exodus.py
 solidipes_solid_mech_plugin/loaders/geo_msh.py
 solidipes_solid_mech_plugin/loaders/geof_mesh.py
 solidipes_solid_mech_plugin/loaders/meshio.py
@@ -31,6 +32,7 @@ solidipes_solid_mech_plugin/viewers/__init__.py
 solidipes_solid_mech_plugin/viewers/pyvista_plotter.py
 solidipes_solid_mech_plugin/viewers/xdmf.py
 tests/conftest.py
+tests/test_exodus_mesh.py
 tests/test_load_modules.py
 tests/test_ontologies.py
 tests/test_pyvista_mesh.py

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/solidipes_solid_mech_plugin.egg-info/requires.txt

@@ -3,9 +3,12 @@ ipython~=8.18.1
 mergedeep~=1.3.4
 meshio[all]~=5.3.5
 streamlit~=1.45.0
-trame~=3.5.2
-streamlit-pyvista==0.0.20rc0
+trame<4,>=3.13.2
+trame-pyvista>=0.2.1
+trame-vtk>=2.11.15
+streamlit-pyvista~=0.0.24
 h5py~=3.13.0
+netCDF4~=1.7.0
 
 [dev]
 pillow~=10.4.0

{solidipes_solid_mech_plugin-0.0.14 → solidipes_solid_mech_plugin-0.0.16}/solidipes_solid_mech_plugin.egg-info/scm_file_list.json

@@ -1,43 +1,45 @@
 {
   "files": [
+    ".flake8",
     ".gitignore",
     ".gitlab-ci.yml",
     ".pre-commit-config.yaml",
     "CHANGELOG.md",
-    ".flake8",
+    "LICENSE",
     "README.md",
-    "uv.lock",
     "pyproject.toml",
-    "LICENSE",
-    "tests/test_pyvista_mesh.py",
-    "tests/conftest.py",
-    "tests/test_ontologies.py",
-    "tests/test_load_modules.py",
-    "tests/test_streamlit_pyvista.py",
-    "tests/test_pyvista_plotter.py",
-    "tests/test_xdmf_mesh.py",
-    "tests/helpers/utils.py",
-    "tests/assets/xdmf_mesh.h5",
-    "tests/assets/test_mesh_loaded.png",
-    "tests/assets/pyvista_mesh.vtu",
-    "tests/assets/generate_assets.py",
-    "tests/assets/pyvista_plot.png",
-    "tests/assets/pyvista_mesh_no_fields.vtu",
-    "tests/assets/xdmf_mesh.xdmf",
     "solidipes_solid_mech_plugin/__init__.py",
     "solidipes_solid_mech_plugin/config.py",
-    "solidipes_solid_mech_plugin/ontologies/solidipes_solid_mech.py",
-    "solidipes_solid_mech_plugin/loaders/parse_inp.py",
+    "solidipes_solid_mech_plugin/loaders/__init__.py",
     "solidipes_solid_mech_plugin/loaders/abaqus.py",
+    "solidipes_solid_mech_plugin/loaders/abaqus_python_code.py",
+    "solidipes_solid_mech_plugin/loaders/exodus.py",
     "solidipes_solid_mech_plugin/loaders/geo_msh.py",
-    "solidipes_solid_mech_plugin/loaders/pyvista_mesh.py",
-    "solidipes_solid_mech_plugin/loaders/__init__.py",
     "solidipes_solid_mech_plugin/loaders/geof_mesh.py",
-    "solidipes_solid_mech_plugin/loaders/xdmf.py",
     "solidipes_solid_mech_plugin/loaders/meshio.py",
-    "solidipes_solid_mech_plugin/loaders/abaqus_python_code.py",
+    "solidipes_solid_mech_plugin/loaders/parse_inp.py",
+    "solidipes_solid_mech_plugin/loaders/pyvista_mesh.py",
+    "solidipes_solid_mech_plugin/loaders/xdmf.py",
+    "solidipes_solid_mech_plugin/ontologies/solidipes_solid_mech.py",
     "solidipes_solid_mech_plugin/viewers/__init__.py",
+    "solidipes_solid_mech_plugin/viewers/pyvista_plotter.py",
     "solidipes_solid_mech_plugin/viewers/xdmf.py",
-    "solidipes_solid_mech_plugin/viewers/pyvista_plotter.py"
+    "tests/assets/generate_assets.py",
+    "tests/assets/pyvista_mesh.vtu",
+    "tests/assets/pyvista_mesh_no_fields.vtu",
+    "tests/assets/pyvista_plot.png",
+    "tests/assets/test_mesh_loaded.png",
+    "tests/assets/xdmf_mesh.h5",
+    "tests/assets/xdmf_mesh.xdmf",
+    "tests/conftest.py",
+    "tests/helpers/utils.py",
+    "tests/test_exodus_mesh.py",
+    "tests/test_load_modules.py",
+    "tests/test_ontologies.py",
+    "tests/test_pyvista_mesh.py",
+    "tests/test_pyvista_plotter.py",
+    "tests/test_streamlit_pyvista.py",
+    "tests/test_xdmf_mesh.py",
+    "uv.lock"
   ]
 }

solidipes_solid_mech_plugin-0.0.16/solidipes_solid_mech_plugin.egg-info/scm_version.json

@@ -0,0 +1,8 @@
+{
+  "tag": "0.0.16",
+  "distance": 0,
+  "node": "gcfd62de0a703eb7a2bab88cd4f3655c523025f43",
+  "dirty": false,
+  "branch": "HEAD",
+  "node_date": "2026-10-21"
+}

solidipes_solid_mech_plugin-0.0.16/tests/test_exodus_mesh.py

@@ -0,0 +1,88 @@
+import os
+
+import pytest
+from solidipes.loaders import load_file
+
+meshio = pytest.importorskip("meshio")
+netCDF4 = pytest.importorskip("netCDF4")
+
+
+@pytest.fixture
+def exodus_file_path(tmp_path, study_dir):
+    """Create a small Exodus II file with a nodal variable and return its path."""
+    import numpy as np
+
+    points = [[0.0, 0.0, 0.0], [1.0, 0.0, 0.0], [1.0, 1.0, 0.0], [0.0, 1.0, 0.0]]
+    cells = [("quad", np.array([[0, 1, 2, 3]]))]
+    mesh = meshio.Mesh(points, cells, point_data={"v": np.array([0.1, 0.2, 0.3, 0.4])})
+
+    path = os.path.join(tmp_path, "test.e")
+    meshio.write(path, mesh, file_format="exodus")
+    return path
+
+
+@pytest.fixture
+def exodus_file(exodus_file_path):
+    file = load_file(exodus_file_path)
+    return file
+
+
+def test_file_type(exodus_file):
+    """The Exodus loader is selected for .e files"""
+    from solidipes.loaders.file import File
+
+    assert isinstance(exodus_file, File)
+    assert type(exodus_file).__name__ == "Exodus"
+
+
+def test_metadata(exodus_file):
+    assert exodus_file.n_nodes == 4
+    assert exodus_file.n_elements == 1
+    assert exodus_file.n_time_steps == 1
+    assert exodus_file.nodal_variable_names == ["v"]
+
+
+def test_mesh(exodus_file):
+    import numpy as np
+    import pyvista as pv
+
+    mesh = exodus_file.mesh
+    assert isinstance(mesh, pv.UnstructuredGrid)
+    assert mesh.n_points == 4
+    assert mesh.n_cells == 1
+    assert "v" in mesh.array_names
+    assert np.allclose(mesh.point_data["v"], [0.1, 0.2, 0.3, 0.4])
+
+
+def test_time_steps(exodus_file_path):
+    """Add a second time step and check that it can be selected"""
+    import numpy as np
+
+    with netCDF4.Dataset(exodus_file_path, "a") as nc:
+        var = nc.variables["vals_nod_var1"]
+        var[1] = np.array([0.5, 0.6, 0.7, 0.8])
+        nc.variables["time_whole"][1] = 1.0
+
+    exodus_file = load_file(exodus_file_path)
+    assert exodus_file.n_time_steps == 2
+
+    mesh_0 = exodus_file.mesh
+    assert np.allclose(mesh_0.point_data["v"], [0.1, 0.2, 0.3, 0.4])
+
+    exodus_file.select_frame(1)
+    mesh_1 = exodus_file.mesh
+    assert np.allclose(mesh_1.point_data["v"], [0.5, 0.6, 0.7, 0.8])
+
+
+def test_validation(exodus_file):
+    assert exodus_file.is_valid
+
+
+def test_inherited_netcdf(exodus_file):
+    """The netCDF loadable and metadata are inherited from the NetCDF loader"""
+    import netCDF4
+
+    assert isinstance(exodus_file.netcdf, netCDF4.Dataset)
+    assert exodus_file.dimensions["num_nodes"] == 4
+    assert exodus_file.global_attributes["title"].startswith("Created by meshio")
+    assert "coord" in exodus_file.variables

solidipes_solid_mech_plugin-0.0.16/tests/test_pyvista_plotter.py

@@ -0,0 +1,59 @@
+import solidipes as sp
+import utils
+from PIL import Image
+from pixelmatch.contrib.PIL import pixelmatch
+
+image_name = "pyvista_plot.png"
+image_size = (1024, 768)
+
+
+def start_xvfb(wait=3, window_size=None):
+    """Start the virtual framebuffer Xvfb.
+
+    Parameters
+    ----------
+    wait : float, optional
+        Time to wait for the virtual framebuffer to start. Set to 0
+        to disable wait.
+
+    window_size : list, optional
+        Window size of the virtual frame buffer. Defaults to ``[1024, 768]``.
+    """
+    import os
+    import time
+
+    if os.name != "posix":
+        raise OSError("`start_xvfb` is only supported on Linux")
+
+    if os.system("which Xvfb > /dev/null"):
+        raise OSError("Xvfb is not installed. Install it with: apt-get install xvfb libgl1-mesa-glx")
+
+    if window_size is None:
+        window_size = [1024, 768]
+    window_size_parm = f"{window_size[0]:d}x{window_size[1]:d}x24"
+    display_num = ":99"
+    os.system(f"Xvfb {display_num} -screen 0 {window_size_parm} > /dev/null 2>&1 &")
+    os.environ["DISPLAY"] = display_num
+    if wait:
+        time.sleep(wait)
+
+
+def test_plotter(study_dir):
+    pyvista_file = sp.load_file(utils.get_asset_path("pyvista_mesh.vtu"))
+    test_image_path = study_dir / image_name
+    reference_image_path = utils.get_asset_path(image_name)
+
+    # Start virtual frame buffer, otherwise segfault
+    start_xvfb()
+
+    # Create a plotter and save the image
+    plotter = sp.viewer.PyvistaPlotter(off_screen=True)
+    plotter.add_mesh(pyvista_file)
+    plotter.show()  # Not shown because off_screen=True
+    plotter.save(test_image_path)
+
+    test_image = Image.open(test_image_path)
+    reference_image = Image.open(reference_image_path)
+    diff = Image.new("RGB", image_size)
+    mismatch = pixelmatch(test_image, reference_image, diff)
+    assert mismatch == 0