#!/usr/bin/env python

# pfh 6/15/09
# Starting from http://itamarst.org/writings/etech04/twisted_internet-22.html
# Goal: Listen to arduino, save data in engineering units, to CSV, also present
# in twisted.web or similar. Socket server?
#
# Design: For now, have arduino send raw ADC counts over and we'll do the
# calibration - seems smarter. I can then use config file parser to move
# all the constants into a config file.
#
# I really want something like rrdtool-based graphs with different timescales.
# Maybe graphite?
# http://somic.org/2009/05/21/graphite-rabbitmq-integration/
# Or even AMQP for fun?
# Serial code from http://twistedmatrix.com/projects/core/documentation/examples/gpsfix.py

#from twisted.internet import iocpreactor
#iocpreactor.install()

from twisted.protocols.basic import LineReceiver

from twisted.internet import reactor
from twisted.internet.serialport import SerialPort
from twisted.web import server, resource, client

from twisted.python import usage
import logging
import sys
import time

## import stuff from Autobahn
##
from autobahn.websocket import listenWS
from autobahn.wamp import WampServerFactory, WampServerProtocol

print "using Twisted reactor", reactor.__class__
print

## WS-MCU protocol
##
class WsMcuProtocol(WampServerProtocol):

    def onSessionOpen(self):
        ## register topic under which we will publish MCU measurements
        ##
        self.registerForPubSub("http://example.com/mcu#measure1")


lastTemp = 0.0
lastRH = 0.0
lastTimestamp = 0

class THOptions(usage.Options):
    optParameters = [
        ['baudrate', 'b', 9600, 'Serial baudrate'],
        ['port', 'p', 3, 'Serial port to use'],
        ]

class indexPage(resource.Resource):
    isLeaf = True

    def render_GET(self, request):
        ccStr = ' Temp:%f Humidity:%f\n' % (lastTemp, lastRH)
        return ccStr

class Echo(LineReceiver):

    ## need a reference to our WS-MCU gateway factory to dispatch PubSub events
    ##
    def __init__(self, wsMcuFactory):
        self.wsMcuFactory = wsMcuFactory

    def processData(self, data):
        """Convert raw ADC counts into SI units as per datasheets"""
        # Skip bad reads
        if len(data) != 2:
            return

        global lastTemp, lastRH, lastTimestamp

        tempCts = int(data[0])
        rhCts = int(data[1])

        rhVolts = rhCts * 0.0048828125
        # 10mV/degree, 1024 count/5V
        temp = tempCts * 0.48828125
        # RH temp correction is -0.7% per deg C
        rhcf = (-0.7 * (temp - 25.0)) / 100.0

        # Uncorrected humidity
        humidity = (rhVolts * 45.25) - 42.76

        # Add correction factor
        humidity = humidity + (rhcf * humidity)


        lastTemp = tempCts
        lastRH = rhCts
        #lastTemp = temp
        #lastRH = humidity
        # Update screen now and then
        if (time.time() - lastTimestamp) > 20.0:

            ## dispatch Temp/Humidity as a PubSub event to all clients subscribed to
            ## topic http://example.com/mcu#measure1
            ##
            self.wsMcuFactory._dispatchEvent("http://example.com/mcu#measure1", (temp, humidity))

            logging.info('Temp: %f C Relative humidity: %f %%' % (temp, humidity))
            logging.debug('Temp: %f counts: %d RH: %f counts: %d volts: %f' % (temp, tempCts, humidity, rhCts, rhVolts))
            lastTimestamp = time.time()

        return temp, humidity

    def connectionMade(self):
        logging.info('Serial connection made!')

    def lineReceived(self, line):
        try:
            data = line.split()
            logging.debug(data)
            self.processData(data)
        except ValueError:
            logging.error('Unable to parse data %s' % line)
            return

if __name__ == '__main__':
    logging.basicConfig(level=logging.INFO, \
                format='%(asctime)s %(levelname)s [%(funcName)s] %(message)s')

    ## create MCU-WS gateway factory/protocol
    ##
    wsMcuFactory = WampServerFactory("ws://localhost:9000", debugWamp = True)
    wsMcuFactory.protocol = WsMcuProtocol
    listenWS(wsMcuFactory)

    o = THOptions()
    try:
        o.parseOptions()
    except usage.UsageError, errortext:
        logging.error('%s %s' % (sys.argv[0], errortext))
        logging.info('Try %s --help for usage details' % sys.argv[0])
        raise SystemExit, 1

    if o.opts['baudrate']:
        baudrate = int(o.opts['baudrate'])

    port = o.opts['port']

    logging.debug('About to open port %s' % port)
    s = SerialPort(Echo(wsMcuFactory), port, reactor, baudrate=baudrate)

    # HTTP interface
    logging.debug('Setting up webserver on http://localhost:2000/')
    root = indexPage()
    site = server.Site(root)
    reactor.listenTCP(2000, site)

    reactor.run()