ndi-python/example/recv_audio_sd.py at 58a6751c7c203db6b9ec819c259a67c7d1d13849 · buresu/ndi-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
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.FRAME_TYPE_STATUS_CHANGE:
                print('Receiver connection status changed.')
                continue

    ndi.recv_destroy(ndi_recv)

    ndi.destroy()

    return 0


if __name__ == "__main__":
    sys.exit(main())