Update send_video.py

This commit is contained in:
Naoto Kondo
2020-04-25 00:33:27 +09:00
parent 7434789358
commit 4218d0d753
2 changed files with 11 additions and 17 deletions

View File

@@ -22,7 +22,7 @@ def main():
while time.time() - start < 60 * 5: while time.time() - start < 60 * 5:
start_send = time.time() start_send = time.time()
for idx in reversed(range(200)): for _ in reversed(range(200)):
ret, img = cap.read() ret, img = cap.read()

View File

@@ -1,30 +1,24 @@
import sys import sys
import time import time
import numpy as np import numpy as np
import NDIlib import NDIlib as ndi
import ctypes
def main(): def main():
if not NDIlib.initialize(): if not ndi.initialize():
return 0 return 0
ndi_send = NDIlib.send_create() ndi_send = ndi.send_create()
if ndi_send is 0: if ndi_send is None:
return 0 return 0
img = np.zeros((1080, 1920, 4), dtype=np.uint8) img = np.zeros((1080, 1920, 4), dtype=np.uint8)
h, w = img.shape[:2]
video_frame = NDIlib.VideoFrameV2() video_frame = ndi.VideoFrameV2()
video_frame.xres = w video_frame.data = img
video_frame.yres = h video_frame.FourCC = ndi.FOURCC_VIDEO_TYPE_BGRX
video_frame.FourCC = NDIlib.FOURCC_TYPE_BGRX
#video_frame.data = img.ctypes.data_as(NDIlib.POINTER(NDIlib.c_uint8))
#video_frame.data = img.ctypes.data_as(ctypes.POINTER(ctypes.c_uint8))
print(type(video_frame.data))
start = time.time() start = time.time()
while time.time() - start < 60 * 5: while time.time() - start < 60 * 5:
@@ -32,13 +26,13 @@ def main():
for idx in reversed(range(200)): for idx in reversed(range(200)):
img.fill(255 if idx % 2 else 0) img.fill(255 if idx % 2 else 0)
NDIlib.send_send_video_v2(ndi_send, video_frame) ndi.send_send_video_v2(ndi_send, video_frame)
print('200 frames sent, at %1.2ffps' % (200.0 / (time.time() - start_send))) print('200 frames sent, at %1.2ffps' % (200.0 / (time.time() - start_send)))
NDIlib.send_destroy(ndi_send) ndi.send_destroy(ndi_send)
NDIlib.destroy() ndi.destroy()
return 0 return 0