import uuid from PyQt6.QtWidgets import ( QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QSpinBox, QLineEdit, QFrame, QSizePolicy ) from PyQt6.QtCore import Qt, QEvent from palette import PALETTE_HEX, PALETTE_VIVID from styles import TITLE_STYLE from zonebutton import ZoneButton from colorswatchpopup import ColorSwatchPopup from midi_sender import send_cc class ToggleWidget(QWidget): def __init__(self, label, default_cc): super().__init__() self.setSizePolicy(QSizePolicy.Policy.Fixed, QSizePolicy.Policy.Preferred) self.setFixedWidth(100) self.current_color = "transparent" self.toggle_state = False self.uid = str(uuid.uuid4()) self.zone = None self.center_index = None self.zone_index = None self.on_zone_change_callback = None outer = QVBoxLayout(self) outer.setContentsMargins(0, 0, 0, 0) outer.setSpacing(2) self.container = QFrame() self.container.setFrameShape(QFrame.Shape.StyledPanel) self.apply_color(self.current_color) inner = QVBoxLayout(self.container) inner.setContentsMargins(4, 6, 4, 6) inner.setSpacing(4) inner.setAlignment(Qt.AlignmentFlag.AlignHCenter) # Learn button self.hw_cc = None self.is_learning = False self.learn_btn = QPushButton("Learn") self.learn_btn.setFixedHeight(20) self.learn_btn.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu) self.learn_btn.customContextMenuRequested.connect(self.on_learn_context_menu) inner.addWidget(self.learn_btn) # CC In label self.cc_input_lbl = QLabel("") self.cc_input_lbl.setAlignment(Qt.AlignmentFlag.AlignHCenter) self.cc_input_lbl.setStyleSheet("background-color: #000000; color: #888888; font-size: 10px; padding: 1px;") inner.addWidget(self.cc_input_lbl) # CC spin self.cc_spin = QSpinBox() self.cc_spin.setMinimum(1) self.cc_spin.setMaximum(127) self.cc_spin.setValue(default_cc) self.cc_spin.setFixedWidth(75) inner.addWidget(self.cc_spin, alignment=Qt.AlignmentFlag.AlignHCenter) self.cc_output_lbl = QLabel(f"Out CC{default_cc} [--]") self.cc_output_lbl.setAlignment(Qt.AlignmentFlag.AlignHCenter) self.cc_output_lbl.setStyleSheet("background-color: #000000; color: #00e676; font-size: 10px; padding: 1px;") inner.addWidget(self.cc_output_lbl) # Toggle button with LED dot overlay btn_container = QWidget() btn_container.setFixedSize(60, 60) self.btn = QPushButton("", btn_container) self.btn.setFixedSize(60, 60) self.btn.setStyleSheet("") self.btn.clicked.connect(self.on_click) self.led_dot = QLabel(btn_container) self.led_dot.setFixedSize(6, 6) self.led_dot.move(5, 49) self.led_dot.setStyleSheet("background-color: #3c3c3c; border-radius: 3px;") inner.addWidget(btn_container, alignment=Qt.AlignmentFlag.AlignHCenter) # Scribble strip self.title_edit = QLineEdit(label) self.title_edit.setAlignment(Qt.AlignmentFlag.AlignHCenter) self.title_edit.setStyleSheet(TITLE_STYLE) inner.addWidget(self.title_edit) # Color swatch button self.color_name = "Gray" self.color_swatch_btn = QPushButton() self.color_swatch_btn.setFixedHeight(10) self.color_swatch_btn.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Fixed) self.color_swatch_btn.setStyleSheet(f"background-color: {PALETTE_VIVID['Gray']}; border-radius: 2px; border: none;") self.color_swatch_btn.clicked.connect(self.show_color_popup) inner.addWidget(self.color_swatch_btn) # + / - buttons inside container below color swatch toggle_btn_row = QHBoxLayout() toggle_btn_row.setContentsMargins(0, 0, 0, 0) toggle_btn_row.setSpacing(2) self.minus_btn = QPushButton("-") self.minus_btn.setFixedSize(30, 20) self.plus_btn = QPushButton("+") self.plus_btn.setFixedSize(30, 20) toggle_btn_row.addWidget(self.minus_btn) toggle_btn_row.addStretch() toggle_btn_row.addWidget(self.plus_btn) inner.addLayout(toggle_btn_row) # Zone button self.zone_btn = ZoneButton() self.zone_btn.left_callback = lambda active: self.on_zone_checked("left", active) self.zone_btn.right_callback = lambda active: self.on_zone_checked("right", active) inner.addWidget(self.zone_btn) # Separator before ID label id_sep = QWidget() id_sep.setFixedHeight(1) id_sep.setStyleSheet("background-color: #3a3a3a;") inner.addWidget(id_sep) # Channel ID label self.id_label = QLabel("") self.id_label.setAlignment(Qt.AlignmentFlag.AlignHCenter) self.id_label.setStyleSheet("color: #ffffff; font-size: 13px; font-weight: bold; background: transparent;") inner.addWidget(self.id_label) outer.addWidget(self.container) self.on_select_callback = None for child in self.findChildren(QWidget): child.installEventFilter(self) def eventFilter(self, obj, event): if event.type() == QEvent.Type.MouseButtonPress: if self.on_select_callback: self.on_select_callback(self) return False def mousePressEvent(self, event): if self.on_select_callback: self.on_select_callback(self) super().mousePressEvent(event) def apply_color(self, hex_color, selected=False): self.current_color = hex_color border = "#00ff88" if selected else "rgba(0,0,0,0.3)" border_width = "3px" if selected else "2px" self.container.setObjectName("stripContainer") self.container.setStyleSheet(f""" QFrame#stripContainer {{ background-color: {hex_color}; border-radius: 6px; border: {border_width} solid {border}; }} """) def set_selected(self, selected): self.apply_color(self.current_color, selected=selected) def start_learn(self): self.is_learning = True self.learn_btn.setText("Listening...") self.learn_btn.setStyleSheet("color: orange; font-weight: bold;") def stop_learn(self): self.is_learning = False self.learn_btn.setStyleSheet("") if self.hw_cc is not None: self.learn_btn.setText(f"HW: CC{self.hw_cc}") self.cc_input_lbl.setText(f"In CC{self.hw_cc} [--]") self.cc_input_lbl.setStyleSheet("background-color: #000000; color: #00e676; font-size: 10px; padding: 1px;") else: self.learn_btn.setText("Learn") self.cc_input_lbl.setText("") self.cc_input_lbl.setStyleSheet("background-color: #000000; color: #888888; font-size: 10px; padding: 1px;") def bind_hw_cc(self, cc_number): self.hw_cc = cc_number self.stop_learn() def on_learn_context_menu(self, pos): from PyQt6.QtWidgets import QMenu if self.hw_cc is None: return menu = QMenu(self) clear_action = menu.addAction("Clear Input Assignment") action = menu.exec(self.learn_btn.mapToGlobal(pos)) if action == clear_action: self.hw_cc = None self.is_learning = False self.learn_btn.setText("Learn") self.learn_btn.setStyleSheet("") self.cc_input_lbl.setText("") def update_id_label(self, zone, index): if zone == "left": self.id_label.setText(f"L{index + 1}") elif zone == "right": self.id_label.setText(f"R{index + 1}") else: self.id_label.setText(f"{index + 1}") def on_zone_checked(self, zone, active): if active: self.zone = zone else: self.zone = None if self.on_zone_change_callback: self.on_zone_change_callback(self) def show_color_popup(self): popup = ColorSwatchPopup(self.on_color_change, self) btn_pos = self.color_swatch_btn.mapToGlobal(self.color_swatch_btn.rect().bottomLeft()) popup.move(btn_pos) popup.show() def on_color_change(self, name, hex_color=None): if hex_color is None: hex_color = PALETTE_HEX.get(name, self.current_color) self.color_name = name self.color_swatch_btn.setStyleSheet(f"background-color: {PALETTE_VIVID.get(self.color_name, hex_color)}; border-radius: 2px; border: none;") self.apply_color(hex_color) self._update_btn_style() def on_click(self): self.toggle_state = not self.toggle_state cc_num = self.cc_spin.value() value = 127 if self.toggle_state else 0 send_cc(cc_num, value) self.cc_output_lbl.setText(f"Out CC{cc_num} [{value}]") self._update_btn_style() def _update_btn_style(self): if self.toggle_state: self.led_dot.setStyleSheet("background-color: #00e676; border-radius: 3px;") else: self.led_dot.setStyleSheet("background-color: #3c3c3c; border-radius: 3px;") def get_state(self): return { "uid": self.uid, "label": self.title_edit.text(), "cc": self.cc_spin.value(), "state": self.toggle_state, "color": self.color_name, "zone": self.zone, "center_index": self.center_index, "zone_index": self.zone_index, "hw_cc": self.hw_cc } def set_state(self, state): if "uid" in state: self.uid = state["uid"] self.title_edit.setText(state.get("label", "")) self.cc_spin.setValue(state.get("cc", 1)) color_name = state.get("color", "Gray") hex_color = PALETTE_HEX.get(color_name, "transparent") self.color_name = color_name self.color_swatch_btn.setStyleSheet(f"background-color: {PALETTE_VIVID.get(self.color_name, hex_color)}; border-radius: 2px; border: none;") self.apply_color(hex_color) self.toggle_state = state.get("state", False) self._update_btn_style() cc_num = self.cc_spin.value() value = 127 if self.toggle_state else 0 self.cc_output_lbl.setText(f"Out CC{cc_num} [{value}]") self.zone = state.get("zone", None) self.center_index = state.get("center_index", None) self.zone_index = state.get("zone_index", None) self.zone_btn.set_state(self.zone == "left", self.zone == "right") self.hw_cc = state.get("hw_cc", None) if self.hw_cc is not None: self.learn_btn.setText(f"HW: CC{self.hw_cc}") self.cc_input_lbl.setText(f"In CC{self.hw_cc} [--]") self.cc_input_lbl.setStyleSheet("background-color: #000000; color: #00e676; font-size: 10px; padding: 1px;") else: self.learn_btn.setText("Learn") self.cc_input_lbl.setText("")