Commit 4d96e938 authored by hackEns's avatar hackEns
Browse files

Working on code for Noel2013

Started serial handling

TODO : Serial in main loop + args
parent c59cbd10
......@@ -21,6 +21,9 @@
from sound4python import sound
from multiprocessing import Process
import math
import serial
import sys
import getopt
class _Getch:
......@@ -46,45 +49,109 @@ def play_wave(frequency=440, nb_secs=1.):
sine_wave = []
for i in range(math.ceil(nb_secs*framerate)+1):
sine_wave.append(int(16384*math.sin(2*math.pi*frequency*i/framerate)))
sound(sine_wave)
def read_serial(ser):
line = ser.readline()
line = line.decode().strip("\r\n")
line = line.split(" ")
return [int(i or 0) for i in line]
framerate = 16000
processes = []
frequency = 440
frequencies = {"FA": 698.5, "SOL": 784, "LA": 880, "SIb": 932, "DO": 1046.5}
notes = frequencies.keys()
thresholds = []
measures = []
serial_port = "/dev/ttyACM0"
serial_speed = 115200
try:
opts, args = getopt.getopt(sys.argv[1:], "hs:", ["help", "serial="])
for opt, arg in opts:
if opt in ("-h", "--help"):
print("Touchless 3D tracking with color mapping")
print("\nUsage : "+sys.argv[0]+" [OPTIONS]")
print("\nTrack the position of your hand in 3D and map it " +
"in RGB space")
print("\nOptions :")
print("\t-h (--help) \t display this help message")
print("\t-s (--serial=) \t change serial port (default is " +
"/dev/tty/ACM0")
sys.exit(0)
elif opt in ("-s", "--serial"):
serial_port = arg
except getopt.GetoptError:
pass
# Handle serial opening
# *** TODO ***
try:
ser = serial.Serial(serial_port, serial_speed)
except Exception as e:
sys.exit("Invalid serial port options : "+str(e))
try:
ser.open()
except Exception as e:
sys.exit("Error while opening serial port : "+str(e))
if not ser.isOpen():
sys.exit("Serial port not opened")
ser.flushInput()
ser.flushOutput()
# Handle calibration
# *** TODO ***
for i in len(notes):
print("Calibrating note "+notes[i])
print("Touch and release the key now... Will get 10k samples.")
for sample in range(10000):
measures[sample] = read_serial(ser)
measures.sort()
max_diff = 0
max_diff_index = 0
for sample in range(10000-1):
if measures[sample+1]-measures[sample]:
max_diff = measures[sample+1]-measures[sample]
max_diff_index = sample
thresholds[i] = (measures[max_diff_index+1]+measures[max_diff_index])/2
# Main loop
# *** TODO *** : Add fetch from the serial also
print("Running... Press q to quit.")
running = True
while running:
serial_input = read_serial(ser)
for note in range(notes):
char = getch()
if char == "q":
print("Exiting...")
running = False
continue
elif char == "a":
frequency = 440
elif char == "b":
frequency = 493.88
elif char == "c":
frequency = 523.25
elif char == "d":
frequency = 587.33
elif char == "e":
frequency = 659.26
elif char == "f":
frequency = 698.46
elif char == "g":
frequency = 783.99
if char == "a":
frequency = frequencies[0]
if char == "b":
frequency = frequencies[1]
if char == "c":
frequency = frequencies[2]
if char == "d":
frequency = frequencies[3]
if char == "e":
frequency = frequencies[4]
if char == "f":
frequency = frequencies[5]
if char == "g":
frequency = frequencies[6]
else:
continue
......
try: import tempfile, wave, subprocess, os, signal, struct
except:
print("E: sound4python is unable to import a combination of %s"%
("tempfile, wave, subprocess, os, signal, struct"))
FNULL = open(os.devnull,'w')
def launchWithoutConsole(args,output=False):
"""Launches args windowless and waits until finished"""
startupinfo = None
if( 'STARTUPINFO' in dir(subprocess) ):
startupinfo = subprocess.STARTUPINFO()
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
if( output ):
return subprocess.Popen(args, stdin=subprocess.PIPE,
stdout=subprocess.PIPE,startupinfo=startupinfo, )
else:
return subprocess.Popen(args, stdin=subprocess.PIPE,
stdout=FNULL ,stderr=FNULL, startupinfo=startupinfo)
# sound4python
def sound(itr,samprate=16000,autoscale=True,output=False):
try: import numpy as np; foundNumpy=True;
except:
foundNumpy=False;
#for now, assume 1-D iterable
mult = 1
if( autoscale ):
mult = 32767.0 / max(itr)
#mult = 128.0 / max(itr)
#create file in memory
#with tempfile.SpooledTemporaryFile() as memFile:
memFile = tempfile.SpooledTemporaryFile()
#create wave write objection pointing to memFile
waveWrite = wave.open(memFile,'wb')
waveWrite.setsampwidth(2) # int16 default
waveWrite.setnchannels(1) # mono default
waveWrite.setframerate(samprate) # 8kHz default
wroteFrames=False
# Not satisfied by the numpy implementation, buzzing and so...
#Let's try to create sound from NumPy vector
#if( foundNumpy ):
# if( type(itr)==np.array ):
# if( itr.ndim == 1 or itr.shape.count(1) == itr.ndim - 1 ):
# waveWrite.writeframes( (mult*itr.flatten()).astype(np.int16).tostring() )
# wroteFrames=True
# else: #we have np, but the iterable isn't a vector
# waveWrite.writeframes( (mult*np.array(itr)).astype(np.int16).tostring() )
# wroteFrames=True
#if( not wroteFrames and not foundNumpy ):
#python w/o np doesn't have "short"/"int16", "@h" is "native,aligned short"
# BAD : waveWrite.writeframes( struct.pack(len(itr)*"@h",[int(mult*itm) for itm in itr]) )
writeValues = []
for itm in itr:
packed_value = struct.pack('h', int(mult*itm))
waveWrite.writeframes(packed_value)
wroteFrames=True
if( not wroteFrames ):
print("E: Unable to create sound. Only 1D numpy arrays and numerical lists are supported.")
waveWrite.close()
return None
#configure the file object, memFile, as if it has just been opened for reading
memFile.seek(0)
try:
# getting here means wroteFrames == True
#print("\nAttempting to play a mono audio stream of length")
#print(" %.2f seconds (%.3f thousand samples at sample rate of %.3f kHz)"%
# ( 1.0*len(itr)/samprate , len(itr)/1000. , int(samprate)/1000.) )
p=launchWithoutConsole(['sox','-','-d'])
except:
print("E: Unable to launch sox.")
print("E: Please ensure that sox is installed and on the path.")
print("E: Try 'sox -h' to test sox installation.")
waveWrite.close()
return None
try:
p.communicate(memFile.read())
p.wait()
except:
print("E: Unable to send in-memory wave file to stdin of sox subprocess.")
waveWrite.close()
return None
#os.kill(p.pid,signal.CTRL_C_EVENT)
#end def sound(itr,samprate=8000,autoscale=True)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment