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