diff --git a/example/send_png.py b/example/send_png.py index 976e3de..0b3e569 100644 --- a/example/send_png.py +++ b/example/send_png.py @@ -1,43 +1,39 @@ import sys import time import numpy as np -import cv2 -import NDIlib +import cv2 as cv +import NDIlib as ndi def main(): - if not NDIlib.initialize(): + if not ndi.initialize(): return 0 - img = cv2.imread('image/test2.png', cv2.IMREAD_ANYCOLOR) - img = cv2.cvtColor(img, cv2.COLOR_BGR2RGBA) + img = cv.imread('image/test2.png', cv.IMREAD_ANYCOLOR) + img = cv.cvtColor(img, cv.COLOR_BGR2RGBA) - h, w = img.shape[:2] + send_settings = ndi.SendCreate() + send_settings.ndi_name = 'ndi-python' - send_settings = NDIlib.SendCreate() - send_settings.p_ndi_name = b'ndi-python' + ndi_send = ndi.send_create(send_settings) - ndi_send = NDIlib.send_create(send_settings) + video_frame = ndi.VideoFrameV2() - video_frame = NDIlib.VideoFrameV2() - video_frame.xres = w - video_frame.yres = h - video_frame.FourCC = NDIlib.FOURCC_TYPE_RGBA - video_frame.p_data = img.ctypes.data_as(NDIlib.POINTER(NDIlib.c_uint8)) - video_frame.line_stride_in_bytes = w * 4 + video_frame.data = img + video_frame.FourCC = ndi.FOURCC_VIDEO_TYPE_RGBX start = time.time() while time.time() - start < 60 * 5: start_send = time.time() - for idx in reversed(range(200)): - NDIlib.send_send_video_v2(ndi_send, video_frame) + for _ in reversed(range(200)): + ndi.send_send_video_v2(ndi_send, video_frame) 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