_damo_records: Implement a function for loading memory footprints from a record file

Signed-off-by: SeongJae Park <sj@kernel.org>

1 files changed, 5 insertions, 0 deletions

diff --git a/_damo_records.py b/_damo_records.py
index 0e41b42d..5ffd0c8f 100644
--- a/_damo_records.py
+++ b/_damo_records.py
@@ -601,6 +601,11 @@ def save_mem_footprint(snapshots, filepath, file_permission):
         json.dump([s.to_kvpairs() for s in snapshots], f, indent=4)
     os.chmod(filepath, file_permission)
 
+def load_mem_footprint(filepath):
+    with open(filepath, 'r') as f:
+        kvpairs = json.load(f)
+    return [MemFootprintsSnapshot.from_kvpairs(x) for x in kvpairs]
+
 # record-polling
 
 def pid_running(pid):