Skip to content

Instantly share code, notes, and snippets.

@arihans
Created March 31, 2020 11:47
Show Gist options
  • Save arihans/a7b63ddfb41a6e2395e842560f0d103e to your computer and use it in GitHub Desktop.
Save arihans/a7b63ddfb41a6e2395e842560f0d103e to your computer and use it in GitHub Desktop.
Convert arduino serial port reading to tone
import numpy
import pyaudio
import math
import time
# from gsr-reader import return_frequency
import serial
ser = serial.Serial('/dev/ttyACM0')
ser.flushInput()
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)
generator = ToneGenerator()
frequency_start = 50 # Frequency to start the sweep from
frequency_end = 10000 # Frequency to end the sweep at
num_frequencies = 200 # Number of frequencies in the sweep
amplitude = 0.50 # Amplitude of the waveform
step_duration = 0.43 # Time (seconds) to play at each step
while True:
ser_bytes = ser.readline()
try :
frequency = float(ser_bytes[0:len(ser_bytes)-2])
except:
frequency = 0
print("Playing tone at {0:0.2f} Hz".format(frequency)+" [{}]".format(ser_bytes))
generator.play(frequency, step_duration, amplitude)
time.sleep(0.1)
# while generator.is_playing():
# pass
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment