Exemplo n.º 1
0
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas = FigureCanvas(self.ui.mplwidget.figure)
        self.canvas.setParent(self.ui.widget)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox = QVBoxLayout()
        # The matplotlib canvas
        vbox.addWidget(self.canvas)
        # The matplotlib toolbar
        vbox.addWidget(self.mpl_toolbar)
        self.ui.widget.setLayout(vbox)

        # Connect the mplwidget with canvas
        self.ui.mplwidget = self.canvas

        self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'),
                     self.Browse)
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.Plot)
        self.connect(self.ui.pushButton_show, SIGNAL('clicked()'), self.Show)
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas = FigureCanvas(self.ui.mplwidget.figure)
        self.canvas.setParent(self.ui.widget)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox = QVBoxLayout()
        # The matplotlib canvas
        vbox.addWidget(self.canvas)
        # The matplotlib toolbar
        vbox.addWidget(self.mpl_toolbar)
        self.ui.widget.setLayout(vbox)

        # Connect the mplwidget with canvas
        self.ui.mplwidget = self.canvas
        self.reset_plot()

        self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'),
                     self.Browse)
        self.connect(self.ui.pushButton_change_name, SIGNAL('clicked()'),
                     self.Change_name)
        self.connect(self.ui.pushButton_open, SIGNAL('clicked()'), self.Open)
        self.connect(self.ui.pushButton_Plot, SIGNAL('clicked()'),
                     self.Plot_ready)
        self.connect(self.ui.pushButton_Next, SIGNAL('clicked()'), self.Next)
        self.connect(self.ui.pushButton_Previous, SIGNAL('clicked()'),
                     self.Previous)
        self.connect(self.ui.pushButton_find, SIGNAL('clicked()'), self.Find)
        self.connect(self.ui.pushButton_modify, SIGNAL('clicked()'),
                     self.Modify)
        self.connect(self.ui.pushButton_return, SIGNAL('clicked()'),
                     self.Return)

        self.ui.pushButton_browse.setEnabled(True)
        self.modify = False
        self.first_round = True
        self.open_dir = ''
        self.open_files_number = 0
        self.all_info = []
Exemplo n.º 3
0
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'), lambda : self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda : self.Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_5, SIGNAL('clicked()'), lambda : self.Close("visa5", self.visa5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"), self.Print_data)
        
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)
        
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "b", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(self.ui.curvewidget_2, "b", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "b", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(self.ui.curvewidget_4, "b", titles = ["Plot 4", "X (x)", "Y (y)"])
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None
 def __init__(self, parent = None):
     
     # Standard GUI code
     QWidget.__init__(self, parent)
     
     # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
     # Thus to do anything on the GUI, the commands must go through this variable
     self.ui = Ui_MainWindow()
     self.ui.setupUi(self)
     
     # For the canvas.
     self.canvas = FigureCanvas(self.ui.mplwidget.figure)
     self.canvas.setParent(self.ui.widget)
     # We need the toolbar widget for the canvas
     self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)
     
     # Create the QVBoxLayout object and add the widget into the layout
     vbox = QVBoxLayout()
     # The matplotlib canvas
     vbox.addWidget(self.canvas)
     # The matplotlib toolbar
     vbox.addWidget(self.mpl_toolbar)
     self.ui.widget.setLayout(vbox)
     
     # Connect the mplwidget with canvas
     self.ui.mplwidget = self.canvas
     
     self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'), self.Browse)
     self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.Plot)
     self.connect(self.ui.pushButton_openfile, SIGNAL('clicked()'), self.OpenFile)
     self.connect(self.ui.radioButton_qmd, SIGNAL('clicked()'), self.Choose1)
     self.connect(self.ui.radioButton_ppms, SIGNAL('clicked()'), self.Choose1)
     self.connect(self.ui.radioButton_other, SIGNAL('clicked()'), self.Choose1)
     self.connect(self.ui.pushButton_choose2, SIGNAL('clicked()'), self.Choose2)
     self.connect(self.ui.pushButton_recent, SIGNAL('clicked()'), self.Open_Recent)
     
     self.first_round = True
     self.open_dir = ''
     self.open_files_number = 0
     self.all_info = []
Exemplo n.º 5
0
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general,
                                                     self.ui.widget_general)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)

        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general

        self.connect(self.ui.actionArray_Builder, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(0))
        self.connect(self.ui.actionKeithley_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(1))
        self.connect(self.ui.actionAgilent_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(2))
        self.connect(self.ui.actionKeithley_Stepper_Single_Scan,
                     SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(3))
        self.connect(self.ui.actionKeithley_Gate_Sweep, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(4))
        self.connect(self.ui.actionLockIn_Single_Scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(5))
        self.connect(self.ui.actionResonant_Single_Scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(6))
        self.connect(self.ui.actionSee_Visa_List, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(7))

        self.List_Visas = List_Visas(main=self, ui=self.ui)
        self.Array_Builder_programs = Array_Builder(main=self, ui=self.ui)

        self.Keithley_programs = Keithley(main=self, ui=self.ui)
        self.Keithley_programs.Refresh_visa()

        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0

        self.ui.lineEdit_directory_save.setText(os.getcwd())

        self.Agilent_Yokogawa_programs = Agilent_Yokogawa(main=self,
                                                          ui=self.ui)

        self.Resonant_Sweeper_functions = Resonance_Sweep(main=self,
                                                          ui=self.ui)

        self.Resonant_Sweeper_functions.update_visa()

        self.Lock_In_programs = Lock_In(main=self, ui=self.ui)

        self.KeithleyGateSweep = KeithleyGateSweep(main=self, ui=self.ui)

    # This method is to transfer the array data from the Array Builder tab to the Keithley tab
    # If the array is valid, data_available is true and execute the data transfer
    # If the array is not valid, execution cannot be done
    def CopyDataFunc(self):
        if self.Array_Builder_programs.data_available == True:
            return self.Array_Builder_programs.Values
        else:
            return None

    def copyData(self):
        try:
            if self.Array_Builder_programs.data_available:
                return self.Array_Builder_programs.Values
            self.ui.output.setText('Array has been copied and plotted.')
        except:
            return False
            self.ui.output.setText('No valid array to copy.')

    # Make sure the user is going to quit the program
    def closeEvent(self, event):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()
Exemplo n.º 6
0
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'), lambda : self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(self.ui.pushButton_update_6, SIGNAL('clicked()'), lambda : self.Update("visa6", self.ui.comboBox_visa_6))
        self.connect(self.ui.pushButton_update_7, SIGNAL('clicked()'), lambda : self.Update("visa7", self.ui.comboBox_visa_7))
        self.connect(self.ui.pushButton_update_8, SIGNAL('clicked()'), lambda : self.Update("visa8", self.ui.comboBox_visa_8))
        self.connect(self.ui.pushButton_update_9, SIGNAL('clicked()'), lambda : self.Update("visa9", self.ui.comboBox_visa_9))
        self.connect(self.ui.pushButton_update_10, SIGNAL('clicked()'), lambda : self.Update("visa10", self.ui.comboBox_visa_10))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda : self.Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_select_6, SIGNAL('clicked()'), lambda : self.Select("visa6", self.visa6, self.ui.comboBox_visa_6, self.ui.label_visa_6, [self.ui.pushButton_select_6, self.ui.pushButton_close_6]))
        self.connect(self.ui.pushButton_select_7, SIGNAL('clicked()'), lambda : self.Select("visa7", self.visa7, self.ui.comboBox_visa_7, self.ui.label_visa_7, [self.ui.pushButton_select_7, self.ui.pushButton_close_7]))
        self.connect(self.ui.pushButton_select_8, SIGNAL('clicked()'), lambda : self.Select("visa8", self.visa8, self.ui.comboBox_visa_8, self.ui.label_visa_8, [self.ui.pushButton_select_8, self.ui.pushButton_close_8]))
        self.connect(self.ui.pushButton_select_9, SIGNAL('clicked()'), lambda : self.Select("visa9", self.visa9, self.ui.comboBox_visa_9, self.ui.label_visa_9, [self.ui.pushButton_select_9, self.ui.pushButton_close_9]))
        self.connect(self.ui.pushButton_select_10, SIGNAL('clicked()'), lambda : self.Select("visa10", self.visa10, self.ui.comboBox_visa_10, self.ui.label_visa_10, [self.ui.pushButton_select_10, self.ui.pushButton_close_10]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_5, SIGNAL('clicked()'), lambda : self.Close("visa5", self.visa5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_close_6, SIGNAL('clicked()'), lambda : self.Close("visa6", self.visa6, self.ui.label_visa_6, [self.ui.pushButton_select_6, self.ui.pushButton_close_6]))
        self.connect(self.ui.pushButton_close_7, SIGNAL('clicked()'), lambda : self.Close("visa7", self.visa7, self.ui.label_visa_7, [self.ui.pushButton_select_7, self.ui.pushButton_close_7]))
        self.connect(self.ui.pushButton_close_8, SIGNAL('clicked()'), lambda : self.Close("visa8", self.visa8, self.ui.label_visa_8, [self.ui.pushButton_select_8, self.ui.pushButton_close_8]))
        self.connect(self.ui.pushButton_close_9, SIGNAL('clicked()'), lambda : self.Close("visa9", self.visa9, self.ui.label_visa_9, [self.ui.pushButton_select_9, self.ui.pushButton_close_9]))
        self.connect(self.ui.pushButton_close_10, SIGNAL('clicked()'), lambda : self.Close("visa10", self.visa10, self.ui.label_visa_10, [self.ui.pushButton_select_10, self.ui.pushButton_close_10]))
        
        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"), self.Print_data)
        
        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'), lambda : self.start_font("C"))
    
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)
        self.Update("visa6", self.ui.comboBox_visa_6)
        self.Update("visa7", self.ui.comboBox_visa_7)
        self.Update("visa8", self.ui.comboBox_visa_8)
        self.Update("visa9", self.ui.comboBox_visa_9)
        self.Update("visa10", self.ui.comboBox_visa_10)
        
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "r", "black", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(self.ui.curvewidget_2, "r", "black", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "r", "black", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(self.ui.curvewidget_4, "r", "black", titles = ["Plot 4", "X (x)", "Y (y)"])
        self.curve_5 = self.make_curveWidgets(self.ui.curvewidget_5, "r", "black", titles = ["Plot 5", "X (x)", "Y (y)"])
        self.curve_6 = self.make_curveWidgets(self.ui.curvewidget_6, "r", "black", titles = ["Plot 6", "X (x)", "Y (y)"])
        self.curve_7 = self.make_curveWidgets(self.ui.curvewidget_7, "r", "black", titles = ["Plot 7", "X (x)", "Y (y)"])
        self.curve_8 = self.make_curveWidgets(self.ui.curvewidget_8, "r", "black", titles = ["Plot 8", "X (x)", "Y (y)"])
        self.curve_9 = self.make_curveWidgets(self.ui.curvewidget_9, "r", "black", titles = ["Plot 9", "X (x)", "Y (y)"])
        self.curve_10 = self.make_curveWidgets(self.ui.curvewidget_10, "r", "black", titles = ["Plot 10", "X (x)", "Y (y)"])
        self.curve_11 = self.make_curveWidgets(self.ui.curvewidget_11, "r", "black", titles = ["Plot 11", "X (x)", "Y (y)"])
        self.curve_12 = self.make_curveWidgets(self.ui.curvewidget_12, "r", "black", titles = ["Plot 12", "X (x)", "Y (y)"])
        
        self.ui.mplwidget_1 = self.make_mplToolBar(self.ui.mplwidget_1, self.ui.widget_1)
        self.ui.mplwidget_2 = self.make_mplToolBar(self.ui.mplwidget_2, self.ui.widget_2)
        self.ui.mplwidget_3 = self.make_mplToolBar(self.ui.mplwidget_3, self.ui.widget_3)
        self.ui.mplwidget_4 = self.make_mplToolBar(self.ui.mplwidget_4, self.ui.widget_4)
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None
        self.visa6 = None
        self.visa7 = None
        self.visa8 = None
        self.visa9 = None
        self.visa10 = None
        
        # Set the visa names of the tabWidget
        # font = QFont()
        # font.setPointSize(8)
        # self.ui.tabWidget_visa.setFont(font)
        # self.ui.tabWidget_visa.setTabText(0, "Visa Name 1")
        # self.ui.tabWidget_visa.setTabText(1, "Visa Name 2")
        # self.ui.tabWidget_visa.setTabText(2, "Visa Name 3")
        # self.ui.tabWidget_visa.setTabText(3, "Visa Name 4")
        # self.ui.tabWidget_visa.setTabText(4, "Visa Name 5")
        # self.ui.tabWidget_visa.setTabText(5, "Visa Name 6")
        # self.ui.tabWidget_visa.setTabText(6, "Visa Name 7")
        # self.ui.tabWidget_visa.setTabText(7, "Visa Name 8")
        # self.ui.tabWidget_visa.setTabText(8, "Visa Name 9")
        # self.ui.tabWidget_visa.setTabText(8, "Visa Name 10")
        
        # Input parameters by string
        # Remember to keep the last one as "Comments: ..." so that the program could see that as an end.
        self.input_string = []
        # self.input_string.append("00 Parameter: 1\n")
        # self.input_string.append("01 Parameter: 1\n")
        # self.input_string.append("02 Parameter: 1\n")
        # self.input_string.append("03 Parameter: 1\n")
        # self.input_string.append("04 Parameter: 1\n")
        # self.input_string.append("05 Parameter: 1\n")
        # self.input_string.append("06 Parameter: 1\n")
        # self.input_string.append("07 Parameter: 1\n")
        # self.input_string.append("08 Parameter: 1\n")
        # self.input_string.append("09 Parameter: 1\n")
        # self.input_string.append("10 Parameter: 1\n")
        # self.input_string.append("11 Parameter: 1\n")
        # self.input_string.append("12 Parameter: 1\n")
        # self.input_string.append("13 Comments: ")
        
        
        self.start_font("C")
        
        try:
            file_par = open("parameters.txt", "r")
            input_string_print = file_par.read()
            file_par.close()
            
        except:
            input_string_print = ""
            for st_i in self.input_string:
                input_string_print = input_string_print + st_i
            
        font = QFont()
        font.setPointSize(8)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)
        
        self.inputted_data = []
        self.array_sweep = []
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas = FigureCanvas(self.ui.mplwidget.figure)
        self.canvas.setParent(self.ui.widget)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)

        # For the canvas.
        self.canvas_first_step = FigureCanvas(
            self.ui.mplwidget_first_step.figure)
        self.canvas_first_step.setParent(self.ui.widget_first_step)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_first_step = NavigationToolbar(
            self.canvas_first_step, self.ui.widget_first_step)

        # For the canvas.
        self.canvas_second_step = FigureCanvas(
            self.ui.mplwidget_second_step.figure)
        self.canvas_second_step.setParent(self.ui.widget_second_step)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_second_step = NavigationToolbar(
            self.canvas_second_step, self.ui.widget_second_step)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox = QVBoxLayout()
        # The matplotlib canvas
        vbox.addWidget(self.canvas)
        # The matplotlib toolbar
        vbox.addWidget(self.mpl_toolbar)
        self.ui.widget.setLayout(vbox)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_first_step = QVBoxLayout()
        # The matplotlib canvas
        vbox_first_step.addWidget(self.canvas_first_step)
        # The matplotlib toolbar
        vbox_first_step.addWidget(self.mpl_toolbar_first_step)
        self.ui.widget_first_step.setLayout(vbox_first_step)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_second_step = QVBoxLayout()
        # The matplotlib canvas
        vbox_second_step.addWidget(self.canvas_second_step)
        # The matplotlib toolbar
        vbox_second_step.addWidget(self.mpl_toolbar_second_step)
        self.ui.widget_second_step.setLayout(vbox_second_step)

        # Connect the mplwidget with canvas
        self.ui.mplwidget = self.canvas

        # Connect the mplwidget with canvas
        self.ui.mplwidget_first_step = self.canvas_first_step

        # Connect the mplwidget with canvas
        self.ui.mplwidget_second_step = self.canvas_second_step

        self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'),
                     self.Browse)
        self.connect(self.ui.pushButton_change_name, SIGNAL('clicked()'),
                     self.Plot)
        self.connect(self.ui.pushButton_openfile, SIGNAL('clicked()'),
                     self.OpenFile)
        self.connect(self.ui.radioButton_qmd, SIGNAL('clicked()'),
                     self.Choose_type)
        self.connect(self.ui.radioButton_ppms, SIGNAL('clicked()'),
                     self.Choose_type)
        self.connect(self.ui.radioButton_frontpanel, SIGNAL('clicked()'),
                     self.Choose_type)
        self.connect(self.ui.radioButton_zbridge, SIGNAL('clicked()'),
                     self.Choose_type)
        self.connect(self.ui.radioButton_compressor, SIGNAL('clicked()'),
                     self.Choose_type)
        self.connect(self.ui.pushButton_Plot, SIGNAL('clicked()'),
                     self.Plot_ready)
        self.connect(self.ui.pushButton_recent, SIGNAL('clicked()'),
                     self.Open_Recent)
        self.connect(self.ui.pushButton_plot_first_step, SIGNAL('clicked()'),
                     self.plot1)
        self.connect(self.ui.pushButton_plot_second_step, SIGNAL('clicked()'),
                     self.plot2)
        self.connect(self.ui.pushButton_find, SIGNAL('clicked()'), self.Find)

        self.first_round = True
        self.open_dir = ''
        self.open_files_number = 0
        self.all_info = []

    def Choose_type(self):
        if self.ui.radioButton_qmd.isChecked():
            self.divider = 'Collected Data'
            self.ui.lineEdit_divider.setText(self.divider)
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')
        elif self.ui.radioButton_ppms.isChecked():
            self.divider = '[Data]'
            self.ui.lineEdit_divider.setText(self.divider)
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')
        elif self.ui.radioButton_frontpanel.isChecked():
            self.divider = 'frontpanel'
            self.ui.lineEdit_divider.setText('None')
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')
        elif self.ui.radioButton_zbridge.isChecked():
            self.divider = '#'
            self.ui.lineEdit_divider.setText(self.divider)
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')
        elif self.ui.radioButton_compressor.isChecked():
            self.divider = 'compressor'
            self.ui.lineEdit_divider.setText('None')
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')

    def Browse(self):
        self.open_recent = False

        prev_dir = os.getcwd()
        fileDir = QFileDialog.getOpenFileName(self, 'Select Folder to Save',
                                              prev_dir)
        if fileDir != '':
            open_dir = ''
            file_list = str(fileDir).split('/')
            for i in range(0, len(file_list) - 1):
                if i < len(file_list) - 1:
                    open_dir += file_list[i] + '\\'
                elif i == len(file_list) - 1:
                    open_dir += file_list[i]
            fileDir.replace('/', '\\')
            self.ui.lineEdit_address.setText(fileDir)

            self.ui.pushButton_openfile.setEnabled(True)
            self.ui.groupBox_labels.setEnabled(True)
            self.first_round = False
            filedir = str(fileDir).split('\\')
            self.file_name = filedir[len(filedir) - 1]
            print filedir
        else:
            self.ui.lineEdit_address.setText('None')
            self.ui.lineEdit_condition1.setText('Failed to Read File')

    def OpenFile(self):

        # Every you click the "Open File" button, the content in the plot combox is refreshed
        self.reset_plot()
        self.ui.mplwidget.draw()
        self.reset_first_step_plot()
        self.reset_second_step_plot()
        self.ui.comboBox_x.clear()
        self.ui.comboBox_y.clear()
        self.ui.comboBox_x_2.clear()
        self.ui.comboBox_y_2.clear()
        self.ui.comboBox_y_first_step.clear()
        self.ui.comboBox_y_second_step.clear()
        self.ui.comboBox_first.clear()
        self.ui.comboBox_second.clear()
        self.ui.lineEdit_x.setText('')
        self.ui.lineEdit_y.setText('')
        self.ui.lineEdit_name.setText('')
        self.ui.lineEdit_condition2.setText('')
        self.ui.lineEdit_minute.setText('')
        self.ui.lineEdit_date.setText('')

        labels = ''

        raw_parameters = []
        units = []
        parameters = []
        new_parameters = []
        self.data = []
        new_data = []
        first_data = []

        item = 0
        length = 0
        # This count is the number of the reading loops for labels. If count is over 1000, that means there's something wrong with the divider line and it helps to stop the program
        count = 0

        # It is false when program cannot find the line divider and runs the while loop 1000 times
        self.divider_found = True

        self.fileDir = self.ui.lineEdit_address.text()
        self.divider = str(self.ui.lineEdit_divider.text())
        fp = open(self.fileDir)
        if self.ui.radioButton_frontpanel.isChecked():
            labels = 'None'
        elif self.ui.radioButton_compressor.isChecked():
            labels = fp.readline()
        else:
            while True:
                if count > 1000:
                    self.ui.lineEdit_condition1.setText(
                        'Line divider is not found.')
                    self.divider_found = False
                    break
                line = fp.readline()
                #print 1, line
                linelist = line.split(',')
                if linelist[0].upper() == self.divider.upper() + '\n':
                    break
                for i in range(0, len(linelist)):
                    labels += linelist[i]
                count += 1

        if self.divider_found == True:
            self.ui.textEdit_labels.setText(labels)
            parameters = fp.readline().replace("\n", '')
            if self.ui.radioButton_frontpanel.isChecked():
                parameters = parameters.split('\t')
                for i in range(0, len(parameters)):
                    if parameters[i] == 'P1':
                        parameters[i] = 'Probe Line (P1)'
                    elif parameters[i] == 'P2':
                        parameters[i] = ' OVC Line (P2)'
                    elif parameters[i] == 'P3':
                        parameters[i] = 'IVC/Aux Port Line (P3)'
                    elif parameters[i] == 'P4':
                        parameters[i] = 'Turbo-to-S3 (P4)'
                    elif parameters[i] == 'P5':
                        parameters[i] = 'S3-to-traps (P5)'
                    elif parameters[i] == 'P6':
                        parameters[i] = '4He Dump (P6)'
                    elif parameters[i] == 'P7':
                        parameters[i] = '3He Dump (P7)'
                    elif parameters[i] == 'MG2':
                        parameters[i] = 'OVC Maxigauge (MG2)'
                    elif parameters[i] == 'MG3':
                        parameters[i] = 'Still Maxiguage (MG3)'
                    elif parameters[i] == 'MG4':
                        parameters[i] = 'IVC Maxiguage (MG4)'
                    elif parameters[i] == 'MG5':
                        parameters[i] = 'Probe Maxiguage (MG5)'
            elif self.ui.radioButton_compressor.isChecked():
                parameters = parameters.split('\t')
            else:
                parameters = parameters.split(',')

            if self.ui.radioButton_ppms.isChecked():
                lines = fp.readline().replace("\n", "")
                first_data = lines.split(",")
                for i in range(0, len(first_data)):
                    par = []
                    if first_data[i] != '':
                        par = parameters[i].split(' (')
                        if len(par) == 2:
                            units.append(par[1].replace(')', ''))
                        elif len(par) == 1:
                            units.append('1')
                        new_parameters.append(par[0])
                parameters = new_parameters
            elif self.ui.radioButton_qmd.isChecked():
                units = fp.readline().replace("\n", '')
                units = units.split(",")
            elif self.ui.radioButton_frontpanel.isChecked():
                for i in range(0, len(parameters)):
                    units.append('')
            elif self.ui.radioButton_compressor.isChecked():
                units.append('')
                new_parameters.append(parameters[0])
                for i in range(1, len(parameters)):
                    par = []
                    par = parameters[i].split('(')
                    if par[1].replace(')', '') == 's':
                        units.append('h')
                    else:
                        units.append(par[1].replace(')', ''))
                    new_parameters.append(par[0])
                parameters = new_parameters

            self.ui.comboBox_x_2.addItem('None')
            self.ui.comboBox_y_2.addItem('None')
            self.ui.comboBox_first.addItem('None')
            self.ui.comboBox_second.addItem('None')
            if self.ui.radioButton_compressor.isChecked():
                for i in range(1, len(parameters)):
                    self.ui.comboBox_x.addItem(parameters[i])
                    self.ui.comboBox_y.addItem(parameters[i])
                    self.ui.comboBox_y_first_step.addItem(parameters[i])
                    self.ui.comboBox_y_second_step.addItem(parameters[i])
                    self.ui.comboBox_x_2.addItem(parameters[i])
                    self.ui.comboBox_y_2.addItem(parameters[i])
                    self.ui.comboBox_first.addItem(parameters[i])
                    self.ui.comboBox_second.addItem(parameters[i])
            else:
                for i in range(0, len(parameters)):
                    self.ui.comboBox_x.addItem(parameters[i])
                    self.ui.comboBox_y.addItem(parameters[i])
                    self.ui.comboBox_y_first_step.addItem(parameters[i])
                    self.ui.comboBox_y_second_step.addItem(parameters[i])
                    self.ui.comboBox_x_2.addItem(parameters[i])
                    self.ui.comboBox_y_2.addItem(parameters[i])
                    self.ui.comboBox_first.addItem(parameters[i])
                    self.ui.comboBox_second.addItem(parameters[i])

            for i in range(0, len(parameters)):
                new_data = [parameters[i], units[i], [], []]
                self.data.append(new_data)

            if self.ui.radioButton_ppms.isChecked():
                for i in range(0, len(first_data)):
                    if first_data[i] != '':
                        self.data[item][2].append(first_data[i])
                        item += 1

                length = item

            while True:
                val = []
                lines = fp.readline().replace("\n", "")
                if lines == '':
                    break
                if self.ui.radioButton_frontpanel.isChecked(
                ) or self.ui.radioButton_compressor.isChecked():
                    values = lines.split('\t')
                else:
                    values = lines.split(',')
                for i in range(0, len(values)):
                    if values[0] != 'Measurement was Aborted':
                        if values[i] != '':
                            val.append(values[i])
                if self.ui.radioButton_ppms.isChecked():
                    if len(val) < length:
                        break
                if self.ui.radioButton_frontpanel.isChecked():
                    date_info = val[0].split(' ')
                    weekday = datetime.datetime.strptime(
                        date_info[0], '%Y-%m-%d').strftime('%a')
                    date = date_info[0] + ' ' + weekday + ' ' + date_info[1]
                    self.data[0][3].append(date)
                    self.data[0][3].append(self.convert_time(val[0]))
                for i in range(0, len(val)):
                    self.data[i][2].append(val[i])

            if self.ui.radioButton_frontpanel.isChecked():
                set_zero = self.data[0][3][1]
                for i in range(0, len(self.data[0][2])):
                    self.data[0][2][i] = (
                        (float(self.data[0][3][2 * i + 1]) - set_zero) / 3600)
                self.data[0][1] = 'hour'

            if self.open_recent == False:
                self.item_info = []
                if self.ui.radioButton_frontpanel.isChecked():
                    self.item_info.append('frontpanel')
                elif self.ui.radioButton_compressor.isChecked():
                    self.item_info.append('compressor')
                else:
                    self.item_info.append(self.divider)
                self.item_info.append(self.file_name)
                self.item_info.append(str(self.ui.lineEdit_address.text()))
                self.all_info.append(self.item_info)
                self.ui.comboBox_recent.addItem(
                    self.all_info[len(self.all_info) - 1][1])

            if self.ui.radioButton_compressor.isChecked():
                self.date_transfer = []
                date_tran = []
                date_tran = self.data[0][2]
                for i in range(0, len(date_tran)):
                    date_info = []
                    date_info = date_tran[i].split(' ')
                    weekday = date_info[0]
                    day = date_info[2]
                    year = date_info[4]
                    mon = date_info[1]
                    if mon == 'Jan':
                        mon = '01'
                    elif mon == 'Feb':
                        mon = '02'
                    elif mon == 'Mar':
                        mon = '03'
                    elif mon == 'Apr':
                        mon = '04'
                    elif mon == 'May':
                        mon = '05'
                    elif mon == 'Jun':
                        mon = '06'
                    elif mon == 'Jul':
                        mon = '07'
                    elif mon == 'Aug':
                        mon = '08'
                    elif mon == 'Sep':
                        mon = '09'
                    elif mon == 'Oct':
                        mon = '10'
                    elif mon == 'Nov':
                        mon = '11'
                    elif mon == 'Dec':
                        mon = '12'
                    date = year + '-' + mon + '-' + day + ' ' + weekday + ' ' + date_info[
                        3]
                    self.date_transfer.append(date)
                #self.date_transfer.append(self.data[1][2])
                self.data = self.data[1:]
                for i in range(0, len(self.data[0][2])):
                    self.data[0][2][i] = float(self.data[0][2][i]) / 3600

            self.ui.groupBox_plot.setEnabled(True)
            self.ui.groupBox_first.setEnabled(True)
            self.ui.groupBox_second.setEnabled(True)
            self.ui.lineEdit_condition2.setText('Ready to Plot.')
            self.ui.label_11.setEnabled(True)
            self.ui.comboBox_recent.setEnabled(True)
            self.ui.pushButton_recent.setEnabled(True)
            self.ui.comboBox_y_first_step.setEnabled(True)
            self.ui.pushButton_plot_first_step.setEnabled(True)
            self.ui.pushButton_plot_second_step.setEnabled(True)
            self.ui.comboBox_y_second_step.setEnabled(True)
            self.ui.label_14.setEnabled(True)
            self.ui.label_15.setEnabled(True)
            if self.ui.radioButton_frontpanel.isChecked(
            ) or self.ui.radioButton_compressor.isChecked():
                self.ui.groupBox_date.setEnabled(True)

    def Open_Recent(self):
        file_number = self.ui.comboBox_recent.currentIndex()
        if self.all_info[file_number][0].upper() == '[Data]'.upper():
            self.ui.radioButton_ppms.setChecked(True)
            self.ui.lineEdit_divider.setText('[Data]')
            self.divider = '[Data]' + '\n'
            self.ui.lineEdit_address.setText(self.all_info[file_number][2])
        elif self.all_info[file_number][0].upper() == 'Collected Data'.upper():
            self.ui.radioButton_qmd.setChecked(True)
            self.ui.lineEdit_divider.setText('Collected Data')
            self.divider = 'Collected Data' + '\n'
            self.ui.lineEdit_address.setText(self.all_info[file_number][2])
        elif self.all_info[file_number][0].upper() == 'frontpanel'.upper():
            self.ui.radioButton_frontpanel.setChecked(True)
            self.ui.lineEdit_divider.setText('')
            self.divider = self.all_info[file_number][1] + '\n'
            self.ui.lineEdit_address.setText(self.all_info[file_number][2])
            self.ui.lineEdit_divider.setText('None')
            self.ui.lineEdit_condition1.setText(
                self.ui.comboBox_recent.currentText() + ' has been opened.')
        elif self.all_info[file_number][0].upper() == 'compressor'.upper():
            self.ui.radioButton_compressor.setChecked(True)
            self.ui.lineEdit_divider.setText('')
            self.divider = self.all_info[file_number][1] + '\n'
            self.ui.lineEdit_address.setText(self.all_info[file_number][2])
            self.ui.lineEdit_divider.setText('None')
            self.ui.lineEdit_condition1.setText(
                self.ui.comboBox_recent.currentText() + ' has been opened.')

        self.reset_plot()
        self.ui.mplwidget.draw()
        self.open_recent = True
        self.OpenFile()

    def Find(self):
        try:
            time = float(self.ui.lineEdit_minute.text())
            if time <= self.data[0][2][0]:
                if self.ui.radioButton_frontpanel.isChecked():
                    self.ui.lineEdit_date.setText(self.data[0][3][0])
                elif self.ui.radioButton_compressor.isChecked():
                    self.ui.lineEdit_date.setText(self.date_transfer[0])
            elif time >= self.data[0][2][len(self.data[0][2]) - 1]:
                if self.ui.radioButton_frontpanel.isChecked():
                    self.ui.lineEdit_date.setText(
                        self.data[0][3][2 * (len(self.data[0][2]) - 1)])
                elif self.ui.radioButton_compressor.isChecked():
                    self.ui.lineEdit_date.setText(
                        self.date_transfer[len(self.date_transfer) - 1])
            else:
                item = 0
                while self.data[0][2][item] - time < 0:
                    item += 1
                if abs(self.data[0][2][item - 1] -
                       time) <= abs(self.data[0][2][item] - time):
                    if self.ui.radioButton_frontpanel.isChecked():
                        self.ui.lineEdit_date.setText(
                            self.data[0][3][2 * (item - 1)])
                    elif self.ui.radioButton_compressor.isChecked():
                        self.ui.lineEdit_date.setText(self.date_transfer[item -
                                                                         1])
                else:
                    if self.ui.radioButton_frontpanel.isChecked():
                        self.ui.lineEdit_date.setText(self.data[0][3][2 *
                                                                      item])
                    elif self.ui.radioButton_compressor.isChecked():
                        self.ui.lineEdit_date.setText(self.date_transfer[item])

        except ValueError:
            self.ui.lineEdit_date.setText('')
            self.ui.lineEdit_condition2.setText('Please enter valid time.')

    def Plot_ready(self):
        self.x_value = 0
        self.y_value = 0
        self.x_value = self.ui.comboBox_x.currentIndex()
        self.y_value = self.ui.comboBox_y.currentIndex()
        xaxis = self.data[self.x_value][0]
        xunit = self.data[self.x_value][1]
        yaxis = self.data[self.y_value][0]
        yunit = self.data[self.y_value][1]

        self.x2_value = 0
        self.y2_value = 0
        self.x2_value = self.ui.comboBox_x_2.currentIndex() - 1
        self.y2_value = self.ui.comboBox_y_2.currentIndex() - 1
        x2axis = self.data[self.x2_value][0]
        x2unit = self.data[self.x2_value][1]
        y2axis = self.data[self.y2_value][0]
        y2unit = self.data[self.y2_value][1]

        if self.x2_value == -1:
            self.ui.lineEdit_x.setText(xaxis + ' (' + xunit + ')')
            self.x_value = self.data[self.x_value][2]
        elif self.x2_value >= 0:
            self.ui.lineEdit_x.setText(xaxis + '/' + x2axis + ' (' + xunit +
                                       '/' + x2unit + ')')
            if self.ui.radioButton_x_divide.isChecked():
                self.x_value = numpy.array(
                    self.data[self.x_value][2], dtype='float') / numpy.array(
                        self.data[self.x2_value][2], dtype='float')
            elif self.ui.radioButton_x_multiply.isChecked():
                self.x_value = numpy.array(
                    self.data[self.x_value][2], dtype='float') * numpy.array(
                        self.data[self.x2_value][2], dtype='float')
            elif self.ui.radioButton_x_plus.isChecked():
                self.x_value = numpy.array(
                    self.data[self.x_value][2], dtype='float') + numpy.array(
                        self.data[self.x2_value][2], dtype='float')
            elif self.ui.radioButton_x_minus.isChecked():
                self.x_value = numpy.array(
                    self.data[self.x_value][2], dtype='float') - numpy.array(
                        self.data[self.x2_value][2], dtype='float')
        if self.ui.checkBox_x_value.isChecked():
            self.x_value = numpy.absolute(self.x_value)

        if self.y2_value == -1:
            self.ui.lineEdit_y.setText(yaxis + ' (' + yunit + ')')
            self.y_value = self.data[self.y_value][2]
        elif self.y2_value >= 0:
            self.ui.lineEdit_y.setText(yaxis + '/' + y2axis + ' (' + yunit +
                                       '/' + y2unit + ')')
            if self.ui.radioButton_y_divide.isChecked():
                self.y_value = numpy.array(
                    self.data[self.y_value][2], dtype='float') / numpy.array(
                        self.data[self.y2_value][2], dtype='float')
            elif self.ui.radioButton_x_multiply.isChecked():
                self.y_value = numpy.array(
                    self.data[self.y_value][2], dtype='float') * numpy.array(
                        self.data[self.y2_value][2], dtype='float')
            elif self.ui.radioButton_x_plus.isChecked():
                self.y_value = numpy.array(
                    self.data[self.y_value][2], dtype='float') + numpy.array(
                        self.data[self.y2_value][2], dtype='float')
            elif self.ui.radioButton_x_minus.isChecked():
                self.y_value = numpy.array(
                    self.data[self.y_value][2], dtype='float') - numpy.array(
                        self.data[self.y2_value][2], dtype='float')
        if self.ui.checkBox_y_value.isChecked():
            self.y_value = numpy.absolute(self.y_value)

        if self.x2_value == -1 and self.y2_value == -1:
            self.ui.lineEdit_name.setText(yaxis + ' vs. ' + xaxis)
        elif self.x2_value >= 0 and self.y2_value == -1:
            self.ui.lineEdit_name.setText(yaxis + ' vs. ' + xaxis + '/' +
                                          x2axis)
        elif self.x2_value == -1 and self.y2_value >= 0:
            self.ui.lineEdit_name.setText(yaxis + '/' + y2axis + ' vs. ' +
                                          xaxis)
        else:
            self.ui.lineEdit_name.setText(yaxis + '/' + y2axis + ' vs. ' +
                                          xaxis + '/' + x2axis)

        self.ui.pushButton_change_name.setEnabled(True)
        self.Plot()

    def Plot(self):
        self.reset_plot()
        self.axes.plot(self.x_value, self.y_value, marker='.', linestyle=':')
        self.axes.grid()
        self.axes.set_title(self.ui.lineEdit_name.text())
        self.axes.set_xlabel(self.ui.lineEdit_x.text())
        self.axes.set_ylabel(self.ui.lineEdit_y.text())
        self.ui.mplwidget.draw()

        self.ui.lineEdit_condition2.setText('Plot successfully.')

    def plot1(self):
        y_value1 = self.ui.comboBox_y_first_step.currentIndex()
        y_value2 = self.ui.comboBox_first.currentIndex() - 1
        y_axis1 = self.data[y_value1][0]
        y_axis2 = self.data[y_value2][0]
        y_unit1 = self.data[y_value1][1]
        y_unit2 = self.data[y_value2][1]
        yv1 = self.data[y_value1][2]
        yv2 = self.data[y_value2][2]
        xv = []

        item = 0
        for i in range(0, len(yv1)):
            xv.append(item)
            item += 1

        if y_value2 == -1:
            y_value = yv1
            y_name = y_axis1 + ' (' + y_unit1 + ')'
            name = y_axis1 + ' Steps'
        elif y_value2 >= 0:
            if self.ui.radioButton_first_divide.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') / numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' / ' + y_axis2 + ' (' + y_unit1 + ' / ' + y_unit2 + ')'
                name = y_axis1 + ' / ' + y_axis2 + ' Steps'
            elif self.ui.radioButton_first_multiply.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') * numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' * ' + y_axis2 + ' (' + y_unit1 + ' * ' + y_unit2 + ')'
                name = y_axis1 + ' * ' + y_axis2 + ' Steps'
            elif self.ui.radioButton_first_plus.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') + numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' + ' + y_axis2 + ' (' + y_unit1 + ' + ' + y_unit2 + ')'
                name = y_axis1 + ' + ' + y_axis2 + ' Steps'
            elif self.ui.radioButton_first_minus.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') - numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' - ' + y_axis2 + ' (' + y_unit1 + ' - ' + y_unit2 + ')'
                name = y_axis1 + ' - ' + y_axis2 + ' Steps'
        if self.ui.checkBox_first.isChecked():
            y_value = numpy.absolute(y_value)

        self.reset_first_step_plot()
        self.axes_first_step.plot(xv, y_value, marker='.', linestyle=':')
        self.axes_first_step.grid()
        self.axes_first_step.set_title(name)
        self.axes_first_step.set_xlabel('Unit Step (1)')
        self.axes_first_step.set_ylabel(y_name)
        self.ui.mplwidget_first_step.draw()

    def plot2(self):
        y_value1 = self.ui.comboBox_y_second_step.currentIndex()
        y_value2 = self.ui.comboBox_second.currentIndex() - 1
        y_axis1 = self.data[y_value1][0]
        y_axis2 = self.data[y_value2][0]
        y_unit1 = self.data[y_value1][1]
        y_unit2 = self.data[y_value2][1]
        yv1 = self.data[y_value1][2]
        yv2 = self.data[y_value2][2]
        xv = []
        item = 0
        for i in range(0, len(yv1)):
            xv.append(item)
            item += 1

        if y_value2 == -1:
            y_value = yv1
            y_name = y_axis1 + ' (' + y_unit1 + ')'
            name = y_axis1 + ' Steps'
        elif y_value2 >= 0:
            if self.ui.radioButton_second_divide.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') / numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' / ' + y_axis2 + ' (' + y_unit1 + ' / ' + y_unit2 + ')'
                name = y_axis1 + ' / ' + y_axis2 + ' Steps'
            elif self.ui.radioButton_second_multiply.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') * numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' * ' + y_axis2 + ' (' + y_unit1 + ' * ' + y_unit2 + ')'
                name = y_axis1 + ' * ' + y_axis2 + ' Steps'
            elif self.ui.radioButton_second_plus.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') + numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' + ' + y_axis2 + ' (' + y_unit1 + ' + ' + y_unit2 + ')'
                name = y_axis1 + ' + ' + y_axis2 + ' Steps'
            elif self.ui.radioButton_second_minus.isChecked():
                y_value = numpy.array(
                    self.data[y_value1][2], dtype='float') - numpy.array(
                        self.data[y_value2][2], dtype='float')
                y_name = y_axis1 + ' - ' + y_axis2 + ' (' + y_unit1 + ' - ' + y_unit2 + ')'
                name = y_axis1 + ' - ' + y_axis2 + ' Steps'
        if self.ui.checkBox_second.isChecked():
            y_value = numpy.absolute(y_value)

        self.reset_second_step_plot()
        self.axes_second_step.plot(xv, y_value, marker='.', linestyle=':')
        self.axes_second_step.grid()
        self.axes_second_step.set_title(name)
        self.axes_second_step.set_xlabel('Unit Step (1)')
        self.axes_second_step.set_ylabel(y_name)
        self.ui.mplwidget_second_step.draw()

    def reset_plot(self):
        self.ui.mplwidget.figure.clear()
        self.axes = self.ui.mplwidget.figure.add_subplot(111)

    def reset_first_step_plot(self):
        self.ui.mplwidget_first_step.figure.clear()
        self.axes_first_step = self.ui.mplwidget_first_step.figure.add_subplot(
            111)

    def reset_second_step_plot(self):
        self.ui.mplwidget_second_step.figure.clear()
        self.axes_second_step = self.ui.mplwidget_second_step.figure.add_subplot(
            111)

    def convert_time(self, d):
        date, times = d.split(" ")
        year, month, day = date.split("-")
        hour, minute, second = times.split(":")

        t = datetime.datetime(int(year), int(month), int(day), int(hour),
                              int(minute), int(second))
        t_sec = time.mktime(t.timetuple())

        return t_sec

    def closeEvent(self, question):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, 'Message', quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        # If your answer is "yes", then quit
        if reply == QMessageBox.Yes:
            question.accept()
        # If your answer is "no", then get back
        else:
            question.ignore()
Exemplo n.º 8
0
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'), lambda : self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda : self.Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_5, SIGNAL('clicked()'), lambda : self.Close("visa5", self.visa5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        
        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"), self.Print_data)
        
        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'), lambda : self.start_font("C"))
        
        
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)
        
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "black", "black", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(self.ui.curvewidget_2, "r", "black", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "g", "black", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(self.ui.curvewidget_4, "o", "black", titles = ["Plot 4", "X (x)", "Y (y)"])
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None

        self.setWindowTitle("Keithley Gate Sweep with Current Bias Lock-in")
        
        font = QFont()
        font.setPointSize(10)
        self.ui.tabWidget_visa.setFont(font)
        self.ui.tabWidget_visa.setTabText(0, "Lock-in Curr")
        self.ui.tabWidget_visa.setTabText(1, "Keithley Gate")
        
        self.input_string = []
        self.input_string.append("00 Run_Number: 01\n\n")
        
        self.input_string.append("01 Start Gate Voltage: 0\n")
        self.input_string.append("02 Step Gate Voltage: 0.1\n")
        self.input_string.append("03 End Gate Voltage: 10\n")
        self.input_string.append("04 Double Linear (y/n): y\n")
        self.input_string.append("05 Turn off Keithley (y/n): y\n\n")
        
        self.input_string.append("06 CurrBias Lock-in Sens: 5e-3\n")
        self.input_string.append("07 Current Bias Resistor: 1.015e6\n")
        self.input_string.append("08 Lock-in Output Voltage: 4e-3\n")
        self.input_string.append("09 Additional Lock-in Volt Gain: 1\n\n")
        
        self.input_string.append("10 User: Vannucci\n")
        self.input_string.append("11 Save Name: \n")
        self.input_string.append("12 TimeStamp (y/n): y\n")
        self.input_string.append("13 Comments: ")
        
        
        self.start_font("C")
        
        try:
            file_par = open("parameters.txt", "r")
            input_string_print = file_par.read()
            file_par.close()
            
        except:
            input_string_print = ""
            for st_i in self.input_string:
                input_string_print = input_string_print + st_i
        
        font = QFont()
        font.setPointSize(8)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)
        
        self.inputted_data = []
        self.array_sweep = []
Exemplo n.º 9
0
class MyForm(QMainWindow):
    
    # The __init__ function is what is everything the user wants to be initialized when the class is called.
    # Here we shall define the trig functions to corresponding variables.
    # Note that the "self" variable means that the function is part of the class and can be called inside and outside the class.(Although __init__ is special.)
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'), lambda : self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(self.ui.pushButton_update_6, SIGNAL('clicked()'), lambda : self.Update("visa6", self.ui.comboBox_visa_6))
        self.connect(self.ui.pushButton_update_7, SIGNAL('clicked()'), lambda : self.Update("visa7", self.ui.comboBox_visa_7))
        self.connect(self.ui.pushButton_update_8, SIGNAL('clicked()'), lambda : self.Update("visa8", self.ui.comboBox_visa_8))
        self.connect(self.ui.pushButton_update_9, SIGNAL('clicked()'), lambda : self.Update("visa9", self.ui.comboBox_visa_9))
        self.connect(self.ui.pushButton_update_10, SIGNAL('clicked()'), lambda : self.Update("visa10", self.ui.comboBox_visa_10))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda : self.Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_select_6, SIGNAL('clicked()'), lambda : self.Select("visa6", self.visa6, self.ui.comboBox_visa_6, self.ui.label_visa_6, [self.ui.pushButton_select_6, self.ui.pushButton_close_6]))
        self.connect(self.ui.pushButton_select_7, SIGNAL('clicked()'), lambda : self.Select("visa7", self.visa7, self.ui.comboBox_visa_7, self.ui.label_visa_7, [self.ui.pushButton_select_7, self.ui.pushButton_close_7]))
        self.connect(self.ui.pushButton_select_8, SIGNAL('clicked()'), lambda : self.Select("visa8", self.visa8, self.ui.comboBox_visa_8, self.ui.label_visa_8, [self.ui.pushButton_select_8, self.ui.pushButton_close_8]))
        self.connect(self.ui.pushButton_select_9, SIGNAL('clicked()'), lambda : self.Select("visa9", self.visa9, self.ui.comboBox_visa_9, self.ui.label_visa_9, [self.ui.pushButton_select_9, self.ui.pushButton_close_9]))
        self.connect(self.ui.pushButton_select_10, SIGNAL('clicked()'), lambda : self.Select("visa10", self.visa10, self.ui.comboBox_visa_10, self.ui.label_visa_10, [self.ui.pushButton_select_10, self.ui.pushButton_close_10]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_5, SIGNAL('clicked()'), lambda : self.Close("visa5", self.visa5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_close_6, SIGNAL('clicked()'), lambda : self.Close("visa6", self.visa6, self.ui.label_visa_6, [self.ui.pushButton_select_6, self.ui.pushButton_close_6]))
        self.connect(self.ui.pushButton_close_7, SIGNAL('clicked()'), lambda : self.Close("visa7", self.visa7, self.ui.label_visa_7, [self.ui.pushButton_select_7, self.ui.pushButton_close_7]))
        self.connect(self.ui.pushButton_close_8, SIGNAL('clicked()'), lambda : self.Close("visa8", self.visa8, self.ui.label_visa_8, [self.ui.pushButton_select_8, self.ui.pushButton_close_8]))
        self.connect(self.ui.pushButton_close_9, SIGNAL('clicked()'), lambda : self.Close("visa9", self.visa9, self.ui.label_visa_9, [self.ui.pushButton_select_9, self.ui.pushButton_close_9]))
        self.connect(self.ui.pushButton_close_10, SIGNAL('clicked()'), lambda : self.Close("visa10", self.visa10, self.ui.label_visa_10, [self.ui.pushButton_select_10, self.ui.pushButton_close_10]))
        
        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"), self.Print_data)
        
        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'), lambda : self.start_font("C"))
    
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)
        self.Update("visa6", self.ui.comboBox_visa_6)
        self.Update("visa7", self.ui.comboBox_visa_7)
        self.Update("visa8", self.ui.comboBox_visa_8)
        self.Update("visa9", self.ui.comboBox_visa_9)
        self.Update("visa10", self.ui.comboBox_visa_10)
        
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "r", "black", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(self.ui.curvewidget_2, "r", "black", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "r", "black", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(self.ui.curvewidget_4, "r", "black", titles = ["Plot 4", "X (x)", "Y (y)"])
        self.curve_5 = self.make_curveWidgets(self.ui.curvewidget_5, "r", "black", titles = ["Plot 5", "X (x)", "Y (y)"])
        self.curve_6 = self.make_curveWidgets(self.ui.curvewidget_6, "r", "black", titles = ["Plot 6", "X (x)", "Y (y)"])
        self.curve_7 = self.make_curveWidgets(self.ui.curvewidget_7, "r", "black", titles = ["Plot 7", "X (x)", "Y (y)"])
        self.curve_8 = self.make_curveWidgets(self.ui.curvewidget_8, "r", "black", titles = ["Plot 8", "X (x)", "Y (y)"])
        self.curve_9 = self.make_curveWidgets(self.ui.curvewidget_9, "r", "black", titles = ["Plot 9", "X (x)", "Y (y)"])
        self.curve_10 = self.make_curveWidgets(self.ui.curvewidget_10, "r", "black", titles = ["Plot 10", "X (x)", "Y (y)"])
        self.curve_11 = self.make_curveWidgets(self.ui.curvewidget_11, "r", "black", titles = ["Plot 11", "X (x)", "Y (y)"])
        self.curve_12 = self.make_curveWidgets(self.ui.curvewidget_12, "r", "black", titles = ["Plot 12", "X (x)", "Y (y)"])
        
        self.ui.mplwidget_1 = self.make_mplToolBar(self.ui.mplwidget_1, self.ui.widget_1)
        self.ui.mplwidget_2 = self.make_mplToolBar(self.ui.mplwidget_2, self.ui.widget_2)
        self.ui.mplwidget_3 = self.make_mplToolBar(self.ui.mplwidget_3, self.ui.widget_3)
        self.ui.mplwidget_4 = self.make_mplToolBar(self.ui.mplwidget_4, self.ui.widget_4)
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None
        self.visa6 = None
        self.visa7 = None
        self.visa8 = None
        self.visa9 = None
        self.visa10 = None
        
        # Set the visa names of the tabWidget
        # font = QFont()
        # font.setPointSize(8)
        # self.ui.tabWidget_visa.setFont(font)
        # self.ui.tabWidget_visa.setTabText(0, "Visa Name 1")
        # self.ui.tabWidget_visa.setTabText(1, "Visa Name 2")
        # self.ui.tabWidget_visa.setTabText(2, "Visa Name 3")
        # self.ui.tabWidget_visa.setTabText(3, "Visa Name 4")
        # self.ui.tabWidget_visa.setTabText(4, "Visa Name 5")
        # self.ui.tabWidget_visa.setTabText(5, "Visa Name 6")
        # self.ui.tabWidget_visa.setTabText(6, "Visa Name 7")
        # self.ui.tabWidget_visa.setTabText(7, "Visa Name 8")
        # self.ui.tabWidget_visa.setTabText(8, "Visa Name 9")
        # self.ui.tabWidget_visa.setTabText(8, "Visa Name 10")
        
        # Input parameters by string
        # Remember to keep the last one as "Comments: ..." so that the program could see that as an end.
        self.input_string = []
        # self.input_string.append("00 Parameter: 1\n")
        # self.input_string.append("01 Parameter: 1\n")
        # self.input_string.append("02 Parameter: 1\n")
        # self.input_string.append("03 Parameter: 1\n")
        # self.input_string.append("04 Parameter: 1\n")
        # self.input_string.append("05 Parameter: 1\n")
        # self.input_string.append("06 Parameter: 1\n")
        # self.input_string.append("07 Parameter: 1\n")
        # self.input_string.append("08 Parameter: 1\n")
        # self.input_string.append("09 Parameter: 1\n")
        # self.input_string.append("10 Parameter: 1\n")
        # self.input_string.append("11 Parameter: 1\n")
        # self.input_string.append("12 Parameter: 1\n")
        # self.input_string.append("13 Comments: ")
        
        
        self.start_font("C")
        
        try:
            file_par = open("parameters.txt", "r")
            input_string_print = file_par.read()
            file_par.close()
            
        except:
            input_string_print = ""
            for st_i in self.input_string:
                input_string_print = input_string_print + st_i
            
        font = QFont()
        font.setPointSize(8)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)
        
        self.inputted_data = []
        self.array_sweep = []
    
    def start_font(self, status):
        if status == "C":
            font = QFont()
            font.setPointSize(8)
            self.ui.pushButton_Start.setFont(font)
            self.ui.pushButton_Start.setText("Check Array")
        elif status == "S":
            font = QFont()
            font.setPointSize(12)
            self.ui.pushButton_Start.setFont(font)
            self.ui.pushButton_Start.setText("Start")
    
    def Update(self, signal, comboBoxVisa):
        rm = visa.ResourceManager()
        try:
            alls = rm.list_resources()
        except:
            alls = "No Visa Available."
        if signal == "visa1":
            self.ui.comboBox_visa_1.clear()
        elif signal == "visa2":
            self.ui.comboBox_visa_2.clear()
        elif signal == "visa3":
            self.ui.comboBox_visa_3.clear()
        elif signal == "visa4":
            self.ui.comboBox_visa_4.clear()
        elif signal == "visa5":
            self.ui.comboBox_visa_5.clear()
        elif signal == "visa6":
            self.ui.comboBox_visa_6.clear()
        elif signal == "visa7":
            self.ui.comboBox_visa_7.clear()
        elif signal == "visa8":
            self.ui.comboBox_visa_8.clear()
        elif signal == "visa9":
            self.ui.comboBox_visa_9.clear()
        elif signal == "visa10":
            self.ui.comboBox_visa_10.clear()
        
        for temp in alls:
            if signal == "visa1":
                self.ui.comboBox_visa_1.addItem(temp)
            elif signal == "visa2":
                self.ui.comboBox_visa_2.addItem(temp)
            elif signal == "visa3":
                self.ui.comboBox_visa_3.addItem(temp)
            elif signal == "visa4":
                self.ui.comboBox_visa_4.addItem(temp)
            elif signal == "visa5":
                self.ui.comboBox_visa_5.addItem(temp)
            elif signal == "visa6":
                self.ui.comboBox_visa_6.addItem(temp)
            elif signal == "visa7":
                self.ui.comboBox_visa_7.addItem(temp)
            elif signal == "visa8":
                self.ui.comboBox_visa_8.addItem(temp)
            elif signal == "visa9":
                self.ui.comboBox_visa_9.addItem(temp)
            elif signal == "visa10":
                self.ui.comboBox_visa_10.addItem(temp)
    
    def Select(self, signal, visa_chosen, comboBoxVisa, lineEditVisa, selectClose, baud = None):
        visa_address = str(comboBoxVisa.currentText())
        rm = visa.ResourceManager()
        rm.list_resources()
        if baud == None:
            inst = rm.open_resource(visa_address)
        else:
            inst = rm.open_resource(visa_address, baud_rate = baud)
        visa_check = self.Check(inst)
        if visa_check == True:
            self.ui.label_condition.setText("Visa is selected succefully!")
            visa_name = inst.query("*IDN?")
            name_list = visa_name.split(',')
            first_name = name_list[0]
            if signal == "visa1":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa1 = inst
            elif signal == "visa2":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa2 = inst
            elif signal == "visa3":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa3 = inst
            elif signal == "visa4":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa4 = inst
            elif signal == "visa5":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa5 = inst
            elif signal == "visa6":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa6 = inst
            elif signal == "visa7":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa7 = inst
            elif signal == "visa8":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa8 = inst
            elif signal == "visa9":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa9 = inst
            elif signal == "visa10":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.visa10 = inst
        elif visa_check == False:
            self.ui.label_condition.setText("Invalid visa address.")
            lineEditVisa.setText("None.")
            visa_chosen = False
    
    def Close(self, signal, visa_chosen, lineEditVisa, selectClose):
        self.ui.label_condition.setText('Visa address is closed')
        lineEditVisa.setText('')
        selectClose[0].setEnabled(True)
        selectClose[1].setEnabled(False)
        if signal == "visa1":
            visa_chosen.close()
            self.visa1 = None
        elif signal == "visa2":
            visa_chosen.close()
            self.visa2 = None
        elif signal == "visa3":
            visa_chosen.close()
            self.visa3 = None
        elif signal == "visa4":
            visa_chosen.close()
            self.visa4 = None
        elif signal == "visa5":
            visa_chosen.close()
            self.visa5 = None
        elif signal == "visa6":
            visa_chosen.close()
            self.visa6 = None
        elif signal == "visa7":
            visa_chosen.close()
            self.visa7 = None
        elif signal == "visa8":
            visa_chosen.close()
            self.visa8 = None
        elif signal == "visa9":
            visa_chosen.close()
            self.visa9 = None
        elif signal == "visa10":
            visa_chosen.close()
            self.visa10 = None
            
            
    def Check(self, inst):
        try:
            inst.ask("*IDN?")
            valid = True
        except:
            valid = False
        return valid
    
    def make_curveWidgets(self, curvewidget, color, markerColor, titles):
        curve_temp = make.curve([], [], color = color, marker = "o", markerfacecolor = markerColor, markersize = 5)
        curvewidget.plot.add_item(curve_temp)
        curvewidget.plot.set_antialiasing(True)
        curvewidget.plot.set_titles(titles[0], titles[1], titles[2])
        return curve_temp
    
    def make_mplToolBar(self, mplwidget, widget):
        canvas_mpl = FigureCanvas(mplwidget.figure)
        canvas_mpl.setParent(widget)
        # This is the toolbar widget for the import canvas
        mpl_toolbar = NavigationToolbar(canvas_mpl, mplwidget)
        vbox_ = QVBoxLayout()
        # The matplotlib canvas
        vbox_.addWidget(canvas_mpl)
        # The matplotlib toolbar
        vbox_.addWidget(mpl_toolbar)
        widget.setLayout(vbox_)
        return canvas_mpl
    
    def mplPlots(self):
        self.ui.tabWidget.setCurrentIndex(1)
        self.ui.mplwidget_1.draw()
        self.ui.mplwidget_2.draw()
        self.ui.mplwidget_3.draw()
        self.ui.mplwidget_4.draw()

    def plot_reset(self, axes, mplwidget):
        mplwidget.figure.clear()
        axes = mplwidget.figure.add_subplot(111)
        return axes
    
    def plot_data(self, axes, x, y, titles, mplwidget):
        axes = self.plot_reset(axes, mplwidget)
        axes.plot(x, y, marker = '.', linestyle = '-')
        axes.grid()
        axes.set_title(titles[0])
        axes.set_xlabel(titles[1])
        axes.set_ylabel(titles[2])
        mplwidget.draw()
        
    def start(self):
        instruments = [self.visa1, self.visa2, self.visa3, self.visa4, self.visa5]
        curves = [self.curve_1, self.curve_2, self.curve_3, self.curve_4]
        curveWidgets =[self.ui.curvewidget_1, self.ui.curvewidget_2, self.ui.curvewidget_3, self.ui.curvewidget_4]
        go_on = True
        
        inputted_string = str(self.ui.textEdit.toPlainText())
        input_data = inputted_string.split("Comments:")
        comments = input_data[1]
        
        input_data = input_data[0].split("\n")
        
        inputted_data = []
        
        for i in range(0, len(input_data)):
            for j in range(0, len(self.input_string)):
                if len(input_data[i].split(self.input_string[j].split(":")[0])) > 1:
                    inputted_data.append(input_data[i].split(self.input_string[j].split(":")[0])[1].replace(":", "").replace(" ", ""))
                    print inputted_data
        
        #Make Array
        
        array_start = float(inputted_data[1])
        array_step = float(inputted_data[3])
        array_stop = float(inputted_data[4])
        double_lin = inputted_data[5].upper()
        
        if array_start < array_stop:
            array_1 = numpy.arange(array_start, array_stop + array_step, array_step)
        elif array_start > array_stop:
            array_1 = numpy.arange(array_stop, array_start + array_step, array_step)[::-1]
        
        
        if double_lin == "Y":
            array_2 = array_1[::-1]
            array_sweep = numpy.append(array_1, array_2)
        else:
             array_sweep = array_1
        
        run_program = True
        for i in range(0, len(inputted_data)):
            try:
                if self.inputted_data[i] != inputted_data[i]:
                    run_program = False
                    self.start_font("C")
            except:
                run_program = False
                self.start_font("C")
        
        for i in range(0, len(array_sweep)):
            try:
                if self.array_sweep[i] != array_sweep[i]:
                    run_program = False
                    self.start_font("C")
            except:
                run_program = False
                self.start_font("C")
        
        if not run_program:
            self.inputted_data = inputted_data
            self.array_sweep = array_sweep
            self.ui.curvewidget_1.plot.set_titles("Magnetic Field Steps", "Steps", "Field (T)")
            self.curve_1.set_data(range(0, len(self.array_sweep)), self.array_sweep)
            self.ui.curvewidget_1.plot.do_autoscale()
            self.curve_1.plot().replot()
            self.start_font("S")
            
        elif run_program:
            if str(self.ui.pushButton_Start.text()) == "Start":
                self.start_font("C")
                self.collect_data_thread.input(self.ui, instruments, curves, curveWidgets, go_on, inputted_data, comments, array_sweep, inputted_string, self.input_string)
            else:
                self.start_font("S")

    def curvePlots_update(self, curveInfo):
        curveWidget = curveInfo[0]
        curve = curveInfo[1]
        curveWidget.plot.do_autoscale()
        curve.plot().replot()
    
    def Print_data(self, display_text):
        
        font = QFont()
        font.setPointSize(8)
        self.ui.textEditDisplay.setFont(font)
        self.ui.textEditDisplay.setText(display_text)
    
    def Switch_scale(self, num):
        temp = abs(num)
        if temp >= 1E9:
            scale = [1E-9, "G"]
        elif temp >= 1E6 and temp < 1E9:
            scale = [1E-6, "M"]
        elif temp >= 1E3 and temp < 1E6:
            scale = [1E-3, "k"]
        elif temp >= 1 and temp < 1000:
            scale = [1, ""]
        elif temp >= 1E-3 and temp < 1:
            scale = [1E3, "m"]
        elif temp >= 1E-6 and temp < 1E-3:
            scale = [1E6, "u"]
        elif temp >= 1E-9 and temp < 1E-6:
            scale = [1E9, "n"]
        elif temp < 1E-9:
            scale = [1E12, "p"]
        return scale
    
    def closeEvent(self, event):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg, QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()
Exemplo n.º 10
0
class MyForm(QMainWindow):
    
    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent = None):
        
        # Standard GUI code
        QWidget.__init__(self, parent)
        
        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general, self.ui.widget_general)
        
        self.canvas_import = FigureCanvas(self.ui.mplwidget_import.figure)
        self.canvas_import.setParent(self.ui.widget_import)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import = NavigationToolbar(self.canvas_import, self.ui.widget_import)
        
        self.canvas_scan = FigureCanvas(self.ui.mplwidget_scan.figure)
        self.canvas_scan.setParent(self.ui.widget_scan)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_scan = NavigationToolbar(self.canvas_scan, self.ui.widget_scan)
        
        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)
        
        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import = QVBoxLayout()
        # The matplotlib canvas
        vbox_import.addWidget(self.canvas_import)
        # The matplotlib toolbar
        vbox_import.addWidget(self.mpl_toolbar_import)
        self.ui.widget_import.setLayout(vbox_import)
        
        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_scan = QVBoxLayout()
        # The matplotlib canvas
        vbox_scan.addWidget(self.canvas_scan)
        # The matplotlib toolbar
        vbox_scan.addWidget(self.mpl_toolbar_scan)
        self.ui.widget_scan.setLayout(vbox_scan)
        
        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general
        self.ui.mplwidget_import = self.canvas_import
        self.ui.mplwidget_scan = self.canvas_scan
        
        self.General_programs = General(self.ui)
        # Connect buttons in the General Array tab
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.General_programs.Plot_general)
        self.connect(self.ui.pushButton_clear, SIGNAL('clicked()'), self.General_programs.Clear)
        self.connect(self.ui.pushButton_save, SIGNAL('clicked()'), self.General_programs.Save)
        self.connect(self.ui.pushButton_browse_save, SIGNAL('clicked()'), self.General_programs.Browse_save)
        
        self.Keithley_programs = Keithley(self.ui)
        self.Keithley_programs.Refresh_visa()
        # Connect buttons in the Keithely tab
        self.connect(self.ui.pushButton_browse_keithley, SIGNAL('clicked()'), self.Keithley_programs.Browse_keithley)
        self.connect(self.ui.pushButton_import_keithley, SIGNAL('clicked()'), self.Keithley_programs.Import_keithley)
        self.connect(self.ui.pushButton_close_keithley, SIGNAL('clicked()'), self.Keithley_programs.Close_keithley)
        self.connect(self.ui.pushButton_select_keithley, SIGNAL('clicked()'), self.Keithley_programs.Select_keithley)
        self.connect(self.ui.pushButton_scan_keithley, SIGNAL('clicked()'), self.Keithley_programs.Scan_keithley)
        self.connect(self.ui.pushButton_stop_keithley, SIGNAL('clicked()'), self.Keithley_programs.Stop_keithley)
        self.connect(self.ui.pushButton_clear_keithley, SIGNAL('clicked()'), self.Keithley_programs.Clear_keithley)
        
        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0
        
        self.ui.lineEdit_directory_save.setText(os.getcwd())

    # Make sure the user is going to quit the program
    def closeEvent(self, event):
        quit_msg = "DO you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg, QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            question.accept()
        else:
            question.ignore()
Exemplo n.º 11
0
class MyForm(QMainWindow):
    
    # The __init__ function is what is everything the user wants to be initialized when the class is called.
    # Here we shall define the trig functions to corresponding variables.
    # Note that the "self" variable means that the function is part of the class and can be called inside and outside the class.(Although __init__ is special.)
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'), lambda : self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda : self.Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_5, SIGNAL('clicked()'), lambda : self.Close("visa5", self.visa5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        
        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"), self.Print_data)
        
        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'), lambda : self.start_font("C"))
        
        
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)
        
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "black", "black", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(self.ui.curvewidget_2, "r", "black", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "g", "black", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(self.ui.curvewidget_4, "o", "black", titles = ["Plot 4", "X (x)", "Y (y)"])
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None

        self.setWindowTitle("Keithley Gate Sweep with Current Bias Lock-in")
        
        font = QFont()
        font.setPointSize(10)
        self.ui.tabWidget_visa.setFont(font)
        self.ui.tabWidget_visa.setTabText(0, "Lock-in Curr")
        self.ui.tabWidget_visa.setTabText(1, "Keithley Gate")
        
        self.input_string = []
        self.input_string.append("00 Run_Number: 01\n\n")
        
        self.input_string.append("01 Start Gate Voltage: 0\n")
        self.input_string.append("02 Step Gate Voltage: 0.1\n")
        self.input_string.append("03 End Gate Voltage: 10\n")
        self.input_string.append("04 Double Linear (y/n): y\n")
        self.input_string.append("05 Turn off Keithley (y/n): y\n\n")
        
        self.input_string.append("06 CurrBias Lock-in Sens: 5e-3\n")
        self.input_string.append("07 Current Bias Resistor: 1.015e6\n")
        self.input_string.append("08 Lock-in Output Voltage: 4e-3\n")
        self.input_string.append("09 Additional Lock-in Volt Gain: 1\n\n")
        
        self.input_string.append("10 User: Vannucci\n")
        self.input_string.append("11 Save Name: \n")
        self.input_string.append("12 TimeStamp (y/n): y\n")
        self.input_string.append("13 Comments: ")
        
        
        self.start_font("C")
        
        try:
            file_par = open("parameters.txt", "r")
            input_string_print = file_par.read()
            file_par.close()
            
        except:
            input_string_print = ""
            for st_i in self.input_string:
                input_string_print = input_string_print + st_i
        
        font = QFont()
        font.setPointSize(8)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)
        
        self.inputted_data = []
        self.array_sweep = []
        
    
    def start_font(self, status):
        if status == "C":
            font = QFont()
            font.setPointSize(8)
            self.ui.pushButton_Start.setFont(font)
            self.ui.pushButton_Start.setText("Check Array")
        elif status == "S":
            font = QFont()
            font.setPointSize(12)
            self.ui.pushButton_Start.setFont(font)
            self.ui.pushButton_Start.setText("Start")
    
    def Update(self, signal, comboBoxVisa):
        rm = visa.ResourceManager()
        try:
            alls = rm.list_resources()
        except:
            alls = "No Visa Available."
        if signal == "visa1":
            self.ui.comboBox_visa_1.clear()
        elif signal == "visa2":
            self.ui.comboBox_visa_2.clear()
        elif signal == "visa3":
            self.ui.comboBox_visa_3.clear()
        elif signal == "visa4":
            self.ui.comboBox_visa_4.clear()
        elif signal == "visa5":
            self.ui.comboBox_visa_5.clear()

        
        for temp in alls:
            if signal == "visa1":
                self.ui.comboBox_visa_1.addItem(temp)
            elif signal == "visa2":
                self.ui.comboBox_visa_2.addItem(temp)
            elif signal == "visa3":
                self.ui.comboBox_visa_3.addItem(temp)
            elif signal == "visa4":
                self.ui.comboBox_visa_4.addItem(temp)
            elif signal == "visa5":
                self.ui.comboBox_visa_5.addItem(temp)
    
    def Select(self, signal, visa_chosen, comboBoxVisa, lineEditVisa, selectClose):
        visa_address = str(comboBoxVisa.currentText())
        rm = visa.ResourceManager()
        rm.list_resources()
        inst = rm.open_resource(visa_address)
        #visa_check = self.Check(inst)
        visa_check = True
        if visa_check == True:
            self.ui.label_condition.setText("Visa is selected succefully!")
            if signal == "visa1":
                visa_name = inst.query("*IDN?")
                print visa_name
                name_list = visa_name.split(',')
                first_name = name_list[0]
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                #self.Enable(signal)
                self.visa1 = inst
            elif signal == "visa2":
                visa_name = inst.query("*IDN?\n")
                print visa_name
                name_list = visa_name.split(',')
                first_name = name_list[0]
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                #self.Enable(signal)
                self.visa2 = inst
            elif signal == "visa3":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                #self.Enable(signal)
                self.visa3 = inst
            elif signal == "visa4":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                #self.Enable(signal)
                self.visa4 = inst
            elif signal == "visa5":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                #self.Enable(signal)
                self.visa5 = inst
        elif visa_check == False:
            self.ui.label_condition.setText("Invalid visa address.")
            lineEditVisa.setText("None.")
            visa_chosen = False
    
    def Close(self, signal, visa_chosen, lineEditVisa, selectClose):
        self.ui.label_condition.setText('Visa address is closed')
        lineEditVisa.setText('')
        selectClose[0].setEnabled(True)
        selectClose[1].setEnabled(False)
        if signal == "visa1":
            visa_chosen.close()
            self.visa1 = None
            self.Disable(signal)
        elif signal == "visa2":
            visa_chosen.close()
            self.visa2 = None
            self.Disable(signal)
        elif signal == "visa3":
            visa_chosen.close()
            self.visa3 = None
            self.Disable(signal)
        elif signal == "visa4":
            visa_chosen.close()
            self.visa4 = None
            self.Disable(signal)
        elif signal == "visa5":
            visa_chosen.close()
            self.visa5 = None
            self.Disable(signal)
            
    def Check(self, inst):
        try:
            inst.ask("*IDN?")
            valid = True
        except:
            valid = False
        return valid
    
    def make_curveWidgets(self, curvewidget, color, markerColor, titles):
        curve_temp = make.curve([], [], color = color, marker = "o", markerfacecolor = markerColor, markersize = 5)
        curvewidget.plot.add_item(curve_temp)
        curvewidget.plot.set_antialiasing(True)
        curvewidget.plot.set_titles(titles[0], titles[1], titles[2])
        return curve_temp
    
    def start(self):
        instruments = [self.visa1, self.visa2, self.visa3, self.visa4, self.visa5]
        curves = [self.curve_1, self.curve_2, self.curve_3, self.curve_4]
        curveWidgets =[self.ui.curvewidget_1, self.ui.curvewidget_2, self.ui.curvewidget_3, self.ui.curvewidget_4]
        go_on = True
        
        inputted_string = str(self.ui.textEdit.toPlainText())
        input_data = inputted_string.split("Comments:")
        comments = input_data[1]
        
        input_data = input_data[0].split("\n")
        
        inputted_data = []
        
        for i in range(0, len(input_data)):
            for j in range(0, len(self.input_string)):
                if len(input_data[i].split(self.input_string[j].split(":")[0])) > 1:
                    inputted_data.append(input_data[i].split(self.input_string[j].split(":")[0])[1].replace(":", "").replace(" ", ""))
                    print inputted_data
        
        #Make Array
        
        if float(inputted_data[1]) < float(inputted_data[3]):
            array_1 = numpy.arange(float(inputted_data[1]), float(inputted_data[3]) + float(inputted_data[2]), float(inputted_data[2]))
        elif float(inputted_data[1]) > float(inputted_data[3]):
            array_1 = numpy.arange(float(inputted_data[3]), float(inputted_data[1]) + float(inputted_data[2]), float(inputted_data[2]))[::-1]
        
        
        if inputted_data[4].upper() == "Y":
            array_2 = array_1[::-1]
            array_sweep = numpy.append(array_1, array_2)
        else:
             array_sweep = array_1
        
        run_program = True
        for i in range(0, len(inputted_data)):
            try:
                if self.inputted_data[i] != inputted_data[i]:
                    run_program = False
                    self.start_font("C")
            except:
                run_program = False
                self.start_font("C")
        
        for i in range(0, len(array_sweep)):
            try:
                if self.array_sweep[i] != array_sweep[i]:
                    run_program = False
                    self.start_font("C")
            except:
                run_program = False
                self.start_font("C")
        
        if not run_program:
            self.inputted_data = inputted_data
            self.array_sweep = array_sweep
            self.ui.curvewidget_1.plot.set_titles("Gate Voltage Steps", "Steps", "Gate Voltage (V)")
            self.curve_1.set_data(range(0, len(self.array_sweep)), self.array_sweep)
            self.ui.curvewidget_1.plot.do_autoscale()
            self.curve_1.plot().replot()
            self.start_font("S")
            
        elif run_program:
            if str(self.ui.pushButton_Start.text()) == "Start":
                self.start_font("C")
                self.collect_data_thread.input(self.ui, instruments, curves, curveWidgets, go_on, inputted_data, comments, array_sweep, inputted_string, self.input_string)
            else:
                self.start_font("S")

    def curvePlots_update(self, curveInfo):
        curveWidget = curveInfo[0]
        curve = curveInfo[1]
        curveWidget.plot.do_autoscale()
        curve.plot().replot()
    
    def Print_data(self, display_text):
        
        font = QFont()
        font.setPointSize(8)
        self.ui.textEditDisplay.setFont(font)
        self.ui.textEditDisplay.setText(display_text)
        
        """self.ui.label_step.setText(step)
        self.ui.label_time.setText(time)
        self.ui.label_V1.setText(V1)
        self.ui.label_V2.setText(V2)
        self.ui.label_V3.setText(V3)
        self.ui.label_V4.setText(V4)
        self.ui.label_V5.setText(V5)
        self.ui.label_V6.setText(V6)"""
        
    
    def Switch_scale(self, num):
        temp = abs(num)
        if temp >= 1E9:
            scale = [1E-9, "G"]
        elif temp >= 1E6 and temp < 1E9:
            scale = [1E-6, "M"]
        elif temp >= 1E3 and temp < 1E6:
            scale = [1E-3, "k"]
        elif temp >= 1 and temp < 1000:
            scale = [1, ""]
        elif temp >= 1E-3 and temp < 1:
            scale = [1E3, "m"]
        elif temp >= 1E-6 and temp < 1E-3:
            scale = [1E6, "u"]
        elif temp >= 1E-9 and temp < 1E-6:
            scale = [1E9, "n"]
        elif temp < 1E-9:
            scale = [1E12, "p"]
            
        return scale
    
    def closeEvent(self, event):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg, QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()
Exemplo n.º 12
0
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # When you open the window, the visas get refreshed
        self.refresh_visa()

        # This is to check whether user click start before they save
        self.data_check = False

        # We set up a QTimer variable called collect_timer.
        # It runs as a countdown clock that runs in the background and emits a timeout signal when it reaches 0.
        # Because this variable, we do not to use a for loop function for collecting and timing since the for loop can not be stopped in progress and can cause program frozen.
        # Also it allows the rest of the commands to run normally
        self.collect_timer = QTimer()

        # Set up guiqwt plot
        self.curve_item = make.curve([], [], color='r')
        self.ui.curvewidgetPlot.plot.add_item(self.curve_item)
        self.ui.curvewidgetPlot.plot.set_antialiasing(True)
        self.ui.curvewidgetPlot.plot.set_titles(
            "Agilent 34460A Digit Multimeter", "Time (s)", "Y-Axis")

        # The following connects the signals sent from an action on the GUI to a function that does something
        # When the button is clicked, it does the corresponding function
        self.connect(self.ui.pushButton_Select_Visa, SIGNAL("clicked()"),
                     self.choose_visa)
        self.connect(self.ui.pushButton_Start, SIGNAL("clicked()"), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL("clicked()"), self.stop)
        self.connect(self.ui.pushButton_Refresh, SIGNAL("clicked()"),
                     self.refresh_visa)

        # This works as a clock running in the background, different functions can use this clock at the same time
        # It is better than .sleep() function because during the sleep time, the whole program is frozen and cannot do anything else
        # Everytime collect_timer gets the time_step, it sends a "timeout()" signal to the device and runs the collect function
        self.connect(self.collect_timer, SIGNAL("timeout()"), self.collect)

        self.connect(self.ui.comboBox_Meas_Type, SIGNAL("clicked()"),
                     self.meas_type)

        # To give a click signal to the checkBox
        self.connect(self.ui.checkBox_Continue, SIGNAL("clicked()"),
                     self.continue_check)

        # I add a slect button for the measurement type
        # Only if you click the button, the computer knows your measurement type
        # It shows the current measurement type and warning at the same time
        self.connect(self.ui.pushButton_Select_Measurement,
                     SIGNAL("clicked()"), self.choose_meas)

        # To save the data
        self.connect(self.ui.pushButton_Save, SIGNAL("clicked()"), self.save)

        # To see and change the directory
        self.connect(self.ui.pushButton_Browse, SIGNAL("clicked()"),
                     self.browse)

        # To choose a name folder to save file
        self.connect(self.ui.pushButton_Select_Directory, SIGNAL("clicked()"),
                     self.select_directory)

        # To select name for the file, date and time or custom name entered by the user
        self.connect(self.ui.radioButton_Timename, SIGNAL("clicked()"),
                     self.select_name)
        self.connect(self.ui.radioButton_Custom_Name, SIGNAL("clicked()"),
                     self.select_name)

        # To select a file type, either .csv or .txt
        self.connect(self.ui.radioButton_csv, SIGNAL('clicked()'),
                     self.select_type)
        self.connect(self.ui.radioButton_csv, SIGNAL('clicked()'),
                     self.select_type)
Exemplo n.º 13
0
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general,
                                                     self.ui.widget_general)

        self.canvas_import = FigureCanvas(self.ui.mplwidget_import.figure)
        self.canvas_import.setParent(self.ui.widget_import)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import = NavigationToolbar(self.canvas_import,
                                                    self.ui.widget_import)

        self.canvas_analysis = FigureCanvas(self.ui.mplwidget_analysis.figure)
        self.canvas_analysis.setParent(self.ui.widget_analysis)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_analysis = NavigationToolbar(self.canvas_analysis,
                                                      self.ui.widget_analysis)

        self.canvas_ct_analysis = FigureCanvas(
            self.ui.mplwidget_ct_analysis.figure)
        self.canvas_ct_analysis.setParent(self.ui.widget_ct_analysis)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_ct_analysis = NavigationToolbar(
            self.canvas_ct_analysis, self.ui.widget_ct_analysis)

        self.canvas_vt_analysis = FigureCanvas(
            self.ui.mplwidget_vt_analysis.figure)
        self.canvas_vt_analysis.setParent(self.ui.widget_vt_analysis)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_vt_analysis = NavigationToolbar(
            self.canvas_vt_analysis, self.ui.widget_vt_analysis)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import = QVBoxLayout()
        # The matplotlib canvas
        vbox_import.addWidget(self.canvas_import)
        # The matplotlib toolbar
        vbox_import.addWidget(self.mpl_toolbar_import)
        self.ui.widget_import.setLayout(vbox_import)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_analysis = QVBoxLayout()
        # The matplotlib canvas
        vbox_analysis.addWidget(self.canvas_analysis)
        # The matplotlib toolbar
        vbox_analysis.addWidget(self.mpl_toolbar_analysis)
        self.ui.widget_analysis.setLayout(vbox_analysis)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_ct_analysis = QVBoxLayout()
        # The matplotlib canvas
        vbox_ct_analysis.addWidget(self.canvas_ct_analysis)
        # The matplotlib toolbar
        vbox_ct_analysis.addWidget(self.mpl_toolbar_ct_analysis)
        self.ui.widget_ct_analysis.setLayout(vbox_ct_analysis)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_vt_analysis = QVBoxLayout()
        # The matplotlib canvas
        vbox_vt_analysis.addWidget(self.canvas_vt_analysis)
        # The matplotlib toolbar
        vbox_vt_analysis.addWidget(self.mpl_toolbar_vt_analysis)
        self.ui.widget_vt_analysis.setLayout(vbox_vt_analysis)

        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general
        self.ui.mplwidget_import = self.canvas_import
        self.ui.mplwidget_analysis = self.canvas_analysis
        self.ui.mplwidget_ct_analysis = self.canvas_ct_analysis
        self.ui.mplwidget_vt_analysis = self.canvas_vt_analysis

        self.connect(self.ui.actionArray_Builder, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(0))
        self.connect(self.ui.actionKeithley_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(1))
        self.connect(self.ui.actionAgilent_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(2))
        self.connect(self.ui.actionLockIn_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(3))
        self.connect(self.ui.actionResonant_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(4))

        self.Array_Builder_programs = Array_Builder(main=self, ui=self.ui)
        # Connect buttons in the General Array tab
        # self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.Array_Builder_programs.Plot_general)
        # self.connect(self.ui.pushButton_clear, SIGNAL('clicked()'), self.Array_Builder_programs.Clear)
        # self.connect(self.ui.pushButton_save, SIGNAL('clicked()'), self.Array_Builder_programs.Save)
        # self.connect(self.ui.pushButton_browse_save, SIGNAL('clicked()'), self.Array_Builder_programs.Browse_save)

        self.Keithley_programs = Keithley(main=self, ui=self.ui)
        self.Keithley_programs.Refresh_visa()
        # Connect buttons in the Keithely tab
        # self.connect(self.ui.pushButton_browse_keithley, SIGNAL('clicked()'), self.Keithley_programs.Browse_keithley)
        # self.connect(self.ui.pushButton_import_keithley, SIGNAL('clicked()'), self.Keithley_programs.Import_keithley)
        # self.connect(self.ui.pushButton_close_keithley, SIGNAL('clicked()'), self.Keithley_programs.Close_keithley)
        # self.connect(self.ui.pushButton_select_keithley, SIGNAL('clicked()'), self.Keithley_programs.Select_keithley)
        # self.connect(self.ui.pushButton_scan_keithley, SIGNAL('clicked()'), self.Keithley_programs.Scan_keithley)
        # self.connect(self.ui.pushButton_pause_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.pause)
        # self.connect(self.ui.pushButton_stop_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.stop)
        # self.connect(self.ui.pushButton_clear_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.Clear_keithley)
        # self.connect(self.ui.pushButton_copy_keithley, SIGNAL('clicked()'), self.CopyData)
        # self.connect(self.ui.pushButton_fit_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.Fit)
        # self.connect(self.ui.radioButton_voltage_keithley, SIGNAL("clicked()"), lambda : self.ui.tabWidget_scan_keithley.setCurrentIndex(0))
        # self.connect(self.ui.radioButton_timescan_keithley, SIGNAL("clicked()"), self.Keithley_programs.collect_data_thread.Pre_MPL_Plot)
        # self.connect(self.ui.radioButton_stepscan_keithley, SIGNAL("clicked()"), self.Keithley_programs.collect_data_thread.Pre_MPL_Plot)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("plot"), self.Keithley_programs.Plot_data)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("mpl_plot"), self.Keithley_programs.Plot_analysis)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("data_available"), self.Keithley_programs.Pre_save)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("clear_plot"), self.Keithley_programs.Clear_plot)

        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0

        self.ui.lineEdit_directory_save.setText(os.getcwd())

        #Agilent_Yokogawa
        self.canvas_import_ay = FigureCanvas(
            self.ui.mplwidget_import_ay.figure)
        self.canvas_import_ay.setParent(self.ui.widget_import_ay)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import_ay = NavigationToolbar(
            self.canvas_import_ay, self.ui.widget_import_ay)

        #Agilent_Yokogawa
        self.canvas_analysis_ay = FigureCanvas(
            self.ui.mplwidget_analysis_ay.figure)
        self.canvas_analysis_ay.setParent(self.ui.widget_analysis_ay)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_analysis_ay = NavigationToolbar(
            self.canvas_analysis_ay, self.ui.widget_analysis_ay)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import_ay = QVBoxLayout()
        # The matplotlib canvas
        vbox_import_ay.addWidget(self.canvas_import_ay)
        # The matplotlib toolbar
        vbox_import_ay.addWidget(self.mpl_toolbar_import_ay)
        self.ui.widget_import_ay.setLayout(vbox_import_ay)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_analysis_ay = QVBoxLayout()
        # The matplotlib canvas
        vbox_analysis_ay.addWidget(self.canvas_analysis_ay)
        # The matplotlib toolbar
        vbox_analysis_ay.addWidget(self.mpl_toolbar_analysis_ay)
        self.ui.widget_analysis_ay.setLayout(vbox_analysis_ay)

        # Connect the mplwidget with canvass
        self.ui.mplwidget_import_ay = self.canvas_import_ay
        self.ui.mplwidget_analysis_ay = self.canvas_analysis_ay

        # Connect buttons in the Agilent Yokogawa tab
        self.Agilent_Yokogawa_programs = Agilent_Yokogawa(self.ui)

        self.connect(self.ui.startButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.start)
        self.connect(self.ui.stopButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.stop)
        #self.connect(self.action_timer, SIGNAL("timeout()"), self.Agilent_Yokogawa_programs.action)

        self.connect(self.ui.pushButton_browse_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Browse_ay)
        self.connect(self.ui.pushButton_import_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Import_ay)
        self.connect(self.ui.pushButton_copy_ay, SIGNAL('clicked()'),
                     self.Copy_ay)
        self.connect(self.ui.selectVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.select_visa)
        self.connect(self.ui.updateVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.update_visa)
        self.connect(self.ui.closeVisaButton0, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa0)
        self.connect(self.ui.closeVisaButton1, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa1)
        self.connect(self.Agilent_Yokogawa_programs.collectDataThread,
                     SIGNAL("plot"), self.Agilent_Yokogawa_programs.plotData)
        self.connect(self.Agilent_Yokogawa_programs.collectDataThread,
                     SIGNAL("analyse"), self.Agilent_Yokogawa_programs.analyse)

        self.connect(self.ui.browseButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.browse)
        self.connect(self.ui.saveButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.save)

        self.ui.mplwidget_analysis_ay.figure.canvas.mpl_connect(
            'button_release_event', self.Agilent_Yokogawa_programs.slope)

        #Connects the buttons in the Resonant Sweeper Tab
        #self.Resonant_programs = Resonance_Sweep(main = self, ui = self.ui)
        self.Resonant_Sweeper_functions = Resonance_Sweep(main=self,
                                                          ui=self.ui)

        self.Resonant_Sweeper_functions.update_visa()
        self.connect(self.ui.pushButtonSelectRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.choose_visa)
        self.connect(self.ui.pushButtonUpdateRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.update_visa)
        self.connect(self.ui.pushButtonSourceSelectRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.choose_visa)
        self.connect(self.ui.pushButtonSourceUpdateRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.update_visa)
        self.connect(self.ui.pushButtonStartRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.start)
        self.connect(self.ui.pushButtonStopRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.stop)
        self.connect(self.ui.pushButtonPauseRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.collectDataThread.pause)
        self.connect(self.Resonant_Sweeper_functions.collectDataThread,
                     SIGNAL("plot"), self.Resonant_Sweeper_functions.plotData)
        self.connect(self.Resonant_Sweeper_functions.collectDataThread,
                     SIGNAL("stop"), self.Resonant_Sweeper_functions.stop)

        #Connects the buttons in the Lock-In Program Tab
        self.Lock_In_programs = Lock_In(main=self, ui=self.ui)
        self.connect(self.ui.pushButtonSelectLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.choose_visa)
        self.connect(self.ui.pushButtonUpdateLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.update_visa)
        self.connect(self.ui.pushButtonStartLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.start)
        self.connect(self.ui.pushButtonStopLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.stop)
        self.connect(self.ui.pushButtonStopLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.final_append)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.append_parameters)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.collectDataThread.pause)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.collectDataThread.appendSessionData)
        self.connect(self.ui.pushButtonStopLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.collectDataThread.appendSessionData)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.tableAppend)
        self.connect(self.ui.pushButtonStopLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.tableAppend)

        self.connect(self.ui.pushButtonSaveLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save)
        self.connect(self.ui.pushButtonBrowseLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.browse)
        self.connect(self.ui.pushButtonFolderSelectLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.select_name)

        self.connect(self.ui.radioButtonDateTimeLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_name)
        self.connect(self.ui.radioButtonCustomLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_name)

        self.connect(self.ui.radioButton_csvLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_type)
        self.connect(self.ui.radioButton_txtLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_type)

        self.connect(self.Lock_In_programs.collectDataThread, SIGNAL("plot"),
                     self.Lock_In_programs.plotData)
        self.connect(self.Lock_In_programs.collectDataThread,
                     SIGNAL("Begin_Save"), self.Lock_In_programs.pre_save)

    # This method is to transfer the array data from the Array Builder tab to the Keithley tab
    # If the array is valid, data_available is true and execute the data transfer
    # If the array is not valid, execution cannot be done
    def CopyDataFunc(self):
        if self.Array_Builder_programs.data_available == True:
            return self.Array_Builder_programs.Values
        else:
            return None

    def Copy_ay(self):
        try:
            #self.Array_Builder_programs.data_available == True:
            self.Agilent_Yokogawa_programs.Copy_ay(
                self.Array_Builder_programs.Values)
            self.ui.output.setText('Array has been copied and plotted.')
        except:
            self.ui.output.setText('No valid array to copy.')

    # Make sure the user is going to quit the program
    def closeEvent(self, event):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()
Exemplo n.º 14
0
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas = FigureCanvas(self.ui.mplwidget.figure)
        self.canvas.setParent(self.ui.widget)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox = QVBoxLayout()
        # The matplotlib canvas
        vbox.addWidget(self.canvas)
        # The matplotlib toolbar
        vbox.addWidget(self.mpl_toolbar)
        self.ui.widget.setLayout(vbox)

        # Connect the mplwidget with canvas
        self.ui.mplwidget = self.canvas

        self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'),
                     self.Browse)
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.Plot)
        self.connect(self.ui.pushButton_show, SIGNAL('clicked()'), self.Show)

    def Browse(self):
        self.prev_dir = 'C:\Users\QMDlab\Desktop'
        self.fileDir = QFileDialog.getOpenFileName(self,
                                                   'Select File to Open:',
                                                   'C:\Users\QMDlab\Desktop')

        if self.fileDir != '':
            self.fileDir.replace('/', '\\')
            self.ui.lineEdit_address.setText(self.fileDir)
            self.ui.lineEdit_condition1.setText('Open Source File')
        else:
            self.ui.lineEdit_address.setText('None')
            self.ui.lineEdit_condition1.setText('Failed to Read File')

        self.ui.groupBox_labels.setEnabled(True)

    def Show(self):
        labels = ''
        row = 1
        raw_parameters = []
        parameters = []
        fp = open(self.fileDir)
        while True:
            line = fp.readline()
            #print 1, line
            linelist = line.split(',')
            if linelist[0] == 'Collected Data\n':
                break
            labels += linelist[0]
            row += 1
        self.ui.textEdit_labels.setText(labels)
        #fp.close()

        #fp = open(self.fileDir)
        """
        raw_parameters = lines[row].split(',')
        for i in range(0, len(raw_parameters) - 1):
            parameters.append(raw_parameters[i])
        parameters.append(raw_parameters[len(raw_parameters) - 1].strip())
        for i in range(0, len(parameters)):
            self.ui.comboBox_x.addItem(parameters[i])
            self.ui.comboBox_y.addItem(parameters[i])
        #fp.close()
        list_bundle = [[] for i in range(0, len(parameters))]
        """

        #Read Parameters
        parameters = fp.readline().replace("\n", '')
        #print 'parameters', parameters
        parameters = parameters.split(',')

        for i in range(0, len(parameters)):
            self.ui.comboBox_x.addItem(parameters[i])
            self.ui.comboBox_y.addItem(parameters[i])

        #Read Units
        units = fp.readline().replace("\n", '')
        #print 'units', units
        units = units.split(",")

        self.data = []

        for i in range(0, len(parameters)):
            new_data = [parameters[i], units[i], []]
            self.data.append(new_data)

        #fp = open(self.fileDir)
        while True:
            #parameters = []
            #fp = open(self.fileDir)
            lines = fp.readline().replace("\n", "")
            #print "lines", lines
            if lines == '':
                break

            values = lines.split(',')

            for i in range(0, len(values)):
                self.data[i][2].append(values[i])
            #row += 1

        #print data

        ###
        #for i in range(0, len(raw_parameters) - 1):
        #    parameters.append(raw_parameters[i])
        #parameters.append(raw_parameters[len(raw_parameters) - 1].strip())
        #for i in range(0, len(parameters)):
        #    list_bundle[i].append(parameters[i])
        ###

        #fp.close()
        self.ui.groupBox_plot.setEnabled(True)
        self.ui.lineEdit_condition2.setText('Ready to Plot.')

    def Plot(self):
        x_value = self.ui.comboBox_x.currentIndex()
        y_value = self.ui.comboBox_y.currentIndex()
        self.reset_plot()
        self.axes.plot(self.data[x_value][2],
                       self.data[y_value][2],
                       marker='.',
                       linestyle=':')
        self.axes.set_title(self.data[x_value][0] + ' vs. ' +
                            self.data[y_value][0])
        self.axes.set_xlabel(self.data[x_value][0] + ' (' +
                             self.data[x_value][1] + ')')
        self.axes.set_ylabel(self.data[y_value][0] + ' (' +
                             self.data[y_value][1] + ')')
        self.ui.mplwidget.draw()

        self.ui.lineEdit_condition2.setText('Plot successfully.')

    def reset_plot(self):
        self.ui.mplwidget.figure.clear()
        self.axes = self.ui.mplwidget.figure.add_subplot(111)

    def closeEvent(self, question):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, 'Message', quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        # If your answer is "yes", then quit
        if reply == QMessageBox.Yes:
            question.accept()
        # If your answer is "no", then get back
        else:
            question.ignore()
Exemplo n.º 15
0
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general,
                                                     self.ui.widget_general)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)

        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general

        self.connect(self.ui.actionArray_Builder, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(0))
        self.connect(self.ui.actionKeithley_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(1))
        self.connect(self.ui.actionAgilent_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(2))
        self.connect(self.ui.actionKeithley_Stepper_Single_Scan,
                     SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(3))
        self.connect(self.ui.actionKeithley_Gate_Sweep, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(4))
        self.connect(self.ui.actionLockIn_Single_Scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(5))
        self.connect(self.ui.actionResonant_Single_Scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(6))
        self.connect(self.ui.actionSee_Visa_List, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(7))

        self.List_Visas = List_Visas(main=self, ui=self.ui)
        self.Array_Builder_programs = Array_Builder(main=self, ui=self.ui)

        self.Keithley_programs = Keithley(main=self, ui=self.ui)
        self.Keithley_programs.Refresh_visa()

        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0

        self.ui.lineEdit_directory_save.setText(os.getcwd())

        self.Agilent_Yokogawa_programs = Agilent_Yokogawa(main=self,
                                                          ui=self.ui)

        self.Resonant_Sweeper_functions = Resonance_Sweep(main=self,
                                                          ui=self.ui)

        self.Resonant_Sweeper_functions.update_visa()

        self.Lock_In_programs = Lock_In(main=self, ui=self.ui)

        self.KeithleyGateSweep = KeithleyGateSweep(main=self, ui=self.ui)
Exemplo n.º 16
0
    def __init__(self, parent=None):

        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'),
                     lambda: self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'),
                     lambda: self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'),
                     lambda: self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'),
                     lambda: self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'),
                     lambda: self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(
            self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda: self.
            Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.
                   label_visa_1,
                   [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(
            self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda: self.
            Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.
                   label_visa_2,
                   [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(
            self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda: self.
            Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.
                   label_visa_3,
                   [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(
            self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda: self.
            Select("visa4",
                   self.visa4,
                   self.ui.comboBox_visa_4,
                   self.ui.label_visa_4,
                   [self.ui.pushButton_select_4, self.ui.pushButton_close_4],
                   baud=115200))
        self.connect(
            self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda: self.
            Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.
                   label_visa_5,
                   [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(
            self.ui.pushButton_close_1, SIGNAL('clicked()'),
            lambda: self.Close("visa1", self.visa1, self.ui.label_visa_1, [
                self.ui.pushButton_select_1, self.ui.pushButton_close_1
            ]))
        self.connect(
            self.ui.pushButton_close_2, SIGNAL('clicked()'),
            lambda: self.Close("visa2", self.visa2, self.ui.label_visa_2, [
                self.ui.pushButton_select_2, self.ui.pushButton_close_2
            ]))
        self.connect(
            self.ui.pushButton_close_3, SIGNAL('clicked()'),
            lambda: self.Close("visa3", self.visa3, self.ui.label_visa_3, [
                self.ui.pushButton_select_3, self.ui.pushButton_close_3
            ]))
        self.connect(
            self.ui.pushButton_close_4, SIGNAL('clicked()'),
            lambda: self.Close("visa4", self.visa4, self.ui.label_visa_4, [
                self.ui.pushButton_select_4, self.ui.pushButton_close_4
            ]))
        self.connect(
            self.ui.pushButton_close_5, SIGNAL('clicked()'),
            lambda: self.Close("visa5", self.visa5, self.ui.label_visa_5, [
                self.ui.pushButton_select_5, self.ui.pushButton_close_5
            ]))

        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'),
                     self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'),
                     self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"),
                     self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"),
                     self.Print_data)

        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'),
                     lambda: self.start_font("C"))

        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)

        self.curve_1 = self.make_curveWidgets(
            self.ui.curvewidget_1,
            "black",
            "black",
            titles=["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(
            self.ui.curvewidget_2,
            "r",
            "black",
            titles=["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(
            self.ui.curvewidget_3,
            "g",
            "black",
            titles=["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(
            self.ui.curvewidget_4,
            "o",
            "black",
            titles=["Plot 4", "X (x)", "Y (y)"])

        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None

        self.setWindowTitle("Magnetic Sweep with Voltage Bias Lock-in")

        font = QFont()
        font.setPointSize(8)
        self.ui.tabWidget_visa.setFont(font)
        self.ui.tabWidget_visa.setTabText(0, "Lock-in Curr")
        self.ui.tabWidget_visa.setTabText(1, "Lock-in Volt")
        self.ui.tabWidget_visa.setTabText(2, "Gate Keithley")
        self.ui.tabWidget_visa.setTabText(3, "Magnet")

        self.input_string = []
        self.input_string.append("00 Run_Number: 01\n\n")

        self.input_string.append("01 Start Field: 0\n")
        self.input_string.append("02 Field Sweep: 0.001\n")
        self.input_string.append("03 Record Step Field: 0.001\n")
        self.input_string.append("04 End Field: 0.001\n")
        self.input_string.append("05 Double Linear (y/n): y\n\n")

        self.input_string.append("06 Lock-in Curr Sens: 1e-3\n")
        self.input_string.append("07 Lock-in Volt Sens: 1e-3\n")
        self.input_string.append("08 Curr Pre-Amplifier Sens: 1e-5\n")
        self.input_string.append("09 Additional Lock-in Volt Gain: 1\n")
        self.input_string.append("10 Additional Lock-in Curr Gain: 1\n\n")

        self.input_string.append("11 User: Vannucci\n")
        self.input_string.append("12 Save Name: \n")
        self.input_string.append("13 TimeStamp (y/n): y\n")
        self.input_string.append("14 Comments: ")

        self.start_font("C")

        input_string_print = ""
        for st_i in self.input_string:
            input_string_print = input_string_print + st_i

        font = QFont()
        font.setPointSize(8)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)

        self.inputted_data = []
        self.array_sweep = []
Exemplo n.º 17
0
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general,
                                                     self.ui.widget_general)

        self.canvas_import = FigureCanvas(self.ui.mplwidget_import.figure)
        self.canvas_import.setParent(self.ui.widget_import)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import = NavigationToolbar(self.canvas_import,
                                                    self.ui.widget_import)

        # self.canvas_scan = FigureCanvas(self.ui.mplwidget_scan.figure)
        # self.canvas_scan.setParent(self.ui.widget_scan)
        # # This is the toolbar widget for the scan canvas
        # self.mpl_toolbar_scan = NavigationToolbar(self.canvas_scan, self.ui.widget_scan)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import = QVBoxLayout()
        # The matplotlib canvas
        vbox_import.addWidget(self.canvas_import)
        # The matplotlib toolbar
        vbox_import.addWidget(self.mpl_toolbar_import)
        self.ui.widget_import.setLayout(vbox_import)

        # # Create the QVBoxLayout object and add the widget into the Layout
        # vbox_scan = QVBoxLayout()
        # # The matplotlib canvas
        # vbox_scan.addWidget(self.canvas_scan)
        # # The matplotlib toolbar
        # vbox_scan.addWidget(self.mpl_toolbar_scan)
        # self.ui.widget_scan.setLayout(vbox_scan)

        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general
        self.ui.mplwidget_import = self.canvas_import
        # self.ui.mplwidget_scan = self.canvas_scan

        self.Array_Builder_programs = Array_Builder(self.ui)
        # Connect buttons in the General Array tab
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Plot_general)
        self.connect(self.ui.pushButton_clear, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Clear)
        self.connect(self.ui.pushButton_save, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Save)
        self.connect(self.ui.pushButton_browse_save, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Browse_save)

        # Connect buttons in the Keithely tab
        self.Keithley_programs = Keithley(self.ui)
        self.Keithley_programs.Refresh_visa()

        self.connect(self.ui.pushButton_browse_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Browse_keithley)
        self.connect(self.ui.pushButton_import_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Import_keithley)
        self.connect(self.ui.pushButton_close_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Close_keithley)
        self.connect(self.ui.pushButton_select_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Select_keithley)
        self.connect(self.ui.pushButton_scan_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Scan_keithley)
        self.connect(self.ui.pushButton_pause_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.collect_data_thread.pause)
        self.connect(self.ui.pushButton_stop_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.collect_data_thread.stop)
        self.connect(self.ui.pushButton_clear_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Clear_keithley)
        self.connect(self.ui.pushButton_copy_keithley, SIGNAL('clicked()'),
                     self.CopyData)
        self.connect(
            self.ui.radioButton_voltage_keithley, SIGNAL("clicked()"),
            lambda: self.ui.tabWidget_scan_keithley.setCurrentIndex(0))
        self.connect(
            self.ui.radioButton_current_keithley, SIGNAL("clicked()"),
            lambda: self.ui.tabWidget_scan_keithley.setCurrentIndex(1))
        self.connect(self.Keithley_programs.collect_data_thread,
                     SIGNAL("plot"), self.Keithley_programs.Plot_data)

        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0

        self.ui.lineEdit_directory_save.setText(os.getcwd())

        self.canvas_import_ay = FigureCanvas(
            self.ui.mplwidget_import_ay.figure)
        self.canvas_import_ay.setParent(self.ui.widget_import_ay)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import_ay = NavigationToolbar(
            self.canvas_import_ay, self.ui.widget_import_ay)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import_ay = QVBoxLayout()
        # The matplotlib canvas
        vbox_import_ay.addWidget(self.canvas_import_ay)
        # The matplotlib toolbar
        vbox_import_ay.addWidget(self.mpl_toolbar_import_ay)
        self.ui.widget_import_ay.setLayout(vbox_import_ay)

        # Connect the mplwidget with canvas

        self.ui.mplwidget_import_ay = self.canvas_import_ay
        # self.ui.mplwidget_scan = self.canvas_scan

        # Connect buttons in the Agilent Yokogawa tab
        self.Agilent_Yokogawa_programs = Agilent_Yokogawa(self.ui)

        self.connect(self.ui.startButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.start)
        #self.connect(self.action_timer, SIGNAL("timeout()"), self.Agilent_Yokogawa_programs.action)

        self.connect(self.ui.pushButton_browse_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Browse_ay)
        self.connect(self.ui.pushButton_import_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Import_ay)
        self.connect(self.ui.pushButton_copy_ay, SIGNAL('clicked()'),
                     self.Copy_ay)
        self.connect(self.ui.selectVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.select_visa)
        self.connect(self.ui.updateVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.update_visa)
        self.connect(self.ui.closeVisaButton0, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa0)
        self.connect(self.ui.closeVisaButton1, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa1)

        self.connect(self.ui.browseButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.browse)
        self.connect(self.ui.saveButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.save)
Exemplo n.º 18
0
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general,
                                                     self.ui.widget_general)

        self.canvas_import = FigureCanvas(self.ui.mplwidget_import.figure)
        self.canvas_import.setParent(self.ui.widget_import)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import = NavigationToolbar(self.canvas_import,
                                                    self.ui.widget_import)

        # self.canvas_scan = FigureCanvas(self.ui.mplwidget_scan.figure)
        # self.canvas_scan.setParent(self.ui.widget_scan)
        # # This is the toolbar widget for the scan canvas
        # self.mpl_toolbar_scan = NavigationToolbar(self.canvas_scan, self.ui.widget_scan)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import = QVBoxLayout()
        # The matplotlib canvas
        vbox_import.addWidget(self.canvas_import)
        # The matplotlib toolbar
        vbox_import.addWidget(self.mpl_toolbar_import)
        self.ui.widget_import.setLayout(vbox_import)

        # # Create the QVBoxLayout object and add the widget into the Layout
        # vbox_scan = QVBoxLayout()
        # # The matplotlib canvas
        # vbox_scan.addWidget(self.canvas_scan)
        # # The matplotlib toolbar
        # vbox_scan.addWidget(self.mpl_toolbar_scan)
        # self.ui.widget_scan.setLayout(vbox_scan)

        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general
        self.ui.mplwidget_import = self.canvas_import
        # self.ui.mplwidget_scan = self.canvas_scan

        self.Array_Builder_programs = Array_Builder(self.ui)
        # Connect buttons in the General Array tab
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Plot_general)
        self.connect(self.ui.pushButton_clear, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Clear)
        self.connect(self.ui.pushButton_save, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Save)
        self.connect(self.ui.pushButton_browse_save, SIGNAL('clicked()'),
                     self.Array_Builder_programs.Browse_save)

        # Connect buttons in the Keithely tab
        self.Keithley_programs = Keithley(self.ui)
        self.Keithley_programs.Refresh_visa()

        self.connect(self.ui.pushButton_browse_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Browse_keithley)
        self.connect(self.ui.pushButton_import_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Import_keithley)
        self.connect(self.ui.pushButton_close_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Close_keithley)
        self.connect(self.ui.pushButton_select_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Select_keithley)
        self.connect(self.ui.pushButton_scan_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Scan_keithley)
        self.connect(self.ui.pushButton_pause_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.collect_data_thread.pause)
        self.connect(self.ui.pushButton_stop_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.collect_data_thread.stop)
        self.connect(self.ui.pushButton_clear_keithley, SIGNAL('clicked()'),
                     self.Keithley_programs.Clear_keithley)
        self.connect(self.ui.pushButton_copy_keithley, SIGNAL('clicked()'),
                     self.CopyData)
        self.connect(
            self.ui.radioButton_voltage_keithley, SIGNAL("clicked()"),
            lambda: self.ui.tabWidget_scan_keithley.setCurrentIndex(0))
        self.connect(
            self.ui.radioButton_current_keithley, SIGNAL("clicked()"),
            lambda: self.ui.tabWidget_scan_keithley.setCurrentIndex(1))
        self.connect(self.Keithley_programs.collect_data_thread,
                     SIGNAL("plot"), self.Keithley_programs.Plot_data)

        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0

        self.ui.lineEdit_directory_save.setText(os.getcwd())

        self.canvas_import_ay = FigureCanvas(
            self.ui.mplwidget_import_ay.figure)
        self.canvas_import_ay.setParent(self.ui.widget_import_ay)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import_ay = NavigationToolbar(
            self.canvas_import_ay, self.ui.widget_import_ay)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import_ay = QVBoxLayout()
        # The matplotlib canvas
        vbox_import_ay.addWidget(self.canvas_import_ay)
        # The matplotlib toolbar
        vbox_import_ay.addWidget(self.mpl_toolbar_import_ay)
        self.ui.widget_import_ay.setLayout(vbox_import_ay)

        # Connect the mplwidget with canvas

        self.ui.mplwidget_import_ay = self.canvas_import_ay
        # self.ui.mplwidget_scan = self.canvas_scan

        # Connect buttons in the Agilent Yokogawa tab
        self.Agilent_Yokogawa_programs = Agilent_Yokogawa(self.ui)

        self.connect(self.ui.startButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.start)
        #self.connect(self.action_timer, SIGNAL("timeout()"), self.Agilent_Yokogawa_programs.action)

        self.connect(self.ui.pushButton_browse_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Browse_ay)
        self.connect(self.ui.pushButton_import_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Import_ay)
        self.connect(self.ui.pushButton_copy_ay, SIGNAL('clicked()'),
                     self.Copy_ay)
        self.connect(self.ui.selectVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.select_visa)
        self.connect(self.ui.updateVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.update_visa)
        self.connect(self.ui.closeVisaButton0, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa0)
        self.connect(self.ui.closeVisaButton1, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa1)

        self.connect(self.ui.browseButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.browse)
        self.connect(self.ui.saveButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.save)

    def CopyData(self):
        if self.Array_Builder_programs.data_available == True:
            self.Keithley_programs.Copy_keithley(
                self.Array_Builder_programs.Values)
            self.ui.lineEdit_condition_keithley.setText(
                'Array has been copied and plotted.')
        else:
            self.ui.lineEdit_condition_keithley.setText(
                'No valid array to copy.')

    def Copy_ay(self):
        if self.Array_Builder_programs.data_available == True:
            self.Agilent_Yokogawa_programs.Copy_ay(
                self.Array_Builder_programs.Values)
            self.ui.output.setText('Array has been copied and plotted.')
        else:
            self.ui.output.setText('No valid array to copy.')

    # Make sure the user is going to quit the program
    def closeEvent(self, event):
        quit_msg = "DO you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()
Exemplo n.º 19
0
    def __init__(self, parent = None):
        
        # Standard GUI code
        QWidget.__init__(self, parent)
        
        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general, self.ui.widget_general)
        
        self.canvas_import = FigureCanvas(self.ui.mplwidget_import.figure)
        self.canvas_import.setParent(self.ui.widget_import)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import = NavigationToolbar(self.canvas_import, self.ui.widget_import)
        
        self.canvas_scan = FigureCanvas(self.ui.mplwidget_scan.figure)
        self.canvas_scan.setParent(self.ui.widget_scan)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_scan = NavigationToolbar(self.canvas_scan, self.ui.widget_scan)
        
        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)
        
        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import = QVBoxLayout()
        # The matplotlib canvas
        vbox_import.addWidget(self.canvas_import)
        # The matplotlib toolbar
        vbox_import.addWidget(self.mpl_toolbar_import)
        self.ui.widget_import.setLayout(vbox_import)
        
        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_scan = QVBoxLayout()
        # The matplotlib canvas
        vbox_scan.addWidget(self.canvas_scan)
        # The matplotlib toolbar
        vbox_scan.addWidget(self.mpl_toolbar_scan)
        self.ui.widget_scan.setLayout(vbox_scan)
        
        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general
        self.ui.mplwidget_import = self.canvas_import
        self.ui.mplwidget_scan = self.canvas_scan
        
        self.General_programs = General(self.ui)
        # Connect buttons in the General Array tab
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.General_programs.Plot_general)
        self.connect(self.ui.pushButton_clear, SIGNAL('clicked()'), self.General_programs.Clear)
        self.connect(self.ui.pushButton_save, SIGNAL('clicked()'), self.General_programs.Save)
        self.connect(self.ui.pushButton_browse_save, SIGNAL('clicked()'), self.General_programs.Browse_save)
        
        self.Keithley_programs = Keithley(self.ui)
        self.Keithley_programs.Refresh_visa()
        # Connect buttons in the Keithely tab
        self.connect(self.ui.pushButton_browse_keithley, SIGNAL('clicked()'), self.Keithley_programs.Browse_keithley)
        self.connect(self.ui.pushButton_import_keithley, SIGNAL('clicked()'), self.Keithley_programs.Import_keithley)
        self.connect(self.ui.pushButton_close_keithley, SIGNAL('clicked()'), self.Keithley_programs.Close_keithley)
        self.connect(self.ui.pushButton_select_keithley, SIGNAL('clicked()'), self.Keithley_programs.Select_keithley)
        self.connect(self.ui.pushButton_scan_keithley, SIGNAL('clicked()'), self.Keithley_programs.Scan_keithley)
        self.connect(self.ui.pushButton_stop_keithley, SIGNAL('clicked()'), self.Keithley_programs.Stop_keithley)
        self.connect(self.ui.pushButton_clear_keithley, SIGNAL('clicked()'), self.Keithley_programs.Clear_keithley)
        
        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0
        
        self.ui.lineEdit_directory_save.setText(os.getcwd())
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas = FigureCanvas(self.ui.mplwidget.figure)
        self.canvas.setParent(self.ui.widget)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox = QVBoxLayout()
        # The matplotlib canvas
        vbox.addWidget(self.canvas)
        # The matplotlib toolbar
        vbox.addWidget(self.mpl_toolbar)
        self.ui.widget.setLayout(vbox)

        # Connect the mplwidget with canvas
        self.ui.mplwidget = self.canvas
        self.reset_plot()

        self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'),
                     self.Browse)
        self.connect(self.ui.pushButton_change_name, SIGNAL('clicked()'),
                     self.Change_name)
        self.connect(self.ui.pushButton_open, SIGNAL('clicked()'), self.Open)
        self.connect(self.ui.pushButton_Plot, SIGNAL('clicked()'),
                     self.Plot_ready)
        self.connect(self.ui.pushButton_Next, SIGNAL('clicked()'), self.Next)
        self.connect(self.ui.pushButton_Previous, SIGNAL('clicked()'),
                     self.Previous)
        self.connect(self.ui.pushButton_find, SIGNAL('clicked()'), self.Find)
        self.connect(self.ui.pushButton_modify, SIGNAL('clicked()'),
                     self.Modify)
        self.connect(self.ui.pushButton_return, SIGNAL('clicked()'),
                     self.Return)

        self.ui.pushButton_browse.setEnabled(True)
        self.modify = False
        self.first_round = True
        self.open_dir = ''
        self.open_files_number = 0
        self.all_info = []

    def Browse(self):
        self.Num = 0
        self.Files = []
        self.Names = []

        prev_dir = os.getcwd()
        fileDir = QFileDialog.getOpenFileNames(self, 'Select Files to Open',
                                               self.ui.lineEdit_address.text())
        if fileDir != '':
            self.comboBox_update = True
            self.Files = fileDir
            open_dir = ''
            file_list = str(fileDir[0]).split('\\')
            for i in range(0, len(file_list) - 1):
                if i < len(file_list) - 2:
                    open_dir += file_list[i] + '\\'
                elif i == len(file_list) - 2:
                    open_dir += file_list[i]
            self.ui.lineEdit_address.setText(open_dir)

            self.ui.pushButton_open.setEnabled(True)
            self.first_round = False
            filedir = str(fileDir).split('\\')
            for i in range(0, len(fileDir)):
                temp = fileDir[i].split('\\')
                name = temp[len(temp) - 1]
                self.Names.append(name)
            if len(fileDir) > 1:
                self.ui.label_condition.setText(
                    str(len(fileDir)) + ' files selected, click Open')
            elif len(fileDir) == 1:
                self.ui.label_condition.setText(
                    str(len(fileDir)) + ' file selected, click Open')
        else:
            self.ui.lineEdit_address.setText('None')
            self.ui.label_condition.setText('Failed to Read File')

    def Open(self):

        # Every you click the "Open File" button, the content in the plot combox is refreshed
        self.reset_plot()
        self.ui.mplwidget.draw()
        self.ui.lineEdit_x.setText('')
        self.ui.lineEdit_y.setText('')
        self.ui.lineEdit_name.setText('')
        self.ui.label_condition.setText('')
        self.ui.lineEdit_minute.setText('')
        self.ui.label_date.setText('')

        labels = ''

        raw_parameters = []
        units = []
        parameters = []
        new_parameters = []
        self.data = []
        new_data = []
        first_data = []
        self.date_time = []
        temp_parameters = []

        item = 0
        length = 0
        # This count is the number of the reading loops for labels. If count is over 1000, that means there's something wrong with the divider line and it helps to stop the program
        count = 0

        # It is false when program cannot find the line divider and runs the while loop 1000 times
        self.divider_found = True

        fp = open(self.Files[self.Num])
        while True:
            if count > 1000:
                self.ui.label_condition.setText('Line divider is not found.')
                self.divider_found = False
                break
            line = fp.readline()
            linelist = line.split(',')
            if linelist[0].upper() == 'COLLECTED DATA\n':
                break
            for i in range(0, len(linelist)):
                labels += linelist[i]
            count += 1

        if self.divider_found == True:
            self.ui.textEdit_labels.setText(labels)
            parameters = fp.readline().replace("\n", '')
            temp_parameters = parameters.split(',')
            if temp_parameters[0] == 'Date':
                parameters = temp_parameters[1:]
            else:
                parameters = temp_parameters

            # The following is for the units
            units = fp.readline().replace("\n", '')
            units = units.split(",")
            if temp_parameters[0] == 'Date':
                units = units[1:]

            if self.comboBox_update:
                self.ui.comboBox_x.clear()
                self.ui.comboBox_y1.clear()
                self.ui.comboBox_y2.clear()
                self.ui.comboBox_y3.clear()
                self.ui.comboBox_y4.clear()
                self.ui.comboBox_y5.clear()
                self.ui.comboBox_y6.clear()
                self.ui.comboBox_y2.addItem('None')
                self.ui.comboBox_y3.addItem('None')
                self.ui.comboBox_y4.addItem('None')
                self.ui.comboBox_y5.addItem('None')
                self.ui.comboBox_y6.addItem('None')
                for i in range(0, len(parameters)):
                    self.ui.comboBox_x.addItem(parameters[i])
                    self.ui.comboBox_y1.addItem(parameters[i])
                    self.ui.comboBox_y2.addItem(parameters[i])
                    self.ui.comboBox_y3.addItem(parameters[i])
                    self.ui.comboBox_y4.addItem(parameters[i])
                    self.ui.comboBox_y5.addItem(parameters[i])
                    self.ui.comboBox_y6.addItem(parameters[i])
                self.ui.comboBox_y1.setCurrentIndex(1)
                self.comboBox_update = False

            for i in range(0, len(parameters)):
                new_data = [parameters[i], units[i], [], []]
                self.data.append(new_data)

            while True:

                val = []
                lines = fp.readline().replace("\n", "")
                if lines == '':
                    break
                values = lines.split(',')
                for i in range(0, len(values)):
                    if values[0] != 'Measurement was Aborted':
                        if values[i] != '':
                            val.append(values[i])
                if temp_parameters[0] == 'Date':
                    date_info = val[0].split(' ')
                    if '/' not in date_info[0]:
                        weekday = datetime.datetime.strptime(
                            date_info[0], '%Y-%m-%d').strftime('%a')
                    else:
                        weekday = datetime.datetime.strptime(
                            date_info[0], '%m/%d/%Y').strftime('%a')
                    date = date_info[0] + ' ' + weekday + ' ' + date_info[1]
                    self.date_time.append(date)
                    start = 1
                else:
                    start = 0
                for i in range(start, len(val)):
                    self.data[i - start][2].append(val[i])

            self.ui.pushButton_Plot.setEnabled(True)
            if self.Num == 0:
                self.ui.pushButton_Previous.setEnabled(False)
            if len(self.Files) == 1:
                self.ui.label_condition.setText(
                    str(len(self.Files)) + ' file ready to Plot')
            else:
                self.ui.label_condition.setText(
                    str(len(self.Files)) + ' files ready to Plot')

    def Find(self):
        try:
            time = float(self.ui.lineEdit_minute.text())
            if time <= float(self.data[0][2][0]):
                self.ui.label_date.setText(self.date_time[0])
            elif time >= float(self.data[0][2][len(self.data[0][2]) - 1]):
                self.ui.label_date.setText(self.date_time[len(self.date_time) -
                                                          1])
            else:
                item = 0
                while float(self.data[0][2][item]) - time < 0:
                    item += 1
                if abs(float(self.data[0][2][item - 1]) -
                       time) <= abs(float(self.data[0][2][item]) - time):
                    self.ui.label_date.setText(self.date_time[item - 1])
                else:
                    self.ui.label_date.setText(self.date_time[item])

        except ValueError:
            self.ui.label_date.setText('')
            self.ui.label_condition.setText('Please enter valid time.')

    def Plot_ready(self):
        self.x_value = []
        self.y1_value = []
        self.y2_value = []
        self.y3_value = []
        self.y4_value = []
        self.y5_value = []
        self.y6_value = []

        # The corresponding number of the comboBox
        self.x_num = 0
        self.x_num = self.ui.comboBox_x.currentIndex()
        self.y1_num = 0
        self.y1_num = self.ui.comboBox_y1.currentIndex()
        self.y2_num = 0
        self.y2_num = self.ui.comboBox_y2.currentIndex() - 1
        self.y3_num = 0
        self.y3_num = self.ui.comboBox_y3.currentIndex() - 1
        self.y4_num = 0
        self.y4_num = self.ui.comboBox_y4.currentIndex() - 1
        self.y5_num = 0
        self.y5_num = self.ui.comboBox_y5.currentIndex() - 1
        self.y6_num = 0
        self.y6_num = self.ui.comboBox_y6.currentIndex() - 1

        x_axis = self.data[self.x_num][0]
        x_unit = self.data[self.x_num][1]
        y1_axis = self.data[self.y1_num][0]
        y1_unit = self.data[self.y1_num][1]
        y2_axis = self.data[self.y2_num][0]
        y2_unit = self.data[self.y2_num][1]
        y3_axis = self.data[self.y3_num][0]
        y3_unit = self.data[self.y3_num][1]
        y4_axis = self.data[self.y4_num][0]
        y4_unit = self.data[self.y4_num][1]
        y5_axis = self.data[self.y5_num][0]
        y5_unit = self.data[self.y5_num][1]
        y6_axis = self.data[self.y6_num][0]
        y6_unit = self.data[self.y6_num][1]

        self.y1_color_num = 0
        self.y1_color_num = self.ui.comboBox_color_y1.currentIndex()
        self.y1_color = self.Color(self.y1_color_num)
        self.y2_color_num = 0
        self.y2_color_num = self.ui.comboBox_color_y2.currentIndex()
        self.y2_color = self.Color(self.y2_color_num)
        self.y3_color_num = 0
        self.y3_color_num = self.ui.comboBox_color_y3.currentIndex()
        self.y3_color = self.Color(self.y3_color_num)
        self.y4_color_num = 0
        self.y4_color_num = self.ui.comboBox_color_y4.currentIndex()
        self.y4_color = self.Color(self.y4_color_num)
        self.y5_color_num = 0
        self.y5_color_num = self.ui.comboBox_color_y5.currentIndex()
        self.y5_color = self.Color(self.y5_color_num)
        self.y6_color_num = 0
        self.y6_color_num = self.ui.comboBox_color_y6.currentIndex()
        self.y6_color = self.Color(self.y6_color_num)

        self.y1_legend = self.ui.comboBox_y1.currentText()
        self.y2_legend = self.ui.comboBox_y2.currentText()
        self.y3_legend = self.ui.comboBox_y3.currentText()
        self.y4_legend = self.ui.comboBox_y4.currentText()
        self.y5_legend = self.ui.comboBox_y5.currentText()
        self.y6_legend = self.ui.comboBox_y6.currentText()

        self.x_value = self.data[self.x_num][2]
        self.y1_value = self.data[self.y1_num][2]
        if self.y2_num != -1:
            self.y2_value = self.data[self.y2_num][2]
        if self.y3_num != -1:
            self.y3_value = self.data[self.y3_num][2]
        if self.y4_num != -1:
            self.y4_value = self.data[self.y4_num][2]
        if self.y5_num != -1:
            self.y5_value = self.data[self.y5_num][2]
        if self.y6_num != -1:
            self.y6_value = self.data[self.y6_num][2]

        self.ui.lineEdit_x.setText(x_axis + ' (' + x_unit + ')')
        self.ui.lineEdit_y.setText(y1_axis + ' (' + y1_unit + ')')
        self.ui.lineEdit_name.setText(self.Names[self.Num])

        self.ui.pushButton_change_name.setEnabled(True)
        if len(self.Files) > 1:
            self.ui.pushButton_Next.setEnabled(True)
        self.Pre_plot()

    def Color(self, num):
        if num == 0:
            return 'b'
        elif num == 1:
            return 'g'
        elif num == 2:
            return 'r'
        elif num == 3:
            return 'y'
        elif num == 4:
            return 'k'
        elif num == 5:
            return 'm'

    def Pre_plot(self):
        self.reset_plot()
        self.axes.grid()

        self.x_value = numpy.array(self.x_value, dtype='float')
        self.y1_value = numpy.array(self.y1_value, dtype='float')
        self.Plot(self.x_value, self.y1_value, self.y1_color, self.y1_legend)
        if self.y2_num != -1:
            self.y2_value = numpy.array(self.y2_value, dtype='float')
            self.Plot(self.x_value, self.y2_value, self.y2_color,
                      self.y2_legend)
        if self.y3_num != -1:
            self.y3_value = numpy.array(self.y3_value, dtype='float')
            self.Plot(self.x_value, self.y3_value, self.y3_color,
                      self.y3_legend)
        if self.y4_num != -1:
            self.y4_value = numpy.array(self.y4_value, dtype='float')
            self.Plot(self.x_value, self.y4_value, self.y4_color,
                      self.y4_legend)
        if self.y5_num != -1:
            self.y5_value = numpy.array(self.y5_value, dtype='float')
            self.Plot(self.x_value, self.y5_value, self.y5_color,
                      self.y5_legend)
        if self.y6_num != -1:
            self.y6_value = numpy.array(self.y6_value, dtype='float')
            self.Plot(self.x_value, self.y6_value, self.y6_color,
                      self.y6_legend)

        self.ui.label_condition.setText('File' + ' (' + str(self.Num + 1) +
                                        '/' + str(len(self.Files)) + ')' +
                                        ': ' + self.Names[self.Num])

        self.Draw()

    def Next(self):
        self.Num = self.Num + 1
        self.Open()
        self.Plot_ready()
        if self.Num == len(self.Files) - 1:
            self.ui.pushButton_Next.setEnabled(False)
        self.ui.pushButton_Previous.setEnabled(True)

    def Previous(self):
        self.Num = self.Num - 1
        self.Open()
        self.Plot_ready()
        if self.Num == -1:
            self.ui.pushButton_Previous.setEnabled(False)
        self.ui.pushButton_Next.setEnabled(True)

    def Modify(self):
        try:
            self.modify = True
            if self.ui.lineEdit_x_range.text() != '':
                self.modify_x = True
                self.x_min = float(
                    self.ui.lineEdit_x_range.text().split(',')[0])
                self.x_max = float(
                    self.ui.lineEdit_x_range.text().split(',')[1])
            else:
                self.modify_x = False
            if self.ui.lineEdit_y_range.text() != '':
                self.modify_y = True
                self.y_min = float(
                    self.ui.lineEdit_y_range.text().split(',')[0])
                self.y_max = float(
                    self.ui.lineEdit_y_range.text().split(',')[1])
            else:
                self.modify_y = False
            if self.ui.checkBox_logx.isChecked():
                self.log_x = True
            else:
                self.log_x = False
            if self.ui.checkBox_logx.isChecked():
                self.log_y = True
            else:
                self.log_y = False
            self.Plot_ready()
        except ValueError:
            self.ui.label_condition.setText('Please enter valid Min and Max.')

    def Return(self):
        self.modify = False
        self.ui.lineEdit_x_range.setText('')
        self.ui.lineEdit_y_range.setText('')
        self.ui.checkBox_logx.setCheckState(False)
        self.ui.checkBox_logx.setCheckState(False)
        self.ui.pushButton_return.setEnabled(False)
        self.Plot_ready()

    def Change_name(self):
        self.axes.set_title(self.ui.lineEdit_name.text())
        self.axes.set_xlabel(self.ui.lineEdit_x.text())
        self.axes.set_ylabel(self.ui.lineEdit_y.text())
        self.Draw()

    def Plot(self, x, y, c, l):
        self.axes.plot(x, y, color=c, marker='.', linestyle=':', label=l)
        self.axes.set_title(self.ui.lineEdit_name.text())
        self.axes.set_xlabel(self.ui.lineEdit_x.text())
        self.axes.set_ylabel(self.ui.lineEdit_y.text())
        self.axes.legend()
        if self.modify:
            if self.modify_x:
                self.axes.set_xlim(self.x_min, self.x_max)
            if self.modify_y:
                self.axes.set_ylim(self.y_min, self.y_max)
            if self.log_x:
                self.axes.set_xscale('log')
            if self.log_y:
                self.axes.set_yscale('log')
        self.ui.label_condition.setText('Plot successfully.')

    def Draw(self):
        self.ui.mplwidget.draw()

    def reset_plot(self):
        self.ui.mplwidget.figure.clear()
        self.axes = self.ui.mplwidget.figure.add_subplot(111)
        self.ui.mplwidget.figure.subplots_adjust(left=0.06,
                                                 right=0.94,
                                                 top=0.92,
                                                 bottom=0.08)

    def convert_time(self, d):
        date, times = d.split(" ")
        year, month, day = date.split("-")
        hour, minute, second = times.split(":")

        t = datetime.datetime(int(year), int(month), int(day), int(hour),
                              int(minute), int(second))
        t_sec = time.mktime(t.timetuple())

        return t_sec

    def closeEvent(self, question):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, 'Message', quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        # If your answer is "yes", then quit
        if reply == QMessageBox.Yes:
            question.accept()
        # If your answer is "no", then get back
        else:
            question.ignore()
Exemplo n.º 21
0
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # When you open the window, the visas get refreshed
        self.refresh_visa()

        # This is to check whether user click start before they save
        self.data_check = False

        # We set up a QTimer variable called collect_timer.
        # It runs as a countdown clock that runs in the background and emits a timeout signal when it reaches 0.
        # Because this variable, we do not to use a for loop function for collecting and timing since the for loop can not be stopped in progress and can cause program frozen.
        # Also it allows the rest of the commands to run normally
        self.collect_timer = QTimer()

        # Set up guiqwt plot
        self.curve_item = make.curve([], [], color='r')
        self.ui.curvewidgetPlot.plot.add_item(self.curve_item)
        self.ui.curvewidgetPlot.plot.set_antialiasing(True)
        self.ui.curvewidgetPlot.plot.set_titles(
            "Agilent 34460A Digit Multimeter", "Time (s)", "Y-Axis")

        # The following connects the signals sent from an action on the GUI to a function that does something
        # When the button is clicked, it does the corresponding function
        self.connect(self.ui.pushButton_Select_Visa, SIGNAL("clicked()"),
                     self.choose_visa)
        self.connect(self.ui.pushButton_Start, SIGNAL("clicked()"), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL("clicked()"), self.stop)
        self.connect(self.ui.pushButton_Refresh, SIGNAL("clicked()"),
                     self.refresh_visa)

        # This works as a clock running in the background, different functions can use this clock at the same time
        # It is better than .sleep() function because during the sleep time, the whole program is frozen and cannot do anything else
        # Everytime collect_timer gets the time_step, it sends a "timeout()" signal to the device and runs the collect function
        self.connect(self.collect_timer, SIGNAL("timeout()"), self.collect)

        self.connect(self.ui.comboBox_Meas_Type, SIGNAL("clicked()"),
                     self.meas_type)

        # To give a click signal to the checkBox
        self.connect(self.ui.checkBox_Continue, SIGNAL("clicked()"),
                     self.continue_check)

        # I add a slect button for the measurement type
        # Only if you click the button, the computer knows your measurement type
        # It shows the current measurement type and warning at the same time
        self.connect(self.ui.pushButton_Select_Measurement,
                     SIGNAL("clicked()"), self.choose_meas)

        # To save the data
        self.connect(self.ui.pushButton_Save, SIGNAL("clicked()"), self.save)

        # To see and change the directory
        self.connect(self.ui.pushButton_Browse, SIGNAL("clicked()"),
                     self.browse)

        # To choose a name folder to save file
        self.connect(self.ui.pushButton_Select_Directory, SIGNAL("clicked()"),
                     self.select_directory)

        # To select name for the file, date and time or custom name entered by the user
        self.connect(self.ui.radioButton_Timename, SIGNAL("clicked()"),
                     self.select_name)
        self.connect(self.ui.radioButton_Custom_Name, SIGNAL("clicked()"),
                     self.select_name)

        # To select a file type, either .csv or .txt
        self.connect(self.ui.radioButton_csv, SIGNAL('clicked()'),
                     self.select_type)
        self.connect(self.ui.radioButton_csv, SIGNAL('clicked()'),
                     self.select_type)

    def refresh_visa(self):
        # Put all the visa resources into all_visas
        # If there is no visa resource at all, an error is returned. So this helps to prevent this kind of situation
        rm = visa.ResourceManager()
        try:
            # put the visa resources into a list
            all_visas = rm.list_resources()
        except:
            # return "No visas available" if there is no visa
            all_visas = 'No visa available right now.'

        # Remove the previous results in the comboBox
        self.ui.comboBox_All_Visas.clear()
        # Show all the visas in the combox
        for item in all_visas:
            self.ui.comboBox_All_Visas.addItem(item)

    # Click select button and choose the visa you want
    def choose_visa(self):
        self.visa_address = str(self.ui.comboBox_All_Visas.currentText())
        rm = visa.ResourceManager()
        rm.list_resources()
        dmm = rm.open_resource(self.visa_address)
        valid = self.check_visa(dmm)
        if valid == True:
            self.ui.lineEdit_Error.setText("Everything goes well right now.")
            self.ui.label_CurrentVisa.setText(self.visa_address)
        elif valid == False:
            self.ui.lineEdit_Error.setText("Invalid visa address.")
            self.ui.label_CurrentVisa.setText('None')

    # This one is used to check whether the Agilent is connected correctly
    # It can be frequently used in the beginning of each function because if the Agilent is not linked it would report error when program running.
    def check_visa(self, dmm):
        try:
            dmm.ask('*IDN?')
            valid = True
        except:
            valid = False
        return valid

    def set_timestep(self):
        try:
            self.timestep = float(self.ui.lineEdit_Steps.text())
        except:
            self.timestep = -1
        if self.timestep <= 0:
            self.ui.lineEdit_Error.setText("Invalid time step input")

    # This is the number of points you want to collect
    # Also it is the number of points plotted on the figure
    def set_points(self):
        try:
            self.points = int(self.ui.lineEdit_Points.text())
            print self.points
        except:
            self.points = -1
        if self.points <= 0:
            self.ui.lineEdit_Error.setText("Invalid data points input")

    # Set two numpy arrays to store data and time
    def init_nparray(self):
        # self.data is used to store the response collected from the device
        self.data = np.array([0], dtype=float)
        # self.time is used to store the time each response is collected
        self.time = np.array([0], dtype=float)

    # To judge whether the checkBox is checked or not
    def continue_check(self):
        # If the checkBox is checked, return true and store it into cont_check
        if self.ui.checkBox_Continue.isChecked() == True:
            self.cont_check = True
        # If the checkBox hasn't been checked, return false and store it into cont_check
        elif self.ui.checkBox_Continue.isChecked() == False:
            self.cont_check = False
        # Return the result
        return self.cont_check

    # To vary from which types of measurements the user want to do
    def meas_type(self):
        self.meas = str(self.ui.comboBox_Meas_Type.currentText())
        self.meas_type_check = True
        self.unit = 'V'
        if self.meas == 'None':
            self.meas_type_check = False
        elif self.meas == 'AC Current':
            self.command = 'MEAS:CURR:AC?'
            self.delay_correct = 0.5143113484
            self.unit = 'A'
        elif self.meas == 'DC Current':
            self.command = 'MEAS:CURR:DC?'
            self.delay_correct = 0.4683864132
            self.unit = 'A'
        elif self.meas == 'AC Voltage':
            self.command = 'MEAS:VOLT:AC?'
            self.delay_correct = 0.7460769785
        elif self.meas == 'DC Voltage':
            self.command = 'MEAS:VOLT:DC?'
            self.delay_correct = 0.3845665093
        elif self.meas == 'Resistance':
            self.unit = 'Ohms'
            self.command = 'MEAS:RES?'

    def choose_meas(self):
        self.meas_type()
        self.ui.label_CurrentMeasurement.setText(self.meas)
        if self.meas == 'AC Current':
            self.ui.label_Step_Reminder.setText(
                "Please enter a time step larger than 0.6s.")
        elif self.meas == 'DC Current':
            self.ui.label_Step_Reminder.setText(
                "Please enter a time step larger than 0.5s.")
        elif self.meas == 'AC Voltage':
            self.ui.label_Step_Reminder.setText(
                "Please enter a time step larger than 0.8s.")
        elif self.meas == 'DC Voltage':
            self.ui.label_Step_Reminder.setText(
                "Please enter a time step larger than 0.4s.")
        elif self.meas == 'None':
            self.ui.lineEdit_Error.setText("Please choose a valid command")

    def start(self):
        self.meas_type()
        # If user does not choose a meas_type
        if self.meas_type_check == False:
            self.ui.lineEdit_Error.setText("Please choose a valid command")
            self.meas_type()
        else:
            start_check = False
            # Collect the time steps entered by the user
            try:
                self.steps = float(self.ui.lineEdit_Steps.text())
            # In case the user type something instead of a number
            except:
                self.steps = -1
            # To judge whether the checkBox is checked or not
            self.continue_check()
            # If the user does not choose the continue measurement mode
            if self.cont_check == False:
                # Collect the points entered by the user in the edit line
                try:
                    self.points = float(self.ui.lineEdit_Points.text())
                # In case the user enters anything strange instead of a number
                except:
                    self.points = -1
            # If the user choose the continue mode, then the points do not matter
            # So set it to be a positive number such as 1
            elif self.cont_check == True:
                self.points = 1

            # Both the time steps and points are invalid
            if self.steps <= 0 and self.points <= 0:
                self.ui.lineEdit_Error.setText(
                    "Invalid time steps and points input")
            # Only the time steps are invalid
            elif self.steps <= 0:
                self.ui.lineEdit_Error.setText("Invalid ti  me steps input")
            # Only the points number are invalid
            elif self.points <= 0:
                self.ui.lineEdit_Error.setText("Invalid points input")
            # Both the time steps and points are valid
            else:
                start_check = True

            # This helps to check whther the input time step is in the range
            # If it is smaller than the minimum step,
            if self.steps < self.delay_correct:
                self.ui.lineEdit_Error.setText(
                    "Your time step is too small. Please enter again.")
                start_check = False

            # Set the address of the digit multimeter
            # I do not use self.rm & self.dmm, so I need to put them here before the data collection
            rm = visa.ResourceManager()
            rm.list_resources()
            dmm = rm.open_resource(self.visa_address)
            # Check if everything is ready to start
            if start_check == True:
                valid = self.check_visa(dmm)
                # If the visa address is valid
                if valid == True:
                    self.data_check = True
                    # Tell user there is no error right now
                    self.ui.lineEdit_Error.setText(
                        "Everything goes well right now.")
                    self.meas_type()
                    # Set the numpy arrays for time and data
                    self.init_nparray()
                    # To calculate the time difference for system responce, set starting time t1
                    self.t1 = time.clock()
                    # The default time step for collect_timer function is 1ms
                    # So for 1 second, we have to multiply 1000
                    self.collect_timer.start(1000 * self.steps)
                    # Run the collect function
                    self.collect()

    # This is to collect the data from the device
    def collect(self):
        self.meas_type()

        self.t2 = time.clock()
        self.t = self.t2 - self.t1
        self.t1 = time.clock()

        self.time_now = self.time[-1] + self.t

        collect_check = True
        rm = visa.ResourceManager()
        rm.list_resources()
        dmm = rm.open_resource(self.visa_address)
        try:
            reading = float(dmm.ask(self.command))
            self.ui.lineEdit_Error.setText("Start to collect data and plot.")
        except:
            collect_check = False
            valid = self.check_visa(self.visa_address)
            if valid == True:
                self.ui.lineEdit_Error.setText("Invalid command syntax.")
            elif valid == False:
                self.ui.lineEdit_Error.setText("Connection disabled")

        # The problem here is that the numpy array is going to append automatically and in the end the length will surplus the number of points we want to plot
        # Therefore we have to make sure the points on the plot is less than or equal to the number we want
        if collect_check == True:
            self.continue_check()
            # If the continue checkBox is checked, run the collect until the user click stop
            self.data = np.append(self.data, reading)
            self.time = np.append(self.time, self.time_now)
            if self.cont_check == True:
                # We do not need the zero point, so we start plot from the first data
                self.x = self.time[1:]
                self.y = self.data[1:]
            elif self.cont_check == False:
                if len(self.time) > self.points:
                    self.x = self.time[(len(self.time) - self.points):]
                else:
                    self.x = self.time[1:]
                if len(self.data) > self.points:
                    self.y = self.data[(len(self.data) - self.points):]
                else:
                    self.y = self.data[1:]

            # To adjust the scale of the result
            # You do not want to see E-9 in the y-axis on the plot
            # Take absolute for all the data in the self.data in case some data are negative
            self.abs_y = np.absolute(self.y)
            # Set the first number in the self.abs_y
            First = self.abs_y[0]
            # Set a specific string for scale
            self.scale = ''
            self.out_y = self.y
            # The scale is E-3
            if First < 1E-1 and First >= 1E-4:
                self.out_y = 1E3 * self.y
                self.scale = 'm'
            # The scale is E-6
            elif First < 1E-4 and First >= 1E-7:
                self.out_y = 1E6 * self.y
                self.scale = 'u'
            # The scale is E-9
            elif First < 1E-7 and First >= 1E-10:
                self.out_y = 1E9 * self.y
                self.scale = 'n'

            # Refresh the lables, especially the scally for yLable
            self.refresh_labels()
            # Sets the data in the guiqwt plot
            self.curve_item.set_data(self.x, self.out_y)
            self.curve_item.plot().replot()
        else:
            self.stop()

    # Refresh the lables when it starts to plot
    def refresh_labels(self):
        self.title = "Measure " + self.meas + " by Agilent 34460A Digit Multimeter"
        self.yLabel = self.meas + ' (' + self.scale + self.unit + ')'
        self.ui.curvewidgetPlot.plot.set_titles(self.title, "Time (s)",
                                                self.yLabel)

    def stop(self):
        self.collect_timer.stop()

    # This function let user to search for the OneDrive folder
    # The lineEdit_OneDrive has already shown the directory of OneDrive in this computer
    # However, the directory maybe different in different computer
    # So if someone use different pc or mac, they can use browse to find the OneDrive folder
    def browse(self):
        # The default OneDrive folder directory
        self.pre_directory = self.ui.lineEdit_OneDrive.text()
        # Open the browse window and let use to choose a new OneDrive directory
        self.directory = QFileDialog.getExistingDirectory(
            self, 'Select GoogleDrive Directory')
        # If there is a new directory, refresh the default OneDrive directory to this new one
        if self.directory != '':
            # Chang the '/' into '\'
            string.replace(self.directory, '/', '\\')
            # Switch the OneDrive to the new directory
            self.ui.lineEdit_OneDrive.setText(self.directory)

    # This function works to locate the address to the namefolder\Data
    def select_directory(self):
        # The previous directory
        self.pre_directory = str(self.ui.lineEdit_OneDrive.text())
        # The name of the folder you want to save
        self.namefolder = str(self.ui.comboBox_Name_Folder.currentText())
        self.select_directory_check = True
        if self.namefolder == 'None':
            self.ui.lineEdit_Error.setText(
                "Please choose a name folder to save file.")
            self.select_directory_check = False
        else:
            self.ui.lineEdit_Error.setText("Save to GoogleDrive\\" +
                                           self.namefolder + "\Data" + '\\' +
                                           self.date)

    # Set the file name
    # User can choose whether to save the data by date and time or their custom name
    def select_name(self):
        self.custom_name_check = True
        if self.ui.radioButton_DateTime.isChecked() == True:
            self.now = datetime.now()
            self.date = '%s-%s-%s' % (now.year, now.month, now.day)
            self.current_time = '%s.%s.%s' % (now.hour, now.minute, now.second)
            self.date_and_time = self.date + ' ' + self.current_time
            self.file_name = str(self.date_and_time)
        else:
            self.custom_name_check = False

    # Let user to choose which type of the file they want to save
    def select_type(self):
        self.select_type_check = True
        if self.ui.radioButton_csv.isChecked() == True:
            self.type = '.csv'
            self.divide = ','
            self.form = ''
        elif self.ui.radioButton_txt.isChecked() == True:
            self.type = '.txt'
            self.divide = '\t'
            self.form = '              '
        else:
            self.select_type_check = False

    # To save the data
    def save(self):
        self.select_directory()
        self.select_name()
        self.select_type()
        # Check whether user collect data before they save
        if self.data_check == False:
            self.ui.lineEdit_Error.setText("There is no data to be saved.")
        else:
            # If the user choose to type their own file name, set their input to be the file name
            if self.ui.radioButton_Custom_Name.isChecked() == True:
                self.custom_name_check = True
                self.file_name = str(self.ui.lineEdit_custom_Name.text())
            # Check if the user choose one of the name folder to save
            # If they do not choose one, ask them to pick up a name folder
            if self.select_directory_check == False:
                self.ui.lineEdit_Error.setText(
                    "Please pick up a name folder to save file.")
            else:
                # If user does not choose any of two radio buttons for the file name, ask them to choose one
                if self.custom_name_check == False:
                    self.ui.lineEdit_Error.setText(
                        "Please choose a kind of file name.")
                # If user input nothing in the text line for the custom file name, ask them to fill in a name
                elif self.file_name == '':
                    self.ui.lineEdit_Error.setText(
                        "Please enter a valid file name.")
                else:
                    # If user does not choose any of two types of the file, ask them to choose one
                    if self.select_type_check == False:
                        self.ui.lineEdit_Error.setText(
                            "Please select a file type to save.")
                    else:
                        self.path = self.pre_directory + '\\' + self.namefolder + '\\' + 'Data' + '\\' + self.date
                        if not os.path.isdir(self.path):
                            os.makedirs(self.path)
                        # Complete the file adddress to OneDrive\NameFolder\Data
                        self.name = self.pre_directory + '\\' + self.namefolder + '\\' + 'Data' + '\\' + self.date + '\\' + self.file_name + self.type
                        # Open a file
                        f = open(self.name, 'w')
                        # Write the name in the first line
                        f.write('Name: ' + self.namefolder + '\n')
                        # Write the time in the second line
                        f.write('Time: ' + str(datetime.now()) + '\n')
                        # This is all the measurements Agilent can do in this program
                        f.write('#' + self.divide + 'Time (s)' + self.divide +
                                'AC Current (A)' + self.divide +
                                'DC Current (A)' + self.divide +
                                'AC Voltage (V)' + self.divide +
                                'DC Voltage (V)' + self.divide +
                                'Resistance (Ohms)' + '\n')
                        # The number of measurements
                        n = 1
                        for i in range(1, len(self.data)):
                            # self.divide is to divide the different types of data. For .csv file it's ',' while for .txt file it's '\t'
                            # self.form is to make the data looks better
                            if self.meas == 'AC Current':
                                f.write(
                                    str(n) + self.divide + str(self.time[i]) +
                                    self.divide + str(self.data[i]) + '\n')
                            elif self.meas == 'DC Current':
                                f.write(
                                    str(n) + self.divide + str(self.time[i]) +
                                    self.divide + self.form + self.divide +
                                    str(self.data[i]) + '\n')
                            elif self.meas == 'AC Voltage':
                                f.write(
                                    str(n) + self.divide + str(self.time[i]) +
                                    self.divide + self.form + self.divide +
                                    self.form + self.divide +
                                    str(self.data[i]) + '\n')
                            elif self.meas == 'DC Voltage':
                                f.write(
                                    str(n) + self.divide + str(self.time[i]) +
                                    self.divide + self.form + self.divide +
                                    self.form + self.divide + self.form +
                                    self.divide + str(self.data[i]) + '\n')
                            elif self.meas == 'Resistance':
                                f.write(
                                    str(n) + self.divide + str(self.time[i]) +
                                    self.divide + self.form + self.divide +
                                    self.form + self.divide + self.form +
                                    self.divide + self.form + self.divide +
                                    str(self.data[i]) + '\n')
                            n += 1
                        # Don't forget to close the file in the end
                        f.close()
                        self.ui.lineEdit_Error.setText(
                            "Your file has been successfully saved!")

    # When you close the window, there will be a new window comes out and ask you "Do you want to quit the program?"
    def closeEvent(self, question):
        print question
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, 'Message', quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        # If your answer is "yes", then quit
        if reply == QMessageBox.Yes:
            question.accept()
        # If your answer is "no", then get back
        else:
            question.ignore()
Exemplo n.º 22
0
class MyForm(QMainWindow):
    
    # The __init__ function is what is everything the user wants to be initialized when the class is called.
    # Here we shall define the trig functions to corresponding variables.
    # Note that the "self" variable means that the function is part of the class and can be called inside and outside the class.(Although __init__ is special.)
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()
        self.dsave_thread = Dynamic_Save_Thread()
        
        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))

        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("freq_update"), self.setup_plot)
        self.connect(self.collect_data_thread, SIGNAL("power_update"), self.setup_plot)
        self.connect(self.collect_data_thread, SIGNAL("print_value"), self.Print_data)
        self.connect(self.ui.checkBox_dsave, SIGNAL("clicked()"), self.Pre_dsave)
        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'), lambda : self.start_font("C"))
        self.connect(self.ui.pushButton_dsave_browse, SIGNAL('clicked()'), self.Dsave_browse)
        
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)

        
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "b", "blue", titles = ["RF Power", "Step", "Power (dBm)"])
        self.curve_3_update = self.make_curveWidgets(self.ui.curvewidget_3, "red", "red", titles = ["RF Power", "Step", "Power (dBm)"])
        self.curve_2_b = self.make_curveWidgets(self.ui.curvewidget_2, "black", "black", titles = ["Keithley", "Step", "Voltage (V)"])
        self.curve_2_r = self.make_curveWidgets(self.ui.curvewidget_2, "red", "red", titles = ["Keithley", "Step", "Voltage (V)"])
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "b", "blue", titles = ["Keithley", "Step", "Voltage (V)"])
        self.curve_1_update = self.make_curveWidgets(self.ui.curvewidget_1, "red", "red", titles = ["Keithley", "Step", "Voltage (V)"])
        
        self.Curve1 = self.make_curveWidgets(self.ui.curvewidget_13, "r", "black", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.Curve2 = self.make_curveWidgets(self.ui.curvewidget_14, "g", "black", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.Curve3 = self.make_curveWidgets(self.ui.curvewidget_15, "r", "black", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.Curve4 = self.make_curveWidgets(self.ui.curvewidget_16, "g", "black", titles = ["Plot 4", "X (x)", "Y (y)"])
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None

        self.setWindowTitle("Sweep with Lock-in without Magnetic")
        
        self.input_string = []
        self.input_string.append("00 Run_Number: 01\n\n")
        
        self.input_string.append("01 Start Gate Voltage (V): 35\n")
        self.input_string.append("02 Step Gate Voltage (V): 0.1\n")
        self.input_string.append("03 03 End Gate Voltage (V): 0\n")
        self.input_string.append("04 Double Linear (y/n): n\n")
        self.input_string.append("05 Turn off Keithley (y/n): n\n")
        self.input_string.append("06 CurrBias Lock-in Sens: 20e-6\n")
        self.input_string.append("07 Reading wait time(s): .25\n\n")
        
        self.input_string.append("08 Current Bias Resistor: 1.009e9\n")
        self.input_string.append("09 Lock-in Output Voltage: 10\n")
        self.input_string.append("10 Additional Lock-in Volt Gain: 1\n\n")
        
        self.input_string.append("11 User: Vannucci\n")
        self.input_string.append("12 File Name: S-SiA0036-V_B1_leads\n")
        self.input_string.append("13 TimeStamp (y/n): y\n")
        self.input_string.append("14 Comments:\n\n")
        self.start_font("C")        
        
        self.out0_1 = "None"
        self.out0_2 = "None"
        self.out0_3 = "None"
        self.out0_4 = "None"
        self.out1_0 = "None"
        self.out1_1 = "None"
        self.out1_2 = "None"
        self.out1_3 = "None"
        self.out2_0 = "None"
        self.out2_1 = "None"
        self.out2_2 = "None"
        self.out2_3 = "None"
        self.out3_0 = "None"
        self.out3_1 = "None"
        self.out3_2 = "None"
        self.out3_3 = "None"
        
        self.scale1_1 = [1, ""]
        self.scale1_2 = [1, ""]
        self.scale1_3 = [1, ""]
        self.scale2_1 = [1, ""]
        self.scale2_2 = [1, ""]
        self.scale2_3 = [1, ""]
        self.scale3_1 = [1, ""]
        self.scale3_2 = [1, ""]
        self.scale3_3 = [1, ""]
        
        self.Coil_Constant = 1E-3/9.1E-3
        
        try:
            file_par = open("parameters.txt", "r")
            input_string_print = file_par.read()
            
        except:
            input_string_print = ""
            for st_i in self.input_string:
                input_string_print = input_string_print + st_i
        
        font = QFont()
        font.setPointSize(10)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)
        
        self.inputted_data = []
        self.array_sweep = []
        
        rm = visa.ResourceManager()
        self.Visa_states = []
        try:
            alls = rm.list_resources()
            for i in range(0, len(alls)):
                self.Visa_states.append([alls[i], True])
        except:
            alls = "No Visa Available."
        
    def start_font(self, status):
        if status == "C":
            font = QFont()
            font.setPointSize(8)
            self.ui.pushButton_Start.setFont(font)
            self.ui.pushButton_Start.setText("Check Array")
        elif status == "S":
            font = QFont()
            font.setPointSize(12)
            self.ui.pushButton_Start.setFont(font)
            self.ui.pushButton_Start.setText("Start")
    
    def Update(self, signal, comboBoxVisa):
        rm = visa.ResourceManager()
        try:
            alls = rm.list_resources()
        except:
            alls = "No Visa Available."
        if signal == "visa1":
            self.ui.comboBox_visa_1.clear()
        elif signal == "visa2":
            self.ui.comboBox_visa_2.clear()
        elif signal == "visa3":
            self.ui.comboBox_visa_3.clear()
        elif signal == "visa4":
            self.ui.comboBox_visa_4.clear()
        
        for temp in alls:
            if signal == "visa1":
                self.ui.comboBox_visa_1.addItem(temp)
            elif signal == "visa2":
                self.ui.comboBox_visa_2.addItem(temp)
            elif signal == "visa3":
                self.ui.comboBox_visa_3.addItem(temp)
            elif signal == "visa4":
                self.ui.comboBox_visa_4.addItem(temp)
        
        # if signal == "visa1":
        #     self.ui.comboBox_visa_1.addItem(alls[3])
        # elif signal == "visa2":
        #     self.ui.comboBox_visa_2.addItem(alls[0])
        # elif signal == "visa3":
        #     self.ui.comboBox_visa_3.addItem(alls[4])
        # elif signal == "visa4":
        #     self.ui.comboBox_visa_4.addItem(alls[1])
    
    def Select(self, signal, visa_chosen, comboBoxVisa, lineEditVisa, selectClose, baud = None):
        visa_address = str(comboBoxVisa.currentText())
        rm = visa.ResourceManager()
        rm.list_resources()
        if baud == None:
            inst = rm.open_resource(visa_address)
        else:
            inst = rm.open_resource(visa_address, baud_rate = baud)
        visa_check = self.Check(inst)
        if visa_check == True:
            
            visa_name = inst.query("*IDN?")
            name_list = visa_name.split(',')
            first_name = name_list[0]
            if signal == "visa1":
                if self.Ready(inst):
                    lineEditVisa.setText(visa_name)
                    selectClose[0].setEnabled(False)
                    selectClose[1].setEnabled(True)
                    self.visa1 = inst
                    self.Mark(inst, False)
                    self.ui.label_condition.setText("Visa is selected succefully!")
                else:
                    self.ui.label_condition.setText("Pick another Lock_In address")
            elif signal == "visa2":
                if self.Ready(inst):
                    lineEditVisa.setText(visa_name)
                    selectClose[0].setEnabled(False)
                    selectClose[1].setEnabled(True)
                    self.visa2 = inst
                    self.Mark(inst, False)
                    self.ui.label_condition.setText("Visa is selected succefully!")
                else:
                    self.ui.label_condition.setText("Pick another Keithley address")
            
        elif visa_check == False:
            self.ui.label_condition.setText("Invalid visa address.")
            lineEditVisa.setText("None.")
            visa_chosen = False
    
    def Ready(self, inst):
        temp = str(inst).split(' ')
        address = temp[len(temp) - 1]
        for i in range(0, len(self.Visa_states)):
            if self.Visa_states[i][0] == address:
                return self.Visa_states[i][1]
        return False
    
    def Mark(self, inst, bol):
        temp = str(inst).split(' ')
        address = temp[len(temp) - 1]
        for i in range(0, len(self.Visa_states)):
            if self.Visa_states[i][0] == address:
                self.Visa_states[i][1] = bol
    
    def Close(self, signal, visa_chosen, lineEditVisa, selectClose):
        self.ui.label_condition.setText('Visa address is closed')
        lineEditVisa.setText('')
        selectClose[0].setEnabled(True)
        selectClose[1].setEnabled(False)
        self.Mark(visa_chosen, True)
        if signal == "visa1":
            visa_chosen.close()
            self.visa1 = None
            #self.Disable(signal)
        elif signal == "visa2":
            visa_chosen.close()
            self.visa2 = None
            #self.Disable(signal)
        elif signal == "visa3":
            visa_chosen.close()
            self.visa3 = None
            #self.Disable(signal)
        elif signal == "visa4":
            visa_chosen.close()
            self.visa4 = None
            #self.Disable(signal)
            
    def Check(self, inst):
        try:
            inst.ask("*IDN?")
            valid = True
        except:
            valid = False
        return valid
    
    def make_curveWidgets(self, curvewidget, color, markerColor, titles):
        curve_temp = make.curve([], [], color = color, marker = "o", markerfacecolor = markerColor, markersize = 5)
        curvewidget.plot.add_item(curve_temp)
        curvewidget.plot.set_antialiasing(True)
        curvewidget.plot.set_titles(titles[0], titles[1], titles[2])
        return curve_temp
    
    def array_builder(self, start, step, stop, double_linear):
        if start < stop:
            array_1 = numpy.arange(start, stop + step, abs(step))
        elif start > stop:
            array_1 = numpy.arange(stop, start - step, abs(step))[::-1]
        else:
            array_1 = numpy.array([start])
        
        if double_linear == "y":
            array = array_1[::-1]
            array_2 = numpy.delete(array, 0)
            
            array_sweep = numpy.append(array_1, array_2)
        else:
             array_sweep = array_1
        return array_sweep
            
    def start(self):
        self.Device1_state = False
        
        self.instruments = []
        curves = [self.Curve1, self.Curve2, self.Curve3, self.Curve4, self.curve_3, self.curve_3_update, self.curve_1, self.curve_1_update]
        curveWidgets =[self.ui.curvewidget_13, self.ui.curvewidget_14, self.ui.curvewidget_15, self.ui.curvewidget_16, self.ui.curvewidget_1]
        array_1 = []
        go_on = True
        
        input_data = []
        temp = str(self.ui.textEdit.toPlainText()).split("\n")
        for i in range(0, len(temp)):
            if temp[i] != '':
                input_data.append(temp[i])
        inputted_data = []
        inputted_data.append([input_data[0].split(":")[1].replace(" ", "")])
        temp = []
        for i in range(1, 8):
            temp.append(input_data[i].split(":")[1].replace(" ", ""))
        inputted_data.append(temp)
        temp = []
        for i in range(8, 11):
            temp.append(input_data[i].split(":")[1].replace(" ", ""))
        inputted_data.append(temp)
        temp = []
        for i in range(11, 15):
            temp.append(input_data[i].split(":")[1].replace(" ", ""))
        inputted_data.append(temp)
        self.Device1_state = True
        device_state = [self.Device1_state]
        
        #Make Array
        count = 0
        if self.Device1_state:
            count += 1
            temp = []
            array_1 = self.array_builder(float(inputted_data[1][0]), float(inputted_data[1][1]), float(inputted_data[1][2]), inputted_data[1][3])
            
        
        if str(self.ui.pushButton_Start.text()) == "Start":
            if self.Check_ready(count):
                if self.ui.checkBox_dsave.isChecked():
                    self.Dsave_directory()
                    if self.dsave_dir_ok:
                        self.Dsave_filename()
                        if self.dsave_filename_ok:    
                            self.Dsave_username()
                            if self.dsave_username_ok:
                                self.Dsave_filetype()
                                dsave = [self.dsave_directory, self.dsave_filename, self.dsave_username, self.dsave_thread, self.dsave_filetype, self.dsave_divide, self.date]
                                self.start_font("C")
                                self.ui.tabWidget.setCurrentIndex(2)
                                self.ui.pushButton_Start.setEnabled(False)
                                self.ui.pushButton_Pause.setEnabled(True)
                                self.ui.pushButton_Stop.setEnabled(True)
                                
                                file_par = open("parameters.txt", "w").write(str(self.ui.textEdit.toPlainText()))
                                self.collect_data_thread.input(self.ui, self.instruments, curves, curveWidgets, go_on, inputted_data, array_1, device_state, count, dsave, input_data)
                            else:
                                self.ui.label_condition.setText('Enter user name for dynamic saving.')
                        else:
                            self.ui.label_condition.setText('Enter file name for dynamic saving.')
                    else:
                        self.ui.label_condition.setText('Choose valid directory for dynamic saving.')
                else:
                    self.ui.label_condition.setText('Choose dynamic saving.')
            else:
                self.ui.label_condition.setText('Make sure enough devices are connected.')
        else:
            # if self.Device1_state:
            #     self.setup_plot_2(self.curve_2_b, self.curve_2_r, array_1)
            self.setup_plot(self.curve_1, array_1)
            self.ui.tabWidget.setCurrentIndex(0)
            self.start_font("S")
    
    def Check_ready(self, count):
        if count == 1:
            self.instruments = [self.visa1, self.visa2]
            return True
        return False
    
    def setup_plot(self, curve, data):
        x = []
        for i in range(0, len(data)):
            x.append(i)
        curve.set_data(x, data)
        self.ui.curvewidget_1.plot.do_autoscale()
        self.ui.curvewidget_2.plot.do_autoscale()
        self.ui.curvewidget_3.plot.do_autoscale()
        curve.plot().replot()
    
    def setup_plot_2(self, curve_b, curve_r, data):
        x1 = []
        y1 = []
        x2 = []
        y2 = []
        num = 0
        for i in range(0, len(data)):
            x1.append(num)
            y1.append(data[i])
            num += 1
        num -= 1
        curve_b.set_data(x1, y1)
        #curve_r.set_data(x2, y2)
        curve_b.plot().replot()
        #curve_r.plot().replot()
            
    def curvePlots_update(self, curveInfo):
        curveWidget = curveInfo[0]
        curve = curveInfo[1]
        curveWidget.plot.do_autoscale()
        curve.plot().replot()
    
    def Print_data(self, device, text):
        font = QFont()
        font.setPointSize(10)
        self.ui.textEditDisplay.setFont(font)
        self.out0_1 = str(format(text[0], '.3f'))
        self.out0_2 = str(format(text[1], '.3f'))
        lockin_params = text[7]
        # print lockin_params
        # print lockin_params[1][0]
        # print lockin_params[0][0]
        # print lockin_params[1][1]
        # print lockin_params[0][1]
        display_text = ""
        display_text = display_text + "Time: " + self.out0_1 + " s\n"
        display_text = display_text + "Time Step: " + self.out0_2 + " s\n"
        out1_1 = float(text[2])
        self.scale1_1 = self.Switch_scale(out1_1)
        self.out1_1 = str(format(out1_1 * self.scale1_1[0], '.3f'))
        out1_2 = float(text[3])
        self.scale1_2 = self.Switch_scale(out1_2)
        self.out1_2 = str(format(out1_2 * self.scale1_2[0], '.3f'))
        out1_3 = float(text[4])
        self.scale1_3 = self.Switch_scale(out1_3)
        self.out1_3 = str(format(out1_3 * self.scale1_3[0], '.3f'))
        out1_4 = float(text[5])
        self.scale1_4 = self.Switch_scale(out1_4)
        self.out1_4 = str(format(out1_4 * self.scale1_4[0], '.3f'))
        self.out1_5 = float(text[6])
        # self.scale1_5 = self.Switch_scale(out1_5)
        # self.out1_5 = str(format(out1_5 * self.scale1_5[0], '.3f'))
            
        if self.Device1_state:
            display_text = display_text + "Output Voltage: " + str(self.out1_1) + " " + str(self.scale1_1[1]) + "Volts\n"
            display_text = display_text + "Output Current: " + str(self.out1_2) + " " + str(self.scale1_2[1])+ "Amps\n"
            display_text = display_text + "X: " + str(self.out1_3) + " "  + str(self.scale1_3[1])+ "Volts\n"
            display_text = display_text + "Y: " + str(self.out1_4) + " "  + str(self.scale1_4[1])+ "Volts\n"
            display_text = display_text + "Phase: " + str(self.out1_5) + " " + "Radians\n"
            display_text = display_text + lockin_params[1][0] + ': ' + lockin_params[0][0] +'\n'
            display_text = display_text + lockin_params[1][1] + ': ' + lockin_params[0][1] +'\n'
            display_text = display_text + lockin_params[1][2] + ': ' + lockin_params[0][2] +'\n'
            display_text = display_text + lockin_params[1][3] + ': ' + lockin_params[0][3] +'\n'
            display_text = display_text + lockin_params[1][4] + ': ' + lockin_params[0][4] +'\n'
            display_text = display_text + lockin_params[1][5] + ': ' + lockin_params[0][5] +'\n'
            display_text = display_text + lockin_params[1][6] + ': ' + lockin_params[0][6] +'\n'
            display_text = display_text + lockin_params[1][7] + ': ' + lockin_params[0][7] +'\n'
            display_text = display_text + lockin_params[1][8] + ': ' + lockin_params[0][8] +'\n'
            display_text = display_text + lockin_params[1][9] + ': ' + lockin_params[0][9] +'\n'
            display_text = display_text + text[8]+ '\n'
            
            
        self.ui.textEditDisplay.setText(display_text)       
    
    def Switch_scale(self, num):
        temp = abs(num)
        if temp >= 1E9:
            scale = [1E-9, "G"]
        elif temp >= 1E6 and temp < 1E9:
            scale = [1E-6, "M"]
        elif temp >= 1E3 and temp < 1E6:
            scale = [1E-3, "k"]
        elif temp >= 1 and temp < 1000:
            scale = [1, ""]
        elif temp >= 1E-3 and temp < 1:
            scale = [1E3, "m"]
        elif temp >= 1E-6 and temp < 1E-3:
            scale = [1E6, "u"]
        elif temp >= 1E-9 and temp < 1E-6:
            scale = [1E9, "n"]
        elif temp < 1E-9:
            scale = [1E12, "p"]
            
        return scale 
    
    def Pre_dsave(self):
        if self.ui.checkBox_dsave.isChecked():
            self.ui.label_dsave_directory.setEnabled(True)
            self.ui.lineEdit_dsave_directory.setEnabled(True)
            self.ui.pushButton_dsave_browse.setEnabled(True)
            self.ui.groupBox_dsave_filename.setEnabled(True)
            self.ui.radioButton_dsave_timename.setEnabled(True)
            self.ui.radioButton_dsave_custname.setEnabled(True)
            self.ui.lineEdit_dsave_custname.setEnabled(True)
            self.ui.groupBox_dsave_filetype.setEnabled(True)
            self.ui.radioButton_csv.setEnabled(True)
            self.ui.radioButton_txt.setEnabled(True)
            self.ui.label_dsave_username.setEnabled(True)
            self.ui.lineEdit_dsave_username.setEnabled(True)
            self.ui.label_dsave_comment.setEnabled(True)
            self.ui.textEdit_dsave_comment.setEnabled(True)
            self.ui.label_condition.setText("Dynamic saving opened.")
        else:
            self.ui.label_dsave_directory.setEnabled(False)
            self.ui.lineEdit_dsave_directory.setEnabled(False)
            self.ui.pushButton_dsave_browse.setEnabled(False)
            self.ui.groupBox_dsave_filename.setEnabled(False)
            self.ui.radioButton_dsave_timename.setEnabled(False)
            self.ui.radioButton_dsave_custname.setEnabled(False)
            self.ui.lineEdit_dsave_custname.setEnabled(False)
            self.ui.groupBox_dsave_filetype.setEnabled(False)
            self.ui.radioButton_csv.setEnabled(False)
            self.ui.radioButton_txt.setEnabled(False)
            self.ui.label_dsave_username.setEnabled(False)
            self.ui.lineEdit_dsave_username.setEnabled(False)
            self.ui.label_dsave_comment.setEnabled(False)
            self.ui.textEdit_dsave_comment.setEnabled(False)
            self.ui.label_condition.setText("Dynamic saving closed.")

    def Dsave_browse(self):
        self.dsave_directory = ''
        prev_dir = os.getcwd()
        fileDir = QFileDialog.getExistingDirectory(None, 'Select Folder to Save', prev_dir)
        if fileDir != '':
            open_dir = ''
            file_list = str(fileDir).split('/')
            for i in range(0, len(file_list) - 1):
                if i < len(file_list) - 1:
                    open_dir += file_list[i] + '\\'
                elif i == len(file_list) - 1:
                    open_dir += file_list[i]
            fileDir.replace('/', '\\')
            self.dsave_directory = fileDir
            self.ui.lineEdit_dsave_directory.setText(fileDir)
            self.ui.label_condition.setText("Dynamic saving directory selected.")
        else:
            self.ui.lineEdit_dsave_directory.setText('None')
            self.ui.label_condition.setText('Choose a directory for dynamic saving.')
            
    def Dsave_directory(self):
        self.dsave_dir_ok = True
        if self.ui.lineEdit_dsave_directory.text() == '' or self.ui.lineEdit_dsave_directory.text() == 'None':
            self.dsave_dir_ok = False
        
    def Dsave_filename(self):
        self.dsave_filename_ok = True
        self.date = ''
        self.dsave_filename = ''
        if self.ui.radioButton_dsave_timename.isChecked():
            now = datetime.datetime.now()
            date = '%s-%s-%s' % (now.year, now.month, now.day)
            current_time = '%s.%s.%s' % (now.hour, now.minute, now.second)
            date_and_time = date + '_' + current_time
            self.date = date
            self.dsave_filename = str(date_and_time)
        elif self.ui.radioButton_dsave_custname.isChecked():
            self.dsave_filename = str(self.ui.lineEdit_dsave_custname.text())
            if self.dsave_filename == '':
                self.dsave_filename_ok = False
    
    def Dsave_filetype(self):
        if self.ui.radioButton_csv.isChecked():
            self.dsave_filetype = '.csv'
            self.dsave_divide = ','
        elif self.ui.radioButton_txt.isChecked():
            self.dsave_filetype = '.txt'
            self.dsave_divide = '\t'
            
    def Dsave_username(self):
        self.dsave_username_ok = True
        self.dsave_username = ''
        self.dsave_username = str(self.ui.lineEdit_dsave_username.text())
        if self.dsave_username == '':
            self.dsave_username_ok = False
            
    def closeEvent(self, event):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg, QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()
class MyForm(QMainWindow):

    # The __init__ function is what everything the user wants to be initialized when the class is called.
    # Here we shall define the tring functions to corresponding variables.
    # The 'self' variable means that the function is part of the class and can be called inside and outside the class.
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas = FigureCanvas(self.ui.mplwidget.figure)
        self.canvas.setParent(self.ui.widget)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar = NavigationToolbar(self.canvas, self.ui.widget)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox = QVBoxLayout()
        # The matplotlib canvas
        vbox.addWidget(self.canvas)
        # The matplotlib toolbar
        vbox.addWidget(self.mpl_toolbar)
        self.ui.widget.setLayout(vbox)

        # Connect the mplwidget with canvas
        self.ui.mplwidget = self.canvas

        self.connect(self.ui.pushButton_browse, SIGNAL('clicked()'),
                     self.Browse)
        self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.Plot)
        self.connect(self.ui.pushButton_show, SIGNAL('clicked()'), self.Show)
        self.connect(self.ui.radioButton_qmd, SIGNAL('clicked()'),
                     self.Choose1)
        self.connect(self.ui.radioButton_ppms, SIGNAL('clicked()'),
                     self.Choose1)
        self.connect(self.ui.radioButton_other, SIGNAL('clicked()'),
                     self.Choose1)
        self.connect(self.ui.pushButton_choose2, SIGNAL('clicked()'),
                     self.Choose2)
        self.connect(self.ui.pushButton_recent, SIGNAL('clicked()'),
                     self.Open_Recent)

        self.first_round = True
        self.open_dir = ''
        self.open_files_number = 0
        self.all_info = []

    def Choose1(self):
        if self.ui.radioButton_qmd.isChecked():
            self.divider = 'Collected Data'
            self.ui.lineEdit_divider.setText(self.divider)
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')
        elif self.ui.radioButton_ppms.isChecked():
            self.divider = '[Data]'
            self.ui.lineEdit_divider.setText(self.divider)
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please choose your file.')
        elif self.ui.radioButton_other.isChecked():
            self.ui.lineEdit_divider.setText('')
            self.ui.pushButton_browse.setEnabled(True)
            self.ui.lineEdit_condition1.setText('Please enter data divider.')

    def Browse(self):

        if self.ui.radioButton_other.isChecked():
            if self.ui.lineEdit_divider.text() == '':
                self.ui.lineEdit_condition1.setText(
                    'Data divider cannot be empty.')
            else:
                self.divider = self.ui.lineEdit_divider.text()
                self.file_list = []
                if self.first_round == True:
                    self.prev_dir = 'C:\Google Drive'
                else:
                    self.prev_dir = self.open_dir

                self.fileDir = QFileDialog.getOpenFileName(
                    self, 'Select File to Open:', self.prev_dir)

                self.open_dir = ''

                if self.fileDir != '':
                    self.file_list = str(self.fileDir).split('/')
                    for i in range(0, len(self.file_list) - 1):
                        if i < len(self.file_list) - 1:
                            self.open_dir += self.file_list[i] + '\\'
                        elif i == len(self.file_list) - 1:
                            self.open_dir += self.file_list[i]
                    self.fileDir.replace('/', '\\')
                    self.ui.lineEdit_address.setText(self.fileDir)
                    self.ui.lineEdit_condition1.setText('Open Source File')
                else:
                    self.ui.lineEdit_address.setText('None')
                    self.ui.lineEdit_condition1.setText('Failed to Read File')

                self.ui.pushButton_show.setEnabled(True)
                self.ui.groupBox_labels.setEnabled(True)
                self.first_round = False
                self.file_name = self.file_list[len(self.file_list) - 1]
        else:
            self.file_list = []

            if self.first_round == True:
                self.prev_dir = 'C:\Google Drive'
            else:
                self.prev_dir = self.open_dir

            self.fileDir = QFileDialog.getOpenFileName(self,
                                                       'Select File to Open:',
                                                       self.prev_dir)

            self.open_dir = ''

            if self.fileDir != '':
                self.file_list = str(self.fileDir).split('/')
                for i in range(0, len(self.file_list) - 1):
                    if i < len(self.file_list) - 1:
                        self.open_dir += self.file_list[i] + '\\'
                    elif i == len(self.file_list) - 1:
                        self.open_dir += self.file_list[i]
                self.fileDir.replace('/', '\\')
                self.ui.lineEdit_address.setText(self.fileDir)
                self.ui.lineEdit_condition1.setText('Open Source File')
            else:
                self.ui.lineEdit_address.setText('None')
                self.ui.lineEdit_condition1.setText('Failed to Read File')

            self.ui.pushButton_show.setEnabled(True)
            self.ui.groupBox_labels.setEnabled(True)
            self.first_round = False
            self.file_name = self.file_list[len(self.file_list) - 1]

    def Show(self):
        '''self.item_info = []
        self.item_info.append(self.divider)
        self.item_info.append(self.file_name)
        self.item_info.append(self.ui.lineEdit_address.text())
        self.all_info.append(self.item_info)'''

        self.ui.comboBox_recent.addItem(self.file_name)
        # Every you click the "Open File" button, the content in the plot combox is refreshed
        self.reset_plot()
        self.ui.comboBox_x.clear()
        self.ui.comboBox_y.clear()
        self.ui.lineEdit_x.setText('')
        self.ui.lineEdit_y.setText('')
        self.ui.lineEdit_name.setText('')
        self.ui.lineEdit_condition2.setText('')

        labels = ''

        raw_parameters = []
        units = []
        parameters = []
        new_parameters = []
        self.data = []
        new_data = []
        first_data = []

        item = 0
        length = 0
        # This count is the number of the reading loops for labels. If count is over 1000, that means there's something wrong with the divider line and it helps to stop the program
        count = 0

        # It is false when program cannot find the line divider and runs the while loop 1000 times
        self.divider_found = True

        self.fileDir = self.ui.lineEdit_address.text()
        fp = open(self.fileDir)
        while True:
            if count > 1000:
                self.ui.lineEdit_condition1.setText(
                    'Line divider is not found.')
                self.divider_found = False
                break
            line = fp.readline()
            #print 1, line
            linelist = line.split(',')
            if linelist[0].upper() == self.divider.upper() + '\n':
                break
            for i in range(0, len(linelist)):
                labels += linelist[i]
            count += 1

        if self.divider_found == True:
            self.ui.textEdit_labels.setText(labels)
            parameters = fp.readline().replace("\n", '')
            parameters = parameters.split(',')

            if self.ui.radioButton_ppms.isChecked():
                lines = fp.readline().replace("\n", "")
                first_data = lines.split(",")
                for i in range(0, len(first_data)):
                    par = []
                    if first_data[i] != '':
                        par = parameters[i].split(' (')
                        if len(par) == 2:
                            units.append(par[1].replace(')', ''))
                        elif len(par) == 1:
                            units.append('1')
                        new_parameters.append(par[0])
                parameters = new_parameters
            elif self.ui.radioButton_qmd.isChecked():
                units = fp.readline().replace("\n", '')
                units = units.split(",")

            for i in range(0, len(parameters)):
                self.ui.comboBox_x.addItem(parameters[i])
                self.ui.comboBox_y.addItem(parameters[i])

            for i in range(0, len(parameters)):
                new_data = [parameters[i], units[i], []]
                self.data.append(new_data)

            if self.ui.radioButton_ppms.isChecked():
                for i in range(0, len(first_data)):
                    if first_data[i] != '':
                        self.data[item][2].append(first_data[i])
                        item += 1

                length = item

            while True:
                val = []
                lines = fp.readline().replace("\n", "")
                if lines == '':
                    break
                values = lines.split(',')
                for i in range(0, len(values)):
                    if values[0] != 'Measurement was Aborted':
                        if values[i] != '':
                            val.append(values[i])
                if self.ui.radioButton_ppms.isChecked():
                    if len(val) < length:
                        break

                for i in range(0, len(val)):
                    self.data[i][2].append(val[i])

            self.ui.groupBox_plot.setEnabled(True)
            self.ui.lineEdit_condition2.setText('Ready to Plot.')
            self.ui.label_11.setEnabled(True)
            self.ui.comboBox_recent.setEnabled(True)
            self.ui.pushButton_recent.setEnabled(True)

    def Open_Recent(self):
        pass

    def Choose2(self):
        self.x_value = 0
        self.y_value = 0
        self.x_value = self.ui.comboBox_x.currentIndex()
        self.y_value = self.ui.comboBox_y.currentIndex()

        xaxis = self.data[self.x_value][0]
        xunit = self.data[self.x_value][1]
        yaxis = self.data[self.y_value][0]
        yunit = self.data[self.y_value][1]

        self.ui.lineEdit_x.setText(xaxis + ' (' + xunit + ')')
        self.ui.lineEdit_y.setText(yaxis + ' (' + yunit + ')')
        self.ui.lineEdit_name.setText(yaxis + ' vs ' + xaxis)

        self.ui.pushButton_plot.setEnabled(True)
        self.Plot()

    def Plot(self):

        self.reset_plot()
        self.axes.plot(self.data[self.x_value][2],
                       self.data[self.y_value][2],
                       marker='.',
                       linestyle=':')
        self.axes.set_title(self.ui.lineEdit_name.text())
        self.axes.set_xlabel(self.ui.lineEdit_x.text())
        self.axes.set_ylabel(self.ui.lineEdit_y.text())
        self.ui.mplwidget.draw()

        self.ui.lineEdit_condition2.setText('Plot successfully.')

    def reset_plot(self):
        self.ui.mplwidget.figure.clear()
        self.axes = self.ui.mplwidget.figure.add_subplot(111)

    def closeEvent(self, question):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, 'Message', quit_msg,
                                     QMessageBox.Yes, QMessageBox.No)
        # If your answer is "yes", then quit
        if reply == QMessageBox.Yes:
            question.accept()
        # If your answer is "no", then get back
        else:
            question.ignore()
Exemplo n.º 24
0
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()
        self.dsave_thread = Dynamic_Save_Thread()
        
        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))

        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("freq_update"), self.setup_plot)
        self.connect(self.collect_data_thread, SIGNAL("power_update"), self.setup_plot)
        self.connect(self.collect_data_thread, SIGNAL("print_value"), self.Print_data)
        self.connect(self.ui.checkBox_dsave, SIGNAL("clicked()"), self.Pre_dsave)
        self.connect(self.ui.textEdit, SIGNAL('textChanged ()'), lambda : self.start_font("C"))
        self.connect(self.ui.pushButton_dsave_browse, SIGNAL('clicked()'), self.Dsave_browse)
        
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)

        
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "b", "blue", titles = ["RF Power", "Step", "Power (dBm)"])
        self.curve_3_update = self.make_curveWidgets(self.ui.curvewidget_3, "red", "red", titles = ["RF Power", "Step", "Power (dBm)"])
        self.curve_2_b = self.make_curveWidgets(self.ui.curvewidget_2, "black", "black", titles = ["Keithley", "Step", "Voltage (V)"])
        self.curve_2_r = self.make_curveWidgets(self.ui.curvewidget_2, "red", "red", titles = ["Keithley", "Step", "Voltage (V)"])
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "b", "blue", titles = ["Keithley", "Step", "Voltage (V)"])
        self.curve_1_update = self.make_curveWidgets(self.ui.curvewidget_1, "red", "red", titles = ["Keithley", "Step", "Voltage (V)"])
        
        self.Curve1 = self.make_curveWidgets(self.ui.curvewidget_13, "r", "black", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.Curve2 = self.make_curveWidgets(self.ui.curvewidget_14, "g", "black", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.Curve3 = self.make_curveWidgets(self.ui.curvewidget_15, "r", "black", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.Curve4 = self.make_curveWidgets(self.ui.curvewidget_16, "g", "black", titles = ["Plot 4", "X (x)", "Y (y)"])
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None

        self.setWindowTitle("Sweep with Lock-in without Magnetic")
        
        self.input_string = []
        self.input_string.append("00 Run_Number: 01\n\n")
        
        self.input_string.append("01 Start Gate Voltage (V): 35\n")
        self.input_string.append("02 Step Gate Voltage (V): 0.1\n")
        self.input_string.append("03 03 End Gate Voltage (V): 0\n")
        self.input_string.append("04 Double Linear (y/n): n\n")
        self.input_string.append("05 Turn off Keithley (y/n): n\n")
        self.input_string.append("06 CurrBias Lock-in Sens: 20e-6\n")
        self.input_string.append("07 Reading wait time(s): .25\n\n")
        
        self.input_string.append("08 Current Bias Resistor: 1.009e9\n")
        self.input_string.append("09 Lock-in Output Voltage: 10\n")
        self.input_string.append("10 Additional Lock-in Volt Gain: 1\n\n")
        
        self.input_string.append("11 User: Vannucci\n")
        self.input_string.append("12 File Name: S-SiA0036-V_B1_leads\n")
        self.input_string.append("13 TimeStamp (y/n): y\n")
        self.input_string.append("14 Comments:\n\n")
        self.start_font("C")        
        
        self.out0_1 = "None"
        self.out0_2 = "None"
        self.out0_3 = "None"
        self.out0_4 = "None"
        self.out1_0 = "None"
        self.out1_1 = "None"
        self.out1_2 = "None"
        self.out1_3 = "None"
        self.out2_0 = "None"
        self.out2_1 = "None"
        self.out2_2 = "None"
        self.out2_3 = "None"
        self.out3_0 = "None"
        self.out3_1 = "None"
        self.out3_2 = "None"
        self.out3_3 = "None"
        
        self.scale1_1 = [1, ""]
        self.scale1_2 = [1, ""]
        self.scale1_3 = [1, ""]
        self.scale2_1 = [1, ""]
        self.scale2_2 = [1, ""]
        self.scale2_3 = [1, ""]
        self.scale3_1 = [1, ""]
        self.scale3_2 = [1, ""]
        self.scale3_3 = [1, ""]
        
        self.Coil_Constant = 1E-3/9.1E-3
        
        try:
            file_par = open("parameters.txt", "r")
            input_string_print = file_par.read()
            
        except:
            input_string_print = ""
            for st_i in self.input_string:
                input_string_print = input_string_print + st_i
        
        font = QFont()
        font.setPointSize(10)
        self.ui.textEdit.setFont(font)
        self.ui.textEdit.setText(input_string_print)
        
        self.inputted_data = []
        self.array_sweep = []
        
        rm = visa.ResourceManager()
        self.Visa_states = []
        try:
            alls = rm.list_resources()
            for i in range(0, len(alls)):
                self.Visa_states.append([alls[i], True])
        except:
            alls = "No Visa Available."
Exemplo n.º 25
0
    def __init__(self, parent=None):

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to "self.ui"
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # For the canvas.
        self.canvas_general = FigureCanvas(self.ui.mplwidget_general.figure)
        self.canvas_general.setParent(self.ui.widget_general)
        # We need the toolbar widget for the canvas
        self.mpl_toolbar_general = NavigationToolbar(self.canvas_general,
                                                     self.ui.widget_general)

        self.canvas_import = FigureCanvas(self.ui.mplwidget_import.figure)
        self.canvas_import.setParent(self.ui.widget_import)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import = NavigationToolbar(self.canvas_import,
                                                    self.ui.widget_import)

        self.canvas_analysis = FigureCanvas(self.ui.mplwidget_analysis.figure)
        self.canvas_analysis.setParent(self.ui.widget_analysis)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_analysis = NavigationToolbar(self.canvas_analysis,
                                                      self.ui.widget_analysis)

        self.canvas_ct_analysis = FigureCanvas(
            self.ui.mplwidget_ct_analysis.figure)
        self.canvas_ct_analysis.setParent(self.ui.widget_ct_analysis)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_ct_analysis = NavigationToolbar(
            self.canvas_ct_analysis, self.ui.widget_ct_analysis)

        self.canvas_vt_analysis = FigureCanvas(
            self.ui.mplwidget_vt_analysis.figure)
        self.canvas_vt_analysis.setParent(self.ui.widget_vt_analysis)
        # This is the toolbar widget for the scan canvas
        self.mpl_toolbar_vt_analysis = NavigationToolbar(
            self.canvas_vt_analysis, self.ui.widget_vt_analysis)

        # Create the QVBoxLayout object and add the widget into the layout
        vbox_general = QVBoxLayout()
        # The matplotlib canvas
        vbox_general.addWidget(self.canvas_general)
        # The matplotlib toolbar
        vbox_general.addWidget(self.mpl_toolbar_general)
        self.ui.widget_general.setLayout(vbox_general)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import = QVBoxLayout()
        # The matplotlib canvas
        vbox_import.addWidget(self.canvas_import)
        # The matplotlib toolbar
        vbox_import.addWidget(self.mpl_toolbar_import)
        self.ui.widget_import.setLayout(vbox_import)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_analysis = QVBoxLayout()
        # The matplotlib canvas
        vbox_analysis.addWidget(self.canvas_analysis)
        # The matplotlib toolbar
        vbox_analysis.addWidget(self.mpl_toolbar_analysis)
        self.ui.widget_analysis.setLayout(vbox_analysis)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_ct_analysis = QVBoxLayout()
        # The matplotlib canvas
        vbox_ct_analysis.addWidget(self.canvas_ct_analysis)
        # The matplotlib toolbar
        vbox_ct_analysis.addWidget(self.mpl_toolbar_ct_analysis)
        self.ui.widget_ct_analysis.setLayout(vbox_ct_analysis)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_vt_analysis = QVBoxLayout()
        # The matplotlib canvas
        vbox_vt_analysis.addWidget(self.canvas_vt_analysis)
        # The matplotlib toolbar
        vbox_vt_analysis.addWidget(self.mpl_toolbar_vt_analysis)
        self.ui.widget_vt_analysis.setLayout(vbox_vt_analysis)

        # Connect the mplwidget with canvas
        self.ui.mplwidget_general = self.canvas_general
        self.ui.mplwidget_import = self.canvas_import
        self.ui.mplwidget_analysis = self.canvas_analysis
        self.ui.mplwidget_ct_analysis = self.canvas_ct_analysis
        self.ui.mplwidget_vt_analysis = self.canvas_vt_analysis

        self.connect(self.ui.actionArray_Builder, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(0))
        self.connect(self.ui.actionKeithley_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(1))
        self.connect(self.ui.actionAgilent_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(2))
        self.connect(self.ui.actionLockIn_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(3))
        self.connect(self.ui.actionResonant_Single_scan, SIGNAL('triggered()'),
                     lambda: self.ui.stackedWidget.setCurrentIndex(4))

        self.Array_Builder_programs = Array_Builder(main=self, ui=self.ui)
        # Connect buttons in the General Array tab
        # self.connect(self.ui.pushButton_plot, SIGNAL('clicked()'), self.Array_Builder_programs.Plot_general)
        # self.connect(self.ui.pushButton_clear, SIGNAL('clicked()'), self.Array_Builder_programs.Clear)
        # self.connect(self.ui.pushButton_save, SIGNAL('clicked()'), self.Array_Builder_programs.Save)
        # self.connect(self.ui.pushButton_browse_save, SIGNAL('clicked()'), self.Array_Builder_programs.Browse_save)

        self.Keithley_programs = Keithley(main=self, ui=self.ui)
        self.Keithley_programs.Refresh_visa()
        # Connect buttons in the Keithely tab
        # self.connect(self.ui.pushButton_browse_keithley, SIGNAL('clicked()'), self.Keithley_programs.Browse_keithley)
        # self.connect(self.ui.pushButton_import_keithley, SIGNAL('clicked()'), self.Keithley_programs.Import_keithley)
        # self.connect(self.ui.pushButton_close_keithley, SIGNAL('clicked()'), self.Keithley_programs.Close_keithley)
        # self.connect(self.ui.pushButton_select_keithley, SIGNAL('clicked()'), self.Keithley_programs.Select_keithley)
        # self.connect(self.ui.pushButton_scan_keithley, SIGNAL('clicked()'), self.Keithley_programs.Scan_keithley)
        # self.connect(self.ui.pushButton_pause_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.pause)
        # self.connect(self.ui.pushButton_stop_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.stop)
        # self.connect(self.ui.pushButton_clear_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.Clear_keithley)
        # self.connect(self.ui.pushButton_copy_keithley, SIGNAL('clicked()'), self.CopyData)
        # self.connect(self.ui.pushButton_fit_keithley, SIGNAL('clicked()'), self.Keithley_programs.collect_data_thread.Fit)
        # self.connect(self.ui.radioButton_voltage_keithley, SIGNAL("clicked()"), lambda : self.ui.tabWidget_scan_keithley.setCurrentIndex(0))
        # self.connect(self.ui.radioButton_timescan_keithley, SIGNAL("clicked()"), self.Keithley_programs.collect_data_thread.Pre_MPL_Plot)
        # self.connect(self.ui.radioButton_stepscan_keithley, SIGNAL("clicked()"), self.Keithley_programs.collect_data_thread.Pre_MPL_Plot)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("plot"), self.Keithley_programs.Plot_data)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("mpl_plot"), self.Keithley_programs.Plot_analysis)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("data_available"), self.Keithley_programs.Pre_save)
        # self.connect(self.Keithley_programs.collect_data_thread, SIGNAL("clear_plot"), self.Keithley_programs.Clear_plot)

        self.Values = []
        self.Step = []
        self.Peak = []
        self.round = 0
        self.new_start = 0

        self.ui.lineEdit_directory_save.setText(os.getcwd())

        #Agilent_Yokogawa
        self.canvas_import_ay = FigureCanvas(
            self.ui.mplwidget_import_ay.figure)
        self.canvas_import_ay.setParent(self.ui.widget_import_ay)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_import_ay = NavigationToolbar(
            self.canvas_import_ay, self.ui.widget_import_ay)

        #Agilent_Yokogawa
        self.canvas_analysis_ay = FigureCanvas(
            self.ui.mplwidget_analysis_ay.figure)
        self.canvas_analysis_ay.setParent(self.ui.widget_analysis_ay)
        # This is the toolbar widget for the import canvas
        self.mpl_toolbar_analysis_ay = NavigationToolbar(
            self.canvas_analysis_ay, self.ui.widget_analysis_ay)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_import_ay = QVBoxLayout()
        # The matplotlib canvas
        vbox_import_ay.addWidget(self.canvas_import_ay)
        # The matplotlib toolbar
        vbox_import_ay.addWidget(self.mpl_toolbar_import_ay)
        self.ui.widget_import_ay.setLayout(vbox_import_ay)

        # Create the QVBoxLayout object and add the widget into the Layout
        vbox_analysis_ay = QVBoxLayout()
        # The matplotlib canvas
        vbox_analysis_ay.addWidget(self.canvas_analysis_ay)
        # The matplotlib toolbar
        vbox_analysis_ay.addWidget(self.mpl_toolbar_analysis_ay)
        self.ui.widget_analysis_ay.setLayout(vbox_analysis_ay)

        # Connect the mplwidget with canvass
        self.ui.mplwidget_import_ay = self.canvas_import_ay
        self.ui.mplwidget_analysis_ay = self.canvas_analysis_ay

        # Connect buttons in the Agilent Yokogawa tab
        self.Agilent_Yokogawa_programs = Agilent_Yokogawa(self.ui)

        self.connect(self.ui.startButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.start)
        self.connect(self.ui.stopButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.stop)
        #self.connect(self.action_timer, SIGNAL("timeout()"), self.Agilent_Yokogawa_programs.action)

        self.connect(self.ui.pushButton_browse_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Browse_ay)
        self.connect(self.ui.pushButton_import_ay, SIGNAL('clicked()'),
                     self.Agilent_Yokogawa_programs.Import_ay)
        self.connect(self.ui.pushButton_copy_ay, SIGNAL('clicked()'),
                     self.Copy_ay)
        self.connect(self.ui.selectVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.select_visa)
        self.connect(self.ui.updateVisaButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.update_visa)
        self.connect(self.ui.closeVisaButton0, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa0)
        self.connect(self.ui.closeVisaButton1, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.close_visa1)
        self.connect(self.Agilent_Yokogawa_programs.collectDataThread,
                     SIGNAL("plot"), self.Agilent_Yokogawa_programs.plotData)
        self.connect(self.Agilent_Yokogawa_programs.collectDataThread,
                     SIGNAL("analyse"), self.Agilent_Yokogawa_programs.analyse)

        self.connect(self.ui.browseButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.browse)
        self.connect(self.ui.saveButton, SIGNAL("clicked()"),
                     self.Agilent_Yokogawa_programs.save)

        self.ui.mplwidget_analysis_ay.figure.canvas.mpl_connect(
            'button_release_event', self.Agilent_Yokogawa_programs.slope)

        #Connects the buttons in the Resonant Sweeper Tab
        #self.Resonant_programs = Resonance_Sweep(main = self, ui = self.ui)
        self.Resonant_Sweeper_functions = Resonance_Sweep(main=self,
                                                          ui=self.ui)

        self.Resonant_Sweeper_functions.update_visa()
        self.connect(self.ui.pushButtonSelectRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.choose_visa)
        self.connect(self.ui.pushButtonUpdateRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.update_visa)
        self.connect(self.ui.pushButtonSourceSelectRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.choose_visa)
        self.connect(self.ui.pushButtonSourceUpdateRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.update_visa)
        self.connect(self.ui.pushButtonStartRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.start)
        self.connect(self.ui.pushButtonStopRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.stop)
        self.connect(self.ui.pushButtonPauseRS, SIGNAL("clicked()"),
                     self.Resonant_Sweeper_functions.collectDataThread.pause)
        self.connect(self.Resonant_Sweeper_functions.collectDataThread,
                     SIGNAL("plot"), self.Resonant_Sweeper_functions.plotData)
        self.connect(self.Resonant_Sweeper_functions.collectDataThread,
                     SIGNAL("stop"), self.Resonant_Sweeper_functions.stop)

        #Connects the buttons in the Lock-In Program Tab
        self.Lock_In_programs = Lock_In(main=self, ui=self.ui)
        self.connect(self.ui.pushButtonSelectLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.choose_visa)
        self.connect(self.ui.pushButtonUpdateLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.update_visa)
        self.connect(self.ui.pushButtonStartLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.start)
        self.connect(self.ui.pushButtonStopLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.stop)
        self.connect(self.ui.pushButtonStopLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.final_append)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.append_parameters)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL("clicked()"),
                     self.Lock_In_programs.collectDataThread.pause)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.collectDataThread.appendSessionData)
        self.connect(self.ui.pushButtonStopLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.collectDataThread.appendSessionData)
        self.connect(self.ui.pushButtonPauseLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.tableAppend)
        self.connect(self.ui.pushButtonStopLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.tableAppend)

        self.connect(self.ui.pushButtonSaveLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save)
        self.connect(self.ui.pushButtonBrowseLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.browse)
        self.connect(self.ui.pushButtonFolderSelectLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.select_name)

        self.connect(self.ui.radioButtonDateTimeLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_name)
        self.connect(self.ui.radioButtonCustomLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_name)

        self.connect(self.ui.radioButton_csvLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_type)
        self.connect(self.ui.radioButton_txtLI, SIGNAL('clicked()'),
                     self.Lock_In_programs.save_type)

        self.connect(self.Lock_In_programs.collectDataThread, SIGNAL("plot"),
                     self.Lock_In_programs.plotData)
        self.connect(self.Lock_In_programs.collectDataThread,
                     SIGNAL("Begin_Save"), self.Lock_In_programs.pre_save)
Exemplo n.º 26
0
class MyForm(QMainWindow):
    
    # The __init__ function is what is everything the user wants to be initialized when the class is called.
    # Here we shall define the trig functions to corresponding variables.
    # Note that the "self" variable means that the function is part of the class and can be called inside and outside the class.(Although __init__ is special.)
    def __init__(self, parent = None):
        
        self.collect_data_thread = Collect_data()

        # Standard GUI code
        QWidget.__init__(self, parent)

        # All the GUI data and widgets in the Ui_MainWindow() class is defined to self.ui
        # Thus to do anything on the GUI, the commands must go through this variable
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)
        
        self.connect(self.ui.pushButton_update_1, SIGNAL('clicked()'), lambda : self.Update("visa1", self.ui.comboBox_visa_1))
        self.connect(self.ui.pushButton_update_2, SIGNAL('clicked()'), lambda : self.Update("visa2", self.ui.comboBox_visa_2))
        self.connect(self.ui.pushButton_update_3, SIGNAL('clicked()'), lambda : self.Update("visa3", self.ui.comboBox_visa_3))
        self.connect(self.ui.pushButton_update_4, SIGNAL('clicked()'), lambda : self.Update("visa4", self.ui.comboBox_visa_4))
        self.connect(self.ui.pushButton_update_5, SIGNAL('clicked()'), lambda : self.Update("visa5", self.ui.comboBox_visa_5))
        self.connect(self.ui.pushButton_select_1, SIGNAL('clicked()'), lambda : self.Select("visa1", self.visa1, self.ui.comboBox_visa_1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_select_2, SIGNAL('clicked()'), lambda : self.Select("visa2", self.visa2, self.ui.comboBox_visa_2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_select_3, SIGNAL('clicked()'), lambda : self.Select("visa3", self.visa3, self.ui.comboBox_visa_3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_select_4, SIGNAL('clicked()'), lambda : self.Select("visa4", self.visa4, self.ui.comboBox_visa_4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_select_5, SIGNAL('clicked()'), lambda : self.Select("visa5", self.visa5, self.ui.comboBox_visa_5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_close_1, SIGNAL('clicked()'), lambda : self.Close("visa1", self.visa1, self.ui.label_visa_1, [self.ui.pushButton_select_1, self.ui.pushButton_close_1]))
        self.connect(self.ui.pushButton_close_2, SIGNAL('clicked()'), lambda : self.Close("visa2", self.visa2, self.ui.label_visa_2, [self.ui.pushButton_select_2, self.ui.pushButton_close_2]))
        self.connect(self.ui.pushButton_close_3, SIGNAL('clicked()'), lambda : self.Close("visa3", self.visa3, self.ui.label_visa_3, [self.ui.pushButton_select_3, self.ui.pushButton_close_3]))
        self.connect(self.ui.pushButton_close_4, SIGNAL('clicked()'), lambda : self.Close("visa4", self.visa4, self.ui.label_visa_4, [self.ui.pushButton_select_4, self.ui.pushButton_close_4]))
        self.connect(self.ui.pushButton_close_5, SIGNAL('clicked()'), lambda : self.Close("visa5", self.visa5, self.ui.label_visa_5, [self.ui.pushButton_select_5, self.ui.pushButton_close_5]))
        self.connect(self.ui.pushButton_Start, SIGNAL('clicked()'), self.start)
        self.connect(self.ui.pushButton_Stop, SIGNAL('clicked()'), self.collect_data_thread.stop)
        self.connect(self.ui.pushButton_Pause, SIGNAL('clicked()'), self.collect_data_thread.pause)
        self.connect(self.collect_data_thread, SIGNAL("curve_plot"), self.curvePlots_update)
        self.connect(self.collect_data_thread, SIGNAL("print"), self.Print_data)
        
        self.Update("visa1", self.ui.comboBox_visa_1)
        self.Update("visa2", self.ui.comboBox_visa_2)
        self.Update("visa3", self.ui.comboBox_visa_3)
        self.Update("visa4", self.ui.comboBox_visa_4)
        self.Update("visa5", self.ui.comboBox_visa_5)
        
        self.curve_1 = self.make_curveWidgets(self.ui.curvewidget_1, "b", titles = ["Plot 1", "X (x)", "Y (y)"])
        self.curve_2 = self.make_curveWidgets(self.ui.curvewidget_2, "b", titles = ["Plot 2", "X (x)", "Y (y)"])
        self.curve_3 = self.make_curveWidgets(self.ui.curvewidget_3, "b", titles = ["Plot 3", "X (x)", "Y (y)"])
        self.curve_4 = self.make_curveWidgets(self.ui.curvewidget_4, "b", titles = ["Plot 4", "X (x)", "Y (y)"])
        
        self.visa1 = None
        self.visa2 = None
        self.visa3 = None
        self.visa4 = None
        self.visa5 = None
        
    def Update(self, signal, comboBoxVisa):
        rm = visa.ResourceManager()
        try:
            alls = rm.list_resources()
        except:
            alls = "No Visa Available."
        if signal == "visa1":
            self.ui.comboBox_visa_1.clear()
        elif signal == "visa2":
            self.ui.comboBox_visa_2.clear()
        elif signal == "visa3":
            self.ui.comboBox_visa_3.clear()
        elif signal == "visa4":
            self.ui.comboBox_visa_4.clear()
        elif signal == "visa5":
            self.ui.comboBox_visa_5.clear()

        
        for temp in alls:
            if signal == "visa1":
                self.ui.comboBox_visa_1.addItem(temp)
            elif signal == "visa2":
                self.ui.comboBox_visa_2.addItem(temp)
            elif signal == "visa3":
                self.ui.comboBox_visa_3.addItem(temp)
            elif signal == "visa4":
                self.ui.comboBox_visa_4.addItem(temp)
            elif signal == "visa5":
                self.ui.comboBox_visa_5.addItem(temp)
    
    def Select(self, signal, visa_chosen, comboBoxVisa, lineEditVisa, selectClose):
        visa_address = str(comboBoxVisa.currentText())
        rm = visa.ResourceManager()
        rm.list_resources()
        inst = rm.open_resource(visa_address)
        visa_check = self.Check(inst)
        if visa_check == True:
            self.ui.label_condition.setText("Visa is selected succefully!")
            visa_name = inst.query("*IDN?")
            name_list = visa_name.split(',')
            first_name = name_list[0]
            if signal == "visa1":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.Enable(signal)
                self.visa1 = inst
            elif signal == "visa2":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.Enable(signal)
                self.visa2 = inst
            elif signal == "visa3":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.Enable(signal)
                self.visa3 = inst
            elif signal == "visa4":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.Enable(signal)
                self.visa4 = inst
            elif signal == "visa5":
                lineEditVisa.setText(visa_name)
                selectClose[0].setEnabled(False)
                selectClose[1].setEnabled(True)
                self.Enable(signal)
                self.visa5 = inst
        elif visa_check == False:
            self.ui.label_condition.setText("Invalid visa address.")
            lineEditVisa.setText("None.")
            visa_chosen = False
    
    def Close(self, signal, visa_chosen, lineEditVisa, selectClose):
        self.ui.label_condition.setText('Visa address is closed')
        lineEditVisa.setText('')
        selectClose[0].setEnabled(True)
        selectClose[1].setEnabled(False)
        if signal == "visa1":
            visa_chosen.close()
            self.visa1 = None
            self.Disable(signal)
        elif signal == "visa2":
            visa_chosen.close()
            self.visa2 = None
            self.Disable(signal)
        elif signal == "visa3":
            visa_chosen.close()
            self.visa3 = None
            self.Disable(signal)
        elif signal == "visa4":
            visa_chosen.close()
            self.visa4 = None
            self.Disable(signal)
        elif signal == "visa5":
            visa_chosen.close()
            self.visa5 = None
            self.Disable(signal)
            
    def Check(self, inst):
        try:
            inst.ask("*IDN?")
            valid = True
        except:
            valid = False
        return valid
    
    def make_curveWidgets(self, curvewidget, color, titles):
        curve_temp = make.curve([], [], color = 'b', marker = "o")
        curvewidget.plot.add_item(curve_temp)
        curvewidget.plot.set_antialiasing(True)
        curvewidget.plot.set_titles(titles[0], titles[1], titles[2])
        return curve_temp
    
    def start(self):
        instruments = [self.visa1, self.visa2, self.visa3, self.visa4, self.visa5]
        curves = [self.curve_1, self.curve_2, self.curve_3, self.curve_4]
        curveWidgets =[self.ui.curvewidget_1, self.ui.curvewidget_2, self.ui.curvewidget_3, self.ui.curvewidget_4]
        go_on = None
        self.collect_data_thread.input(self.ui, instruments, curves, curveWidgets, go_on)
                        
    def curvePlots_update(self, curveInfo):
        curveWidget = curveInfo[0]
        curve = curveInfo[1]
        curveWidget.plot.do_autoscale()
        curve.plot().replot()
    
    def Print_data(self, step, time, V1, V2, V3, V4, V5, V6):
        self.ui.label_step.setText(step)
        self.ui.label_time.setText(time)
        self.ui.label_V1.setText(V1)
        self.ui.label_V2.setText(V2)
        self.ui.label_V3.setText(V3)
        self.ui.label_V4.setText(V4)
        self.ui.label_V5.setText(V5)
        self.ui.label_V6.setText(V6)
    
    def Switch_scale(self, num):
        temp = abs(num)
        if temp >= 1E9:
            scale = [1E-9, "G"]
        elif temp >= 1E6 and temp < 1E9:
            scale = [1E-6, "M"]
        elif temp >= 1E3 and temp < 1E6:
            scale = [1E-3, "k"]
        elif temp >= 1 and temp < 1000:
            scale = [1, ""]
        elif temp >= 1E-3 and temp < 1:
            scale = [1E3, "m"]
        elif temp >= 1E-6 and temp < 1E-3:
            scale = [1E6, "u"]
        elif temp >= 1E-9 and temp < 1E-6:
            scale = [1E9, "n"]
        elif temp < 1E-9:
            scale = [1E12, "p"]
            
        return scale
    
    def closeEvent(self, event):
        quit_msg = "Do you want to quit the program?"
        reply = QMessageBox.question(self, "Message", quit_msg, QMessageBox.Yes, QMessageBox.No)
        if reply == QMessageBox.Yes:
            event.accept()
        else:
            event.ignore()