Merge PR #424: add script for n-color RGB Fusion 2.0 color cycling

This commit is contained in:
Peter Eckersley 2022-02-13 05:05:56 -08:00 committed by GitHub
parent 4a3a6e2067
commit fedf566415
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 91 additions and 0 deletions

View File

@ -1,6 +1,7 @@
import sys
collect_ignore = ['setup.py']
collect_ignore.append('extra/contrib/fusion_rgb_cycle.py') # depends on coloraide
if sys.platform != 'linux':
collect_ignore.append('tests/test_smbus.py')

View File

@ -0,0 +1,90 @@
#!/usr/bin/env python
# Copyright (C) 20222022 Peter Eckersley <pde@pde.is>
# SPDX-License-Identifier: GPL-3.0-or-later
import liquidctl.cli as lc
import liquidctl
import coloraide as c
import argparse
import time
import sys
teal = c.Color("#2de544")
bluer = c.Color("#2d8a8d")
deep_blue = c.Color("#001030")
purple = c.Color("#c000c0")
red = c.Color("#d00000")
# set up argparse
parser = argparse.ArgumentParser(
description="Cycle through colors on an Gigabyte RGB Fusion 2.0 device."
)
parser.add_argument(
"--space",
type=str,
default="srgb",
help="Color space to use; see https://facelessuser.github.io/coloraide/colors/",
)
# you can provide 0 or more colors to cycle through
parser.add_argument("colors", nargs="*", help="Colors to cycle through")
parser.add_argument(
"--steps", default=30, type=int, help="Number of steps per color fade"
)
parser.add_argument("--debug", action="store_true", help="Print debug messages")
parser.add_argument(
"--channel",
type=str,
default="led6",
help="Channel to use; see https://github.com/liquidctl/liquidctl/blob/main/docs/gigabyte-rgb-fusion2-guide.md",
)
parser.add_argument(
"--speed", type=float, default=1.0, help="Speed (seconds per color transition)"
)
args = parser.parse_args()
colors = [c.Color("#" + x) for x in args.colors]
if len(colors) < 2:
colors.append(teal)
if len(colors) < 2:
colors.append(bluer)
devs = liquidctl.driver.rgb_fusion2.RgbFusion2.find_supported_devices()
if not devs:
print("No Gigabyte RGB Fusion 2.0 device found.")
sys.exit(1)
elif len(devs) > 1:
print("Multiple Gigabyte RGB Fusion 2.0 devices found, using first one.")
dev = devs[0]
lookup = []
def to_int(color):
string = color.to_string() # "rgb(255 75 0)"
string = string.partition("(")[2][:-1] # "255 75 0"
string = string.split() # ["255", "75", "0"]
return [int(float(x)) for x in string]
for i in range(len(colors) - 1):
gradient = colors[i].interpolate(colors[i + 1], space=args.space)
cols = [to_int(gradient(x / args.steps)) for x in range(args.steps)]
lookup += cols
lookup += reversed(lookup)
if args.debug:
print("Cycling through", lookup)
while True:
# reconnect occasionally, just in case these connections die and we can bring them back
with dev.connect():
for x in range(2 * args.steps):
c = lookup[x]
dev.set_color(channel=args.channel, mode="fixed", colors=[c])
time.sleep(args.speed * 2.0 / args.steps)