summaryrefslogtreecommitdiff
path: root/gm_platform/fw/tw_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'gm_platform/fw/tw_test.py')
-rw-r--r--gm_platform/fw/tw_test.py104
1 files changed, 52 insertions, 52 deletions
diff --git a/gm_platform/fw/tw_test.py b/gm_platform/fw/tw_test.py
index a05a44b..cde59f0 100644
--- a/gm_platform/fw/tw_test.py
+++ b/gm_platform/fw/tw_test.py
@@ -74,56 +74,56 @@ if __name__ == '__main__':
while True:
#ser.write(cobs.encode(b'\x01\xff') + b'\0')
data = ser.read_until(b'\0')
- if not data or data[-1] != 0x00:
- #print(f'{time():>7.3f} Timeout: resetting')
- #ser.write(cobs.encode(b'\x01\xff') + b'\0') # reset
- continue
-
- try:
- if len(data) <= 1: # delimiting zero for retransmission
- cur.execute('INSERT INTO errors(run_id, rx_ts, type) VALUES (?, ?, "retransmission")',
- (run_id, int(time()*1000)))
- continue
- crc32, payload = unpack_head('I', cobs.decode(data[:-1]))
- pid, seq, data = unpack_head('xBH', payload)
- ts = time()
-
- # Calculate byte-wise CRC32
- our_crc = zlib.crc32(bytes(b for x in payload for b in (0, 0, 0, x)))
- #log.append((time(), seq, crc32, our_crc, pid, data))
- print(f'{ts:>7.3f} {seq:05d} {crc32:08x} {our_crc:08x} {pid} {hexlify(data).decode()}', end='')
-
- error = False
- suppress_ack = False
- if crc32 != our_crc:
- print(' CRC ERROR', end='')
- suppress_ack = True
- error = True
-
- if last_pid is not None and pid != (last_pid+1)%8:
- print(' PID ERROR', end='')
- error = True
- else:
- last_pid = pid
-
- if not suppress_ack:
- ser.write(ctrl_ack(pid))
- ser.flushOutput()
-
- if not error:
- cur.execute('INSERT INTO measurements VALUES (?, ?, ?, ?)', (run_id, int(ts*1000), seq, data))
- else:
- cur.execute('INSERT INTO errors VALUES (?, ?, "pid", ?, ?, ?, ?, ?, ?)',
- (run_id, int(ts*1000), seq, pid, (last_pid+1)%8, crc32, our_crc, data))
-
- print()
- lines_written += 1
- if lines_written == 80:
- lines_written = 0
- print('\033[2J\033[H', end='')
- db.commit()
-
- except Exception as e:
- print(e, len(data))
- ser.write(ctrl_ack(0)) # FIXME delet this
+ for data in data.split(b'\0')[:-1]: # data always ends on \0 due to read_until, so split off the trailing empty bytes()
+ try:
+ if not data:
+ #print(f'{time():>7.3f} Timeout: resetting')
+ #ser.write(cobs.encode(b'\x01\xff') + b'\0') # reset
+ ser.write(ctrl_ack(0)) # FIXME delet this
+ cur.execute('INSERT INTO errors(run_id, rx_ts, type) VALUES (?, ?, "retransmission")',
+ (run_id, int(time()*1000)))
+ continue
+
+ crc32, payload = unpack_head('I', cobs.decode(data))
+ pid, seq, data = unpack_head('xBH', payload)
+ ts = time()
+
+ # Calculate byte-wise CRC32
+ our_crc = zlib.crc32(bytes(b for x in payload for b in (0, 0, 0, x)))
+ #log.append((time(), seq, crc32, our_crc, pid, data))
+ print(f'{ts:>7.3f} {seq:05d} {crc32:08x} {our_crc:08x} {pid} {hexlify(data).decode()}', end='')
+
+ error = False
+ suppress_ack = False
+ if crc32 != our_crc:
+ print(' CRC ERROR', end='')
+ suppress_ack = True
+ error = True
+
+ if last_pid is not None and pid != (last_pid+1)%8:
+ print(' PID ERROR', end='')
+ error = True
+ else:
+ last_pid = pid
+
+ if not suppress_ack:
+ ser.write(ctrl_ack(pid))
+ ser.flushOutput()
+
+ if not suppress_ack:
+ cur.execute('INSERT INTO measurements VALUES (?, ?, ?, ?)', (run_id, int(ts*1000), seq, data))
+ if error:
+ cur.execute('INSERT INTO errors VALUES (?, ?, "pid", ?, ?, ?, ?, ?, ?)',
+ (run_id, int(ts*1000), seq, pid, (last_pid+1)%8, crc32, our_crc, data))
+
+ print()
+ lines_written += 1
+ if lines_written == 80:
+ lines_written = 0
+ print('\033[2J\033[H', end='')
+ db.commit()
+
+ except Exception as e:
+ print(e, len(data))
+ ser.write(ctrl_ack(0)) # FIXME delet this