implement fake lakeshore 336, 340 and 370 in Python with curses display

site_ansto/instrument/TEST_SICS/fakeTempControl/lakeshore/LS370.py

@@ -0,0 +1,124 @@
+#!/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 Lakeshore370 import Lakeshore370 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()