mirror of
https://github.com/Athemis/PyDSF.git
synced 2025-04-05 14:46:03 +00:00
296 lines
10 KiB
Python
296 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Module implementing MainWindow.
|
|
"""
|
|
from PyQt5.QtCore import (pyqtSlot, QObject, pyqtSignal, QThreadPool,
|
|
QRunnable, QCoreApplication)
|
|
from PyQt5.QtWidgets import (QMainWindow, QProgressBar, QDialogButtonBox,
|
|
QFileDialog, QMessageBox, QApplication)
|
|
|
|
from .Ui_mainwindow import Ui_MainWindow
|
|
from .mplwidget import MplWidget
|
|
from pydsf import Experiment, PlotResults
|
|
|
|
VERSION = "1.0"
|
|
|
|
|
|
class WorkerSignals(QObject):
|
|
|
|
finished = pyqtSignal()
|
|
|
|
|
|
class Worker(QRunnable):
|
|
finished = pyqtSignal(int)
|
|
|
|
def __init__(self, owner):
|
|
super(Worker, self).__init__()
|
|
self.exp = None
|
|
self.owner = owner
|
|
self.signals = WorkerSignals()
|
|
|
|
def run(self):
|
|
c_lower = None
|
|
c_upper = None
|
|
cbar_range = None
|
|
signal_threshold = None
|
|
instrument_type = self.owner.comboBox_instrument.currentText()
|
|
if self.owner.groupBox_cutoff.isChecked():
|
|
c_lower = self.owner.doubleSpinBox_lower.value()
|
|
c_upper = self.owner.doubleSpinBox_upper.value()
|
|
if self.owner.groupBox_cbar.isChecked():
|
|
cbar_range = (self.owner.doubleSpinBox_cbar_start,
|
|
self.owner.doubleSpinBox_cbar_end)
|
|
if self.owner.groupBox_signal_threshold.isChecked():
|
|
signal_threshold = self.owner.spinBox_signal_threshold.value()
|
|
|
|
items = (self.owner.listWidget_data.item(i)
|
|
for i in range(self.owner.listWidget_data.count()))
|
|
|
|
files = []
|
|
for item in items:
|
|
files.append(item.text())
|
|
self.exp = Experiment(exp_type=instrument_type,
|
|
files=files,
|
|
t1=self.owner.doubleSpinBox_tmin.value(),
|
|
t2=self.owner.doubleSpinBox_tmax.value(),
|
|
dt=self.owner.doubleSpinBox_dt.value(),
|
|
cols=12,
|
|
rows=8,
|
|
cutoff_low=c_lower,
|
|
cutoff_high=c_upper,
|
|
signal_threshold=signal_threshold,
|
|
color_range=cbar_range)
|
|
self.exp.analyze()
|
|
self.signals.finished.emit()
|
|
|
|
|
|
class TaskSignals(QObject):
|
|
finished = pyqtSignal(list)
|
|
|
|
|
|
class Tasks(QObject):
|
|
def __init__(self):
|
|
super(Tasks, self).__init__()
|
|
|
|
self.pool = QThreadPool()
|
|
self.pool.setMaxThreadCount(1)
|
|
self.tasks = []
|
|
self.data = []
|
|
self.signals = TaskSignals()
|
|
|
|
def add_task(self, task):
|
|
self.tasks.append(task)
|
|
|
|
def get_data(self):
|
|
self.pool.waitForDone()
|
|
return self.data
|
|
|
|
def start(self):
|
|
for task in self.tasks:
|
|
self.pool.start(task)
|
|
self.pool.waitForDone()
|
|
|
|
for task in self.tasks:
|
|
self.data.append(task.exp)
|
|
|
|
self.signals.finished.emit(self.data)
|
|
|
|
|
|
class MainWindow(QMainWindow, Ui_MainWindow):
|
|
"""
|
|
Class documentation goes here.
|
|
"""
|
|
|
|
def __init__(self, parent=None):
|
|
"""
|
|
Constructor
|
|
|
|
@param parent reference to the parent widget (QWidget)
|
|
"""
|
|
QMainWindow.__init__(self, parent)
|
|
self.setupUi(self)
|
|
self.progressBar = QProgressBar()
|
|
self.progressBar.setMaximum(0)
|
|
self.progressBar.setMaximum(0)
|
|
self.progressBar.setEnabled(False)
|
|
self.statusBar.addPermanentWidget(self.progressBar)
|
|
self.statusBar.showMessage("Welcome to PyDSF")
|
|
|
|
self.buttonBox_process.addButton(
|
|
QCoreApplication.translate("MainWindow", "&Start Processing"),
|
|
QDialogButtonBox.AcceptRole)
|
|
|
|
self.tasks = Tasks()
|
|
self.worker = Worker(self)
|
|
|
|
@pyqtSlot("QAbstractButton*")
|
|
def on_buttonBox_open_reset_clicked(self, button):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
if button == self.buttonBox_open_reset.button(QDialogButtonBox.Open):
|
|
filenames = QFileDialog.getOpenFileNames(
|
|
self,
|
|
QCoreApplication.translate("MainWindow", "Open data file"), '',
|
|
QCoreApplication.translate("MainWindow",
|
|
"Text files (*.txt *.csv)"))
|
|
self.listWidget_data.addItems(filenames[0])
|
|
if self.listWidget_data.count() > 1:
|
|
self.groupBox_replicates.setChecked(True)
|
|
self.radioButton_rep_files.setEnabled(True)
|
|
elif button == self.buttonBox_open_reset.button(
|
|
QDialogButtonBox.Reset):
|
|
self.listWidget_data.clear()
|
|
|
|
@pyqtSlot("QString")
|
|
def on_comboBox_instrument_currentIndexChanged(self, p0):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
if p0 == 'Analytik Jena qTOWER 2.0/2.2':
|
|
self.groupBox_temp.setEnabled(True)
|
|
else:
|
|
self.groupBox_temp.setEnabled(False)
|
|
|
|
def generate_plot_tab(self, name):
|
|
tab = MplWidget(parent=self)
|
|
tab.setObjectName(name)
|
|
return tab
|
|
|
|
def generate_plate_tabs(self, plate):
|
|
plotter = PlotResults()
|
|
|
|
if id != 'average':
|
|
tab = self.generate_plot_tab("tab_heatmap_{}".format(id))
|
|
self.tabWidget.addTab(tab, QCoreApplication.translate(
|
|
"MainWindow", "Heatmap #{}".format(plate.id)))
|
|
plotter.plot_tm_heatmap_single(plate, tab)
|
|
|
|
tab = self.generate_plot_tab("tab_raw_{}".format(id))
|
|
self.tabWidget.addTab(tab, QCoreApplication.translate(
|
|
"MainWindow", "Raw Data #{}".format(plate.id)))
|
|
plotter.plot_raw(plate, tab)
|
|
|
|
tab = self.generate_plot_tab("tab_derivative_{}".format(id))
|
|
self.tabWidget.addTab(tab, QCoreApplication.translate(
|
|
"MainWindow", "Derivatives #{}".format(plate.id)))
|
|
plotter.plot_derivative(plate, tab)
|
|
else:
|
|
tab = self.generate_plot_tab("tab_heatmap_{}".format(id))
|
|
self.tabWidget.addTab(tab, QCoreApplication.translate(
|
|
"MainWindow", "Heatmap ({})".format(plate.id)))
|
|
plotter.plot_tm_heatmap_single(plate, tab)
|
|
|
|
@pyqtSlot()
|
|
def on_buttonBox_process_accepted(self):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
|
|
if self.listWidget_data.count() < 1:
|
|
QMessageBox.critical(
|
|
self, QCoreApplication.translate("MainWindow", "Error"),
|
|
QCoreApplication.translate("MainWindow",
|
|
"No data file loaded!"),
|
|
QMessageBox.Close, QMessageBox.Close)
|
|
return
|
|
if self.spinBox_signal_threshold.value(
|
|
) == 0 and self.groupBox_signal_threshold.isChecked():
|
|
QMessageBox.warning(
|
|
self, QCoreApplication.translate("MainWindow", "Warning"),
|
|
QCoreApplication.translate(
|
|
"MainWindow",
|
|
"Signal threshold is currently set to zero."),
|
|
QMessageBox.Ok, QMessageBox.Ok)
|
|
|
|
self.progressBar.setEnabled(True)
|
|
self.statusBar.showMessage(QCoreApplication.translate("MainWindow",
|
|
"Processing..."))
|
|
|
|
self.tasks.signals.finished.connect(self.on_processing_finished)
|
|
self.tasks.add_task(self.worker)
|
|
self.tasks.start()
|
|
|
|
@pyqtSlot()
|
|
def on_processing_finished(self):
|
|
# For now, only read the first entry
|
|
exp = self.tasks.data[0]
|
|
|
|
save_data = QMessageBox.question(
|
|
self, QCoreApplication.translate("MainWindow", "Save data"),
|
|
QCoreApplication.translate(
|
|
"MainWindow", "Calculations are finished. Save results?"),
|
|
QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes)
|
|
if save_data == QMessageBox.Yes:
|
|
dialog = QFileDialog()
|
|
dialog.setFileMode(QFileDialog.Directory)
|
|
folder = dialog.getExistingDirectory(
|
|
self, QCoreApplication.translate("MainWindow",
|
|
"Choose path for results"))
|
|
for plate in exp.plates:
|
|
plate.write_tm_table(
|
|
'{}/plate_{}_04_tm.csv'.format(folder, str(plate.id)))
|
|
plate.write_derivative_table(
|
|
'{}/plate_{}_03_dI_dT.csv'.format(folder, str(plate.id)))
|
|
plate.write_filtered_table(
|
|
'{}/plate_{}_02_filtered_data.csv'.format(folder,
|
|
str(plate.id)))
|
|
plate.write_raw_table('{}/plate_{}_01_raw_data.csv'.format(
|
|
folder, str(plate.id)))
|
|
|
|
if exp.avg_plate:
|
|
exp.avg_plate.write_avg_tm_table(
|
|
'{}/plate_{}_05_tm_avg.csv'.format(
|
|
folder, str(self.worker.exp.avg_plate.id)))
|
|
|
|
for i in range(len(self.worker.exp.plates)):
|
|
|
|
plate = exp.plates[i]
|
|
self.generate_plate_tabs(plate)
|
|
|
|
if exp.avg_plate:
|
|
|
|
plate = exp.avg_plate
|
|
self.generate_plate_tabs(plate)
|
|
|
|
self.progressBar.setEnabled(False)
|
|
self.statusBar.showMessage(QCoreApplication.translate("MainWindow",
|
|
"Finished!"))
|
|
|
|
@pyqtSlot()
|
|
def on_buttonBox_process_rejected(self):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
QApplication.quit()
|
|
|
|
pyqtSlot()
|
|
|
|
def on_actionQuit_triggered(self):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
QApplication.quit()
|
|
|
|
@pyqtSlot("bool")
|
|
def on_groupBox_cutoff_toggled(self):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
self.doubleSpinBox_upper.setValue(self.doubleSpinBox_tmax.value())
|
|
self.doubleSpinBox_lower.setValue(self.doubleSpinBox_tmin.value())
|
|
|
|
@pyqtSlot()
|
|
def on_actionAbout_triggered(self):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
# TODO: not implemented yet
|
|
raise NotImplementedError
|
|
|
|
@pyqtSlot()
|
|
def on_actionAbout_Qt_triggered(self):
|
|
"""
|
|
Slot documentation goes here.
|
|
"""
|
|
QApplication.aboutQt()
|