touchless3dtracking/color.py

144 lines
4.2 KiB
Python
Raw Normal View History

#!/bin/env python
import serial
import pygame
import sys
import getopt
def compute_value(line, minimum, maximum):
if len(line) < 3:
return False
2013-10-05 17:46:40 +02:00
value = [0, 0, 0]
for i in range(3):
2013-10-05 17:46:40 +02:00
if maximum[i]-minimum[i] != 0:
value[i] = int(255*(line[i]-minimum[i])/(maximum[i]-minimum[i]))
else:
value[i] = 0
if value[i] > 255:
value[i] = 255
elif value[i] < 0:
value[i] = 0
2013-10-05 17:46:40 +02:00
return value
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)
pygame.init()
# Keep old value to determine a mean value
value = [[0, 0, 0], [0, 0, 0]]
font = pygame.font.Font(None, 36)
size = width, height = 640, 480
screen = pygame.display.set_mode(size)
pygame.display.set_caption("Touchless 3D tracking")
2013-10-05 17:46:40 +02:00
screen.fill((0, 0, 0))
try:
ser.open()
except Exception as e:
print("Error while opening serial port : "+str(e))
if ser.isOpen():
try:
ser.flushInput()
ser.flushOutput()
maximum = [-1, -1, -1]
minimum = [-1, -1, -1]
print("Calibration :")
2013-10-05 17:46:40 +02:00
print("Press any key to launch the program when calibration is " +
2013-09-30 23:49:24 +02:00
"finished.")
# Display info in window
label = font.render("Calibration...", 1, (255, 255, 255))
label_pos = label.get_rect()
label_pos.centerx = screen.get_rect().centerx
label_pos.centery = 20
screen.blit(label, label_pos)
pygame.display.flip()
2013-09-30 23:49:24 +02:00
running = True
while running:
for event in pygame.event.get():
2013-10-05 17:46:40 +02:00
if event.type == pygame.KEYDOWN:
2013-09-30 23:49:24 +02:00
running = False
2013-10-05 17:46:40 +02:00
continue
2013-09-30 23:49:24 +02:00
2013-10-05 17:46:40 +02:00
line = ser.readline()
line = line.decode().strip("\r\n")
line = line.split(" ")
line = [int(j or 0) for j in line]
2013-10-05 17:46:40 +02:00
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)
2013-09-30 23:49:24 +02:00
print("Running...")
running = True
while running:
# Quit if window is closed
for event in pygame.event.get():
2013-09-30 23:49:24 +02:00
if event.type == pygame.QUIT or event.type == pygame.KEYDOWN:
pygame.quit()
running = False
2013-10-05 17:46:40 +02:00
continue
# Read line from serial
2013-10-05 17:46:40 +02:00
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)
2013-10-05 17:46:40 +02:00
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)]
if value is not False:
screen.fill((value_bg[0], value_bg[1], value_bg[2]))
2013-09-30 23:49:24 +02:00
# Display info in window
label = font.render("Running...", 1, (value_text[0],
value_text[1], value_text[2]))
2013-09-30 23:49:24 +02:00
label_pos = label.get_rect()
label_pos.centerx = screen.get_rect().centerx
label_pos.centery = 20
screen.blit(label, label_pos)
pygame.display.flip()
ser.close()
except Exception as e:
print("Error while fetching data from serial : "+str(e))