/
sampler.py
48 lines (35 loc) · 1.38 KB
/
sampler.py
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
import numpy
from .pulseaudio import PulseAudioSignalAnalayzer
class Sampler(PulseAudioSignalAnalayzer):
"""
Samples the audio signal recorded from the PulseAudio source
and optionally appends the samples to a fixed-size buffer.
"""
def __init__(self, sample_size, buffer_size=None, **kwargs):
super().__init__(**kwargs)
self._samples = self._pulseaudio_client.create_buffer(size=sample_size)
if buffer_size is None:
buffer_size = sample_size
if buffer_size >= sample_size:
self._buffer = numpy.zeros(buffer_size, dtype='f')
else:
raise Exception('buffer_size is too small: {0} < {1}'.format(buffer_size, sample_size))
@property
def sample_size(self):
return len(self._samples)
@property
def buffer(self):
if self._buffer is None:
raise Exception('buffer_size has not been set.')
return self._buffer
@property
def buffer_size(self):
if self._buffer is None:
raise Exception('buffer_size has not been set.')
return len(self._buffer)
def _sample(self):
self._pulseaudio_client.read_into_buffer(self._samples)
if self._buffer is not None:
n = self.buffer_size - self.sample_size
self._buffer[0:n] = self._buffer[self.sample_size:]
self._buffer[n:] = self._samples