166 lines
4.7 KiB
Python
Executable File
166 lines
4.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# ============================================================================
|
|
# This script allows you to play music using a chorizo (or whatever food you
|
|
# like) keyboard ! See README for more info and links.
|
|
|
|
# Specially designed for our demo for Christmas 2013. Designed to fully handle
|
|
# 5 notes and play jingle bells.
|
|
|
|
# As all the other scripts in this repository, I release it under a very
|
|
# permissive license. To make a long story short : do whatever you want with
|
|
# this script (but try to have fun :), I don't mind. It would be cool to quote
|
|
# the origin of the script if you reuse it, but you don't have to. I'd like to
|
|
# be noticed of what you did cool with it (if you think it's worth). :)
|
|
# Ah, I almost forgot : If by chance we ever meet and you think this script is
|
|
# worth, you can buy me a soda :)
|
|
#
|
|
# Phyks
|
|
# =============================================================================
|
|
|
|
from sound4python import sound
|
|
from multiprocessing import Process
|
|
import math
|
|
import serial
|
|
import sys
|
|
import getopt
|
|
|
|
|
|
class _Getch:
|
|
"""Gets a single character from standard input. Does not echo to the
|
|
screen."""
|
|
def __call__(self):
|
|
import sys
|
|
import tty
|
|
import termios
|
|
fd = sys.stdin.fileno()
|
|
old_settings = termios.tcgetattr(fd)
|
|
try:
|
|
tty.setraw(sys.stdin.fileno())
|
|
ch = sys.stdin.read(1)
|
|
finally:
|
|
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
|
|
return ch
|
|
|
|
getch = _Getch()
|
|
|
|
|
|
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 = []
|
|
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
|
|
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
|
|
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
|
|
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
|
|
|
|
print("Playing "+char.upper())
|
|
processes.append(Process(target=play_wave,
|
|
args=(frequency,
|
|
math.floor(0.2 * frequency + 1) / frequency)))
|
|
processes[-1].start()
|
|
|
|
for i in processes:
|
|
i.join()
|