mirror of
https://github.com/dcs-retribution/dcs-retribution.git
synced 2025-11-10 15:41:24 +00:00
95 lines
3.2 KiB
Python
95 lines
3.2 KiB
Python
from PySide6.QtCore import Signal
|
|
from PySide6.QtWidgets import (
|
|
QHBoxLayout,
|
|
QLabel,
|
|
QPushButton,
|
|
QWidget,
|
|
)
|
|
|
|
from game.radio.radios import RadioFrequency, RadioRange, MHz, kHz
|
|
from game.theater import NavalControlPoint
|
|
from qt_ui.models import GameModel
|
|
from qt_ui.windows.QRadioFrequencyDialog import QRadioFrequencyDialog
|
|
|
|
|
|
class QLink4Widget(QWidget):
|
|
freq_changed = Signal(QWidget)
|
|
|
|
def __init__(self, cp: NavalControlPoint, game_model: GameModel) -> None:
|
|
super().__init__()
|
|
|
|
self.cp = cp
|
|
self.gm = game_model
|
|
|
|
columns = QHBoxLayout()
|
|
self.setLayout(columns)
|
|
|
|
self.freq = QLabel(self._get_label_text())
|
|
self.check_freq()
|
|
columns.addWidget(self.freq)
|
|
columns.addStretch()
|
|
|
|
self.set_freq_btn = QPushButton("Set LINK4")
|
|
self.set_freq_btn.setProperty("class", "comms")
|
|
self.set_freq_btn.setFixedWidth(100)
|
|
columns.addWidget(self.set_freq_btn)
|
|
self.set_freq_btn.clicked.connect(self.open_freq_dialog)
|
|
|
|
self.reset_freq_btn = QPushButton("Reset LINK4")
|
|
self.reset_freq_btn.setProperty("class", "btn-danger comms")
|
|
self.reset_freq_btn.setFixedWidth(100)
|
|
columns.addWidget(self.reset_freq_btn)
|
|
self.reset_freq_btn.clicked.connect(self.reset_freq)
|
|
|
|
def _get_label_text(self) -> str:
|
|
freq = "AUTO" if self.cp.link4 is None else self.cp.link4
|
|
return f"<b>LINK4: {freq}</b>"
|
|
|
|
def open_freq_dialog(self) -> None:
|
|
ranges = [RadioRange(MHz(225), MHz(400), kHz(25))]
|
|
self.frequency_dialog = QRadioFrequencyDialog(self, self.cp, ranges, link4=True)
|
|
self.frequency_dialog.accepted.connect(self.assign_frequency)
|
|
self.frequency_dialog.show()
|
|
|
|
def assign_frequency(self) -> None:
|
|
hz = round(self.frequency_dialog.frequency_input.value() * 10**6)
|
|
self._try_remove()
|
|
self.cp.link4 = RadioFrequency(hertz=hz)
|
|
self.gm.allocated_freqs.append(self.cp.link4)
|
|
self.freq.setText(self._get_label_text())
|
|
self.check_freq()
|
|
self.freq_changed.emit(self)
|
|
|
|
def reset_freq(self):
|
|
self._try_remove()
|
|
self.cp.link4 = None
|
|
self.freq.setText(self._get_label_text())
|
|
self._reset_color_and_tooltip()
|
|
self.freq_changed.emit(self)
|
|
|
|
def check_freq(self):
|
|
if self.cp.link4 is None:
|
|
return
|
|
if self.gm.allocated_freqs.count(self.cp.link4) > 1:
|
|
self.freq.setStyleSheet("color: orange")
|
|
self.freq.setToolTip(
|
|
"Double booked frequency, verify that this was your intention."
|
|
)
|
|
elif self.gm.allocated_freqs.count(self.cp.link4) == 1:
|
|
self._reset_color_and_tooltip()
|
|
if self.cp.link4 == MHz(243) or self.cp.link4 == MHz(121, 500):
|
|
self.freq.setStyleSheet("color: red")
|
|
self.freq.setToolTip(
|
|
"GUARD Freq. was assigned, verify that this was your intention."
|
|
)
|
|
|
|
def _reset_color_and_tooltip(self):
|
|
self.freq.setStyleSheet("color: white")
|
|
self.freq.setToolTip(None)
|
|
|
|
def _try_remove(self):
|
|
try:
|
|
self.gm.allocated_freqs.remove(self.cp.link4)
|
|
except ValueError:
|
|
pass
|