Correctly restore ui state when selectedNodeId's parent is not in dee…

…pestExpandedGroupNodeIds. PiperOrigin-RevId: 689444740
google-ai-edge · Oct 24, 2024 · 18a2459 · 18a2459
1 parent c2bbd1c
commit 18a2459
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/ui/src/components/visualizer/webgl_renderer.ts b/src/ui/src/components/visualizer/webgl_renderer.ts
@@ -886,6 +886,20 @@ export class WebglRenderer implements OnInit, OnDestroy {
           );
           deepestExpandedGroupNodeIds = groupNodeIds;
         }
+        // Add the parent node of the selected node if it is not set in
+        // deepestExpandedGroupNodeIds.
+        else {
+          const selectedNode =
+            this.curModelGraph.nodesById[paneState.selectedNodeId];
+          const nsParentId = selectedNode?.nsParentId || '';
+          if (
+            selectedNode &&
+            nsParentId &&
+            !deepestExpandedGroupNodeIds.includes(nsParentId)
+          ) {
+            deepestExpandedGroupNodeIds.push(nsParentId);
+          }
+        }
         if (
           paneState.selectedNodeId !== '' ||
           deepestExpandedGroupNodeIds.length > 0