105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
|
#!/bin/env python
|
||
|
|
|
||
|
|
# NOTE : This script is *not* working yet...
|
||
|
|
|
||
|
|
# ============================================================================
|
||
|
|
# I had to play "Petit papa Noël" in an original way for some events. So, why
|
||
|
|
# not using these cool electrodes to handle it ? Left hand would control the
|
||
|
|
# volume and right hand, the note which is played.
|
||
|
|
#
|
||
|
|
# Disclaimer : Even with a lot of practice, I'm not sure you will be able to
|
||
|
|
# play a nice with this script :)
|
||
|
|
#
|
||
|
|
# 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
|
||
|
|
# =============================================================================
|
||
|
|
|
||
|
|
import serial
|
||
|
|
import pygame
|
||
|
|
import sys
|
||
|
|
import getopt
|
||
|
|
|
||
|
|
|
||
|
|
serial_port = "/dev/ttyACM0"
|
||
|
|
|
||
|
|
try:
|
||
|
|
opts, args = getopt.getopt(sys.argv[1:], "hs:", ["help", "serial="])
|
||
|
|
|
||
|
|
for opt, arg in opts:
|
||
|
|
if opt == "-h" or opt == "-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
|
||
|
|
|
||
|
|
ser = serial.Serial(serial_port, 115200)
|
||
|
|
|
||
|
|
try:
|
||
|
|
ser.open()
|
||
|
|
except Exception as e:
|
||
|
|
print("Error while opening serial port : "+str(e))
|
||
|
|
|
||
|
|
if ser.isOpen():
|
||
|
|
try:
|
||
|
|
ser.flushInput()
|
||
|
|
ser.flushOutput()
|
||
|
|
|
||
|
|
print("Calibration :")
|
||
|
|
print("Press any key to launch the program when calibration is " +
|
||
|
|
"finished.")
|
||
|
|
running = True
|
||
|
|
while running:
|
||
|
|
for event in pygame.event.get():
|
||
|
|
if event.type == pygame.KEYDOWN:
|
||
|
|
running = False
|
||
|
|
continue
|
||
|
|
|
||
|
|
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] < minimum[i] or minimum[i] < 0:
|
||
|
|
minimum[i] = line[i]
|
||
|
|
if line[i] > maximum[i]:
|
||
|
|
maximum[i] = line[i]
|
||
|
|
|
||
|
|
print(line)
|
||
|
|
|
||
|
|
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]
|
||
|
|
|
||
|
|
# Compute the value for red
|
||
|
|
value[1] = value[0]
|
||
|
|
value[0] = compute_value(line, minimum, maximum)
|
||
|
|
value_bg = [(value[0][i] + value[1][i])/2 for i in range(3)]
|
||
|
|
value_text = [255 - i for i in compute_value(line, minimum,
|
||
|
|
maximum)]
|
||
|
|
|
||
|
|
ser.close()
|
||
|
|
except Exception as e:
|
||
|
|
print("Error while fetching data from serial : "+str(e))
|