Paul Lipscomb
350 lines
14 KiB
Python
350 lines
14 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(115)
|
|
self.current_color = PALETTE_HEX.get("Gray", "#263238")
|
|
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)
|
|
|
|
_grp_style = "QFrame { border: 1px solid rgba(255,255,255,0.2); border-radius: 4px; }"
|
|
|
|
# Group 1: Learn + input readout
|
|
self.hw_cc = None
|
|
self.hw_channel = None
|
|
self.is_learning = False
|
|
self.grp_learn = QFrame()
|
|
self.grp_learn.setStyleSheet(_grp_style)
|
|
_grp1_layout = QVBoxLayout(self.grp_learn)
|
|
_grp1_layout.setContentsMargins(4, 4, 4, 4)
|
|
_grp1_layout.setSpacing(3)
|
|
|
|
self.learn_btn = QPushButton("Learn")
|
|
self.learn_btn.setFixedHeight(20)
|
|
self.learn_btn.setStyleSheet("border: none;")
|
|
self.learn_btn.setContextMenuPolicy(Qt.ContextMenuPolicy.CustomContextMenu)
|
|
self.learn_btn.customContextMenuRequested.connect(self.on_learn_context_menu)
|
|
_grp1_layout.addWidget(self.learn_btn)
|
|
|
|
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; border: none;")
|
|
self.cc_input_lbl.setFixedHeight(16)
|
|
_grp1_layout.addWidget(self.cc_input_lbl)
|
|
|
|
inner.addWidget(self.grp_learn)
|
|
|
|
self.arrow_lbl = QLabel("▼")
|
|
self.arrow_lbl.setAlignment(Qt.AlignmentFlag.AlignHCenter)
|
|
self.arrow_lbl.setStyleSheet("color: rgba(255,255,255,0.3); font-size: 9px; background: transparent;")
|
|
self.arrow_lbl.setFixedHeight(12)
|
|
inner.addWidget(self.arrow_lbl)
|
|
|
|
# Group 2: CC spin + CH spin + output readout
|
|
self.grp_cc = QFrame()
|
|
self.grp_cc.setStyleSheet(_grp_style)
|
|
_grp2_layout = QVBoxLayout(self.grp_cc)
|
|
_grp2_layout.setContentsMargins(4, 4, 4, 4)
|
|
_grp2_layout.setSpacing(3)
|
|
|
|
self.cc_spin = QSpinBox()
|
|
self.cc_spin.setMinimum(0)
|
|
self.cc_spin.setMaximum(127)
|
|
self.cc_spin.setValue(default_cc)
|
|
self.cc_spin.setFixedWidth(55)
|
|
_cc_lbl = QLabel("CC")
|
|
_cc_lbl.setStyleSheet("color: #fff; font-size: 10px; border: none;")
|
|
self.cc_spin_row = QWidget()
|
|
_hbox_cc = QHBoxLayout(self.cc_spin_row)
|
|
_hbox_cc.setContentsMargins(0, 0, 0, 0)
|
|
_hbox_cc.setSpacing(4)
|
|
_hbox_cc.addWidget(_cc_lbl)
|
|
_hbox_cc.addWidget(self.cc_spin)
|
|
_grp2_layout.addWidget(self.cc_spin_row)
|
|
|
|
self.ch_spin = QSpinBox()
|
|
self.ch_spin.setMinimum(1)
|
|
self.ch_spin.setMaximum(16)
|
|
self.ch_spin.setValue(1)
|
|
self.ch_spin.setFixedWidth(55)
|
|
_ch_lbl = QLabel("CH")
|
|
_ch_lbl.setStyleSheet("color: #fff; font-size: 10px; border: none;")
|
|
self.ch_spin_row = QWidget()
|
|
_hbox_ch = QHBoxLayout(self.ch_spin_row)
|
|
_hbox_ch.setContentsMargins(0, 0, 0, 0)
|
|
_hbox_ch.setSpacing(4)
|
|
_hbox_ch.addWidget(_ch_lbl)
|
|
_hbox_ch.addWidget(self.ch_spin)
|
|
_grp2_layout.addWidget(self.ch_spin_row)
|
|
|
|
self.cc_output_lbl = QLabel(f"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; border: none;")
|
|
self.cc_output_lbl.setFixedHeight(16)
|
|
_grp2_layout.addWidget(self.cc_output_lbl)
|
|
|
|
inner.addWidget(self.grp_cc)
|
|
|
|
# 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 _cc_input_label(self, vel="--"):
|
|
if self.hw_cc is None:
|
|
return ""
|
|
ch = f" CH{self.hw_channel}" if self.hw_channel is not None else ""
|
|
return f"CC{self.hw_cc}{ch} [{vel}]"
|
|
|
|
def _cc_output_label(self, vel="--"):
|
|
out_cc = self.cc_spin.value()
|
|
out_ch = self.ch_spin.value()
|
|
return f"CC{out_cc} CH{out_ch} [{vel}]"
|
|
|
|
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("border: none;")
|
|
if self.hw_cc is not None:
|
|
self.learn_btn.setText(f"HW: CC{self.hw_cc}")
|
|
self.cc_input_lbl.setText(self._cc_input_label())
|
|
self.cc_input_lbl.setStyleSheet("background-color: #000000; color: #00e676; font-size: 10px; padding: 1px; border: none;")
|
|
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; border: none;")
|
|
|
|
def bind_hw_cc(self, cc_number, channel=None):
|
|
self.hw_cc = cc_number
|
|
self.hw_channel = channel
|
|
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.hw_channel = None
|
|
self.is_learning = False
|
|
self.learn_btn.setText("Learn")
|
|
self.learn_btn.setStyleSheet("border: none;")
|
|
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()
|
|
out_ch = self.ch_spin.value()
|
|
value = 127 if self.toggle_state else 0
|
|
print(f"[toggle] CC{cc_num} CH{out_ch} ({value})")
|
|
send_cc(cc_num, value, channel=out_ch)
|
|
self.cc_output_lbl.setText(self._cc_output_label(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(),
|
|
"ch": self.ch_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,
|
|
"hw_channel": self.hw_channel,
|
|
}
|
|
|
|
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))
|
|
self.ch_spin.setValue(state.get("ch", 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()
|
|
value = 127 if self.toggle_state else 0
|
|
self.cc_output_lbl.setText(self._cc_output_label(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)
|
|
self.hw_channel = state.get("hw_channel", None)
|
|
if self.hw_cc is not None:
|
|
self.learn_btn.setText(f"HW: CC{self.hw_cc}")
|
|
self.cc_input_lbl.setText(self._cc_input_label())
|
|
self.cc_input_lbl.setStyleSheet("background-color: #000000; color: #00e676; font-size: 10px; padding: 1px; border: none;")
|
|
else:
|
|
self.learn_btn.setText("Learn")
|
|
self.cc_input_lbl.setText("") |