include orientation icons in dropdown

spacetelescope · Nov 13, 2024 · 306ce37 · 306ce37
1 parent c9e9bfa
commit 306ce37
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jdaviz/configs/default/plugins/data_menu/data_menu.py b/jdaviz/configs/default/plugins/data_menu/data_menu.py
@@ -60,6 +60,7 @@ class DataMenu(TemplateMixin, LayerSelectMixin, DatasetSelectMixin):
     viewer_id = Unicode().tag(sync=True)
     viewer_reference = Unicode().tag(sync=True)
 
+    icons = Dict().tag(sync=True)
     layer_icons = Dict().tag(sync=True)  # read-only, see app.state.layer_icons
     viewer_icons = Dict().tag(sync=True)  # read-only, see app.state.viewer_icons
 
@@ -141,6 +142,8 @@ def data_not_in_viewer(data):
                              for k, v in self._viewer.toolbar.tools_data.items()
                              if k in SUBSET_TOOL_IDS.values()]
 
+        self.icons = {k: v for k, v in self.app.state.icons.items()}
+
     @property
     def user_api(self):
         expose = ['layer', 'set_layer_visibility', 'toggle_layer_visibility',