aprl/treppe-gl
Watch
0
0
Fork
You've already forked treppe-gl
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add brightness option and run/dry wrapper scripts

Browse source 
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
This commit is contained in:
April John 2026-10-16 14:38:55 +02:00
commit 0ff391bfa1
1 changed files with 38 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

15
treppe_gl.py
View file

@ -35,23 +35,26 @@ WHITE_MODES = ("extract", "alpha", "off")
# --- Protocol --------------------------------------------------------------
def to_rgbw(pixels, white="extract"):
def to_rgbw(pixels, white="extract", brightness=100.0):
"""
Convert float RGBA pixels (STEPS x 4, 0..1) to RGBW (STEPS x 4).
extract: w = min(r, g, b), subtracted from r, g, b
alpha: the shader's alpha channel is the white value
off: w = 0
brightness (0..100, percent) scales the result linearly.
"""
scale = np.float32(min(max(brightness, 0.0), 100.0) / 100.0)
px = np.clip(np.asarray(pixels, dtype=np.float32), 0.0, 1.0)
rgb = px[:, :3]
if white == "extract":
w = rgb.min(axis=1, keepdims=True)
return np.hstack([rgb - w, w])
return np.hstack([rgb - w, w]) * scale
if white == "alpha":
return np.hstack([rgb, px[:, 3:4]])
return np.hstack([rgb, px[:, 3:4]]) * scale
if white == "off":
return np.hstack([rgb, np.zeros((len(px), 1), dtype=np.float32)])
return np.hstack([rgb, np.zeros((len(px), 1), dtype=np.float32)]) * scale
raise ValueError("unknown white mode: {}".format(white))
@ -380,7 +383,7 @@ def run(args):
last = now
pixels = renderer.render_strip(t, dt, frame)
rgbw = to_rgbw(pixels, args.white)
rgbw = to_rgbw(pixels, args.white, args.brightness)
if sock:
sock.sendto(encode_frame(rgbw), target)
@ -414,6 +417,8 @@ def parse_args(argv=None):
p.add_argument("--port", type=int, default=PORT_DEFAULT)
p.add_argument("--fps", type=float, default=60.0)
p.add_argument("--white", choices=WHITE_MODES, default="extract")
p.add_argument("--brightness", type=float, default=100.0,
help="0..100 percent, scales what is sent (default 100)")
p.add_argument("--dry-run", action="store_true",
help="do not send anything over the network")
p.add_argument("--no-window", dest="window", action="store_false",