From 887451da398befdfbb2fc28a8e4b1a171ade2f23 Mon Sep 17 00:00:00 2001
From: aranega <vincent.aranega@gmail.com>
Date: Tue, 14 Jan 2025 04:36:20 -0600
Subject: [PATCH] feat: Add dimension tool in palette in the bottom of the
 screen

---
 .../state_generator.py                        | 11 ++++++++
 tests/test_state_generators.py                | 27 ++++++++++++++++++-
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/cryoet_data_portal_neuroglancer/state_generator.py b/cryoet_data_portal_neuroglancer/state_generator.py
index 1d23c85..74cd47d 100644
--- a/cryoet_data_portal_neuroglancer/state_generator.py
+++ b/cryoet_data_portal_neuroglancer/state_generator.py
@@ -293,6 +293,17 @@ def combine_json_layers(
             "row": 2,
             "visible": False,
         },
+        "toolPalettes": {
+            "Dimensions": {
+                "side": "bottom",
+                "row": 1,
+                "tools": [
+                    {"type": "dimension", "dimension": "x"},
+                    {"type": "dimension", "dimension": "y"},
+                    {"type": "dimension", "dimension": "z"},
+                ],
+            },
+        },
     }
     if len(image_layers) > 0 and "_position" in image_layers[0]:
         combined_json["position"] = image_layers[0]["_position"]
diff --git a/tests/test_state_generators.py b/tests/test_state_generators.py
index 332220c..fe5106d 100644
--- a/tests/test_state_generators.py
+++ b/tests/test_state_generators.py
@@ -20,8 +20,33 @@ def test__generate_segmentation_layer_default_values():
     assert state["pick"] is False
 
 
-def test__generate_color_legend():
+def test__generate_configuration_default_values():
     state = combine_json_layers(layers=[{"type": "image", "volumeRendering": "OK", "name": "myname"}], scale=1.0)
 
     assert "enableLayerColorWidget" in state
     assert state["enableLayerColorWidget"] is True
+
+    assert "toolPalettes" in state
+    assert len(state["toolPalettes"]) == 1
+
+    palette = state["toolPalettes"]["Dimensions"]
+    assert palette.get("side") == "bottom"
+    assert palette.get("row") == 1
+    assert len(palette.get("tools", [])) == 3
+
+    tools = palette["tools"]
+    assert "type" in tools[0]
+    assert "type" in tools[1]
+    assert "type" in tools[2]
+
+    assert tools[0]["type"] == "dimension"
+    assert tools[1]["type"] == "dimension"
+    assert tools[2]["type"] == "dimension"
+
+    assert "dimension" in tools[0]
+    assert "dimension" in tools[1]
+    assert "dimension" in tools[2]
+
+    assert tools[0]["dimension"] == "x"
+    assert tools[1]["dimension"] == "y"
+    assert tools[2]["dimension"] == "z"