116 lines
3.5 KiB
Python
Executable File
116 lines
3.5 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.
|
|
|
|
# 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
|
|
|
|
|
|
def play_wave(frequency=440, nb_secs=1):
|
|
sine_wave = []
|
|
for i in range(nb_secs*framerate):
|
|
sine_wave.append(int(16384*math.sin(2*math.pi*frequency*i/framerate)))
|
|
|
|
sound(sine_wave)
|
|
|
|
serial_port = "/dev/ttyACM0"
|
|
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:], "hs:", ["help", "serial="])
|
|
|
|
for opt, arg in opts:
|
|
if opt in ("-h", "--help"):
|
|
print("Play music with chorizos !")
|
|
print("\nUsage : "+sys.argv[0]+" [OPTIONS]")
|
|
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
|
|
|
|
try:
|
|
ser = serial.Serial(serial_port, 115200)
|
|
except Exception as e:
|
|
sys.exit("Error with serial port :"+str(e))
|
|
|
|
framerate = 16000
|
|
frequencies = [440, 600, 880]
|
|
gap_times = 10
|
|
p = []
|
|
for i in range(3):
|
|
processes.append(Process(target=play_wave, args=(frequencies[i], 1)))
|
|
processes[-1].start()
|
|
|
|
try:
|
|
ser.open()
|
|
except Exception as e:
|
|
sys.exit("Error while opening serial port : "+str(e))
|
|
|
|
if ser.isOpen():
|
|
try:
|
|
ser.flushInput()
|
|
ser.flushOutput()
|
|
|
|
measures = [[], [], []]
|
|
|
|
print("Calibration :")
|
|
print("Touch each chorizo electrodes before starting")
|
|
|
|
for i in range(10000):
|
|
line = ser.readline()
|
|
line = line.decode().strip("\r\n")
|
|
line = line.split(" ")
|
|
line = [int(j or 0) for j in line]
|
|
|
|
for j in range(3):
|
|
measures[j][i] = line[j]
|
|
|
|
|
|
# Get threshold
|
|
threshold = [-1, -1, -1]
|
|
for i in range(3):
|
|
gap = -1
|
|
for j in range(1, 10000):
|
|
if measures[i][j] - measures[i][j-1] > gap:
|
|
gap = measures[i][j] - measures[i][j-1]
|
|
threshold[i] = (measures[i][j]+measures[i][j-1])/2
|
|
|
|
print("Running...")
|
|
running = True
|
|
while running:
|
|
# Read line from serial
|
|
line = ser.readline()
|
|
line = line.decode().strip("\r\n")
|
|
line = line.split(" ")
|
|
line = [int(j or 0) for j in line]
|
|
|
|
for i in range(3):
|
|
if line[i] > threshold[i]:
|
|
print("Playing")
|
|
p[i].start()
|
|
|
|
p[i].join()
|
|
except Exception as e:
|
|
sys.exit("Error while handling data from serial : "+str(e))
|