watch_reports

#!/usr/bin/env python

import os, os.path
import sys
import time
import json
import zmq
import serial
import serial_controller
import wordlists

def auto_device() :
	# TODO work on non-linux (for you, the person who cares! Who isn't me! Have fun!)
	devdir = '/dev'
	look = lambda: [(dev, os.stat(dev)) for dev in [os.path.join(devdir, dev) for dev in os.listdir(devdir) if 'ttyUSB' in dev]]
	devs = look()
	while not devs :
		time.sleep(0.5)
		print 'auto_device waiting for devices...'
		devs = look()

	dev = None
	ts = 0
	for d, st in devs :
		ts_this = max(st.st_mtime, st.st_ctime)
		if ts_this > ts :
			dev = d

	print 'auto selected device %s' % dev
	return dev

def device_selector(arg) :
	if arg == 'auto' :
		return lambda: auto_device()
	return lambda: arg

if __name__ == '__main__' :
	zmq_url = sys.argv[1]
	device_arg = 'auto'
	try :
		device_arg = sys.argv[2]
	except IndexError :
		pass

	device = device_selector(device_arg)

	c = zmq.Context(1)
	s = c.socket(zmq.SUB)
	s.connect(zmq_url)
	s.setsockopt (zmq.SUBSCRIBE, "")

	serial_opener = lambda: serial.Serial(device(), 115200, timeout=1)
	ctl = serial_controller.ChatlightController(serial_opener)

	for n in ctl.lightnames :
		ctl.off(getattr(ctl, n))

	ctl.set_main({})

	while True :
		r, w, x = zmq.core.poll.select([s], [], [], 0.1)
		if r :
			msg = s.recv()
			msg = json.loads(msg)
	
			print msg

			# TODO missing
			if msg['type'] == 'chat_report' :
				n, r, c = msg['chatters'], msg['rate'], msg['category']

				n = min(n, ctl.TOPLIGHTS)
				for i in range(ctl.TOPLIGHTS) :
					if i < n :
						# fix dimming TODO
						ctl.set_parameters(getattr(ctl, 'LN_TOP%d' % i), 1, 12, 1, 0)
					else :
						ctl.off(getattr(ctl, 'LN_TOP%d' % i))
				print 'set up top lights'

				color = "white"

				if c in wordlists.COLORS :
					color = wordlists.COLORS[c]
				
				blink_on = 0

				# ratecap
				if r > 2.0 :
					r = 2.0

				if r > 1/60.0 :
					blink_on = int(1000 * 1/r * 3)

				blink_on = min(5535, blink_on)

				print 'blink_on = %d' % blink_on

				if blink_on > 0 :
					ctl.set_main({color : (1, 4, blink_on, 800)})
				else :
					ctl.set_main(dict())
			elif msg['type'] == 'page' :
				ctl.set_main({"strobe" : (1, 0, 500, 500)})
				print 'set paging settings.'
				time.sleep(10)
				if ctl.revert_main() :
					print 'reverted paging settings'
				else :
					print 'failed to revert paging settings.'
			else :
				print 'unhandled'