Bug correction in color.py
Computed value could be outside range [0, 255]. Now, it won't.
This commit is contained in:
parent
92d85d46e8
commit
4a8fe2911f
51
color.py
Executable file
51
color.py
Executable 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
45
main.py
@ -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()
|
|
Loading…
Reference in New Issue
Block a user