feat: add segmentation property objects

cryoet_data_portal_neuroglancer/models/json_generator.py

@@ -48,6 +48,33 @@ def __str__(self):
         return self.name.lower()
 
 
+@dataclass
+class SegmentPropertyJSONGenerator:
+    """Generates a JSON file for segmentation properties.
+
+    Supports a subset of the properties that can be set in Neuroglancer.
+    See https://github.com/google/neuroglancer/blob/3efc90465e702453916d2b03d472c16378848132/src/datasource/precomputed/segment_properties.md
+    """
+
+    ids: list[int]
+    labels: list[str]
+
+    def generate_json(self) -> dict:
+        return {
+            "inline": {
+                "ids": [str(val) for val in self.ids],
+                "properties": [
+                    {
+                        "values": self.labels,
+                        "type": "label",
+                        "id": "label",
+                    },
+                ],
+            },
+            "@type": "neuroglancer_segment_properties",
+        }
+
+
 @dataclass
 class RenderingJSONGenerator:
     """Generates a JSON file for Neuroglancer to read."""

cryoet_data_portal_neuroglancer/precompute/segmentation_properties.py

@@ -0,0 +1,12 @@
+import json
+from pathlib import Path
+
+from cryoet_data_portal_neuroglancer.models.json_generator import SegmentPropertyJSONGenerator
+
+
+def write_segment_properties(base_folder: str | Path, ids: list[int], labels: list[str]):
+    segment_generator = SegmentPropertyJSONGenerator(ids=ids, labels=labels)
+    segment_properties = segment_generator.generate_json()
+    segment_properties_path = Path(base_folder) / "segment_properties" / "info"
+    segment_properties_path.parent.mkdir(exist_ok=True, parents=True)
+    segment_properties_path.write_text(json.dumps(segment_properties, indent=2))