add tests

Author: benjaminpkane

app/packages/looker/src/processOverlays.test.ts

@@ -0,0 +1,20 @@
+import { describe, expect, it } from "vitest";
+
+import HeatmapOverlay from "./overlays/heatmap";
+import SegmentationOverlay from "./overlays/segmentation";
+import { filter } from "./processOverlays";
+
+const EMPTY = {
+  id: "",
+  tags: [],
+};
+
+describe("test overlay processing", () => {
+  it("filters heatmap without a map", () => {
+    expect(filter(new HeatmapOverlay("test", EMPTY), {})).toBe(true);
+  });
+
+  it("filters segmentations without a mask", () => {
+    expect(filter(new SegmentationOverlay("test", EMPTY), {})).toBe(true);
+  });
+});

app/packages/looker/src/processOverlays.ts

@@ -2,11 +2,12 @@
  * Copyright 2017-2025, Voxel51, Inc.
  */
 
-import { CONTAINS, Overlay } from "./overlays/base";
+import type { Overlay } from "./overlays/base";
+import { CONTAINS } from "./overlays/base";
 import { ClassificationsOverlay } from "./overlays/classifications";
 import HeatmapOverlay from "./overlays/heatmap";
 import SegmentationOverlay from "./overlays/segmentation";
-import { BaseState } from "./state";
+import type { BaseState } from "./state";
 import { rotate } from "./util";
 
 const processOverlays = <State extends BaseState>(
@@ -30,25 +31,17 @@ const processOverlays = <State extends BaseState>(
       continue;
     }
 
-    if (!(overlay.field && overlay.field in bins)) continue;
-
-    // todo: find a better approach / place for this.
-    // for instance, this won't work in detection overlay, where
-    // we might want the bounding boxes but masks might not have been loaded
-    if (overlay instanceof SegmentationOverlay && !overlay.label.mask) {
-      continue;
-    }
-
-    if (overlay instanceof HeatmapOverlay && !overlay.label.map) {
-      continue;
-    }
-
     if (!overlay.isShown(state)) continue;
 
+    if (filter(overlay, bins)) continue;
+
     bins[overlay.field].push(overlay);
   }
 
-  let ordered = activePaths.reduce((acc, cur) => [...acc, ...bins[cur]], []);
+  let ordered = activePaths.reduce((acc, cur) => {
+    acc.push(...bins[cur]);
+    return acc;
+  }, []);
 
   if (classifications && !state.config.thumbnail) {
     ordered = [classifications, ...ordered];
@@ -89,4 +82,26 @@ const processOverlays = <State extends BaseState>(
   return [[...contained, ...outside], newRotate];
 };
 
+export const filter = <State extends BaseState>(
+  overlay: Overlay<State>,
+  bins: {
+    [k: string]: Overlay<State>[];
+  }
+) => {
+  if (!(overlay.field && overlay.field in bins)) return true;
+
+  // todo: find a better approach / place for this.
+  // for instance, this won't work in detection overlay, where
+  // we might want the bounding boxes but masks might not have been loaded
+  if (overlay instanceof HeatmapOverlay && !overlay.label.map) {
+    return true;
+  }
+
+  if (overlay instanceof SegmentationOverlay && !overlay.label.mask) {
+    return true;
+  }
+
+  return false;
+};
+
 export default processOverlays;