all_gui.py

# -*- coding: utf-8 -*-
"""
Created on Tue Apr 24 17:43:57 2018

gui to run everything

@author: Patrick
"""

#import necessary modules
import os
import sys
# os.chdir(getattr(sys, '_MEIPASS', os.path.dirname(os.path.abspath(__file__))))

import subprocess
import spike_analysis.gui
import kilosort_control.sort_gui

#import GUI objects/Widgets
from PySide2.QtCore import QRect

from PySide2.QtWidgets import (QApplication, QMainWindow, QFrame, QSizePolicy,
                          QVBoxLayout, QMenuBar, QPushButton)

from PySide2.QtGui import QGuiApplication

#make sure we're using the right qt API
os.environ['QT_API'] = 'pyside2'


"""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""
Let's begin!
"""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""


#class for MainWindow instance
class MainWindow(QMainWindow):
    
    def __init__(self, parent=None):
        ''' sets up the whole main window '''
        
        #standard init
        super(MainWindow, self).__init__(parent)

        #set the window title
        self.setWindowTitle('SpikeSuite')
        
        #get screen dimensions
        self.screen_height = QGuiApplication.primaryScreen().availableGeometry().height()
        self.screen_width = QGuiApplication.primaryScreen().availableGeometry().width()
        
        self.make_buttons()

        #create a QMenuBar and set geometry
        self.menubar = QMenuBar(self)
        self.menubar.setGeometry(QRect(0, 0, self.screen_width*.5, self.screen_height*.03))
        #set the QMenuBar as menu bar for main window
        self.setMenuBar(self.menubar)
        
        #show the window fullscreen
        self.resize(600, 400)
        
    def make_buttons(self):
        
        buttons = QFrame(self)
        buttons_layout = QVBoxLayout()
        
        self.acquire_button = QPushButton('Acquire',buttons)
        self.sort_button = QPushButton('Sort',buttons)
        self.analyze_button = QPushButton('Analyze',buttons)
        
        self.acquire_button.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Expanding)
        self.sort_button.setSizePolicy(QSizePolicy.Preferred,QSizePolicy.Expanding)
        self.analyze_button.setSizePolicy(QSizePolicy.Preferred,QSizePolicy.Expanding)

        self.acquire_button.setStyleSheet('background-color: darkRed')
        self.sort_button.setStyleSheet('background-color: darkBlue')
        self.analyze_button.setStyleSheet('background-color: darkGreen')
        
        self.analyze_button.clicked.connect(self.run_analysis)
        self.acquire_button.clicked.connect(self.run_acq)
        self.sort_button.clicked.connect(self.run_sort)
            
        buttonstyle = 'QPushButton{border-style: solid;border-width: 2px;border-radius: 40px;font: bold 30px; color: white}'
        buttonstyle += 'QPushButton:hover{border-width: 4px; border-color:black}'
        buttons.setStyleSheet(buttonstyle)

        buttons_layout.addWidget(self.acquire_button)
        buttons_layout.addWidget(self.sort_button)
        buttons_layout.addWidget(self.analyze_button)
        
        buttons.setLayout(buttons_layout)
        self.setCentralWidget(buttons)
        
    def run_analysis(self):        
        os.chdir(os.path.dirname(os.path.abspath(__file__)) + '/spike_analysis')
        #create and show the main window
        self.analysis_frame = spike_analysis.gui.MainWindow(launcher=self)
        self.analysis_frame.show()
    
        #set up stream for stdout and stderr based on outputStream class
        self.outputStream = spike_analysis.gui.outputStream()
        #when outputStream sends messages, connect to appropriate function for
        #writing to terminal window
        self.outputStream.message.connect(self.analysis_frame.print_message)

        #connect stdout and stderr to outputStream
        sys.stdout = self.outputStream
        sys.stderr = self.outputStream
        
        self.close()
        
    def run_sort(self,recdir=None):
        os.chdir(os.path.dirname(os.path.abspath(__file__)) + '/kilosort_control')
        #create and show the main window

        self.recdir = recdir
            
        self.sort_frame = kilosort_control.sort_gui.MainWindow(launcher=self)
        self.sort_frame.show()
    
        #set up stream for stdout and stderr based on outputStream class
        self.outputStream = kilosort_control.sort_gui.outputStream()
        #when outputStream sends messages, connect to appropriate function for
        #writing to terminal window
        self.outputStream.message.connect(self.sort_frame.print_message)

        #connect stdout and stderr to outputStream
        sys.stdout = self.outputStream
        sys.stderr = self.outputStream
        
        self.close()
        
    def run_acq(self):
        
        subprocess.Popen('python ./oe_control/control_gui.py',
                               shell=True) 
        
        # self.close()
        
        sys.exit(app.exec_())


    def show_launcher(self):
        #create and show the main window
        os.chdir(os.path.dirname(os.path.abspath(__file__)))
        self.show()
        
        try:
            self.analysis_frame.close()
        except:
            pass      
        
        try:
            self.sort_frame.close()
        except:
            pass     
        
if __name__ == '__main__':
    #create a QApplication if one doesn't already exist
    app = QApplication.instance()
    if app == None:
        app = QApplication([])
    
    #create and show the main window
    frame = MainWindow()
    
    if len(sys.argv) > 1:
        frame.run_sort(sys.argv[1])
    else:
        frame.show()
    
    #exit the app when we're all done
    sys.exit(app.exec_())