camerastream: update receive.py with some options
parent
814741dafe
commit
fa4b434e43
|
@ -11,23 +11,32 @@ import cereal.messaging as messaging
|
||||||
# also start bridge
|
# also start bridge
|
||||||
# then run this "./receive.py <ip>"
|
# then run this "./receive.py <ip>"
|
||||||
|
|
||||||
SCALE = 1
|
if "FULL" in os.environ:
|
||||||
XMIN = 771
|
SCALE = 2
|
||||||
XMAX = 1156
|
XMIN, XMAX = 0, 1927
|
||||||
YMIN = 483
|
YMIN, YMAX = 0, 1207
|
||||||
YMAX = 724
|
else:
|
||||||
|
SCALE = 1
|
||||||
|
XMIN = 771
|
||||||
|
XMAX = 1156
|
||||||
|
YMIN = 483
|
||||||
|
YMAX = 724
|
||||||
H, W = ((YMAX-YMIN+1)//SCALE, (XMAX-XMIN+1)//SCALE)
|
H, W = ((YMAX-YMIN+1)//SCALE, (XMAX-XMIN+1)//SCALE)
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
cameras = ['roadCameraState', 'wideRoadCameraState', 'driverCameraState']
|
cameras = ['roadCameraState', 'wideRoadCameraState', 'driverCameraState']
|
||||||
|
if "CAM" in os.environ:
|
||||||
|
cam = int(os.environ['CAM'])
|
||||||
|
cameras = cameras[cam:cam+1]
|
||||||
sm = messaging.SubMaster(cameras, addr=sys.argv[1])
|
sm = messaging.SubMaster(cameras, addr=sys.argv[1])
|
||||||
win = Window(W*3, H)
|
win = Window(W*len(cameras), H)
|
||||||
bdat = np.zeros((H, W*3, 3), dtype=np.uint8)
|
bdat = np.zeros((H, W*len(cameras), 3), dtype=np.uint8)
|
||||||
|
|
||||||
while 1:
|
while 1:
|
||||||
sm.update()
|
sm.update()
|
||||||
for i,k in enumerate(cameras):
|
for i,k in enumerate(cameras):
|
||||||
if sm.updated[k]:
|
if sm.updated[k]:
|
||||||
|
#print("update", k)
|
||||||
bgr_raw = sm[k].image
|
bgr_raw = sm[k].image
|
||||||
dat = np.frombuffer(bgr_raw, dtype=np.uint8).reshape(H, W, 3)[:, :, [2,1,0]]
|
dat = np.frombuffer(bgr_raw, dtype=np.uint8).reshape(H, W, 3)[:, :, [2,1,0]]
|
||||||
bdat[:, W*i:W*(i+1)] = dat
|
bdat[:, W*i:W*(i+1)] = dat
|
||||||
|
|
Loading…
Reference in New Issue