Merge pull request #389 from jburel/issue_380

Pass the node and not the Data object
ome · Aug 17, 2023 · 919d9af · 919d9af
2 parents 7e07e35 + 140f2b0
commit 919d9af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/main/java/org/openmicroscopy/shoola/agents/treeviewer/RefreshExperimenterDataLoader.java b/src/main/java/org/openmicroscopy/shoola/agents/treeviewer/RefreshExperimenterDataLoader.java
@@ -329,7 +329,7 @@ public void load()
                                     smartFolders = new 
                                     HashMap<SecurityContext, TreeImageSet>();
                                 smartFolders.put(ctx, (TreeImageSet) ob);
-                            } else nl.add(ob);
+                            } else nl.add(((TreeImageSet) ob).getUserObject());
                         }
                         m.put(ctx, nl);
                     }

diff --git a/src/main/java/org/openmicroscopy/shoola/agents/treeviewer/cmd/RefreshVisitor.java b/src/main/java/org/openmicroscopy/shoola/agents/treeviewer/cmd/RefreshVisitor.java
@@ -106,7 +106,7 @@ public void visit(TreeImageSet node)
         	&& node.isExpanded()) {
         	parent = node.getParentDisplay();
     		if (parent.isExpanded()) 
-    			foundNodes.add(userObject);
+    			foundNodes.add(node);
     		if (!(parent.getUserObject() instanceof ProjectData)) {
     			long id = ((DataObject) userObject).getId();
                 List<Long> l = expandedTopNodes.get(DatasetData.class);
@@ -123,7 +123,7 @@ public void visit(TreeImageSet node)
         	String ns = tag.getNameSpace();
     		if (parent.isExpanded() &&
     			!TagAnnotationData.INSIGHT_TAGSET_NS.equals(ns)) 
-    			foundNodes.add(userObject);
+    			foundNodes.add(node);
     		if (!(parent.getUserObject() instanceof TagAnnotationData)) {
     			long id = ((DataObject) userObject).getId();
                 List<Long> l = expandedTopNodes.get(TagAnnotationData.class);