mirror of
https://github.com/brmlab/ledbar.git
synced 2025-06-09 13:24:01 +02:00
76 lines
1.8 KiB
Python
Executable file
76 lines
1.8 KiB
Python
Executable file
#!/usr/bin/python
|
|
# vim:et:sw=4:ts=4:sts=4
|
|
|
|
import sys
|
|
import getopt
|
|
import pygame
|
|
|
|
def print_usage():
|
|
print '''\
|
|
USAGE:
|
|
%s [-n number] [-h]
|
|
OPTIONS:
|
|
-n number number of controlled boxes
|
|
-h --help show this help
|
|
''' % sys.argv[0]
|
|
|
|
def read_byte():
|
|
r = sys.stdin.read(1)
|
|
if len(r) == 0: raise EOFError
|
|
return ord(r)
|
|
|
|
def write_byte(b):
|
|
sys.stdout.write(chr(b))
|
|
|
|
def main():
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:], 'n:h', ['help'])
|
|
except getopt.GetOptError:
|
|
print_usage()
|
|
return 1
|
|
if len(args):
|
|
print_usage()
|
|
return 1
|
|
number = 20
|
|
show_help = False
|
|
for k, v in opts:
|
|
if k == '-n':
|
|
if not v.isdigit():
|
|
print_usage()
|
|
return 1
|
|
number = int(v)
|
|
elif k == '-h' or k == '--help': show_help = True
|
|
if show_help:
|
|
print_usage()
|
|
return 0
|
|
|
|
pygame.init()
|
|
screen_size = [800, 600]
|
|
screen = pygame.display.set_mode(screen_size)
|
|
pygame.display.set_caption("ledbar demo viewer")
|
|
offset = 5
|
|
pixel_width = (screen_size[0]-offset) / number
|
|
try:
|
|
exit = False
|
|
while not exit:
|
|
for event in pygame.event.get():
|
|
if event.type == pygame.QUIT:
|
|
exit = True
|
|
continue
|
|
screen.fill([0, 0, 0])
|
|
for i in xrange(number):
|
|
r = read_byte()
|
|
g = read_byte()
|
|
b = read_byte()
|
|
pygame.draw.rect(screen, [r, g, b], [pixel_width*i, 0, pixel_width-offset, pixel_width-offset])
|
|
write_byte(r); write_byte(g); write_byte(b)
|
|
sys.stdout.flush()
|
|
pygame.display.flip()
|
|
except EOFError:
|
|
pass
|
|
finally:
|
|
pygame.quit()
|
|
|
|
return 0
|
|
|
|
sys.exit(main())
|