wildflow-splat 0.1.3__tar.gz → 0.1.4__tar.gz

{wildflow_splat-0.1.3 → wildflow_splat-0.1.4}/Cargo.lock

@@ -4,21 +4,21 @@ version = 4
 
 [[package]]
 name = "autocfg"
-version = "1.4.0"
+version = "1.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ace50bade8e6234aa140d9a2f552bbee1db4d353f69b8217bc503490fc1a9f26"
+checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
 
 [[package]]
 name = "bumpalo"
-version = "3.18.0"
+version = "3.19.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c1b094a32014c3d1f3944e4808e0e7c70e97dae0660886a8eb6dbc52d745badc"
+checksum = "46c5e41b57b8bba42a04676d81cb89e9ee8e859a1a66f80a5a72e1cb76b34d43"
 
 [[package]]
 name = "cfg-if"
-version = "1.0.0"
+version = "1.0.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
+checksum = "9555578bc9e57714c812a1f84e4fc5b4d21fcb063490c624de019f7464c91268"
 
 [[package]]
 name = "console"
@@ -93,9 +93,9 @@ dependencies = [
 
 [[package]]
 name = "libc"
-version = "0.2.172"
+version = "0.2.174"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d750af042f7ef4f724306de029d18836c26c1765a54a6a3f094cbd23a7267ffa"
+checksum = "1171693293099992e19cddea4e8b849964e9846f4acee11b3948bcc337be8776"
 
 [[package]]
 name = "log"
@@ -105,9 +105,9 @@ checksum = "13dc2df351e3202783a1fe0d44375f7295ffb4049267b0f3018346dc122a1d94"
 
 [[package]]
 name = "memchr"
-version = "2.7.4"
+version = "2.7.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
+checksum = "32a282da65faaf38286cf3be983213fcf1d2e2a58700e808f83f4ea9a4804bc0"
 
 [[package]]
 name = "memoffset"
@@ -132,9 +132,9 @@ checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
 
 [[package]]
 name = "portable-atomic"
-version = "1.11.0"
+version = "1.11.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "350e9b48cbc6b0e028b0473b114454c6316e57336ee184ceab6e53f72c178b3e"
+checksum = "f84267b20a16ea918e43c6a88433c2d54fa145c92a811b5b047ccbe153674483"
 
 [[package]]
 name = "ppv-lite86"
@@ -156,9 +156,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3"
-version = "0.25.0"
+version = "0.25.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f239d656363bcee73afef85277f1b281e8ac6212a1d42aa90e55b90ed43c47a4"
+checksum = "8970a78afe0628a3e3430376fc5fd76b6b45c4d43360ffd6cdd40bdde72b682a"
 dependencies = [
  "indoc",
  "libc",
@@ -173,9 +173,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-build-config"
-version = "0.25.0"
+version = "0.25.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "755ea671a1c34044fa165247aaf6f419ca39caa6003aee791a0df2713d8f1b6d"
+checksum = "458eb0c55e7ece017adeba38f2248ff3ac615e53660d7c71a238d7d2a01c7598"
 dependencies = [
  "once_cell",
  "target-lexicon",
@@ -183,9 +183,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-ffi"
-version = "0.25.0"
+version = "0.25.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "fc95a2e67091e44791d4ea300ff744be5293f394f1bafd9f78c080814d35956e"
+checksum = "7114fe5457c61b276ab77c5055f206295b812608083644a5c5b2640c3102565c"
 dependencies = [
  "libc",
  "pyo3-build-config",
@@ -193,9 +193,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-macros"
-version = "0.25.0"
+version = "0.25.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a179641d1b93920829a62f15e87c0ed791b6c8db2271ba0fd7c2686090510214"
+checksum = "a8725c0a622b374d6cb051d11a0983786448f7785336139c3c94f5aa6bef7e50"
 dependencies = [
  "proc-macro2",
  "pyo3-macros-backend",
@@ -205,9 +205,9 @@ dependencies = [
 
 [[package]]
 name = "pyo3-macros-backend"
-version = "0.25.0"
+version = "0.25.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "9dff85ebcaab8c441b0e3f7ae40a6963ecea8a9f5e74f647e33fcf5ec9a1e89e"
+checksum = "4109984c22491085343c05b0dbc54ddc405c3cf7b4374fc533f5c3313a572ccc"
 dependencies = [
  "heck",
  "proc-macro2",
@@ -295,9 +295,9 @@ dependencies = [
 
 [[package]]
 name = "syn"
-version = "2.0.101"
+version = "2.0.104"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "8ce2b7fc941b3a24138a0a7cf8e858bfc6a992e7978a068a5c760deb0ed43caf"
+checksum = "17b6f705963418cdb9927482fa304bc562ece2fdd4f616084c50b7023b435a40"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -318,9 +318,9 @@ checksum = "5a5f39404a5da50712a4c1eecf25e90dd62b613502b7e925fd4e4d19b5c96512"
 
 [[package]]
 name = "unicode-width"
-version = "0.2.0"
+version = "0.2.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "1fc81956842c57dac11422a97c3b8195a1ff727f06e85c84ed2e8aa277c9a0fd"
+checksum = "4a1a07cc7db3810833284e8d372ccdc6da29741639ecc70c9ec107df0fa6154c"
 
 [[package]]
 name = "unindent"
@@ -330,9 +330,9 @@ checksum = "7264e107f553ccae879d21fbea1d6724ac785e8c3bfc762137959b5802826ef3"
 
 [[package]]
 name = "wasi"
-version = "0.11.0+wasi-snapshot-preview1"
+version = "0.11.1+wasi-snapshot-preview1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
+checksum = "ccf3ec651a847eb01de73ccad15eb7d99f80485de043efb2f370cd654f4ea44b"
 
 [[package]]
 name = "wasm-bindgen"
@@ -487,18 +487,18 @@ checksum = "589f6da84c646204747d1270a2a5661ea66ed1cced2631d546fdfb155959f9ec"
 
 [[package]]
 name = "zerocopy"
-version = "0.8.25"
+version = "0.8.26"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a1702d9583232ddb9174e01bb7c15a2ab8fb1bc6f227aa1233858c351a3ba0cb"
+checksum = "1039dd0d3c310cf05de012d8a39ff557cb0d23087fd44cad61df08fc31907a2f"
 dependencies = [
  "zerocopy-derive",
 ]
 
 [[package]]
 name = "zerocopy-derive"
-version = "0.8.25"
+version = "0.8.26"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "28a6e20d751156648aa063f3800b706ee209a32c0b4d9f24be3d980b01be55ef"
+checksum = "9ecf5b4cc5364572d7f4c329661bcc82724222973f2cab6f050a4e5c22f75181"
 dependencies = [
  "proc-macro2",
  "quote",

wildflow_splat-0.1.4/PKG-INFO

@@ -0,0 +1,74 @@
+Metadata-Version: 2.4
+Name: wildflow-splat
+Version: 0.1.4
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Rust
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Multimedia :: Graphics :: 3D Modeling
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: maturin>=1.0 ; extra == 'dev'
+Requires-Dist: pytest>=7.0 ; extra == 'dev'
+Provides-Extra: dev
+License-File: LICENSE
+Summary: Fast PLY point cloud processing for 3D Gaussian splatting workflows
+Keywords: ply,point-cloud,3d-gaussian-splatting,colmap,photogrammetry,computer-vision
+Author-email: Wildflow AI <info@wildflow.ai>
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: repository, https://github.com/wildflowai/wildflow-splat
+Project-URL: documentation, https://docs.wildflow.ai
+
+Super hacky library to work with coral reef splats.
+
+You can swim with a few GoPros around a reef (e.g. [wildflow.ai/protocol](https://wildflow.ai/protocol)) and then turn the footage into 3D models (e.g. [wildflow.ai/demo](https://wildflow.ai/demo)) to track changes over time, run different analysis on top of it, and ultimately see which conservation/restoration methods work best.
+
+This is a bunch of primitives to process the data.
+
+# Usage
+Install with
+```
+pip install wildflow
+```
+So you can play with it from python:
+```py
+from wildflow import splat
+splat.split(...)
+```
+# Workflow
+
+## SfM workflow
+Turns images from cameras 3D point cloud and 
+
+![](/images/wildflow-3dgs-wf.svg)
+
+# Local Development
+
+This library uses Rust extensions built with Maturin. To set up locally:
+
+```bash
+# Create virtual environment
+python3 -m venv venv
+source venv/bin/activate
+
+# Install Rust (if not already installed)
+curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+source "$HOME/.cargo/env"
+
+# Install dependencies and build
+pip install maturin
+pip install -r requirements.txt
+maturin develop
+```
+
+After making changes to Rust code, rebuild with `maturin develop`.
+

wildflow_splat-0.1.4/README.md

@@ -0,0 +1,43 @@
+Super hacky library to work with coral reef splats.
+
+You can swim with a few GoPros around a reef (e.g. [wildflow.ai/protocol](https://wildflow.ai/protocol)) and then turn the footage into 3D models (e.g. [wildflow.ai/demo](https://wildflow.ai/demo)) to track changes over time, run different analysis on top of it, and ultimately see which conservation/restoration methods work best.
+
+This is a bunch of primitives to process the data.
+
+# Usage
+Install with
+```
+pip install wildflow
+```
+So you can play with it from python:
+```py
+from wildflow import splat
+splat.split(...)
+```
+# Workflow
+
+## SfM workflow
+Turns images from cameras 3D point cloud and 
+
+![](/images/wildflow-3dgs-wf.svg)
+
+# Local Development
+
+This library uses Rust extensions built with Maturin. To set up locally:
+
+```bash
+# Create virtual environment
+python3 -m venv venv
+source venv/bin/activate
+
+# Install Rust (if not already installed)
+curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+source "$HOME/.cargo/env"
+
+# Install dependencies and build
+pip install maturin
+pip install -r requirements.txt
+maturin develop
+```
+
+After making changes to Rust code, rebuild with `maturin develop`.

wildflow_splat-0.1.4/examples/patches_example.py

@@ -0,0 +1,140 @@
+"""
+Example usage of wildflow.splat.patches library for COLMAP camera partitioning.
+Interactive sliders allow you to explore how different parameters affect the partitioning.
+"""
+
+import pycolmap
+import matplotlib.pyplot as plt
+import matplotlib.patches as patches
+from matplotlib.widgets import Slider
+import math
+from wildflow.splat import patches as splat_patches, BoundingBox
+
+# Constants
+COLMAP_FOLDER = "/Users/nsv/corals/soneva/0/"
+MAX_CAMERAS = 50
+BUFFER_METERS = 0.0
+TARGET_BINS = 100
+
+def rotate30(cameras: list[tuple[float, float]]) -> list[tuple[float, float]]:
+    """Rotate camera positions by 30 degrees counter-clockwise."""
+    angle_rad = math.radians(30)
+    cos_angle = math.cos(angle_rad)
+    sin_angle = math.sin(angle_rad)
+    
+    rotated = []
+    for x, y in cameras:
+        x_rot = x * cos_angle - y * sin_angle
+        y_rot = x * sin_angle + y * cos_angle
+        rotated.append((x_rot, y_rot))
+    
+    return rotated
+
+def create_interactive_plot(cameras: list[tuple[float, float]]) -> None:
+    """Create interactive plot with sliders for parameter adjustment."""
+    # Create figure with space for sliders at bottom (smaller to fit on screen)
+    fig = plt.figure(figsize=(12, 9))
+    
+    # Main plot takes up most of the space
+    ax_main = plt.axes((0.1, 0.3, 0.8, 0.6))
+    
+    # Create slider axes with proper spacing
+    ax_max_cameras = plt.axes((0.2, 0.2, 0.6, 0.03))
+    ax_buffer = plt.axes((0.2, 0.15, 0.6, 0.03))
+    ax_target_bins = plt.axes((0.2, 0.1, 0.6, 0.03))
+    
+    # Create sliders with nice snapping values
+    slider_max_cameras = Slider(ax_max_cameras, 'Max Cameras', 1, 2000, valinit=MAX_CAMERAS, valfmt='%d')
+    # Buffer slider with 100 beautiful steps (0.02 increments)
+    slider_buffer = Slider(ax_buffer, 'Buffer (m)', 0.0, 2.0, valinit=BUFFER_METERS, valfmt='%.2f', 
+                          valstep=0.02)
+    slider_target_bins = Slider(ax_target_bins, 'Target Bins', 10, 300, valinit=TARGET_BINS, valfmt='%d')
+    
+    def update_plot(val=None):
+        """Update the plot when sliders change."""
+        # Clear the main plot
+        ax_main.clear()
+        
+        # Get current slider values
+        max_cameras = int(slider_max_cameras.val)
+        buffer_meters = slider_buffer.val
+        target_bins = int(slider_target_bins.val)
+        
+        # Run partitioning with current parameters
+        try:
+            bboxes = splat_patches(
+                cameras, 
+                max_cameras=max_cameras,
+                buffer_meters=buffer_meters,
+                target_bins=target_bins
+            )
+            
+            # Plot camera positions
+            x_coords = [pos[0] for pos in cameras]
+            y_coords = [pos[1] for pos in cameras]
+            ax_main.scatter(x_coords, y_coords, c='red', s=20, alpha=0.7,
+                           edgecolors='black', linewidth=0.3, zorder=2)
+            
+            # Draw bounding boxes
+            for i, bbox in enumerate(bboxes):
+                rect = patches.Rectangle(
+                    (bbox.min_x, bbox.min_y), bbox.width, bbox.height,
+                    linewidth=2, edgecolor='blue', facecolor='lightblue', alpha=0.3, zorder=1
+                )
+                ax_main.add_patch(rect)
+                
+                # Add box number
+                center_x = bbox.min_x + bbox.width / 2
+                center_y = bbox.min_y + bbox.height / 2
+                ax_main.text(center_x, center_y, str(i+1),
+                            horizontalalignment='center', verticalalignment='center',
+                            fontsize=12, color='blue', fontweight='bold', zorder=3,
+                            bbox=dict(boxstyle="round,pad=0.3", facecolor='white', alpha=0.8))
+            
+            # Update title with current parameters
+            ax_main.set_title(f'Camera Partitioning: {len(bboxes)} patches '
+                             f'(max_cameras={max_cameras}, buffer={buffer_meters:.2f}m, bins={target_bins})')
+            
+        except Exception as e:
+            # Show error message if partitioning fails
+            ax_main.text(0.5, 0.5, f'Error: {str(e)}', 
+                        transform=ax_main.transAxes, ha='center', va='center',
+                        fontsize=14, color='red', bbox=dict(boxstyle="round", facecolor='yellow', alpha=0.8))
+            ax_main.set_title('Partitioning Failed - Try Different Parameters')
+        
+        ax_main.set_xlabel('X Position')
+        ax_main.set_ylabel('Y Position')
+        ax_main.grid(True, alpha=0.3)
+        ax_main.set_aspect('equal')
+        
+        # Redraw
+        fig.canvas.draw()
+    
+    # Connect sliders to update function
+    slider_max_cameras.on_changed(update_plot)
+    slider_buffer.on_changed(update_plot)
+    slider_target_bins.on_changed(update_plot)
+    
+    # Initial plot
+    update_plot()
+    
+    plt.show()
+
+def main():
+    # Load camera positions using the exact pattern from docstring
+    print("Loading camera positions...")
+    model = pycolmap.Reconstruction(COLMAP_FOLDER)
+    cameras = [(img.projection_center()[0], img.projection_center()[1]) 
+               for img in model.images.values()]
+    print(f"Loaded {len(cameras)} camera positions")
+    
+    # Rotate cameras (can be commented out easily)
+    cameras = rotate30(cameras)
+    
+    # Create interactive visualization
+    print("Creating interactive visualization...")
+    print("Use the sliders to adjust parameters and see how they affect the partitioning!")
+    create_interactive_plot(cameras)
+
+if __name__ == "__main__":
+    main() 

{wildflow_splat-0.1.3 → wildflow_splat-0.1.4}/meta-package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wildflow"
-version = "0.1.3"
+version = "0.1.4"
 description = "Modeling natural ecosystems"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -28,7 +28,7 @@ classifiers = [
     "Topic :: Software Development :: Libraries :: Python Modules",
 ]
 dependencies = [
-    "wildflow-splat>=0.1.3",
+    "wildflow-splat>=0.1.4",
 ]
 
 [project.urls]

{wildflow_splat-0.1.3 → wildflow_splat-0.1.4}/native/lib.rs

@@ -1,6 +1,7 @@
 use pyo3::prelude::*;
 
 mod split;
+mod patches;
 
 /// The main PyO3 module for wildflow.splat
 #[pymodule]
@@ -9,5 +10,10 @@ fn _core(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<split::Config>()?;
     m.add_class::<split::Patch>()?;
     m.add_function(wrap_pyfunction!(split::split_ply, m)?)?;
+    
+    // Export patches functionality
+    m.add_class::<patches::BoundingBox>()?;
+    m.add_function(wrap_pyfunction!(patches::patches, m)?)?;
+    
     Ok(())
 }