shuttle_gui/mainFile.py at master · SARL-Engineering/shuttle_gui · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# Authors: Aaron Rito, Corwin Perren
# Date: 10/2/17
# Project: SARL Shuttlebox Behavior System
# Client: Oregon State University, SARL lab

########################################################################################################################
#   Welcome to Tanguay Lab Shuttlebox Behaviour System. This program will detect SARL shuttleboxes and configure them  #
#   for use. The system is designed to run 48 boxes. In theory, with the right hardware, one may be able to run "n"    #
#   boxes using this software. (SUCCESSFULLY TESTED 5 SYSTEMS AT ONCE 10/2/17 A.R.)                                    #
#                                                                                                                      #
#   For support email: aaronrito@gmail.com                                                                             #
########################################################################################################################
import sys
import signal
import os
from PyQt5 import QtCore, QtWidgets, QtGui, uic
import serialHandler
from Framework.BoxHandlerCore import BoxHandler
from settings_core import ShuttleSettings
import qdarkstyle

UI_FILE_PATH = "Shuttlebox_form.ui"
UI_LOGO = "logo.png"


class NewWindow(QtWidgets.QMainWindow):
    stop_all_threads = QtCore.pyqtSignal()  # signal from class QmainWindow to stop
    welcome_signal = QtCore.pyqtSignal()

    def __init__(self, parent=None):
        super(NewWindow, self).__init__(parent)
        uic.loadUi(UI_FILE_PATH, self)

        # load the Tanguay logo before starting
        self.settings_class = ShuttleSettings(self)
        directory = self.settings_class.settings.value("results_directory")
        print("Saved directory set to: ", directory)
        if not directory or directory == 0:
            file_dialog = QtWidgets.QFileDialog(self)
            file_dialog.setFileMode(QtWidgets.QFileDialog.DirectoryOnly)
            file_dialog.setDirectory(QtCore.QStandardPaths.standardLocations(QtCore.QStandardPaths.HomeLocation)[0])
            if directory != "":
                directory = file_dialog.getExistingDirectory()
            if directory != "":
                print("Setting directory to: \"" + directory + "\".")
                self.settings_class.settings.setValue("results_directory", directory)
            else:
                print("No folder selected.")
            if not os.path.isdir(directory):
                os.makedirs(directory)
            print("New directory set to: ", directory)
        self.box_handler_class = BoxHandler(self)
        self.welcome_signal.connect(self.welcome_open)
        self.welcome = QtWidgets.QProgressDialog(parent)
        self.box_handler_class.boxes_ready_signal.connect(self.welcome_slot)
        self.welcome.setCancelButton(None)
        self.welcome.setRange(0, 0)
        self.my_font = QtGui.QFont()
        self.my_font.setPointSizeF(12)
        self.my_font.setBold(True)
        self.welcome.setFont(self.my_font)
        self.welcome.setWindowTitle("SARL Shuttlebox Behavior System")
        self.welcome_label = QtWidgets.QLabel(
            "Welcome to the SARL Shuttlebox Behavior System. Concept by: Aaron Rito," +
            " Corwin Perren, and Vance Langer.")
        self.welcome_button = QtWidgets.QPushButton(None)
        self.welcome_window = QtWidgets.QGraphicsScene()
        self.welcome_image = QtGui.QImage("big_logo.png")
        self.lab = QtWidgets.QLabel()
        layout = QtWidgets.QVBoxLayout()
        self.imageLabel = QtWidgets.QLabel()
        self.imageLabel.setPixmap(QtGui.QPixmap.fromImage(self.welcome_image))
        layout.addWidget(self.imageLabel)
        layout.addWidget(self.welcome_label)
        layout.addSpacing(30)
        self.welcome.setLayout(layout)
        self.welcome.resize(850, 500)
        self.welcome.setModal(True)
        self.welcome.setWindowFlags(QtCore.Qt.WindowStaysOnTopHint)
        self.should_run = True
        self.welcome_signal.emit()

    def welcome_open(self):
        self.welcome.exec()

    def welcome_slot(self):
        print("welcome slot")
        self.welcome.done(0)

    def closeEvent(self, event):
        self.stop_all_threads.emit()


if __name__ == "__main__":

    signal.signal(signal.SIGINT, signal.SIG_DFL)

    app = QtWidgets.QApplication(sys.argv)
    app.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())

    app_icon = QtGui.QIcon()
    app_icon.addFile(UI_LOGO, QtCore.QSize(16, 16))
    app_icon.addFile(UI_LOGO, QtCore.QSize(24, 24))
    app_icon.addFile(UI_LOGO, QtCore.QSize(32, 32))
    app_icon.addFile(UI_LOGO, QtCore.QSize(48, 48))
    app_icon.addFile(UI_LOGO, QtCore.QSize(256, 256))
    app.setWindowIcon(app_icon)

    myWindow = NewWindow()

    myWindow.show()

    app.exec_()