Fix: refresh plot for unselected images when viewing in a new window

Fix #159

Author: PierreRaybaut

CHANGELOG.md

@@ -14,6 +14,7 @@ See DataLab [roadmap page](https://datalab-platform.com/en/contributing/roadmap.
 * Fixed [Issue #152](https://github.com/DataLab-Platform/DataLab/issues/152) - Ignore `nan` values for image normalization, flatfield correction, offset correction, and centroid computation
 * Fixed [Issue #153](https://github.com/DataLab-Platform/DataLab/issues/153) - Ignore `nan` values for signal normalization and statistics computations (both analysis result and interactive tool)
 * Fixed [Issue #158](https://github.com/DataLab-Platform/DataLab/issues/158) - When editing ROI of a list of images, the first image of the selection is shown (instead of the last as in the image panel)
+* Fixed [Issue #159](https://github.com/DataLab-Platform/DataLab/issues/159) - When selecting multiple images just after opening an HDF5 file, the "View in a new window" feature does not work (`KeyError` exception)
 
 ## DataLab Version 0.19.0 ##
 

cdl/core/gui/panel/base.py

@@ -1098,6 +1098,16 @@ def open_separate_view(
             oids = self.objview.get_sel_object_uuids(include_groups=True)
         obj = self.objmodel[oids[0]]
 
+        if not all([oid in self.plothandler for oid in oids]):
+            # This happens for example when opening an already saved workspace with
+            # multiple images, and if the user tries to view in a new window a group of
+            # images without having selected any object yet. In this case, only the
+            # last image is actually plotted (because if the other have the same size
+            # and position, they are hidden), and the plot item of every other image is
+            # not created yet. So we need to refresh the plot to create the plot item of
+            # those images.
+            self.plothandler.refresh_plot("selected", True, True)
+
         # Create a new dialog and add plot items to it
         dlg = self.create_new_dialog(
             title=obj.title if len(oids) == 1 else None,