Changeset 159

Timestamp:

Jul 16, 2010, 12:38:38 PM (14 years ago)

Author:

Rick van der Zwet

Message:

Features :-)

File:

• powerbar/serial-npm4000.py (modified) (13 diffs)

powerbar/serial-npm4000.py

@@ -24 +24 @@
 opt_address_code = 0xFFFF
 opt_baudrate = 19200
+opt_delay = 0.0
 
 # Serial connection port status is cached globally to avoid overhead
@@ -29 +30 @@
 port_status_synced = False
 ports_status = None
+ports_ampere = None
+# Segment A, B, C
+grid_status = None
 
 def make_checksum(command):
@@ -195 +199 @@
 def action_port_on(port):
   """ Enable port on device """
-  global ports_status
-  ports_status = None
+  global port_status_synced
+  port_status_synced = False
 
   action_login()
@@ -203 +207 @@
 def action_port_off(port):
   """ Disable port on device """
-  global ports_status
-  ports_status = None
+  global port_status_synced
+  port_status_synced = False
 
   action_login()
@@ -210 +214 @@
 
 def get_ports_status():
-   global ports_status
+   global ports_status, port_status_synced, ports_ampere, grid_status
+   ports_status_synced = False
    # Example port 1 is off
    # d1 28 ff ff fe ff ff
@@ -221 +226 @@
    #[01:38:14] Comm: Refresh
    port_array = [False] * 25
+   ports_ampere = [0] * 25
+   grid_status = [0] * 3
 
    action_login()
@@ -228 +235 @@
       if (status_array & (1 << port)) > 0:
         port_array[port+1] = True
+   # retval[7] is the sensor state?
+   ports_ampere = [0] + retval[8:16] + retval[17:25] + retval[26:34]
+   grid_status = [retval[16], retval[25], retval[34]]
 
    # Update global state
    ports_status = port_array
+   port_status_synced = True
+
    return port_array
+
 
 
 def get_port_status(port):
    """ Get specific port status """
-   global ports_status
-
-   if ports_status:
-     return ports_status[port]
-   else:
-     return get_ports_status()[port]
-
-
+   global ports_status, port_status_synced
+
+   if not port_status_synced:
+     get_ports_status()
+   return ports_status[port]
+
+
+
+def get_port_ampere(port):
+   """ Get specific port ampere usage """
+   global ports_ampere, port_status_synced
+
+   if not port_status_synced:
+     get_ports_status()
+   return ports_ampere[port]
+
+
+
+def get_grid_status():
+   """ Get grid ampere usage """
+   global grid_status, ports_status_synced
+   
+   if not port_status_synced:
+     get_ports_status()
+   return grid_status
   
 
-
-def runTest():
-    print "TEST: Start will all port Offline"
-    doCommand('allPortsOff')
-
-    print "TEST: All On and Off again, in mass execution"
-    doCommand('allPortsOn',0x01)
-    doCommand('allPortsOff')
-
-    print "TEST: Enable and disable ports one by one"
-    for i in range(1,25):
-        print getPortName(i)
-        doCommand('portOn', getPortHex(i))
-        doCommand('portOff', getPortHex(i))
-
-    print "TEST: Send known error"
-    doCommand('knownError')
 
 def bool_to_str(boolean, raw=False):
@@ -280 +293 @@
 Arguments:
   [-h|--help]                   Reading right know
+  [-a|--ampere]                 Display ampere readings
   [-d|--debug]                  Print extra communication output
   [-r|--raw]                    Status(es) is bits like output
-  --serialport=                 Serial Port to connect to [%s]
-  --password=                   Password to use in hex notation [%s]
-  --addresscode=                Internal device number in hex notation [%s]
+  --serialport=<path>           Serial Port to connect to [%s]
+  --password=<hex>              Password to use in hex notation [%s]
+  --addresscode=<hex>           Internal device number in hex notation [%s]
+  --delay=<int>                 Delay used between port operations [%s]
+  --baudrate=<int>              Bautrate used for communication (19200,9600) [%s]
   [-s <port>|--status=<port>]   Current port(s) configuration
   [-t <port>|--toggle=<port>]   Toggle port(s)
@@ -294 +310 @@
   --changeaddresscode=          Change addresscode [TODO: Implement]
   --changetimerdelay=           Change internal timer delay [TODO: Implement]
-  --delay=                      Delay used between port operations [TODO: Implement]
-  --baudrate=                   Bautrate used for communication (19200,9600) [%s]
+
+Note: [TODO: Implement] bit codes are in the source code, feel free to drop me
+an email <info@rickvanderzwet.nl> if you really to need to be in there.
 
 Note: <port> has different notations:
@@ -301 +318 @@
   Actual value of port          A1,..,A8,B1,..,B8,C1,..,C8 
   All ports                     all
-    """ % (msg, sys.argv[0], opt_serial_port, hex(opt_password), hex(opt_address_code), opt_baudrate)
+    """ % (msg, sys.argv[0], opt_serial_port, hex(opt_password), hex(opt_address_code), opt_delay, opt_baudrate)
     if exitcode:
         sys.exit(exitcode)   
 
 def main():
-    global DEBUG, ser, opt_serial_port, opt_password, opt_address_code, opt_baudrate
+    global DEBUG, ser, opt_serial_port, opt_password, opt_address_code, opt_baudrate, opt_delay
     try:
         opts, args = getopt.getopt(sys.argv[1:], 
-            "df:s:t:ro:v", 
-             ["debug", "verbose", "serialport=", "password=", 
+            "adf:s:t:ro:v", 
+             ["ampere", "debug", "delay=", "verbose", "serialport=", "password=", 
              "addresscode=","toggle=","off=", "on=", "status=", "buadrate=", "raw="])
     except getopt.GetoptError, err:
@@ -318 +335 @@
     opt_action = None
     opt_raw = False
+    opt_ampere = False
     for o, a in opts:
         debug("%s : %s" % (o, a))
-        if o in ["-d", "--debug"]:
+        if o in ["-a", "--ampere"]:
+            opt_ampere = True
+        elif o in ["-d", "--debug"]:
             DEBUG = True
+        elif o in ["--delay"]:
+            opt_delay = float(a)
         elif o in ["-h", "--help"]:
             usage("",0)
@@ -368 +390 @@
     debug("Operating on ports " + str(ports))
 
+    # Open serial port
     ser = serial.Serial(opt_serial_port, opt_baudrate, timeout=5)
     debug(serial)
-      
 
     # Status needs real integers, hack
     for port in ports:
       if opt_action == "status":
-        print bool_to_str(get_port_status(port),opt_raw),
+        if opt_raw:
+          print bool_to_str(get_port_status(port),opt_raw),
+        else:
+          ampere_str = ""
+          if opt_ampere:
+            ampere_str = "[%s]" % get_port_ampere(port)
+          print "Port %02i : %s %s" % (port, get_port_status(port), ampere_str)
       elif opt_action == "toggle":
         if get_port_status(port):
@@ -387 +415 @@
       else:
           assert False, "Option '%s' invalid" % opt_action
-
+      # Backoff if we need to be patient
+      time.sleep(opt_delay)
+      sys.stdout.flush()
+
+    # Be nice and close correctly
     ser.close()
+
+    if opt_ampere:
+      if opt_raw:
+        print " ".join(map(str, get_grid_status()))
+      else:
+        print "Grid A: %s" % grid_status[0]
+        print "Grid B: %s" % grid_status[1]
+        print "Grid C: %s" % grid_status[2]
 
 if __name__ == "__main__":