#!/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


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.]

  while True:
    sm.update()

    if sm.updated['thermal']:
      t = sm['thermal']
      last_temp = np.mean([t.cpu0, t.cpu1, t.cpu2, t.cpu3]) / 10.
      last_mem = t.memUsedPercent

    if sm.updated['procLog']:
      m = sm['procLog']


      cores = [0., 0., 0., 0.]
      total_times_new = [0., 0., 0., 0.]
      busy_times_new = [0., 0., 0.0, 0.]

      for c in m.cpuTimes:
        n = c.cpuNum
        total_times_new[n] = cputime_total(c)
        busy_times_new[n] = cputime_busy(c)

      for n in range(4):
        t_busy = busy_times_new[n] - busy_times[n]
        t_total = total_times_new[n] - total_times[n]
        cores[n] = t_busy / t_total

      total_times = total_times_new[:]
      busy_times = busy_times_new[:]

      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()