add example for the recv audio sd
This commit is contained in:
95
example/recv_audio_sd.py
Normal file
95
example/recv_audio_sd.py
Normal file
@@ -0,0 +1,95 @@
|
|||||||
|
import sys
|
||||||
|
import queue
|
||||||
|
import numpy as np
|
||||||
|
import sounddevice as sd
|
||||||
|
import NDIlib as ndi
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
|
||||||
|
if not ndi.initialize():
|
||||||
|
print('Cannot run NDI.')
|
||||||
|
return 0
|
||||||
|
|
||||||
|
find_create_desc = ndi.FindCreate()
|
||||||
|
ndi_find = ndi.find_create_v2(find_create_desc)
|
||||||
|
|
||||||
|
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)
|
||||||
|
|
||||||
|
if len(sources) < 1:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
recv_create_desc = ndi.RecvCreateV3()
|
||||||
|
recv_create_desc.source_to_connect_to = sources[0]
|
||||||
|
recv_create_desc.ndi_recv_name = 'Example Audio Converter Receiver'
|
||||||
|
|
||||||
|
ndi_recv = ndi.recv_create_v3(recv_create_desc)
|
||||||
|
|
||||||
|
if ndi_recv is None:
|
||||||
|
ndi.find_destroy(ndi_find)
|
||||||
|
return 0
|
||||||
|
|
||||||
|
ndi.find_destroy(ndi_find)
|
||||||
|
|
||||||
|
q = queue.Queue(maxsize=20)
|
||||||
|
|
||||||
|
def audio_callback(outdata, frames, time, status):
|
||||||
|
try:
|
||||||
|
data = q.get_nowait()
|
||||||
|
if not status.output_underflow:
|
||||||
|
outdata[:] = data
|
||||||
|
except queue.Empty as e:
|
||||||
|
return
|
||||||
|
|
||||||
|
stream = sd.RawOutputStream(
|
||||||
|
samplerate=48000, blocksize=1024, channels=2, dtype='int16', callback=audio_callback)
|
||||||
|
|
||||||
|
with stream:
|
||||||
|
while True:
|
||||||
|
t, v, a, m = ndi.recv_capture_v2(ndi_recv, 1000)
|
||||||
|
|
||||||
|
if t == ndi.FRAME_TYPE_NONE:
|
||||||
|
print('No data received.')
|
||||||
|
continue
|
||||||
|
|
||||||
|
if t == ndi.FRAME_TYPE_VIDEO:
|
||||||
|
print('Video data received (%dx%d).' % (v.xres, v.yres))
|
||||||
|
ndi.recv_free_video_v2(ndi_recv, v)
|
||||||
|
continue
|
||||||
|
|
||||||
|
if t == ndi.FRAME_TYPE_AUDIO:
|
||||||
|
print('Audio data received (%d samples).' % a.no_samples)
|
||||||
|
data = np.zeros((a.no_channels, a.no_samples), np.int16)
|
||||||
|
interleaved = ndi.AudioFrameInterleaved16s()
|
||||||
|
interleaved.data = data
|
||||||
|
ndi.util_audio_to_interleaved_16s_v2(a, interleaved)
|
||||||
|
timeout = interleaved.no_samples * 20 / interleaved.sample_rate
|
||||||
|
q.put(data, timeout=timeout)
|
||||||
|
ndi.recv_free_audio_v2(ndi_recv, a)
|
||||||
|
continue
|
||||||
|
|
||||||
|
if t == ndi.FRAME_TYPE_METADATA:
|
||||||
|
print('Meta data received.')
|
||||||
|
ndi.recv_free_metadata(ndi_recv, m)
|
||||||
|
continue
|
||||||
|
|
||||||
|
if t == ndi.FRANE_TYPE_STATUS_CHANGE:
|
||||||
|
print('Receiver connection status changed.')
|
||||||
|
continue
|
||||||
|
|
||||||
|
ndi.recv_destroy(ndi_recv)
|
||||||
|
|
||||||
|
ndi.destroy()
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
sys.exit(main())
|
||||||
@@ -1,2 +1,3 @@
|
|||||||
numpy
|
numpy
|
||||||
opencv-python
|
opencv-python
|
||||||
|
sounddevice
|
||||||
Reference in New Issue
Block a user