parent
da079d47d7
commit
f467642a1c
66 changed files with 4113 additions and 0 deletions
@ -0,0 +1,42 @@ |
||||
#!/usr/bin/env python3 |
||||
import binascii |
||||
import os |
||||
import sys |
||||
from collections import defaultdict |
||||
|
||||
import cereal.messaging as messaging |
||||
from common.realtime import sec_since_boot |
||||
|
||||
|
||||
def can_printer(bus=0, max_msg=None, addr="127.0.0.1"): |
||||
logcan = messaging.sub_sock('can', addr=addr) |
||||
|
||||
start = sec_since_boot() |
||||
lp = sec_since_boot() |
||||
msgs = defaultdict(list) |
||||
canbus = int(os.getenv("CAN", bus)) |
||||
while 1: |
||||
can_recv = messaging.drain_sock(logcan, wait_for_one=True) |
||||
for x in can_recv: |
||||
for y in x.can: |
||||
if y.src == canbus: |
||||
msgs[y.address].append(y.dat) |
||||
|
||||
if sec_since_boot() - lp > 0.1: |
||||
dd = chr(27) + "[2J" |
||||
dd += "%5.2f\n" % (sec_since_boot() - start) |
||||
for k,v in sorted(zip(msgs.keys(), map(lambda x: binascii.hexlify(x[-1]), msgs.values()))): |
||||
if max_msg is None or k < max_msg: |
||||
dd += "%s(%6d) %s\n" % ("%04X(%4d)" % (k,k),len(msgs[k]), v.decode('ascii')) |
||||
print(dd) |
||||
lp = sec_since_boot() |
||||
|
||||
if __name__ == "__main__": |
||||
if len(sys.argv) > 3: |
||||
can_printer(int(sys.argv[1]), int(sys.argv[2]), sys.argv[3]) |
||||
elif len(sys.argv) > 2: |
||||
can_printer(int(sys.argv[1]), int(sys.argv[2])) |
||||
elif len(sys.argv) > 1: |
||||
can_printer(int(sys.argv[1])) |
||||
else: |
||||
can_printer() |
@ -0,0 +1,43 @@ |
||||
#!/usr/bin/env python3 |
||||
import argparse |
||||
import numpy as np |
||||
from collections import defaultdict, deque |
||||
from common.realtime import sec_since_boot |
||||
import cereal.messaging as messaging |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
context = messaging.Context() |
||||
poller = messaging.Poller() |
||||
|
||||
parser = argparse.ArgumentParser() |
||||
parser.add_argument("socket", type=str, nargs='*', help="socket name") |
||||
args = parser.parse_args() |
||||
|
||||
socket_names = args.socket |
||||
sockets = {} |
||||
|
||||
rcv_times = defaultdict(lambda: deque(maxlen=100)) |
||||
|
||||
t = sec_since_boot() |
||||
for name in socket_names: |
||||
sock = messaging.sub_sock(name, poller=poller) |
||||
sockets[sock] = name |
||||
|
||||
prev_print = t |
||||
while True: |
||||
for socket in poller.poll(100): |
||||
msg = messaging.recv_one(socket) |
||||
name = msg.which() |
||||
|
||||
t = sec_since_boot() |
||||
rcv_times[name].append(msg.logMonoTime / 1e9) |
||||
|
||||
if t - prev_print > 1: |
||||
print() |
||||
for name in socket_names: |
||||
dts = np.diff(rcv_times[name]) |
||||
mean = np.mean(dts) |
||||
print("%s: Freq %.2f Hz, Min %.2f%%, Max %.2f%%" % (name, 1.0 / mean, np.min(dts) / mean * 100, np.max(dts) / mean * 100)) |
||||
|
||||
prev_print = t |
@ -0,0 +1,18 @@ |
||||
#!/usr/bin/env python3 |
||||
|
||||
# put 2 fingeprints and print the diffs |
||||
f1 = { |
||||
168: 8, 257: 5, 258: 8, 264: 8, 268: 8, 270: 8, 274: 2, 280: 8, 284: 8, 288: 7, 290: 6, 291: 8, 292: 8, 294: 8, 300: 8, 308: 8, 320: 8, 324: 8, 331: 8, 332: 8, 344: 8, 368: 8, 376: 3, 384: 8, 388: 4, 448: 6, 456: 4, 464: 8, 469: 8, 480: 8, 500: 8, 501: 8, 512: 8, 514: 8, 520: 8, 528: 8, 532: 8, 544: 8, 557: 8, 559: 8, 560: 8, 564: 8, 571: 3, 579: 8, 584: 8, 608: 8, 624: 8, 625: 8, 632: 8, 639: 8, 653: 8, 654: 8, 655: 8, 658: 6, 660: 8, 669: 3, 671: 8, 672: 8, 678: 8, 680: 8, 701: 8, 703: 8, 704: 8, 705: 8, 706: 8, 709: 8, 710: 8, 719: 8, 720: 6, 729: 5, 736: 8, 737: 8, 746: 5, 752: 2, 754: 8, 760: 8, 764: 8, 766: 8, 770: 8, 773: 8, 779: 8, 782: 8, 784: 8, 792: 8, 799: 8, 800: 8, 804: 8, 816: 8, 817: 8, 820: 8, 825: 2, 826: 8, 832: 8, 838: 2, 848: 8, 853: 8, 856: 4, 860: 6, 863: 8, 878: 8, 882: 8, 897: 8, 906: 8, 908: 8, 924: 8, 926: 3, 929: 8, 937: 8, 938: 8, 939: 8, 940: 8, 941: 8, 942: 8, 943: 8, 947: 8, 948: 8, 958: 8, 959: 8, 962: 8, 969: 4, 973: 8, 974: 5, 979: 8, 980: 8, 981: 8, 982: 8, 983: 8, 984: 8, 992: 8, 993: 7, 995: 8, 996: 8, 1000: 8, 1001: 8, 1002: 8, 1003: 8, 1008: 8, 1009: 8, 1010: 8, 1011: 8, 1012: 8, 1013: 8, 1014: 8, 1015: 8, 1024: 8, 1025: 8, 1026: 8, 1031: 8, 1033: 8, 1050: 8, 1059: 8, 1082: 8, 1083: 8, 1098: 8, 1100: 8, 1537: 8, 1538: 8, 1562: 8 |
||||
} |
||||
|
||||
f2 = { |
||||
168: 8, 257: 5, 258: 8, 264: 8, 268: 8, 270: 8, 274: 2, 280: 8, 284: 8, 288: 7, 290: 6, 291: 8, 292: 8, 294: 8, 300: 8, 308: 8, 320: 8, 324: 8, 331: 8, 332: 8, 344: 8, 368: 8, 376: 3, 384: 8, 388: 4, 448: 6, 456: 4, 464: 8, 469: 8, 480: 8, 500: 8, 501: 8, 512: 8, 514: 8, 515: 7, 516: 7, 517: 7, 518: 7, 520: 8, 528: 8, 532: 8, 542: 8, 544: 8, 557: 8, 559: 8, 560: 8, 564: 8, 571: 3, 579: 8, 584: 8, 608: 8, 624: 8, 625: 8, 632: 8, 639: 8, 653: 8, 654: 8, 655: 8, 658: 6, 660: 8, 669: 3, 671: 8, 672: 8, 678: 8, 680: 8, 701: 8, 703: 8, 704: 8, 705: 8, 706: 8, 709: 8, 710: 8, 719: 8, 720: 6, 729: 5, 736: 8, 737: 8, 746: 5, 752: 2, 754: 8, 760: 8, 764: 8, 766: 8, 770: 8, 773: 8, 779: 8, 782: 8, 784: 8, 792: 8, 799: 8, 800: 8, 804: 8, 816: 8, 817: 8, 820: 8, 825: 2, 826: 8, 832: 8, 838: 2, 848: 8, 853: 8, 856: 4, 860: 6, 863: 8, 878: 8, 882: 8, 897: 8, 906: 8, 908: 8, 924: 8, 926: 3, 929: 8, 937: 8, 938: 8, 939: 8, 940: 8, 941: 8, 942: 8, 943: 8, 947: 8, 948: 8, 958: 8, 959: 8, 962: 8, 969: 4, 973: 8, 974: 5, 979: 8, 980: 8, 981: 8, 982: 8, 983: 8, 984: 8, 992: 8, 993: 7, 995: 8, 996: 8, 1000: 8, 1001: 8, 1002: 8, 1003: 8, 1008: 8, 1009: 8, 1010: 8, 1011: 8, 1012: 8, 1013: 8, 1014: 8, 1015: 8, 1024: 8, 1025: 8, 1026: 8, 1031: 8, 1033: 8, 1050: 8, 1059: 8, 1082: 8, 1083: 8, 1098: 8, 1100: 8 |
||||
} |
||||
|
||||
for k in f1: |
||||
if k not in f2 or f1[k] != f2[k]: |
||||
print(k, "not in f2") |
||||
|
||||
for k in f2: |
||||
if k not in f1 or f2[k] != f1[k]: |
||||
print(k, "not in f1") |
@ -0,0 +1,120 @@ |
||||
#!/usr/bin/env python3 |
||||
import psutil |
||||
import time |
||||
import os |
||||
import sys |
||||
import numpy as np |
||||
import argparse |
||||
import re |
||||
from collections import defaultdict |
||||
|
||||
''' |
||||
System tools like top/htop can only show current cpu usage values, so I write this script to do statistics jobs. |
||||
Features: |
||||
Use psutil library to sample cpu usage(avergage for all cores) of OpenPilot processes, at a rate of 5 samples/sec. |
||||
Do cpu usage statistics periodically, 5 seconds as a cycle. |
||||
Caculate the average cpu usage within this cycle. |
||||
Caculate minumium/maximium/accumulated_average cpu usage as long term inspections. |
||||
Monitor multiple processes simuteneously. |
||||
Sample usage: |
||||
root@localhost:/data/openpilot$ python selfdrive/debug/cpu_usage_stat.py boardd,ubloxd |
||||
('Add monitored proc:', './boardd') |
||||
('Add monitored proc:', 'python locationd/ubloxd.py') |
||||
boardd: 1.96%, min: 1.96%, max: 1.96%, acc: 1.96% |
||||
ubloxd.py: 0.39%, min: 0.39%, max: 0.39%, acc: 0.39% |
||||
''' |
||||
|
||||
# Do statistics every 5 seconds |
||||
PRINT_INTERVAL = 5 |
||||
SLEEP_INTERVAL = 0.2 |
||||
|
||||
monitored_proc_names = [ |
||||
'ubloxd', 'thermald', 'uploader', 'deleter', 'controlsd', 'plannerd', 'radard', 'mapd', 'loggerd' , 'logmessaged', 'tombstoned', |
||||
'logcatd', 'proclogd', 'boardd', 'pandad', './ui', 'ui', 'calibrationd', 'params_learner', 'modeld', 'monitoringd', 'camerad', 'sensord', 'updated', 'gpsd', 'athena'] |
||||
cpu_time_names = ['user', 'system', 'children_user', 'children_system'] |
||||
|
||||
timer = getattr(time, 'monotonic', time.time) |
||||
|
||||
def get_arg_parser(): |
||||
parser = argparse.ArgumentParser( |
||||
description="Unlogger and UI", |
||||
formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
||||
|
||||
parser.add_argument("proc_names", nargs="?", default='', |
||||
help="Process names to be monitored, comma seperated") |
||||
parser.add_argument("--list_all", action='store_true', |
||||
help="Show all running processes' cmdline") |
||||
parser.add_argument("--detailed_times", action='store_true', |
||||
help="show cpu time details (split by user, system, child user, child system)") |
||||
return parser |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
args = get_arg_parser().parse_args(sys.argv[1:]) |
||||
if args.list_all: |
||||
for p in psutil.process_iter(): |
||||
print('cmdline', p.cmdline(), 'name', p.name()) |
||||
sys.exit(0) |
||||
|
||||
if len(args.proc_names) > 0: |
||||
monitored_proc_names = args.proc_names.split(',') |
||||
monitored_procs = [] |
||||
stats = {} |
||||
for p in psutil.process_iter(): |
||||
if p == psutil.Process(): |
||||
continue |
||||
matched = any([l for l in p.cmdline() if any([pn for pn in monitored_proc_names if re.match(r'.*{}.*'.format(pn), l, re.M | re.I)])]) |
||||
if matched: |
||||
k = ' '.join(p.cmdline()) |
||||
print('Add monitored proc:', k) |
||||
stats[k] = {'cpu_samples': defaultdict(list), 'min': defaultdict(lambda: None), 'max': defaultdict(lambda: None), |
||||
'avg': defaultdict(lambda: 0.0), 'last_cpu_times': None, 'last_sys_time':None} |
||||
stats[k]['last_sys_time'] = timer() |
||||
stats[k]['last_cpu_times'] = p.cpu_times() |
||||
monitored_procs.append(p) |
||||
i = 0 |
||||
interval_int = int(PRINT_INTERVAL / SLEEP_INTERVAL) |
||||
while True: |
||||
for p in monitored_procs: |
||||
k = ' '.join(p.cmdline()) |
||||
cur_sys_time = timer() |
||||
cur_cpu_times = p.cpu_times() |
||||
cpu_times = np.subtract(cur_cpu_times, stats[k]['last_cpu_times']) / (cur_sys_time - stats[k]['last_sys_time']) |
||||
stats[k]['last_sys_time'] = cur_sys_time |
||||
stats[k]['last_cpu_times'] = cur_cpu_times |
||||
cpu_percent = 0 |
||||
for num, name in enumerate(cpu_time_names): |
||||
stats[k]['cpu_samples'][name].append(cpu_times[num]) |
||||
cpu_percent += cpu_times[num] |
||||
stats[k]['cpu_samples']['total'].append(cpu_percent) |
||||
time.sleep(SLEEP_INTERVAL) |
||||
i += 1 |
||||
if i % interval_int == 0: |
||||
l = [] |
||||
for k, stat in stats.items(): |
||||
if len(stat['cpu_samples']) <= 0: |
||||
continue |
||||
for name, samples in stat['cpu_samples'].items(): |
||||
samples = np.array(samples) |
||||
avg = samples.mean() |
||||
c = samples.size |
||||
min_cpu = np.amin(samples) |
||||
max_cpu = np.amax(samples) |
||||
if stat['min'][name] is None or min_cpu < stat['min'][name]: |
||||
stat['min'][name] = min_cpu |
||||
if stat['max'][name] is None or max_cpu > stat['max'][name]: |
||||
stat['max'][name] = max_cpu |
||||
stat['avg'][name] = (stat['avg'][name] * (i - c) + avg * c) / (i) |
||||
stat['cpu_samples'][name] = [] |
||||
|
||||
msg = 'avg: {1:.2%}, min: {2:.2%}, max: {3:.2%} {0}'.format(os.path.basename(k), stat['avg']['total'], stat['min']['total'], stat['max']['total']) |
||||
if args.detailed_times: |
||||
for stat_type in ['avg', 'min', 'max']: |
||||
msg += '\n {}: {}'.format(stat_type, [name + ':' + str(round(stat[stat_type][name]*100, 2)) for name in cpu_time_names]) |
||||
l.append((os.path.basename(k), stat['avg']['total'], msg)) |
||||
l.sort(key= lambda x: -x[1]) |
||||
for x in l: |
||||
print(x[2]) |
||||
print('avg sum: {0:.2%} over {1} samples {2} seconds\n'.format( |
||||
sum([stat['avg']['total'] for k, stat in stats.items()]), i, i * SLEEP_INTERVAL |
||||
)) |
@ -0,0 +1,64 @@ |
||||
#!/usr/bin/env python3 |
||||
import os |
||||
import sys |
||||
import argparse |
||||
import json |
||||
from hexdump import hexdump |
||||
|
||||
from cereal import log |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
if __name__ == "__main__": |
||||
|
||||
parser = argparse.ArgumentParser(description='Sniff a communcation socket') |
||||
parser.add_argument('--pipe', action='store_true') |
||||
parser.add_argument('--raw', action='store_true') |
||||
parser.add_argument('--json', action='store_true') |
||||
parser.add_argument('--dump-json', action='store_true') |
||||
parser.add_argument('--no-print', action='store_true') |
||||
parser.add_argument('--addr', default='127.0.0.1') |
||||
parser.add_argument('--values', help='values to monitor (instead of entire event)') |
||||
parser.add_argument("socket", type=str, nargs='*', help="socket name") |
||||
args = parser.parse_args() |
||||
|
||||
if args.addr != "127.0.0.1": |
||||
os.environ["ZMQ"] = "1" |
||||
messaging.context = messaging.Context() |
||||
|
||||
poller = messaging.Poller() |
||||
|
||||
for m in args.socket if len(args.socket) > 0 else service_list: |
||||
sock = messaging.sub_sock(m, poller, addr=args.addr) |
||||
|
||||
values = None |
||||
if args.values: |
||||
values = [s.strip().split(".") for s in args.values.split(",")] |
||||
|
||||
while 1: |
||||
polld = poller.poll(1000) |
||||
for sock in polld: |
||||
msg = sock.receive() |
||||
evt = log.Event.from_bytes(msg) |
||||
|
||||
if not args.no_print: |
||||
if args.pipe: |
||||
sys.stdout.write(msg) |
||||
sys.stdout.flush() |
||||
elif args.raw: |
||||
hexdump(msg) |
||||
elif args.json: |
||||
print(json.loads(msg)) |
||||
elif args.dump_json: |
||||
print(json.dumps(evt.to_dict())) |
||||
elif values: |
||||
print("logMonotime = {}".format(evt.logMonoTime)) |
||||
for value in values: |
||||
if hasattr(evt, value[0]): |
||||
item = evt |
||||
for key in value: |
||||
item = getattr(item, key) |
||||
print("{} = {}".format(".".join(value), item)) |
||||
print("") |
||||
else: |
||||
print(evt) |
@ -0,0 +1,29 @@ |
||||
#!/usr/bin/env python3 |
||||
|
||||
# simple script to get a vehicle fingerprint. |
||||
|
||||
# Instructions: |
||||
# - connect to a Panda |
||||
# - run selfdrive/boardd/boardd |
||||
# - launching this script |
||||
# - turn on the car in STOCK MODE (set giraffe switches properly). |
||||
# Note: it's very important that the car is in stock mode, in order to collect a complete fingerprint |
||||
# - since some messages are published at low frequency, keep this script running for at least 30s, |
||||
# until all messages are received at least once |
||||
|
||||
import cereal.messaging as messaging |
||||
|
||||
logcan = messaging.sub_sock('can') |
||||
msgs = {} |
||||
while True: |
||||
lc = messaging.recv_sock(logcan, True) |
||||
for c in lc.can: |
||||
# read also msgs sent by EON on CAN bus 0x80 and filter out the |
||||
# addr with more than 11 bits |
||||
if c.src in [0, 2] and c.address < 0x800: |
||||
msgs[c.address] = len(c.dat) |
||||
|
||||
fingerprint = ', '.join("%d: %d" % v for v in sorted(msgs.items())) |
||||
|
||||
print("number of messages {0}:".format(len(msgs))) |
||||
print("fingerprint {0}".format(fingerprint)) |
@ -0,0 +1,21 @@ |
||||
import time |
||||
import numpy as np |
||||
|
||||
from common.realtime import sec_since_boot |
||||
|
||||
N = 1000 |
||||
|
||||
times = [] |
||||
for i in range(1000): |
||||
t1 = sec_since_boot() |
||||
time.sleep(0.01) |
||||
t2 = sec_since_boot() |
||||
dt = t2 - t1 |
||||
times.append(dt) |
||||
|
||||
|
||||
print("Mean", np.mean(times)) |
||||
print("Max", np.max(times)) |
||||
print("Min", np.min(times)) |
||||
print("Variance", np.var(times)) |
||||
print("STD", np.sqrt(np.var(times))) |
@ -0,0 +1,113 @@ |
||||
#!/usr/bin/env python3 |
||||
import os |
||||
|
||||
from common.basedir import BASEDIR |
||||
os.environ['BASEDIR'] = BASEDIR |
||||
SCALE = 3 |
||||
|
||||
import argparse |
||||
import zmq |
||||
import pygame |
||||
import numpy as np |
||||
import cv2 |
||||
import sys |
||||
import traceback |
||||
from collections import namedtuple |
||||
|
||||
from cereal import car |
||||
from common.params import Params |
||||
from common.lazy_property import lazy_property |
||||
from cereal.messaging import sub_sock, recv_one_or_none, recv_one |
||||
from cereal.services import service_list |
||||
|
||||
_BB_OFFSET = 290, 332 |
||||
_BB_TO_FULL_FRAME = np.asarray([[1., 0., _BB_OFFSET[0]], [0., 1., _BB_OFFSET[1]], |
||||
[0., 0., 1.]]) |
||||
_FULL_FRAME_TO_BB = np.linalg.inv(_BB_TO_FULL_FRAME) |
||||
_FULL_FRAME_SIZE = 1164, 874 |
||||
|
||||
|
||||
|
||||
def pygame_modules_have_loaded(): |
||||
return pygame.display.get_init() and pygame.font.get_init() |
||||
|
||||
|
||||
def ui_thread(addr, frame_address): |
||||
context = zmq.Context() |
||||
|
||||
pygame.init() |
||||
pygame.font.init() |
||||
assert pygame_modules_have_loaded() |
||||
|
||||
size = (640 * SCALE, 480 * SCALE) |
||||
pygame.display.set_caption("comma one debug UI") |
||||
screen = pygame.display.set_mode(size, pygame.DOUBLEBUF) |
||||
|
||||
camera_surface = pygame.surface.Surface((640 * SCALE, 480 * SCALE), 0, 24).convert() |
||||
|
||||
frame = context.socket(zmq.SUB) |
||||
frame.connect(frame_address or "tcp://%s:%d" % (addr, 'frame')) |
||||
frame.setsockopt(zmq.SUBSCRIBE, "") |
||||
|
||||
img = np.zeros((480, 640, 3), dtype='uint8') |
||||
imgff = np.zeros((_FULL_FRAME_SIZE[1], _FULL_FRAME_SIZE[0], 3), dtype=np.uint8) |
||||
|
||||
while 1: |
||||
list(pygame.event.get()) |
||||
screen.fill((64, 64, 64)) |
||||
|
||||
# ***** frame ***** |
||||
fpkt = recv_one(frame) |
||||
yuv_img = fpkt.frame.image |
||||
|
||||
if fpkt.frame.transform: |
||||
yuv_transform = np.array(fpkt.frame.transform).reshape(3, 3) |
||||
else: |
||||
# assume frame is flipped |
||||
yuv_transform = np.array([[-1.0, 0.0, _FULL_FRAME_SIZE[0] - 1], |
||||
[0.0, -1.0, _FULL_FRAME_SIZE[1] - 1], [0.0, 0.0, 1.0]]) |
||||
|
||||
if yuv_img and len(yuv_img) == _FULL_FRAME_SIZE[0] * _FULL_FRAME_SIZE[1] * 3 // 2: |
||||
yuv_np = np.frombuffer( |
||||
yuv_img, dtype=np.uint8).reshape(_FULL_FRAME_SIZE[1] * 3 // 2, -1) |
||||
cv2.cvtColor(yuv_np, cv2.COLOR_YUV2RGB_I420, dst=imgff) |
||||
cv2.warpAffine( |
||||
imgff, |
||||
np.dot(yuv_transform, _BB_TO_FULL_FRAME)[:2], (img.shape[1], img.shape[0]), |
||||
dst=img, |
||||
flags=cv2.WARP_INVERSE_MAP) |
||||
else: |
||||
img.fill(0) |
||||
|
||||
height, width = img.shape[:2] |
||||
img_resized = cv2.resize( |
||||
img, (SCALE * width, SCALE * height), interpolation=cv2.INTER_CUBIC) |
||||
# *** blits *** |
||||
pygame.surfarray.blit_array(camera_surface, img_resized.swapaxes(0, 1)) |
||||
screen.blit(camera_surface, (0, 0)) |
||||
|
||||
# this takes time...vsync or something |
||||
pygame.display.flip() |
||||
|
||||
|
||||
def get_arg_parser(): |
||||
parser = argparse.ArgumentParser( |
||||
description="Show replay data in a UI.", |
||||
formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
||||
|
||||
parser.add_argument( |
||||
"ip_address", |
||||
nargs="?", |
||||
default="127.0.0.1", |
||||
help="The ip address on which to receive zmq messages.") |
||||
|
||||
parser.add_argument( |
||||
"--frame-address", |
||||
default=None, |
||||
help="The ip address on which to receive zmq messages.") |
||||
return parser |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
args = get_arg_parser().parse_args(sys.argv[1:]) |
||||
ui_thread(args.ip_address, args.frame_address) |
@ -0,0 +1,98 @@ |
||||
#!/usr/bin/env python |
||||
from common.kalman.ned import ecef2geodetic |
||||
|
||||
import csv |
||||
import numpy as np |
||||
import webbrowser |
||||
import os |
||||
import sys |
||||
import json |
||||
import numpy.linalg as LA |
||||
import gmplot |
||||
from dateutil.parser import parse |
||||
from common.numpy_helpers import deep_interp |
||||
# import cvxpy as cvx |
||||
MPH_TO_MS = 0.44704 |
||||
|
||||
|
||||
def downsample(positions, speeds, start_idx, end_idx, dist): |
||||
# TODO: save headings too |
||||
track = [] |
||||
last_position = positions[start_idx] |
||||
valid_indeces = [] |
||||
track_speeds = [] |
||||
for pi in range(start_idx, end_idx): |
||||
# only save points that are at least 10 cm far away |
||||
if LA.norm(positions[pi] - last_position) >= dist: |
||||
#print LA.norm(positions[pi] - last_position) |
||||
last_position = positions[pi] |
||||
track.append(positions[pi]) |
||||
valid_indeces.append(pi) |
||||
track_speeds.append(speeds[pi]) |
||||
print(-start_idx + end_idx, len(valid_indeces)) |
||||
# this compare the original point count vs the filtered count |
||||
|
||||
track = np.array(track) |
||||
track_speeds = np.array(track_speeds) |
||||
return track, track_speeds |
||||
|
||||
def converter(date): |
||||
|
||||
filename = "/home/batman/one/selfdrive/locationd/liveloc_dumps/" + date + "/canonical.csv" # Point one (OK!) |
||||
|
||||
c = csv.DictReader(open(filename, 'rb'), delimiter=',') |
||||
|
||||
start_time = None |
||||
|
||||
t = [] |
||||
ll_positions = [] |
||||
positions = [] |
||||
sats = [] |
||||
flag = [] |
||||
speeds = [] |
||||
|
||||
for row in c: |
||||
t.append(float(row['pctime'])) |
||||
x = float(row['ecefX']) |
||||
y = float(row['ecefY']) |
||||
z = float(row['ecefZ']) |
||||
ecef = np.array((x, y, z)) |
||||
speeds.append(float(row['velSpeed'])) |
||||
|
||||
pos = ecef2geodetic(ecef) |
||||
ll_positions.append(pos) |
||||
positions.append(ecef) |
||||
|
||||
t = np.array(t) |
||||
ll_positions = np.array(ll_positions) |
||||
positions = np.array(positions) |
||||
|
||||
#distances = ll_positions[:,0:2] - START_POS[:2] |
||||
#i_start = np.argmin(LA.norm(distances, axis=1)) |
||||
|
||||
#for i in range(i_start + 500): |
||||
# distances[i] += np.array([100, 100]) |
||||
#i_end = np.argmin(LA.norm(distances, axis=1)) |
||||
|
||||
i_start = 0 |
||||
i_end = len(positions) |
||||
|
||||
print(i_start, i_end) |
||||
track, track_speeds = downsample(positions, speeds, i_start, i_end, 0.2) |
||||
ll_track = np.array([ecef2geodetic(pos) for pos in track]) |
||||
|
||||
track_struct = {} |
||||
print(track_speeds.shape) |
||||
print(track.shape) |
||||
track_struct['race'] = np.hstack((track, |
||||
np.expand_dims(track_speeds, axis=1), |
||||
np.zeros((len(track_speeds), 1)))) |
||||
|
||||
f = open('/home/batman/one/selfdrive/controls/tracks/loop_city.npy', 'w') |
||||
np.save(f, track_struct) |
||||
f.close() |
||||
print("SAVED!") |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
converter(sys.argv[1]) |
@ -0,0 +1,53 @@ |
||||
# USAGE: python cycle_alerts.py [duration_millis=1000] |
||||
# Then start manager |
||||
|
||||
import argparse |
||||
import time |
||||
import zmq |
||||
|
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
from selfdrive.controls.lib.alerts import ALERTS |
||||
|
||||
def now_millis(): return time.time() * 1000 |
||||
|
||||
default_alerts = sorted(ALERTS, key=lambda alert: (alert.alert_size, len(alert.alert_text_2))) |
||||
|
||||
def cycle_alerts(duration_millis, alerts=None): |
||||
if alerts is None: |
||||
alerts = default_alerts |
||||
|
||||
controls_state = messaging.pub_sock('controlsState') |
||||
|
||||
last_pop_millis = now_millis() |
||||
alert = alerts.pop() |
||||
while 1: |
||||
if (now_millis() - last_pop_millis) > duration_millis: |
||||
alerts.insert(0, alert) |
||||
alert = alerts.pop() |
||||
last_pop_millis = now_millis() |
||||
print('sending {}'.format(str(alert))) |
||||
|
||||
dat = messaging.new_message() |
||||
dat.init('controlsState') |
||||
|
||||
dat.controlsState.alertType = alert.alert_type |
||||
dat.controlsState.alertText1 = alert.alert_text_1 |
||||
dat.controlsState.alertText2 = alert.alert_text_2 |
||||
dat.controlsState.alertSize = alert.alert_size |
||||
dat.controlsState.alertStatus = alert.alert_status |
||||
dat.controlsState.alertSound = alert.audible_alert |
||||
controls_state.send(dat.to_bytes()) |
||||
|
||||
time.sleep(0.01) |
||||
|
||||
if __name__ == '__main__': |
||||
parser = argparse.ArgumentParser() |
||||
parser.add_argument('--duration', type=int, default=1000) |
||||
parser.add_argument('--alert-types', nargs='+') |
||||
args = parser.parse_args() |
||||
alerts = None |
||||
if args.alert_types: |
||||
alerts = [next(a for a in ALERTS if a.alert_type==alert_type) for alert_type in args.alert_types] |
||||
|
||||
cycle_alerts(args.duration, alerts=alerts) |
@ -0,0 +1,32 @@ |
||||
#!/usr/bin/env python3 |
||||
import numpy as np |
||||
import control |
||||
|
||||
dt = 0.01 |
||||
A = np.array([[ 0. , 1. ], [-0.78823806, 1.78060701]]) |
||||
B = np.array([[-2.23399437e-05], [ 7.58330763e-08]]) |
||||
C = np.array([[1., 0.]]) |
||||
|
||||
|
||||
# Kalman tuning |
||||
Q = np.diag([1, 1]) |
||||
R = np.atleast_2d(1e5) |
||||
|
||||
(_, _, L) = control.dare(A.T, C.T, Q, R) |
||||
L = L.T |
||||
|
||||
# LQR tuning |
||||
Q = np.diag([2e5, 1e-5]) |
||||
R = np.atleast_2d(1) |
||||
(_, _, K) = control.dare(A, B, Q, R) |
||||
|
||||
A_cl = (A - B.dot(K)) |
||||
sys = control.ss(A_cl, B, C, 0, dt) |
||||
dc_gain = control.dcgain(sys) |
||||
|
||||
print(("self.A = np." + A.__repr__()).replace('\n', '')) |
||||
print(("self.B = np." + B.__repr__()).replace('\n', '')) |
||||
print(("self.C = np." + C.__repr__()).replace('\n', '')) |
||||
print(("self.K = np." + K.__repr__()).replace('\n', '')) |
||||
print(("self.L = np." + L.__repr__()).replace('\n', '')) |
||||
print("self.dc_gain = " + str(dc_gain)) |
@ -0,0 +1,48 @@ |
||||
#!/usr/bin/env python3 |
||||
import time |
||||
import sys |
||||
import argparse |
||||
import zmq |
||||
import json |
||||
import pyproj |
||||
import numpy as np |
||||
ecef = pyproj.Proj(proj='geocent', ellps='WGS84', datum='WGS84') |
||||
lla = pyproj.Proj(proj='latlong', ellps='WGS84', datum='WGS84') |
||||
|
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
poller = zmq.Poller() |
||||
ll = messaging.sub_sock("liveLocation", poller) |
||||
tll = messaging.sub_sock("testLiveLocation", poller) |
||||
|
||||
l, tl = None, None |
||||
|
||||
lp = time.time() |
||||
|
||||
while 1: |
||||
polld = poller.poll(timeout=1000) |
||||
for sock, mode in polld: |
||||
if mode != zmq.POLLIN: |
||||
continue |
||||
if sock == ll: |
||||
l = messaging.recv_one(sock) |
||||
elif sock == tll: |
||||
tl = messaging.recv_one(sock) |
||||
if l is None or tl is None: |
||||
continue |
||||
|
||||
alt_err = np.abs(l.liveLocation.alt - tl.liveLocation.alt) |
||||
l1 = pyproj.transform(lla, ecef, l.liveLocation.lon, l.liveLocation.lat, l.liveLocation.alt) |
||||
l2 = pyproj.transform(lla, ecef, tl.liveLocation.lon, tl.liveLocation.lat, tl.liveLocation.alt) |
||||
|
||||
al1 = pyproj.transform(lla, ecef, l.liveLocation.lon, l.liveLocation.lat, l.liveLocation.alt) |
||||
al2 = pyproj.transform(lla, ecef, tl.liveLocation.lon, tl.liveLocation.lat, l.liveLocation.alt) |
||||
|
||||
tdiff = np.abs(l.logMonoTime - tl.logMonoTime) / 1e9 |
||||
|
||||
if time.time()-lp > 0.1: |
||||
print("tm: %f mse: %f mse(flat): %f alterr: %f" % (tdiff, np.mean((np.array(l1)-np.array(l2))**2), np.mean((np.array(al1)-np.array(al2))**2), alt_err)) |
||||
lp = time.time() |
||||
|
||||
|
@ -0,0 +1,33 @@ |
||||
#!/usr/bin/env python3 |
||||
|
||||
import sys |
||||
from tools.lib.logreader import MultiLogIterator |
||||
from xx.chffr.lib.route import Route |
||||
|
||||
|
||||
def get_fingerprint(lr): |
||||
can_msgs = [m for m in lr if m.which() == 'can'] |
||||
|
||||
msgs = {} |
||||
|
||||
for msg in can_msgs: |
||||
for c in msg.can: |
||||
# read also msgs sent by EON on CAN bus 0x80 and filter out the |
||||
# addr with more than 11 bits |
||||
if c.src % 0x80 == 0 and c.address < 0x800: |
||||
msgs[c.address] = len(c.dat) |
||||
|
||||
fingerprint = ', '.join("%d: %d" % v for v in sorted(msgs.items())) |
||||
print("number of messages {0}:".format(len(msgs))) |
||||
print("fingerprint {0}".format(fingerprint)) |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
if len(sys.argv) < 2: |
||||
print("Usage: ./get_fingerprint_internal.py <route>") |
||||
sys.exit(1) |
||||
|
||||
route = sys.argv[1] |
||||
route = Route(route) |
||||
lr = MultiLogIterator(route.log_paths(), wraparound=False) |
||||
get_fingerprint(lr) |
@ -0,0 +1,89 @@ |
||||
#!/usr/bin/env python3 |
||||
import sys |
||||
import time |
||||
|
||||
import matplotlib.pyplot as plt |
||||
import numpy as np |
||||
import cereal.messaging as messaging |
||||
import zmq |
||||
from common.transformations.coordinates import LocalCoord |
||||
from cereal.services import service_list |
||||
|
||||
SCALE = 20. |
||||
|
||||
def mpc_vwr_thread(addr="127.0.0.1"): |
||||
plt.ion() |
||||
fig = plt.figure(figsize=(15, 15)) |
||||
ax = fig.add_subplot(1,1,1) |
||||
ax.set_xlim([-SCALE, SCALE]) |
||||
ax.set_ylim([-SCALE, SCALE]) |
||||
ax.grid(True) |
||||
|
||||
line, = ax.plot([0.0], [0.0], ".b") |
||||
line2, = ax.plot([0.0], [0.0], 'r') |
||||
|
||||
ax.set_aspect('equal', 'datalim') |
||||
plt.show() |
||||
|
||||
live_location = messaging.sub_sock('liveLocation', addr=addr, conflate=True) |
||||
gps_planner_points = messaging.sub_sock('gpsPlannerPoints', conflate=True) |
||||
gps_planner_plan = messaging.sub_sock('gpsPlannerPlan', conflate=True) |
||||
|
||||
last_points = messaging.recv_one(gps_planner_points) |
||||
last_plan = messaging.recv_one(gps_planner_plan) |
||||
while True: |
||||
p = messaging.recv_one_or_none(gps_planner_points) |
||||
pl = messaging.recv_one_or_none(gps_planner_plan) |
||||
ll = messaging.recv_one(live_location).liveLocation |
||||
|
||||
if p is not None: |
||||
last_points = p |
||||
if pl is not None: |
||||
last_plan = pl |
||||
|
||||
if not last_plan.gpsPlannerPlan.valid: |
||||
time.sleep(0.1) |
||||
line2.set_color('r') |
||||
continue |
||||
|
||||
p0 = last_points.gpsPlannerPoints.points[0] |
||||
p0 = np.array([p0.x, p0.y, p0.z]) |
||||
|
||||
n = LocalCoord.from_geodetic(np.array([ll.lat, ll.lon, ll.alt])) |
||||
points = [] |
||||
print(len(last_points.gpsPlannerPoints.points)) |
||||
for p in last_points.gpsPlannerPoints.points: |
||||
ecef = np.array([p.x, p.y, p.z]) |
||||
points.append(n.ecef2ned(ecef)) |
||||
|
||||
points = np.vstack(points) |
||||
line.set_xdata(points[:, 1]) |
||||
line.set_ydata(points[:, 0]) |
||||
|
||||
y = np.matrix(np.arange(-100, 100.0, 0.5)) |
||||
x = -np.matrix(np.polyval(last_plan.gpsPlannerPlan.poly, y)) |
||||
xy = np.hstack([x.T, y.T]) |
||||
|
||||
cur_heading = np.radians(ll.heading - 90) |
||||
c, s = np.cos(cur_heading), np.sin(cur_heading) |
||||
R = np.array([[c, -s], [s, c]]) |
||||
xy = xy.dot(R) |
||||
|
||||
line2.set_xdata(xy[:, 1]) |
||||
line2.set_ydata(-xy[:, 0]) |
||||
line2.set_color('g') |
||||
|
||||
|
||||
ax.set_xlim([-SCALE, SCALE]) |
||||
ax.set_ylim([-SCALE, SCALE]) |
||||
|
||||
fig.canvas.draw() |
||||
fig.canvas.flush_events() |
||||
|
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
if len(sys.argv) > 1: |
||||
mpc_vwr_thread(sys.argv[1]) |
||||
else: |
||||
mpc_vwr_thread() |
@ -0,0 +1,113 @@ |
||||
#!/usr/bin/env python3 |
||||
import os |
||||
import zmq |
||||
|
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
from panda.lib.panda import Panda |
||||
from hexdump import hexdump |
||||
import time |
||||
|
||||
def raw_panda(): |
||||
p = Panda() |
||||
print(p) |
||||
|
||||
p.set_uart_baud(2, 9600) |
||||
p.set_uart_baud(3, 9600) |
||||
|
||||
p.set_uart_parity(2, 1) |
||||
p.set_uart_parity(3, 1) |
||||
|
||||
p.set_uart_callback(2, 1) |
||||
p.set_uart_callback(3, 1) |
||||
|
||||
idx = 0 |
||||
while 1: |
||||
""" |
||||
dat = p.serial_read(2) |
||||
if len(dat) > 0: |
||||
print "2:", |
||||
hexdump(dat) |
||||
|
||||
dat = p.serial_read(3) |
||||
if len(dat) > 0: |
||||
print "3:", |
||||
hexdump(dat) |
||||
|
||||
print "read done, waiting" |
||||
time.sleep(0.01) |
||||
""" |
||||
|
||||
if idx%2 == 1: |
||||
dat = "\x20\x80\xc0\xa0" |
||||
else: |
||||
dat = "\x00\x80\xc0\xc0" |
||||
p.can_send(0, dat, 8) |
||||
|
||||
for r in p.can_recv(): |
||||
if r[-1] in [8, 9]: |
||||
print(r[-1], r[2].encode("hex")) |
||||
|
||||
time.sleep(0.01) |
||||
idx += 1 |
||||
|
||||
if __name__ == "__main__": |
||||
#raw_panda() |
||||
#exit(0) |
||||
|
||||
logcan = messaging.sub_sock('can') |
||||
|
||||
t1 = [] |
||||
t2 = [] |
||||
t3 = [] |
||||
|
||||
while len(t1) < 1000 or os.uname()[-1] == "aarch64": |
||||
rr = messaging.recv_sock(logcan, wait=True) |
||||
for c in rr.can: |
||||
if c.src in [9] and len(c.dat) == 5: |
||||
aa = map(lambda x: ord(x)&0x7f, c.dat) |
||||
|
||||
# checksum |
||||
assert (-(aa[0]+aa[1]+aa[2]+aa[3]))&0x7f == aa[4] |
||||
|
||||
#print map(bin, aa[0:4]) |
||||
|
||||
aa[0] &= ~0x20 |
||||
aa[1] &= ~0x20 |
||||
|
||||
st = (aa[0] << 5) + aa[1] |
||||
if st >= 256: |
||||
st = -(512-st) |
||||
|
||||
mt = ((aa[2] >> 3) << 7) + aa[3] |
||||
if mt >= 512: |
||||
mt = -(1024-mt) |
||||
|
||||
print(st, mt) |
||||
t1.append(st) |
||||
t2.append(mt) |
||||
#print map(bin, aa), "apply", st |
||||
|
||||
if c.src in [8] and len(c.dat) == 4: |
||||
aa = map(lambda x: ord(x)&0x7f, c.dat) |
||||
|
||||
# checksum |
||||
assert (-(aa[0]+aa[1]+aa[2]))&0x7f == aa[3] |
||||
|
||||
aa[0] &= ~0x20 |
||||
aa[1] &= ~0x20 |
||||
|
||||
st = (aa[0] << 5) + aa[1] |
||||
if st >= 256: |
||||
st = -(512-st) |
||||
print(aa, "apply", st) |
||||
|
||||
t3.append(st) |
||||
|
||||
import matplotlib.pyplot as plt |
||||
plt.plot(t1) |
||||
plt.plot(t2) |
||||
plt.plot(t3) |
||||
plt.show() |
||||
|
@ -0,0 +1,57 @@ |
||||
#!/usr/bin/env python3 |
||||
import os |
||||
import sys |
||||
import argparse |
||||
import struct |
||||
|
||||
from cereal import log |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
if __name__ == "__main__": |
||||
|
||||
parser = argparse.ArgumentParser(description='Sniff a communcation socket') |
||||
parser.add_argument('--addr', default='127.0.0.1') |
||||
args = parser.parse_args() |
||||
|
||||
if args.addr != "127.0.0.1": |
||||
os.environ["ZMQ"] = "1" |
||||
messaging.context = messaging.Context() |
||||
|
||||
poller = messaging.Poller() |
||||
messaging.sub_sock('can', poller, addr=args.addr) |
||||
|
||||
active = 0 |
||||
start_t = 0 |
||||
start_v = 0 |
||||
max_v = 0 |
||||
max_t = 0 |
||||
window = [0] * 10 |
||||
avg = 0 |
||||
while 1: |
||||
polld = poller.poll(1000) |
||||
for sock in polld: |
||||
msg = sock.receive() |
||||
evt = log.Event.from_bytes(msg) |
||||
|
||||
for item in evt.can: |
||||
if item.address == 0xe4 and item.src == 128: |
||||
torque_req = struct.unpack('!h', item.dat[0:2])[0] |
||||
# print(torque_req) |
||||
active = abs(torque_req) > 0 |
||||
if abs(torque_req) < 100: |
||||
if max_v > 5: |
||||
print(f'{start_v} -> {max_v} = {round(max_v - start_v, 2)} over {round(max_t - start_t, 2)}s') |
||||
start_t = evt.logMonoTime / 1e9 |
||||
start_v = avg |
||||
max_t = 0 |
||||
max_v = 0 |
||||
if item.address == 0x1ab and item.src == 0: |
||||
motor_torque = ((item.dat[0] & 0x3) << 8) + item.dat[1] |
||||
window.append(motor_torque) |
||||
window.pop(0) |
||||
avg = sum(window) / len(window) |
||||
#print(f'{evt.logMonoTime}: {avg}') |
||||
if active and avg > max_v + 0.5: |
||||
max_v = avg |
||||
max_t = evt.logMonoTime / 1e9 |
@ -0,0 +1,2 @@ |
||||
sender |
||||
receiver |
@ -0,0 +1,61 @@ |
||||
CC = clang
|
||||
CXX = clang++
|
||||
|
||||
ARCH := $(shell uname -m)
|
||||
OS := $(shell uname -o)
|
||||
|
||||
BASEDIR = ../../../..
|
||||
PHONELIBS = ../../../../phonelibs
|
||||
|
||||
WARN_FLAGS = -Werror=implicit-function-declaration \
|
||||
-Werror=incompatible-pointer-types \
|
||||
-Werror=int-conversion \
|
||||
-Werror=return-type \
|
||||
-Werror=format-extra-args
|
||||
|
||||
CFLAGS = -std=gnu11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
CXXFLAGS = -std=c++11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
|
||||
ifeq ($(ARCH),aarch64) |
||||
CFLAGS += -mcpu=cortex-a57
|
||||
CXXFLAGS += -mcpu=cortex-a57
|
||||
endif |
||||
|
||||
|
||||
EXTRA_LIBS = -lpthread
|
||||
|
||||
ifeq ($(ARCH),x86_64) |
||||
BOOST_LIBS = -lboost_system -lboost_locale -lrt
|
||||
else |
||||
EXTRA_LIBS += -llog -luuid
|
||||
endif |
||||
|
||||
.PHONY: all |
||||
all: sender receiver |
||||
|
||||
receiver: receiver.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(CEREAL_LIBS) \
|
||||
$(BOOST_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
sender: sender.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(CEREAL_LIBS) \
|
||||
$(BOOST_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
%.o: %.cc |
||||
@echo "[ CXX ] $@"
|
||||
$(CXX) $(CXXFLAGS) -MMD \
|
||||
-Iinclude -I.. -I../.. \
|
||||
-I../ \
|
||||
-I../../ \
|
||||
-c -o '$@' '$<'
|
||||
|
||||
|
||||
.PHONY: clean |
||||
clean: |
||||
rm -f *.d sender receiver *.o
|
@ -0,0 +1,54 @@ |
||||
#include <boost/interprocess/ipc/message_queue.hpp> |
||||
#include <iostream> |
||||
#include <vector> |
||||
#include <thread> |
||||
|
||||
using namespace boost::interprocess; |
||||
#define N 1024 |
||||
|
||||
message_queue *sub_queue(const char *name){ |
||||
while (true){ |
||||
try { |
||||
message_queue *mq = new message_queue(open_only, name); |
||||
return mq; |
||||
} |
||||
catch(interprocess_exception &ex){ |
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10)); |
||||
} |
||||
|
||||
} |
||||
} |
||||
|
||||
message_queue *pub_queue(const char *name){ |
||||
message_queue::remove(name); |
||||
message_queue *mq = new message_queue(create_only, name, 100, N); |
||||
return mq; |
||||
} |
||||
|
||||
|
||||
|
||||
int main () |
||||
{ |
||||
|
||||
message_queue::remove("queue_1"); |
||||
message_queue::remove("queue_2"); |
||||
|
||||
message_queue *pq = pub_queue("queue_2"); |
||||
message_queue *sq = sub_queue("queue_1"); |
||||
std::cout << "Ready" << std::endl; |
||||
|
||||
unsigned int priority; |
||||
std::size_t recvd_size; |
||||
|
||||
char * rcv_msg = new char[N]; |
||||
|
||||
while (true){ |
||||
|
||||
sq->receive(rcv_msg, N, recvd_size, priority); |
||||
assert(N == recvd_size); |
||||
|
||||
pq->send(rcv_msg, N, 0); |
||||
} |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,62 @@ |
||||
#include <boost/interprocess/ipc/message_queue.hpp> |
||||
#include <iostream> |
||||
#include <vector> |
||||
#include <thread> |
||||
#include <chrono> |
||||
#include <cassert> |
||||
|
||||
#define N 1024 |
||||
#define MSGS 1e5 |
||||
|
||||
using namespace boost::interprocess; |
||||
|
||||
message_queue *sub_queue(const char *name){ |
||||
while (true){ |
||||
try { |
||||
message_queue *mq = new message_queue(open_only, name); |
||||
return mq; |
||||
} |
||||
catch(interprocess_exception &ex){ |
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10)); |
||||
} |
||||
|
||||
} |
||||
} |
||||
|
||||
message_queue *pub_queue(const char *name){ |
||||
message_queue::remove(name); |
||||
message_queue *mq = new message_queue(create_only, name, 100, N); |
||||
return mq; |
||||
} |
||||
|
||||
|
||||
|
||||
int main () |
||||
{ |
||||
message_queue *pq = pub_queue("queue_1"); |
||||
message_queue *sq = sub_queue("queue_2"); |
||||
std::cout << "Ready" << std::endl; |
||||
|
||||
auto start = std::chrono::steady_clock::now(); |
||||
char * rcv_msg = new char[N]; |
||||
char * snd_msg = new char[N]; |
||||
|
||||
unsigned int priority; |
||||
std::size_t recvd_size; |
||||
|
||||
for (int i = 0; i < MSGS; i++){ |
||||
sprintf(snd_msg, "%d", i); |
||||
|
||||
pq->send(snd_msg, N, 0); |
||||
sq->receive(rcv_msg, N, recvd_size, priority); |
||||
} |
||||
|
||||
auto end = std::chrono::steady_clock::now(); |
||||
double elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count() / 1e9; |
||||
double throughput = ((double) MSGS / (double) elapsed); |
||||
|
||||
std::cout << "Elapsed: " << elapsed << " s" << std::endl; |
||||
std::cout << "Throughput: " << throughput << " msg/s" << std::endl; |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,2 @@ |
||||
receiver |
||||
sender |
@ -0,0 +1,65 @@ |
||||
CC = clang
|
||||
CXX = clang++
|
||||
|
||||
ARCH := $(shell uname -m)
|
||||
OS := $(shell uname -o)
|
||||
|
||||
BASEDIR = ../../../..
|
||||
PHONELIBS = ../../../../phonelibs
|
||||
|
||||
WARN_FLAGS = -Werror=implicit-function-declaration \
|
||||
-Werror=incompatible-pointer-types \
|
||||
-Werror=int-conversion \
|
||||
-Werror=return-type \
|
||||
-Werror=format-extra-args
|
||||
|
||||
CFLAGS = -std=gnu11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
CXXFLAGS = -std=c++11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
# NANOMSG_LIBS = -l:libnanomsg.a
|
||||
|
||||
ifeq ($(ARCH),aarch64) |
||||
CFLAGS += -mcpu=cortex-a57
|
||||
CXXFLAGS += -mcpu=cortex-a57
|
||||
endif |
||||
|
||||
|
||||
EXTRA_LIBS = -lpthread
|
||||
|
||||
ifeq ($(ARCH),x86_64) |
||||
NANOMSG_FLAGS = -I$(BASEDIR)/phonelibs/nanomsg/x64/include
|
||||
NANOMSG_LIBS = -L$(BASEDIR)/phonelibs/nanomsg/x64/lib \
|
||||
-lnanomsg -Wl,-rpath,$(BASEDIR)/phonelibs/nanomsg/x64/lib
|
||||
else |
||||
EXTRA_LIBS += -llog -luuid
|
||||
endif |
||||
|
||||
.PHONY: all |
||||
all: sender receiver |
||||
|
||||
receiver: receiver.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(NANOMSG_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
sender: sender.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(NANOMSG_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
%.o: %.cc |
||||
@echo "[ CXX ] $@"
|
||||
$(CXX) $(CXXFLAGS) -MMD \
|
||||
-Iinclude -I.. -I../.. \
|
||||
$(NANOMSG_FLAGS) \
|
||||
$(JSON11_FLAGS) \
|
||||
$(JSON_FLAGS) \
|
||||
-I../ \
|
||||
-I../../ \
|
||||
-c -o '$@' '$<'
|
||||
|
||||
|
||||
.PHONY: clean |
||||
clean: |
||||
rm -f *.d sender receiver
|
@ -0,0 +1,48 @@ |
||||
#include <future> |
||||
#include <cassert> |
||||
#include <iostream> |
||||
#include <cstring> |
||||
#include <thread> |
||||
|
||||
#include <nanomsg/nn.h> |
||||
#include <nanomsg/pubsub.h> |
||||
#include <nanomsg/tcp.h> |
||||
|
||||
#define N 1024 |
||||
|
||||
int sub_sock(const char *endpoint) { |
||||
int sock = nn_socket(AF_SP, NN_SUB); |
||||
assert(sock >= 0); |
||||
|
||||
nn_setsockopt(sock, NN_SUB, NN_SUB_SUBSCRIBE, "", 0); |
||||
assert(nn_connect(sock, endpoint) >= 0); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
int pub_sock(const char *endpoint){ |
||||
int sock = nn_socket(AF_SP, NN_PUB); |
||||
assert(sock >= 0); |
||||
|
||||
int b = 1; |
||||
nn_setsockopt(sock, NN_TCP, NN_TCP_NODELAY, &b, sizeof(b)); |
||||
|
||||
assert(nn_bind(sock, endpoint) >= 0); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
int main(int argc, char *argv[]) { |
||||
auto p_sock = pub_sock("tcp://*:10011"); |
||||
auto s_sock = sub_sock("tcp://127.0.0.1:10010"); |
||||
std::cout << "Ready!" << std::endl; |
||||
|
||||
char * msg = new char[N]; |
||||
|
||||
while (true){ |
||||
int bytes = nn_recv(s_sock, msg, N, 0); |
||||
nn_send(p_sock, msg, bytes, 0); |
||||
} |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,73 @@ |
||||
#include <iostream> |
||||
#include <cassert> |
||||
#include <chrono> |
||||
#include <thread> |
||||
|
||||
#include <nanomsg/nn.h> |
||||
#include <nanomsg/pubsub.h> |
||||
#include <nanomsg/tcp.h> |
||||
|
||||
|
||||
#define N 1024 |
||||
#define MSGS 1e5 |
||||
|
||||
int sub_sock(const char *endpoint) { |
||||
int sock = nn_socket(AF_SP, NN_SUB); |
||||
assert(sock >= 0); |
||||
|
||||
nn_setsockopt(sock, NN_SUB, NN_SUB_SUBSCRIBE, "", 0); |
||||
|
||||
int timeout = 100; |
||||
nn_setsockopt(sock, NN_SOL_SOCKET, NN_RCVTIMEO, &timeout , sizeof(timeout)); |
||||
|
||||
assert(nn_connect(sock, endpoint) >= 0); |
||||
return sock; |
||||
} |
||||
|
||||
int pub_sock(const char *endpoint){ |
||||
int sock = nn_socket(AF_SP, NN_PUB); |
||||
assert(sock >= 0); |
||||
|
||||
int b = 1; |
||||
nn_setsockopt(sock, NN_TCP, NN_TCP_NODELAY, &b, sizeof(b)); |
||||
|
||||
assert(nn_bind(sock, endpoint) >= 0); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
|
||||
int main(int argc, char *argv[]) { |
||||
auto p_sock = pub_sock("tcp://*:10010"); |
||||
auto s_sock = sub_sock("tcp://127.0.0.1:10011"); |
||||
std::cout << "Ready!" << std::endl; |
||||
|
||||
// auto p_sock = pub_sock("ipc:///tmp/feeds/3");
|
||||
// auto s_sock = sub_sock("ipc:///tmp/feeds/2");
|
||||
|
||||
char * msg = new char[N]; |
||||
auto start = std::chrono::steady_clock::now(); |
||||
|
||||
|
||||
for (int i = 0; i < MSGS; i++){ |
||||
sprintf(msg, "%d", i); |
||||
|
||||
nn_send(p_sock, msg, N, 0); |
||||
int bytes = nn_recv(s_sock, msg, N, 0); |
||||
|
||||
if (bytes < 0) { |
||||
std::cout << "Timeout" << std::endl; |
||||
} |
||||
|
||||
} |
||||
auto end = std::chrono::steady_clock::now(); |
||||
|
||||
|
||||
double elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count() / 1e9; |
||||
double throughput = ((double) MSGS / (double) elapsed); |
||||
|
||||
std::cout << "Elapsed: " << elapsed << " s" << std::endl; |
||||
std::cout << "Throughput: " << throughput << " msg/s" << std::endl; |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,2 @@ |
||||
receiver |
||||
sender |
@ -0,0 +1,68 @@ |
||||
CC = clang
|
||||
CXX = clang++
|
||||
|
||||
ARCH := $(shell uname -m)
|
||||
OS := $(shell uname -o)
|
||||
|
||||
BASEDIR = ../../../..
|
||||
PHONELIBS = ../../../../phonelibs
|
||||
|
||||
WARN_FLAGS = -Werror=implicit-function-declaration \
|
||||
-Werror=incompatible-pointer-types \
|
||||
-Werror=int-conversion \
|
||||
-Werror=return-type \
|
||||
-Werror=format-extra-args
|
||||
|
||||
CFLAGS = -std=gnu11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
CXXFLAGS = -std=c++11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
NNG_LIBS = -l:libnng.a
|
||||
|
||||
ifeq ($(ARCH),aarch64) |
||||
CFLAGS += -mcpu=cortex-a57
|
||||
CXXFLAGS += -mcpu=cortex-a57
|
||||
endif |
||||
|
||||
|
||||
EXTRA_LIBS = -lpthread
|
||||
|
||||
ifeq ($(ARCH),x86_64) |
||||
ZMQ_FLAGS = -I$(BASEDIR)/phonelibs/nng/x64/include
|
||||
NNG_LIBS = -L$(BASEDIR)/phonelibs/nng/x64/lib \
|
||||
-l:libnng.a
|
||||
else |
||||
EXTRA_LIBS += -llog -luuid
|
||||
endif |
||||
|
||||
.PHONY: all |
||||
all: sender receiver |
||||
|
||||
receiver: receiver.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(CEREAL_LIBS) \
|
||||
$(NNG_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
sender: sender.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(CEREAL_LIBS) \
|
||||
$(NNG_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
%.o: %.cc |
||||
@echo "[ CXX ] $@"
|
||||
$(CXX) $(CXXFLAGS) -MMD \
|
||||
-Iinclude -I.. -I../.. \
|
||||
$(CEREAL_CXXFLAGS) \
|
||||
$(ZMQ_FLAGS) \
|
||||
$(JSON11_FLAGS) \
|
||||
$(JSON_FLAGS) \
|
||||
-I../ \
|
||||
-I../../ \
|
||||
-c -o '$@' '$<'
|
||||
|
||||
|
||||
.PHONY: clean |
||||
clean: |
||||
rm -f *.d sender receiver
|
@ -0,0 +1,56 @@ |
||||
#include <future> |
||||
#include <cassert> |
||||
#include <iostream> |
||||
#include <cstring> |
||||
#include <thread> |
||||
|
||||
#include <nng/nng.h> |
||||
#include <nng/protocol/pubsub0/pub.h> |
||||
#include <nng/protocol/pubsub0/sub.h> |
||||
|
||||
nng_socket sub_sock(const char *endpoint) { |
||||
nng_socket sock; |
||||
int r; |
||||
r = nng_sub0_open(&sock); |
||||
assert(r == 0); |
||||
|
||||
nng_setopt(sock, NNG_OPT_SUB_SUBSCRIBE, "", 0); |
||||
|
||||
while (true){ |
||||
r = nng_dial(sock, endpoint, NULL, 0); |
||||
|
||||
if (r == 0){ |
||||
break; |
||||
} |
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10)); |
||||
|
||||
} |
||||
return sock; |
||||
} |
||||
|
||||
nng_socket pub_sock(const char *endpoint){ |
||||
nng_socket sock; |
||||
int r; |
||||
r = nng_pub0_open(&sock); |
||||
assert(r == 0); |
||||
r = nng_listen(sock, endpoint, NULL, 0); |
||||
assert(r == 0); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
int main(int argc, char *argv[]) { |
||||
// auto p_sock = pub_sock("tcp://*:10004");
|
||||
// auto s_sock = sub_sock("tcp://127.0.0.1:10003");
|
||||
|
||||
auto p_sock = pub_sock("ipc:///tmp/feeds/2"); |
||||
auto s_sock = sub_sock("ipc:///tmp/feeds/3"); |
||||
|
||||
while (true){ |
||||
nng_msg *msg; |
||||
nng_recvmsg(s_sock, &msg, 0); |
||||
nng_sendmsg(p_sock, msg, 0); |
||||
} |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,78 @@ |
||||
#include <iostream> |
||||
#include <cassert> |
||||
#include <chrono> |
||||
#include <thread> |
||||
|
||||
#include <nng/nng.h> |
||||
#include <nng/protocol/pubsub0/pub.h> |
||||
#include <nng/protocol/pubsub0/sub.h> |
||||
|
||||
|
||||
#define N 1024 |
||||
#define MSGS 1e5 |
||||
|
||||
nng_socket sub_sock(const char *endpoint) { |
||||
nng_socket sock; |
||||
int r; |
||||
r = nng_sub0_open(&sock); |
||||
assert(r == 0); |
||||
|
||||
nng_setopt(sock, NNG_OPT_SUB_SUBSCRIBE, "", 0); |
||||
nng_setopt_ms(sock, NNG_OPT_RECVTIMEO, 100); |
||||
|
||||
while (true){ |
||||
r = nng_dial(sock, endpoint, NULL, 0); |
||||
|
||||
if (r == 0){ |
||||
break; |
||||
} |
||||
std::this_thread::sleep_for(std::chrono::milliseconds(10)); |
||||
} |
||||
return sock; |
||||
} |
||||
|
||||
nng_socket pub_sock(const char *endpoint){ |
||||
nng_socket sock; |
||||
int r; |
||||
r = nng_pub0_open(&sock); |
||||
assert(r == 0); |
||||
r = nng_listen(sock, endpoint, NULL, 0); |
||||
assert(r == 0); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
|
||||
int main(int argc, char *argv[]) { |
||||
// auto p_sock = pub_sock("tcp://*:10003");
|
||||
// auto s_sock = sub_sock("tcp://127.0.0.1:10004");
|
||||
|
||||
auto p_sock = pub_sock("ipc:///tmp/feeds/3"); |
||||
auto s_sock = sub_sock("ipc:///tmp/feeds/2"); |
||||
|
||||
|
||||
auto start = std::chrono::steady_clock::now(); |
||||
|
||||
for (int i = 0; i < MSGS; i++){ |
||||
nng_msg *msg; |
||||
nng_msg_alloc(&msg, N); |
||||
nng_sendmsg(p_sock, msg, 0); |
||||
|
||||
nng_msg *rmsg; |
||||
int r = nng_recvmsg(s_sock, &rmsg, 0); |
||||
|
||||
if (r) { |
||||
std::cout << "Timeout" << std::endl; |
||||
} |
||||
} |
||||
auto end = std::chrono::steady_clock::now(); |
||||
|
||||
|
||||
double elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count() / 1e9; |
||||
double throughput = ((double) MSGS / (double) elapsed); |
||||
|
||||
std::cout << "Elapsed: " << elapsed << " s" << std::endl; |
||||
std::cout << "Throughput: " << throughput << " msg/s" << std::endl; |
||||
|
||||
return 0; |
||||
} |
@ -0,0 +1,2 @@ |
||||
receiver |
||||
sender |
@ -0,0 +1,71 @@ |
||||
CC = clang
|
||||
CXX = clang++
|
||||
|
||||
ARCH := $(shell uname -m)
|
||||
OS := $(shell uname -o)
|
||||
|
||||
BASEDIR = ../../../../../
|
||||
PHONELIBS = ../../../../../phonelibs
|
||||
|
||||
WARN_FLAGS = -Werror=implicit-function-declaration \
|
||||
-Werror=incompatible-pointer-types \
|
||||
-Werror=int-conversion \
|
||||
-Werror=return-type \
|
||||
-Werror=format-extra-args
|
||||
|
||||
CFLAGS = -std=gnu11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
CXXFLAGS = -std=c++11 -g -fPIC -O2 $(WARN_FLAGS) -Wall
|
||||
ZMQ_LIBS = -l:libczmq.a -l:libzmq.a
|
||||
|
||||
ifeq ($(ARCH),aarch64) |
||||
CFLAGS += -mcpu=cortex-a57
|
||||
CXXFLAGS += -mcpu=cortex-a57
|
||||
ZMQ_LIBS += -lgnustl_shared
|
||||
endif |
||||
|
||||
|
||||
EXTRA_LIBS = -lpthread
|
||||
|
||||
ifeq ($(ARCH),x86_64) |
||||
ZMQ_FLAGS = -I$(BASEDIR)/phonelibs/zmq/x64/include
|
||||
ZMQ_LIBS = -L$(BASEDIR)/external/zmq/lib \
|
||||
-l:libczmq.a -l:libzmq.a
|
||||
ZMQ_SHARED_LIBS = -L$(BASEDIR)/external/zmq/lib \
|
||||
-lczmq -lzmq
|
||||
else |
||||
EXTRA_LIBS += -llog -luuid
|
||||
endif |
||||
|
||||
.PHONY: all |
||||
all: sender receiver |
||||
|
||||
receiver: receiver.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(CEREAL_LIBS) \
|
||||
$(ZMQ_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
sender: sender.o |
||||
@echo "[ LINK ] $@"
|
||||
$(CXX) -fPIC -o '$@' $^ \
|
||||
$(CEREAL_LIBS) \
|
||||
$(ZMQ_LIBS) \
|
||||
$(EXTRA_LIBS)
|
||||
|
||||
%.o: %.cc |
||||
@echo "[ CXX ] $@"
|
||||
$(CXX) $(CXXFLAGS) -MMD \
|
||||
-Iinclude -I.. -I../.. \
|
||||
$(CEREAL_CXXFLAGS) \
|
||||
$(ZMQ_FLAGS) \
|
||||
$(JSON11_FLAGS) \
|
||||
$(JSON_FLAGS) \
|
||||
-I../ \
|
||||
-I../../ \
|
||||
-c -o '$@' '$<'
|
||||
|
||||
|
||||
.PHONY: clean |
||||
clean: |
||||
rm -f *.d sender receiver
|
@ -0,0 +1,49 @@ |
||||
#include <future> |
||||
#include <iostream> |
||||
#include <cstring> |
||||
|
||||
#include <zmq.h> |
||||
|
||||
// #define IPC
|
||||
|
||||
void *sub_sock(void *ctx, const char *endpoint) { |
||||
void* sock = zmq_socket(ctx, ZMQ_SUB); |
||||
zmq_connect(sock, endpoint); |
||||
zmq_setsockopt(sock, ZMQ_SUBSCRIBE, "", 0); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
void *pub_sock(void *ctx, const char *endpoint){ |
||||
void * sock = zmq_socket(ctx, ZMQ_PUB); |
||||
|
||||
zmq_bind(sock, endpoint); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
int main(int argc, char *argv[]) { |
||||
auto ctx = zmq_ctx_new(); |
||||
|
||||
#ifdef IPC |
||||
auto s_sock = sub_sock(ctx, "ipc:///tmp/q0"); |
||||
auto p_sock = pub_sock(ctx, "ipc:///tmp/q1"); |
||||
#else |
||||
auto s_sock = sub_sock(ctx, "tcp://localhost:10005"); |
||||
auto p_sock = pub_sock(ctx, "tcp://*:10004"); |
||||
#endif |
||||
|
||||
zmq_msg_t msg; |
||||
zmq_msg_init(&msg); |
||||
|
||||
|
||||
while (true){ |
||||
zmq_msg_recv(&msg, s_sock, 0); |
||||
zmq_msg_send(&msg, p_sock, ZMQ_DONTWAIT); |
||||
} |
||||
|
||||
zmq_msg_close(&msg); |
||||
zmq_close(p_sock); |
||||
zmq_close(s_sock); |
||||
return 0; |
||||
} |
@ -0,0 +1,65 @@ |
||||
#include <iostream> |
||||
#include <zmq.h> |
||||
#include <chrono> |
||||
|
||||
#define N 1024 |
||||
#define MSGS 1e5 |
||||
|
||||
// #define IPC
|
||||
|
||||
void *sub_sock(void *ctx, const char *endpoint) { |
||||
void* sock = zmq_socket(ctx, ZMQ_SUB); |
||||
zmq_connect(sock, endpoint); |
||||
zmq_setsockopt(sock, ZMQ_SUBSCRIBE, "", 0); |
||||
|
||||
int timeout = 100; |
||||
zmq_setsockopt(sock, ZMQ_RCVTIMEO, &timeout, sizeof(int)); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
void *pub_sock(void *ctx, const char *endpoint){ |
||||
void * sock = zmq_socket(ctx, ZMQ_PUB); |
||||
zmq_bind(sock, endpoint); |
||||
|
||||
return sock; |
||||
} |
||||
|
||||
|
||||
int main(int argc, char *argv[]) { |
||||
auto ctx = zmq_ctx_new(); |
||||
|
||||
#ifdef IPC |
||||
auto s_sock = sub_sock(ctx, "ipc:///tmp/q1"); |
||||
auto p_sock = pub_sock(ctx, "ipc:///tmp/q0"); |
||||
#else |
||||
auto s_sock = sub_sock(ctx, "tcp://127.0.0.1:10004"); |
||||
auto p_sock = pub_sock(ctx, "tcp://*:10005"); |
||||
#endif |
||||
|
||||
zmq_msg_t msg; |
||||
zmq_msg_init_size (&msg, N); |
||||
|
||||
auto start = std::chrono::steady_clock::now(); |
||||
|
||||
for (int i = 0; i < MSGS; i++){ |
||||
zmq_msg_send(&msg, p_sock, ZMQ_DONTWAIT); |
||||
int r = zmq_msg_recv(&msg, s_sock, 0); |
||||
if (r) { |
||||
start = std::chrono::steady_clock::now(); |
||||
std::cout << "Timeout" << std::endl; |
||||
} |
||||
} |
||||
auto end = std::chrono::steady_clock::now(); |
||||
|
||||
|
||||
double elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - start).count() / 1e9; |
||||
double throughput = ((double) MSGS / (double) elapsed); |
||||
|
||||
std::cout << "Elapsed: " << elapsed << " s" << std::endl; |
||||
std::cout << "Throughput: " << throughput << " msg/s" << std::endl; |
||||
|
||||
zmq_close(p_sock); |
||||
zmq_close(s_sock); |
||||
return 0; |
||||
} |
@ -0,0 +1,17 @@ |
||||
#!/usr/bin/env python3 |
||||
import time |
||||
import cereal.messaging as messaging |
||||
|
||||
|
||||
def init_message_bench(N=100000): |
||||
t = time.time() |
||||
for _ in range(N): |
||||
dat = messaging.new_message() |
||||
dat.init('controlsState') |
||||
|
||||
dt = time.time() - t |
||||
print("Init message %d its, %f s" % (N, dt)) |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
init_message_bench() |
@ -0,0 +1,21 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
from cereal.services import service_list |
||||
import cereal.messaging as messaging |
||||
from cereal import log |
||||
|
||||
def mock(): |
||||
traffic_events = messaging.pub_sock('uiNavigationEvent') |
||||
|
||||
while 1: |
||||
m = messaging.new_message() |
||||
m.init('uiNavigationEvent') |
||||
m.uiNavigationEvent.type = log.UiNavigationEvent.Type.mergeRight |
||||
m.uiNavigationEvent.status = log.UiNavigationEvent.Status.active |
||||
m.uiNavigationEvent.distanceTo = 100. |
||||
traffic_events.send(m.to_bytes()) |
||||
time.sleep(0.01) |
||||
|
||||
if __name__=="__main__": |
||||
mock() |
@ -0,0 +1,22 @@ |
||||
#!/usr/bin/env python3 |
||||
import time |
||||
import zmq |
||||
from hexdump import hexdump |
||||
|
||||
from common.realtime import Ratekeeper |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
if __name__ == "__main__": |
||||
controls_state = messaging.pub_sock('controlsState') |
||||
|
||||
rk = Ratekeeper(100) |
||||
while 1: |
||||
dat = messaging.new_message() |
||||
dat.init('controlsState') |
||||
|
||||
dat.controlsState.vEgo = 25. |
||||
dat.controlsState.enabled = True |
||||
controls_state.send(dat.to_bytes()) |
||||
|
||||
rk.keep_time() |
@ -0,0 +1,27 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
from hexdump import hexdump |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
from cereal import log |
||||
|
||||
def leadRange(start, end, step): |
||||
x = start |
||||
while x < end: |
||||
yield x |
||||
x += (x * step) |
||||
|
||||
def mock_lead(): |
||||
radarState = messaging.pub_sock('radarState') |
||||
while 1: |
||||
m = messaging.new_message() |
||||
m.init('radarState') |
||||
m.radarState.leadOne.status = True |
||||
for x in leadRange(3.0, 65.0, 0.005): |
||||
m.radarState.leadOne.dRel = x |
||||
radarState.send(m.to_bytes()) |
||||
time.sleep(0.01) |
||||
|
||||
if __name__=="__main__": |
||||
mock_lead() |
@ -0,0 +1,46 @@ |
||||
# mock_gps.py: Publishes a generated path moving at 15m/s to gpsLocation |
||||
# USAGE: python mock_gps.py |
||||
# Then start manager |
||||
|
||||
from itertools import cycle |
||||
import time |
||||
import zmq |
||||
|
||||
from cereal import log |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
degrees_per_meter = 0.000009000009 # approximation |
||||
start_lat = 43.64199141443989 |
||||
start_lng = -94.97520411931725 |
||||
|
||||
def gen_path(length_seconds, speed=15): |
||||
return [{"lat": start_lat, |
||||
"lng": start_lng + speed * i * degrees_per_meter, # moving along longitudinal axis at speed m/s |
||||
"speed": speed} |
||||
for i in range(1, length_seconds + 1)] |
||||
|
||||
if __name__ == '__main__': |
||||
gpsLocation = messaging.pub_sock('gpsLocation') |
||||
|
||||
path_stopped_5s = [{"lat": start_lat, "lng": start_lng, "speed": 0}] * 5 |
||||
path_moving = gen_path(30, speed=15) |
||||
path_stopped_5s_then_moving = path_stopped_5s + path_moving |
||||
|
||||
for point in cycle(path_stopped_5s_then_moving): |
||||
print('sending gpsLocation from point: {}'.format(str(point))) |
||||
dat = messaging.new_message() |
||||
dat.init('gpsLocation') |
||||
dat.gpsLocation.latitude = point['lat'] |
||||
dat.gpsLocation.longitude = point['lng'] |
||||
dat.gpsLocation.speed = point['speed'] |
||||
dat.gpsLocation.flags = 0 |
||||
dat.gpsLocation.altitude = 0 |
||||
dat.gpsLocation.bearing = 0 # todo we can mock this |
||||
dat.gpsLocation.accuracy = 1 |
||||
dat.gpsLocation.timestamp = int(time.time() * 1000) |
||||
dat.gpsLocation.source = log.GpsLocationData.SensorSource.android |
||||
|
||||
gpsLocation.send(dat.to_bytes()) |
||||
time.sleep(1) |
||||
|
@ -0,0 +1,27 @@ |
||||
#!/usr/bin/env python3 |
||||
import time |
||||
import zmq |
||||
|
||||
from cereal import log |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
gpsLocationExternal = messaging.pub_sock('gpsLocationExternal') |
||||
|
||||
while True: |
||||
dat = messaging.new_message() |
||||
dat.init('gpsLocationExternal') |
||||
dat.gpsLocationExternal.latitude = 37.6513687 |
||||
dat.gpsLocationExternal.longitude = -122.4535056 |
||||
dat.gpsLocationExternal.speed = 28.2 |
||||
dat.gpsLocationExternal.flags = 1 |
||||
dat.gpsLocationExternal.altitude = 75. |
||||
dat.gpsLocationExternal.bearing = 145.5 |
||||
dat.gpsLocationExternal.accuracy = 1. |
||||
dat.gpsLocationExternal.timestamp = int(time.time() * 1000) |
||||
dat.gpsLocationExternal.source = log.GpsLocationData.SensorSource.ublox |
||||
|
||||
gpsLocationExternal.send(dat.to_bytes()) |
||||
time.sleep(.1) |
@ -0,0 +1,22 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
from hexdump import hexdump |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
from cereal import log |
||||
|
||||
def mock_x(): |
||||
liveMpc = messaging.pub_sock('liveMpc') |
||||
while 1: |
||||
m = messaging.new_message() |
||||
mx = [] |
||||
m.init('liveMpc') |
||||
for x in range(0, 100): |
||||
mx.append(x*1.0) |
||||
m.liveMpc.x = mx |
||||
|
||||
liveMpc.send(m.to_bytes()) |
||||
|
||||
if __name__=="__main__": |
||||
mock_x() |
@ -0,0 +1,22 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
from cereal.services import service_list |
||||
import cereal.messaging as messaging |
||||
from cereal import log |
||||
|
||||
def mock(): |
||||
traffic_events = messaging.pub_sock('trafficEvents') |
||||
|
||||
while 1: |
||||
m = messaging.new_message() |
||||
m.init('trafficEvents', 1) |
||||
m.trafficEvents[0].type = log.TrafficEvent.Type.stopSign |
||||
m.trafficEvents[0].resuming = False |
||||
m.trafficEvents[0].distance = 100. |
||||
m.trafficEvents[0].action = log.TrafficEvent.Action.stop |
||||
traffic_events.send(m.to_bytes()) |
||||
time.sleep(0.01) |
||||
|
||||
if __name__=="__main__": |
||||
mock() |
@ -0,0 +1,41 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
|
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
if __name__ == "__main__": |
||||
poller = zmq.Poller() |
||||
|
||||
fsock = messaging.sub_sock("frame", poller) |
||||
msock = messaging.sub_sock("model", poller) |
||||
|
||||
frmTimes = {} |
||||
proc = [] |
||||
|
||||
last100 = [] |
||||
|
||||
while 1: |
||||
polld = poller.poll(timeout=1000) |
||||
for sock, mode in polld: |
||||
if mode != zmq.POLLIN: |
||||
continue |
||||
if sock == fsock: |
||||
f = messaging.recv_one(sock) |
||||
frmTimes[f.frame.frameId] = f.frame.timestampEof |
||||
else: |
||||
proc.append(messaging.recv_one(sock)) |
||||
nproc = [] |
||||
for mm in proc: |
||||
fid = mm.model.frameId |
||||
|
||||
if fid in frmTimes: |
||||
tm = (mm.logMonoTime-frmTimes[fid])/1e6 |
||||
del frmTimes[fid] |
||||
last100.append(tm) |
||||
last100 = last100[-100:] |
||||
print("%10d: %.2f ms min: %.2f max: %.2f" % (fid, tm, min(last100), max(last100))) |
||||
else: |
||||
nproc.append(mm) |
||||
proc = nproc |
||||
|
@ -0,0 +1,96 @@ |
||||
import timeit |
||||
|
||||
import numpy as np |
||||
import numpy.linalg |
||||
from scipy.linalg import cho_factor, cho_solve |
||||
|
||||
# We are trying to solve the following system |
||||
# (A.T * A) * x = A.T * b |
||||
# Where x are the polynomial coefficients and b is are the input points |
||||
|
||||
# First we build A |
||||
deg = 3 |
||||
x = np.arange(50 * 1.0) |
||||
A = np.vstack(tuple(x**n for n in range(deg, -1, -1))).T |
||||
|
||||
# The first way to solve this is using the pseudoinverse, which can be precomputed |
||||
# x = (A.T * A)^-1 * A^T * b = PINV b |
||||
PINV = np.linalg.pinv(A) |
||||
|
||||
# Another way is using the Cholesky decomposition |
||||
# We can note that at (A.T * A) is always positive definite |
||||
# By precomputing the Cholesky decomposition we can efficiently solve |
||||
# systems of the form (A.T * A) x = c |
||||
CHO = cho_factor(np.dot(A.T, A)) |
||||
|
||||
|
||||
def model_polyfit_old(points, deg=3): |
||||
A = np.vstack(tuple(x**n for n in range(deg, -1, -1))).T |
||||
pinv = np.linalg.pinv(A) |
||||
return np.dot(pinv, map(float, points)) |
||||
|
||||
|
||||
def model_polyfit(points, deg=3): |
||||
A = np.vander(x, deg + 1) |
||||
pinv = np.linalg.pinv(A) |
||||
return np.dot(pinv, map(float, points)) |
||||
|
||||
|
||||
def model_polyfit_cho(points, deg=3): |
||||
A = np.vander(x, deg + 1) |
||||
cho = cho_factor(np.dot(A.T, A)) |
||||
c = np.dot(A.T, points) |
||||
return cho_solve(cho, c, check_finite=False) |
||||
|
||||
|
||||
def model_polyfit_np(points, deg=3): |
||||
return np.polyfit(x, points, deg) |
||||
|
||||
|
||||
def model_polyfit_lstsq(points, deg=3): |
||||
A = np.vander(x, deg + 1) |
||||
return np.linalg.lstsq(A, points, rcond=None)[0] |
||||
|
||||
|
||||
TEST_DATA = np.linspace(0, 5, num=50) + 1. |
||||
|
||||
|
||||
def time_pinv_old(): |
||||
model_polyfit_old(TEST_DATA) |
||||
|
||||
|
||||
def time_pinv(): |
||||
model_polyfit(TEST_DATA) |
||||
|
||||
|
||||
def time_cho(): |
||||
model_polyfit_cho(TEST_DATA) |
||||
|
||||
|
||||
def time_np(): |
||||
model_polyfit_np(TEST_DATA) |
||||
|
||||
|
||||
def time_lstsq(): |
||||
model_polyfit_lstsq(TEST_DATA) |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
# Verify correct results |
||||
pinv_old = model_polyfit_old(TEST_DATA) |
||||
pinv = model_polyfit(TEST_DATA) |
||||
cho = model_polyfit_cho(TEST_DATA) |
||||
numpy = model_polyfit_np(TEST_DATA) |
||||
lstsq = model_polyfit_lstsq(TEST_DATA) |
||||
|
||||
assert all(np.isclose(pinv, pinv_old)) |
||||
assert all(np.isclose(pinv, cho)) |
||||
assert all(np.isclose(pinv, numpy)) |
||||
assert all(np.isclose(pinv, lstsq)) |
||||
|
||||
# Run benchmark |
||||
print("Pseudo inverse (old)", timeit.timeit("time_pinv_old()", setup="from __main__ import time_pinv_old", number=10000)) |
||||
print("Pseudo inverse", timeit.timeit("time_pinv()", setup="from __main__ import time_pinv", number=10000)) |
||||
print("Cholesky", timeit.timeit("time_cho()", setup="from __main__ import time_cho", number=10000)) |
||||
print("Numpy leastsq", timeit.timeit("time_lstsq()", setup="from __main__ import time_lstsq", number=10000)) |
||||
print("Numpy polyfit", timeit.timeit("time_np()", setup="from __main__ import time_np", number=10000)) |
@ -0,0 +1,64 @@ |
||||
#!/usr/bin/env python3 |
||||
import os |
||||
import time |
||||
import sys |
||||
from datetime import datetime |
||||
|
||||
def average(avg, sample): |
||||
# Weighted avg between existing value and new sample |
||||
return ((avg[0] * avg[1] + sample) / (avg[1] + 1), avg[1] + 1) |
||||
|
||||
|
||||
if __name__ == '__main__': |
||||
try: |
||||
if len(sys.argv) > 1 and sys.argv[1] == "--charge": |
||||
print("not disabling charging") |
||||
else: |
||||
print("disabling charging") |
||||
os.system('echo "0" > /sys/class/power_supply/battery/charging_enabled') |
||||
|
||||
voltage_average = (0., 0) # average, count |
||||
current_average = (0., 0) |
||||
power_average = (0., 0) |
||||
capacity_average = (0., 0) |
||||
bat_temp_average = (0., 0) |
||||
start_time = datetime.now() |
||||
while 1: |
||||
with open("/sys/class/power_supply/bms/voltage_now") as f: |
||||
voltage = int(f.read()) / 1e6 # volts |
||||
|
||||
with open("/sys/class/power_supply/bms/current_now") as f: |
||||
current = int(f.read()) / 1e3 # ma |
||||
|
||||
power = voltage * current |
||||
|
||||
with open("/sys/class/power_supply/bms/capacity_raw") as f: |
||||
capacity = int(f.read()) / 1e2 # percent |
||||
|
||||
with open("/sys/class/power_supply/bms/temp") as f: |
||||
bat_temp = int(f.read()) / 1e1 # celsius |
||||
|
||||
# compute averages |
||||
voltage_average = average(voltage_average, voltage) |
||||
current_average = average(current_average, current) |
||||
power_average = average(power_average, power) |
||||
capacity_average = average(capacity_average, capacity) |
||||
bat_temp_average = average(bat_temp_average, bat_temp) |
||||
|
||||
print("%.2f volts %12.2f ma %12.2f mW %8.2f%% battery %8.1f degC" % (voltage, current, power, capacity, bat_temp)) |
||||
time.sleep(0.1) |
||||
finally: |
||||
stop_time = datetime.now() |
||||
print("\n----------------------Average-----------------------------------") |
||||
voltage = voltage_average[0] |
||||
current = current_average[0] |
||||
power = power_average[0] |
||||
capacity = capacity_average[0] |
||||
bat_temp = bat_temp_average[0] |
||||
print("%.2f volts %12.2f ma %12.2f mW %8.2f%% battery %8.1f degC" % (voltage, current, power, capacity, bat_temp)) |
||||
print(" {:.2f} Seconds {} samples".format((stop_time-start_time).total_seconds(), voltage_average[1])) |
||||
print("----------------------------------------------------------------") |
||||
|
||||
# reenable charging |
||||
os.system('echo "1" > /sys/class/power_supply/battery/charging_enabled') |
||||
print("charging enabled\n") |
@ -0,0 +1,22 @@ |
||||
#!/usr/bin/env python3 |
||||
import time |
||||
|
||||
from common.realtime import sec_since_boot, monotonic_time |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
N = 100000 |
||||
|
||||
t = time.time() |
||||
for _ in range(N): |
||||
monotonic_time() |
||||
dt = time.time() - t |
||||
|
||||
print("Monotonic", dt) |
||||
|
||||
t = time.time() |
||||
for _ in range(N): |
||||
sec_since_boot() |
||||
dt = time.time() - t |
||||
|
||||
print("Boot", dt) |
@ -0,0 +1,79 @@ |
||||
#!/usr/bin/env python3 |
||||
|
||||
import argparse |
||||
import time |
||||
import os |
||||
|
||||
from tqdm import tqdm |
||||
|
||||
from cereal.messaging import PubMaster, recv_one, sub_sock |
||||
from cereal.services import service_list |
||||
from tools.lib.logreader import LogReader |
||||
from xx.chffr.lib.route import Route, RouteSegment |
||||
from tools.lib.route_framereader import RouteFrameReader |
||||
from common.column_store import save_dict_as_column_store |
||||
from xx.pipeline.lib.log_time_series import append_dict |
||||
from selfdrive.test.process_replay.compare_logs import save_log |
||||
|
||||
if __name__ == "__main__": |
||||
parser = argparse.ArgumentParser(description="Run visiond on segment") |
||||
parser.add_argument("segment_name", help="The segment to run") |
||||
parser.add_argument("output_path", help="The output file") |
||||
|
||||
args = parser.parse_args() |
||||
segment = RouteSegment.from_canonical_name(args.segment_name) |
||||
route = Route(segment._name._route_name) |
||||
|
||||
frame_id_lookup = {} |
||||
frame_reader = RouteFrameReader(route.camera_paths(), None, frame_id_lookup, readahead=True) |
||||
|
||||
msgs = list(LogReader(segment.log_path)) |
||||
|
||||
pm = PubMaster(['liveCalibration', 'frame']) |
||||
model_sock = sub_sock('model') |
||||
|
||||
# Read encodeIdx |
||||
for msg in msgs: |
||||
if msg.which() == 'encodeIdx': |
||||
frame_id_lookup[msg.encodeIdx.frameId] = (msg.encodeIdx.segmentNum, msg.encodeIdx.segmentId) |
||||
|
||||
# Send some livecalibration messages to initalize visiond |
||||
for msg in msgs: |
||||
if msg.which() == 'liveCalibration': |
||||
pm.send('liveCalibration', msg.as_builder()) |
||||
|
||||
time.sleep(1.0) |
||||
values = {} |
||||
|
||||
out_msgs = [] |
||||
for msg in tqdm(msgs): |
||||
w = msg.which() |
||||
|
||||
if w == 'liveCalibration': |
||||
pm.send(w, msg.as_builder()) |
||||
|
||||
if w == 'frame': |
||||
msg = msg.as_builder() |
||||
|
||||
frame_id = msg.frame.frameId |
||||
img = frame_reader.get(frame_id, pix_fmt="rgb24")[:,:,::-1] |
||||
|
||||
msg.frame.image = img.flatten().tobytes() |
||||
pm.send(w, msg) |
||||
|
||||
model = recv_one(model_sock) |
||||
model = model.as_builder() |
||||
model.logMonoTime = 0 |
||||
model = model.as_reader() |
||||
out_msgs.append(model) |
||||
|
||||
save_log(args.output_path, out_msgs) |
||||
|
||||
# tm = model.logMonoTime / 1.0e9 |
||||
# model = model.model |
||||
# append_dict("model/data/path", tm, model.path.to_dict(), values) |
||||
# append_dict("model/data/left_lane", tm, model.leftLane.to_dict(), values) |
||||
# append_dict("model/data/right_lane", tm, model.rightLane.to_dict(), values) |
||||
# append_dict("model/data/lead", tm, model.lead.to_dict(), values) |
||||
|
||||
# save_dict_as_column_store(values, os.path.join(args.output_path, "LiveVisionD", args.segment_name)) |
@ -0,0 +1,22 @@ |
||||
#!/usr/bin/env python3 |
||||
import time |
||||
import zmq |
||||
from hexdump import hexdump |
||||
|
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
if __name__ == "__main__": |
||||
controls_state = messaging.pub_sock('controlsState') |
||||
|
||||
while 1: |
||||
dat = messaging.new_message() |
||||
dat.init('controlsState') |
||||
|
||||
dat.controlsState.alertText1 = "alert text 1" |
||||
dat.controlsState.alertText2 = "alert text 2" |
||||
dat.controlsState.alertType = "test" |
||||
dat.controlsState.alertSound = "chimeDisengage" |
||||
controls_state.send(dat.to_bytes()) |
||||
|
||||
time.sleep(0.01) |
@ -0,0 +1,107 @@ |
||||
import os |
||||
import sys |
||||
|
||||
import zmq |
||||
from lru import LRU |
||||
|
||||
from cereal import log |
||||
from common.realtime import Ratekeeper |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
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 |
||||
|
||||
def cpu_dtotal(l1, l2): |
||||
t1_total = sum(cputime_total(ct) for ct in l1.cpuTimes) |
||||
t2_total = sum(cputime_total(ct) for ct in l2.cpuTimes) |
||||
return t2_total - t1_total |
||||
|
||||
def cpu_percent(l1, l2): |
||||
dtotal = cpu_dtotal(l1, l2) |
||||
t1_busy = sum(cputime_busy(ct) for ct in l1.cpuTimes) |
||||
t2_busy = sum(cputime_busy(ct) for ct in l2.cpuTimes) |
||||
|
||||
dbusy = t2_busy - t1_busy |
||||
|
||||
if dbusy < 0 or dtotal <= 0: |
||||
return 0.0 |
||||
return dbusy / dtotal |
||||
|
||||
def proc_cpu_percent(proc1, proc2, l1, l2): |
||||
dtotal = cpu_dtotal(l1, l2) |
||||
|
||||
dproc = (proc2.cpuUser+proc2.cpuSystem) - (proc1.cpuUser+proc1.cpuSystem) |
||||
if dproc < 0: |
||||
return 0.0 |
||||
|
||||
return dproc / dtotal |
||||
|
||||
def display_cpu(pl1, pl2): |
||||
l1, l2 = pl1.procLog, pl2.procLog |
||||
|
||||
print(cpu_percent(l1, l2)) |
||||
|
||||
procs1 = dict((proc.pid, proc) for proc in l1.procs) |
||||
procs2 = dict((proc.pid, proc) for proc in l2.procs) |
||||
|
||||
procs_print = 4 |
||||
|
||||
procs_with_percent = sorted((proc_cpu_percent(procs1[proc.pid], proc, l1, l2), proc) for proc in l2.procs |
||||
if proc.pid in procs1) |
||||
for percent, proc in procs_with_percent[-1:-procs_print-1:-1]: |
||||
print(percent, proc.name) |
||||
|
||||
print() |
||||
|
||||
|
||||
def main(): |
||||
frame_cache = LRU(16) |
||||
md_cache = LRU(16) |
||||
plan_cache = LRU(16) |
||||
|
||||
frame_sock = messaging.sub_sock('frame') |
||||
md_sock = messaging.sub_sock('model') |
||||
plan_sock = messaging.sub_sock('plan') |
||||
controls_state_sock = messaging.sub_sock('controlsState') |
||||
|
||||
proc = messaging.sub_sock('procLog') |
||||
pls = [None, None] |
||||
|
||||
rk = Ratekeeper(10) |
||||
while True: |
||||
|
||||
for msg in messaging.drain_sock(frame_sock): |
||||
frame_cache[msg.frame.frameId] = msg |
||||
|
||||
for msg in messaging.drain_sock(md_sock): |
||||
md_cache[msg.logMonoTime] = msg |
||||
|
||||
for msg in messaging.drain_sock(plan_sock): |
||||
plan_cache[msg.logMonoTime] = msg |
||||
|
||||
controls_state = messaging.recv_sock(controls_state_sock) |
||||
if controls_state is not None: |
||||
plan_time = controls_state.controlsState.planMonoTime |
||||
if plan_time != 0 and plan_time in plan_cache: |
||||
plan = plan_cache[plan_time] |
||||
md_time = plan.plan.mdMonoTime |
||||
if md_time != 0 and md_time in md_cache: |
||||
md = md_cache[md_time] |
||||
frame_id = md.model.frameId |
||||
if frame_id != 0 and frame_id in frame_cache: |
||||
frame = frame_cache[frame_id] |
||||
print("controls lag: %.2fms" % ((controls_state.logMonoTime - frame.frame.timestampEof) / 1e6)) |
||||
|
||||
|
||||
pls = (pls+messaging.drain_sock(proc))[-2:] |
||||
if None not in pls: |
||||
display_cpu(*pls) |
||||
|
||||
rk.keep_time() |
||||
|
||||
if __name__ == "__main__": |
||||
main() |
@ -0,0 +1,83 @@ |
||||
import numpy as np |
||||
import matplotlib.pyplot as plt |
||||
from mpl_toolkits.mplot3d import Axes3D |
||||
from matplotlib import cm |
||||
from matplotlib.ticker import LinearLocator, FormatStrFormatter |
||||
from scipy.optimize import minimize |
||||
|
||||
a = -9.81 |
||||
dt = 0.1 |
||||
|
||||
r = 2.0 |
||||
|
||||
v_ls = [] |
||||
x_ls = [] |
||||
v_egos = [] |
||||
|
||||
for vv_ego in np.arange(35, 40, 1): |
||||
for vv_l in np.arange(35, 40, 1): |
||||
for xx_l in np.arange(0, 100, 1.0): |
||||
x_l = xx_l |
||||
v_l = vv_l |
||||
v_ego = vv_ego |
||||
x_ego = 0.0 |
||||
|
||||
ttc = None |
||||
for t in np.arange(0, 100, dt): |
||||
x_l += v_l * dt |
||||
v_l += a * dt |
||||
v_l = max(v_l, 0.0) |
||||
|
||||
x_ego += v_ego * dt |
||||
if t > r: |
||||
v_ego += a * dt |
||||
v_ego = max(v_ego, 0.0) |
||||
|
||||
if x_ego >= x_l: |
||||
ttc = t |
||||
break |
||||
|
||||
if ttc is None: |
||||
if xx_l < 0.1: |
||||
break |
||||
|
||||
v_ls.append(vv_l) |
||||
x_ls.append(xx_l) |
||||
v_egos.append(vv_ego) |
||||
break |
||||
|
||||
|
||||
def eval_f(x, v_ego, v_l): |
||||
est = x[0] * v_l + x[1] * v_l**2 \ |
||||
+ x[2] * v_ego + x[3] * v_ego**2 |
||||
return est |
||||
|
||||
def f(x): |
||||
r = 0.0 |
||||
for v_ego, v_l, x_l in zip(v_egos, v_ls, x_ls): |
||||
est = eval_f(x, v_ego, v_l) |
||||
r += (x_l - est)**2 |
||||
|
||||
return r |
||||
|
||||
x0 = [0.5, 0.5, 0.5, 0.5] |
||||
res = minimize(f, x0, method='Nelder-Mead') |
||||
print(res) |
||||
print(res.x) |
||||
|
||||
g = 9.81 |
||||
t_r = 1.8 |
||||
|
||||
estimated = [4.0 + eval_f(res.x, v_ego, v_l) for (v_ego, v_l) in zip(v_egos, v_ls)] |
||||
new_formula = [4.0 + v_ego * t_r - (v_l - v_ego) * t_r + v_ego**2/(2*g) - v_l**2 / (2*g) for (v_ego, v_l) in zip(v_egos, v_ls)] |
||||
|
||||
fig = plt.figure() |
||||
ax = fig.add_subplot(111, projection='3d') |
||||
surf = ax.scatter(v_egos, v_ls, x_ls, s=1) |
||||
# surf = ax.scatter(v_egos, v_ls, estimated, s=1) |
||||
surf = ax.scatter(v_egos, v_ls, new_formula, s=1) |
||||
|
||||
ax.set_xlabel('v ego') |
||||
ax.set_ylabel('v lead') |
||||
ax.set_zlabel('min distance') |
||||
plt.show() |
@ -0,0 +1,178 @@ |
||||
#!/usr/bin/env python3 |
||||
import sys |
||||
import math |
||||
import pygame |
||||
import pyproj |
||||
|
||||
import zmq |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
import numpy as np |
||||
|
||||
METER = 25 |
||||
YSCALE = 1 |
||||
|
||||
def to_grid(pt): |
||||
return (int(round(pt[0] * METER + 100)), int(round(pt[1] * METER * YSCALE + 500))) |
||||
|
||||
def gps_latlong_to_meters(gps_values, zero): |
||||
inProj = pyproj.Proj(init='epsg:4326') |
||||
outProj = pyproj.Proj(("+proj=tmerc +lat_0={:f} +lon_0={:f} +units=m" |
||||
" +k=1. +x_0=0 +y_0=0 +ellps=WGS84 +datum=WGS84 +no_defs" |
||||
"+towgs84=-90.7,-106.1,-119.2,4.09,0.218,-1.05,1.37").format(*zero)) |
||||
gps_x, gps_y = pyproj.transform(inProj, outProj, gps_values[1], gps_values[0]) |
||||
return gps_x, gps_y |
||||
|
||||
def rot(hrad): |
||||
return [[math.cos(hrad), -math.sin(hrad)], |
||||
[math.sin(hrad), math.cos(hrad)]] |
||||
|
||||
class Car(): |
||||
CAR_WIDTH = 2.0 |
||||
CAR_LENGTH = 4.5 |
||||
|
||||
def __init__(self, c): |
||||
self.car = pygame.Surface((METER*self.CAR_LENGTH*YSCALE, METER*self.CAR_LENGTH)) |
||||
self.car.set_alpha(64) |
||||
self.car.fill((0,0,0)) |
||||
self.car.set_colorkey((0,0,0)) |
||||
pygame.draw.rect(self.car, c, (METER*1.25*YSCALE, 0, METER*self.CAR_WIDTH*YSCALE, METER*self.CAR_LENGTH), 1) |
||||
|
||||
self.x = 0.0 |
||||
self.y = 0.0 |
||||
self.heading = 0.0 |
||||
|
||||
def from_car_frame(self, pts): |
||||
ret = [] |
||||
for x, y in pts: |
||||
rx, ry = np.dot(rot(math.radians(self.heading)), [x,y]) |
||||
ret.append((self.x + rx, self.y + ry)) |
||||
return ret |
||||
|
||||
def draw(self, screen): |
||||
cars = pygame.transform.rotate(self.car, 90-self.heading) |
||||
pt = (self.x - self.CAR_LENGTH/2, self.y - self.CAR_LENGTH/2) |
||||
screen.blit(cars, to_grid(pt)) |
||||
|
||||
|
||||
def ui_thread(addr="127.0.0.1"): |
||||
#from selfdrive.radar.nidec.interface import RadarInterface |
||||
#RI = RadarInterface() |
||||
|
||||
pygame.display.set_caption("comma top down UI") |
||||
size = (1920,1000) |
||||
screen = pygame.display.set_mode(size, pygame.DOUBLEBUF) |
||||
|
||||
liveLocation = messaging.sub_sock('liveLocation', addr=addr) |
||||
|
||||
#model = messaging.sub_sock('testModel', addr=addr) |
||||
model = messaging.sub_sock('model', addr=addr) |
||||
|
||||
plan = messaging.sub_sock('plan', addr=addr) |
||||
frame = messaging.sub_sock('frame', addr=addr) |
||||
liveTracks = messaging.sub_sock('liveTracks', addr=addr) |
||||
|
||||
car = Car((255,0,255)) |
||||
|
||||
base = None |
||||
|
||||
lb = [] |
||||
|
||||
ts_map = {} |
||||
|
||||
while 1: |
||||
lloc = messaging.recv_sock(liveLocation, wait=True) |
||||
lloc_ts = lloc.logMonoTime |
||||
lloc = lloc.liveLocation |
||||
|
||||
# 50 ms of lag |
||||
lb.append(lloc) |
||||
if len(lb) < 2: |
||||
continue |
||||
lb = lb[-1:] |
||||
|
||||
lloc = lb[0] |
||||
|
||||
# spacebar reset |
||||
for event in pygame.event.get(): |
||||
if event.type == pygame.KEYDOWN and event.key == pygame.K_SPACE: |
||||
base = None |
||||
|
||||
# offscreen reset |
||||
rp = to_grid((car.x, car.y)) |
||||
if rp[0] > (size[0] - 100) or rp[1] > (size[1] - 100) or rp[0] < 0 or rp[1] < 100: |
||||
base = None |
||||
|
||||
|
||||
if base == None: |
||||
screen.fill((10,10,10)) |
||||
base = lloc |
||||
|
||||
# transform pt into local |
||||
pt = gps_latlong_to_meters((lloc.lat, lloc.lon), (base.lat, base.lon)) |
||||
hrad = math.radians(270+base.heading) |
||||
pt = np.dot(rot(hrad), pt) |
||||
|
||||
car.x, car.y = pt[0], -pt[1] |
||||
car.heading = lloc.heading - base.heading |
||||
|
||||
#car.draw(screen) |
||||
pygame.draw.circle(screen, (192,64,192,128), to_grid((car.x, car.y)), 4) |
||||
|
||||
""" |
||||
lt = messaging.recv_sock(liveTracks, wait=False) |
||||
if lt is not None: |
||||
for track in lt.liveTracks: |
||||
pt = car.from_car_frame([[track.dRel, -track.yRel]])[0] |
||||
if track.stationary: |
||||
pygame.draw.circle(screen, (192,128,32,64), to_grid(pt), 1) |
||||
""" |
||||
|
||||
|
||||
""" |
||||
rr = RI.update() |
||||
for pt in rr.points: |
||||
cpt = car.from_car_frame([[pt.dRel + 2.7, -pt.yRel]])[0] |
||||
if (pt.vRel + lloc.speed) < 1.0: |
||||
pygame.draw.circle(screen, (192,128,32,64), to_grid(cpt), 1) |
||||
""" |
||||
|
||||
|
||||
for f in messaging.drain_sock(frame): |
||||
ts_map[f.frame.frameId] = f.frame.timestampEof |
||||
|
||||
def draw_model_data(mm, c): |
||||
pts = car.from_car_frame(zip(np.arange(0.0, 50.0), -np.array(mm))) |
||||
lt = 255 |
||||
for pt in pts: |
||||
screen.set_at(to_grid(pt), (c[0]*lt,c[1]*lt,c[2]*lt,lt)) |
||||
lt -= 2 |
||||
#pygame.draw.lines(screen, (c[0]*lt,c[1]*lt,c[2]*lt,lt), False, map(to_grid, pts), 1) |
||||
|
||||
md = messaging.recv_sock(model, wait=False) |
||||
if md: |
||||
if md.model.frameId in ts_map: |
||||
f_ts = ts_map[md.model.frameId] |
||||
print((lloc_ts - f_ts) * 1e-6,"ms") |
||||
|
||||
#draw_model_data(md.model.path.points, (1,0,0)) |
||||
if md.model.leftLane.prob > 0.3: |
||||
draw_model_data(md.model.leftLane.points, (0,1,0)) |
||||
if md.model.rightLane.prob > 0.3: |
||||
draw_model_data(md.model.rightLane.points, (0,1,0)) |
||||
#if md.model.leftLane.prob > 0.3 and md.model.rightLane.prob > 0.3: |
||||
# draw_model_data([(x+y)/2 for x,y in zip(md.model.leftLane.points, md.model.rightLane.points)], (1,1,0)) |
||||
|
||||
tplan = messaging.recv_sock(plan, wait=False) |
||||
if tplan: |
||||
pts = np.polyval(tplan.plan.dPoly, np.arange(0.0, 50.0)) |
||||
draw_model_data(pts, (1,1,1)) |
||||
|
||||
pygame.display.flip() |
||||
|
||||
if __name__ == "__main__": |
||||
if len(sys.argv) > 1: |
||||
ui_thread(sys.argv[1]) |
||||
else: |
||||
ui_thread() |
||||
|
@ -0,0 +1,152 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
import random |
||||
from collections import defaultdict, OrderedDict |
||||
|
||||
from selfdrive.boardd.boardd import can_list_to_can_capnp |
||||
from selfdrive.car.toyota.toyotacan import make_can_msg |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
|
||||
fields = range(0, 256) |
||||
fields = [105, 225] |
||||
field_results = defaultdict(lambda: "\x00\x00") |
||||
cur_field = 97 |
||||
|
||||
def send(sendcan, addr, m): |
||||
packet = make_can_msg(addr, m, 0, False) |
||||
packets = can_list_to_can_capnp([packet], msgtype='sendcan') |
||||
sendcan.send(packets.to_bytes()) |
||||
|
||||
|
||||
def recv(can, addr): |
||||
received = False |
||||
r = [] |
||||
|
||||
while not received: |
||||
c = messaging.recv_one(can) |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
return r |
||||
|
||||
|
||||
def recv_timeout(can, addr): |
||||
received = False |
||||
r = [] |
||||
t = time.time() |
||||
|
||||
while not received: |
||||
c = messaging.recv_one_or_none(can) |
||||
|
||||
if c is not None: |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
|
||||
if time.time() - t > 0.05: |
||||
received = True |
||||
|
||||
return r |
||||
|
||||
|
||||
def print_hex(d): |
||||
s = map(ord, d) |
||||
s = "".join(["{:02X}".format(b) for b in s]) |
||||
print(s) |
||||
|
||||
|
||||
TYPES = { |
||||
0: 'single', |
||||
1: 'first', |
||||
2: 'consecutive', |
||||
3: 'flow' |
||||
} |
||||
|
||||
FIRST = "\x42\x02\xA8\x01\x00\x00\x00\x00" |
||||
CONTINUE = "\x42\x30\x01\x00\x00\x00\x00\x00" |
||||
|
||||
TEST_ON = "\x42\x02\x10\x60\x00\x00\x00\x00" |
||||
TEST_OFF = "\x42\x02\x10\x5F\x00\x00\x00\x00" |
||||
|
||||
POLL = "\x42\x02\x21\x69\x00\x00\x00\x00" |
||||
|
||||
prev_rcv_t = "" |
||||
recv_data = [] |
||||
l = 0 |
||||
index = 0 |
||||
|
||||
|
||||
can = messaging.sub_sock('can') |
||||
sendcan = messaging.pub_sock('sendcan') |
||||
|
||||
time.sleep(0.5) |
||||
|
||||
send(sendcan, 1872, FIRST) |
||||
results = [] |
||||
|
||||
test_mode = False |
||||
|
||||
while True: |
||||
# Send flow control if necessary |
||||
if prev_rcv_t == 'first' or prev_rcv_t == 'consecutive': |
||||
send(sendcan, 1872, CONTINUE) |
||||
|
||||
received = recv_timeout(can, 1880) |
||||
|
||||
if len(received) == 0: |
||||
print(chr(27) + "[2J") |
||||
print(time.time()) |
||||
if results[0] != "\x7F\x21\x31": |
||||
field_results[cur_field] = results[0] |
||||
else: |
||||
fields.remove(cur_field) |
||||
for k in fields: |
||||
if field_results[k] == "\x00\x00": |
||||
continue |
||||
print(k, end=' ') |
||||
print_hex(field_results[k]) |
||||
results = [] |
||||
|
||||
if not test_mode: |
||||
send(sendcan, 1872, TEST_ON) |
||||
test_mode = True |
||||
else: |
||||
cur_field = random.choice(fields) |
||||
send(sendcan, 1872, POLL.replace('\x69', chr(cur_field))) |
||||
|
||||
for r in received: |
||||
data = r.dat |
||||
|
||||
# Check message type |
||||
t = TYPES[ord(data[1]) >> 4] |
||||
if t == 'single': |
||||
l = ord(data[1]) & 0x0F |
||||
elif t == 'first': |
||||
a = ord(data[1]) & 0x0F |
||||
b = ord(data[2]) |
||||
l = b + (a << 8) |
||||
recv_data = [] |
||||
|
||||
prev_rcv_t = t |
||||
|
||||
if t == 'single': |
||||
recv_data = data[2: 2 + l] |
||||
results.append(recv_data) |
||||
if t == 'first': |
||||
index = 0 |
||||
recv_data += data[3: min(8, 3 + l)] |
||||
if t == 'consecutive': |
||||
index += 1 |
||||
assert index == ord(data[1]) & 0x0F |
||||
|
||||
pending_l = l - len(recv_data) |
||||
recv_data += data[2: min(8, 2 + pending_l)] |
||||
|
||||
if len(recv_data) == l: |
||||
prev_rcv_t = "" |
||||
results.append(recv_data) |
@ -0,0 +1,151 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
import random |
||||
from collections import defaultdict, OrderedDict |
||||
|
||||
from selfdrive.boardd.boardd import can_list_to_can_capnp |
||||
from selfdrive.car.toyota.toyotacan import make_can_msg |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
|
||||
fields = range(0, 256) |
||||
# fields = [105, 225] |
||||
fields = [105] |
||||
field_results = defaultdict(lambda: "\x00\x00") |
||||
cur_field = 97 |
||||
|
||||
def send(sendcan, addr, m): |
||||
packet = make_can_msg(addr, m, 0, False) |
||||
packets = can_list_to_can_capnp([packet], msgtype='sendcan') |
||||
sendcan.send(packets.to_bytes()) |
||||
|
||||
|
||||
def recv(can, addr): |
||||
received = False |
||||
r = [] |
||||
|
||||
while not received: |
||||
c = messaging.recv_one(can) |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
return r |
||||
|
||||
|
||||
def recv_timeout(can, addr): |
||||
received = False |
||||
r = [] |
||||
t = time.time() |
||||
|
||||
while not received: |
||||
c = messaging.recv_one_or_none(can) |
||||
|
||||
if c is not None: |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
|
||||
if time.time() - t > 0.05: |
||||
received = True |
||||
|
||||
return r |
||||
|
||||
|
||||
def print_hex(d): |
||||
s = map(ord, d) |
||||
s = "".join(["{:02X}".format(b) for b in s]) |
||||
print(s) |
||||
|
||||
|
||||
TYPES = { |
||||
0: 'single', |
||||
1: 'first', |
||||
2: 'consecutive', |
||||
3: 'flow' |
||||
} |
||||
|
||||
FIRST = "\xFF\x02\xA8\x01\x00\x00\x00\x00" |
||||
CONTINUE = "\xFF\x30\x01\x00\x00\x00\x00\x00" |
||||
TEST_ON = "\xFF\x02\x10\x01\x00\x00\x00\x00" |
||||
POLL = "\xFF\x02\x21\x69\x00\x00\x00\x00" |
||||
|
||||
prev_rcv_t = "" |
||||
recv_data = [] |
||||
l = 0 |
||||
index = 0 |
||||
|
||||
|
||||
can = messaging.sub_sock('can') |
||||
sendcan = messaging.pub_sock('sendcan') |
||||
|
||||
time.sleep(0.5) |
||||
|
||||
send(sendcan, 1872, FIRST) |
||||
results = [] |
||||
|
||||
test_mode = False |
||||
|
||||
while True: |
||||
# Send flow control if necessary |
||||
if prev_rcv_t == 'first' or prev_rcv_t == 'consecutive': |
||||
send(sendcan, 1872, CONTINUE) |
||||
|
||||
received = recv_timeout(can, 1880) |
||||
|
||||
if len(received) == 0: |
||||
print_hex(results[0]) |
||||
# print chr(27) + "[2J" |
||||
# print time.time() |
||||
# if results[0] != "\x7F\x21\x31": |
||||
# field_results[cur_field] = results[0] |
||||
# else: |
||||
# fields.remove(cur_field) |
||||
# for k in fields: |
||||
# if field_results[k] == "\x00\x00": |
||||
# continue |
||||
# print k, |
||||
# print_hex(field_results[k]) |
||||
results = [] |
||||
|
||||
if not test_mode: |
||||
send(sendcan, 1872, TEST_ON) |
||||
test_mode = True |
||||
else: |
||||
cur_field = random.choice(fields) |
||||
send(sendcan, 1872, POLL.replace('\x69', chr(cur_field))) |
||||
|
||||
for r in received: |
||||
data = r.dat |
||||
|
||||
# Check message type |
||||
t = TYPES[ord(data[1]) >> 4] |
||||
if t == 'single': |
||||
l = ord(data[1]) & 0x0F |
||||
elif t == 'first': |
||||
a = ord(data[1]) & 0x0F |
||||
b = ord(data[2]) |
||||
l = b + (a << 8) |
||||
recv_data = [] |
||||
|
||||
prev_rcv_t = t |
||||
|
||||
if t == 'single': |
||||
recv_data = data[2: 2 + l] |
||||
results.append(recv_data) |
||||
if t == 'first': |
||||
index = 0 |
||||
recv_data += data[3: min(8, 3 + l)] |
||||
if t == 'consecutive': |
||||
index += 1 |
||||
assert index == ord(data[1]) & 0x0F |
||||
|
||||
pending_l = l - len(recv_data) |
||||
recv_data += data[2: min(8, 2 + pending_l)] |
||||
|
||||
if len(recv_data) == l: |
||||
prev_rcv_t = "" |
||||
results.append(recv_data) |
@ -0,0 +1,41 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
from collections import defaultdict, OrderedDict |
||||
|
||||
from selfdrive.boardd.boardd import can_list_to_can_capnp |
||||
from selfdrive.car.toyota.toyotacan import make_can_msg |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
can = messaging.sub_sock('can') |
||||
sendcan = messaging.pub_sock('sendcan') |
||||
|
||||
|
||||
BEFORE = [ |
||||
"\x10\x15\x30\x0B\x00\x00\x00\x00", |
||||
"\x21\x00\x00\x00\x00\x00\x00\x00", |
||||
] |
||||
|
||||
LEFT = "\x22\x00\x00\x08\x00\x00\x00\x00" |
||||
RIGHT = "\x22\x00\x00\x04\x00\x00\x00\x00" |
||||
OFF = "\x22\x00\x00\x00\x00\x00\x00\x00" |
||||
|
||||
AFTER = "\x23\x00\x00\x00\x00\x00\x00\x00" |
||||
|
||||
i = 0 |
||||
j = 0 |
||||
while True: |
||||
i += 1 |
||||
|
||||
if i % 10 == 0: |
||||
j += 1 |
||||
|
||||
cur = RIGHT if j % 2 == 0 else OFF |
||||
can_list = [make_can_msg(1984, d, 0, False) for d in BEFORE] |
||||
can_list.append(make_can_msg(1984, cur, 0, False)) |
||||
can_list.append(make_can_msg(1984, AFTER, 0, False)) |
||||
|
||||
for m in can_list: |
||||
sendcan.send(can_list_to_can_capnp([m], msgtype='sendcan').to_bytes()) |
||||
time.sleep(0.01) |
@ -0,0 +1,61 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
import time |
||||
from collections import defaultdict, OrderedDict |
||||
|
||||
from selfdrive.boardd.boardd import can_list_to_can_capnp |
||||
from selfdrive.car.toyota.toyotacan import make_can_msg |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
|
||||
def send(sendcan, addr, m): |
||||
packet = make_can_msg(addr, m, 0, False) |
||||
packets = can_list_to_can_capnp([packet], msgtype='sendcan') |
||||
sendcan.send(packets.to_bytes()) |
||||
|
||||
|
||||
def recv_timeout(can, addr): |
||||
received = False |
||||
r = [] |
||||
t = time.time() |
||||
|
||||
while not received: |
||||
c = messaging.recv_one_or_none(can) |
||||
|
||||
if c is not None: |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
|
||||
if time.time() - t > 0.1: |
||||
received = True |
||||
|
||||
return r |
||||
|
||||
|
||||
can = messaging.sub_sock('can') |
||||
sendcan = messaging.pub_sock('sendcan') |
||||
|
||||
PUBLIC = 0 |
||||
PRIVATE = 1 |
||||
|
||||
time.sleep(0.5) |
||||
|
||||
# 1, 112 |
||||
|
||||
TEST_ON = "\xFF\x02\x10\x70\x00\x00\x00\x00" |
||||
POLL = "\xFF\x02\x21\x69\x00\x00\x00\x00" |
||||
send(sendcan, 1872, TEST_ON) |
||||
r = recv_timeout(can, 1880) |
||||
print(r) |
||||
|
||||
|
||||
for i in range(0, 256): |
||||
send(sendcan, 1872, POLL.replace('\x69', chr(i))) |
||||
r = recv_timeout(can, 1880) |
||||
if len(r): |
||||
print(i, end=' ') |
||||
for m in r: |
||||
print(m.dat.encode('hex')) |
@ -0,0 +1,26 @@ |
||||
#!/usr/bin/env python3 |
||||
import zmq |
||||
from collections import OrderedDict |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
can = messaging.sub_sock('can') |
||||
|
||||
addr = OrderedDict() |
||||
|
||||
while True: |
||||
c = messaging.recv_one(can) |
||||
for msg in c.can: |
||||
s = map(ord, msg.dat) |
||||
s = "".join(["\\x{:02X}".format(b) for b in s]) |
||||
s = "\"" + s + "\"," |
||||
|
||||
if msg.address == 1872: |
||||
print("s:", s) |
||||
if msg.address == 1880: |
||||
print("r:", s) |
||||
|
||||
if msg.address not in addr: |
||||
addr[msg.address] = list() |
||||
if msg.dat not in addr[msg.address]: |
||||
addr[msg.address].append(s) |
@ -0,0 +1,159 @@ |
||||
#!/usr/bin/env python3 |
||||
import sys |
||||
import zmq |
||||
import os |
||||
import time |
||||
import random |
||||
from collections import defaultdict, OrderedDict |
||||
|
||||
from selfdrive.boardd.boardd import can_list_to_can_capnp |
||||
from selfdrive.car.toyota.toyotacan import make_can_msg |
||||
import cereal.messaging as messaging |
||||
from cereal.services import service_list |
||||
|
||||
changing = [] |
||||
fields = range(0, 256) |
||||
# fields = [225, 50, 39, 40] |
||||
fields = [50] |
||||
field_results = defaultdict(lambda: "\x00\x00") |
||||
cur_field = 97 |
||||
|
||||
def send(sendcan, addr, m): |
||||
packet = make_can_msg(addr, m, 0, False) |
||||
packets = can_list_to_can_capnp([packet], msgtype='sendcan') |
||||
sendcan.send(packets.to_bytes()) |
||||
|
||||
|
||||
def recv(can, addr): |
||||
received = False |
||||
r = [] |
||||
|
||||
while not received: |
||||
c = messaging.recv_one(can) |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
return r |
||||
|
||||
|
||||
def recv_timeout(can, addr): |
||||
received = False |
||||
r = [] |
||||
t = time.time() |
||||
|
||||
while not received: |
||||
c = messaging.recv_one_or_none(can) |
||||
|
||||
if c is not None: |
||||
for msg in c.can: |
||||
if msg.address == addr: |
||||
r.append(msg) |
||||
received = True |
||||
|
||||
if time.time() - t > 0.05: |
||||
received = True |
||||
|
||||
return r |
||||
|
||||
|
||||
def print_hex(d): |
||||
s = map(ord, d) |
||||
s = "".join(["{:02X}".format(b) for b in s]) |
||||
print(s) |
||||
|
||||
|
||||
TYPES = { |
||||
0: 'single', |
||||
1: 'first', |
||||
2: 'consecutive', |
||||
3: 'flow' |
||||
} |
||||
|
||||
CONTINUE = "\x67\x30\x01\x00\x00\x00\x00\x00" |
||||
TEST_ON = "\x67\x02\x10\x74\x00\x00\x00\x00" |
||||
POLL = "\x67\x02\x21\x69\x00\x00\x00\x00" |
||||
# POLL = "\x67\x02\x10\x69\x00\x00\x00\x00" |
||||
|
||||
prev_rcv_t = "" |
||||
recv_data = [] |
||||
l = 0 |
||||
index = 0 |
||||
|
||||
|
||||
can = messaging.sub_sock('can') |
||||
sendcan = messaging.pub_sock('sendcan') |
||||
|
||||
time.sleep(0.5) |
||||
|
||||
results = [] |
||||
|
||||
test_mode = False |
||||
|
||||
while True: |
||||
# Send flow control if necessary |
||||
if prev_rcv_t == 'first' or prev_rcv_t == 'consecutive': |
||||
send(sendcan, 1872, CONTINUE) |
||||
|
||||
received = recv_timeout(can, 1880) |
||||
|
||||
if len(received) == 0: |
||||
sys.stdout.flush() |
||||
print(chr(27) + "[2J") |
||||
print(time.time()) |
||||
print(changing) |
||||
|
||||
if len(results): |
||||
if results[0] != "\x7F\x21\x31": |
||||
old = field_results[cur_field] |
||||
if old != '\x00\x00' and old != results[0] and cur_field not in changing: |
||||
changing.append(cur_field) |
||||
field_results[cur_field] = results[0] |
||||
else: |
||||
fields.remove(cur_field) |
||||
|
||||
for k in fields: |
||||
# if field_results[k] == "\x00\x00": |
||||
# continue |
||||
print(k, end=' ') |
||||
print_hex(field_results[k]) |
||||
results = [] |
||||
|
||||
if not test_mode: |
||||
send(sendcan, 1872, TEST_ON) |
||||
test_mode = True |
||||
else: |
||||
cur_field = random.choice(fields) |
||||
send(sendcan, 1872, POLL.replace('\x69', chr(cur_field))) |
||||
|
||||
for r in received: |
||||
data = r.dat |
||||
|
||||
# Check message type |
||||
t = TYPES[ord(data[1]) >> 4] |
||||
if t == 'single': |
||||
l = ord(data[1]) & 0x0F |
||||
elif t == 'first': |
||||
a = ord(data[1]) & 0x0F |
||||
b = ord(data[2]) |
||||
l = b + (a << 8) |
||||
recv_data = [] |
||||
|
||||
prev_rcv_t = t |
||||
|
||||
if t == 'single': |
||||
recv_data = data[2: 2 + l] |
||||
results.append(recv_data) |
||||
if t == 'first': |
||||
index = 0 |
||||
recv_data += data[3: min(8, 3 + l)] |
||||
if t == 'consecutive': |
||||
index += 1 |
||||
assert index == ord(data[1]) & 0x0F |
||||
|
||||
pending_l = l - len(recv_data) |
||||
recv_data += data[2: min(8, 2 + pending_l)] |
||||
|
||||
if len(recv_data) == l: |
||||
prev_rcv_t = "" |
||||
results.append(recv_data) |
@ -0,0 +1,54 @@ |
||||
#!/usr/bin/env python3 |
||||
import numpy as np |
||||
from selfdrive.controls.lib.vehicle_model import VehicleModel, calc_slip_factor |
||||
from selfdrive.car.honda.interface import CarInterface |
||||
|
||||
def mpc_path_prediction(sa, u, psi_0, dt, VM): |
||||
# sa and u needs to be numpy arrays |
||||
sa_w = sa * np.pi / 180. / VM.CP.steerRatio |
||||
x = np.zeros(len(sa)) |
||||
y = np.zeros(len(sa)) |
||||
psi = np.ones(len(sa)) * psi_0 |
||||
|
||||
for i in range(0, len(sa)-1): |
||||
x[i+1] = x[i] + np.cos(psi[i]) * u[i] * dt |
||||
y[i+1] = y[i] + np.sin(psi[i]) * u[i] * dt |
||||
psi[i+1] = psi[i] + sa_w[i] * u[i] * dt * VM.curvature_factor(u[i]) |
||||
|
||||
return x, y, psi |
||||
|
||||
|
||||
def model_path_prediction(sa, u, psi_0, dt, VM): |
||||
# steady state solution |
||||
sa_r = sa * np.pi / 180. |
||||
x = np.zeros(len(sa)) |
||||
y = np.zeros(len(sa)) |
||||
psi = np.ones(len(sa)) * psi_0 |
||||
for i in range(0, len(sa)-1): |
||||
|
||||
out = VM.steady_state_sol(sa_r[i], u[i]) |
||||
|
||||
x[i+1] = x[i] + np.cos(psi[i]) * u[i] * dt - np.sin(psi[i]) * out[0] * dt |
||||
y[i+1] = y[i] + np.sin(psi[i]) * u[i] * dt + np.cos(psi[i]) * out[0] * dt |
||||
psi[i+1] = psi[i] + out[1] * dt |
||||
|
||||
return x, y, psi |
||||
|
||||
if __name__ == "__main__": |
||||
CP = CarInterface.get_params("HONDA CIVIC 2016 TOURING") |
||||
print(CP) |
||||
VM = VehicleModel(CP) |
||||
print(VM.steady_state_sol(.1, 0.15)) |
||||
print(calc_slip_factor(VM)) |
||||
print("Curv", VM.curvature_factor(30.)) |
||||
|
||||
dt = 0.05 |
||||
st = 20 |
||||
u = np.ones(st) * 1. |
||||
sa = np.ones(st) * 1. |
||||
|
||||
out = mpc_path_prediction(sa, u, dt, VM) |
||||
out_model = model_path_prediction(sa, u, dt, VM) |
||||
|
||||
print("mpc", out) |
||||
print("model", out_model) |
@ -0,0 +1,49 @@ |
||||
#!/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 |
||||
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. |
||||
|
||||
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%% - Temp %.2f" % (100. * np.mean(cores), last_temp )) |
@ -0,0 +1,110 @@ |
||||
#!/usr/bin/env python3 |
||||
import matplotlib |
||||
matplotlib.use('TkAgg') |
||||
|
||||
import sys |
||||
import cereal.messaging as messaging |
||||
import numpy as np |
||||
import matplotlib.pyplot as plt |
||||
|
||||
# debug liateral MPC by plotting its trajectory. To receive liveLongitudinalMpc packets, |
||||
# set on LOG_MPC env variable and run plannerd on a replay |
||||
|
||||
|
||||
def mpc_vwr_thread(addr="127.0.0.1"): |
||||
|
||||
plt.ion() |
||||
fig = plt.figure(figsize=(15, 20)) |
||||
ax = fig.add_subplot(131) |
||||
aa = fig.add_subplot(132, sharey=ax) |
||||
ap = fig.add_subplot(133, sharey=ax) |
||||
|
||||
ax.set_xlim([-10, 10]) |
||||
ax.set_ylim([0., 100.]) |
||||
aa.set_xlim([-20., 20]) |
||||
ap.set_xlim([-5, 5]) |
||||
|
||||
ax.set_xlabel('x [m]') |
||||
ax.set_ylabel('y [m]') |
||||
aa.set_xlabel('steer_angle [deg]') |
||||
ap.set_xlabel('asset angle [deg]') |
||||
ax.grid(True) |
||||
aa.grid(True) |
||||
ap.grid(True) |
||||
|
||||
path_x = np.arange(0, 100) |
||||
mpc_path_x = np.arange(0, 49) |
||||
|
||||
p_path_y = np.zeros(100) |
||||
|
||||
l_path_y = np.zeros(100) |
||||
r_path_y = np.zeros(100) |
||||
mpc_path_y = np.zeros(49) |
||||
mpc_steer_angle = np.zeros(49) |
||||
mpc_psi = np.zeros(49) |
||||
|
||||
line1, = ax.plot(mpc_path_y, mpc_path_x) |
||||
# line1b, = ax.plot(mpc_path_y, mpc_path_x, 'o') |
||||
|
||||
lineP, = ax.plot(p_path_y, path_x) |
||||
lineL, = ax.plot(l_path_y, path_x) |
||||
lineR, = ax.plot(r_path_y, path_x) |
||||
line3, = aa.plot(mpc_steer_angle, mpc_path_x) |
||||
line4, = ap.plot(mpc_psi, mpc_path_x) |
||||
ax.invert_xaxis() |
||||
aa.invert_xaxis() |
||||
plt.show() |
||||
|
||||
|
||||
# *** log *** |
||||
livempc = messaging.sub_sock('liveMpc', addr=addr) |
||||
model = messaging.sub_sock('model', addr=addr) |
||||
path_plan_sock = messaging.sub_sock('pathPlan', addr=addr) |
||||
|
||||
while 1: |
||||
lMpc = messaging.recv_sock(livempc, wait=True) |
||||
md = messaging.recv_sock(model) |
||||
pp = messaging.recv_sock(path_plan_sock) |
||||
|
||||
if md is not None: |
||||
p_poly = np.array(md.model.path.poly) |
||||
l_poly = np.array(md.model.leftLane.poly) |
||||
r_poly = np.array(md.model.rightLane.poly) |
||||
|
||||
p_path_y = np.polyval(p_poly, path_x) |
||||
l_path_y = np.polyval(r_poly, path_x) |
||||
r_path_y = np.polyval(l_poly, path_x) |
||||
|
||||
if pp is not None: |
||||
p_path_y = np.polyval(pp.pathPlan.dPoly, path_x) |
||||
lineP.set_xdata(p_path_y) |
||||
lineP.set_ydata(path_x) |
||||
|
||||
if lMpc is not None: |
||||
mpc_path_x = list(lMpc.liveMpc.x)[1:] |
||||
mpc_path_y = list(lMpc.liveMpc.y)[1:] |
||||
mpc_steer_angle = list(lMpc.liveMpc.delta)[1:] |
||||
mpc_psi = list(lMpc.liveMpc.psi)[1:] |
||||
|
||||
line1.set_xdata(mpc_path_y) |
||||
line1.set_ydata(mpc_path_x) |
||||
lineL.set_xdata(l_path_y) |
||||
lineL.set_ydata(path_x) |
||||
lineR.set_xdata(r_path_y) |
||||
lineR.set_ydata(path_x) |
||||
line3.set_xdata(np.asarray(mpc_steer_angle)*180./np.pi * 14) |
||||
line3.set_ydata(mpc_path_x) |
||||
line4.set_xdata(np.asarray(mpc_psi)*180./np.pi) |
||||
line4.set_ydata(mpc_path_x) |
||||
|
||||
aa.relim() |
||||
aa.autoscale_view(True, scaley=True, scalex=True) |
||||
|
||||
fig.canvas.draw() |
||||
fig.canvas.flush_events() |
||||
|
||||
if __name__ == "__main__": |
||||
if len(sys.argv) > 1: |
||||
mpc_vwr_thread(sys.argv[1]) |
||||
else: |
||||
mpc_vwr_thread() |
@ -0,0 +1,106 @@ |
||||
#!/usr/bin/env python3 |
||||
|
||||
import sys |
||||
import cereal.messaging as messaging |
||||
import numpy as np |
||||
import matplotlib.pyplot as plt |
||||
|
||||
N = 21 |
||||
|
||||
# debug longitudinal MPC by plotting its trajectory. To receive liveLongitudinalMpc packets, |
||||
# set on LOG_MPC env variable and run plannerd on a replay |
||||
|
||||
def plot_longitudinal_mpc(addr="127.0.0.1"): |
||||
# *** log *** |
||||
livempc = messaging.sub_sock('liveLongitudinalMpc', addr=addr, conflate=True) |
||||
radarstate = messaging.sub_sock('radarState', addr=addr, conflate=True) |
||||
|
||||
plt.ion() |
||||
fig = plt.figure() |
||||
|
||||
t = np.hstack([np.arange(0.0, 0.8, 0.2), np.arange(0.8, 10.6, 0.6)]) |
||||
|
||||
p_x_ego = fig.add_subplot(3, 2, 1) |
||||
p_v_ego = fig.add_subplot(3, 2, 3) |
||||
p_a_ego = fig.add_subplot(3, 2, 5) |
||||
# p_x_l = fig.add_subplot(3, 2, 2) |
||||
# p_a_l = fig.add_subplot(3, 2, 6) |
||||
p_d_l = fig.add_subplot(3, 2, 2) |
||||
p_d_l_v = fig.add_subplot(3, 2, 4) |
||||
p_d_l_vv = fig.add_subplot(3, 2, 6) |
||||
|
||||
p_v_ego.set_ylim([0, 30]) |
||||
p_a_ego.set_ylim([-4, 4]) |
||||
p_d_l.set_ylim([-1, 10]) |
||||
|
||||
p_x_ego.set_title('x') |
||||
p_v_ego.set_title('v') |
||||
p_a_ego.set_title('a') |
||||
p_d_l.set_title('rel dist') |
||||
|
||||
l_x_ego, = p_x_ego.plot(t, np.zeros(N)) |
||||
l_v_ego, = p_v_ego.plot(t, np.zeros(N)) |
||||
l_a_ego, = p_a_ego.plot(t, np.zeros(N)) |
||||
l_x_l, = p_x_ego.plot(t, np.zeros(N)) |
||||
l_v_l, = p_v_ego.plot(t, np.zeros(N)) |
||||
l_a_l, = p_a_ego.plot(t, np.zeros(N)) |
||||
l_d_l, = p_d_l.plot(t, np.zeros(N)) |
||||
l_d_l_v, = p_d_l_v.plot(np.zeros(N)) |
||||
l_d_l_vv, = p_d_l_vv.plot(np.zeros(N)) |
||||
p_x_ego.legend(['ego', 'l']) |
||||
p_v_ego.legend(['ego', 'l']) |
||||
p_a_ego.legend(['ego', 'l']) |
||||
p_d_l_v.set_xlabel('d_rel') |
||||
p_d_l_v.set_ylabel('v_rel') |
||||
p_d_l_v.set_ylim([-20, 20]) |
||||
p_d_l_v.set_xlim([0, 100]) |
||||
p_d_l_vv.set_xlabel('d_rel') |
||||
p_d_l_vv.set_ylabel('v_rel') |
||||
p_d_l_vv.set_ylim([-5, 5]) |
||||
p_d_l_vv.set_xlim([10, 40]) |
||||
|
||||
while True: |
||||
lMpc = messaging.recv_sock(livempc, wait=True) |
||||
rs = messaging.recv_sock(radarstate, wait=True) |
||||
|
||||
if lMpc is not None: |
||||
|
||||
if lMpc.liveLongitudinalMpc.mpcId != 1: |
||||
continue |
||||
|
||||
x_ego = list(lMpc.liveLongitudinalMpc.xEgo) |
||||
v_ego = list(lMpc.liveLongitudinalMpc.vEgo) |
||||
a_ego = list(lMpc.liveLongitudinalMpc.aEgo) |
||||
x_l = list(lMpc.liveLongitudinalMpc.xLead) |
||||
v_l = list(lMpc.liveLongitudinalMpc.vLead) |
||||
# a_l = list(lMpc.liveLongitudinalMpc.aLead) |
||||
a_l = rs.radarState.leadOne.aLeadK * np.exp(-lMpc.liveLongitudinalMpc.aLeadTau * t**2 / 2) |
||||
#print(min(a_ego), lMpc.liveLongitudinalMpc.qpIterations) |
||||
|
||||
l_x_ego.set_ydata(x_ego) |
||||
l_v_ego.set_ydata(v_ego) |
||||
l_a_ego.set_ydata(a_ego) |
||||
|
||||
l_x_l.set_ydata(x_l) |
||||
l_v_l.set_ydata(v_l) |
||||
l_a_l.set_ydata(a_l) |
||||
|
||||
l_d_l.set_ydata(np.array(x_l) - np.array(x_ego)) |
||||
l_d_l_v.set_ydata(np.array(v_l) - np.array(v_ego)) |
||||
l_d_l_v.set_xdata(np.array(x_l) - np.array(x_ego)) |
||||
l_d_l_vv.set_ydata(np.array(v_l) - np.array(v_ego)) |
||||
l_d_l_vv.set_xdata(np.array(x_l) - np.array(x_ego)) |
||||
|
||||
p_x_ego.relim() |
||||
p_x_ego.autoscale_view(True, scaley=True, scalex=True) |
||||
fig.canvas.draw() |
||||
fig.canvas.flush_events() |
||||
|
||||
|
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
if len(sys.argv) > 1: |
||||
plot_longitudinal_mpc(sys.argv[1]) |
||||
else: |
||||
plot_longitudinal_mpc() |
@ -0,0 +1,129 @@ |
||||
#! /usr/bin/env python |
||||
import matplotlib.pyplot as plt |
||||
from selfdrive.controls.lib.lateral_mpc import libmpc_py |
||||
from selfdrive.controls.lib.drive_helpers import MPC_COST_LAT |
||||
import math |
||||
|
||||
libmpc = libmpc_py.libmpc |
||||
libmpc.init(MPC_COST_LAT.PATH, MPC_COST_LAT.LANE, MPC_COST_LAT.HEADING, 1.) |
||||
|
||||
cur_state = libmpc_py.ffi.new("state_t *") |
||||
cur_state[0].x = 0.0 |
||||
cur_state[0].y = 0.0 |
||||
cur_state[0].psi = 0.0 |
||||
cur_state[0].delta = 0.0 |
||||
|
||||
mpc_solution = libmpc_py.ffi.new("log_t *") |
||||
xx = [] |
||||
yy = [] |
||||
deltas = [] |
||||
psis = [] |
||||
times = [] |
||||
|
||||
curvature_factor = 0.3 |
||||
v_ref = 1.0 * 20.12 # 45 mph |
||||
|
||||
LANE_WIDTH = 3.7 |
||||
p = [0.0, 0.0, 0.0, 0.0] |
||||
p_l = p[:] |
||||
p_l[3] += LANE_WIDTH / 2.0 |
||||
|
||||
p_r = p[:] |
||||
p_r[3] -= LANE_WIDTH / 2.0 |
||||
|
||||
|
||||
l_poly = libmpc_py.ffi.new("double[4]", p_l) |
||||
r_poly = libmpc_py.ffi.new("double[4]", p_r) |
||||
p_poly = libmpc_py.ffi.new("double[4]", p) |
||||
|
||||
l_prob = 1.0 |
||||
r_prob = 1.0 |
||||
p_prob = 1.0 |
||||
|
||||
for i in range(1): |
||||
cur_state[0].delta = math.radians(510. / 13.) |
||||
libmpc.run_mpc(cur_state, mpc_solution, l_poly, r_poly, p_poly, l_prob, r_prob, |
||||
curvature_factor, v_ref, LANE_WIDTH) |
||||
|
||||
timesi = [] |
||||
ct = 0 |
||||
for i in range(21): |
||||
timesi.append(ct) |
||||
if i <= 4: |
||||
ct += 0.05 |
||||
else: |
||||
ct += 0.15 |
||||
|
||||
|
||||
xi = list(mpc_solution[0].x) |
||||
yi = list(mpc_solution[0].y) |
||||
psii = list(mpc_solution[0].psi) |
||||
deltai = list(mpc_solution[0].delta) |
||||
print("COST: ", mpc_solution[0].cost) |
||||
|
||||
|
||||
plt.figure(0) |
||||
plt.subplot(3, 1, 1) |
||||
plt.plot(timesi, psii) |
||||
plt.ylabel('psi') |
||||
plt.grid(True) |
||||
plt.subplot(3, 1, 2) |
||||
plt.plot(timesi, deltai) |
||||
plt.ylabel('delta') |
||||
plt.grid(True) |
||||
plt.subplot(3, 1, 3) |
||||
plt.plot(timesi, yi) |
||||
plt.ylabel('y') |
||||
plt.grid(True) |
||||
plt.show() |
||||
|
||||
|
||||
#### UNCOMMENT TO CHECK ITERATIVE SOLUTION |
||||
#### |
||||
####for i in range(100): |
||||
#### libmpc.run_mpc(cur_state, mpc_solution, l_poly, r_poly, p_poly, l_prob, r_prob, |
||||
#### curvature_factor, v_ref, LANE_WIDTH) |
||||
#### print "x", list(mpc_solution[0].x) |
||||
#### print "y", list(mpc_solution[0].y) |
||||
#### print "delta", list(mpc_solution[0].delta) |
||||
#### print "psi", list(mpc_solution[0].psi) |
||||
#### # cur_state[0].x = mpc_solution[0].x[1] |
||||
#### # cur_state[0].y = mpc_solution[0].y[1] |
||||
#### # cur_state[0].psi = mpc_solution[0].psi[1] |
||||
#### cur_state[0].delta = radians(200 / 13.)#mpc_solution[0].delta[1] |
||||
#### |
||||
#### xx.append(cur_state[0].x) |
||||
#### yy.append(cur_state[0].y) |
||||
#### psis.append(cur_state[0].psi) |
||||
#### deltas.append(cur_state[0].delta) |
||||
#### times.append(i * 0.05) |
||||
#### |
||||
#### |
||||
####def f(x): |
||||
#### return p_poly[0] * x**3 + p_poly[1] * x**2 + p_poly[2] * x + p_poly[3] |
||||
#### |
||||
#### |
||||
##### planned = map(f, xx) |
||||
##### plt.figure(1) |
||||
##### plt.plot(yy, xx, 'r-') |
||||
##### plt.plot(planned, xx, 'b--', linewidth=0.5) |
||||
##### plt.axes().set_aspect('equal', 'datalim') |
||||
##### plt.gca().invert_xaxis() |
||||
#### |
||||
##### planned = map(f, map(float, list(mpc_solution[0].x)[1:])) |
||||
##### plt.figure(1) |
||||
##### plt.plot(map(float, list(mpc_solution[0].y)[1:]), map(float, list(mpc_solution[0].x)[1:]), 'r-') |
||||
##### plt.plot(planned, map(float, list(mpc_solution[0].x)[1:]), 'b--', linewidth=0.5) |
||||
##### plt.axes().set_aspect('equal', 'datalim') |
||||
##### plt.gca().invert_xaxis() |
||||
#### |
||||
####plt.figure(2) |
||||
####plt.subplot(2, 1, 1) |
||||
####plt.plot(times, psis) |
||||
####plt.ylabel('psi') |
||||
####plt.subplot(2, 1, 2) |
||||
####plt.plot(times, deltas) |
||||
####plt.ylabel('delta') |
||||
#### |
||||
#### |
||||
####plt.show() |
@ -0,0 +1,186 @@ |
||||
#! /usr/bin/env python |
||||
import numpy as np |
||||
from collections import OrderedDict |
||||
import matplotlib.pyplot as plt |
||||
from selfdrive.car.honda.interface import CarInterface |
||||
from selfdrive.controls.lib.lateral_mpc import libmpc_py |
||||
from selfdrive.controls.lib.vehicle_model import VehicleModel |
||||
|
||||
# plot lateral MPC trajectory by defining boundary conditions: |
||||
# lane lines, p_poly and vehicle states. Use this script to tune MPC costs |
||||
|
||||
libmpc = libmpc_py.libmpc |
||||
|
||||
mpc_solution = libmpc_py.ffi.new("log_t *") |
||||
|
||||
points_l = np.array([1.1049711, 1.1053879, 1.1073375, 1.1096942, 1.1124474, 1.1154714, 1.1192677, 1.1245866, 1.1321017, 1.1396152, 1.146443, 1.1555313, 1.1662073, 1.1774249, 1.1888939, 1.2009926, 1.2149779, 1.2300836, 1.2450289, 1.2617753, 1.2785473, 1.2974714, 1.3151019, 1.3331807, 1.3545501, 1.3763691, 1.3983455, 1.4215056, 1.4446729, 1.4691089, 1.4927692, 1.5175346, 1.5429921, 1.568854, 1.5968665, 1.6268958, 1.657122, 1.6853137, 1.7152609, 1.7477539, 1.7793678, 1.8098511, 1.8428392, 1.8746407, 1.9089606, 1.9426043, 1.9775689, 2.0136933, 2.0520134, 2.0891454]) |
||||
|
||||
points_r = np.array([-2.4442139, -2.4449506, -2.4448867, -2.44377, -2.4422617, -2.4393811, -2.4374201, -2.4334245, -2.4286852, -2.4238286, -2.4177458, -2.4094386, -2.3994849, -2.3904033, -2.380136, -2.3699453, -2.3594661, -2.3474073, -2.3342307, -2.3194637, -2.3046403, -2.2881098, -2.2706163, -2.2530098, -2.235604, -2.2160542, -2.1967411, -2.1758952, -2.1544619, -2.1325269, -2.1091819, -2.0850561, -2.0621953, -2.0364127, -2.0119917, -1.9851667, -1.9590458, -1.9306552, -1.9024918, -1.8745357, -1.8432863, -1.8131843, -1.7822732, -1.7507075, -1.7180918, -1.6845931, -1.650871, -1.6157099, -1.5787286, -1.5418037]) |
||||
|
||||
|
||||
points_c = (points_l + points_r) / 2.0 |
||||
|
||||
def compute_path_pinv(): |
||||
deg = 3 |
||||
x = np.arange(50.0) |
||||
X = np.vstack(tuple(x**n for n in range(deg, -1, -1))).T |
||||
pinv = np.linalg.pinv(X) |
||||
return pinv |
||||
|
||||
|
||||
def model_polyfit(points): |
||||
path_pinv = compute_path_pinv() |
||||
return np.dot(path_pinv, map(float, points)) |
||||
|
||||
|
||||
xx = [] |
||||
yy = [] |
||||
deltas = [] |
||||
psis = [] |
||||
times = [] |
||||
|
||||
CP = CarInterface.get_params("HONDA CIVIC 2016 TOURING") |
||||
VM = VehicleModel(CP) |
||||
|
||||
v_ref = 32.00 # 45 mph |
||||
curvature_factor = VM.curvature_factor(v_ref) |
||||
print(curvature_factor) |
||||
|
||||
LANE_WIDTH = 3.9 |
||||
p_l = map(float, model_polyfit(points_l)) |
||||
p_r = map(float, model_polyfit(points_r)) |
||||
p_p = map(float, model_polyfit(points_c)) |
||||
|
||||
l_poly = libmpc_py.ffi.new("double[4]", p_l) |
||||
r_poly = libmpc_py.ffi.new("double[4]", p_r) |
||||
p_poly = libmpc_py.ffi.new("double[4]", p_p) |
||||
l_prob = 1.0 |
||||
r_prob = 1.0 |
||||
p_prob = 1.0 # This is always 1 |
||||
|
||||
|
||||
mpc_x_points = np.linspace(0., 2.5*v_ref, num=50) |
||||
points_poly_l = np.polyval(p_l, mpc_x_points) |
||||
points_poly_r = np.polyval(p_r, mpc_x_points) |
||||
points_poly_p = np.polyval(p_p, mpc_x_points) |
||||
print(points_poly_l) |
||||
|
||||
lanes_x = np.linspace(0, 49) |
||||
|
||||
cur_state = libmpc_py.ffi.new("state_t *") |
||||
cur_state[0].x = 0.0 |
||||
cur_state[0].y = 0.5 |
||||
cur_state[0].psi = 0.0 |
||||
cur_state[0].delta = 0.0 |
||||
|
||||
xs = [] |
||||
ys = [] |
||||
deltas = [] |
||||
titles = [ |
||||
'Steer rate cost', |
||||
'Heading cost', |
||||
'Lane cost', |
||||
'Path cost', |
||||
] |
||||
|
||||
# Steer rate cost |
||||
sol_x = OrderedDict() |
||||
sol_y = OrderedDict() |
||||
delta = OrderedDict() |
||||
for cost in np.logspace(-1, 1.0, 5): |
||||
libmpc.init(1.0, 3.0, 1.0, cost) |
||||
for _ in range(10): |
||||
libmpc.run_mpc(cur_state, mpc_solution, l_poly, r_poly, p_poly, l_prob, r_prob, |
||||
curvature_factor, v_ref, LANE_WIDTH) |
||||
sol_x[cost] = map(float, list(mpc_solution[0].x)) |
||||
sol_y[cost] = map(float, list(mpc_solution[0].y)) |
||||
delta[cost] = map(float, list(mpc_solution[0].delta)) |
||||
xs.append(sol_x) |
||||
ys.append(sol_y) |
||||
deltas.append(delta) |
||||
|
||||
# Heading cost |
||||
sol_x = OrderedDict() |
||||
sol_y = OrderedDict() |
||||
delta = OrderedDict() |
||||
for cost in np.logspace(-1, 1.0, 5): |
||||
libmpc.init(1.0, 3.0, cost, 1.0) |
||||
for _ in range(10): |
||||
libmpc.run_mpc(cur_state, mpc_solution, l_poly, r_poly, p_poly, l_prob, r_prob, |
||||
curvature_factor, v_ref, LANE_WIDTH) |
||||
sol_x[cost] = map(float, list(mpc_solution[0].x)) |
||||
sol_y[cost] = map(float, list(mpc_solution[0].y)) |
||||
delta[cost] = map(float, list(mpc_solution[0].delta)) |
||||
xs.append(sol_x) |
||||
ys.append(sol_y) |
||||
deltas.append(delta) |
||||
|
||||
# Lane cost |
||||
sol_x = OrderedDict() |
||||
sol_y = OrderedDict() |
||||
delta = OrderedDict() |
||||
for cost in np.logspace(-1, 2.0, 5): |
||||
libmpc.init(1.0, cost, 1.0, 1.0) |
||||
for _ in range(10): |
||||
libmpc.run_mpc(cur_state, mpc_solution, l_poly, r_poly, p_poly, l_prob, r_prob, |
||||
curvature_factor, v_ref, LANE_WIDTH) |
||||
sol_x[cost] = map(float, list(mpc_solution[0].x)) |
||||
sol_y[cost] = map(float, list(mpc_solution[0].y)) |
||||
delta[cost] = map(float, list(mpc_solution[0].delta)) |
||||
xs.append(sol_x) |
||||
ys.append(sol_y) |
||||
deltas.append(delta) |
||||
|
||||
|
||||
# Path cost |
||||
sol_x = OrderedDict() |
||||
sol_y = OrderedDict() |
||||
delta = OrderedDict() |
||||
for cost in np.logspace(-1, 1.0, 5): |
||||
libmpc.init(cost, 3.0, 1.0, 1.0) |
||||
for _ in range(10): |
||||
libmpc.run_mpc(cur_state, mpc_solution, l_poly, r_poly, p_poly, l_prob, r_prob, |
||||
curvature_factor, v_ref, LANE_WIDTH) |
||||
sol_x[cost] = map(float, list(mpc_solution[0].x)) |
||||
sol_y[cost] = map(float, list(mpc_solution[0].y)) |
||||
delta[cost] = map(float, list(mpc_solution[0].delta)) |
||||
xs.append(sol_x) |
||||
ys.append(sol_y) |
||||
deltas.append(delta) |
||||
|
||||
|
||||
|
||||
plt.figure() |
||||
|
||||
for i in range(len(xs)): |
||||
ax = plt.subplot(2, 2, i + 1) |
||||
sol_x = xs[i] |
||||
sol_y = ys[i] |
||||
for cost in sol_x.keys(): |
||||
plt.plot(sol_x[cost], sol_y[cost]) |
||||
|
||||
plt.plot(lanes_x, points_r, '.b') |
||||
plt.plot(lanes_x, points_l, '.b') |
||||
plt.plot(lanes_x, (points_l + points_r) / 2.0, '--g') |
||||
plt.plot(mpc_x_points, points_poly_l, 'b') |
||||
plt.plot(mpc_x_points, points_poly_r, 'b') |
||||
plt.plot(mpc_x_points, (points_poly_l + points_poly_r) / 2.0, 'g') |
||||
plt.legend(map(lambda x: str(round(x, 2)), sol_x.keys()) + ['right', 'left', 'center'], loc=3) |
||||
plt.title(titles[i]) |
||||
plt.grid(True) |
||||
# ax.set_aspect('equal', 'datalim') |
||||
|
||||
|
||||
plt.figure() |
||||
for i in range(len(xs)): |
||||
plt.subplot(2, 2, i + 1) |
||||
sol_x = xs[i] |
||||
delta = deltas[i] |
||||
|
||||
for cost in sol_x.keys(): |
||||
plt.plot(delta[cost]) |
||||
plt.title(titles[i]) |
||||
plt.legend(map(lambda x: str(round(x, 2)), sol_x.keys()), loc=3) |
||||
plt.grid(True) |
||||
|
||||
plt.show() |
@ -0,0 +1,168 @@ |
||||
#! /usr/bin/env python |
||||
import numpy as np |
||||
import matplotlib.pyplot as plt |
||||
from selfdrive.controls.lib.longitudinal_mpc import libmpc_py |
||||
from selfdrive.controls.lib.drive_helpers import MPC_COST_LONG |
||||
import math |
||||
|
||||
# plot liongitudinal MPC trajectory by defining boundary conditions: |
||||
# ego and lead vehicles state. Use this script to tune MPC costs |
||||
|
||||
def RW(v_ego, v_l): |
||||
TR = 1.8 |
||||
G = 9.81 |
||||
return (v_ego * TR - (v_l - v_ego) * TR + v_ego*v_ego/(2*G) - v_l*v_l / (2*G)) |
||||
|
||||
|
||||
def NORM_RW_ERROR(v_ego, v_l, p): |
||||
return (RW(v_ego, v_l) + 4.0 - p) |
||||
return (RW(v_ego, v_l) + 4.0 - p) / (np.sqrt(v_ego + 0.5) + 0.1) |
||||
|
||||
|
||||
v_ego = 20.0 |
||||
a_ego = 0 |
||||
|
||||
x_lead = 10.0 |
||||
v_lead = 20.0 |
||||
a_lead = -3.0 |
||||
a_lead_tau = 0. |
||||
|
||||
# v_ego = 7.02661012716 |
||||
# a_ego = -1.26143024772 |
||||
|
||||
# x_lead = 29.625 + 20 |
||||
# v_lead = 0.725235462189 + 1 |
||||
# a_lead = -1.00025629997 |
||||
|
||||
# a_lead_tau = 2.90729817665 |
||||
|
||||
min_a_lead_tau = (a_lead**2 * math.pi) / (2 * (v_lead + 0.01)**2) |
||||
min_a_lead_tau = 0.0 |
||||
|
||||
print(a_lead_tau, min_a_lead_tau) |
||||
a_lead_tau = max(a_lead_tau, min_a_lead_tau) |
||||
|
||||
ffi, libmpc = libmpc_py.get_libmpc(1) |
||||
libmpc.init(MPC_COST_LONG.TTC, MPC_COST_LONG.DISTANCE, MPC_COST_LONG.ACCELERATION, MPC_COST_LONG.JERK) |
||||
libmpc.init_with_simulation(v_ego, x_lead, v_lead, a_lead, a_lead_tau) |
||||
|
||||
cur_state = ffi.new("state_t *") |
||||
cur_state[0].x_ego = 0.0 |
||||
cur_state[0].v_ego = v_ego |
||||
cur_state[0].a_ego = a_ego |
||||
cur_state[0].x_l = x_lead |
||||
cur_state[0].v_l = v_lead |
||||
|
||||
mpc_solution = ffi.new("log_t *") |
||||
|
||||
for _ in range(10): |
||||
print(libmpc.run_mpc(cur_state, mpc_solution, a_lead_tau, a_lead)) |
||||
|
||||
|
||||
for i in range(21): |
||||
print("t: %.2f\t x_e: %.2f\t v_e: %.2f\t a_e: %.2f\t" % (mpc_solution[0].t[i], mpc_solution[0].x_ego[i], mpc_solution[0].v_ego[i], mpc_solution[0].a_ego[i])) |
||||
print("x_l: %.2f\t v_l: %.2f\t \t" % (mpc_solution[0].x_l[i], mpc_solution[0].v_l[i])) |
||||
|
||||
t = np.hstack([np.arange(0., 1.0, 0.2), np.arange(1.0, 10.1, 0.6)]) |
||||
|
||||
print(map(float, mpc_solution[0].x_ego)[-1]) |
||||
print(map(float, mpc_solution[0].x_l)[-1] - map(float, mpc_solution[0].x_ego)[-1]) |
||||
|
||||
plt.figure(figsize=(8, 8)) |
||||
|
||||
plt.subplot(4, 1, 1) |
||||
x_l = np.array(map(float, mpc_solution[0].x_l)) |
||||
plt.plot(t, map(float, mpc_solution[0].x_ego)) |
||||
plt.plot(t, x_l) |
||||
plt.legend(['ego', 'lead']) |
||||
plt.title('x') |
||||
plt.grid() |
||||
|
||||
plt.subplot(4, 1, 2) |
||||
v_ego = np.array(map(float, mpc_solution[0].v_ego)) |
||||
v_l = np.array(map(float, mpc_solution[0].v_l)) |
||||
plt.plot(t, v_ego) |
||||
plt.plot(t, v_l) |
||||
plt.legend(['ego', 'lead']) |
||||
plt.ylim([-1, max(max(v_ego), max(v_l))]) |
||||
plt.title('v') |
||||
plt.grid() |
||||
|
||||
plt.subplot(4, 1, 3) |
||||
plt.plot(t, map(float, mpc_solution[0].a_ego)) |
||||
plt.plot(t, map(float, mpc_solution[0].a_l)) |
||||
plt.legend(['ego', 'lead']) |
||||
plt.title('a') |
||||
plt.grid() |
||||
|
||||
|
||||
plt.subplot(4, 1, 4) |
||||
d_l = np.array(map(float, mpc_solution[0].x_l)) - np.array(map(float, mpc_solution[0].x_ego)) |
||||
desired = 4.0 + RW(v_ego, v_l) |
||||
|
||||
plt.plot(t, d_l) |
||||
plt.plot(t, desired, '--') |
||||
plt.ylim(-1, max(max(desired), max(d_l))) |
||||
plt.legend(['relative distance', 'desired distance']) |
||||
plt.grid() |
||||
|
||||
plt.show() |
||||
|
||||
# c1 = np.exp(0.3 * NORM_RW_ERROR(v_ego, v_l, d_l)) |
||||
# c2 = np.exp(4.5 - d_l) |
||||
# print(c1) |
||||
# print(c2) |
||||
|
||||
# plt.figure() |
||||
# plt.plot(t, c1, label="NORM_RW_ERROR") |
||||
# plt.plot(t, c2, label="penalty function") |
||||
# plt.legend() |
||||
|
||||
# ## OLD MPC |
||||
# a_lead_tau = 1.5 |
||||
# a_lead_tau = max(a_lead_tau, -a_lead / (v_lead + 0.01)) |
||||
|
||||
# ffi, libmpc = libmpc_py.get_libmpc(1) |
||||
# libmpc.init(MPC_COST_LONG.TTC, MPC_COST_LONG.DISTANCE, MPC_COST_LONG.ACCELERATION, MPC_COST_LONG.JERK) |
||||
# libmpc.init_with_simulation(v_ego, x_lead, v_lead, a_lead, a_lead_tau) |
||||
|
||||
# cur_state = ffi.new("state_t *") |
||||
# cur_state[0].x_ego = 0.0 |
||||
# cur_state[0].v_ego = v_ego |
||||
# cur_state[0].a_ego = a_ego |
||||
# cur_state[0].x_lead = x_lead |
||||
# cur_state[0].v_lead = v_lead |
||||
# cur_state[0].a_lead = a_lead |
||||
|
||||
# mpc_solution = ffi.new("log_t *") |
||||
|
||||
# for _ in range(10): |
||||
# print libmpc.run_mpc(cur_state, mpc_solution, a_lead_tau) |
||||
|
||||
# t = np.hstack([np.arange(0., 1.0, 0.2), np.arange(1.0, 10.1, 0.6)]) |
||||
|
||||
# print(map(float, mpc_solution[0].x_ego)[-1]) |
||||
# print(map(float, mpc_solution[0].x_lead)[-1] - map(float, mpc_solution[0].x_ego)[-1]) |
||||
# plt.subplot(4, 2, 2) |
||||
# plt.plot(t, map(float, mpc_solution[0].x_ego)) |
||||
# plt.plot(t, map(float, mpc_solution[0].x_lead)) |
||||
# plt.legend(['ego', 'lead']) |
||||
# plt.title('x') |
||||
|
||||
# plt.subplot(4, 2, 4) |
||||
# plt.plot(t, map(float, mpc_solution[0].v_ego)) |
||||
# plt.plot(t, map(float, mpc_solution[0].v_lead)) |
||||
# plt.legend(['ego', 'lead']) |
||||
# plt.title('v') |
||||
|
||||
# plt.subplot(4, 2, 6) |
||||
# plt.plot(t, map(float, mpc_solution[0].a_ego)) |
||||
# plt.plot(t, map(float, mpc_solution[0].a_lead)) |
||||
# plt.legend(['ego', 'lead']) |
||||
# plt.title('a') |
||||
|
||||
|
||||
# plt.subplot(4, 2, 8) |
||||
# plt.plot(t, np.array(map(float, mpc_solution[0].x_lead)) - np.array(map(float, mpc_solution[0].x_ego))) |
||||
|
||||
# plt.show() |
@ -0,0 +1,27 @@ |
||||
#!/usr/bin/env python3 |
||||
from selfdrive.car.fingerprints import eliminate_incompatible_cars, all_known_cars |
||||
import cereal.messaging as messaging |
||||
|
||||
|
||||
# Prius and Leuxs es 300H |
||||
fingerprint = {898: 8, 905: 8, 810: 2, 1552: 8, 1553: 8, 1556: 8, 1557: 8, 921: 8, 800: 8, 944: 8, 1570: 8, 1059: 1, 36: 8, 37: 8, 550: 8, 295: 8, 296: 8, 170: 8, 1071: 8, 560: 7, 945: 8, 562: 6, 180: 8, 1077: 8, 950: 8, 951: 8, 953: 8, 1595: 8, 1084: 8, 829: 2, 1086: 8, 1568: 8, 452: 8, 581: 5, 1057: 8, 713: 8, 971: 7, 975: 5, 1571: 8, 466: 8, 467: 8, 1572: 8, 1114: 8, 933: 8, 863: 8, 608: 8, 993: 8, 610: 8, 955: 8, 166: 8, 1056: 8, 956: 8, 1132: 8, 1085: 8, 552: 4, 1779: 8, 1017: 8, 1020: 8, 426: 6, 1279: 8} |
||||
|
||||
# rav4 2019 and corolla tss2 |
||||
fingerprint = {896: 8, 898: 8, 976: 1, 1541: 8, 905: 8, 1164: 8, 1165: 8, 1166: 8, 1167: 8, 1552: 8, 1553: 8, 1556: 8, 921: 8, 1056: 8, 544: 4, 1570: 8, 1059: 1, 36: 8, 37: 8, 550: 8, 552: 4, 170: 8, 812: 8, 944: 8, 945: 8, 562: 6, 180: 8, 1077: 8, 951: 8, 824: 8, 1076: 8, 186: 4, 955: 8, 956: 8, 705: 8, 452: 8, 1592: 8, 464: 8, 1571: 8, 466: 8, 467: 8, 761: 8, 728: 8, 1572: 8, 1114: 8, 933: 8, 800: 8, 608: 8, 865: 8, 610: 8, 1595: 8, 1745: 8, 764: 8, 1002: 8, 1649: 8, 1779: 8, 1568: 8, 1017: 8, 1279: 8, 1020: 8, 810: 2, 426: 6} |
||||
|
||||
# rav4 2019 and corolla tss2 |
||||
fingerprint = {896: 8, 898: 8, 900: 6, 976: 1, 1541: 8, 902: 6, 905: 8, 810: 2, 1164: 8, 1165: 8, 1166: 8, 1167: 8, 1552: 8, 1553: 8, 1556: 8, 1571: 8, 921: 8, 1056: 8, 544: 4, 1570: 8, 1059: 1, 36: 8, 37: 8, 550: 8, 935: 8, 552: 4, 170: 8, 812: 8, 944: 8, 945: 8, 562: 6, 180: 8, 1077: 8, 951: 8, 1592: 8, 1076: 8, 186: 4, 955: 8, 956: 8, 1001: 8, 705: 8, 452: 8, 1788: 8, 464: 8, 824: 8, 466: 8, 467: 8, 761: 8, 728: 8, 1572: 8, 1114: 8, 933: 8, 800: 8, 608: 8, 865: 8, 610: 8, 1595: 8, 934: 8, 998: 5, 1745: 8, 1000: 8, 764: 8, 1002: 8, 999: 7, 1789: 8, 1649: 8, 1779: 8, 1568: 8, 1017: 8, 1786: 8, 1787: 8, 1020: 8, 426: 6, 1279: 8} |
||||
|
||||
candidate_cars = all_known_cars() |
||||
|
||||
|
||||
for addr, l in fingerprint.items(): |
||||
dat = messaging.new_message() |
||||
dat.init('can', 1) |
||||
|
||||
msg = dat.can[0] |
||||
msg.address = addr |
||||
msg.dat = " " * l |
||||
|
||||
candidate_cars = eliminate_incompatible_cars(msg, candidate_cars) |
||||
print(candidate_cars) |
@ -0,0 +1,67 @@ |
||||
#!/usr/bin/env python3 |
||||
""" |
||||
This tool can be used to quickly changes the values in a JSON file used for tuning |
||||
Keys like in vim: |
||||
- h: decrease by 0.05 |
||||
- l: increase by 0.05 |
||||
- k: move pointer up |
||||
- j: move pointer down |
||||
""" |
||||
|
||||
import tty |
||||
import sys |
||||
import json |
||||
import termios |
||||
from collections import OrderedDict |
||||
|
||||
FILENAME = '/data/tuning.json' |
||||
|
||||
def read_tuning(): |
||||
while True: |
||||
try: |
||||
return json.loads(open(FILENAME).read()) |
||||
except: |
||||
pass |
||||
|
||||
def main(): |
||||
dat = json.loads(open(FILENAME, 'r').read()) |
||||
dat = OrderedDict(sorted(dat.items(), key=lambda i: i[0])) |
||||
|
||||
cur = 0 |
||||
while True: |
||||
sys.stdout.write("\x1Bc") |
||||
|
||||
for i, (k, v) in enumerate(dat.items()): |
||||
prefix = "> " if cur == i else " " |
||||
print((prefix + k).ljust(20) + "%.2f" % v) |
||||
|
||||
key = sys.stdin.read(1)[0] |
||||
|
||||
write = False |
||||
if key == "k": |
||||
cur = max(0, cur - 1) |
||||
elif key == "j": |
||||
cur = min(len(dat.keys()) - 1, cur + 1) |
||||
elif key == "l": |
||||
dat[dat.keys()[cur]] += 0.05 |
||||
write = True |
||||
elif key == "h": |
||||
dat[dat.keys()[cur]] -= 0.05 |
||||
write = True |
||||
elif key == "q": |
||||
break |
||||
|
||||
if write: |
||||
open(FILENAME, 'w').write(json.dumps(dat)) |
||||
|
||||
|
||||
if __name__ == "__main__": |
||||
orig_settings = termios.tcgetattr(sys.stdin) |
||||
tty.setcbreak(sys.stdin) |
||||
|
||||
try: |
||||
main() |
||||
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, orig_settings) |
||||
except: |
||||
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, orig_settings) |
||||
raise |
Loading…
Reference in new issue