def main():
import argparse
parser = argparse.ArgumentParser(description='Animate colors.')
parser.add_argument('device', type=str, default=None, nargs='?',
help='serial device')
parser.add_argument('--num_leds', type=int, default=5,
help='number of leds')
parser.add_argument('--max_intensity', type=chr, default=0xf,
help='maximum intensity')
parser.add_argument('--min_delay', type=float, default=1/20.,
help='minimum delay in seconds')
parser.add_argument('--stepsize', type=float, default=math.pi/256.,
help='sinus stepsize')
parser.add_argument('--speed', type=int, default=115200,
help='serial port baud rate')
parser.add_argument('--timeout', type=int, default=1,
help='serial port timeout')
args = parser.parse_args()
size = max( [ len(arg) for arg in vars(args).keys() ])
for name,val in vars(args).items():
name = name.ljust(size)
if isinstance(val,float):
print "%s: %.3f" % (name,val)
else:
print "%s: %s" % (name,val)
print
port = args.device
strip = Strip()
if not port:
port = strip.findDevice()
if not port:
print "No device found"
sys.exit(1)
print "Opening '%s'" % port
if not strip.connect(port, speed=args.speed, timeout=args.timeout):
sys.exit(1)
retries = 10
while retries > 0:
if strip.pingDevice():
break
retries -= 1
time.sleep(1)
else:
print "Error connecting to device"
sys.exit(1)
strip.updateConfig(True)
if int(strip.config['debug']) == 1:
strip.toggleDebug()
strip.setSize( int(args.num_leds))
print "Starting effect"
r = Rainbow( int( args.num_leds),
int( args.max_intensity),
float(args.stepsize))
min_delay = float(args.min_delay)
prev = time.time()
cur = time.time()
signal.signal(signal.SIGINT, signal_handler)
while doIterate:
try:
s = strip.tryReadSerial()
if len(s):
print "Serial hickup, read %d bytes" % len(s)
time.sleep(1)
strip.pingDevice()
cur = time.time()
diff = cur - prev - min_delay
if diff < 0: time.sleep(-diff)
prev = cur
strip.setState(r.iterate())
except select.error, v:
if v[0] != errno.EINTR:
raise
else:
break
