add bounding box property for synapses

chunkflow/lib/bounding_boxes.py

@@ -48,6 +48,22 @@ def adjust(self, size: Union[int, tuple, list, Vec]):
         self.maxpt += size
         return self
 
+    def union(self, bbox2: Union[BoundingBox, Bbox]):
+        """Merge another bounding box
+
+        Args:
+            bbox2 (Union[BoundingBox, Bbox]): another bounding box
+
+        Returns:
+            BoundingBox: bounding box after merging
+        """
+        if isinstance(bbox2, BoundingBox):
+            assert self.voxel_size == bbox2.voxel_size
+
+        self.minpt = np.minimum(self.minpt, bbox2.minpt)
+        self.maxpt = np.maximum(self.maxpt, bbox2.maxpt)
+        return self
+
     def contains(self, point: Union[tuple, Vec, list]):
         assert 3 == len(point)
         return np.all(np.asarray(

chunkflow/lib/synapses.py

@@ -7,6 +7,8 @@
 import numpy as np
 import h5py
 
+from .bounding_boxes import BoundingBox
+
 
 class Synapses():
     def __init__(self, pre: np.ndarray, pre_confidence: np.ndarray = None, 
@@ -176,8 +178,14 @@ def pre_with_physical_coordinate(self):
             return self.pre * self.resolution
         else:
             return self.pre
-
-
+
+    @property
+    def bounding_box(self) -> BoundingBox:
+        bbox = BoundingBox.from_points(self.pre)
+        bbox_post = BoundingBox.from_points(self.post[:, 1:])
+        bbox.union(bbox_post)
+        return bbox
+
     @property
     def post_with_physical_coordinate(self):
         """ post synapses with physical coordinate. Note that the first column is the index of