openpilot_comma/selfdrive/debug/live_cpu_and_temp.py

#!/usr/bin/env python3
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



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

    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.]
    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()
    print("CPU %.2f%% - RAM: %.2f - Temp %.2f" % (100. * np.mean(cores), last_mem, last_temp))
    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")