Run controlsd without sockets

old-commit-hash: 4d03b8536f

selfdrive/test/profiling/controlsd.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+
+from tools.lib.logreader import LogReader
+from selfdrive.controls.controlsd import controlsd_thread
+from selfdrive.test.profiling.lib import SubMaster, PubMaster, SubSocket, ReplayDone
+
+
+BASE_URL = "https://commadataci.blob.core.windows.net/openpilotci/"
+SEGMENT = "99c94dc769b5d96e|2019-08-03--14-19-59/2"
+
+
+if __name__ == "__main__":
+  segment = SEGMENT.replace('|', '/')
+  rlog_url = f"{BASE_URL}{segment}/rlog.bz2"
+  msgs = list(LogReader(rlog_url))
+
+  pm = PubMaster(['sendcan', 'controlsState', 'carState', 'carControl', 'carEvents', 'carParams'])
+  sm = SubMaster(msgs, 'can', ['thermal', 'health', 'liveCalibration', 'dMonitoringState', 'plan', 'pathPlan', 'model'])
+  can_sock = SubSocket(msgs, 'can')
+
+  try:
+    controlsd_thread(sm, pm, can_sock)
+  except ReplayDone:
+    pass

selfdrive/test/profiling/lib.py

@@ -0,0 +1,107 @@
+import cereal.messaging as messaging
+import capnp
+
+
+class ReplayDone(Exception):
+  pass
+
+
+class SubSocket():
+  def __init__(self, msgs, trigger):
+    self.i = 0
+    self.trigger = trigger
+    self.msgs = [m for m in msgs if m.which() == trigger]
+    self.max_i = len(self.msgs) - 1
+
+  def receive(self, non_blocking=False):
+    if non_blocking:
+      return None
+
+    if self.i == self.max_i:
+      raise ReplayDone
+
+    while True:
+      msg = self.msgs[self.i]
+      msg = msg.as_builder()
+
+      self.i += 1
+
+      return msg.to_bytes()
+
+
+class PubSocket():
+  def send(self, data):
+    pass
+
+
+class SubMaster():
+  def __init__(self, msgs, trigger, services):
+    self.max_i = len(msgs) - 1
+    self.i = 0
+    self.frame = 0
+    self.trigger = trigger
+    self.msgs = msgs
+    self.data = {}
+
+    self.alive = {s: True for s in services}
+    self.updated = {s: False for s in services}
+    self.rcv_time = {s: 0. for s in services}
+    self.rcv_frame = {s: 0 for s in services}
+    self.valid = {s: True for s in services}
+    self.logMonoTime = {}
+    self.sock = {}
+
+    for s in services:
+      try:
+        data = messaging.new_message(s)
+      except capnp.lib.capnp.KjException:
+        # lists
+        data = messaging.new_message(s, 0)
+
+      self.data[s] = getattr(data, s)
+      self.logMonoTime[s] = 0
+      self.sock[s] = SubSocket(msgs, s)
+
+  def update(self, timeout=None):
+    if self.i == self.max_i:
+      raise ReplayDone
+
+    self.updated = dict.fromkeys(self.updated, False)
+    self.frame += 1
+
+    while True:
+      msg = self.msgs[self.i]
+      w = msg.which()
+
+      self.updated[w] = True
+      self.rcv_time[w] = msg.logMonoTime / 1e9
+      self.rcv_frame[w] = self.frame
+      self.data[w] = getattr(msg, w)
+      self.logMonoTime[w] = msg.logMonoTime
+
+      self.i += 1
+
+      if w == self.trigger:
+        break
+
+  def all_alive(self):
+    return True
+
+  def all_valid(self):
+    return True
+
+  def all_alive_and_valid(self):
+    return True
+
+  def __getitem__(self, s):
+    return self.data[s]
+
+
+class PubMaster():
+  def __init__(self, services):
+    self.sock = {}
+    for s in services:
+      self.sock[s] = PubSocket()
+
+  def send(self, s, dat):
+    pass