#!/usr/bin/env python
# vim: ts=8 sts=4 sw=4 expandtab
# Author: Douglas Clowes (dcl@ansto.gov.au) 2013-06-03

from twisted.internet.task import LoopingCall
from twisted.internet import reactor
from twisted.python import log, usage

from Lakeshore340 import Lakeshore340 as MYBASE
from LakeshoreFactory import LakeshoreFactory
from LakeshoreProtocol import LakeshoreProtocol
import sys

from displayscreen import Screen

class MyOptions(usage.Options):
    optFlags = [
            ["window", "w", "Create a display window"],
            ]
    optParameters = [
            ["logfile", "l", None, "output logfile name"],
            ["port", "p", None, "port number to listen on"],
            ]
    def __init__(self):
        usage.Options.__init__(self)
        self['files'] = []

    def parseArgs(self, *args):
        for arg in args:
            self['files'].append(arg)

class MyScreen(Screen):
    def __init__(self, stdscr):
        Screen.__init__(self, stdscr)

    def sendLine(self, txt):
        global myDev
        myDev.protocol = self
        myDev.dataReceived(txt)

    def write(self, txt):
        try:
            newLine = self.lines[-1] + " => " + txt
            del self.lines[-1]
            self.addLine(newLine)
        except:
            pass

class MYDEV(MYBASE):
    def __init__(self):
        MYBASE.__init__(self)
        print MYDEV.__name__, "ctor"

def device_display():
    global stdscr, myDev, myOpts, myPort
    try:
        myDev.doIteration();
    except:
        raise

    if not myOpts["window"]:
        return

    try:
        stdscr.addstr(0, 25, "Identity   : %s (%d)" % (myDev.IDN, myPort))
        stdscr.addstr(1, 25, "Temperature: %8.3f" % myDev.KRDG[1])
        stdscr.addstr(2, 25, "Setpoint   : %8.3f" % myDev.SETP[1])
        stdscr.addstr(3, 25, "Difference : %8.3f" % (myDev.KRDG[1] - myDev.SETP[1]))
        stdscr.addstr(5, 25, "Target     : %8.3f" % myDev.TARGET[1])
        stdscr.addstr(6, 25, "Ramp Rate  : %8.3f" % myDev.RAMP_RATE[1])
        stdscr.addstr(7, 25, "Ramp On    : %8s" % ["No", "Yes"][myDev.RAMP_ON[1]])
        stdscr.addstr(8, 25, "Ramping    : %8s" % ["No", "Yes"][myDev.RAMP_ST[1]])
        stdscr.addstr(0, 0, "Random     : %8.3f" % myDev.RANDOM)
        stdscr.refresh()
    except:
        raise

if __name__ == "__main__":
    global stdscr, myDev, myOpts, myPort

    myOpts = MyOptions()
    try:
        myOpts.parseOptions()
    except usage.UsageError, errortext:
        print '%s: %s' % (sys.argv[0], errortext)
        print '%s: Try --help for usage details.' % (sys.argv[0])
        raise SystemExit, 1

    myDev = MYDEV()
    digits = myDev.IDN[10:13]
    default_port = 7000 + int(digits)
    myPort = default_port
    logfile = None

    if myOpts["port"]:
        myPort = int(myOpts["port"])
        if myPort < 1025 or myPort > 65535:
            myPort = default_port

    if myOpts["window"]:
        logfile = "/tmp/Fake_LS%s_%d.log" % (digits, myPort)

    if myOpts["logfile"]:
        logfile = myOpts["logfile"]

    if logfile:
        log.startLogging(open(logfile, "w"))
    else:
        log.startLogging(sys.stdout)
        #log.startLogging(sys.stderr)

    if myOpts["window"]:
        import curses

        stdscr = curses.initscr()
        screen = MyScreen(stdscr)
        # add screen object as a reader to the reactor
        reactor.addReader(screen)

    myFactory = LakeshoreFactory(LakeshoreProtocol, myDev, "\r")
    lc = LoopingCall(device_display)
    lc.start(0.250)
    reactor.listenTCP(myPort, myFactory) # server
    reactor.run()