pyro/YAFI
1
0
Fork 0
mirror of https://github.com/Steve-Tech/YAFI.git synced 2026-04-19 16:50:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make LEDs page interactive

Browse Source
This commit is contained in:
Stephen Horvath
2025-03-19 09:56:53 +10:00
parent 5c1bf1c630
commit 17307fc7d5
4 changed files with 259 additions and 229 deletions
Download Patch File Download Diff File Expand all files Collapse all files

129
yafi/yafi.py
View File

@@ -1,17 +1,19 @@
import sys
import os
import gi
gi.require_version('Gtk', '4.0')
gi.require_version('Adw', '1')
gi.require_version("Gtk", "4.0")
gi.require_version("Adw", "1")
from gi.repository import Gtk, Adw, GLib
from cros_ec_python import get_cros_ec
import cros_ec_python.commands as ec_commands
class YAFI(Adw.Application):
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.connect('activate', self.on_activate)
self.connect("activate", self.on_activate)
self.script_dir = os.path.dirname(os.path.abspath(__file__))
self.cros_ec = get_cros_ec()
@@ -57,14 +59,16 @@ class YAFI(Adw.Application):
fan_set_rpm.set_visible(False)
fan_set_percent.set_visible(False)
ec_commands.thermal.thermal_auto_fan_ctrl(self.cros_ec)
case 1: # Percent
case 1: # Percent
fan_set_rpm.set_visible(False)
fan_set_percent.set_visible(True)
case 2: # RPM
case 2: # RPM
fan_set_rpm.set_visible(True)
fan_set_percent.set_visible(False)
fan_mode.connect("notify::selected", lambda combo, _: handle_fan_mode(combo.get_selected()))
fan_mode.connect(
"notify::selected", lambda combo, _: handle_fan_mode(combo.get_selected())
)
def handle_fan_percent(scale):
percent = int(scale.get_value())
@@ -92,7 +96,9 @@ class YAFI(Adw.Application):
self._update_thermals(fan_rpm, temp_items, fan_set_rpm)
# Schedule _update_thermals to run every second
GLib.timeout_add_seconds(1, self._update_thermals, fan_rpm, temp_items, fan_set_rpm)
GLib.timeout_add_seconds(
1, self._update_thermals, fan_rpm, temp_items, fan_set_rpm
)
def _leds_page(self, builder):
# Load the leds.ui file
@@ -104,6 +110,97 @@ class YAFI(Adw.Application):
self._change_page(builder, leds_root)
# Power LED
led_pwr = leds_builder.get_object("led-pwr")
led_pwr_scale = leds_builder.get_object("led-pwr-scale")
def handle_led_pwr(scale):
value = int(abs(scale.get_value() - 2))
ec_commands.framework_laptop.set_fp_led_level(self.cros_ec, value)
led_pwr.set_subtitle(["High", "Medium", "Low"][value])
current_fp_level = ec_commands.framework_laptop.get_fp_led_level(
self.cros_ec
).value
led_pwr_scale.set_value(abs(current_fp_level - 2))
led_pwr.set_subtitle(["High", "Medium", "Low"][current_fp_level])
led_pwr_scale.connect("value-changed", handle_led_pwr)
# Keyboard backlight
led_kbd = leds_builder.get_object("led-kbd")
led_kbd_scale = leds_builder.get_object("led-kbd-scale")
def handle_led_kbd(scale):
value = int(scale.get_value())
ec_commands.pwm.pwm_set_keyboard_backlight(self.cros_ec, value)
led_kbd.set_subtitle(f"{value} %")
current_kb_level = ec_commands.pwm.pwm_get_keyboard_backlight(self.cros_ec)[
"percent"
]
led_kbd_scale.set_value(current_kb_level)
led_kbd.set_subtitle(f"{current_kb_level} %")
led_kbd_scale.connect("value-changed", handle_led_kbd)
# Advanced: Power LED
led_pwr_colour = leds_builder.get_object("led-pwr-colour")
led_pwr_colour_strings = led_pwr_colour.get_model()
all_colours = ["Red", "Green", "Blue", "Yellow", "White", "Amber"]
def add_colours(strings, led_id):
supported_colours = ec_commands.leds.led_control_get_max_values(
self.cros_ec, led_id
)
for i, colour in enumerate(all_colours):
if supported_colours[i]:
strings.append(colour)
add_colours(led_pwr_colour_strings, ec_commands.leds.EcLedId.EC_LED_ID_POWER_LED)
def handle_led_colour(combobox, led_id):
colour = combobox.get_selected() - 2
match colour:
case -2: # Auto
ec_commands.leds.led_control_set_auto(self.cros_ec, led_id)
case -1: # Off
ec_commands.leds.led_control(
self.cros_ec,
led_id,
0,
[0] * ec_commands.leds.EcLedColors.EC_LED_COLOR_COUNT.value,
)
case _: # Colour
colour_idx = all_colours.index(
combobox.get_selected_item().get_string()
)
ec_commands.leds.led_control_set_color(
self.cros_ec,
led_id,
100,
ec_commands.leds.EcLedColors(colour_idx),
)
led_pwr_colour.connect(
"notify::selected",
lambda combo, _: handle_led_colour(
combo, ec_commands.leds.EcLedId.EC_LED_ID_POWER_LED
),
)
# Advanced: Charging LED
led_charge_colour = leds_builder.get_object("led-chg-colour")
led_charge_colour_strings = led_charge_colour.get_model()
add_colours(led_charge_colour_strings, ec_commands.leds.EcLedId.EC_LED_ID_BATTERY_LED)
led_charge_colour.connect(
"notify::selected",
lambda combo, _: handle_led_colour(
combo, ec_commands.leds.EcLedId.EC_LED_ID_BATTERY_LED
),
)
def _battery_page(self, builder):
# Load the battery.ui file
battery_builder = Gtk.Builder()
@@ -135,8 +232,11 @@ class YAFI(Adw.Application):
# Reset the selection in the navbar
navbar = app_builder.get_object("navbar")
about.connect("close-request", lambda _: navbar.select_row(navbar.get_row_at_index(self.current_page)))
about.connect(
"close-request",
lambda _: navbar.select_row(navbar.get_row_at_index(self.current_page)),
)
about.present()
def on_activate(self, app):
@@ -147,7 +247,13 @@ class YAFI(Adw.Application):
self.current_page = 0
self._thermals_page(builder)
pages = (("Thermals", self._thermals_page), ("LEDs", self._leds_page), ("Battery", self._battery_page), ("Hardware", self._hardware_page), ("About", self._about_page))
pages = (
("Thermals", self._thermals_page),
("LEDs", self._leds_page),
("Battery", self._battery_page),
("Hardware", self._hardware_page),
("About", self._about_page),
)
# Build the navbar
navbar = builder.get_object("navbar")
@@ -161,7 +267,7 @@ class YAFI(Adw.Application):
if page != len(pages) - 1:
self.current_page = page
pages[page][1](builder)
navbar.connect("row-activated", lambda box, row: switch_page(row.get_index()))
# Obtain and show the main window
@@ -169,5 +275,6 @@ class YAFI(Adw.Application):
self.win.set_application(self)
self.win.present()
app = YAFI(application_id="au.stevetech.yafi")
app.run(sys.argv)