49 lines
1.4 KiB
Python
Executable File
49 lines
1.4 KiB
Python
Executable File
# flake8: noqa
|
|
# pylint: disable=W
|
|
|
|
#!/usr/bin/env python
|
|
import numpy as np
|
|
import cv2
|
|
from time import time, sleep
|
|
|
|
H, W = (604*2//6, 964*2//6)
|
|
# H, W = (604, 964)
|
|
|
|
cam_bufs = np.zeros((3,H,W,3), dtype=np.uint8)
|
|
hist_bufs = np.zeros((3,H,200,3), dtype=np.uint8)
|
|
|
|
if __name__ == '__main__':
|
|
import zmq
|
|
context = zmq.Context()
|
|
socket = context.socket(zmq.PULL)
|
|
socket.bind("tcp://192.168.3.4:7768")
|
|
while True:
|
|
try:
|
|
message = socket.recv()
|
|
except Exception as ex:
|
|
print(ex)
|
|
message = b"123"
|
|
|
|
dat = np.frombuffer(message, dtype=np.uint8)
|
|
cam_id = (dat[0] + 1) % 3
|
|
# import pdb; pdb.set_trace()
|
|
b = dat[::3].reshape(H, W)
|
|
g = dat[1::3].reshape(H, W)
|
|
r = dat[2::3].reshape(H, W)
|
|
cam_bufs[cam_id] = cv2.merge((r, g, b))
|
|
cam_bufs[cam_id]= cv2.cvtColor(cam_bufs[cam_id], cv2.COLOR_RGB2BGR)
|
|
|
|
hist = cv2.calcHist([cv2.cvtColor(cam_bufs[cam_id], cv2.COLOR_BGR2GRAY)],[0],None,[32],[0,256])
|
|
hist = (H*hist/hist.max()).astype(np.uint8)
|
|
hist_bufs[cam_id] = 0
|
|
for i,bb in enumerate(hist):
|
|
hist_bufs[cam_id, H-bb[0]:,i*(200//32):(i+1)*(200//32), :] = (222,222,222)
|
|
|
|
out = cam_bufs.reshape((3*H,W,3))
|
|
hist_bufs_out = hist_bufs.reshape((3*H,200,3))
|
|
out = np.hstack([out, hist_bufs_out])
|
|
cv2.imshow('RGB', out)
|
|
cv2.waitKey(55)
|
|
#dat.tofile('/tmp/c3rgb.img')
|
|
#cv2.imwrite('/tmp/c3rgb.png', out)
|