From 154d084933116aa273751480b4faad401a14c85f Mon Sep 17 00:00:00 2001
From: jaseg <git@jaseg.de>
Date: Tue, 26 Dec 2023 15:08:07 +0100
Subject: [PATCH] Add idle message loop

---
 idle_rotation.txt |  12 +++++
 queue.py          | 110 +++++++++++++++++++++++++++++++++++++++-------
 2 files changed, 106 insertions(+), 16 deletions(-)
 create mode 100644 idle_rotation.txt

diff --git a/idle_rotation.txt b/idle_rotation.txt
new file mode 100644
index 0000000..6607e1b
--- /dev/null
+++ b/idle_rotation.txt
@@ -0,0 +1,12 @@
+5:Send your own text at
+15:8seg.jaseg.de
+5:Send your own text at
+15:8seg.jaseg.de
+15:37C3
+15:UNLOCKED
+15:37C3
+15:UNLOCKED
+15:37C3
+15:UNLOCKED
+15:37C3
+15:UNLOCKED
diff --git a/queue.py b/queue.py
index 3a83715..4140cf8 100644
--- a/queue.py
+++ b/queue.py
@@ -4,6 +4,7 @@ import math
 import time
 import sqlite3
 from datetime import datetime
+from pathlib import Path
 
 import serial
 import click
@@ -65,18 +66,83 @@ seg_map = {
         "#": 0xff,
     }
 
+def shuffle_segments(d):
+    bits_in = [7, 6, 5, 4, 3, 2, 1, 0]
+    bits_out = [7, 6, 5, 4, 3, 2, 1, 0]
+    
+    out = 0
+    for bit_in, bit_out in zip(bits_in, bits_out):
+        if d & (1<<bit_in):
+            out |= 1<<bit_out
+    return out
+
+digit_addresses = [
+    (0, 0),  # 0
+    (0, 1),  # 1
+    (0, 2),  # 2
+    (0, 3),  # 3
+    (0, 4),  # 4
+    (0, 5),  # 5
+    (0, 6),  # 6
+    (0, 7),  # 7
+    (1, 8),  # 8
+    (1, 9),  # 9
+    (1, 10), # 10
+    (1, 11), # 11
+    (1, 12), # 12
+    (1, 13), # 13
+    (1, 14), # 14
+    (1, 15), # 15
+    (2, 0),  # 16
+    (2, 1),  # 17
+    (2, 2),  # 18
+    (2, 3),  # 19
+    (2, 4),  # 20
+    (2, 5),  # 21
+    (2, 6),  # 22
+    (2, 7),  # 23
+    (3, 8),  # 24
+    (3, 9),  # 25
+    (3, 10), # 26
+    (3, 11), # 27
+    (3, 12), # 28
+    (3, 13), # 29
+    (3, 14), # 30
+    (3, 15), # 31
+]
+
+def set_display(ser, message, interval, display_width, global_brightness):
+    lines = [ line for line in message.splitlines() if line.strip() ]
+    for i, line in enumerate(lines):
+        line = line.center(display_width)[:display_width]
+        line_mapped = [shuffle_segments(seg_map.get(c, seg_map['#'])) for c in line]
+        print(f'    [Line {i+1}/{len(lines)}] Sending control data.')
+
+        digit_data = [0x00] * 16 * 4
+        for (driver, address), data in zip(digit_addresses, line_mapped):
+            digit_data[driver*16 + address] = data
+
+        formatted = ''.join(f'{global_brightness:1x}{d:02x}' for d in digit_data) + '\n'
+        if ser is not None:
+            ser.write(formatted)
+            ser.flush()
+        time.sleep(interval)
 
 @click.command()
+@click.option('--dry-run/--wet-run')
 @click.option('-b', '--baudrate', type=int, default=9600)
 @click.option('-w', '--display-width', type=int, default=32)
-@click.option('-i', '--message-interval', type=float, default=30)
+@click.option('-m', '--message-interval', type=float, default=30)
 @click.option('-g', '--global-brightness', type=click.IntRange(0, 15), default=15)
+@click.option('-r', '--idle-rotation', type=click.Path(dir_okay=False, path_type=Path), default='idle_rotation.txt')
 @click.argument('database')
-@click.argument('port')
-def cli(database, port, baudrate, message_interval, display_width, global_brightness):
+@click.argument('port', default=None, required=False)
+def cli(database, port, baudrate, message_interval, display_width, global_brightness, idle_rotation, dry_run):
     db = sqlite3.connect(database)
-    ser = serial.Serial(port, baudrate)
+    ser = serial.Serial(port, baudrate) if port is not None and not dry_run else None
 
+    idle_index = 0
+    idle_seconds = 0
     discard = 0
     while True:
         count, = db.execute('''
@@ -85,12 +151,32 @@ def cli(database, port, baudrate, message_interval, display_width, global_bright
                 ORDER BY timestamp_received''').fetchone()
 
         if count == 0:
-            print(f'Queue is empty. Sleeping.')
+            print(idle_seconds)
+            if idle_seconds > 0:
+                idle_seconds -= 1
+            else:
+                if idle_rotation.is_file():
+                    idle_messages = idle_rotation.read_text().splitlines()
+                else:
+                    idle_messages = ['10:37C3', '10:UNLOCKED']
+
+                if idle_index >= len(idle_messages):
+                    idle_index = 0
+
+                interval, _, message = idle_messages[idle_index].partition(':')
+                idle_seconds = int(interval)
+                idle_index += 1
+
+                print(f'    [Idle {idle_index}/{len(idle_messages)}] {message}')
+                set_display(ser, message, 0, display_width, global_brightness)
+
             time.sleep(1)
             continue
 
         queue_pressure = max(0, min((math.log10(max(1, count-3)) - 0.3) * message_interval*2/3, message_interval*1/3))
         interval = message_interval - queue_pressure
+        print(f'    Waiting interval of {message_interval:.1f} s with queue pressure {queue_pressure:.1f} from n={count} items for a total of {interval:.1f} s')
+
 
         rowid, message, timestamp, remote_ip, suppress_display = db.execute('''
                 SELECT rowid, message, timestamp_received, remote_ip, suppress_display FROM messages
@@ -114,17 +200,9 @@ def cli(database, port, baudrate, message_interval, display_width, global_bright
             print(f'Queueing message by {remote_ip} received at {timestamp} ({delta//60:.0f}m {delta%60:.0f}s ago):')
             print('   ', repr(message))
 
-            lines = [ line for line in message.splitlines() if line.strip() ]
-            interval = max(5, interval/len(lines))
-            for i, line in enumerate(lines):
-                print(f'    [Line {i+1}/{len(lines)}] Waiting interval of {message_interval:.1f} s with queue pressure {queue_pressure:.1f} from n={count} items for a total of {interval:.1f} s')
-
-                line = line.center(display_width)[:display_width]
-                line_mapped = [seg_map.get(c, seg_map['#']) for c in line]
-                formatted = ''.join(f'{global_brightness:1x}{d:02x}' for d in line_mapped) + '\n'
-                time.sleep(interval)
-                ser.write(formatted)
-                ser.flush()
+            idle_seconds = 0
+            interval = max(5, interval/len(message.splitlines()))
+            set_display(ser, message, interval, display_width, global_brightness)
 
 
 if __name__ == '__main__':