126 lines
3.9 KiB
Python
126 lines
3.9 KiB
Python
import sys
|
|
import winput
|
|
from PySide2.QtCore import Qt
|
|
from PySide2.QtWidgets import QApplication, QMainWindow, QRadioButton, QGroupBox
|
|
from PySide2.QtWidgets import QVBoxLayout, QHBoxLayout, QWidget, QLabel, QSpinBox
|
|
from PySide2.QtWidgets import QAction
|
|
from PySide2.QtGui import QIcon
|
|
|
|
WM_MOUSEWHEEL = 0x020A # mouse wheel moved
|
|
|
|
class ArtyWidget(QWidget):
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
|
|
self.radio_type1 = QRadioButton("Type 1: MILRAD 622 - 978")
|
|
self.radio_type2 = QRadioButton("Type 2: MILRAD 800 - 1120")
|
|
|
|
gbox_layout = QVBoxLayout()
|
|
gbox_layout.addWidget(self.radio_type1)
|
|
gbox_layout.addWidget(self.radio_type2)
|
|
|
|
self.gbox = QGroupBox("Settings (Ctrl+H to hide the window bar)")
|
|
self.gbox.setLayout(gbox_layout)
|
|
|
|
self.target_range = QSpinBox()
|
|
self.target_range.setMinimum(100)
|
|
self.target_range.setMaximum(1600)
|
|
self.elevation = QSpinBox()
|
|
self.elevation.setReadOnly(True)
|
|
|
|
font = self.elevation.font()
|
|
font.setPointSize(14)
|
|
self.target_range.setFont(font)
|
|
self.elevation.setFont(font)
|
|
|
|
self.radio_type1.toggled.connect(self.toggle_type)
|
|
self.radio_type1.setChecked(True)
|
|
self.target_range.valueChanged.connect(self.update_elevation)
|
|
|
|
layout = QHBoxLayout()
|
|
layout.addWidget(QLabel("Range:"))
|
|
layout.addWidget(self.target_range)
|
|
layout.addWidget(QLabel("Elevation:"))
|
|
layout.addWidget(self.elevation)
|
|
|
|
main_layout = QVBoxLayout()
|
|
main_layout.addWidget(self.gbox)
|
|
main_layout.addLayout(layout)
|
|
|
|
self.setLayout(main_layout)
|
|
|
|
def toggle_type(self):
|
|
if self.radio_type1.isChecked():
|
|
self.elevation.setMinimum(622)
|
|
self.elevation.setMaximum(978)
|
|
else:
|
|
self.elevation.setMinimum(800)
|
|
self.elevation.setMaximum(1120)
|
|
|
|
self.update_elevation(self.target_range.value())
|
|
|
|
def update_elevation(self, value):
|
|
x0, x1 = 100, 1600
|
|
if self.radio_type1.isChecked():
|
|
y0, y1 = 978, 622
|
|
else:
|
|
y0, y1 = 1120, 800
|
|
|
|
elevation = round(self.interpolator(value, x0, x1, y0, y1))
|
|
self.elevation.setValue(elevation)
|
|
|
|
def interpolator(self, x, x0, x1, y0, y1):
|
|
return y0 + (y1 - y0) * (x - x0) / (x1 - x0)
|
|
|
|
def increment_range(self, mouse_event):
|
|
if mouse_event.action == WM_MOUSEWHEEL:
|
|
if not self.isActiveWindow():
|
|
current_range = self.target_range.value()
|
|
increment = mouse_event.additional_data
|
|
self.target_range.setValue(current_range + increment)
|
|
|
|
|
|
class ArtyGUI(QMainWindow):
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.arty_widget = ArtyWidget(self)
|
|
|
|
self.setCentralWidget(self.arty_widget)
|
|
self.setWindowTitle("Arty")
|
|
self.setWindowIcon(QIcon("icon.png"))
|
|
self.setWindowFlag(Qt.WindowStaysOnTopHint)
|
|
self.resize(150, 20)
|
|
|
|
self.hideBarAction = QAction('&Hide Window Bar', self)
|
|
self.hideBarAction.setShortcut('Ctrl+H')
|
|
self.hideBarAction.triggered.connect(self.hide_bar)
|
|
self.addAction(self.hideBarAction)
|
|
|
|
self.quitAction = QAction('&Quit', self)
|
|
self.quitAction.setShortcut('Ctrl+Q')
|
|
self.quitAction.triggered.connect(qApp.quit)
|
|
self.addAction(self.quitAction)
|
|
|
|
self.arty_widget.target_range.setFocus()
|
|
|
|
def hide_bar(self):
|
|
self.setWindowFlag(Qt.FramelessWindowHint)
|
|
self.arty_widget.gbox.setTitle("Settings (Ctrl+Q to quit)")
|
|
self.show()
|
|
self.arty_widget.target_range.setFocus()
|
|
|
|
def wheelEvent(self, event):
|
|
self.arty_widget.target_range.event(event)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app = QApplication(sys.argv)
|
|
|
|
arty_gui = ArtyGUI()
|
|
arty_gui.show()
|
|
|
|
winput.hook_mouse(arty_gui.arty_widget.increment_range)
|
|
|
|
sys.exit(app.exec_())
|