Add Orion example

speedup_studies/compute_speedup.py

@@ -97,6 +97,26 @@ def getrms(cube, fr1=10, fr2=10):
             "min_delta": min_delta,
             "min_npix": min_npix,
         }
+    elif args["example"] == "orion":
+        # Open the cube
+        path = f"{__file__[: __file__.index(os.path.basename(__file__))]}../data/orion_12CO.fits"
+        hdu = fits.open(path)[0]
+        data = hdu.data
+        hd = hdu.header
+
+        # Survey designs
+        sigma = 0.3  # K, noise level
+        ppb = 1.3  # pixels/beam
+
+        data = np.array(data, dtype=float)
+
+        return {
+            "data": data,
+            "min_npix": 3 * ppb,
+            "min_delta": 2 * sigma,
+            "min_value": sigma,
+        }
+
     else:
         raise NotImplementedError
 

speedup_studies/timing_data/orion.json

@@ -0,0 +1,12 @@
+{
+    "v3": {
+        "1": 13.861861704994226,
+        "2": 13.693961618002504,
+        "4": 11.044378619990312,
+        "8": 23.26084513301612,
+        "16": 22.725151990016457
+    },
+    "astrodendro": {
+        "1": 9.042727890977403
+    }
+}

src/dendro/distributed_dendrogram_v3.py

@@ -292,7 +292,7 @@ def split_structure(self, structure, split_at):
         if not isinstance(structure._indices, np.ndarray):
             structure._indices = np.array(structure._indices)
 
-        top_mask = structure._values > split_at
+        top_mask = structure._values >= split_at
 
         bottom_part = Structure(
             indices=structure._indices[~top_mask],