Add video receiving example
This commit is contained in:
57
example/recv.py
Normal file
57
example/recv.py
Normal file
@@ -0,0 +1,57 @@
|
|||||||
|
import sys
|
||||||
|
import numpy as np
|
||||||
|
import cv2 as cv
|
||||||
|
import NDIlib as ndi
|
||||||
|
|
||||||
|
def main():
|
||||||
|
|
||||||
|
if not ndi.initialize():
|
||||||
|
return 0
|
||||||
|
|
||||||
|
ndi_find = ndi.find_create_v2()
|
||||||
|
|
||||||
|
if ndi_find is None:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
sources = []
|
||||||
|
while not len(sources) > 0:
|
||||||
|
print('Looking for sources ...')
|
||||||
|
ndi.find_wait_for_sources(ndi_find, 1000)
|
||||||
|
sources = ndi.find_get_current_sources(ndi_find)
|
||||||
|
|
||||||
|
ndi_recv_create = ndi.RecvCreateV3()
|
||||||
|
ndi_recv_create.color_format = ndi.RECV_COLOR_FORMAT_BGRX_BGRA
|
||||||
|
|
||||||
|
ndi_recv = ndi.recv_create_v3(ndi_recv_create)
|
||||||
|
|
||||||
|
if ndi_recv is None:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
ndi.recv_connect(ndi_recv, sources[0])
|
||||||
|
|
||||||
|
ndi.find_destroy(ndi_find)
|
||||||
|
|
||||||
|
cv.startWindowThread()
|
||||||
|
|
||||||
|
frame = np.array((1920))
|
||||||
|
|
||||||
|
while True:
|
||||||
|
t,v,_,_ = ndi.recv_capture_v2(ndi_recv, 5000)
|
||||||
|
|
||||||
|
if t == ndi.FRAME_TYPE_VIDEO:
|
||||||
|
print("Video data received (%dx%d)." % (v.xres, v.yres))
|
||||||
|
frame = np.copy(v.data)
|
||||||
|
cv.imshow('ndi image',frame)
|
||||||
|
ndi.recv_free_video_v2(ndi_recv, v)
|
||||||
|
|
||||||
|
if cv.waitKey(1) & 0xff == 27:
|
||||||
|
break
|
||||||
|
|
||||||
|
ndi.recv_destroy(ndi_recv)
|
||||||
|
ndi.destroy()
|
||||||
|
cv.destroyAllWindows()
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
||||||
Reference in New Issue
Block a user