Backup RFP Infinity controller state before Resolume changes
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Local Infinity visualizer for the exact Global-2D layer.
|
||||
"""Local Infinity visualizer for the Infinity Global-2D layer.
|
||||
|
||||
The browser does not reimplement WLED effects. This server proxies the master
|
||||
state and renders only the Infinity Global-2D layer for the 6 x 3 x 106 layout.
|
||||
The browser proxies the master state and renders the Infinity Global-2D layer.
|
||||
Effect-mask modes use a synthetic color preview so geometry, masks and timing
|
||||
can be checked without reimplementing the full WLED effect engine.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
@@ -24,28 +25,43 @@ NODE_COUNT = 6
|
||||
ROWS = 3
|
||||
LEDS_PER_PANEL = 106
|
||||
OUTPUT_LABELS = ["UART6", "UART5", "UART4"]
|
||||
MODE_NAMES = ["Off", "Center Pulse", "Checkerd", "Arrow", "Scan", "Snake", "Wave Line"]
|
||||
MODE_NAMES = ["Off", "Center Pulse", "Checkerd", "Arrow", "Scan", "Snake", "Wave Line", "Strobe", "Schlängeln", "Sunburst"]
|
||||
VARIANT_NAMES = ["Expand / Classic / Line", "Reverse / Diagonal / Bands", "Outline / Checkerd", "Outline Reverse"]
|
||||
BLEND_NAMES = ["Replace", "Add", "Multiply Mask", "Palette Tint"]
|
||||
SCHLAENGELN_VARIANT_NAMES = ["Top Left", "Top Right", "Bottom Left", "Bottom Right"]
|
||||
SUNBURST_VARIANT_NAMES = ["Still", "Wobble", "Rotate"]
|
||||
DIRECTION_NAMES = ["Left -> Right", "Right -> Left", "Top -> Bottom", "Bottom -> Top", "Outward", "Inward", "Ping Pong"]
|
||||
BPM_MIN = 20
|
||||
BPM_MAX = 240
|
||||
PANEL_GAP_RATIO = 0.50 # 8 cm gap for roughly 16 cm active panel aperture.
|
||||
|
||||
# Keep in sync with wled00/infinity_sync.cpp. Values are:
|
||||
# rotation quarter-turns clockwise, mirror X, mirror Y.
|
||||
PANEL_TRANSFORMS: list[list[tuple[int, bool, bool]]] = [
|
||||
[(0, False, False), (0, False, False), (0, False, False), (0, False, False), (0, False, False), (0, False, False)],
|
||||
[(0, False, False), (0, False, False), (0, False, False), (0, False, False), (0, False, False), (0, False, False)],
|
||||
[(0, False, False), (0, False, False), (0, False, False), (0, False, False), (0, False, False), (0, False, False)],
|
||||
]
|
||||
|
||||
HTML = r"""<!doctype html>
|
||||
<html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width,initial-scale=1"><title>Infinity Local Visualizer</title>
|
||||
<style>
|
||||
:root{--bg:#0f1115;--panel:#181b22;--line:#303640;--text:#eef2f6;--muted:#9aa5b5;--good:#35d07f;--bad:#ff637d;--warn:#ffd166;--accent:#4da3ff}*{box-sizing:border-box}body{margin:0;background:var(--bg);color:var(--text);font:14px system-ui,-apple-system,BlinkMacSystemFont,"Segoe UI",sans-serif}main{max-width:1320px;margin:0 auto;padding:14px}header{display:flex;align-items:center;justify-content:space-between;gap:12px;margin-bottom:12px}h1{margin:0;font-size:22px}.sub,.muted{color:var(--muted)}button,input{font:inherit;border:1px solid var(--line);background:#11151b;color:var(--text);border-radius:6px;padding:8px 10px}button{cursor:pointer;font-weight:650}.toolbar{display:flex;gap:8px;flex-wrap:wrap;align-items:center}.toolbar input{width:170px}.status{display:grid;grid-template-columns:repeat(6,minmax(0,1fr));gap:8px;margin-bottom:12px}.metric,.panel,.log{background:var(--panel);border:1px solid var(--line);border-radius:8px}.metric{padding:10px;min-width:0}.metric span{display:block;color:var(--muted);font-size:11px;text-transform:uppercase}.metric strong{display:block;margin-top:3px;overflow:hidden;text-overflow:ellipsis;white-space:nowrap}.nodes{display:grid;grid-template-columns:repeat(6,minmax(128px,1fr));gap:12px}.panel{padding:8px;min-width:0}.panel-head{display:flex;justify-content:space-between;gap:8px;margin-bottom:6px;color:var(--muted);font-size:12px}.panel-head strong{color:var(--text)}canvas{width:100%;aspect-ratio:1/1;background:#05070b;border:1px solid #222b35;border-radius:6px;display:block}.log{margin-top:12px;padding:10px;min-height:38px}.ok{color:var(--good)}.bad{color:var(--bad)}.warn{color:var(--warn)}@media(max-width:900px){header{align-items:flex-start;flex-direction:column}.status{grid-template-columns:1fr}.nodes{grid-template-columns:repeat(3,minmax(92px,1fr));gap:8px}.toolbar,.toolbar input{width:100%}button{flex:1 1 auto}}
|
||||
</style></head><body><main>
|
||||
<header><div><h1>Infinity Local Visualizer</h1><div class="sub">Exact Global-2D layer preview. WLED base effects are not simulated.</div></div><div class="toolbar"><input id="master" aria-label="Master IP"><button id="apply">Connect</button><button id="pause">Pause</button><a id="masterLink" href="#" target="_blank"><button type="button">Master UI</button></a></div></header>
|
||||
<header><div><h1>Infinity Local Visualizer</h1><div class="sub">Global-2D preview with synthetic effect colors and panel-orientation calibration.</div></div><div class="toolbar"><input id="master" aria-label="Master IP"><button id="apply">Connect</button><button id="pause">Pause</button><button id="calibrate">Calibrate</button><a id="masterLink" href="#" target="_blank"><button type="button">Master UI</button></a></div></header>
|
||||
<section class="status" id="status"></section><section class="nodes" id="nodes"></section><div class="log" id="log">Starting...</div>
|
||||
<script>
|
||||
const NODE_COUNT=6, ROWS=3, LEDS=106, OUTPUT_LABELS=["UART6","UART5","UART4"];
|
||||
const PANEL_TRANSFORMS=[[[0,0,0],[0,0,0],[0,0,0],[0,0,0],[0,0,0],[0,0,0]],[[0,0,0],[0,0,0],[0,0,0],[0,0,0],[0,0,0],[0,0,0]],[[0,0,0],[0,0,0],[0,0,0],[0,0,0],[0,0,0],[0,0,0]]];
|
||||
let frame=null, paused=false, refreshInFlight=false;const q=id=>document.getElementById(id);const masterIp=()=>q("master").value.trim()||new URLSearchParams(location.search).get("master")||"10.42.0.213";
|
||||
let calibration=new URLSearchParams(location.search).get("cal")==="1";
|
||||
function ensureNodes(){if(q("nodes").children.length)return;q("nodes").innerHTML=Array.from({length:ROWS},(_,row)=>Array.from({length:NODE_COUNT},(_,node)=>`<article class="panel"><div class="panel-head"><strong>ESP${node+1} ${OUTPUT_LABELS[row]}</strong><span id="meta${node}_${row}">106 LEDs</span></div><canvas width="160" height="160" id="c${node}_${row}"></canvas></article>`).join("")).join("")}
|
||||
function drawPanel(canvas, leds){const ctx=canvas.getContext("2d");ctx.clearRect(0,0,160,160);ctx.fillStyle="#05070b";ctx.fillRect(0,0,160,160);ctx.strokeStyle="#1e2732";ctx.lineWidth=2;ctx.strokeRect(23,23,114,114);for(let i=0;i<leds.length;i++){let x=0,y=0;if(i<25){x=26+i*(108/24);y=24}else if(i<52){x=136;y=26+(i-25)*(108/26)}else if(i<79){x=136-(i-52)*(108/26);y=136}else{x=24;y=136-(i-79)*(108/26)}const c=leds[i];ctx.fillStyle=`rgb(${c[0]},${c[1]},${c[2]})`;ctx.beginPath();ctx.arc(x,y,2.1,0,Math.PI*2);ctx.fill()}}
|
||||
function render(){ensureNodes();const s=frame?.scene||{}, spatial=s.spatial||{};for(let row=0;row<ROWS;row++){for(let node=0;node<NODE_COUNT;node++){q(`meta${node}_${row}`).textContent=frame?.node_ips?.[node]||"virtual";drawPanel(q(`c${node}_${row}`),frame?.panels?.[row]?.[node]||Array.from({length:LEDS},()=>[0,0,0]));}}q("status").innerHTML=[["Master",masterIp()],["2D Mode",frame?.mode_name||"Off"],["Variant",frame?.variant_name||"-"],["Direction",frame?.direction_name||"-"],["Blend",frame?.blend_name||"-"],["Layer",frame?.note||"2D layer exact"]].map(([k,v])=>`<div class="metric"><span>${k}</span><strong>${v}</strong></div>`).join("")}
|
||||
async function refresh(){if(paused||refreshInFlight)return;refreshInFlight=true;const ip=masterIp();q("masterLink").href=`http://${ip}/infinity`;try{const response=await fetch(`/api/frame?master=${encodeURIComponent(ip)}`,{cache:"no-store"});if(!response.ok)throw new Error((await response.text()).replace(/[{}\"]/g,""));frame=await response.json();q("log").innerHTML=`<span class="ok">connected</span> ${new Date().toLocaleTimeString()} via ${ip}. <span class="warn">WLED base not simulated; showing exact Infinity 2D layer only.</span>`;render()}catch(error){q("log").innerHTML=`<span class="bad">master offline</span> ${ip} · ${error.message}`}finally{refreshInFlight=false}}
|
||||
q("master").value=masterIp();q("apply").onclick=refresh;q("pause").onclick=()=>{paused=!paused;q("pause").textContent=paused?"Resume":"Pause"};setInterval(refresh,250);refresh();
|
||||
function ledLocal(i){if(i<25)return [(i+.5)/25,0];if(i<52)return [1,(i-25+.5)/27];if(i<79)return [1-((i-52+.5)/27),1];return [0,1-((i-79+.5)/27)]}
|
||||
function transformedLocal(i,row,node){let [x,y]=ledLocal(i);const t=PANEL_TRANSFORMS[row]?.[node]||[0,0,0];if(t[1])x=1-x;if(t[2])y=1-y;for(let r=0;r<(t[0]&3);r++){const ox=x;x=1-y;y=ox}return [x,y]}
|
||||
function ledPos(i,row,node){const [x,y]=transformedLocal(i,row,node);return [24+x*112,24+y*112]}
|
||||
function drawPanel(canvas, leds, row, node, panelInfo){const ctx=canvas.getContext("2d");ctx.clearRect(0,0,160,160);ctx.fillStyle="#05070b";ctx.fillRect(0,0,160,160);ctx.strokeStyle="#1e2732";ctx.lineWidth=2;ctx.strokeRect(23,23,114,114);for(let i=0;i<leds.length;i++){const [x,y]=ledPos(i,row,node);const c=leds[i];ctx.fillStyle=`rgb(${c[0]},${c[1]},${c[2]})`;ctx.beginPath();ctx.arc(x,y,2.1,0,Math.PI*2);ctx.fill()}if(calibration){ctx.fillStyle="#ff4d4d";let [sx,sy]=ledPos(0,row,node);ctx.beginPath();ctx.arc(sx,sy,5,0,Math.PI*2);ctx.fill();ctx.strokeStyle="#35d07f";ctx.lineWidth=3;ctx.beginPath();ctx.moveTo(sx,sy);const [ex,ey]=ledPos(12,row,node);ctx.lineTo(ex,ey);ctx.stroke();ctx.fillStyle="#eef2f6";ctx.font="12px system-ui";ctx.fillText(panelInfo?.label||"",30,46);ctx.fillText(panelInfo?.transform||"",30,62);ctx.strokeStyle="#ffd166";ctx.lineWidth=1;ctx.beginPath();ctx.moveTo(80,18);ctx.lineTo(80,142);ctx.moveTo(18,80);ctx.lineTo(142,80);ctx.stroke()}}
|
||||
function render(){ensureNodes();const s=frame?.scene||{}, spatial=s.spatial||{};for(let row=0;row<ROWS;row++){for(let node=0;node<NODE_COUNT;node++){q(`meta${node}_${row}`).textContent=frame?.node_ips?.[node]||"virtual";drawPanel(q(`c${node}_${row}`),frame?.panels?.[row]?.[node]||Array.from({length:LEDS},()=>[0,0,0]),row,node,frame?.panel_info?.[row]?.[node]);}}q("status").innerHTML=[["Master",masterIp()],["2D Mode",frame?.mode_name||"Off"],["Variant",frame?.variant_name||"-"],["Direction",frame?.direction_name||"-"],["Layer",frame?.note||"2D preview"]].map(([k,v])=>`<div class="metric"><span>${k}</span><strong>${v}</strong></div>`).join("")}
|
||||
async function refresh(){if(paused||refreshInFlight)return;refreshInFlight=true;const ip=masterIp();q("masterLink").href=`http://${ip}/infinity`;try{const response=await fetch(`/api/frame?master=${encodeURIComponent(ip)}`,{cache:"no-store"});if(!response.ok)throw new Error((await response.text()).replace(/[{}\"]/g,""));frame=await response.json();q("log").innerHTML=`<span class="ok">connected</span> ${new Date().toLocaleTimeString()} via ${ip}. <span class="warn">Synthetic effect preview; panel orientation uses shared calibration table.</span>`;render()}catch(error){q("log").innerHTML=`<span class="bad">master offline</span> ${ip} · ${error.message}`}finally{refreshInFlight=false}}
|
||||
q("master").value=masterIp();q("apply").onclick=refresh;q("pause").onclick=()=>{paused=!paused;q("pause").textContent=paused?"Resume":"Pause"};q("calibrate").onclick=()=>{calibration=!calibration;q("calibrate").textContent=calibration?"Hide Cal":"Calibrate";render()};q("calibrate").textContent=calibration?"Hide Cal":"Calibrate";setInterval(refresh,250);refresh();
|
||||
</script></main></body></html>"""
|
||||
|
||||
|
||||
@@ -65,12 +81,169 @@ def speed_to_bpm(speed: int) -> int:
|
||||
return round(BPM_MIN + speed * (BPM_MAX - BPM_MIN) / 255.0)
|
||||
|
||||
|
||||
def spatial_phase(now_us: int, speed: int) -> float:
|
||||
def spatial_beat_position(now_us: int, speed: int) -> float:
|
||||
seconds = (now_us % 60_000_000) / 1_000_000.0
|
||||
cps = speed_to_bpm(speed) / 60.0
|
||||
return seconds * cps
|
||||
|
||||
|
||||
def spatial_step_position(now_us: int, speed: int) -> float:
|
||||
# Two visible animation phases make one measured on/off panel cycle match the BPM value.
|
||||
return spatial_beat_position(now_us, speed) * 2.0
|
||||
|
||||
|
||||
def spatial_beat_index(now_us: int, speed: int) -> int:
|
||||
return int(math.floor(spatial_beat_position(now_us, speed)))
|
||||
|
||||
|
||||
def spatial_step_index(now_us: int, speed: int) -> int:
|
||||
return int(math.floor(spatial_step_position(now_us, speed)))
|
||||
|
||||
|
||||
def spatial_beat_frac(now_us: int, speed: int) -> float:
|
||||
beat = spatial_beat_position(now_us, speed)
|
||||
return beat - math.floor(beat)
|
||||
|
||||
|
||||
def strobe_amount(now_us: int, speed: int, pulse_width: int) -> int:
|
||||
pulse_width = max(0, min(255, int(pulse_width)))
|
||||
duty = 0.01 + (pulse_width / 255.0) * 0.34
|
||||
phase = spatial_beat_position(now_us, speed) * 8.0
|
||||
return 255 if (phase - math.floor(phase)) < duty else 0
|
||||
|
||||
|
||||
def serpentine_panel_index(col: int, row: int) -> int:
|
||||
return row * NODE_COUNT + (NODE_COUNT - 1 - col if row & 1 else col)
|
||||
|
||||
|
||||
def mirrored_serpentine_panel_index(col: int, row: int, variant: int) -> int:
|
||||
if variant in (1, 3):
|
||||
col = NODE_COUNT - 1 - col
|
||||
if variant in (2, 3):
|
||||
row = ROWS - 1 - row
|
||||
return serpentine_panel_index(col, row)
|
||||
|
||||
|
||||
def chain_length_from_size(size: int) -> int:
|
||||
size = max(0, min(255, int(size)))
|
||||
if size <= 64:
|
||||
return max(1, size // 16)
|
||||
return min(18, 4 + ((size - 64) * 14 + 95) // 191)
|
||||
|
||||
|
||||
def snake_length_from_size(size: int) -> int:
|
||||
return min(17, 3 + max(1, max(0, min(255, int(size))) // 64))
|
||||
|
||||
|
||||
def spatial_hash(value: int) -> int:
|
||||
value &= 0xFFFFFFFF
|
||||
value ^= value >> 16
|
||||
value = (value * 0x7FEB352D) & 0xFFFFFFFF
|
||||
value ^= value >> 15
|
||||
value = (value * 0x846CA68B) & 0xFFFFFFFF
|
||||
value ^= value >> 16
|
||||
return value & 0xFFFFFFFF
|
||||
|
||||
|
||||
def grid_panel_index(col: int, row: int) -> int:
|
||||
return row * NODE_COUNT + col
|
||||
|
||||
|
||||
def grid_col(index: int) -> int:
|
||||
return index % NODE_COUNT
|
||||
|
||||
|
||||
def grid_row(index: int) -> int:
|
||||
return index // NODE_COUNT
|
||||
|
||||
|
||||
def snake_body_contains(body: list[int], position: int, skip_tail: int = 0) -> bool:
|
||||
end = max(0, len(body) - skip_tail)
|
||||
return position in body[:end]
|
||||
|
||||
|
||||
def snake_spawn_apple(body: list[int], seed: int, generation: int) -> int:
|
||||
candidate = spatial_hash(seed ^ (generation * 0x9E3779B9)) % (NODE_COUNT * ROWS)
|
||||
for _ in range(NODE_COUNT * ROWS):
|
||||
if not snake_body_contains(body, candidate):
|
||||
return candidate
|
||||
candidate = (candidate + 7) % (NODE_COUNT * ROWS)
|
||||
return body[0]
|
||||
|
||||
|
||||
def snake_neighbors(position: int) -> list[tuple[int, int]]:
|
||||
col, row = grid_col(position), grid_row(position)
|
||||
out: list[tuple[int, int]] = []
|
||||
if col + 1 < NODE_COUNT:
|
||||
out.append((0, grid_panel_index(col + 1, row)))
|
||||
if row + 1 < ROWS:
|
||||
out.append((2, grid_panel_index(col, row + 1)))
|
||||
if col > 0:
|
||||
out.append((1, grid_panel_index(col - 1, row)))
|
||||
if row > 0:
|
||||
out.append((3, grid_panel_index(col, row - 1)))
|
||||
return out
|
||||
|
||||
|
||||
def snake_distance(a: int, b: int) -> int:
|
||||
return abs(grid_col(a) - grid_col(b)) + abs(grid_row(a) - grid_row(b))
|
||||
|
||||
|
||||
def snake_reset(seed: int, epoch: int, target_length: int, generation: int) -> tuple[list[int], int, int]:
|
||||
head = spatial_hash(seed ^ (epoch * 0x45D9F3B)) % (NODE_COUNT * ROWS)
|
||||
body = [head] * min(target_length, 3)
|
||||
generation += 1
|
||||
apple = snake_spawn_apple(body, seed ^ epoch, generation)
|
||||
return body, apple, generation
|
||||
|
||||
|
||||
def snake_state_at(local_step: int, target_length: int, seed: int) -> tuple[list[int], int]:
|
||||
body, apple, generation = snake_reset(seed, 0, target_length, 0)
|
||||
base_order = [0, 2, 1, 3]
|
||||
for step in range(local_step):
|
||||
order_offset = spatial_hash(seed ^ (step * 0x9E3779B1) ^ generation) & 3
|
||||
ordered = base_order[order_offset:] + base_order[:order_offset]
|
||||
by_dir = dict(snake_neighbors(body[0]))
|
||||
best = None
|
||||
best_distance = 999
|
||||
for direction in ordered:
|
||||
if direction not in by_dir:
|
||||
continue
|
||||
nxt = by_dir[direction]
|
||||
will_eat = nxt == apple
|
||||
if snake_body_contains(body, nxt, 0 if will_eat else 1):
|
||||
continue
|
||||
distance = snake_distance(nxt, apple)
|
||||
if distance < best_distance:
|
||||
best_distance = distance
|
||||
best = nxt
|
||||
if best is None:
|
||||
body, apple, generation = snake_reset(seed, step + 1, target_length, generation)
|
||||
continue
|
||||
ate = best == apple
|
||||
new_len = min(target_length, len(body) + 1) if ate else len(body)
|
||||
body = [best] + body[:new_len - 1]
|
||||
if ate:
|
||||
generation += 1
|
||||
apple = snake_spawn_apple(body, seed ^ step, generation)
|
||||
return body, apple
|
||||
|
||||
def triangle_step(step: int, max_index: int) -> int:
|
||||
if max_index <= 0:
|
||||
return 0
|
||||
period = max_index * 2
|
||||
phase = step % period
|
||||
return period - phase if phase > max_index else phase
|
||||
|
||||
|
||||
def schlaengeln_pingpong_position(step: int, offset: int, max_index: int) -> int:
|
||||
if max_index <= 0:
|
||||
return 0
|
||||
period = max_index * 2
|
||||
phase = (step + period - (offset % period)) % period
|
||||
return period - phase if phase > max_index else phase
|
||||
|
||||
|
||||
def panel_led_position(led: int) -> tuple[float, float, int]:
|
||||
if led < 25:
|
||||
return (led + 0.5) / 25.0, 0.0, 0
|
||||
@@ -81,11 +254,75 @@ def panel_led_position(led: int) -> tuple[float, float, int]:
|
||||
return 0.0, 1.0 - ((led - 79 + 0.5) / 27.0), 3
|
||||
|
||||
|
||||
def apply_panel_transform(col: int, row: int, x: float, y: float) -> tuple[float, float]:
|
||||
try:
|
||||
rotation, mirror_x, mirror_y = PANEL_TRANSFORMS[row][col]
|
||||
except IndexError:
|
||||
return x, y
|
||||
if mirror_x:
|
||||
x = 1.0 - x
|
||||
if mirror_y:
|
||||
y = 1.0 - y
|
||||
for _ in range(rotation & 3):
|
||||
x, y = 1.0 - y, x
|
||||
return x, y
|
||||
|
||||
|
||||
def physical_panel_led_position(col: int, row: int, led: int) -> tuple[float, float]:
|
||||
lx, ly, _ = panel_led_position(led)
|
||||
lx, ly = apply_panel_transform(col, row, lx, ly)
|
||||
pitch = 1.0 + PANEL_GAP_RATIO
|
||||
return col * pitch + lx, row * pitch + ly
|
||||
|
||||
|
||||
def apply_sunburst_panel_transform(x: float, y: float) -> tuple[float, float]:
|
||||
# Match firmware: Sunburst gets the observed 90-degree-left correction,
|
||||
# while Scan and all other modes keep the neutral global transform.
|
||||
for _ in range(3):
|
||||
x, y = 1.0 - y, x
|
||||
return x, y
|
||||
|
||||
|
||||
def sunburst_panel_led_position(col: int, row: int, led: int) -> tuple[float, float]:
|
||||
lx, ly, _ = panel_led_position(led)
|
||||
lx, ly = apply_sunburst_panel_transform(lx, ly)
|
||||
pitch = 1.0 + PANEL_GAP_RATIO
|
||||
return col * pitch + lx, row * pitch + ly
|
||||
|
||||
|
||||
def physical_panel_center() -> tuple[float, float]:
|
||||
pitch = 1.0 + PANEL_GAP_RATIO
|
||||
return ((NODE_COUNT - 1) * pitch + 1.0) * 0.5, ((ROWS - 1) * pitch + 1.0) * 0.5
|
||||
|
||||
|
||||
def hsv_to_rgb(hue: float, sat: float = 1.0, val: float = 1.0) -> list[int]:
|
||||
hue = hue % 1.0
|
||||
sector = hue * 6.0
|
||||
i = int(math.floor(sector))
|
||||
f = sector - i
|
||||
p = val * (1.0 - sat)
|
||||
q = val * (1.0 - sat * f)
|
||||
t = val * (1.0 - sat * (1.0 - f))
|
||||
if i == 0:
|
||||
r, g, b = val, t, p
|
||||
elif i == 1:
|
||||
r, g, b = q, val, p
|
||||
elif i == 2:
|
||||
r, g, b = p, val, t
|
||||
elif i == 3:
|
||||
r, g, b = p, q, val
|
||||
elif i == 4:
|
||||
r, g, b = t, p, val
|
||||
else:
|
||||
r, g, b = val, p, q
|
||||
return [clamp_byte(r * 255), clamp_byte(g * 255), clamp_byte(b * 255)]
|
||||
|
||||
|
||||
def center_pulse_amount(col: int, row: int, led: int, now_us: int, speed: int, variant: int) -> int:
|
||||
distance = abs(row - 1.0) + abs(col - 2.5)
|
||||
max_distance = 3.5
|
||||
span = max_distance + 1.0
|
||||
front = (spatial_phase(now_us, speed) * span) % span
|
||||
front = spatial_step_position(now_us, speed) % span
|
||||
if variant in (1, 3):
|
||||
front = max_distance - front
|
||||
amount = 1.0 - smoothstep(0.0, 0.70, abs(distance - front))
|
||||
@@ -107,18 +344,18 @@ def center_pulse_amount(col: int, row: int, led: int, now_us: int, speed: int, v
|
||||
|
||||
def checker_amount(col: int, row: int, led: int, now_us: int, speed: int, variant: int) -> int:
|
||||
parity = (row + col) & 1
|
||||
step = int(math.floor(spatial_phase(now_us, speed)))
|
||||
step = spatial_step_index(now_us, speed)
|
||||
if variant in (1, 2):
|
||||
x, y, _ = panel_led_position(led)
|
||||
slash = variant == 2 and (step & 1)
|
||||
first = y <= (1.0 - x if slash else x)
|
||||
return 255 if ((parity == 0) == first) else 0
|
||||
return 255 if ((((parity + step) & 1) == 0) == first) else 0
|
||||
return 255 if ((parity + step) & 1) == 0 else 0
|
||||
|
||||
|
||||
def wave_line_amount(col: int, row: int, now_us: int, speed: int, direction: int) -> int:
|
||||
triangle = [0, 1, 2, 1]
|
||||
step = int(math.floor(spatial_phase(now_us, speed)))
|
||||
step = spatial_step_index(now_us, speed)
|
||||
if direction in (2, 3):
|
||||
phase = step if direction == 2 else -step
|
||||
target = round(triangle[(row - phase) % 4] * ((NODE_COUNT - 1) / 2.0) / 2.0)
|
||||
@@ -136,7 +373,7 @@ def arrow_amount(col: int, row: int, now_us: int, speed: int, direction: int, si
|
||||
minor = row if horizontal else col
|
||||
gap = max(1, 1 + size // 86) - 1
|
||||
span = 3 + gap
|
||||
movement = int(math.floor(spatial_phase(now_us, speed)))
|
||||
movement = spatial_step_index(now_us, speed)
|
||||
band = 0 if abs(minor - ((minor_count - 1) / 2.0)) <= 0.55 else 1
|
||||
orientation_right = direction in (0, 2, 4)
|
||||
target = 1 if band == 0 else (0 if orientation_right else 2)
|
||||
@@ -150,17 +387,24 @@ def scan_amount(col: int, row: int, led: int, now_us: int, speed: int, size: int
|
||||
radians = math.radians((angle + (90 if vertical else 0)) % 360)
|
||||
vx, vy = math.cos(radians), math.sin(radians)
|
||||
progress = (col + x) * vx + (row + y) * vy
|
||||
min_progress, max_progress = -3.0, 8.0
|
||||
p00 = 0.0
|
||||
p10 = NODE_COUNT * vx
|
||||
p01 = ROWS * vy
|
||||
p11 = p10 + p01
|
||||
min_progress = min(p00, p10, p01, p11)
|
||||
max_progress = max(p00, p10, p01, p11)
|
||||
width = 0.15 + (size / 255.0) * (1.60 if option == 1 else 0.85)
|
||||
travel = (max_progress - min_progress) + width
|
||||
phase = (spatial_phase(now_us, speed) * travel) % travel
|
||||
travel = max_progress - min_progress
|
||||
if travel <= 0.001:
|
||||
return 0
|
||||
phase = spatial_step_position(now_us, speed) % travel
|
||||
if direction == 6:
|
||||
phase = (spatial_phase(now_us, speed) * travel) % (travel * 2.0)
|
||||
phase = spatial_step_position(now_us, speed) % (travel * 2.0)
|
||||
if phase > travel:
|
||||
phase = (travel * 2.0) - phase
|
||||
elif direction in (1, 3):
|
||||
phase = travel - phase
|
||||
center = min_progress + phase
|
||||
center = min_progress + max(0.0, min(travel, phase))
|
||||
if option == 1:
|
||||
period = width * 2.0 + 0.35
|
||||
d = abs(((progress - center + period * 64.0) % period) - period * 0.5)
|
||||
@@ -168,41 +412,111 @@ def scan_amount(col: int, row: int, led: int, now_us: int, speed: int, size: int
|
||||
return clamp_byte((1.0 - smoothstep(width * 0.5, width * 0.5 + 0.55, abs(progress - center))) * 255.0)
|
||||
|
||||
|
||||
def snake_amount(col: int, row: int, now_us: int, speed: int, size: int, seed: int) -> int:
|
||||
def snake_sample(col: int, row: int, now_us: int, speed: int, size: int, seed: int) -> tuple[int, str]:
|
||||
panel_index = grid_panel_index(col, row)
|
||||
target_length = snake_length_from_size(size)
|
||||
body, apple = snake_state_at(spatial_step_index(now_us, speed) % 240, target_length, seed)
|
||||
for offset, position in enumerate(body):
|
||||
if panel_index == position:
|
||||
return 255, "primary"
|
||||
return (255, "secondary") if panel_index == apple else (0, "gradient")
|
||||
|
||||
|
||||
def schlaengeln_sample(col: int, row: int, now_us: int, speed: int, size: int, variant: int, direction: int) -> tuple[int, str]:
|
||||
path_len = NODE_COUNT * ROWS
|
||||
panel_index = row * NODE_COUNT + (NODE_COUNT - 1 - col if row & 1 else col)
|
||||
step = int(math.floor(spatial_phase(now_us, speed)))
|
||||
head = (step + seed % path_len) % path_len
|
||||
length = 3 + max(1, size // 64)
|
||||
panel_index = mirrored_serpentine_panel_index(col, row, variant)
|
||||
length = chain_length_from_size(size)
|
||||
step = spatial_step_index(now_us, speed)
|
||||
for offset in range(length):
|
||||
if panel_index == (head + path_len - (offset % path_len)) % path_len:
|
||||
return max(55, 255 - offset * 38)
|
||||
apple = (seed * 17 + (step // path_len) * 11 + 7) % path_len
|
||||
return 180 if panel_index == apple else 0
|
||||
if direction == 6:
|
||||
pos = schlaengeln_pingpong_position(step, offset, path_len - 1)
|
||||
elif direction == 1:
|
||||
head = (path_len - 1) - (step % path_len)
|
||||
pos = (head + path_len - (offset % path_len)) % path_len
|
||||
else:
|
||||
pos = (step + path_len - (offset % path_len)) % path_len
|
||||
if panel_index == pos:
|
||||
return max(55, 255 - offset * 30), "gradient"
|
||||
return 0, "gradient"
|
||||
|
||||
|
||||
def sunburst_sample(col: int, row: int, led: int, now_us: int, speed: int, variant: int, option: int) -> tuple[int, str, float]:
|
||||
x, y = sunburst_panel_led_position(col, row, led)
|
||||
cx, cy = physical_panel_center()
|
||||
dx, dy = x - cx, y - cy
|
||||
wobble = math.sin(spatial_beat_position(now_us, speed) * math.tau) * 0.18 if variant == 1 else 0.0
|
||||
rotation = spatial_beat_position(now_us, speed) * (math.tau / 24.0) if variant == 2 else 0.0
|
||||
angle = (math.atan2(dy, dx) + wobble - rotation) % math.tau
|
||||
active = math.cos(angle * 12.0) >= 0.0
|
||||
if option == 1:
|
||||
return 255, "primary" if active else "secondary", 0.0
|
||||
if option == 2:
|
||||
hue = (angle / math.tau) + (spatial_beat_position(now_us, speed) * 8.0 / 255.0)
|
||||
return (255, "rainbow", hue) if active else (255, "black", 0.0)
|
||||
return (255, "gradient", 0.0) if active else (255, "black", 0.0)
|
||||
|
||||
|
||||
def blend(primary: list[int], secondary: list[int], amount: int) -> list[int]:
|
||||
return [clamp_byte(secondary[i] + (primary[i] - secondary[i]) * amount / 255.0) for i in range(3)]
|
||||
|
||||
|
||||
def layer_amount(mode: int, col: int, row: int, led: int, now_us: int, spatial: dict[str, Any], scene: dict[str, Any]) -> int:
|
||||
def layer_sample(mode: int, col: int, row: int, led: int, now_us: int, spatial: dict[str, Any], scene: dict[str, Any]) -> tuple[int, str]:
|
||||
speed = int(scene.get("speed", 128))
|
||||
variant = int(spatial.get("variant", 0))
|
||||
direction = int(spatial.get("direction", 0))
|
||||
size = int(spatial.get("size", 64))
|
||||
if mode == 1:
|
||||
return center_pulse_amount(col, row, led, now_us, speed, variant)
|
||||
return center_pulse_amount(col, row, led, now_us, speed, variant), "gradient"
|
||||
if mode == 2:
|
||||
return checker_amount(col, row, led, now_us, speed, variant)
|
||||
return checker_amount(col, row, led, now_us, speed, variant), "gradient"
|
||||
if mode == 3:
|
||||
return arrow_amount(col, row, now_us, speed, direction, size)
|
||||
return arrow_amount(col, row, now_us, speed, direction, size), "gradient"
|
||||
if mode == 4:
|
||||
return scan_amount(col, row, led, now_us, speed, size, int(spatial.get("angle", 0)), int(spatial.get("option", 0)), direction)
|
||||
return scan_amount(col, row, led, now_us, speed, size, int(spatial.get("angle", 0)), int(spatial.get("option", 0)), direction), "gradient"
|
||||
if mode == 5:
|
||||
return snake_amount(col, row, now_us, speed, size, int(scene.get("seed", 1)))
|
||||
return snake_sample(col, row, now_us, speed, size, int(scene.get("seed", 1)))
|
||||
if mode == 6:
|
||||
return wave_line_amount(col, row, now_us, speed, direction)
|
||||
return 0
|
||||
return wave_line_amount(col, row, now_us, speed, direction), "gradient"
|
||||
if mode == 7:
|
||||
return strobe_amount(now_us, speed, size), "gradient"
|
||||
if mode == 8:
|
||||
return schlaengeln_sample(col, row, now_us, speed, size, variant, direction)
|
||||
if mode == 9:
|
||||
amount, role, hue = sunburst_sample(col, row, led, now_us, speed, variant, int(spatial.get("option", 0)))
|
||||
return amount, f"rainbow:{hue}" if role == "rainbow" else role
|
||||
return 0, "gradient"
|
||||
|
||||
|
||||
def synthetic_effect_color(col: int, row: int, led: int, now_us: int, scene: dict[str, Any]) -> list[int]:
|
||||
x, y, _ = panel_led_position(led)
|
||||
effect = int(scene.get("effect", 0))
|
||||
palette = int(scene.get("palette", 0))
|
||||
speed = int(scene.get("speed", 128))
|
||||
phase = spatial_beat_position(now_us, speed)
|
||||
hue = (col * 0.10 + row * 0.17 + x * 0.12 + y * 0.08 + phase * 0.07 + effect * 0.013 + palette * 0.021) % 1.0
|
||||
if effect == 0:
|
||||
return scene.get("primary", [255, 160, 80])[:3]
|
||||
return hsv_to_rgb(hue, 0.82, 1.0)
|
||||
|
||||
|
||||
def sample_color(primary: list[int], secondary: list[int], effect_color: list[int], amount: int, role: str) -> list[int]:
|
||||
if role == "primary":
|
||||
return [clamp_byte(channel * amount / 255.0) for channel in primary]
|
||||
if role == "secondary":
|
||||
return [clamp_byte(channel * amount / 255.0) for channel in secondary]
|
||||
if role == "black":
|
||||
return [0, 0, 0]
|
||||
if role.startswith("rainbow:"):
|
||||
try:
|
||||
hue = float(role.split(":", 1)[1])
|
||||
except (IndexError, ValueError):
|
||||
hue = 0.0
|
||||
return [clamp_byte(channel * amount / 255.0) for channel in hsv_to_rgb(hue)]
|
||||
return [clamp_byte(channel * amount / 255.0) for channel in effect_color]
|
||||
|
||||
|
||||
def is_direct_role(role: str) -> bool:
|
||||
return role in ("primary", "secondary") or role.startswith("rainbow:")
|
||||
|
||||
|
||||
def render_frame(state: dict[str, Any]) -> dict[str, Any]:
|
||||
@@ -213,26 +527,47 @@ def render_frame(state: dict[str, Any]) -> dict[str, Any]:
|
||||
primary = scene.get("primary", [255, 160, 80])[:3]
|
||||
secondary = scene.get("secondary", [0, 32, 255])[:3]
|
||||
now_us = int(time.monotonic() * 1_000_000) + int(scene.get("phase", 0)) * 1000
|
||||
speed = int(scene.get("speed", 128))
|
||||
panels: list[list[list[list[int]]]] = []
|
||||
for row in range(ROWS):
|
||||
row_panels = []
|
||||
for col in range(NODE_COUNT):
|
||||
leds = []
|
||||
for led in range(LEDS_PER_PANEL):
|
||||
amount = layer_amount(mode, col, row, led, now_us, spatial, scene)
|
||||
amount = clamp_byte(amount * strength / 255.0)
|
||||
leds.append(blend(primary, secondary, amount) if amount else [0, 0, 0])
|
||||
amount, role = layer_sample(mode, col, row, led, now_us, spatial, scene)
|
||||
if not is_direct_role(role):
|
||||
amount = clamp_byte(amount * strength / 255.0)
|
||||
effect_color = synthetic_effect_color(col, row, led, now_us, scene)
|
||||
leds.append(sample_color(primary, secondary, effect_color, amount, role) if amount else [0, 0, 0])
|
||||
row_panels.append(leds)
|
||||
panels.append(row_panels)
|
||||
panel_info = [
|
||||
[
|
||||
{
|
||||
"label": f"ESP{col + 1} {OUTPUT_LABELS[row]}",
|
||||
"transform": f"r{PANEL_TRANSFORMS[row][col][0] * 90} mx{int(PANEL_TRANSFORMS[row][col][1])} my{int(PANEL_TRANSFORMS[row][col][2])}",
|
||||
}
|
||||
for col in range(NODE_COUNT)
|
||||
]
|
||||
for row in range(ROWS)
|
||||
]
|
||||
return {
|
||||
"scene": scene,
|
||||
"node_ips": state.get("node_ips", []),
|
||||
"panels": panels,
|
||||
"panel_info": panel_info,
|
||||
"mode_name": MODE_NAMES[mode] if 0 <= mode < len(MODE_NAMES) else "Unknown",
|
||||
"variant_name": VARIANT_NAMES[int(spatial.get("variant", 0))] if int(spatial.get("variant", 0)) < len(VARIANT_NAMES) else "Unknown",
|
||||
"blend_name": BLEND_NAMES[int(spatial.get("blend", 2))] if int(spatial.get("blend", 2)) < len(BLEND_NAMES) else "Unknown",
|
||||
"variant_name": (
|
||||
SCHLAENGELN_VARIANT_NAMES[int(spatial.get("variant", 0))]
|
||||
if mode == 8 and int(spatial.get("variant", 0)) < len(SCHLAENGELN_VARIANT_NAMES)
|
||||
else SUNBURST_VARIANT_NAMES[int(spatial.get("variant", 0))]
|
||||
if mode == 9 and int(spatial.get("variant", 0)) < len(SUNBURST_VARIANT_NAMES)
|
||||
else VARIANT_NAMES[int(spatial.get("variant", 0))]
|
||||
if int(spatial.get("variant", 0)) < len(VARIANT_NAMES)
|
||||
else "Unknown"
|
||||
),
|
||||
"direction_name": DIRECTION_NAMES[int(spatial.get("direction", 0))] if int(spatial.get("direction", 0)) < len(DIRECTION_NAMES) else "Unknown",
|
||||
"note": "2D layer exact; WLED base not simulated",
|
||||
"note": "2D layer with synthetic effect-color preview",
|
||||
}
|
||||
|
||||
|
||||
@@ -347,7 +682,7 @@ def main() -> int:
|
||||
host, port = server.server_address[:2]
|
||||
print(f"Infinity visualizer: http://{host}:{port}/?master={args.master}")
|
||||
print(f"Proxying master: http://{args.master}/json/infinity")
|
||||
print("Rendering exact Infinity Global-2D layer; WLED base effects are not simulated.")
|
||||
print("Rendering Infinity Global-2D layer with synthetic effect colors and panel calibration.")
|
||||
try:
|
||||
server.serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
|
||||
Reference in New Issue
Block a user