touchless3dtracking/color.py

101 lines
2.7 KiB
Python
Raw Normal View History

#!/bin/env python
import serial
import pygame
2013-09-30 23:49:24 +02:00
def compute_value(line, minimum, maximum, inv):
line = line.split(" ")
if len(line) < 3:
return False
for i in range(3):
value[i] = int(255*(line[i]-minimum[i])/(maximum[i]-minimum[i]))
if value[i] > 255:
value[i] = 255
elif value[i] < 0:
value[i] = 0
2013-09-30 23:49:24 +02:00
if inv is True:
return 255-value[0], 255-value[1], 255-value[2]
else:
return value[0], value[1], value[2]
ser = serial.Serial("/dev/ttyACM0", 115200)
pygame.init()
size = width, height = 640, 480
screen = pygame.display.set_mode(size)
pygame.display.set_caption("Touchless 3D tracking")
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-09-30 23:49:24 +02:00
print("Press any key to launch the program when calibration is" +
"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():
if event.type == pygame.K_RETURN:
running = False
line = float(ser.readline())
if line < minimum or minimum < 0:
minimumR = line
if line > maximum:
maximumR = 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
# Read line from serial
line = float(ser.readline())
# Compute the value for red
2013-09-30 23:49:24 +02:00
value = compute_value(line, minimum, maximum, False)
value_text = compute_value(line, minimum, maximum, True)
if value is not False:
screen.fill(value)
2013-09-30 23:49:24 +02:00
# Display info in window
label = font.render("Running...", 1, value_text)
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))