reverse3d 0.8.0__tar.gz → 0.8.1__tar.gz

{reverse3d-0.8.0/reverse3d.egg-info → reverse3d-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reverse3d
-Version: 0.8.0
+Version: 0.8.1
 Summary: AI-agent-ready toolkit for 3D reverse engineering: image analysis, photogrammetry, mesh repair, primitive fitting and headless rendering.
 Author-email: Voxel Enterprise <info@voxel.lat>
 Maintainer-email: Voxel Enterprise <info@voxel.lat>
@@ -215,10 +215,11 @@ Cada archivo modificado se **respalda antes** (`*.reverse3d-backup-<fecha>`). So
 
 | bash (`install.sh`) | PowerShell (`install.ps1`) | Efecto |
 |---------------------|----------------------------|--------|
-| *(sin opciones)* | *(sin opciones)* | **Todo**: núcleo, MCP, render, mallado, IA de profundidad, Blender y COLMAP |
-| `--lite` | `-Lite` | Todo menos la IA de profundidad (PyTorch); ahorra hasta ~2.5 GB |
-| `--minimal` | `-Minimal` | Solo núcleo + MCP, sin herramientas externas |
+| *(sin opciones)* | *(sin opciones)* | **Todo**: núcleo, MCP, render, mallado, IA local (con CUDA si hay NVIDIA), Blender y COLMAP |
+| `--lite` | `-Lite` | Todo, con la IA local en la CPU (PyTorch CPU, ~250 MB en vez de ~2.5 GB) |
+| `--minimal` | `-Minimal` | Núcleo + MCP + IA local, sin herramientas externas |
 | `--no-tools` | `-NoTools` | No instalar Blender ni COLMAP |
+| *(en todos los modos)* | *(en todos los modos)* | **La IA local se instala y se descarga siempre** (clasificador, segmentación y profundidad, unos 750 MB) para que todo funcione sin internet |
 | `--no-auto-approve` | `-NoAutoApprove` | Que los agentes pregunten antes de cada herramienta |
 | `--with recon,depth` | `-With recon,depth` | Extras a elección |
 | `--device auto\|cpu\|gpu` | `-Device …` | Dispositivo por defecto |

{reverse3d-0.8.0 → reverse3d-0.8.1}/README.md

@@ -147,10 +147,11 @@ Cada archivo modificado se **respalda antes** (`*.reverse3d-backup-<fecha>`). So
 
 | bash (`install.sh`) | PowerShell (`install.ps1`) | Efecto |
 |---------------------|----------------------------|--------|
-| *(sin opciones)* | *(sin opciones)* | **Todo**: núcleo, MCP, render, mallado, IA de profundidad, Blender y COLMAP |
-| `--lite` | `-Lite` | Todo menos la IA de profundidad (PyTorch); ahorra hasta ~2.5 GB |
-| `--minimal` | `-Minimal` | Solo núcleo + MCP, sin herramientas externas |
+| *(sin opciones)* | *(sin opciones)* | **Todo**: núcleo, MCP, render, mallado, IA local (con CUDA si hay NVIDIA), Blender y COLMAP |
+| `--lite` | `-Lite` | Todo, con la IA local en la CPU (PyTorch CPU, ~250 MB en vez de ~2.5 GB) |
+| `--minimal` | `-Minimal` | Núcleo + MCP + IA local, sin herramientas externas |
 | `--no-tools` | `-NoTools` | No instalar Blender ni COLMAP |
+| *(en todos los modos)* | *(en todos los modos)* | **La IA local se instala y se descarga siempre** (clasificador, segmentación y profundidad, unos 750 MB) para que todo funcione sin internet |
 | `--no-auto-approve` | `-NoAutoApprove` | Que los agentes pregunten antes de cada herramienta |
 | `--with recon,depth` | `-With recon,depth` | Extras a elección |
 | `--device auto\|cpu\|gpu` | `-Device …` | Dispositivo por defecto |

reverse3d-0.8.1/reverse3d/_version.py

@@ -0,0 +1 @@
+__version__ = "0.8.1"

{reverse3d-0.8.0 → reverse3d-0.8.1}/reverse3d/recipes.py

@@ -30,6 +30,8 @@ from dataclasses import dataclass, field
 from pathlib import Path
 from typing import Any, Literal, TypeVar
 
+import numpy as np
+
 from reverse3d.config import Settings
 from reverse3d.errors import (
     CaptureQualityError,
@@ -135,29 +137,70 @@ def _clean(
     parts, and tiny debris is already dropped by the area filter."""
     from reverse3d.reverse_eng.mesh_cleaner import CleanOptions, clean_mesh_file
 
+    options = CleanOptions(
+        keep_largest_only=keep_largest,
+        fill_holes=fill_holes,
+        align=align,
+        ground=align != "none",
+        scale=scale,
+        max_hole_edges=2000 if repair == "strong" else 200,
+    )
     report = run.step(
         "clean",
-        lambda: clean_mesh_file(
-            src,
-            dst,
-            run.settings,
-            CleanOptions(
-                keep_largest_only=keep_largest,
-                fill_holes=fill_holes,
-                align=align,
-                ground=align != "none",
-                scale=scale,
-                max_hole_edges=2000 if repair == "strong" else 200,
-            ),
-            overwrite=overwrite,
-        ),
+        lambda: clean_mesh_file(src, dst, run.settings, options, overwrite=overwrite),
     )
     run.warnings.extend(report.warnings)
     run.outputs["model"] = str(dst)
     run.details["clean"] = report.to_dict()
+    if _has_vertex_colors(src):
+        # Same cleaning on the coloured mesh (deterministic, colours follow the
+        # vertices): model_color.ply for renders and model.glb for games and viewers.
+        try:
+            color_ply = dst.with_name("model_color.ply")
+            clean_mesh_file(src, color_ply, run.settings, options, overwrite=True)
+            import trimesh
+
+            glb = dst.with_name("model.glb")
+            trimesh.load(color_ply, process=False, force="mesh").export(glb)
+            run.outputs["model_color"] = str(color_ply)
+            run.outputs["model_glb"] = str(glb)
+        except Exception as exc:  # colour is a bonus: never fail the model for it
+            logger.info("Coloured model skipped: %s", exc)
     return report.after
 
 
+def _has_vertex_colors(path: Path) -> bool:
+    import trimesh
+
+    try:
+        mesh: Any = trimesh.load(path, process=False, force="mesh")
+    except Exception:
+        return False
+    colors = getattr(mesh.visual, "vertex_colors", None)
+    if colors is None or len(colors) != len(mesh.vertices):
+        return False
+    rgb = np.asarray(colors)[:, :3]
+    return float(np.max(np.std(rgb, axis=0))) > 1.0  # not just a uniform default colour
+
+
+def _color_from_cloud(mesh_path: Path, cloud_path: Path, out: Path) -> Path | None:
+    """Paint a mesh with the colours of the point cloud it was built from."""
+    import trimesh
+    from scipy.spatial import cKDTree
+
+    cloud: Any = trimesh.load(cloud_path, process=False)
+    colors = getattr(cloud, "colors", None)
+    if colors is None or len(colors) != len(getattr(cloud, "vertices", [])):
+        return None
+    mesh: Any = trimesh.load(mesh_path, process=False, force="mesh")
+    _, idx = cKDTree(np.asarray(cloud.vertices)).query(np.asarray(mesh.vertices), k=4)
+    rgb = np.asarray(colors)[idx][..., :3].astype(np.float64).mean(axis=1)
+    rgba = np.column_stack([rgb, np.full(len(rgb), 255.0)]).clip(0, 255).astype(np.uint8)
+    mesh.visual = trimesh.visual.ColorVisuals(mesh, vertex_colors=rgba)
+    mesh.export(out)
+    return out
+
+
 def _render(run: _Run, mesh: Path, overwrite: bool) -> None:
     from reverse3d.rendering.blender_render import RenderOptions, preset_options, render_mesh
 
@@ -210,28 +253,6 @@ _DEFAULT_LOOK = {
 }
 
 
-def _color_outputs(run: _Run, colored: Path, model: Path, scale: float) -> None:
-    """Give the final model the photo colours of the carved hull: model.glb (games,
-    web, viewers) and model_color.ply (renders)."""
-    import numpy as np
-    import trimesh
-    from scipy.spatial import cKDTree
-
-    source: Any = trimesh.load(colored, process=False, force="mesh")
-    target: Any = trimesh.load(model, process=False, force="mesh")
-    colors = getattr(source.visual, "vertex_colors", None)
-    if colors is None or len(colors) != len(source.vertices):
-        return
-    _, nearest = cKDTree(np.asarray(source.vertices) * scale).query(np.asarray(target.vertices))
-    target.visual = trimesh.visual.ColorVisuals(target, vertex_colors=np.asarray(colors)[nearest])
-    glb = run.out / "model.glb"
-    ply = run.out / "model_color.ply"
-    target.export(glb)
-    target.export(ply)
-    run.outputs["model_glb"] = str(glb)
-    run.outputs["model_color"] = str(ply)
-
-
 def _photos_to_model(run: _Run, src: Path, opt: RecipeOptions) -> MeshStats:
     from reverse3d.reconstruction.photogrammetry import (
         PhotogrammetryOptions,
@@ -352,6 +373,15 @@ def _photos_to_model(run: _Run, src: Path, opt: RecipeOptions) -> MeshStats:
             "Point cloud created but no mesh could be built.",
             hint='Install the meshing extra: pip install "reverse3d[recon]" (Python ≤ 3.12).',
         )
+    cloud = photo.dense_points_path or photo.sparse_points_path
+    if not upright and cloud:
+        try:
+            colored = _color_from_cloud(
+                Path(mesh_path), Path(cloud), run.out / "reconstruction" / "mesh_color.ply"
+            )
+            mesh_path = str(colored) if colored else mesh_path
+        except Exception as exc:  # colour is a bonus
+            logger.info("Colouring from the point cloud skipped: %s", exc)
     model = run.out / "model.stl"
     _clean(
         run,
@@ -364,11 +394,6 @@ def _photos_to_model(run: _Run, src: Path, opt: RecipeOptions) -> MeshStats:
         keep_largest=preset.keep_largest,
         repair=opt.repair,
     )
-    if upright:
-        try:
-            _color_outputs(run, Path(mesh_path), model, opt.scale)
-        except Exception as exc:  # colour is a bonus: never fail the model for it
-            logger.info("Colour transfer skipped: %s", exc)
     return _measure(run, model)
 
 

{reverse3d-0.8.0 → reverse3d-0.8.1/reverse3d.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: reverse3d
-Version: 0.8.0
+Version: 0.8.1
 Summary: AI-agent-ready toolkit for 3D reverse engineering: image analysis, photogrammetry, mesh repair, primitive fitting and headless rendering.
 Author-email: Voxel Enterprise <info@voxel.lat>
 Maintainer-email: Voxel Enterprise <info@voxel.lat>
@@ -215,10 +215,11 @@ Cada archivo modificado se **respalda antes** (`*.reverse3d-backup-<fecha>`). So
 
 | bash (`install.sh`) | PowerShell (`install.ps1`) | Efecto |
 |---------------------|----------------------------|--------|
-| *(sin opciones)* | *(sin opciones)* | **Todo**: núcleo, MCP, render, mallado, IA de profundidad, Blender y COLMAP |
-| `--lite` | `-Lite` | Todo menos la IA de profundidad (PyTorch); ahorra hasta ~2.5 GB |
-| `--minimal` | `-Minimal` | Solo núcleo + MCP, sin herramientas externas |
+| *(sin opciones)* | *(sin opciones)* | **Todo**: núcleo, MCP, render, mallado, IA local (con CUDA si hay NVIDIA), Blender y COLMAP |
+| `--lite` | `-Lite` | Todo, con la IA local en la CPU (PyTorch CPU, ~250 MB en vez de ~2.5 GB) |
+| `--minimal` | `-Minimal` | Núcleo + MCP + IA local, sin herramientas externas |
 | `--no-tools` | `-NoTools` | No instalar Blender ni COLMAP |
+| *(en todos los modos)* | *(en todos los modos)* | **La IA local se instala y se descarga siempre** (clasificador, segmentación y profundidad, unos 750 MB) para que todo funcione sin internet |
 | `--no-auto-approve` | `-NoAutoApprove` | Que los agentes pregunten antes de cada herramienta |
 | `--with recon,depth` | `-With recon,depth` | Extras a elección |
 | `--device auto\|cpu\|gpu` | `-Device …` | Dispositivo por defecto |

{reverse3d-0.8.0 → reverse3d-0.8.1}/reverse3d.egg-info/SOURCES.txt

@@ -60,6 +60,7 @@ tests/test_authorship.py
 tests/test_automation.py
 tests/test_cli.py
 tests/test_cli_automation.py
+tests/test_color.py
 tests/test_config.py
 tests/test_geometry_fit.py
 tests/test_heic.py

reverse3d-0.8.1/tests/test_color.py

@@ -0,0 +1,49 @@
+from __future__ import annotations
+
+from pathlib import Path
+
+import numpy as np
+import trimesh
+
+from reverse3d.config import Settings
+from reverse3d.recipes import _clean, _color_from_cloud, _has_vertex_colors, _Run
+
+
+def _two_tone_cloud(path: Path) -> Path:
+    """Points on a sphere: red on top, blue at the bottom (like a COLMAP fused.ply)."""
+    pts = trimesh.creation.icosphere(subdivisions=5).vertices
+    rgb = np.where(pts[:, 2:3] > 0, [220, 30, 30], [30, 30, 220])
+    trimesh.PointCloud(pts, colors=np.column_stack([rgb, np.full(len(pts), 255)])).export(path)
+    return path
+
+
+def test_mesh_takes_the_colours_of_its_point_cloud(tmp_path: Path) -> None:
+    plain = tmp_path / "mesh.ply"
+    trimesh.creation.icosphere(subdivisions=3).export(plain)
+    assert not _has_vertex_colors(plain)
+    colored = _color_from_cloud(plain, _two_tone_cloud(tmp_path / "cloud.ply"), tmp_path / "c.ply")
+    assert colored is not None and _has_vertex_colors(colored)
+    mesh = trimesh.load(colored, process=False)
+    top = mesh.vertices[:, 2] > 0.3
+    assert mesh.visual.vertex_colors[top, 0].mean() > 150  # red on top
+    assert mesh.visual.vertex_colors[~top & (mesh.vertices[:, 2] < -0.3), 2].mean() > 150
+
+
+def test_cleaning_a_coloured_mesh_also_writes_glb(tmp_path: Path, settings: Settings) -> None:
+    plain = tmp_path / "mesh.ply"
+    trimesh.creation.icosphere(subdivisions=3).export(plain)
+    colored = _color_from_cloud(plain, _two_tone_cloud(tmp_path / "cloud.ply"), tmp_path / "c.ply")
+    assert colored is not None
+    run = _Run(recipe="photos_to_model", settings=settings, out=tmp_path)
+    _clean(run, colored, tmp_path / "model.stl", scale=1.0, overwrite=True)
+    assert {"model", "model_color", "model_glb"} <= set(run.outputs)
+    glb = trimesh.load(run.outputs["model_glb"], force="mesh")
+    assert len(np.unique(np.asarray(glb.visual.vertex_colors)[:, :3], axis=0)) > 1
+
+
+def test_plain_meshes_get_no_colour_outputs(tmp_path: Path, settings: Settings) -> None:
+    plain = tmp_path / "mesh.ply"
+    trimesh.creation.icosphere(subdivisions=3).export(plain)
+    run = _Run(recipe="mesh_report", settings=settings, out=tmp_path)
+    _clean(run, plain, tmp_path / "model.stl", scale=1.0, overwrite=True)
+    assert "model_glb" not in run.outputs

reverse3d-0.8.0/reverse3d/_version.py

@@ -1 +0,0 @@
-__version__ = "0.8.0"