30 lines
1.5 KiB
Python
30 lines
1.5 KiB
Python
 
|
# Send colours compressed by the JS encoder (tuz.js) and check the fan confirms them.
|
||
|
|
import os, subprocess, sys, time
|
||
|
|
TOOLS = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")
|
||
|
|
sys.path.insert(0, TOOLS)
|
||
|
|
import lianli_poc as p
|
||
|
|
|
||
|
|
def js_tinyuz(raw):
|
||
|
|
js = "const {tuzCompress}=require(process.argv[1]);process.stdout.write(Buffer.from(tuzCompress([...Buffer.from(process.argv[2],'hex')])))"
|
||
|
|
return subprocess.run(["node", "-e", js, TOOLS + "/tuz.js", raw.hex()], capture_output=True, check=True).stdout
|
||
|
|
|
||
|
|
p.tinyuz = js_tinyuz
|
||
|
|
tx, rx = p.Dongle(p.TX_ID, "TX"), p.Dongle(p.RX_ID, "RX")
|
||
|
|
try:
|
||
|
|
master, ch, _ = p.find_master(tx)
|
||
|
|
dev = [d for d in p.get_devices(rx) if d["master"] == master][0]
|
||
|
|
# gradient frame exercises dict matches and literals, then back to purple
|
||
|
|
grad = [(i * 5, 0, 255 - i * 5) for i in range(44)]
|
||
|
|
for name, colors in [("gradient", grad), ("purple", [(128, 0, 255)] * 44)]:
|
||
|
|
up = p.build_upload(colors)
|
||
|
|
# reuse set_color's send path with a custom frame
|
||
|
|
n = (len(up["compressed"]) + p.CHUNK_BYTES - 1) // p.CHUNK_BYTES + 1
|
||
|
|
for i in range(n):
|
||
|
|
pkt = p.rgb_packet(up, dev["mac"], master, i)
|
||
|
|
for _ in range(2 if i == 0 else 1):
|
||
|
|
p.send_rf(tx, ch, dev["rx"], pkt); time.sleep(0.002)
|
||
|
|
ok = p.wait_for_effect(rx, dev["mac"], up["effect"])
|
||
|
|
print(name, len(up["compressed"]), "bytes", "confirmed" if ok else "NOT confirmed")
|
||
|
|
time.sleep(3)
|
||
|
|
finally:
|
||
|
|
tx.close(); rx.close()
|