conquer3d 0.3.4__tar.gz → 0.3.5__tar.gz

{conquer3d-0.3.4/conquer3d.egg-info → conquer3d-0.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conquer3d
-Version: 0.3.4
+Version: 0.3.5
 Summary: Geometric Cuda Tool Box
 Author-email: Do Hoang Khoi <khoido8899@gmail.com>
 License-Expression: MIT
@@ -13,6 +13,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: torch>=2.0.0
+Requires-Dist: meshio>=5.3.5
 Requires-Dist: gdown
 Provides-Extra: cuda
 Requires-Dist: cuda-toolkit==12.8.2; extra == "cuda"
@@ -49,6 +50,7 @@ pip install git+https://github.com/KhoiDOO/geocutool.git --no-build-isolation
 ```bash
 conda create -c conda-forge -n geocutool python=3.10 gxx_linux-64=13 gcc_linux-64=13 -y
 conda activate geocutool
+conda install -c conda-forge sparsehash -y
 
 conda install nvidia::cuda-toolkit==12.8.2 -y
 

{conquer3d-0.3.4 → conquer3d-0.3.5}/README.md

@@ -20,6 +20,7 @@ pip install git+https://github.com/KhoiDOO/geocutool.git --no-build-isolation
 ```bash
 conda create -c conda-forge -n geocutool python=3.10 gxx_linux-64=13 gcc_linux-64=13 -y
 conda activate geocutool
+conda install -c conda-forge sparsehash -y
 
 conda install nvidia::cuda-toolkit==12.8.2 -y
 

{conquer3d-0.3.4 → conquer3d-0.3.5}/conquer3d/data_structure/grid.py

@@ -62,4 +62,21 @@ def compute_active_voxels(
     Returns:
         torch.Tensor: An int64 tensor containing the indices of the active voxels.
     """
-    return _C.compute_active_voxels(voxels.contiguous(), sdf.contiguous(), iso)
+    return _C.compute_active_voxels(voxels.contiguous(), sdf.contiguous(), iso)
+
+def create_random_points_ball(n_points: int, radius: float = 1.0, device: str = 'cuda') -> torch.Tensor:
+    """
+    Creates a set of random points uniformly distributed inside a sphere.
+
+    Args:
+        n_points (int): Number of points to sample.
+        radius (float): Radius of the sphere. Defaults to 1.0.
+        device (str): Device to place the tensor on. Defaults to 'cuda'.
+
+    Returns:
+        torch.Tensor: Random points of shape (n_points, 3).
+    """
+    points = torch.randn(n_points, 3, device=device)
+    radii = torch.rand(n_points, 1, device=device) ** (1/3) * radius
+    points = (points / points.norm(dim=-1, keepdim=True)) * radii
+    return points

conquer3d-0.3.5/conquer3d/io/__init__.py

@@ -0,0 +1,2 @@
+from .obj import read_obj, write_obj
+from .off import read_off

conquer3d-0.3.5/conquer3d/io/obj.py

@@ -0,0 +1,42 @@
+import torch
+import meshio
+
+def read_obj(file_obj):
+    """
+    Reads an OBJ file and returns vertices, faces, and optional vertex colors using meshio.
+    
+    Args:
+        file_obj: A file-like object or a string path.
+    """
+    mesh = meshio.read(file_obj, file_format="obj")
+    vertices = torch.tensor(mesh.points, dtype=torch.float32)
+    
+    faces = None
+    if "triangle" in mesh.cells_dict:
+        faces = torch.tensor(mesh.cells_dict["triangle"], dtype=torch.long)
+    else:
+        faces = torch.empty((0, 3), dtype=torch.long)
+        
+    # Meshio sometimes places vertex colors in point_data under 'obj:vc'
+    colors = None
+    if mesh.point_data is not None and "obj:vc" in mesh.point_data:
+        colors = torch.tensor(mesh.point_data["obj:vc"], dtype=torch.float32)
+        
+    return vertices, faces, colors
+
+def write_obj(filepath, vertices, faces, colors=None):
+    """
+    Writes vertices, faces, and optional vertex colors to an OBJ file using meshio.
+    """
+    point_data = {}
+    if colors is not None:
+        point_data["obj:vc"] = colors.detach().cpu().numpy()
+        
+    mesh = meshio.Mesh(
+        points=vertices.detach().cpu().numpy(),
+        cells=[("triangle", faces.detach().cpu().numpy())],
+        point_data=point_data if len(point_data) > 0 else None
+    )
+    meshio.write(filepath, mesh, file_format="obj")
+
+

conquer3d-0.3.5/conquer3d/io/off.py

@@ -0,0 +1,23 @@
+import torch
+import meshio
+
+def read_off(filepath_or_filelike):
+    """
+    Read an OFF (Object File Format) file and return vertices and faces as PyTorch tensors using meshio.
+    
+    Args:
+        filepath_or_filelike: A string path or a file-like object.
+        
+    Returns:
+        tuple: (vertices, faces) as torch.Tensor
+    """
+    mesh = meshio.read(filepath_or_filelike, file_format="off")
+    vertices = torch.tensor(mesh.points, dtype=torch.float32)
+    
+    faces = None
+    if "triangle" in mesh.cells_dict:
+        faces = torch.tensor(mesh.cells_dict["triangle"], dtype=torch.long)
+    else:
+        faces = torch.empty((0, 3), dtype=torch.long)
+        
+    return vertices, faces

{conquer3d-0.3.4 → conquer3d-0.3.5/conquer3d.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: conquer3d
-Version: 0.3.4
+Version: 0.3.5
 Summary: Geometric Cuda Tool Box
 Author-email: Do Hoang Khoi <khoido8899@gmail.com>
 License-Expression: MIT
@@ -13,6 +13,7 @@ Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: torch>=2.0.0
+Requires-Dist: meshio>=5.3.5
 Requires-Dist: gdown
 Provides-Extra: cuda
 Requires-Dist: cuda-toolkit==12.8.2; extra == "cuda"
@@ -49,6 +50,7 @@ pip install git+https://github.com/KhoiDOO/geocutool.git --no-build-isolation
 ```bash
 conda create -c conda-forge -n geocutool python=3.10 gxx_linux-64=13 gcc_linux-64=13 -y
 conda activate geocutool
+conda install -c conda-forge sparsehash -y
 
 conda install nvidia::cuda-toolkit==12.8.2 -y
 

{conquer3d-0.3.4 → conquer3d-0.3.5}/conquer3d.egg-info/requires.txt

@@ -1,4 +1,5 @@
 torch>=2.0.0
+meshio>=5.3.5
 gdown
 
 [cuda]

{conquer3d-0.3.4 → conquer3d-0.3.5}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "conquer3d"
-version = "0.3.4"
+version = "0.3.5"
 description = "Geometric Cuda Tool Box"
 readme = "README.md"
 license = "MIT"
@@ -20,6 +20,7 @@ requires-python = ">=3.9"
 keywords = ["geometry"]
 dependencies = [
     "torch>=2.0.0",
+    "meshio>=5.3.5",
     "gdown",
 ]
 

conquer3d-0.3.4/conquer3d/io/obj.py

@@ -1,46 +0,0 @@
-import torch
-
-def read_obj(file_obj):
-    """
-    Reads an OBJ file and returns vertices, faces, and optional vertex colors.
-    
-    Args:
-        file_obj: A file-like object or a string path.
-    """
-    vertices = []
-    faces = []
-    colors = []
-    
-    close_file = False
-    if isinstance(file_obj, str):
-        file_obj = open(file_obj, 'r', encoding='utf-8')
-        close_file = True
-
-    try:
-        for line in file_obj:
-            if isinstance(line, bytes):
-                line = line.decode('utf-8', errors='ignore')
-            line = line.strip()
-            if line.startswith('v '):
-                parts = line.split()
-                vertices.append([float(parts[1]), float(parts[2]), float(parts[3])])
-                if len(parts) >= 7:
-                    colors.append([float(parts[4]), float(parts[5]), float(parts[6])])
-            elif line.startswith('f '):
-                parts = line.split()
-                # Face elements can be v, v/vt, or v/vt/vn. We just want the first index (v).
-                face = [int(p.split('/')[0]) - 1 for p in parts[1:4]]
-                faces.append(face)
-    finally:
-        if close_file:
-            file_obj.close()
-            
-    vertices = torch.tensor(vertices, dtype=torch.float32)
-    faces = torch.tensor(faces, dtype=torch.long)
-    
-    if len(colors) == len(vertices):
-        colors = torch.tensor(colors, dtype=torch.float32)
-    else:
-        colors = None
-        
-    return vertices, faces, colors

conquer3d-0.3.4/conquer3d/io/off.py

@@ -1,84 +0,0 @@
-import torch
-import numpy as np
-import io
-
-def read_off(filepath_or_filelike):
-    """
-    Read an OFF (Object File Format) file and return vertices and faces as PyTorch tensors.
-    Supports reading from a file path or a file-like object.
-    
-    Args:
-        filepath_or_filelike: A string path or a file-like object.
-        
-    Returns:
-        tuple: (vertices, faces) as torch.Tensor
-    """
-    if isinstance(filepath_or_filelike, str):
-        with open(filepath_or_filelike, "r") as f:
-            lines = f.readlines()
-    else:
-        # Assuming text-based or bytes file-like object
-        lines = filepath_or_filelike.readlines()
-        if len(lines) > 0 and isinstance(lines[0], bytes):
-            lines = [line.decode('utf-8') for line in lines]
-
-    # Clean up empty lines and comments
-    lines = [line.strip() for line in lines if line.strip() and not line.startswith('#')]
-    
-    if not lines:
-        raise ValueError("Empty OFF file.")
-
-    first_line = lines[0].strip()
-    if first_line.upper().startswith("OFF"):
-        if len(first_line) > 3:
-            # Format: 'OFF 123 456 0'
-            parts = first_line[3:].strip().split()
-            # If the rest of the line doesn't contain counts, maybe they are on the next line
-            if parts:
-                lines = [parts] + lines[1:]
-            else:
-                lines = lines[1:]
-        else:
-            lines = lines[1:]
-            
-    # If the format parsing above resulted in a list of parts at index 0, handle it
-    if isinstance(lines[0], list):
-        counts = lines[0]
-    else:
-        counts = lines[0].split()
-        
-    num_verts, num_faces, num_edges = map(int, counts[:3])
-    
-    # Fast parsing with numpy
-    verts_lines = lines[1:num_verts + 1]
-    verts_str = '\n'.join(verts_lines)
-    verts = np.fromstring(verts_str, sep=' ').reshape(num_verts, -1)[:, :3]
-    
-    faces_lines = lines[num_verts + 1 : num_verts + 1 + num_faces]
-    faces_str = '\n'.join(faces_lines)
-    
-    # In an OFF file, face lines look like: `3 v1 v2 v3` for triangles.
-    # We parse the flat array and reshape based on the number of elements per line
-    faces_raw = np.fromstring(faces_str, sep=' ', dtype=int)
-    
-    # We assume all faces are triangles! 
-    # The first element of each chunk will be 3, followed by 3 indices.
-    # Total length should be num_faces * 4
-    if len(faces_raw) == num_faces * 4:
-        faces = faces_raw.reshape(num_faces, 4)[:, 1:4]
-    else:
-        # If faces are mixed (e.g. some quads), we fall back to a slower parsing
-        faces_list = []
-        idx = 0
-        for _ in range(num_faces):
-            n_v = faces_raw[idx]
-            if n_v == 3:
-                faces_list.append([faces_raw[idx+1], faces_raw[idx+2], faces_raw[idx+3]])
-            else:
-                # Naive triangulation for ngons (fan triangulation)
-                for i in range(1, n_v - 1):
-                    faces_list.append([faces_raw[idx+1], faces_raw[idx+i+1], faces_raw[idx+i+2]])
-            idx += n_v + 1
-        faces = np.array(faces_list, dtype=int)
-        
-    return torch.from_numpy(verts).float(), torch.from_numpy(faces).long()