Add video receiving example

This commit is contained in:
Naoto Kondo
2020-04-25 07:32:47 +09:00
parent 300075443b
commit b096ab9366

57
example/recv.py Normal file
View 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())