From 7fb26813eb7da26f88f1f7ad3b02ab612396588a Mon Sep 17 00:00:00 2001
From: Polina Cherkasova <polina.c@live.com>
Date: Sat, 27 Apr 2024 16:16:02 -0700
Subject: [PATCH] Update diff_pane_controller.dart

---
 .../panes/diff/controller/diff_pane_controller.dart  | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/packages/devtools_app/lib/src/screens/memory/panes/diff/controller/diff_pane_controller.dart b/packages/devtools_app/lib/src/screens/memory/panes/diff/controller/diff_pane_controller.dart
index 91cf9b81656..6a6ba0e148d 100644
--- a/packages/devtools_app/lib/src/screens/memory/panes/diff/controller/diff_pane_controller.dart
+++ b/packages/devtools_app/lib/src/screens/memory/panes/diff/controller/diff_pane_controller.dart
@@ -48,9 +48,19 @@ class DiffPaneController extends DisposableController {
     final snapshots = (json[_Json.snapshots] as List)
         .map((e) => SnapshotDataItem.fromJson(e))
         .toList();
+
+    final diffWith = json[_Json.diffWith] as List<int?>;
+
+    for (var i = 0; i < snapshots.length; i++) {
+      final diffIndex = diffWith[i];
+      if (diffIndex != null) {
+        snapshots[i].diffWith.value = snapshots[diffIndex];
+      }
+    }
+
     return DiffPaneController(
       loader: null,
-      //snapshots: snapshots,
+      snapshots: snapshots,
     );
   }