subsurface-terra 2025.1.0rc15__py3-none-any.whl → 2025.1.0rc16__py3-none-any.whl

subsurface/modules/reader/mesh/_trimesh_reader.py

@@ -1,447 +1,447 @@
-import enum
-from typing import Union, TextIO, Optional
-import io
-import os
-
-import numpy as np
-from ....core.structs import UnstructuredData
-from .... import optional_requirements
-from ....core.structs import TriSurf, StructuredData
-
-
-class TriMeshTransformations(enum.Flag):
-    UP_Z = 2**1
-    UP_Y = 2**2
-    FORWARD_MINUS_Z = 2**3
-    FORWARD_PLUS_Z = 2**4
-    RIGHT_HANDED_Z_UP_Y_REVERSED = UP_Y | FORWARD_MINUS_Z
-    RIGHT_HANDED_Z_UP = UP_Y | FORWARD_PLUS_Z
-    ORIGINAL = UP_Z | FORWARD_MINUS_Z
-
-
-def load_with_trimesh(path_to_file_or_buffer, file_type: Optional[str] = None,
-                      coordinate_system: TriMeshTransformations = TriMeshTransformations.RIGHT_HANDED_Z_UP, *, plot=False):
-    """
-    Load a mesh with trimesh and convert to the specified coordinate system.
-
-    """
-    trimesh = optional_requirements.require_trimesh()
-    scene_or_mesh = LoadWithTrimesh.load_with_trimesh(path_to_file_or_buffer, file_type, plot)
-
-    match coordinate_system:
-        case TriMeshTransformations.ORIGINAL:
-            return scene_or_mesh
-        # * Forward -Z up Y
-        case TriMeshTransformations.RIGHT_HANDED_Z_UP:
-            # Transform from Y-up (modeling software) to Z-up (scientific)
-            # This rotates the model so that:
-            # Old Y axis → New Z axis (pointing up)
-            # Old Z axis → New -Y axis
-            # Old X axis → Remains as X axis
-            transform = np.array([
-                    [1, 0, 0, 0],  
-                    [0, 0, -1, 0],  
-                    [0, 1, 0, 0], 
-                    [0, 0, 0, 1]
-            ])
-        case TriMeshTransformations.RIGHT_HANDED_Z_UP_Y_REVERSED:
-            # * Forward Z Up Y
-            transform=np.array([
-                    [-1, 0, 0, 0],
-                    [0, 0, 1, 0],
-                    [0, 1, 0, 0],
-                    [0, 0, 0, 1],
-            ])
-            # Apply the coordinate transformation
-        # TODO: Add all the options of blender
-        case _:
-            raise ValueError(f"Invalid coordinate system: {coordinate_system}")
-
-    if isinstance(scene_or_mesh, trimesh.Scene):
-        for geometry in scene_or_mesh.geometry.values():
-            geometry.apply_transform(transform)
-    else:
-        scene_or_mesh.apply_transform(transform)
-    return scene_or_mesh
-
-
-def trimesh_to_unstruct(scene_or_mesh: Union["trimesh.Trimesh", "trimesh.Scene"]) -> TriSurf:
-    return TrimeshToSubsurface.trimesh_to_unstruct(scene_or_mesh)
-
-
-class LoadWithTrimesh:
-    @classmethod
-    def load_with_trimesh(cls, path_to_file_or_buffer, file_type: Optional[str] = None, plot=False):
-        trimesh = optional_requirements.require_trimesh()
-        # Load the OBJ with Trimesh using the specified options
-        scene_or_mesh = trimesh.load(
-            file_obj=path_to_file_or_buffer,
-            file_type=file_type,
-            force="mesh"
-        )
-        # Process single mesh vs. scene
-        if isinstance(scene_or_mesh, trimesh.Scene):
-            print("Loaded a Scene with multiple geometries.")
-            cls._process_scene(scene_or_mesh)
-            if plot:
-                scene_or_mesh.show()
-        else:
-            print("Loaded a single Trimesh object.")
-            print(f" - Vertices: {len(scene_or_mesh.vertices)}")
-            print(f" - Faces: {len(scene_or_mesh.faces)}")
-            cls.handle_material_info(scene_or_mesh)
-            if plot:
-                scene_or_mesh.show()
-
-        return scene_or_mesh
-
-    @classmethod
-    def handle_material_info(cls, geometry):
-        """
-        Handle and print material information for a single geometry,
-        explicitly injecting the PIL image if provided.
-        """
-        if geometry.visual and hasattr(geometry.visual, 'material'):
-            material = geometry.visual.material
-
-            print("Trimesh material:", material)
-
-            # If there's already an image reference in the material, let the user know
-            if hasattr(material, 'image') and material.image is not None:
-                print("  -> Material already has an image:", material.image)
-            
-            if geometry.visual.uv is None:
-                raise ValueError("Geometry does not have UV coordinates for texture mapping, despite having a material."
-                                 "This can also happen if the geometry is given in quads instead of triangles.")
-        else:
-            print("No material found or no 'material' attribute on this geometry.")
-
-    @classmethod
-    def _process_scene(cls, scene):
-        """Process a scene with multiple geometries."""
-        geometries = scene.geometry
-        assert len(geometries) > 0, "No geometries found in the scene."
-
-        print(f"Loaded a Scene with {len(scene.geometry)} geometry object(s).")
-        for geom_name, geom in geometries.items():
-            print(f" Submesh: {geom_name}")
-            print(f"  - Vertices: {len(geom.vertices)}")
-            print(f"  - Faces: {len(geom.faces)}")
-
-            print(f"Geometry '{geom_name}':")
-            cls.handle_material_info(geom)
-
-
-class TrimeshToSubsurface:
-    @classmethod
-    def trimesh_to_unstruct(cls, scene_or_mesh: Union["trimesh.Trimesh", "trimesh.Scene"]) -> TriSurf:
-        """
-        Convert a Trimesh or Scene object to a subsurface TriSurf object.
-
-        This function takes either a `trimesh.Trimesh` object or a `trimesh.Scene` 
-        object and converts it to a `subsurface.TriSurf` object. If the input is 
-        a scene containing multiple geometries, it processes all geometries and 
-        combines them into a single TriSurf object. If the input is a single 
-        Trimesh object, it directly converts it to a TriSurf object. An error 
-        is raised if the input is neither a `trimesh.Trimesh` nor a `trimesh.Scene` 
-        object.
-
-        Parameters:
-            scene_or_mesh (Union[trimesh.Trimesh, trimesh.Scene]): 
-                Input geometry data, either as a Trimesh object representing 
-                a single mesh or a Scene object containing multiple geometries.
-
-        Note:
-            ! Multimesh with multiple materials will read the uvs but not the textures since in that case is better
-            ! to read directly the multiple images (compressed) whenever the user wants to work with them. 
-
-        Returns:
-            subsurface.TriSurf: Converted subsurface representation of the 
-            provided geometry data.
-
-        Raises:
-            ValueError: If the input is neither a `trimesh.Trimesh` object nor 
-            a `trimesh.Scene` object.
-        """
-        trimesh = optional_requirements.require_trimesh()
-        if isinstance(scene_or_mesh, trimesh.Scene):
-            # Process scene with multiple geometries
-            ts = cls._trisurf_from_scene(scene_or_mesh, trimesh)
-
-        elif isinstance(scene_or_mesh, trimesh.Trimesh):
-            ts = cls._trisurf_from_trimesh(scene_or_mesh)
-
-
-        else:
-            raise ValueError("Input must be a Trimesh object or a Scene with multiple geometries.")
-
-        return ts
-
-    @classmethod
-    def _trisurf_from_trimesh(cls, scene_or_mesh):
-        # Process single mesh
-        tri = scene_or_mesh
-        pandas = optional_requirements.require_pandas()
-        frame = pandas.DataFrame(tri.face_attributes)
-        # Check frame has a valid shape for cells_attr if not make None
-        if frame.shape[0] != tri.faces.shape[0]:
-            frame = None
-        # Get UV coordinates if they exist
-        vertex_attr = None
-        if hasattr(tri.visual, 'uv') and tri.visual.uv is not None:
-            vertex_attr = pandas.DataFrame(
-                tri.visual.uv,
-                columns=['u', 'v']
-            )
-        unstruct = UnstructuredData.from_array(
-            np.array(tri.vertices),
-            np.array(tri.faces),
-            cells_attr=frame,
-            vertex_attr=vertex_attr,
-            xarray_attributes={
-                    "bounds": tri.bounds.tolist(),
-            },
-        )
-
-        texture = cls._extract_texture_from_material(tri)
-
-        ts = TriSurf(
-            mesh=unstruct,
-            texture=texture,
-        )
-        return ts
-
-    @classmethod
-    def _trisurf_from_scene(cls, scene_or_mesh: 'Scene', trimesh: 'trimesh') -> TriSurf:
-        pandas = optional_requirements.require_pandas()
-        geometries = scene_or_mesh.geometry
-        assert len(geometries) > 0, "No geometries found in the scene."
-        all_vertex = []
-        all_cells = []
-        cell_attr = []
-        all_vertex_attr = []
-        _last_cell = 0
-        texture = None
-        for i, (geom_name, geom) in enumerate(geometries.items()):
-            geom: trimesh.Trimesh
-            LoadWithTrimesh.handle_material_info(geom)
-
-            # Append vertices
-            all_vertex.append(np.array(geom.vertices))
-
-            # Adjust cell indices and append
-            cells = np.array(geom.faces)
-            if len(all_cells) > 0:
-                cells = cells + _last_cell
-            all_cells.append(cells)
-
-            # Create attribute array for this geometry
-            cell_attr.append(np.ones(len(cells)) * i)
-
-            _last_cell = cells.max() + 1
-
-            # Get UV coordinates if they exist
-            if hasattr(geom.visual, 'uv') and geom.visual.uv is not None:
-                vertex_attr = pandas.DataFrame(
-                    geom.visual.uv,
-                    columns=['u', 'v']
-                )
-                all_vertex_attr.append(vertex_attr)
-
-            # Extract texture from material if it is only one geometry
-            if len(geometries) == 1:
-                texture = cls._extract_texture_from_material(geom)
-
-        # Create the combined UnstructuredData
-        unstruct = UnstructuredData.from_array(
-            vertex=np.vstack(all_vertex),
-            cells=np.vstack(all_cells),
-            vertex_attr=pandas.concat(all_vertex_attr, ignore_index=True) if len(all_vertex_attr) > 0 else None,
-            cells_attr=pandas.DataFrame(np.hstack(cell_attr), columns=["Geometry id"]),
-            xarray_attributes={
-                    "bounds": scene_or_mesh.bounds.tolist(),
-            },
-        )
-
-        # If there is a texture
-        ts = TriSurf(
-            mesh=unstruct,
-            texture=texture,
-        )
-
-        return ts
-
-    @classmethod
-    def _extract_texture_from_material(cls, geom):
-        from PIL.JpegImagePlugin import JpegImageFile
-        from PIL.PngImagePlugin import PngImageFile
-        import trimesh
-
-        if geom.visual is None or getattr(geom.visual, 'material', None) is None:
-            return None
-
-        array = np.empty(0)
-        if isinstance(geom.visual.material, trimesh.visual.material.SimpleMaterial):
-            image: JpegImageFile = geom.visual.material.image
-            if image is None:
-                return None
-            array = np.array(image)
-        elif isinstance(geom.visual.material, trimesh.visual.material.PBRMaterial):
-            image: PngImageFile = geom.visual.material.baseColorTexture
-            array = np.array(image.convert('RGBA'))
-
-            if image is None:
-                return None
-        else:
-            raise ValueError(f"Unsupported material type: {type(geom.visual.material)}")
-
-        # Asser that image has 3 channels    assert array.shape[2] == 3    from PIL.PngImagePlugin import PngImageFile
-        assert array.shape[2] == 3 or array.shape[2] == 4
-        texture = StructuredData.from_numpy(array)
-        return texture
-
-    @classmethod
-    def _validate_texture_path(cls, texture_path):
-        """Validate the texture file path."""
-        if texture_path and not texture_path.lower().endswith(('.png', '.jpg', '.jpeg')):
-            raise ValueError("Texture path must be a PNG or JPEG file")
-
-
-class TriMeshReaderFromBlob:
-    @classmethod
-    def OBJ_stream_to_trisurf(cls, obj_stream: TextIO, mtl_stream: list[TextIO],
-                              texture_stream: list[io.BytesIO], coord_system: TriMeshTransformations) -> TriSurf:
-        """
-        Load an OBJ file from a stream and convert it to a TriSurf object.
-        
-        Parameters:
-            obj_stream: TextIO containing the OBJ file data (text format)
-            mtl_stream: TextIO containing the MTL file data (text format)
-            texture_stream: BytesIO containing the texture file data (binary format)
-        
-        Returns:
-            TriSurf: The loaded mesh with textures if available
-        """
-        trimesh = optional_requirements.require_trimesh()
-        import tempfile
-
-        path_in = "file.obj"
-
-        # Create a temporary directory to store associated files
-        with tempfile.TemporaryDirectory() as temp_dir:
-            # Write the OBJ content to a temp file
-            obj_path = os.path.join(temp_dir, os.path.basename(path_in))
-            with open(obj_path, 'w') as f:  # Use text mode 'w' for text files
-                obj_stream.seek(0)
-                f.write(obj_stream.read())
-                obj_stream.seek(0)
-
-            if mtl_stream is not None:
-                cls.write_material_files(
-                    mtl_streams=mtl_stream,
-                    obj_stream=obj_stream,
-                    temp_dir=temp_dir,
-                    texture_streams=texture_stream
-                )
-
-            # Now load the OBJ with all associated files available
-            scene_or_mesh = load_with_trimesh(
-                path_to_file_or_buffer=obj_path,
-                file_type="obj",
-                coordinate_system=coord_system
-            )
-
-            # Convert to a TriSurf object
-            tri_surf = TrimeshToSubsurface.trimesh_to_unstruct(scene_or_mesh)
-
-        return tri_surf
-
-    @classmethod
-    def write_material_files(cls, mtl_streams: list[TextIO], obj_stream: TextIO, temp_dir, texture_streams: list[io.BytesIO]):
-        # Extract mtl references from the OBJ file
-        mtl_files = cls._extract_mtl_references(obj_stream)
-        # Download and save MTL files
-        for e, mtl_file in enumerate(mtl_files):
-            mtl_path = f"{temp_dir}/{mtl_file}" if temp_dir else mtl_file
-            mtl_stream = mtl_streams[e] if mtl_streams else None
-            try:
-                # Save the MTL file to temp directory
-                mtl_temp_path = os.path.join(temp_dir, mtl_file)
-                with open(mtl_temp_path, 'w') as f:  # Use text mode 'w' for text files
-                    mtl_stream.seek(0)
-                    f.write(mtl_stream.read())
-
-                # Extract texture references from MTL
-                mtl_stream.seek(0)
-                texture_files = cls._extract_texture_references(mtl_stream)
-
-                if texture_streams is None:
-                    continue
-
-                # Download texture files
-                for ee, texture_file in enumerate(texture_files):
-                    texture_path = f"{temp_dir}/{texture_file}" if temp_dir else texture_file
-                    texture_stream = texture_streams[ee] if texture_streams else None
-                    try:
-                        # Save the texture file to temp directory
-                        with open(os.path.join(temp_dir, texture_file), 'wb') as f:  # Binary mode for textures
-                            texture_stream.seek(0)
-                            f.write(texture_stream.read())
-                    except Exception as e:
-                        print(f"Failed to load texture {texture_file}: {e}")
-            except Exception as e:
-                print(f"Failed to load MTL file {mtl_file}: {e}")
-
-    @classmethod
-    def _extract_mtl_references(cls, obj_stream):
-        """Extract MTL file references from an OBJ file."""
-        obj_stream.seek(0)
-        mtl_files = []
-
-        # TextIO stream already contains decoded text, so no need to decode
-        obj_text = obj_stream.read()
-        obj_stream.seek(0)
-
-        for line in obj_text.splitlines():
-            if line.startswith('mtllib '):
-                mtl_name = line.split(None, 1)[1].strip()
-                mtl_files.append(mtl_name)
-
-        return mtl_files
-
-    @classmethod
-    def _extract_texture_references(cls, mtl_stream):
-        """
-        Extract texture file references from an MTL file.
-        Works with both TextIO and BytesIO streams.
-        
-        Parameters:
-            mtl_stream: TextIO or BytesIO containing the MTL file data
-            
-        Returns:
-            list[str]: List of texture file names referenced in the MTL
-        """
-        mtl_stream.seek(0)
-        texture_files = []
-
-        # Handle both TextIO and BytesIO
-        if isinstance(mtl_stream, io.TextIOWrapper):
-            # TextIO stream already contains decoded text
-            mtl_text = mtl_stream.read()
-        else:
-            # BytesIO stream needs to be decoded
-            mtl_text = mtl_stream.read().decode('utf-8', errors='replace')
-
-        mtl_stream.seek(0)
-
-        for line in mtl_text.splitlines():
-            # Check for texture map definitions
-            for prefix in ['map_Kd ', 'map_Ka ', 'map_Ks ', 'map_Bump ', 'map_d ']:
-                if line.startswith(prefix):
-                    parts = line.split(None, 1)
-                    if len(parts) > 1:
-                        texture_name = parts[1].strip()
-                        texture_files.append(texture_name)
-                    break
-
-        return texture_files
+import enum
+from typing import Union, TextIO, Optional
+import io
+import os
+
+import numpy as np
+from ....core.structs import UnstructuredData
+from .... import optional_requirements
+from ....core.structs import TriSurf, StructuredData
+
+
+class TriMeshTransformations(enum.Flag):
+    UP_Z = 2**1
+    UP_Y = 2**2
+    FORWARD_MINUS_Z = 2**3
+    FORWARD_PLUS_Z = 2**4
+    RIGHT_HANDED_Z_UP_Y_REVERSED = UP_Y | FORWARD_MINUS_Z
+    RIGHT_HANDED_Z_UP = UP_Y | FORWARD_PLUS_Z
+    ORIGINAL = UP_Z | FORWARD_MINUS_Z
+
+
+def load_with_trimesh(path_to_file_or_buffer, file_type: Optional[str] = None,
+                      coordinate_system: TriMeshTransformations = TriMeshTransformations.RIGHT_HANDED_Z_UP, *, plot=False):
+    """
+    Load a mesh with trimesh and convert to the specified coordinate system.
+
+    """
+    trimesh = optional_requirements.require_trimesh()
+    scene_or_mesh = LoadWithTrimesh.load_with_trimesh(path_to_file_or_buffer, file_type, plot)
+
+    match coordinate_system:
+        case TriMeshTransformations.ORIGINAL:
+            return scene_or_mesh
+        # * Forward -Z up Y
+        case TriMeshTransformations.RIGHT_HANDED_Z_UP:
+            # Transform from Y-up (modeling software) to Z-up (scientific)
+            # This rotates the model so that:
+            # Old Y axis → New Z axis (pointing up)
+            # Old Z axis → New -Y axis
+            # Old X axis → Remains as X axis
+            transform = np.array([
+                    [1, 0, 0, 0],  
+                    [0, 0, -1, 0],  
+                    [0, 1, 0, 0], 
+                    [0, 0, 0, 1]
+            ])
+        case TriMeshTransformations.RIGHT_HANDED_Z_UP_Y_REVERSED:
+            # * Forward Z Up Y
+            transform=np.array([
+                    [-1, 0, 0, 0],
+                    [0, 0, 1, 0],
+                    [0, 1, 0, 0],
+                    [0, 0, 0, 1],
+            ])
+            # Apply the coordinate transformation
+        # TODO: Add all the options of blender
+        case _:
+            raise ValueError(f"Invalid coordinate system: {coordinate_system}")
+
+    if isinstance(scene_or_mesh, trimesh.Scene):
+        for geometry in scene_or_mesh.geometry.values():
+            geometry.apply_transform(transform)
+    else:
+        scene_or_mesh.apply_transform(transform)
+    return scene_or_mesh
+
+
+def trimesh_to_unstruct(scene_or_mesh: Union["trimesh.Trimesh", "trimesh.Scene"]) -> TriSurf:
+    return TrimeshToSubsurface.trimesh_to_unstruct(scene_or_mesh)
+
+
+class LoadWithTrimesh:
+    @classmethod
+    def load_with_trimesh(cls, path_to_file_or_buffer, file_type: Optional[str] = None, plot=False):
+        trimesh = optional_requirements.require_trimesh()
+        # Load the OBJ with Trimesh using the specified options
+        scene_or_mesh = trimesh.load(
+            file_obj=path_to_file_or_buffer,
+            file_type=file_type,
+            force="mesh"
+        )
+        # Process single mesh vs. scene
+        if isinstance(scene_or_mesh, trimesh.Scene):
+            print("Loaded a Scene with multiple geometries.")
+            cls._process_scene(scene_or_mesh)
+            if plot:
+                scene_or_mesh.show()
+        else:
+            print("Loaded a single Trimesh object.")
+            print(f" - Vertices: {len(scene_or_mesh.vertices)}")
+            print(f" - Faces: {len(scene_or_mesh.faces)}")
+            cls.handle_material_info(scene_or_mesh)
+            if plot:
+                scene_or_mesh.show()
+
+        return scene_or_mesh
+
+    @classmethod
+    def handle_material_info(cls, geometry):
+        """
+        Handle and print material information for a single geometry,
+        explicitly injecting the PIL image if provided.
+        """
+        if geometry.visual and hasattr(geometry.visual, 'material'):
+            material = geometry.visual.material
+
+            print("Trimesh material:", material)
+
+            # If there's already an image reference in the material, let the user know
+            if hasattr(material, 'image') and material.image is not None:
+                print("  -> Material already has an image:", material.image)
+            
+            if geometry.visual.uv is None:
+                raise ValueError("Geometry does not have UV coordinates for texture mapping, despite having a material."
+                                 "This can also happen if the geometry is given in quads instead of triangles.")
+        else:
+            print("No material found or no 'material' attribute on this geometry.")
+
+    @classmethod
+    def _process_scene(cls, scene):
+        """Process a scene with multiple geometries."""
+        geometries = scene.geometry
+        assert len(geometries) > 0, "No geometries found in the scene."
+
+        print(f"Loaded a Scene with {len(scene.geometry)} geometry object(s).")
+        for geom_name, geom in geometries.items():
+            print(f" Submesh: {geom_name}")
+            print(f"  - Vertices: {len(geom.vertices)}")
+            print(f"  - Faces: {len(geom.faces)}")
+
+            print(f"Geometry '{geom_name}':")
+            cls.handle_material_info(geom)
+
+
+class TrimeshToSubsurface:
+    @classmethod
+    def trimesh_to_unstruct(cls, scene_or_mesh: Union["trimesh.Trimesh", "trimesh.Scene"]) -> TriSurf:
+        """
+        Convert a Trimesh or Scene object to a subsurface TriSurf object.
+
+        This function takes either a `trimesh.Trimesh` object or a `trimesh.Scene` 
+        object and converts it to a `subsurface.TriSurf` object. If the input is 
+        a scene containing multiple geometries, it processes all geometries and 
+        combines them into a single TriSurf object. If the input is a single 
+        Trimesh object, it directly converts it to a TriSurf object. An error 
+        is raised if the input is neither a `trimesh.Trimesh` nor a `trimesh.Scene` 
+        object.
+
+        Parameters:
+            scene_or_mesh (Union[trimesh.Trimesh, trimesh.Scene]): 
+                Input geometry data, either as a Trimesh object representing 
+                a single mesh or a Scene object containing multiple geometries.
+
+        Note:
+            ! Multimesh with multiple materials will read the uvs but not the textures since in that case is better
+            ! to read directly the multiple images (compressed) whenever the user wants to work with them. 
+
+        Returns:
+            subsurface.TriSurf: Converted subsurface representation of the 
+            provided geometry data.
+
+        Raises:
+            ValueError: If the input is neither a `trimesh.Trimesh` object nor 
+            a `trimesh.Scene` object.
+        """
+        trimesh = optional_requirements.require_trimesh()
+        if isinstance(scene_or_mesh, trimesh.Scene):
+            # Process scene with multiple geometries
+            ts = cls._trisurf_from_scene(scene_or_mesh, trimesh)
+
+        elif isinstance(scene_or_mesh, trimesh.Trimesh):
+            ts = cls._trisurf_from_trimesh(scene_or_mesh)
+
+
+        else:
+            raise ValueError("Input must be a Trimesh object or a Scene with multiple geometries.")
+
+        return ts
+
+    @classmethod
+    def _trisurf_from_trimesh(cls, scene_or_mesh):
+        # Process single mesh
+        tri = scene_or_mesh
+        pandas = optional_requirements.require_pandas()
+        frame = pandas.DataFrame(tri.face_attributes)
+        # Check frame has a valid shape for cells_attr if not make None
+        if frame.shape[0] != tri.faces.shape[0]:
+            frame = None
+        # Get UV coordinates if they exist
+        vertex_attr = None
+        if hasattr(tri.visual, 'uv') and tri.visual.uv is not None:
+            vertex_attr = pandas.DataFrame(
+                tri.visual.uv,
+                columns=['u', 'v']
+            )
+        unstruct = UnstructuredData.from_array(
+            np.array(tri.vertices),
+            np.array(tri.faces),
+            cells_attr=frame,
+            vertex_attr=vertex_attr,
+            xarray_attributes={
+                    "bounds": tri.bounds.tolist(),
+            },
+        )
+
+        texture = cls._extract_texture_from_material(tri)
+
+        ts = TriSurf(
+            mesh=unstruct,
+            texture=texture,
+        )
+        return ts
+
+    @classmethod
+    def _trisurf_from_scene(cls, scene_or_mesh: 'Scene', trimesh: 'trimesh') -> TriSurf:
+        pandas = optional_requirements.require_pandas()
+        geometries = scene_or_mesh.geometry
+        assert len(geometries) > 0, "No geometries found in the scene."
+        all_vertex = []
+        all_cells = []
+        cell_attr = []
+        all_vertex_attr = []
+        _last_cell = 0
+        texture = None
+        for i, (geom_name, geom) in enumerate(geometries.items()):
+            geom: trimesh.Trimesh
+            LoadWithTrimesh.handle_material_info(geom)
+
+            # Append vertices
+            all_vertex.append(np.array(geom.vertices))
+
+            # Adjust cell indices and append
+            cells = np.array(geom.faces)
+            if len(all_cells) > 0:
+                cells = cells + _last_cell
+            all_cells.append(cells)
+
+            # Create attribute array for this geometry
+            cell_attr.append(np.ones(len(cells)) * i)
+
+            _last_cell = cells.max() + 1
+
+            # Get UV coordinates if they exist
+            if hasattr(geom.visual, 'uv') and geom.visual.uv is not None:
+                vertex_attr = pandas.DataFrame(
+                    geom.visual.uv,
+                    columns=['u', 'v']
+                )
+                all_vertex_attr.append(vertex_attr)
+
+            # Extract texture from material if it is only one geometry
+            if len(geometries) == 1:
+                texture = cls._extract_texture_from_material(geom)
+
+        # Create the combined UnstructuredData
+        unstruct = UnstructuredData.from_array(
+            vertex=np.vstack(all_vertex),
+            cells=np.vstack(all_cells),
+            vertex_attr=pandas.concat(all_vertex_attr, ignore_index=True) if len(all_vertex_attr) > 0 else None,
+            cells_attr=pandas.DataFrame(np.hstack(cell_attr), columns=["Geometry id"]),
+            xarray_attributes={
+                    "bounds": scene_or_mesh.bounds.tolist(),
+            },
+        )
+
+        # If there is a texture
+        ts = TriSurf(
+            mesh=unstruct,
+            texture=texture,
+        )
+
+        return ts
+
+    @classmethod
+    def _extract_texture_from_material(cls, geom):
+        from PIL.JpegImagePlugin import JpegImageFile
+        from PIL.PngImagePlugin import PngImageFile
+        import trimesh
+
+        if geom.visual is None or getattr(geom.visual, 'material', None) is None:
+            return None
+
+        array = np.empty(0)
+        if isinstance(geom.visual.material, trimesh.visual.material.SimpleMaterial):
+            image: JpegImageFile = geom.visual.material.image
+            if image is None:
+                return None
+            array = np.array(image)
+        elif isinstance(geom.visual.material, trimesh.visual.material.PBRMaterial):
+            image: PngImageFile = geom.visual.material.baseColorTexture
+            array = np.array(image.convert('RGBA'))
+
+            if image is None:
+                return None
+        else:
+            raise ValueError(f"Unsupported material type: {type(geom.visual.material)}")
+
+        # Asser that image has 3 channels    assert array.shape[2] == 3    from PIL.PngImagePlugin import PngImageFile
+        assert array.shape[2] == 3 or array.shape[2] == 4
+        texture = StructuredData.from_numpy(array)
+        return texture
+
+    @classmethod
+    def _validate_texture_path(cls, texture_path):
+        """Validate the texture file path."""
+        if texture_path and not texture_path.lower().endswith(('.png', '.jpg', '.jpeg')):
+            raise ValueError("Texture path must be a PNG or JPEG file")
+
+
+class TriMeshReaderFromBlob:
+    @classmethod
+    def OBJ_stream_to_trisurf(cls, obj_stream: TextIO, mtl_stream: list[TextIO],
+                              texture_stream: list[io.BytesIO], coord_system: TriMeshTransformations) -> TriSurf:
+        """
+        Load an OBJ file from a stream and convert it to a TriSurf object.
+        
+        Parameters:
+            obj_stream: TextIO containing the OBJ file data (text format)
+            mtl_stream: TextIO containing the MTL file data (text format)
+            texture_stream: BytesIO containing the texture file data (binary format)
+        
+        Returns:
+            TriSurf: The loaded mesh with textures if available
+        """
+        trimesh = optional_requirements.require_trimesh()
+        import tempfile
+
+        path_in = "file.obj"
+
+        # Create a temporary directory to store associated files
+        with tempfile.TemporaryDirectory() as temp_dir:
+            # Write the OBJ content to a temp file
+            obj_path = os.path.join(temp_dir, os.path.basename(path_in))
+            with open(obj_path, 'w') as f:  # Use text mode 'w' for text files
+                obj_stream.seek(0)
+                f.write(obj_stream.read())
+                obj_stream.seek(0)
+
+            if mtl_stream is not None:
+                cls.write_material_files(
+                    mtl_streams=mtl_stream,
+                    obj_stream=obj_stream,
+                    temp_dir=temp_dir,
+                    texture_streams=texture_stream
+                )
+
+            # Now load the OBJ with all associated files available
+            scene_or_mesh = load_with_trimesh(
+                path_to_file_or_buffer=obj_path,
+                file_type="obj",
+                coordinate_system=coord_system
+            )
+
+            # Convert to a TriSurf object
+            tri_surf = TrimeshToSubsurface.trimesh_to_unstruct(scene_or_mesh)
+
+        return tri_surf
+
+    @classmethod
+    def write_material_files(cls, mtl_streams: list[TextIO], obj_stream: TextIO, temp_dir, texture_streams: list[io.BytesIO]):
+        # Extract mtl references from the OBJ file
+        mtl_files = cls._extract_mtl_references(obj_stream)
+        # Download and save MTL files
+        for e, mtl_file in enumerate(mtl_files):
+            mtl_path = f"{temp_dir}/{mtl_file}" if temp_dir else mtl_file
+            mtl_stream = mtl_streams[e] if mtl_streams else None
+            try:
+                # Save the MTL file to temp directory
+                mtl_temp_path = os.path.join(temp_dir, mtl_file)
+                with open(mtl_temp_path, 'w') as f:  # Use text mode 'w' for text files
+                    mtl_stream.seek(0)
+                    f.write(mtl_stream.read())
+
+                # Extract texture references from MTL
+                mtl_stream.seek(0)
+                texture_files = cls._extract_texture_references(mtl_stream)
+
+                if texture_streams is None:
+                    continue
+
+                # Download texture files
+                for ee, texture_file in enumerate(texture_files):
+                    texture_path = f"{temp_dir}/{texture_file}" if temp_dir else texture_file
+                    texture_stream = texture_streams[ee] if texture_streams else None
+                    try:
+                        # Save the texture file to temp directory
+                        with open(os.path.join(temp_dir, texture_file), 'wb') as f:  # Binary mode for textures
+                            texture_stream.seek(0)
+                            f.write(texture_stream.read())
+                    except Exception as e:
+                        print(f"Failed to load texture {texture_file}: {e}")
+            except Exception as e:
+                print(f"Failed to load MTL file {mtl_file}: {e}")
+
+    @classmethod
+    def _extract_mtl_references(cls, obj_stream):
+        """Extract MTL file references from an OBJ file."""
+        obj_stream.seek(0)
+        mtl_files = []
+
+        # TextIO stream already contains decoded text, so no need to decode
+        obj_text = obj_stream.read()
+        obj_stream.seek(0)
+
+        for line in obj_text.splitlines():
+            if line.startswith('mtllib '):
+                mtl_name = line.split(None, 1)[1].strip()
+                mtl_files.append(mtl_name)
+
+        return mtl_files
+
+    @classmethod
+    def _extract_texture_references(cls, mtl_stream):
+        """
+        Extract texture file references from an MTL file.
+        Works with both TextIO and BytesIO streams.
+        
+        Parameters:
+            mtl_stream: TextIO or BytesIO containing the MTL file data
+            
+        Returns:
+            list[str]: List of texture file names referenced in the MTL
+        """
+        mtl_stream.seek(0)
+        texture_files = []
+
+        # Handle both TextIO and BytesIO
+        if isinstance(mtl_stream, io.TextIOWrapper):
+            # TextIO stream already contains decoded text
+            mtl_text = mtl_stream.read()
+        else:
+            # BytesIO stream needs to be decoded
+            mtl_text = mtl_stream.read().decode('utf-8', errors='replace')
+
+        mtl_stream.seek(0)
+
+        for line in mtl_text.splitlines():
+            # Check for texture map definitions
+            for prefix in ['map_Kd ', 'map_Ka ', 'map_Ks ', 'map_Bump ', 'map_d ']:
+                if line.startswith(prefix):
+                    parts = line.split(None, 1)
+                    if len(parts) > 1:
+                        texture_name = parts[1].strip()
+                        texture_files.append(texture_name)
+                    break
+
+        return texture_files