From 61d3e220625b12ed292f324425b209b7ecb72e8d Mon Sep 17 00:00:00 2001
From: Alex Mikhalev <alexmikhalevalex@gmail.com>
Date: Thu, 31 Jan 2019 20:52:27 -0800
Subject: [PATCH] start working on python test code

---
 e32_client/e32.py  | 136 +++++++++++++++++++++++++++++++++++++++++++++
 main/e32_driver.cc |   2 +-
 2 files changed, 137 insertions(+), 1 deletion(-)
 create mode 100644 e32_client/e32.py

diff --git a/e32_client/e32.py b/e32_client/e32.py
new file mode 100644
index 0000000..6740079
--- /dev/null
+++ b/e32_client/e32.py
@@ -0,0 +1,136 @@
+import serial
+import struct
+
+PARITY_NONE = 0
+PARITY_ODD = 1
+PARITY_EVEN = 2
+
+BAUD_TABLE = {
+  0: 1200,
+  1: 2400,
+  2: 4800,
+  3: 9600,
+  4: 19200,
+  5: 38400,
+  6: 57600,
+  7: 115200,
+}
+
+AIR_DATA_RATE_TABLE = {
+  0: 300,
+  1: 1200,
+  2: 2400,
+  3: 4800,
+  4: 9600,
+  5: 19200,
+  6: 19200,
+  7: 19200,
+}
+
+TX_MODE_TRANSPARENT = 0
+TX_MODE_FIXED = 1
+
+IO_MODE_OPEN_COLLECTOR = 0
+IO_MODE_PUSH_PULL = 1
+
+TX_POWER_TABLE = {
+  0: 30,
+  1: 27,
+  2: 24,
+  3: 21,
+}
+
+def least_gte(value, dic):
+  items = sorted(dic.items(), key = lambda item: item[1])
+  last_k = None
+  for k, v in items:
+    last_k = k
+    if v >= value:
+      break
+  return last_k
+
+class E32_Params:
+  save: bool
+  address: int
+  parity: int
+  baud: int
+  air_data_rate: int
+  channel: int
+  tx_mode: int
+  io_mode: int
+  wake_up_time: int
+  fec_enabled: bool
+  tx_power: int
+
+  def default():
+    p = E32_Params()
+    p.save = True
+    p.address = 0
+    p.parity = PARITY_NONE
+    p.baud = 9600
+    p.air_data_rate = 2400
+    p.channel = 0x17
+    p.tx_mode = TX_MODE_TRANSPARENT
+    p.io_mode = IO_MODE_PUSH_PULL
+    p.wake_up_time = 250
+    p.fec_enabled = True
+    p.tx_power = 30
+    return p
+
+  def unpack(data):
+    p = E32_Params()
+    datab = bytes(data)
+    if datab[0] == 0xC0:
+      p.save = True
+    elif datab[0] == 0xC2:
+      p.save = False
+    else:
+      raise Exception('invalid E32_Params data')
+    p.address = (datab[1] << 8) | datab[2]
+    p.parity = (datab[3] >> 6) & 0b11
+    p.baud = BAUD_TABLE[(datab[3] >> 3) & 0b111]
+    p.air_data_rate = AIR_DATA_RATE_TABLE[datab[3] & 0b111]
+    p.channel = datab[4]
+    p.tx_mode = (datab[5] >> 7) & 1
+    p.io_mode = (datab[5] >> 6) & 1
+    p.wake_up_time = 250 * (((datab[5] >> 3) & 0b111) + 1)
+    p.fec_enabled = (datab[5] & (1 << 2)) != 0
+    p.tx_power = TX_POWER_TABLE[datab[5] & 0b11]
+    return p
+
+  def pack(self):
+    p = self
+    datab = bytearray(range(6))
+    if p.save:
+      datab[0] = 0xC0
+    else:
+      datab[0] = 0xC2
+    datab[1] = (p.address >> 8) & 0xFF
+    datab[2] = (p.address) & 0xFF
+    datab[3] = 0
+    datab[3] |= p.parity << 6
+    datab[3] |= (least_gte(p.baud, BAUD_TABLE) << 3)
+    datab[3] |= (least_gte(p.air_data_rate, AIR_DATA_RATE_TABLE))
+    datab[4] = p.channel
+    datab[5] = 0
+    datab[5] |= p.tx_mode << 7
+    datab[5] |= p.io_mode << 6
+    datab[5] |= (int((p.wake_up_time / 250) - 1) & 0b111) << 3
+    datab[5] |= p.fec_enabled << 2
+    datab[5] |= least_gte(p.tx_power, TX_POWER_TABLE)
+    return datab
+
+
+class E32:
+  ser: serial.Serial
+
+  def __init__(self, serial_port: serial.Serial):
+    self.ser = serial_port
+
+if __name__ == "__main__":
+  p = E32_Params.default()
+  print("params: ", p.__dict__)
+  data = p.pack()
+  print("packed data: ", ', '.join(format(x, '02x') for x in data))
+  p2 = E32_Params.unpack(data)
+  print("unpacked params: ", p2.__dict__)
diff --git a/main/e32_driver.cc b/main/e32_driver.cc
index c1d13b6..e581563 100644
--- a/main/e32_driver.cc
+++ b/main/e32_driver.cc
@@ -157,7 +157,7 @@ esp_err_t E32_Driver::ReadParams(Params& params) {
     case 7: params.air_data_rate = 19200; break;
   }
 
-  params.comm_channel = (param_data[4] * 0x0F);
+  params.comm_channel = param_data[4];
 
   params.tx_mode      = (TxMode)(param_data[5] & (1 << 7));
   params.io_mode      = (IoMode)(param_data[5] & (1 << 6));