-
Notifications
You must be signed in to change notification settings - Fork 7
/
Sound_Generator.py
57 lines (50 loc) · 2.21 KB
/
Sound_Generator.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
49
50
51
52
53
54
55
56
57
import numpy
import pyaudio
import math
class ToneGenerator(object):
def __init__(self, samplerate=44100, frames_per_buffer=4410):
self.p = pyaudio.PyAudio()
self.samplerate = samplerate
self.frames_per_buffer = frames_per_buffer
self.streamOpen = False
def sinewave(self):
if self.buffer_offset + self.frames_per_buffer - 1 > self.x_max:
# We don't need a full buffer or audio so pad the end with 0's
xs = numpy.arange(self.buffer_offset,
self.x_max)
tmp = self.amplitude * numpy.sin(xs * self.omega)
out = numpy.append(tmp,
numpy.zeros(self.frames_per_buffer - len(tmp)))
else:
xs = numpy.arange(self.buffer_offset,
self.buffer_offset + self.frames_per_buffer)
out = self.amplitude * numpy.sin(xs * self.omega)
self.buffer_offset += self.frames_per_buffer
return out
def callback(self, in_data, frame_count, time_info, status):
if self.buffer_offset < self.x_max:
data = self.sinewave().astype(numpy.float32)
return (data.tostring(), pyaudio.paContinue)
else:
return (None, pyaudio.paComplete)
def is_playing(self):
if self.stream.is_active():
return True
else:
if self.streamOpen:
self.stream.stop_stream()
self.stream.close()
self.streamOpen = False
return False
def play(self, frequency, duration, amplitude):
self.omega = float(frequency) * (math.pi * 2) / self.samplerate
self.amplitude = amplitude
self.buffer_offset = 0
self.streamOpen = True
self.x_max = math.ceil(self.samplerate * duration) - 1
self.stream = self.p.open(format=pyaudio.paFloat32,
channels=1,
rate=self.samplerate,
output=True,
frames_per_buffer=self.frames_per_buffer,
stream_callback=self.callback)