Make a fake Hiden XCS device for testing the script context driver

site_ansto/instrument/TEST_SICS/fakeHiden/xcs/HidenXCS.py

@@ -0,0 +1,180 @@
+# vim: ft=python ts=8 sts=4 sw=4 expandtab autoindent smartindent nocindent
+# Fake Hiden Temperature Controller
+#
+# Author: Douglas Clowes 2014
+#
+from HidenDevice import HidenDevice
+import random
+import re
+import os
+import sys
+import time
+
+sys.path.insert(0, os.path.realpath(os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]),"../../../util"))))
+from fopdt import fopdt, fopdt_sink
+from pid import PID
+
+class Loop(fopdt):
+    def __init__(self, the_temp, the_nick):
+        fopdt.__init__(self, the_temp)
+        self.setpoint = the_temp
+        self.sensor = the_temp
+        # P, I, D
+        self.pid = PID(0.05, 0.02, 1.2)
+        self.pid.setPoint(the_temp)
+        # Value, Kp, Tp, Td, Absolute
+        self.AddSource(fopdt_sink(0, 2, 13, 10, True))
+        self.AddSink(fopdt_sink(the_temp, 1, 30, 1, False))
+        self.power = 0
+        self.nick = the_nick
+        self.count = 0
+
+    def Setpoint(self, the_sp):
+        self.setpoint = the_sp
+        self.pid.setPoint(the_sp)
+
+    def doIteration(self):
+        self.pid_delta = self.pid.update(self.pv)
+        self.sources[0].value = self.pid_delta
+        if self.sources[0].value > 100.0:
+            self.sources[0].value = 100.0
+        if self.sources[0].value < 0.0:
+            self.sources[0].value = 0.0
+        self.count += 1
+        self.iterate(self.count)
+        self.sensor = 0.9 * self.sensor + 0.1 * self.setpoint
+
+class HidenXCS(HidenDevice):
+    """Hiden XCS humidity controller object - simulates the device"""
+
+    def __init__(self):
+        HidenDevice.__init__(self)
+        print HidenXCS.__name__, "ctor"
+        self.RANDOM = 0.0
+        self.IDN = "Simulated Hiden XCS"
+        self.CONFIG_LOOPS = [0, 1, 2]
+        self.AnalogIn = {0:0, 1:0, 2:0, 8:0, 9:0, 12:0}
+        self.AnalogOut = {0:0, 1:0, 2:0}
+        self.DigitalOut = {2:0}
+        self.humidity = 0.0
+        self.temperature = 49.22
+        self.Loops = {}
+        self.Loops[0] = Loop(0, "Dry")
+        self.Loops[1] = Loop(0, "Wet")
+        self.Loops[2] = Loop(0, "Special")
+        self.reset_powerup()
+
+    def doCommand(self, command, params):
+        print HidenXCS.__name__, "Command:", command, params
+        return HidenDevice.doCommand(self, command, params)
+
+    def doQuery(self, command, params):
+        print HidenXCS.__name__, "Query:", command, params
+        return HidenDevice.doQuery(self, command, params)
+
+    def reset_powerup(self):
+        print HidenXCS.__name__, "reset_powerup"
+        self.LAST_ITERATION = 0
+
+    def doIteration(self):
+        delta_time = time.time() - self.LAST_ITERATION
+        if delta_time < 1:
+            return
+        #print "DoIteration:", delta_time
+        self.LAST_ITERATION = time.time()
+        self.flow = 0.0
+        for idx in self.CONFIG_LOOPS:
+            if self.DigitalOut[2]:
+                self.Loops[idx].setpoint = 500.0 * float(self.AnalogOut[idx]) / 4096.0
+            else:
+                self.Loops[idx].setpoint = 0.0
+            self.Loops[idx].doIteration()
+            self.AnalogIn[idx] = int(self.Loops[idx].sensor / 500.0 * 65536.0)
+            self.flow += self.Loops[idx].sensor
+        self.temperature = 50.00
+        if self.DigitalOut[2] and (self.AnalogIn[0] + self.AnalogIn[1] + self.AnalogIn[2]) > 0:
+            self.humidity = 100.0
+            self.humidity *= self.AnalogIn[1]
+            self.humidity /= (self.AnalogIn[0] + self.AnalogIn[1] + self.AnalogIn[2]) 
+
+    def doCommandAOUT(self, cmd, args):
+        idx = int(args[0])
+        if idx in self.AnalogOut:
+            self.AnalogOut[idx] = int(args[1])
+        else:
+            print "Unknown index"
+        print "Command: \"AOUT\" in \"" + cmd + ":" + ":".join(args) + "\""
+
+    def doCommandDOUT(self, cmd, args):
+        idx = int(args[0])
+        if idx in self.DigitalOut:
+            self.DigitalOut[idx] = int(args[1])
+        else:
+            print "Unknown index"
+        print "Command: \"DOUT\" in \"" + cmd + ":" + ":".join(args) + "\""
+
+    def doQueryAIN(self, cmd, args):
+        idx = int(args[0])
+        if idx in self.AnalogIn:
+            self.write("AIN = %d" % self.AnalogIn[idx])
+        else:
+            self.write("AIN = %d" % 0)
+        print "Query: \"AIN\" in \"" + cmd + ":" + ":".join(args) + "\""
+
+    def doQueryAOUT(self, cmd, args):
+        idx = int(args[0])
+        if idx in self.AnalogOut:
+            self.write("AOUT = %d" % self.AnalogOut[idx])
+        else:
+            self.write("AOUT = %d" % 0)
+        print "Query: \"AOUT\" in \"" + cmd + ":" + ":".join(args) + "\""
+
+    def doQueryDOUT(self, cmd, args):
+        idx = int(args[0])
+        if idx in self.DigitalOut:
+            self.write("DOUT = %d" % self.DigitalOut[idx])
+        else:
+            self.write("DOUT = %d" % 0)
+        print "Query: \"DOUT\" in \"" + cmd + ":" + ":".join(args) + "\""
+
+    def doQueryALL(self, cmd, args):
+        line = "A "
+        line += "%4.2f" % self.humidity
+        line += ",%4.2f" % self.temperature
+        line += ",5051.5"
+        line += ",%5.2f" % (500.0 * self.AnalogIn[0] / 65536.0)
+        line += ",%5.2f" % (500.0 * self.AnalogIn[1] / 65536.0)
+        line += ",%5.2f" % (500.0 * self.AnalogIn[2] / 65536.0)
+        line += ",%-99.59,1100"
+        self.write(line)
+        print "Query: \"ALL\" in \"" + cmd + ":" + ":".join(args) + "\""
+
+if __name__ == '__main__':
+    from HidenProtocol import HidenProtocol
+
+    class TestFactory:
+        def __init__(self):
+            print self.__class__.__name__, "ctor"
+            self.numProtocols = 0
+        def write(self, data):
+            print "test write:", data,
+        def loseConnection(self):
+            print "test lose connection"
+    test_factory = TestFactory()
+    test_device = HidenXCS()
+    test_protocol = HidenProtocol(test_device, "\n")
+    test_protocol.factory = test_factory
+    test_protocol.transport = test_factory
+    test_device.protocol = test_protocol
+    test_device.protocol.connectionMade()
+    commands = [
+        "!DOUT,2,1",
+        "?DOUT,2",
+        "!AOUT,2,1",
+        "?AOUT,2",
+        "?AIN,2",
+        "?ALL DATA"]
+    for cmd in commands:
+        test_device.protocol.dataReceived(cmd)
+        test_device.protocol.dataReceived(test_protocol.term)
+    test_device.protocol.connectionLost("Dunno")