Implemented 3D connected components as wrapper class for scipy.ndimage.label

Ticket: https://www.notion.so/qim-dtu/3D-connected-components-ca5c410fef49444eac7c6efd0b56ead4?pvs=4

I have made a simple plot functions that takes the ConnectedComponents object and extracts the connected_components volume.

However I don't know if it is necessary as the user could do this manually.

def plot_connected_components(connected_components: ConnectedComponents, **kwargs):
    """ Plots connected components from the ConnectedComponents class as 2d slices.

    Args:
        connected_components (ConnectedComponents): The connected components class from the qim3d.utils.connected_components module.
        **kwargs: Additional keyword arguments to pass to qim3d.viz.img.slice_viz.

    Returns:
        matplotlib.pyplot: the 3D plot of the connected components.
    """
    fig = slice_viz(connected_components.connected_components, **kwargs)
    return fig

qim3d/utils/cc.py

@@ -16,6 +16,8 @@ class CC:
         """
         self._connected_components = connected_components
         self.cc_count = num_connected_components
+        
+        self.shape = connected_components.shape
     
     def __len__(self):
         """
@@ -37,7 +39,7 @@ class CC:
         if index is None:
             volume = self._connected_components
         else:
-            assert 1 <= index <= self.cc_count, "Index out of range."
+            assert 1 <= index <= self.cc_count, "Index out of range. Needs to be in range [1, cc_count]."
             volume = self._connected_components == index
             
         if crop: