Rand cmap

Created function that creates a rand_cmap object. Set as default for plot_cc function

Link: https://www.notion.so/qim-dtu/Colormaps-for-qim3d-b8500b578dd54278aa03f6a2866df97d?pvs=4

Code to run

import qim3d
qim3d.io.logger.level("info")
#Prepare data
vol = qim3d.examples.cement_128x128x128[50:150]
binary = qim3d.processing.filters.gaussian(vol, 2)<60
binary = qim3d.processing.filters.median(binary, size=6)
qim3d.viz.slices(binary)
cc = qim3d.utils.cc.get_3d_cc(binary)
qim3d.viz.slices(cc.get_cc(), cmap=qim3dCmap(len(cc)), interpolation="none")
qim3d.viz.img.plot_cc(cc, crop=True, show=True, overlay=None, n_slices=5, max_cc_to_plot=5)

qim3d/utils/qim3d_cmap.py

@@ -6,7 +6,7 @@ from matplotlib.colors import LinearSegmentedColormap
 from qim3d.io.logger import log
 
 
-def qim3dCmap(nlabels, type="bright", first_color_black=True, last_color_black=False):
+def qim3dCmap(nlabels, type="bright", first_color_black=True, last_color_black=False,seed=19):
     """
     Creates a random colormap to be used together with matplotlib. Useful for segmentation tasks
 
@@ -15,6 +15,7 @@ def qim3dCmap(nlabels, type="bright", first_color_black=True, last_color_black=F
         type (str, optional): 'bright' for strong colors, 'soft' for pastel colors. Defaults to 'bright'.
         first_color_black (bool, optional): Option to use first color as black. Defaults to True.
         last_color_black (bool, optional): Option to use last color as black. Defaults to False.
+        seed (int, optional): Seed for random number generator. Defaults to 19.
 
     Returns:
         matplotlib.colors.LinearSegmentedColormap: Colormap for matplotlib
@@ -24,12 +25,14 @@ def qim3dCmap(nlabels, type="bright", first_color_black=True, last_color_black=F
         return
 
     # Generate color map for bright colors, based on hsv
+    # Create a new random generator, to locally set seed
+    rng = np.random.default_rng(seed)
     if type == "bright":
         randHSVcolors = [
             (
-                np.random.uniform(low=0.0, high=1),
-                np.random.uniform(low=0.2, high=1),
-                np.random.uniform(low=0.9, high=1),
+                rng.uniform(low=0.0, high=1),
+                rng.uniform(low=0.2, high=1),
+                rng.uniform(low=0.9, high=1),
             )
             for i in range(nlabels)
         ]
@@ -57,9 +60,9 @@ def qim3dCmap(nlabels, type="bright", first_color_black=True, last_color_black=F
         high = 0.95
         randRGBcolors = [
             (
-                np.random.uniform(low=low, high=high),
-                np.random.uniform(low=low, high=high),
-                np.random.uniform(low=low, high=high),
+                rng.uniform(low=low, high=high),
+                rng.uniform(low=low, high=high),
+                rng.uniform(low=low, high=high),
             )
             for i in range(nlabels)
         ]