Python PyMcaGui.PyMcaFileDialogs类代码示例

    def _getFileList(self, fileTypeList, message=None, getfilter=None):
        if message is None:
            message = "Please select a file"
        if getfilter is None:
            getfilter = False
        wdir = PyMcaDirs.inputDir
        filterused = None
        if getfilter:
            filelist, filterused = PyMcaFileDialogs.getFileList(self.parent,
                            filetypelist=fileTypeList,
                            mode="OPEN",
                            message=message,
                            currentdir=wdir,
                            getfilter=True,
                            single=False,
                            native=True)
        else:
            filelist = PyMcaFileDialogs.getFileList(self.parent,
                            filetypelist=fileTypeList,
                            mode="OPEN",
                            message=message,
                            currentdir=wdir,
                            getfilter=False,
                            single=False,
                            native=True)
        if not(len(filelist)):
            return []
        PyMcaDirs.inputDir = os.path.dirname(filelist[0])
        if PyMcaDirs.outputDir is None:
            PyMcaDirs.outputDir = os.path.dirname(filelist[0])

        if getfilter:
            return filelist, filterused
        else:
            return filelist

 def loadButtonClicked(self):
     if self.inputDir is None:
         if self.inputDir is not None:
             self.inputDir = self.outputDir
         else:
             self.inputDir = PyMcaDirs.inputDir
     wdir = self.inputDir
     if not os.path.exists(wdir):
         wdir = os.getcwd()
     filename = PyMcaFileDialogs.getFileList(self,
                         filetypelist=["Energy table files (*.csv)"],
                         mode="OPEN",
                         message="Choose energy table file",
                         currentdir=wdir,
                         single=True)
     if len(filename):
         filename = qt.safe_str(filename[0])
         if len(filename):
             try:
                 self.loadEnergyTableParameters(filename)
                 self.inputDir = os.path.dirname(filename)
                 PyMcaDirs.inputDir = self.inputDir
             except:
                 msg = qt.QMessageBox(self)
                 msg.setIcon(qt.QMessageBox.Critical)
                 msg.setText("Error loading energy table: %s" % (sys.exc_info()[1]))
                 msg.exec_()

 def browseFitConfiguration(self):
     if self._inputDir is None:
         self._inputDir = PyMcaDirs.inputDir
     elif os.path.exists(self._inputDir):
         PyMcaDirs.inputDir = self._inputDir
     filetypes  = ["Configuration Files (*.cfg)"]
     if self._inputDir is None:
         self._inputDir = PyMcaDirs.inputDir
     elif os.path.exists(self._inputDir):
         PyMcaDirs.inputDir = self._inputDir
     message = "Select a Simple Fit Configuration File"
     mode = "OPEN"
     getfilter = False
     currentfilter = None #self._lastInputFileFilter
     fileList = PyMcaFileDialogs.getFileList(self,
                                              filetypelist=filetypes,
                                              message=message,
                                              mode=mode,
                                              getfilter=getfilter,
                                              single=True,
                                              currentfilter=currentfilter)
     if not len(fileList):
         return
     self._inputDir = os.path.dirname(fileList[0])
     self.setFitConfigurationFile(fileList[0])
     self.raise_()

 def _getOutputTiffFilename(self):
     fileTypes = "TIFF Files (*.tif *.tiff *.TIF *.TIFF)"
     message = "Enter output filename"
     wdir = PyMcaDirs.outputDir
     filename = PyMcaFileDialogs.getFileList(self,
                                     message=message,
                                     mode="SAVE",
                                     currentdir=wdir,
                                     filetypelist=[fileTypes],
                                     getfilter=False,
                                     single=True)
     if len(filename):
         filename = filename[0]
     if len(filename):
         try:
             fname = qt.safe_str(filename)
             if fname.endswith('.tif') or\
                fname.endswith('.tiff') or\
                fname.endswith('.TIF') or\
                fname.endswith('.TIFF'):
                 return fname
             else:
                 return fname + ".tif"
         except UnicodeEncodeError:
             msg = qt.QMessageBox(self)
             msg.setWindowTitle("Encoding error")
             msg.setIcon(qt.QMessageBox.Critical)
             msg.setText("Please use ASCII characters in file name and path")
             msg.exec_()
     return ""

 def saveSignal(self):
     fileList = PyMcaFileDialogs.getFileList(self,
                                             filetypelist=['cfg file (*.cfg)'],
                                             mode="SAVE",
                                             single=True,
                                             getfilter=False)
     if len(fileList):
         self.saveConfiguration(fileList[0])

 def browseConfigurationFile(self):
     f = PyMcaFileDialogs.getFileList(parent=self,
                                  filetypelist=["Configuration files (*.cfg)"],
                                  message="Open a fit configuration file",
                                  mode="OPEN",
                                  single=True)
     if len(f):
         self._configLine.setText(f[0])

 def saveImage(self):
     filelist = PyMcaFileDialogs.getFileList(
         parent=self,
         filetypelist=["Image files (*.png)"],
         message="Please give output file name",
         mode="SAVE",
         getfilter=False)
     if len(filelist):
         self.glWidget.saveImage(filelist[0])

 def load(self):
     fileList = PyMcaFileDialogs.getFileList(parent=self,
                                             filetypelist=["Fit files (*.cfg)"],
                                             message="Select a fit configuration file",
                                             mode="OPEN",
                                             getfilter=False,
                                             single=True)
     if len(fileList):
         d = ConfigDict.ConfigDict()
         d.read(fileList[0])
         self.setParameters(d)

 def browseFile(self):
     filelist = PyMcaFileDialogs.getFileList(self,
                             filetypelist=['HDF5 files (*.h5)'],
                             message="Please enter output file",
                             mode="SAVE",
                             single=True)
     if len(filelist):
         name = filelist[0]
         if not name.endswith('.h5'):
             name = name + ".h5"
         self.fileName.setText(name)

def getObject3DInstance(config=None):
    #for the time being a former configuration
    #for serializing purposes is not implemented

    #I do the import here for the case PyMca is not installed
    #because the modules could be instanstiated without using
    #this method
    try:
        from PyMca5.PyMcaIO import EDFStack
        from PyMca5.PyMcaIO import TiffStack
    except ImportError:
        import EDFStack
        import TiffStack

    fileTypeList = ['EDF Z Stack (*edf *ccd)',
                    'EDF X Stack (*edf *ccd)',
                    'TIFF Stack (*tif *tiff)']
    old = PyMcaFileDialogs.PyMcaDirs.nativeFileDialogs * 1
    PyMcaFileDialogs.PyMcaDirs.nativeFileDialogs = False
    fileList, filterUsed = PyMcaFileDialogs.getFileList(
        parent=None,
        filetypelist=fileTypeList,
        message="Please select the object file(s)",
        mode="OPEN",
        getfilter=True)
    PyMcaFileDialogs.PyMcaDirs.nativeFileDialogs = old
    if not len(fileList):
        return None
    if filterUsed == fileTypeList[0]:
        fileindex = 2
    else:
        fileindex = 1
    #file index is irrelevant in case of an actual 3D stack.
    filename = fileList[0]
    legend = os.path.basename(filename)
    if filterUsed == fileTypeList[2]:
        #TIFF
        stack = TiffStack.TiffStack(dtype=numpy.float32, imagestack=False)
        stack.loadFileList(fileList, fileindex=1)
    elif len(fileList) == 1:
        stack = EDFStack.EDFStack(dtype=numpy.float32, imagestack=False)
        stack.loadIndexedStack(filename, fileindex=fileindex)
    else:
        stack = EDFStack.EDFStack(dtype=numpy.float32, imagestack=False)
        stack.loadFileList(fileList, fileindex=fileindex)
    if stack is None:
        raise IOError("Problem reading stack.")
    object3D = Object3DStack(name=legend)
    object3D.setStack(stack)
    return object3D

 def saveParameters(self, fname=None):
     if fname is None:
         fname = PyMcaFileDialogs.getFileList(self,
                                      filetypelist=["Configuration (*.ini)",
                                                    "Configuration (*.cfg)"],
                                      message="Please enter output file name",
                                      mode="SAVE",
                                      getfilter=False,
                                      single=True)
         if len(fname):
             fname = fname[0]
         else:
             return
     ddict = ConfigDict.ConfigDict()
     ddict["XASParameters"] = self.getParameters()
     ddict.write(fname)

 def loadParameters(self, fname=None):
     if fname is None:
         fname = PyMcaFileDialogs.getFileList(self,
                                      filetypelist=["Configuration (*.ini)",
                                                    "Configuration (*.cfg)",
                                                    "All files (*)"],
                                      message="Please set input file name",
                                      mode="OPEN",
                                      getfilter=False,
                                      single=True)
         if len(fname):
             fname = fname[0]
         else:
             return
     d = ConfigDict.ConfigDict()
     d.read(fname)
     self.setParameters(d["XASParameters"])

 def browseOutputDirectory(self):
     if self._outputDir is None:
         self._outputDir = PyMcaDirs.outputDir
     elif os.path.exists(self._outputDir):
         PyMcaDirs.inputDir = self._outputDir
     message = "Select a Simple Fit Configuration File"
     mode = "OPEN"
     fileList = PyMcaFileDialogs.getExistingDirectory(self,
                                              message=message,
                                              mode=mode)
     if not len(fileList):
         return
     if type(fileList) != type([]):
         fileList = [fileList]
     self._outputDir = os.path.dirname(fileList[0])
     self.setOutputDirectory(fileList[0])
     self.raise_()

 def funevent(self,item):
     item=str(item)
     if item in self.specfit.theorylist:
         self.specfit.settheory(item)
     else:
         filelist = PyMcaFileDialogs.getFileList(self,
                      message="Select python module with your function(s)",
                      filetypelist=["Python Files (*.py)",
                                    "All Files (*)"],
                      mode="OPEN",
                      single=True,
                      getfilter=False)
                             
         if not len(filelist):           
             functionsfile = ""
         else:
             functionsfile = filelist[0]
         if len(functionsfile):
             try:
                 if self.specfit.importfun(functionsfile):
                     qt.QMessageBox.critical(self, "ERROR",
                                             "Function not imported")
                     return
                 else:
                     #empty the ComboBox
                     n=self.guiconfig.FunComBox.count()
                     while(self.guiconfig.FunComBox.count()>1):
                       self.guiconfig.FunComBox.removeItem(1)
                     #and fill it again
                     for key in self.specfit.theorylist:
                         if QTVERSION < '4.0.0':
                             self.guiconfig.FunComBox.insertItem(str(key))
                         else:
                             self.guiconfig.FunComBox.addItem(str(key))
             except:
                 qt.QMessageBox.critical(self, "ERROR",
                                         "Function not imported")
         i=1+self.specfit.theorylist.index(self.specfit.fitconfig['fittheory'])
         if QTVERSION < '4.0.0':
             self.guiconfig.FunComBox.setCurrentItem(i)
         else:
             self.guiconfig.FunComBox.setCurrentIndex(i)
     self.__initialparameters()
     return

 def browseList(self):
     if self._inputDir is None:
         self._inputDir = PyMcaDirs.inputDir
     elif os.path.exists(self._inputDir):
         PyMcaDirs.inputDir = self._inputDir
     filetypes  = ["Mca Files (*.mca)",
                   "Edf Files (*.edf)"]
     if HDF5SUPPORT:
         filetypes.append("HDF5 Files(*.nxs *.h5 *.hdf)")
     filetypes.append("SPEC Files (*.spec)")
     filetypes.append("SPEC Files (*.dat)")
     filetypes.append("All files (*)")
     message = "Open a set of files"
     mode = "OPEN"
     getfilter = True
     currentfilter = self._lastInputFileFilter
     fileList, fileFilter  = PyMcaFileDialogs.getFileList(self,
                                              filetypelist=filetypes,
                                              message=message,
                                              mode=mode,
                                              getfilter=getfilter,
                                              single=False,
                                              currentfilter=currentfilter)
     if not len(fileList):
         return
     else:
         self._lastInputFileFilter = fileFilter
     self._inputDir = os.path.dirname(fileList[0])
     if (QTVERSION < '4.2.0') or (not len(self._fileList)):
         self.setFileList(fileList)
         self.raise_()
         return
     msg = qt.QMessageBox()
     msg.setWindowTitle("Append or replace")
     msg.setIcon(qt.QMessageBox.Information)
     msg.setText("Do you want to delete current file list?")
     msg.setStandardButtons(qt.QMessageBox.Yes|qt.QMessageBox.No)
     answer=msg.exec_()
     if answer == qt.QMessageBox.Yes:
         append = False
     else:
         append = True
     self.setFileList(fileList, append=append)
     self.raise_()