add mem flag to live cpu and temp script

selfdrive/debug/live_cpu_and_temp.py

@@ -1,8 +1,11 @@
 #!/usr/bin/env python3
+import argparse
+
 import numpy as np
 
 from cereal.messaging import SubMaster
 
+
 def cputime_total(ct):
   return ct.user + ct.nice + ct.system + ct.idle + ct.iowait + ct.irq + ct.softirq
 
@@ -11,16 +14,19 @@ def cputime_busy(ct):
   return ct.user + ct.nice + ct.system + ct.irq + ct.softirq
 
 
+if __name__ == "__main__":
+  parser = argparse.ArgumentParser()
+  parser.add_argument('--mem', action='store_true')
+  args = parser.parse_args()
 
-sm = SubMaster(['thermal', 'procLog'])
-
-last_temp = 0.0
-last_mem = 0.0
-total_times = [0., 0., 0., 0.]
-busy_times = [0., 0., 0.0, 0.]
+  sm = SubMaster(['thermal', 'procLog'])
 
+  last_temp = 0.0
+  last_mem = 0.0
+  total_times = [0., 0., 0., 0.]
+  busy_times = [0., 0., 0.0, 0.]
 
-while True:
+  while True:
     sm.update()
 
     if sm.updated['thermal']:
@@ -31,14 +37,6 @@ while True:
     if sm.updated['procLog']:
       m = sm['procLog']
 
-    mems = {}
-    for proc in m.procs:
-      name = proc.name
-      if len(proc.cmdline):
-        name = proc.cmdline[0]
-      if len(proc.exe):
-        name = proc.exe + " - " + name
-      mems[name] = float(proc.memRss) / 1e6
 
       cores = [0., 0., 0., 0.]
       total_times_new = [0., 0., 0., 0.]
@@ -57,8 +55,18 @@ while True:
       total_times = total_times_new[:]
       busy_times = busy_times_new[:]
 
-    print()
       print("CPU %.2f%% - RAM: %.2f - Temp %.2f" % (100. * np.mean(cores), last_mem, last_temp))
+
+      if args.mem:
+        mems = {}
+        for proc in m.procs:
+          name = proc.name
+          if len(proc.cmdline):
+            name = proc.cmdline[0]
+          if len(proc.exe):
+            name = proc.exe + " - " + name
+          mems[name] = float(proc.memRss) / 1e6
         print("Top memory usage:")
         for k, v in sorted(mems.items(), key=lambda item: item[1], reverse=True)[:10]:
           print(f"{k.rjust(70)}   {v:.2f} MB")
+        print()