damo_record: Implement a class for memory foontprints recording

Implement a class for extending 'damo record' to further record memory
footprints.  At the moment, it uses '/proc/<pid>/statm' files.

Signed-off-by: SeongJae Park <sj38.park@gmail.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/damo_record.py b/damo_record.py
index d5fb6e2d..8abdd8aa 100644
--- a/damo_record.py
+++ b/damo_record.py
@@ -62,6 +62,16 @@ def handle_args(args):
         print(err)
         exit(-3)
 
+class MemFootprintsSnapshot:
+    time = None
+    pid_statms = {}
+
+    def __init__(self, pids):
+        self.time = time.time()
+        for pid in pids:
+            with open('/proc/%s/statm' % pid, 'r') as f:
+                self.pid_statms[pid] = f.read()
+
 def pid_running(pid):
     '''pid should be string'''
     try: