Implement blocking and migration guide (#6)

* Implement blocking functionality

* Add migration guide

Author: constantinpape

.gitignore

@@ -27,3 +27,5 @@ venv/
 
 # Data
 *.h5
+
+CLAUDE.md

AGENTS.md

@@ -435,6 +435,11 @@ Public functions should have concise docstrings documenting:
 - background-label behavior, if relevant
 - axis and coordinate conventions
 
+When adding or changing public functionality, update `MIGRATION_GUIDE.md` so
+users migrating from `nifty`, `affogato`, or related packages know the
+corresponding `bioimage-cpp` API, important behavioral differences, and any
+intentional improvements.
+
 ## What an agent should do when modifying this repository
 
 When adding or changing code:

CMakeLists.txt

@@ -11,6 +11,7 @@ find_package(nanobind CONFIG REQUIRED)
 
 nanobind_add_module(_core
     NB_STATIC
+    src/bindings/blocking.cxx
     src/bindings/module.cxx
     src/bindings/graph.cxx
     src/bindings/segmentation.cxx

MIGRATION_GUIDE.md

@@ -0,0 +1,293 @@
+# Migration Guide
+
+This guide explains how to migrate code that used selected `nifty` or
+`affogato` functionality to `bioimage-cpp`.
+
+`bioimage-cpp` is not a drop-in compatibility layer. The package keeps a
+smaller, NumPy-first API with Python-style method names, explicit dtype support,
+and no I/O dependencies in the C++ core.
+
+## Imports
+
+Use:
+
+```python
+import bioimage_cpp as bic
+```
+
+Graph functionality is under `bic.graph`, segmentation functionality is under
+`bic.segmentation`, and small utility functions are under `bic.utils`.
+
+## Blocking
+
+Nifty:
+
+```python
+import nifty.tools as nt
+
+blocking = nt.blocking([0, 0], [100, 80], [32, 32])
+block = blocking.getBlock(0)
+block_with_halo = blocking.getBlockWithHalo(0, [8, 8])
+```
+
+bioimage-cpp:
+
+```python
+import bioimage_cpp as bic
+
+blocking = bic.Blocking([0, 0], [100, 80], [32, 32])
+block = blocking.get_block(0)
+block_with_halo = blocking.get_block_with_halo(0, [8, 8])
+```
+
+Name changes:
+
+| nifty-style name | bioimage-cpp name |
+| --- | --- |
+| `roiBegin` | `roi_begin` |
+| `roiEnd` | `roi_end` |
+| `blockShape` | `block_shape` |
+| `blockShift` | `block_shift` |
+| `blocksPerAxis` | `blocks_per_axis` |
+| `numberOfBlocks` | `number_of_blocks` |
+| `blockGridPosition` | `block_grid_position` |
+| `getNeighborId` | `get_neighbor_id` |
+| `getBlock` | `get_block` |
+| `getBlockWithHalo` | `get_block_with_halo` |
+| `addHalo` | `add_halo` |
+| `coordinatesToBlockId` | `coordinates_to_block_id` |
+| `getBlockIdsInBoundingBox` | `get_block_ids_in_bounding_box` |
+| `getBlockIdsOverlappingBoundingBox` | `get_block_ids_overlapping_bounding_box` |
+| `getLocalOverlaps` | `get_local_overlaps` |
+| `getBlockIdsInSlice` | `get_block_ids_in_slice` |
+
+Intentional improvements over nifty:
+
+- `coordinates_to_block_id` accounts for both `roi_begin` and `block_shift`.
+- `get_block_ids_overlapping_bounding_box` works for any dimensionality, not
+  only 3D.
+- Bounding boxes use NumPy-style half-open intervals: `[begin, end)`.
+- `get_local_overlaps` returns `None` if blocks do not overlap; otherwise it
+  returns `(begin_a, end_a, begin_b, end_b)` in local coordinates.
+
+## Undirected Graphs
+
+Nifty:
+
+```python
+import nifty.graph as ng
+
+graph = ng.undirectedGraph(4)
+edge_id = graph.insertEdge(0, 1)
+uvs = graph.uvIds()
+```
+
+bioimage-cpp:
+
+```python
+import bioimage_cpp as bic
+
+graph = bic.graph.UndirectedGraph(4)
+edge_id = graph.insert_edge(0, 1)
+uvs = graph.uv_ids()
+```
+
+The convenience constructor is:
+
+```python
+graph = bic.graph.undirected_graph(4)
+graph = bic.graph.UndirectedGraph.from_edges(4, [[0, 1], [1, 2]])
+```
+
+Important differences:
+
+- Nodes are fixed at construction and have ids `0 .. number_of_nodes - 1`.
+- Re-inserting an existing undirected edge returns the existing edge id.
+- Bulk methods accept array-like inputs and return NumPy arrays.
+- Python-style names are preferred. A few nifty-style aliases are still present
+  on `UndirectedGraph` for convenience, but new code should use snake_case.
+
+Common method/property mapping:
+
+| nifty-style name | bioimage-cpp name |
+| --- | --- |
+| `numberOfNodes` | `number_of_nodes` |
+| `numberOfEdges` | `number_of_edges` |
+| `nodeIdUpperBound` | `node_id_upper_bound` |
+| `edgeIdUpperBound` | `edge_id_upper_bound` |
+| `insertEdge` | `insert_edge` |
+| `insertEdges` | `insert_edges` |
+| `findEdge` | `find_edge` |
+| `findEdges` | `find_edges` |
+| `uvIds` | `uv_ids` |
+| `nodeAdjacency` | `node_adjacency` |
+| `serializationSize` | `serialization_size` |
+| `extractSubgraphFromNodes` | `extract_subgraph_from_nodes` |
+| `edgesFromNodeList` | `edges_from_node_list` |
+
+## Region Adjacency Graphs
+
+Nifty:
+
+```python
+import nifty.graph.rag as nrag
+
+rag = nrag.gridRag(labels)
+uvs = rag.uvIds()
+```
+
+bioimage-cpp:
+
+```python
+import bioimage_cpp as bic
+
+rag = bic.graph.region_adjacency_graph(labels)
+uvs = rag.uv_ids()
+```
+
+Notes:
+
+- Supported label dtypes are `uint32`, `uint64`, `int32`, and `int64`.
+- Labels must be 2D or 3D.
+- Negative signed labels are rejected.
+- Nodes correspond to label ids from `0` to `labels.max()`.
+- Edge ids are deterministic; RAG edges are sorted lexicographically by
+  endpoint ids.
+- Non-contiguous labels are copied to contiguous memory before entering C++.
+
+## RAG Boundary and Affinity Features
+
+Nifty has RAG feature helpers such as `accumulateEdgeMeanAndLength`,
+`accumulateEdgeStandartFeatures`, and affinity feature accumulation helpers.
+In `bioimage-cpp`, these are exposed as explicit NumPy-returning functions.
+
+Simple edge-map features:
+
+```python
+rag = bic.graph.region_adjacency_graph(labels)
+features = bic.graph.edge_map_features(rag, labels, edge_map)
+```
+
+The columns are:
+
+```python
+bic.graph.SIMPLE_EDGE_FEATURE_NAMES
+# ("mean", "size")
+```
+
+Complex edge-map features:
+
+```python
+features = bic.graph.edge_map_features_complex(rag, labels, edge_map)
+```
+
+The columns are:
+
+```python
+bic.graph.COMPLEX_EDGE_FEATURE_NAMES
+# ("mean", "median", "std", "min", "max", "p5", "p10",
+#  "p25", "p75", "p90", "p95", "size")
+```
+
+Affinity features:
+
+```python
+features = bic.graph.affinity_features(
+    rag,
+    labels,
+    affinities,
+    offsets=[[0, 1], [1, 0]],
+)
+```
+
+Complex affinity features:
+
+```python
+features = bic.graph.affinity_features_complex(
+    rag,
+    labels,
+    affinities,
+    offsets=[[0, 1], [1, 0]],
+)
+```
+
+Notes:
+
+- `edge_map` must have the same shape as `labels`.
+- `affinities` must have shape `(channels, *labels.shape)`.
+- `offsets` must have one offset per channel in NumPy axis order.
+- Feature arrays use `float64` output.
+- `number_of_threads=0` uses the library default; pass a positive integer for a
+  fixed thread count.
+
+## Mutex Watershed
+
+Affogato:
+
+```python
+from affogato.segmentation import compute_mws_segmentation
+
+seg = compute_mws_segmentation(
+    weights,
+    offsets,
+    number_of_attractive_channels=3,
+    strides=[1, 1, 1],
+)
+```
+
+bioimage-cpp:
+
+```python
+import bioimage_cpp as bic
+
+seg = bic.segmentation.mutex_watershed(
+    weights,
+    offsets,
+    number_of_attractive_channels=3,
+    strides=[1, 1, 1],
+)
+```
+
+Important migration notes:
+
+- `bioimage-cpp` expects the first `number_of_attractive_channels` channels to
+  be attractive merge-edge weights and the remaining channels to be mutex-edge
+  weights.
+- Supported affinity dtypes are `float32` and `float64`.
+- Inputs must represent 2D or 3D grids with shapes `(channels, y, x)` or
+  `(channels, z, y, x)`.
+- Non-contiguous affinity arrays are copied to contiguous memory.
+- `strides` sub-sample mutex edges only; attractive edges are always kept.
+- `randomized_strides=True` uses NumPy's global random state, so existing
+  `np.random.seed(...)` workflows remain deterministic.
+- A boolean `mask` may be passed. Edges touching `False` pixels are ignored and
+  masked pixels are set to label `0`.
+- Output labels are `uint64`, consecutive, and 1-based for foreground pixels.
+
+## Dictionary-Based Relabeling
+
+If you used a small helper to apply a dictionary to an integer label array, use
+`take_dict`:
+
+```python
+labels = np.array([1, 3, 2, 1], dtype=np.uint64)
+relabeling = {1: 10, 2: 20, 3: 30}
+
+out = bic.utils.take_dict(relabeling, labels)
+```
+
+Notes:
+
+- Supported input dtypes are `uint32`, `uint64`, `int32`, and `int64`.
+- Output has the same shape and dtype as the input array.
+- Every value in the input must be present in the mapping.
+- Non-contiguous inputs are copied before entering C++.
+
+## I/O and Build Dependencies
+
+`bioimage-cpp` intentionally does not replace nifty or affogato I/O helpers.
+Load TIFF, HDF5, zarr, N5, OME-NGFF, and related formats with existing Python
+libraries, then pass NumPy arrays to `bioimage-cpp`.
+
+The package is designed for small PyPI wheels and does not depend on nifty,
+vigra, HDF5, z5, xtensor, pybind11, or other large C++ libraries.