import os
import sys
from importlib import resources as impresources
from PySide6.QtCore import QByteArray, QSettings, QThread, Signal
from PySide6.QtGui import QAction, QIcon, QDropEvent, QDragLeaveEvent
from PySide6.QtWidgets import (
QApplication,
QCheckBox,
QMainWindow,
QMenu,
QMessageBox,
QSizePolicy,
QSystemTrayIcon,
QTextEdit,
QVBoxLayout,
QWidget,
)
from . import __version__, convert_raw_file, docker, types, vendor
DATA_DIR = os.path.join(str(impresources.files("mzx")), "..", "data")
[docs]
class ConverterThread(QThread):
finished = Signal() # Explicitly declare signal to satisfy mypy
def __init__(
self,
params: types.TConfig,
parent: QWidget | None = None,
):
super().__init__(parent)
self.params = params
[docs]
def run(self):
_outfile = convert_raw_file(self.params)
# Emit finished signal automatically when the thread ends
[docs]
class MainWindow(QMainWindow):
def __init__(self):
super().__init__()
self.initUI()
[docs]
def initUI(self):
settings = QSettings("mzx", "app")
self.central_widget = QWidget()
layout = QVBoxLayout(self.central_widget)
self.setWindowTitle(f"mzx Version {__version__}")
self.setCentralWidget(self.central_widget)
# Restore window geometry (position and size)
geometry = settings.value("window_geometry")
if geometry and isinstance(geometry, QByteArray):
self.restoreGeometry(geometry)
else:
self.setGeometry(100, 100, 600, 400)
# Peak Picking Option
self.peakpicking_checkbox = QCheckBox("Enable Peakpicking", self)
peak_picking = bool(settings.value("peakpicking", False, bool))
self.peakpicking_checkbox.setChecked(peak_picking)
layout.addWidget(self.peakpicking_checkbox)
# Remove Zeros Option
self.removezeros_checkbox = QCheckBox("Remove Zeros", self)
remove_zeros = bool(settings.value("removezeros", False, bool))
self.removezeros_checkbox.setChecked(remove_zeros)
layout.addWidget(self.removezeros_checkbox)
# Create a blank panel at the bottom
blank_panel = QWidget(self)
blank_panel.setSizePolicy(
QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Expanding
)
blank_panel.setMinimumHeight(self.height() // 2)
layout.addWidget(blank_panel)
# Log output
self.log_text_edit = QTextEdit(self)
self.log_text_edit.setReadOnly(True)
layout.addWidget(self.log_text_edit)
self._layout = layout
[docs]
def closeEvent(self, event):
settings = QSettings("mzx", "app")
settings.setValue("window_geometry", self.saveGeometry())
settings.setValue("peakpicking", self.peakpicking_checkbox.isChecked())
settings.setValue("removezeros", self.removezeros_checkbox.isChecked())
super().closeEvent(event)
[docs]
def dragEnterEvent(self, event: QDropEvent) -> None:
if event.mimeData().hasUrls():
self.setStyleSheet("background-color: lightgreen;")
event.accept()
else:
self.setStyleSheet("background-color: lightred;")
event.ignore()
[docs]
def dragLeaveEvent(self, _event: QDragLeaveEvent) -> None:
self.setStyleSheet("") # Reset background color on leave
[docs]
def dropEvent(self, event: QDropEvent) -> None:
self.setStyleSheet("") # Reset background color after drop
for url in event.mimeData().urls():
path = url.toLocalFile()
# if not os.path.isfile(path) or os.path.isdir(path):
# logger.error(f"Invalid path: {path}")
# return
self.convert(path)
[docs]
def convert(self, path: str) -> None:
if not docker.check_running():
self.show_popup("Docker is not running. Please start Docker and try again.")
self.log_text_edit.append("Docker is not running. Conversion cancelled.")
return
self.log_text_edit.append(f"Launching thread to convert {path}...")
# Create thread and connect its finishe
#
vendor_name = vendor.vendor_name_from_file(path)
params: types.TConfig = {
"infile": path,
"index": False,
"sortbyscan": False,
"peak_picking": "all" if self.peakpicking_checkbox.isChecked() else "off",
"remove_zeros": self.removezeros_checkbox.isChecked(),
"vendor": vendor_name,
"outfile": None,
"type": "mzml",
"overwrite": False,
"debug": False,
"verbose": False,
"lockmass": None,
"lockmass_disabled": None,
"lockmass_function_exclude": None,
"lockmass_tolerance": None,
"neg_lockmass": None,
"pos_lockmass": None,
}
self.convert_thread = ConverterThread(params)
self.convert_thread.finished.connect(self.on_conversion_complete)
self.convert_thread.start()
[docs]
def on_conversion_complete(self):
"""Slot that runs when the conversion thread finishes."""
self.log_text_edit.append("Conversion task completed.")
[docs]
def show_popup(self, message: str) -> QMessageBox:
dialog = QMessageBox()
dialog.setIcon(QMessageBox.Icon.Warning)
dialog.setText(message)
dialog.setWindowTitle("Warning")
dialog.setStandardButtons(QMessageBox.StandardButton.Ok)
dialog.show()
return dialog
[docs]
def main():
app = QApplication(sys.argv)
window = MainWindow()
# Tray Icon
icon_path = os.path.join(DATA_DIR, "MMLogo-white-bg.png")
icon = QIcon(icon_path)
tray = QSystemTrayIcon()
tray.setIcon(icon)
tray.setVisible(True)
menu = QMenu()
quit = QAction("Quit")
quit.triggered.connect(app.quit)
menu.addAction(quit)
tray.setContextMenu(menu)
window.setWindowIcon(icon)
window.show()
sys.exit(app.exec())
if __name__ == "__main__":
main()