statapp/statapp/main_window.py

#
# Copyright (c) 2023 Maxim Slipenko, Eugene Lazurenko.
#
# This file is part of Statapp
# (see https://github.com/shizand/statapp).
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import numpy as np
from PySide2.QtCore import Slot
from PySide2.QtWidgets import QMainWindow, QMessageBox

from statapp.calculations import generateXValues, generateYValues
from statapp.generate_factor_window import GenerateFactorWindow
from statapp.models.input_values_model import InputValuesModel
from statapp.generate_window import GenerateWindow
from statapp.about_window import AboutWindow
from statapp.models.fileslc_model import FileSLCModel
from statapp.ui.ui_main_window import Ui_MainWindow
from statapp.utils import buildMessageBox, addIcon
from statapp.variance_analysis import VarianceAnalysisWindow
from statapp.correlation_analysis import CorrelationAnalysisWindow


class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        addIcon(self)

        self.ui.generateXaction.setEnabled(False)
        self.ui.varianceAnalysisAction.setEnabled(False)
        self.ui.correlationAnalisisAction.setEnabled(False)

        self.aboutWindow = None

        self.isDataChanged = False
        self.model = InputValuesModel()
        self.fileModel = FileSLCModel()
        self.ui.tableView.setModel(self.model)

    @Slot()
    def on_openfileaction_triggered(self):
        currentData = self.model.getData()
        data = np.array([])
        if currentData.size > 1:
            file = ''
            if self.fileModel.fileName:
                file = '\nФайл сохранения: ' + self.fileModel.fileName

            msgBox = buildMessageBox \
                ('Сохранение данных',
                 "Сохранить данные?" + file,
                 QMessageBox.Question,
                 QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel,
                 QMessageBox.Cancel)

            reply = msgBox.exec_()
            if reply == QMessageBox.StandardButton.Yes:
                self.fileModel.saveFile(self.model.getData())
            elif reply == QMessageBox.StandardButton.Cancel:
                return
            else:
                data = self.fileModel.loadFile()
                if data is not None and data.shape[0] > 0:
                    self.model.updateAllData(data)
                    self.isDataChanged = False
        else:
            data = self.fileModel.loadFile()
            if data is not None and data.shape[0] > 0:
                self.model.updateAllData(data)
                self.isDataChanged = False

        if data.shape[1] == 1:
            self.ui.generateXaction.setEnabled(True)
            self.ui.varianceAnalysisAction.setEnabled(False)
            self.ui.correlationAnalisisAction.setEnabled(False)
        elif data.shape[1] > 1:
            self.ui.generateXaction.setEnabled(True)
            self.ui.varianceAnalysisAction.setEnabled(True)
            self.ui.correlationAnalisisAction.setEnabled(True)
        else:
            self.ui.generateXaction.setEnabled(False)
            self.ui.varianceAnalysisAction.setEnabled(False)
            self.ui.correlationAnalisisAction.setEnabled(False)

    @Slot()
    def on_savefileaction_triggered(self):
        self.isDataChanged = not self.fileModel.saveFile(self.model.getData())

    @Slot()
    def on_closefileaction_triggered(self):
        self.fileModel.closeFile()
        self.isDataChanged = False

    @Slot()
    def on_generateYaction_triggered(self):
        gw = GenerateWindow()

        if gw.exec():
            y = generateYValues(gw.mat, gw.deviation, gw.count)
            self.model.updateAllData(y.round(2))
            self.isDataChanged = True
            self.ui.generateXaction.setEnabled(True)

    @Slot()
    def on_generateXaction_triggered(self):
        gfw = GenerateFactorWindow()

        if gfw.exec():
            data = self.model.getData()
            y = self.model.getY()
            xValues = generateXValues(gfw.mat, gfw.deviation, gfw.typeConnection, y)
            data = np.concatenate((data, xValues.round(2)), axis=1)
            self.model.updateAllData(data)
            self.ui.varianceAnalysisAction.setEnabled(True)
            self.ui.correlationAnalisisAction.setEnabled(True)
            self.isDataChanged = True

    @Slot()
    def on_aboutmenuaction_triggered(self):
        self.aboutWindow = AboutWindow()
        self.aboutWindow.show()

    @Slot()
    def on_varianceAnalysisAction_triggered(self):
        dw = VarianceAnalysisWindow(self.model.getData())
        dw.exec()

    @Slot()
    def on_correlationAnalisisAction_triggered(self):
        dw = CorrelationAnalysisWindow(self.model.getData())
        dw.exec()

    def closeEvent(self, event):
        if self.isDataChanged:
            file = ''
            if self.fileModel.fileName:
                file = '\nФайл сохранения: ' + self.fileModel.fileName

            msgBox = buildMessageBox \
                ('Завершение работы',
                 "Сохранить данные?" + file,
                 QMessageBox.Question,
                 QMessageBox.Yes | QMessageBox.No | QMessageBox.Cancel,
                 QMessageBox.Cancel)

            reply = msgBox.exec_()
            if reply == QMessageBox.StandardButton.Yes:
                self.fileModel.saveFile(self.model.getData())
                event.accept()
            elif reply == QMessageBox.StandardButton.No:
                event.accept()
            else:
                event.ignore()
        else:
            event.accept()