finish basic UI

src/main.py

@@ -1,19 +1,28 @@
 import customtkinter as ctk
 
 import serial
+import serial.tools
 import serial.tools.list_ports as ls_ports
+import serial.tools.list_ports
 
-from typing import Optional
+from typing import Optional, Any
+import functools
 
 
 class App(ctk.CTk):
     __window_width = 1280
-    __window_height = 720
-    __column_weight = [1, 5]
+    __window_height = 1350
+    __column_weight = [1, 9, 3]
+    serial_status = False
+    serial_baudrate = 115200
+    gain_values = [0.0 for i in range(5)]
+    gain_default = [1.3, 0.7, 1.1, 2.2, 0.5]
+    gain_enable = [True for i in range(5)]
 
     def __init__(self) -> None:
         super().__init__()
         self.__windowInit(self.__window_width, self.__window_height)
+        self.__data_frame()
 
     def __windowInit(self, width: int, height: int) -> None:
         self.title("ISP上位机")
@@ -31,33 +40,241 @@ class App(ctk.CTk):
         center_y = int(screen_height / 2 - height / 2)
         # set the position of the self to the center of the screen
         self.geometry(f"{width}x{height}+{center_x}+{center_y}")
-        # seperate the column
-        self.columnconfigure(0, weight=self.__column_weight[0])
-        self.columnconfigure(1, weight=self.__column_weight[1])
 
+        # create gain_frame
+        top_frame = ctk.CTkFrame(self)
+        top_frame.pack(side="top", fill=ctk.X, padx=10)
+        top_frame.columnconfigure(0, weight=self.__column_weight[0])
+        top_frame.columnconfigure(1, weight=self.__column_weight[1])
+        top_frame.columnconfigure(2, weight=self.__column_weight[2])
+
+        # Serial Port
         # create label
-        self.serial_label = ctk.CTkLabel(self, text="串口:", font=self.font)
-        self.serial_label.grid(column=0, row=1, sticky=ctk.EW)
+        self.serial_label = ctk.CTkLabel(top_frame, text="串口：", font=self.font)
+        self.serial_label.grid(column=0, row=0, sticky=ctk.EW, pady=10, padx=10)
         # create combobox
         self.serial_combobox = ctk.CTkComboBox(
-            self,
+            top_frame,
             state="readonly",
-            command=self.updateSerial,
+            command=self.selectSerial,
             font=self.font,
             dropdown_font=self.font,
         )
-        self.serial_combobox.bind()
+        self.serial_combobox.bind("<Button>", self.updateSerial)
         self.updateSerial()
-        self.serial_combobox.grid(column=1, row=1, sticky=ctk.EW)
+        self.serial_combobox.grid(column=1, row=0, sticky=ctk.EW, pady=10, padx=20)
+        # create button
+        self.serial_button = ctk.CTkButton(
+            top_frame,
+            command=self.connectSerial,
+            font=self.font,
+        )
+        self.serial_button_text = ctk.StringVar(self.serial_button, "连接")
+        self.serial_button.configure(textvariable=self.serial_button_text)
+        self.serial_button.grid(column=2, row=0, sticky=ctk.EW, pady=10, padx=10)
+
+        # Serial Baudrate
+        # create label
+        self.baud_label = ctk.CTkLabel(top_frame, text="波特率：", font=self.font)
+        self.baud_label.grid(column=0, row=1, sticky=ctk.EW, pady=10, padx=10)
+        # create combobox
+        self.baud_combobox = ctk.CTkComboBox(
+            top_frame,
+            values=[str(9600), str(19200), str(38400), str(57600), str(115200)],
+            command=self.setBaudrate,
+            font=self.font,
+            dropdown_font=self.font,
+        )
+        self.baud_combobox.grid(column=1, row=1, sticky=ctk.EW, pady=10, padx=20)
+        # create Set Button
+        self.send_button = ctk.CTkButton(
+            top_frame, text="设置", font=self.font, command=self.setISP
+        )
+        self.send_button.grid(column=2, row=1, sticky=ctk.EW, pady=10, padx=10)
         return None
 
-    def updateSerial(self, port_info: Optional[str] = None):
+    def __data_frame(self) -> None:
+        def setGain(value, i:int, stringVar: ctk.StringVar)-> None:
+            self.gain_values[i] = value
+            stringVar.set(f'{value:.2f}')
+            return None
+
+        def switchSlider(i: int, slider: ctk.CTkSlider) -> None:
+            if slider.cget("state") == "normal":
+                self.gain_values[i] = self.gain_default[i]
+                slider.configure(button_color="grey")
+                slider.configure(state="disabled")
+            else:
+                self.gain_values[i] = slider.get()
+                slider.configure(button_color="#1f538d")
+                slider.configure(state="normal")
+            return None
+
+        def switchCorrection(i: int) -> None:
+            if self.gain_enable[i]:
+                self.gain_enable[i] = False
+            else:
+                self.gain_enable[i] = True
+            return None
+
+        text = [
+            "红色校正系数",
+            "绿色校正系数",
+            "蓝色校正系数",
+            "伽马校正系数",
+            "饱和度调整系数",
+        ]
+        gain_min = [-2.0, -2.0, -2.0, 0.0, -1.0]
+        gain_max = [2.0, 2.0, 2.0, 5.0, 1.0]
+        fg_clolor = [
+            "#4d140b",
+            "#005f50",
+            "#456789",
+            "#596062",
+            "#596062",
+        ]
+
+        data_frame = ctk.CTkFrame(self)
+        data_frame.pack(side="top", fill=ctk.X, pady=10)
+        data_frame.columnconfigure(0, weight=10)
+        data_frame.columnconfigure(1, weight=3)
+
+        gain_frame = []
+        gain_str = []
+        gain_label = []
+        gain_slider = []
+        gain_button = []
+        gain_checkbox = []
+        for i in range(5):
+            # gain_frame
+            gain_frame.append(ctk.CTkFrame(data_frame, fg_color=fg_clolor[i]))
+            gain_frame[i].grid(column=0, sticky=ctk.EW, pady=10, padx=20, ipady=10)
+
+            # label
+            if i < 3:
+                ctk.CTkLabel(gain_frame[i], text=text[i], font=self.font).pack(
+                    side="top", ipady=10
+                )
+            else:
+                gain_checkbox.append(
+                    ctk.CTkCheckBox(
+                        gain_frame[i],
+                        text=text[i],
+                        font=self.font,
+                        command=functools.partial(switchCorrection, i),
+                    )
+                )
+                gain_checkbox[i - 3].select()
+                gain_checkbox[i - 3].pack(side="top", ipady=10)
+
+            # slider
+            gain_slider.append(
+                ctk.CTkSlider(gain_frame[i], from_=gain_min[i], to=gain_max[i])
+            )
+            gain_slider[i].set(self.gain_default[i])
+            gain_slider[i].pack(side="top", fill=ctk.X, padx=10)
+            gain_str.append(
+                ctk.StringVar(gain_slider[i], f"{self.gain_default[i]:.2f}")
+            )
+            gain_slider[i].configure(command=functools.partial(setGain, i=i, stringVar=gain_str[i]))
+
+            # gain value
+            ctk.CTkLabel(gain_frame[i], text=str(gain_min[i]), font=self.font).pack(
+                side="left", padx=10
+            )
+            ctk.CTkLabel(gain_frame[i], text=str(gain_max[i]), font=self.font).pack(
+                side="right", padx=10
+            )
+            gain_label.append(
+                ctk.CTkLabel(
+                    gain_frame[i],
+                    textvariable=gain_str[i],
+                    font=self.font,
+                )
+            )
+            gain_label[i].pack(side="top")
+
+            # button
+            gain_button.append(
+                ctk.CTkButton(
+                    data_frame,
+                    text="使用推荐配置",
+                    command=functools.partial(switchSlider, i=i, slider=gain_slider[i]),
+                    font=self.font,
+                )
+            )
+            gain_button[i].grid(column=1, row=i, sticky=ctk.NSEW, pady=10, padx=20)
+
+        return None
+
+    def updateSerial(self, event: Optional[Any] = None):
+        print("Update Serial Ports")
         port_list = []
         for info in ls_ports.comports():
             port, desc, hwid = info
             port_list.append(port + " " + desc)
         self.serial_combobox.configure(values=port_list)
 
+    def selectSerial(self, port_info: Optional[Any] = None):
+        print("Select Serial port")
+        try:
+            self.select_port = port_info.split()[0]
+        except Exception:
+            print("Get None")
+        finally:
+            print(self.select_port)
+
+    def connectSerial(self, info: Optional[Any] = None):
+        try:
+            print(self.select_port)
+        except AttributeError:
+            print("Please Select Serial Port")
+        else:
+            if not self.serial_status:
+                try:
+                    self.serial = serial.Serial(self.select_port, self.serial_baudrate)
+                except AttributeError:
+                    print("Failed to Connect Serial")
+                else:
+                    self.serial_status = True
+                    self.serial_button_text.set("断开连接")
+                    print("Select Serial Port:")
+                    print(self.select_port)
+            else:
+                try:
+                    self.serial.close()
+                except Exception:
+                    print("Failed to Close Serial")
+                else:
+                    self.serial_status = False
+                    self.serial_button_text.set("连接")
+                    print("Succeed to Close Serial")
+
+    def setBaudrate(self, baudrate: Optional[Any] = None):
+        self.serial_baudrate = int(baudrate)
+        if self.serial_status:
+            try:
+                self.serial.baudrate = int(baudrate)
+            except Exception:
+                print("Failed to Set Serial Baudrate")
+            else:
+                print("Succeed to Set Serial Baudrate")
+        else:
+            print("Please Connect to Serial")
+
+    def setISP(self, event: Optional[Any] = None):
+        if self.serial_status:
+            self.serial.open()
+            for i in range(5):
+                if self.gain_enable[i]:
+                    match i:
+                        case 1:
+                            self.serial.write(b'r')
+                            
+            self.serial.close()
+        else:
+            print("Please Connect to Serial")
+
     def run(self):
         self.mainloop()