control.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import tkinter
import tkinter.messagebox
import subprocess
import customtkinter
import pickle
from math import *
from dataclasses import dataclass

customtkinter.set_appearance_mode("System")  # Modes: "System" (standard), "Dark", "Light"
customtkinter.set_default_color_theme("blue")  # Themes: "blue" (standard), "green", "dark-blue"



@dataclass
class rgb_profile:
    mode: int = 0
    red: float = 0
    green: float = 0
    blue: float = 0

profile_1 = rgb_profile()
print(profile_1.mode)

def clamp(num, min_n, max_n):
    return max(min(num, max_n), min_n)





class App(customtkinter.CTk):

    WIDTH = 780
    HEIGHT = 700

    def __init__(self):
        super().__init__()

        self.title("RGB Control")
        self.geometry(f"{App.WIDTH}x{App.HEIGHT}")

        self.protocol("WM_DELETE_WINDOW", self.on_closing) #call .on_closing() when app gets closed

        # ============ create two frames ============

        # configure grid layout (2x1)
        self.grid_columnconfigure(1, weight=1)
        self.grid_rowconfigure(0, weight=1)

        self.frame_left = customtkinter.CTkFrame(master=self,
                                                 width=180,
                                                 corner_radius=0)
        self.frame_left.grid(row=0, column=0, sticky="nswe")

        self.frame_right = customtkinter.CTkFrame(master=self)
        self.frame_right.grid(row=0, column=1, sticky="nswe", padx=20, pady=20)

        # ============ frame_left ============

        # configure grid layout (1x11)
        self.frame_left.grid_rowconfigure(0, minsize=10)   # empty row with minsize as spacing
        self.frame_left.grid_rowconfigure(7, weight=1)  # empty row as spacing
        self.frame_left.grid_rowconfigure(8, minsize=20)    # empty row with minsize as spacing
        self.frame_left.grid_rowconfigure(11, minsize=10)  # empty row with minsize as spacing

        self.label_1 = customtkinter.CTkLabel(master=self.frame_left,
                                              text="RGB Control",
                                              font=("Roboto Medium", -16))  # font name and size in px
        self.label_1.grid(row=1, column=0, pady=10, padx=10)

        self.button_1 = customtkinter.CTkButton(master=self.frame_left,
                                                text="RGB7",
                                                fg_color=("gray75", "gray30"),  # <- custom tuple-color
                                                command=self.rgb_init_7)
        self.button_1.grid(row=4, column=0, pady=10, padx=20)

        self.button_2 = customtkinter.CTkButton(master=self.frame_left,
                                                text="RGB3",
                                                fg_color=("gray75", "gray30"),  # <- custom tuple-color
                                                command=self.rgb_init_3)
        self.button_2.grid(row=3, column=0, pady=10, padx=20)

        self.button_3 = customtkinter.CTkButton(master=self.frame_left,
                                                text="RGB6",
                                                fg_color=("gray75", "gray30"),  # <- custom tuple-color
                                                command=self.rgb_init_6)
        self.button_3.grid(row=2, column=0, pady=10, padx=20)

        self.button_4 = customtkinter.CTkButton(master=self.frame_left,
                                                text="Autostart",
                                                fg_color=("gray75", "gray30"),  # <- custom tuple-color
                                                command=self.set_autostart)
        self.button_4.grid(row=5, column=0, pady=10, padx=20)

        self.button_5 = customtkinter.CTkButton(master=self.frame_left,
                                                text="rAnDoM",
                                                fg_color=("gray75", "gray30"),  # <- custom tuple-color
                                                command=self.rgb_init_9)

        self.button_5.grid(row=6, column=0, pady=10, padx=20)


        self.switch_2 = customtkinter.CTkSwitch(master=self.frame_left,
                                                text="Dark Mode",
                                                command=self.change_mode)
        self.switch_2.grid(row=10, column=0, pady=10, padx=20, sticky="w")

        # ============ frame_right ============

        # configure grid layout (3x7)
        self.frame_right.rowconfigure((0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10), weight=1)
        self.frame_right.rowconfigure((11), weight=9)
        self.frame_right.columnconfigure(1, weight=2)
        self.frame_right.columnconfigure(5, weight=1)

        self.frame_info = customtkinter.CTkFrame(master=self.frame_right)
        self.frame_info.grid(row=0, column=0, columnspan=3, rowspan=4, pady=20, padx=20, sticky="nsew")

        # ============ frame_info ============

        # configure grid layout (1x1)
        self.frame_info.rowconfigure(0, weight=1)
        self.frame_info.columnconfigure((0, 4), weight=1)

        self.label_preview = customtkinter.CTkLabel(master=self.frame_info,
                                                   text="Preview" ,
                                                   font=('Roboto medium', -16),
                                                   height=300,
                                                   corner_radius=16,
                                                   fg_color='#000000',  # <- custom tuple-color
                                                   justify=tkinter.LEFT,
                                                   bg_color='#000000')

        self.label_preview.grid(column=0, row=0, columnspan=6, rowspan=4, sticky="nwe", padx=15, pady=15)


        # ============ frame_right ============
# SLIDER & LABEL RED
        self.slider_r = customtkinter.CTkSlider(master=self.frame_right,
                                                from_=0,
                                                to=1,
                                                number_of_steps=255,
                                                command=self.update_red,
                                                progress_color='#FF0000',
                                                button_color='#FF0000',
                                                button_hover_color='#FFAAAA')
        self.slider_r.grid(row=4, column=1, columnspan=1, pady=10, padx=20, sticky="we")

        self.label_red_value = customtkinter.CTkEntry(master=self.frame_right,
                                                      font=("Roboto Medium", -16))  # font name and size in px
        self.label_red_value.bind("<KeyRelease>", self.update_slider_r)
        self.label_red_value.grid(row=4, column=2, columnspan=1, pady=10, padx=0)

        self.label_r= customtkinter.CTkLabel(master=self.frame_right,
                                              text='Red',
                                              font=("Roboto Medium", -16))  # font name and size in px

        self.label_r.grid(row=4, column=0, columnspan=1, pady=10, padx=0)


# SLIDER & LABEL GREEN
        self.slider_g = customtkinter.CTkSlider(master=self.frame_right,
                                                from_=0,
                                                to=1,
                                                number_of_steps=255,
                                                command=self.update_green,
                                                progress_color='#00FF00',
                                                button_color='#00FF00',
                                                button_hover_color='#AAFFAA')
        self.slider_g.grid(row=5, column=1, columnspan=1, pady=10, padx=20, sticky="we")


        self.label_green_value = customtkinter.CTkEntry(master=self.frame_right,
                                              font=("Roboto Medium", -16))  # font name and size in px
        self.label_green_value.bind("<KeyRelease>", self.update_slider_g)
        self.label_green_value.grid(row=5, column=2, columnspan=1, pady=10, padx=0)

        self.label_g= customtkinter.CTkLabel(master=self.frame_right,
                                              text='Green',
                                              font=("Roboto Medium", -16))  # font name and size in px

        self.label_g.grid(row=5, column=0, pady=10, padx=0)

# SLIDER & LABEL BLUE
        self.slider_b = customtkinter.CTkSlider(master=self.frame_right,
                                                from_=0,
                                                to=1,
                                                number_of_steps=255,
                                                command=self.update_blue,
                                                progress_color='#0000FF',
                                                button_color='#0000FF',
                                                button_hover_color='#AAAAFF')
        self.slider_b.grid(row=6, column=1, columnspan=1, pady=10, padx=20, sticky="we")


        self.label_blue_value = customtkinter.CTkEntry(master=self.frame_right,
                                                       font=("Roboto Medium", -16))  # font name and size in px
        self.label_blue_value.bind("<KeyRelease>", self.update_slider_b)
        self.label_blue_value.grid(row=6, column=2, columnspan=1, pady=10, padx=0)

        self.label_b= customtkinter.CTkLabel(master=self.frame_right,
                                             text='Blue',
                                             font=("Roboto Medium", -16))  # font name and size in px

        self.label_b.grid(row=6, column=0, pady=10, padx=0)

# SLIDER & LABEL SPEED
        self.slider_s = customtkinter.CTkSlider(master=self.frame_right,
                                                from_=0,
                                                to=1,
                                                number_of_steps=255,
                                                command=self.update_speed,
                                                progress_color='#AAAAAA',
                                                button_color='#AAAAAA',
                                                button_hover_color='#BBBBBB')
        self.slider_s.grid(row=7, column=1, columnspan=1, pady=10, padx=10, sticky="we")


        self.label_speed_value = customtkinter.CTkEntry(master=self.frame_right,
                                                        font=("Roboto Medium", -16)) # font name and size in px
        self.label_speed_value.bind("<KeyRelease>", self.update_slider_s)
        self.label_speed_value.grid(row=7, column=2, columnspan=1, pady=10, padx=0)

        self.label_s = customtkinter.CTkLabel(master=self.frame_right,
                                              text='Speed',
                                              font=("Roboto Medium", -16)) # font name and size in px

        self.label_s.grid(row=7, column=0, pady=10, padx=0)




        self.button_set_colour = customtkinter.CTkButton(master=self.frame_right,
                                                       height=25,
                                                       text="Set colour",
                                                       command=self.set_colour)
        self.button_set_colour.grid(row=8, column=2, columnspan=1, pady=10, padx=20, sticky="we")

        self.set_autostart_button = customtkinter.CTkButton(master=self.frame_right,
                                                       height=25,
                                                       text="Set autostart",
                                                       command=self.set_autostart)
        # self.set_autostart_button.grid(row=8, column=2, columnspan=1, pady=10, padx=20, sticky="we")

        self.load_autostart_button = customtkinter.CTkButton(master=self.frame_right,
                                                       height=25,
                                                       text="Load autostart",
                                                       command=self.load_autostart)
        # self.load_autostart_button.grid(row=9, column=2, columnspan=1, pady=10, padx=20, sticky="we")

        self.entry = customtkinter.CTkEntry(master=self.frame_right,
                                            width=120,
                                            placeholder_text="Save as")
        self.entry.grid(row=12, column=0, columnspan=2, pady=20, padx=20, sticky="we")

        self.button_5 = customtkinter.CTkButton(master=self.frame_right,
                                                text="Save profile",
                                                command=self.save_profile)
        self.button_5.grid(row=12, column=2, columnspan=1, pady=20, padx=20, sticky="we")

        self.slider_button_2 = customtkinter.CTkButton(master=self.frame_right,
                                                       height=25,
                                                       text="Load profile",
                                                       command=self.load_profile)
        self.slider_button_2.grid(row=13, column=2, columnspan=1, pady=10, padx=20, sticky="we")

        self.mode = tkinter.IntVar(value=0)


        # RADIO BUTTONS, MODE SELECT
        self.label_red_valueadio_group = customtkinter.CTkLabel(master=self.frame_right,
                                                        text="Mode select:")
        self.label_red_valueadio_group.grid(row=0, column=3, columnspan=1, pady=20, padx=10, sticky="")

        self.radio_button_0 = customtkinter.CTkRadioButton(master=self.frame_right,
                                                           variable=self.mode,
                                                           value=0,
                                                           text='Static')
        self.radio_button_0.grid(row=1, column=3, pady=5, padx=20, sticky="n")


        self.radio_button_1 = customtkinter.CTkRadioButton(master=self.frame_right,
                                                           variable=self.mode,
                                                           value=1,
                                                           text='Breath')
        self.radio_button_1.grid(row=2, column=3, pady=5, padx=20, sticky="n")

        self.radio_button_2 = customtkinter.CTkRadioButton(master=self.frame_right,
                                                           variable=self.mode,
                                                           value=2,
                                                           text='Neon' )
        self.radio_button_2.grid(row=3, column=3, pady=5, padx=20, sticky="n")

        self.radio_button_3 = customtkinter.CTkRadioButton(master=self.frame_right,
                                                           variable=self.mode,
                                                           value=3,
                                                           text='Wave')
        self.radio_button_3.grid(row=4, column=3, pady=5, padx=20, sticky="n")



        # set default values
        self.radio_button_0.select()
        self.switch_2.select()

        self.slider_r.set(0)
        self.label_red_value.delete(0, 'end')
        self.label_red_value.insert(0, '0')

        self.slider_g.set(0)
        self.label_green_value.delete(0, 'end')
        self.label_green_value.insert(0, '0')

        self.slider_b.set(0)
        self.label_blue_value.delete(0, 'end')
        self.label_blue_value.insert(0, '0')

        self.slider_s.set(0)
        self.label_speed_value.delete(0, 'end')
        self.label_speed_value.insert(0, '0')

        self.profile_template = """
#!/bin/bash

# Link to Github readme:
# https://github.com/JafarAkhondali/acer-predator-turbo-and-rgb-keyboard-linux-module


mode=0
red=100
green=30
blue=235


acer-predator-turbo-and-rgb-keyboard-linux-module/facer_rgb.py -m {mode} -z 1 -cR {red} -cG {green} -cB {blue}
acer-predator-turbo-and-rgb-keyboard-linux-module/facer_rgb.py -m {mode} -z 2 -cR {red} -cG {green} -cB {blue}
acer-predator-turbo-and-rgb-keyboard-linux-module/facer_rgb.py -m {mode} -z 3 -cR {red} -cG {green} -cB {blue}
acer-predator-turbo-and-rgb-keyboard-linux-module/facer_rgb.py -m {mode} -z 4 -cR {red} -cG {green} -cB {blue}
        """


    rgb_1 = rgb_profile(1, 255, 0, 0)
    print(rgb_1)
    profile_default = rgb_profile(0, 0, 255, 0)

    def create_bin(self, profile):
        return None


    # SAVES THE CURRENT OPTIONS TO THE AUTOSTART PICKLE OBJECT AND THE bin/rgb_autostart.sh PROGRAM
    #@todo
    # NOTE: SHOULD BE SAVED IN A WAY THAT CAN BE EASILY RETIREVED AND r, g AND b VARIABLES USED TO UPDATE SLIDERS AND THE COLOR PREVIEW
    def set_autostart(self):
        template = """
        #!/bin/bash
        echo {}
        """
        print(template.format('THIS JUST A TEST!!!'))
        facer_path = 'acer-predator-turbo-and-rgb-keyboard-linux-module/facer_rgb.py'
        return template

    # RETREIVES THE CURRENTLY STORED AUTOSTART PROFILE AND SETS THE SLIDERS ACCORDINGLY
    def retreive_autostart(self):
        return None

    # LOADS THE CURRENTLY SET AUTOSTART PICKLE OBJECT 
    def load_autostart(self):
        return None

    # SAVES THE CURRENT OPTIONS TO rgb_profile_<PROFILE NAME>
    def save_profile(self):

        profile_name = "profiles/rgb_profile_" + self.entry.get()
        profile = rgb_profile(
            self.mode.get(),
            self.slider_r.get(),
            self.slider_g.get(),
            self.slider_b.get()
        )
        print(profile)
        with open(profile_name, "wb") as file:
            pickle.dump(profile, file)

    # LOADS THE GIVEN PROFILE FROM rgb_profile_<PROFILE NAME>
    def load_profile(self):
        profile_name = "profiles/rgb_profile_" + self.entry.get()
        with open(profile_name, "rb") as file:
            profile_obj = pickle.load(file)
            self.mode.set(profile_obj.mode)
            self.slider_r.set(profile_obj.red)
            self.slider_g.set(profile_obj.green)
            self.slider_b.set(profile_obj.blue)
            print(profile_obj)
        self.update_preview()

    def button_event(self):
        print("Button pressed")

    def rgb_autostart(self):
        subprocess.run(["profiles/rgb_autostart"])

    def rgb_init_6(self):
        subprocess.run(["profiles/rgb_init.sh_6"])

    def rgb_init_7(self):
        subprocess.run(["profiles/rgb_init.sh_7"])

    def rgb_init_8(self):
        subprocess.run(["profiles/rgb_init.sh_9"])

    def rgb_init_9(self):
        subprocess.run(["profiles/rgb_init.sh_9"])

    def rgb_init_3(self):
        subprocess.run(["profiles/rgb_init.sh_3"])

    def rgb_init_2(self):
        subprocess.run(["profiles/rgb_init.sh_2"])

    def update_red(self, slider):
        red = slider
        self.label_red_value.delete(0, 'end')
        self.label_red_value.insert(0, str(int(red*255)))
        self.update_preview()

    def update_slider_r(self, event):
        try:
            red = int(self.label_red_value.get())
            self.slider_r.set(red / 255.)
        except:
            pass

    def update_green(self, slider):
        green = slider
        self.label_green_value.delete(0, 'end')
        self.label_green_value.insert(0, str(int(green*255)))
        self.update_preview()

    def update_slider_g(self, event):
        try:
            green = int(self.label_green_value.get())
            self.slider_g.set(green / 255.)
        except:
            pass

    def update_blue(self, slider):
        blue = slider
        self.label_blue_value.delete(0, 'end')
        self.label_blue_value.insert(0, str(int(blue*255)))
        self.update_preview()

    def update_slider_b(self, event):
        try:
            blue = int(self.label_blue_value.get())
            self.slider_b.set(blue / 255.)
        except:
            pass

    def update_speed(self, slider):
        speed = slider
        self.label_speed_value.delete(0, 'end')
        self.label_speed_value.insert(0, str(int(speed*9)))
        self.update_preview()

    def update_slider_s(self, event):
        try:
            speed = int(self.label_speed_value.get())
            self.slider_s.set(speed / 255.)
        except:
            pass

    def return_red(self):
        red = int(self.slider_r.get() * 255)
        return red

    def return_green(self):
        green = int(self.slider_g.get() * 255)
        return green

    def return_blue(self):
        blue = int(self.slider_b.get() * 255)
        return blue

    def update_preview(self):
        red = self.return_red()
        green = self.return_green()
        blue = self.return_blue()
        rgb_255 = (red, green, blue)
        rgb_hex = '#%02x%02x%02x' % rgb_255
        self.label_preview.configure(fg_color=rgb_hex)
        return rgb_hex

    # RETRIEVES THE VALUES OF THE MODE AND THE R G AND B SLIDERS AND EXECUTES A COMMAND TO SET IT
    def set_colour(self):
        mode = self.mode.get()
        red = str(clamp(int(self.label_red_value.get()), 0, 255))
        green = str(clamp(int(self.label_green_value.get()), 0, 255))
        blue = str(clamp(int(self.label_blue_value.get()), 0, 255))
        speed = str(clamp(int(self.label_speed_value.get()), 0, 9))
        facer_path = 'acer-predator-turbo-and-rgb-keyboard-linux-module/facer_rgb.py'
        for i in range(1, 5):
            zone = i
            print(facer_path +' -m {} -z {} -cR {} -cG {} -cB {} -s {}'.format(mode, zone, red, green, blue, speed)
                 )
            subprocess.run(
                [facer_path,
                 '-m', '{}'.format(mode),
                 '-z', '{}'.format(zone),
                 '-cR', '{}'.format(red),
                 '-cG', '{}'.format(green),
                 '-cB', '{}'.format(blue),
                 '-s', '{}'.format(speed)]
            )

    def change_mode(self):
        if self.switch_2.get() == 1:
            customtkinter.set_appearance_mode("dark")
        else:
            customtkinter.set_appearance_mode("light")

    def on_closing(self, event=0):
        self.destroy()

    def start(self):
        self.mainloop()


if __name__ == "__main__":
    app = App()
    #App.update_blue(App, App.return_blue)
    #App.update_red(App.return_red)
    #App.update_green(App.return_green)
    app.start()