pyDSF/ui/mainwindow.py

# -*- coding: utf-8 -*-

"""
Module implementing MainWindow.
"""

VERSION = "1.0"

from PyQt5.QtCore import pyqtSlot
from PyQt5.QtWidgets import *

from .Ui_mainwindow import Ui_MainWindow
from pydsf import *


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("&Start Processing", QDialogButtonBox.AcceptRole)



    @pyqtSlot("QAbstractButton*")
    def on_buttonBox_open_reset_clicked(self, button):
        """
        Slot documentation goes here.
        """
        if button == self.buttonBox_open_reset.button(QDialogButtonBox.Open):
            filenames, filter = QFileDialog.getOpenFileNames(self, 'Open data file', '', "Text files (*.txt *.csv)")
            self.listWidget_data.addItems(filenames)
            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()
            print("Data cleared")
        # self.radioButton_rep_rows.setEnabled(False)
        #            self.radioButton_rep_columns.setEnabled(False)


    @pyqtSlot("QString")
    def on_comboBox_instrument_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        if p0 == 'Analytik Jena qTOWER 2.0/2.2':
            print(p0)
            self.groupBox_temp.setEnabled(True)
        else:
            self.groupBox_temp.setEnabled(False)
        # self.groupBox_data.setEnabled(True)
        #        self.groupBox_cutoff.setEnabled(True)
        #        self.groupBox_cbar.setEnabled(True)
        #        self.groupBox_signal_threshold.setEnabled(True)

    @pyqtSlot()
    def on_buttonBox_process_accepted(self):
        """
        Slot documentation goes here.
        """

        if self.listWidget_data.count() < 1:
            QMessageBox.critical(self, 'Error', "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, 'Warning', "Signal threshold is currently set to zero.", QMessageBox.Ok,
                                QMessageBox.Ok)

        self.progressBar.setEnabled(True)
        self.statusBar.showMessage("Processing...")

        c_lower = None
        c_upper = None
        cbar_range = None
        signal_threshold = None
        type = self.comboBox_instrument.currentText()
        if self.groupBox_cutoff.isChecked():
            c_lower = self.doubleSpinBox_lower.value()
            c_upper = self.doubleSpinBox_upper.value()
        if self.groupBox_cbar.isChecked():
            cbar_range = (self.doubleSpinBox_cbar_start, self.doubleSpinBox_cbar_end)
        if self.groupBox_signal_threshold.isChecked():
            signal_threshold = self.spinBox_signal_threshold.value()

        items = (self.listWidget_data.item(i) for i in range(self.listWidget_data.count()))

        files = []
        for item in items:
            files.append(item.text())
        exp = Experiment(type=type, files=files, t1=self.doubleSpinBox_tmin.value(), t2=self.doubleSpinBox_tmax.value(),
                         dt=self.doubleSpinBox_dt.value(), cols=12, rows=8, cutoff_low=c_lower, cutoff_high=c_upper,
                         signal_threshold=signal_threshold, color_range=cbar_range)
        exp.analyze()

        # plate = Plate(type=type, filename=self.lineEdit_data_file.text(), t1=self.doubleSpinBox_tmin.value(), t2=self.doubleSpinBox_tmax.value(), dt=self.doubleSpinBox_dt.value(), cols=12, rows=8,  cutoff_low=c_lower,  cutoff_high=c_upper,  signal_threshold=signal_threshold,  color_range=cbar_range)
        # self.statusBar.addWidget(self.pb, 100)
        # plate.analyze(gui=self)
        save_data = QMessageBox.question(self, 'Save data', "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, '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(exp.avg_plate.id)))
                #plot(plate, self)

        plotter = PlotResults(exp)
        for i in range(self.tabWidget.count()):
            for plate in exp.plates:
                if i == 0:
                    plotter.plot_raw(plate, self.tabWidget.widget(i))
                elif i == 1:
                    plotter.plot_derivative(plate, self.tabWidget.widget(i))
                elif i == 2:
                    plotter.plot_tm_heatmap_single(plate, self.tabWidget.widget(i))
                elif exp.avg_plate and i == 3:
                    plotter.plot_tm_heatmap_single(exp.avg_plate, self.tabWidget.widget(i))
                    self.tabWidget.setTabEnabled(i, True)
                else:
                    self.tabWidget.setTabEnabled(i, False)

        #fig, ax = figures[0]
        #self.tabWidget.widget(0).canvas.fig = fig
        #self.tabWidget.widget(0).canvas.ax = ax

        #self.tabWidget.widget(0).canvas.draw()

        self.progressBar.setEnabled(False)
        self.statusBar.showMessage("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, p0):
        """
        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()