Bug correction in color.py

Computed value could be outside range [0, 255]. Now, it won't.
This commit is contained in:
Phyks 2013-09-30 22:16:17 +02:00
parent 92d85d46e8
commit 4a8fe2911f
2 changed files with 51 additions and 45 deletions

51
color.py Executable file
View File

@ -0,0 +1,51 @@
#!/bin/env python
import serial
import pygame
ser = serial.Serial("/dev/ttyACM0", 115200)
pygame.init()
size = width, height = 640, 480
screen = pygame.display.set_mode(size)
screen.fill((0, 0, 0))
try:
ser.open()
except Exception:
print("Error while opening serial port.")
if ser.isOpen():
try:
ser.flushInput()
ser.flushOutput()
maximum = -1
minimum = -1
print("Calibration :")
pygame.display.flip()
for i in range(50) :
line = float(ser.readline())
if line < minimum or minimum < 0:
minimum = line
if line > maximum:
maximum = line
print("Go :")
while True:
line = float(ser.readline())
valueR = int(255*(line-minimum)/(maximum-minimum))
if valueR < 0:
valueR = 0
if valueR > 255:
valueR = 255
screen.fill((valueR, 0, 0))
pygame.display.flip()
ser.close()
except Exception:
print("Error while fetching data from serial.")

45
main.py
View File

@ -1,45 +0,0 @@
#!/bin/env python
import serial
import pygame
ser = serial.Serial("/dev/ttyACM0", 115200)
pygame.init()
size = width, height = 640, 480
black = 0, 0, 0
screen = pygame.display.set_mode(size)
background = pygame.Surface(screen.get_size())
background = background.convert()
background.fill(black)
try:
ser.open()
except Exception, e:
print("Error while opening serial port : "+str(e))
if ser.isOpen():
ser.flushInput()
ser.flushOutput()
maximum = -1.
minimum = -1.
print("Calibration")
screen.blit(background, (0,0))
pygame.display.flip()
for i in range(50) :
line = float(ser.readline())
if line < minimum or minimum < 0:
minimum = line
if line > maximum:
maximum = line
print("Go")
while True:
line = float(ser.readline())
background.fill((int(255.*(line-minimum)/(maximum-minimum)), 0, 0))
screen.blit(background, (0,0))
pygame.display.flip()
ser.close()