rename recv.py to recv_cv.py
This commit is contained in:
@@ -1,55 +0,0 @@
|
||||
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()
|
||||
|
||||
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