From 78744c5ec9debbb4930d74fd99e33b41606d1953 Mon Sep 17 00:00:00 2001
From: Diogo Correia <diogo@metacell.us>
Date: Mon, 14 Oct 2024 19:28:06 +0100
Subject: [PATCH] CELE-107 Initialize segment with style depending on selected
 neurons

---
 .../viewers/EM/EMStackTilesViewer.tsx         | 67 +++++++++++++------
 1 file changed, 47 insertions(+), 20 deletions(-)

diff --git a/applications/visualizer/frontend/src/components/viewers/EM/EMStackTilesViewer.tsx b/applications/visualizer/frontend/src/components/viewers/EM/EMStackTilesViewer.tsx
index 65ffd06e..b3e3c053 100644
--- a/applications/visualizer/frontend/src/components/viewers/EM/EMStackTilesViewer.tsx
+++ b/applications/visualizer/frontend/src/components/viewers/EM/EMStackTilesViewer.tsx
@@ -17,10 +17,10 @@ import Stroke from "ol/style/Stroke";
 import Style from "ol/style/Style";
 import Text from "ol/style/Text";
 import { TileGrid } from "ol/tilegrid";
-import { useEffect, useMemo, useRef } from "react";
+import { useCallback, useEffect, useMemo, useRef } from "react";
 import { useGlobalContext } from "../../../contexts/GlobalContext.tsx";
 import { SlidingRing } from "../../../helpers/slidingRing";
-import { getEMDataURL, getSegmentationURL } from "../../../models/models.ts";
+import { getEMDataURL, getSegmentationURL, ViewerType } from "../../../models/models.ts";
 import type { Dataset } from "../../../rest/index.ts";
 import SceneControls from "./SceneControls.tsx";
 
@@ -40,16 +40,12 @@ const getFeatureStyle = (feature: FeatureLike) => {
   });
 };
 
-const resetStyle = (feature: Feature) => {
-  feature.setStyle(getFeatureStyle(feature));
-};
-
-const setHighlightStyle = (feature: Feature) => {
+const getHighlightFeatureStyle = (feature: FeatureLike) => {
   const opacity = 0.5;
   const [r, g, b] = feature.get("color");
   const rgbaColor = `rgba(${r}, ${g}, ${b}, ${opacity})`;
 
-  const style = new Style({
+  return new Style({
     stroke: new Stroke({
       color: [r, g, b],
       width: 4,
@@ -62,8 +58,14 @@ const setHighlightStyle = (feature: Feature) => {
       scale: 2,
     }),
   });
+};
 
-  feature.setStyle(style);
+const resetStyle = (feature: Feature) => {
+  feature.setStyle(getFeatureStyle(feature));
+};
+
+const setHighlightStyle = (feature: Feature) => {
+  feature.setStyle(getHighlightFeatureStyle(feature));
 };
 
 const newEMLayer = (dataset: Dataset, slice: number, tilegrid: TileGrid, projection: Projection): TileLayer<XYZ> => {
@@ -78,17 +80,6 @@ const newEMLayer = (dataset: Dataset, slice: number, tilegrid: TileGrid, project
   });
 };
 
-const newSegLayer = (dataset: Dataset, slice: number) => {
-  return new VectorLayer({
-    source: new VectorSource({
-      url: getSegmentationURL(dataset, slice),
-      format: new GeoJSON(),
-    }),
-    style: getFeatureStyle,
-    zIndex: 1,
-  });
-};
-
 const scale = new ScaleLine({
   units: "metric",
 });
@@ -104,6 +95,7 @@ const interactions = defaultInteractions({
 // const EMStackViewer = ({ dataset }: EMStackViewerParameters) => {
 const EMStackViewer = () => {
   const currentWorkspace = useGlobalContext().getCurrentWorkspace();
+  const selectedNeurons = currentWorkspace.getViewerSelectedNeurons(ViewerType.EM);
 
   // We take the first active dataset at the moment (will change later)
   const firstActiveDataset = Object.values(currentWorkspace.activeDatasets)?.[0];
@@ -146,6 +138,41 @@ const EMStackViewer = () => {
     });
   }, [extent, firstActiveDataset.emData.tileSize]);
 
+  const segLayerStyle = (feature: FeatureLike) => {
+    const properties = feature.getProperties();
+    if (!properties.hasOwnProperty("name")) {
+      throw Error("segment doesn't have a name property");
+    }
+    const neuronName = properties["name"];
+    if (typeof neuronName !== "string") {
+      throw Error("segment name is not a string");
+    }
+
+    if (!selectedNeurons.length) {
+      return getFeatureStyle(feature);
+    }
+
+    if (neuronName in selectedNeurons) {
+      return getHighlightFeatureStyle(feature);
+    }
+
+    return new Style({});
+  };
+
+  const newSegLayer = useCallback(
+    (dataset: Dataset, slice: number) => {
+      return new VectorLayer({
+        source: new VectorSource({
+          url: getSegmentationURL(dataset, slice),
+          format: new GeoJSON(),
+        }),
+        style: segLayerStyle,
+        zIndex: 1,
+      });
+    },
+    [selectedNeurons],
+  );
+
   // const debugLayer = new TileLayer({
   // 	source: new TileDebug({
   // 		projection: projection,