Paul Lipscomb
282 lines
11 KiB
Python
282 lines
11 KiB
Python
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("") |