1
0
mirror of https://github.com/pommi/python-itho-wpu.git synced 2024-11-22 14:02:15 +01:00

feat(itho-wpu): rm slave thread + pass result to I2CMaster via a queue + add retry mechanism

Remove the slave thread, set_callback already spawns a thread.

I2CSlave.callback() now puts the result (if valid) in a queue. This item
can be picked up by I2CMaster for further processing.

I2CMaster will wait 0.21s for a result in the queue. 0.21s, because I
always see a response in < 0.2s. If there is no result, it will send the
request again (max 20 retries). This is because I don't get reliable
responses (via pigpiod?). In an earlier commit I mentioned to configure
pigpiod with -s 2 (sample rate 2 microseconds).  This doesn't seem to be
true. For example -s 10 gives the same reliability (and less CPU usage
by pigpiod).
This commit is contained in:
Pim van den Berg 2020-11-05 13:35:57 +01:00
parent e3bc670050
commit d00e948208

View File

@ -6,7 +6,6 @@ import logging
import pigpio import pigpio
import queue import queue
import sys import sys
import threading
import time import time
consolelogformatter = logging.Formatter("%(asctime)-15s %(levelname)s: %(message)s") consolelogformatter = logging.Formatter("%(asctime)-15s %(levelname)s: %(message)s")
@ -33,30 +32,26 @@ def parse_args():
help="Loglevel") help="Loglevel")
parser.add_argument('--master-only', action='store_true', help="Only run I2C master") parser.add_argument('--master-only', action='store_true', help="Only run I2C master")
parser.add_argument('--slave-only', action='store_true', help="Only run I2C slave") parser.add_argument('--slave-only', action='store_true', help="Only run I2C slave")
parser.add_argument('--slave-timeout', nargs='?', type=int, default=2, parser.add_argument('--slave-timeout', nargs='?', type=int, default=60,
help="Slave timeout in seconds.") help="Slave timeout in seconds when --slave-only")
args = parser.parse_args() args = parser.parse_args()
return args return args
class I2CSlave(): class I2CSlave():
def __init__(self, address): def __init__(self, address, queue):
self.address = address self.address = address
self.queue = queue
self.pi = pigpio.pi() self.pi = pigpio.pi()
if not self.pi.connected: if not self.pi.connected:
logger.error("not pi.connected") logger.error("not pi.connected")
return return
def set_callback(self, q, timeout): def set_callback(self):
logger.debug("set_callback()") logger.debug("set_callback()")
e = self.pi.event_callback(pigpio.EVENT_BSC, self.callback) self.event_callback = self.pi.event_callback(pigpio.EVENT_BSC, self.callback)
self.pi.bsc_i2c(self.address) self.pi.bsc_i2c(self.address)
logger.debug(f"Waiting {timeout}s for activity")
time.sleep(timeout)
e.cancel()
self.pi.bsc_i2c(0) # Disable BSC peripheral
self.pi.stop()
def callback(self, id, tick): def callback(self, id, tick):
logger.debug(f"callback({id}, {tick})") logger.debug(f"callback({id}, {tick})")
@ -65,10 +60,11 @@ class I2CSlave():
if b: if b:
logger.debug(f"Received {b} bytes! Status {s}") logger.debug(f"Received {b} bytes! Status {s}")
result = [hex(c) for c in d] result = [hex(c) for c in d]
logger.debug(f"Callback Response: {result}")
if self.is_checksum_valid(result): if self.is_checksum_valid(result):
logger.info(f"Response: {result}") self.queue.put(result)
else: else:
logger.error(f"Received number of bytes was {b}") logger.debug(f"Received number of bytes was {b}")
def is_checksum_valid(self, b): def is_checksum_valid(self, b):
s = 0x80 s = 0x80
@ -78,18 +74,20 @@ class I2CSlave():
if checksum == 256: if checksum == 256:
checksum = 0 checksum = 0
if checksum != int(b[-1], 0): if checksum != int(b[-1], 0):
logger.debug(f"Checksum invalid (0x{checksum:02x})") logger.debug(f"Checksum invalid (0x{checksum:02x} != {b[-1]})")
return False return False
return True return True
def close(self): def close(self):
self.event_callback.cancel()
self.pi.bsc_i2c(0) self.pi.bsc_i2c(0)
self.pi.stop() self.pi.stop()
class I2CMaster: class I2CMaster:
def __init__(self, address, bus): def __init__(self, address, bus, queue):
self.i = i2c_raw.I2CRaw(address=address, bus=bus) self.i = i2c_raw.I2CRaw(address=address, bus=bus)
self.queue = queue
def execute_action(self, action): def execute_action(self, action):
actions = { actions = {
@ -98,8 +96,19 @@ class I2CMaster:
"getdatatype": [0x80, 0xA4, 0x00, 0x04, 0x00, 0x56], "getdatatype": [0x80, 0xA4, 0x00, 0x04, 0x00, 0x56],
"getdatalog": [0x80, 0xA4, 0x01, 0x04, 0x00, 0x55], "getdatalog": [0x80, 0xA4, 0x01, 0x04, 0x00, 0x55],
} }
logger.info(f"Executing action: {action}") result = None
for i in range(0, 20):
logger.debug(f"Executing action: {action}")
self.i.write_i2c_block_data(actions[action]) self.i.write_i2c_block_data(actions[action])
time.sleep(0.21)
logger.debug("Queue size: {}".format(self.queue.qsize()))
if self.queue.qsize() > 0:
result = self.queue.get()
break
if result is None:
logger.error(f"No valid result in 20 requests")
return result
def close(self): def close(self):
self.i.close() self.i.close()
@ -114,12 +123,17 @@ if __name__ == "__main__":
q = queue.Queue() q = queue.Queue()
if not args.master_only: if not args.master_only:
slave = I2CSlave(address=0x40) slave = I2CSlave(address=0x40, queue=q)
slave_thread = threading.Thread(target=slave.set_callback, args=[q, args.slave_timeout]) slave.set_callback()
slave_thread.start() if args.slave_only:
time.sleep(args.slave_timeout)
if not args.slave_only: if not args.slave_only:
master = I2CMaster(address=0x41, bus=1) master = I2CMaster(address=0x41, bus=1, queue=q)
if args.action: if args.action:
master.execute_action(args.action) result = master.execute_action(args.action)
logger.info(f"Response: {result}")
master.close() master.close()
if not args.master_only:
slave.close()