def apply_fit(self): msgBox = QMessageBox() msgBox.setText("Apply corrections?") msgBox.setInformativeText( "Select to which files will be applied the correction.") CancelButton = msgBox.addButton("Cancel", msgBox.RejectRole) SelectButton = msgBox.addButton("Select", msgBox.NoRole) AllButton = msgBox.addButton("All", msgBox.YesRole) #msgBox.setDefaultButton(QMessageBox.Cancel) msgBox.exec_() if msgBox.clickedButton() == CancelButton: return -1 elif msgBox.clickedButton() == AllButton: filelist = None elif msgBox.clickedButton() == SelectButton: flret = getopenfilenames(self, _("Select file list"), self.data_ref.dir_name)[0] filelist = [] for f in flret: filelist.append((basename(str(f)), dirname(str(f)))) msgBox = QMessageBox() msgBox.setText("Select the correction file") msgBox.addButton(QMessageBox.Ok) msgBox.exec_() filename = getopenfilename( self, _("Correction file"), self.data_ref.dir_name + "/" + basename( normpath(self.data_ref.dir_name) + "_fit." + self.data_ref.files_type)) try: self.data_ref.applycorr(str(filename[0]), filelist) except: msgBox = QMessageBox() msgBox.setText("Not a correction file!") msgBox.exec_() return -1
def fitfile(self): if self.fitc == None: msgBox = QMessageBox() msgBox.setText("Fit not yet executed!") msgBox.exec_() return -1 msgBox = QMessageBox() msgBox.setText("Save correction file?") #msgBox.setInformativeText("A correction file will be saved in the root directory of the observation.") msgBox.addButton(QMessageBox.Cancel) msgBox.addButton(QMessageBox.Ok) msgBox.setDefaultButton(QMessageBox.Cancel) ret = msgBox.exec_() if ret != QMessageBox.Ok: return -1 name = getsavefilename(self, _("Save File"), self.data_ref.dir_name + "/" + basename( normpath(self.data_ref.dir_name) + "_fit." + self.data_ref.files_type), options=QFileDialog.DontConfirmOverwrite) if str(name[0]): if exists(str(name[0])): msgBox = QMessageBox() msgBox.setText("File exists!") msgBox.setInformativeText("Do you want to update it?") msgBox.addButton(QMessageBox.Cancel) msgBox.addButton(QMessageBox.Ok) msgBox.setDefaultButton(QMessageBox.Cancel) ret = msgBox.exec_() if ret != QMessageBox.Ok: return -1 self.data_ref.fitfile(str(name[0]), self.fitc)