Esempio n. 1
0
 def runMapper(self, dataSource, transform):
     '''
     Run the selected mapper
     '''
     self.dataSource = dataSource
     nx = int(self.xDimTxt.text())
     ny = int(self.yDimTxt.text())
     nz = int(self.zDimTxt.text())
     if self.outputFileName == "":
         self.outputFileName = os.path.join(dataSource.projectDir,  \
                                            "%s.vti" %dataSource.projectName)
         self.emit(qtCore.SIGNAL(SET_FILE_NAME_SIGNAL), self.outputFileName)
     if os.access(os.path.dirname(self.outputFileName), os.W_OK):
         self.mapper = QGridMapper(dataSource, \
                                  self.outputFileName, \
                                  nx=nx, ny=ny, nz=nz,
                                  transform = transform,
                                  appConfig=self.appConfig)
         self.mapper.setProgressUpdater(self.updateProgress)
         self.mapper.doMap()
     else:
         self.emit(qtCore.SIGNAL(PROCESS_ERROR_SIGNAL), \
                      "The specified directory \n" + \
                      str(os.path.dirname(self.outputFileName)) + \
                      "\nis not writable")
Esempio n. 2
0
 def runMapper(self, dataSource, transform):
     '''
     Run the selected mapper.  Writer specific class should be specified
     in the application specific subclass.  A list of forms is provided in 
     dataSource classes.
     '''
     logger.debug ("Entering " + self.FORM_TITLE + " " + \
         str(self.gridWriter))
     self.dataSource = dataSource
     nx = int(self.xDimTxt.text())
     ny = int(self.yDimTxt.text())
     nz = int(self.zDimTxt.text())
     self.outputFileName = self.getOutputFileName()
     if self.outputFileName == "":
         self.outputFileName = os.path.join(dataSource.projectDir,  \
             "%s%s" %(dataSource.projectName,self.gridWriter.FILE_EXTENSION) )
         self.setFileName.emit(self.outputFileName)
     if os.access(os.path.dirname(self.outputFileName), os.W_OK):
         self.mapper = QGridMapper(dataSource, \
                                  self.outputFileName, \
                                  self.outputType,\
                                  nx=nx, ny=ny, nz=nz,
                                  transform = transform,
                                  gridWriter = self.gridWriter,
                                  appConfig=self.appConfig)
         self.mapper.setProgressUpdater(self._updateProgress)
         self.mapper.doMap()
     else:
         self.processError.emit("The specified directory \n" + \
                                str(os.path.dirname(self.outputFileName)) + \
                                "\nis not writable")
     logger.debug("Exit " + self.FORM_TITLE)
Esempio n. 3
0
 def runMapper(self, dataSource, transform, gridWriter=None):
     '''
     Run the selected mapper
     '''
     logger.debug(METHOD_ENTER_STR)
     self.dataSource = dataSource
     nx = int(self.xDimTxt.text())
     ny = int(self.yDimTxt.text())
     nz = int(self.zDimTxt.text())
     logger.debug( "nx,ny,nz %d,%d,%d" % (nx, ny, nz))
     outType = self.outputType
     if self.outputFileName == "":
         self.outputFileName = os.path.join(dataSource.projectDir,  \
                                            "%s.vti" %dataSource.projectName)
         self.setFileName[str].emit(self.outputFileName)
     if os.access(os.path.dirname(self.outputFileName), os.W_OK):
         self.mapper = QGridMapper(dataSource, \
                                  self.outputFileName, \
                                  outType, \
                                  nx=nx, ny=ny, nz=nz, \
                                  transform = transform, \
                                  gridWriter = gridWriter,
                                  appConfig=self.appConfig)
         self.mapper.setGridWriter(VTIGridWriter())
         self.mapper.setProgressUpdater(self._updateProgress)
         self.mapper.doMap()
     else:
         self.processError.emit("The specified directory \n" + \
                                str(os.path.dirname(self.outputFileName)) + \
                                "\nis not writable")
     logger.debug(METHOD_EXIT_STR)
Esempio n. 4
0
class AbstractGridOutputForm(AbstractOutputView):
    '''
    A mid level abstract class for output of grid data.  This adds inputs for the output grid
    dimensions.  Subclasses will need to add to _createDataBox in order to provide application 
    level inputs
    '''
    FORM_TITLE = "AbstractOutputForm"

    def __init__(self, **kwargs):
        '''
        Constructor
        '''
        super(AbstractGridOutputForm, self).__init__(**kwargs)
        self.gridWriter = VTIGridWriter()
        self.outputFileName = ""

    def _createDataBox(self):
        '''
        Create Widgets to collect output info.  This class adds parameters for the size of 
        the output grid.  Subclasses will override this method (Super'ing the method) to add 
        application specific inputs.
        '''
        dataBox = super(AbstractGridOutputForm, self)._createDataBox()
        layout = dataBox.layout()

        row = layout.rowCount()
        row += 1
        self._createGridDimensionInput(layout, row)

        return dataBox

    def _createGridDimensionInput(self, layout, row):
        '''
        provide parameters for output grid size
        '''
        label = qtGui.QLabel("Grid Dimensions")
        layout.addWidget(label, row, 0)
        row += 1
        label = qtGui.QLabel(X_STR)
        layout.addWidget(label, row, 0)
        self.xDimTxt = qtGui.QLineEdit()
        self.xDimTxt.setText(str(INITIAL_DIM))
        self.xDimValidator = qtGui.QIntValidator()
        self.xDimTxt.setValidator(self.xDimValidator)
        layout.addWidget(self.xDimTxt, row, 1)

        row += 1
        label = qtGui.QLabel(Y_STR)
        layout.addWidget(label, row, 0)
        self.yDimTxt = qtGui.QLineEdit()
        self.yDimTxt.setText(str(INITIAL_DIM))
        self.yDimValidator = qtGui.QIntValidator()
        self.yDimTxt.setValidator(self.yDimValidator)
        layout.addWidget(self.yDimTxt, row, 1)

        row += 1
        label = qtGui.QLabel(Z_STR)
        layout.addWidget(label, row, 0)
        self.zDimTxt = qtGui.QLineEdit()
        self.zDimTxt.setText(str(INITIAL_DIM))
        self.zDimValidator = qtGui.QIntValidator()
        self.zDimTxt.setValidator(self.zDimValidator)
        layout.addWidget(self.zDimTxt, row, 1)

    def getOutputFileName(self):
        '''
        dummy method to return self.outputFileName.  This is used by runMapper to 
        supply a filename for output files.  If the Writer used needs a fileName, 
        then the the derived class should supply self.outputFileName before running
        the mapper.  Otherwise a filename will be constructed using the project name 
        and gridWriter supplied extension.
        '''
        return self.outputFileName

    def runMapper(self, dataSource, transform):
        '''
        Run the selected mapper.  Writer specific class should be specified
        in the application specific subclass.  A list of forms is provided in 
        dataSource classes.
        '''
        logger.debug ("Entering " + self.FORM_TITLE + " " + \
            str(self.gridWriter))
        self.dataSource = dataSource
        nx = int(self.xDimTxt.text())
        ny = int(self.yDimTxt.text())
        nz = int(self.zDimTxt.text())
        self.outputFileName = self.getOutputFileName()
        if self.outputFileName == "":
            self.outputFileName = os.path.join(dataSource.projectDir,  \
                "%s%s" %(dataSource.projectName,self.gridWriter.FILE_EXTENSION) )
            self.setFileName.emit(self.outputFileName)
        if os.access(os.path.dirname(self.outputFileName), os.W_OK):
            self.mapper = QGridMapper(dataSource, \
                                     self.outputFileName, \
                                     self.outputType,\
                                     nx=nx, ny=ny, nz=nz,
                                     transform = transform,
                                     gridWriter = self.gridWriter,
                                     appConfig=self.appConfig)
            self.mapper.setProgressUpdater(self._updateProgress)
            self.mapper.doMap()
        else:
            self.processError.emit("The specified directory \n" + \
                                   str(os.path.dirname(self.outputFileName)) + \
                                   "\nis not writable")
        logger.debug("Exit " + self.FORM_TITLE)

    def stopMapper(self):
        '''
        Halt the mapping _process
        '''
        self.mapper.stopMap()
Esempio n. 5
0
    ds.setRangeBounds(ds.getOverallRanges())
    imageToBeUsed = ds.getImageToBeUsed()
    
    print("imageToBeUsed %s" % imageToBeUsed)
#    wavelen = ENERGY_WAVELENGTH_CONVERT_FACTOR/ds.getIncidentEnergy()[scans[0]]
    imageSize = np.prod(ds.getDetectorDimensions())
    print scanRange[0]
    for imageInScan in range(1, len(imageToBeUsed[scanRange[0]])+1):
        print ("Scan Line %d" % imageInScan)
        
        outputFileName = os.path.join(projectDir, specName + \
                                      ('_N%d.vti' % imageInScan))
        gridWriter = VTIGridWriter()

        tmpImageUsed = imageToBeUsed[scanRange[0]]
        savImageUsed = imageToBeUsed[scanRange[0]]
        tmpImageUsed[imageInScan] = False
        ds.imageToBeUsed[scanRange[0]]  = [not i for i in tmpImageUsed]
        gridMapper = QGridMapper(ds,
                                 outputFileName, 
                                 outputType=BINARY_OUTPUT,
                                 transform=UnityTransform3D(),
                                 gridWriter=gridWriter,
                                 appConfig=appConfig,
                                 nx=nx, ny=ny, nz=nz)
    
        gridMapper.setProgressUpdater(updateMapperProgress)
        gridMapper.doMap()
        ds.imageToBeUsed[scanRange[0]]  = savImageUsed
    
        
Esempio n. 6
0
class ProcessScans(qtGui.QDialog):
    '''
    This class presents a form to select to start analysis.  This display
    allows switching between Grid map and pole figure.
    '''
    def __init__(self, parent=None):
        '''
        Constructor - Layout widgets on the page & link up actions.
        '''
        super(ProcessScans, self).__init__(parent)
        self.Mapper = None
        layout = qtGui.QVBoxLayout()

        self.dataBox = self._createDataBox()
        controlBox = self._createControlBox()

        layout.addWidget(self.dataBox)
        layout.addWidget(controlBox)
        self.setLayout(layout)

    def _browseForOutputFile(self):
        '''
        Launch file browser to select the output file.  Checks are done to make
        sure the selected directory exists and that the selected file is 
        writable
        '''
        if self.outFileTxt.text() == "":
            fileName = str(qtGui.QFileDialog.getSaveFileName(None, \
                                               SAVE_FILE_STR, \
                                               filter=VTI_FILTER_STR))
        else:
            inFileName = str(self.outFileTxt.text())
            fileName = str(qtGui.QFileDialog.getSaveFileName(None,
                                               SAVE_FILE_STR,
                                               filter=VTI_FILTER_STR, \
                                               directory = inFileName))
        if fileName != "":
            if os.path.exists(os.path.dirname(str(fileName))):
                self.outFileTxt.setText(fileName)
                self.outputFileName = fileName
                self.outFileTxt.emit(qtCore.SIGNAL(EDIT_FINISHED_SIGNAL))
            else:
                message = qtGui.QMessageBox()
                message.warning(self, \
                             WARNING_STR, \
                             "The specified directory does not exist")
                self.outFileTxt.setText(fileName)
                self.outputFileName = fileName
                self.outFileTxt.emit(qtCore.SIGNAL(EDIT_FINISHED_SIGNAL))
            if not os.access(os.path.dirname(fileName), os.W_OK):
                message = qtGui.QMessageBox()
                message.warning(self, \
                             WARNING_STR, \
                             "The specified file is not writable")

    def _cancelProcess(self):
        '''
        Emit a signal to trigger the cancellation of processing.
        '''
        self.emit(qtCore.SIGNAL(CANCEL_PROCESS_SIGNAL))

    def _createControlBox(self):
        '''
        Create box wih the GUI controls Run & Cancel
        '''
        controlBox = qtGui.QGroupBox()
        controlLayout = qtGui.QGridLayout()
        row = 0
        self.progressBar = qtGui.QProgressBar()
        controlLayout.addWidget(self.progressBar, row, 1)

        self.runButton = qtGui.QPushButton(RUN_STR)
        controlLayout.addWidget(self.runButton, row, 3)

        self.cancelButton = qtGui.QPushButton(CANCEL_STR)
        self.cancelButton.setDisabled(True)

        controlLayout.addWidget(self.cancelButton, row, 4)

        self.connect(self.runButton, \
                     qtCore.SIGNAL(CLICKED_SIGNAL), \
                     self._process)
        self.connect(self.cancelButton, \
                     qtCore.SIGNAL(CLICKED_SIGNAL), \
                     self._cancelProcess)
        self.connect(self, \
                     qtCore.SIGNAL(UPDATE_PROGRESS_SIGNAL), \
                     self.setProgress)
        controlBox.setLayout(controlLayout)
        return controlBox

    def _createDataBox(self):
        '''
        Create Sub Layout for data gathering widgets
        '''
        dataBox = qtGui.QGroupBox()
        dataLayout = qtGui.QGridLayout()
        row = 0

        label = qtGui.QLabel("Grid Dimensions")
        dataLayout.addWidget(label, row, 0)
        row += 1
        label = qtGui.QLabel(X_STR)
        dataLayout.addWidget(label, row, 0)
        self.xDimTxt = qtGui.QLineEdit()
        self.xDimTxt.setText("200")
        self.xDimValidator = qtGui.QIntValidator()
        self.xDimTxt.setValidator(self.xDimValidator)
        dataLayout.addWidget(self.xDimTxt, row, 1)

        row += 1
        label = qtGui.QLabel(Y_STR)
        dataLayout.addWidget(label, row, 0)
        self.yDimTxt = qtGui.QLineEdit()
        self.yDimTxt.setText("200")
        self.yDimValidator = qtGui.QIntValidator()
        self.yDimTxt.setValidator(self.yDimValidator)
        dataLayout.addWidget(self.yDimTxt, row, 1)

        row += 1
        label = qtGui.QLabel(Z_STR)
        dataLayout.addWidget(label, row, 0)
        self.zDimTxt = qtGui.QLineEdit()
        self.zDimTxt.setText("200")
        self.zDimValidator = qtGui.QIntValidator()
        self.zDimTxt.setValidator(self.zDimValidator)
        dataLayout.addWidget(self.zDimTxt, row, 1)

        row += 1
        label = qtGui.QLabel("Output File")
        dataLayout.addWidget(label, row, 0)
        self.outputFileName = ""
        self.outFileTxt = qtGui.QLineEdit()
        self.outFileTxt.setText(self.outputFileName)
        dataLayout.addWidget(self.outFileTxt, row, 1)
        self.outputFileButton = qtGui.QPushButton(BROWSE_STR)
        dataLayout.addWidget(self.outputFileButton, row, 2)

        self.connect(self.outputFileButton, \
                     qtCore.SIGNAL(CLICKED_SIGNAL),
                     self._browseForOutputFile)
        self.connect(self.outputFileButton, \
                     qtCore.SIGNAL(EDIT_FINISHED_SIGNAL),
                     self._editFinishedOutputFile)
        self.connect(self.outFileTxt, \
                     qtCore.SIGNAL(EDIT_FINISHED_SIGNAL), \
                     self._editFinishedOutputFile)
        self.connect(self, qtCore.SIGNAL(SET_FILE_NAME_SIGNAL),
                     self.outFileTxt.setText)

        dataBox.setLayout(dataLayout)
        return dataBox

    def _editFinishedOutputFile(self):
        '''
        When editing is finished the a check is done to make sure that the 
        directory exists and the file is writable
        '''
        fileName = str(self.outFileTxt.text())
        if fileName != "":
            if os.path.exists(os.path.dirname(fileName)):
                self.outputFileName = fileName
            else:
                if os.path.dirname(fileName) == "":
                    curDir = os.path.realpath(os.path.curdir)
                    fileName = str(os.path.join(curDir, fileName))
                else:
                    message = qtGui.QMessageBox()
                    message.warning(self, \
                                 WARNING_STR, \
                                 "The specified directory \n" + \
                                 str(os.path.dirname(fileName)) + \
                                 "\ndoes not exist")

#               self.outputFileName = fileName
                self.emit(qtCore.SIGNAL(SET_FILE_NAME_SIGNAL), fileName)

            if not os.access(os.path.dirname(fileName), os.W_OK):
                message = qtGui.QMessageBox()
                message.warning(self, \
                             WARNING_STR, \
                             "The specified file is not writable")

    def _process(self):
        '''
        Emit a signal to trigger the start of processing.
        '''
        self.emit(qtCore.SIGNAL(PROCESS_SIGNAL))

    def runMapper(self, dataSource, transform):
        '''
        Run the selected mapper
        '''
        self.dataSource = dataSource
        nx = int(self.xDimTxt.text())
        ny = int(self.yDimTxt.text())
        nz = int(self.zDimTxt.text())
        if self.outputFileName == "":
            self.outputFileName = os.path.join(dataSource.projectDir,  \
                                               "%s.vti" %dataSource.projectName)
            self.emit(qtCore.SIGNAL(SET_FILE_NAME_SIGNAL), self.outputFileName)
        if os.access(os.path.dirname(self.outputFileName), os.W_OK):
            self.mapper = QGridMapper(dataSource, \
                                     self.outputFileName, \
                                     nx=nx, ny=ny, nz=nz,
                                     transform = transform,
                                     appConfig=self.appConfig)
            self.mapper.setProgressUpdater(self.updateProgress)
            self.mapper.doMap()
        else:
            self.emit(qtCore.SIGNAL(PROCESS_ERROR_SIGNAL), \
                         "The specified directory \n" + \
                         str(os.path.dirname(self.outputFileName)) + \
                         "\nis not writable")

    def setCancelOK(self):
        '''
        If Cancel is OK the run button is disabled and the cancel button is 
        enabled
        '''
        self.runButton.setDisabled(True)
        self.cancelButton.setDisabled(False)
        self.dataBox.setDisabled(True)

    def setOutFileName(self, name):
        '''
        Write a filename to the text widget and to the stored output file name
        :param name: Name of output file
        '''
        self.outFileTxt.setText(name)
        self.outputFileName = name

    def setProgress(self, value):
        '''
        Set the value in the progress bar
        :param value: value to write to the progress bar
        '''
        self.progressBar.setValue(value)

    def setProgressLimits(self, progressMin, progressMax):
        '''
        Set the limits on the progress bar.
        :param progressMin: Minimum value to store in the progress bar
        :param progressMax: Maximum value to store in the progress bar
        '''
        self.progressBar.setMinimum(progressMin)
        self.progressBar.setMaximum(progressMax)

    def setRunOK(self):
        '''
        If Run is OK the load button is enabled and the cancel button is 
        disabled
        '''
        self.runButton.setDisabled(False)
        self.cancelButton.setDisabled(True)
        self.dataBox.setDisabled(False)

    def _stopMapper(self):
        '''
        Halt the mapping _process
        '''
        self.mapper.stopMap()

    def updateProgress(self, value):
        '''
        Send signal to update the progress bar.
        :param value: value to be put on the progress bar.
        '''
        self.emit(qtCore.SIGNAL(UPDATE_PROGRESS_SIGNAL), value)
Esempio n. 7
0
class ProcessVTIOutputForm(AbstractOutputView):
    FORM_TITLE = "VTI Grid Output"
    
    @staticmethod
    def createInstance(parent=None, appConfig=None):
        return ProcessVTIOutputForm(parent=parent, appConfig=appConfig)
    
    def __init__(self, **kwargs):
        super(ProcessVTIOutputForm, self).__init__(**kwargs)
        logger.debug(METHOD_ENTER_STR)
        self.mapper = None
        layout = qtGui.QVBoxLayout()
        self.dataBox = self._createDataBox()
        controlBox = self._createControlBox()
        
        layout.addWidget(self.dataBox)
        layout.addWidget(controlBox)
        self.setLayout(layout)
        self.outputType = BINARY_OUTPUT
        logger.debug(METHOD_EXIT_STR)
        
    @Slot()
    def _browseForOutputFile(self):
        '''
        Launch file browser to select the output file.  Checks are done to make
        sure the selected directory exists and that the selected file is 
        writable
        '''
        logger.debug(METHOD_ENTER_STR)
        if self.outFileTxt.text() == EMPTY_STR:
            fileName = str(qtGui.QFileDialog.getSaveFileName(None, \
                                               SAVE_FILE_STR, \
                                               filter=VTI_FILTER_STR))
        else:
            inFileName = str(self.outFileTxt.text())
            fileName = str(qtGui.QFileDialog.getSaveFileName(None, 
                                               SAVE_FILE_STR, 
                                               filter=VTI_FILTER_STR, \
                                               directory = inFileName))
        if fileName != EMPTY_STR:
            if os.path.exists(os.path.dirname(str(fileName))):
                self.outFileTxt.setText(fileName)
                self.outputFileName = fileName
                self.outFileTxt.editingFinished.emit()
            else:
                message = qtGui.QMessageBox()
                message.warning(self, \
                             WARNING_STR, \
                             "The specified directory does not exist")
                self.outFileTxt.setText(fileName)
                self.outputFileName = fileName
                self.outFileTxt.editingFinished.emit()
            if not os.access(os.path.dirname(fileName), os.W_OK):
                message = qtGui.QMessageBox()
                message.warning(self, \
                             WARNING_STR, \
                             "The specified file is not writable")
        else:
            self.outputFileName = EMPTY_STR
            self.setOutFileText.emit(EMPTY_STR)
        logger.debug(METHOD_EXIT_STR)
#     @Slot()
#     def _cancelProcess(self):
#         '''
#         Emit a signal to trigger the cancellation of processing.
#         '''
#         self.cancel.emit(qtCore.SIGNAL(CANCEL_PROCESS_SIGNAL))
        

    def _createDataBox(self):
        '''
        Create Widgets to collect output info
        '''
        logger.debug(METHOD_ENTER_STR)
        dataBox = super(ProcessVTIOutputForm, self)._createDataBox()
        dataLayout = dataBox.layout()
        row = dataLayout.rowCount()
        
        label = qtGui.QLabel("Grid Dimensions")
        dataLayout.addWidget(label, row,0)
        row += 1
        label = qtGui.QLabel(X_STR)
        dataLayout.addWidget(label, row,0)
        self.xDimTxt = qtGui.QLineEdit()
        self.xDimTxt.setText("200")
        self.xDimValidator = qtGui.QIntValidator()
        self.xDimTxt.setValidator(self.xDimValidator)
        dataLayout.addWidget(self.xDimTxt, row,1)
        
        row += 1
        label = qtGui.QLabel(Y_STR)
        dataLayout.addWidget(label, row,0)
        self.yDimTxt = qtGui.QLineEdit()
        self.yDimTxt.setText("200")
        self.yDimValidator = qtGui.QIntValidator()
        self.yDimTxt.setValidator(self.yDimValidator)
        dataLayout.addWidget(self.yDimTxt, row,1)
        
        row += 1
        label = qtGui.QLabel(Z_STR)
        dataLayout.addWidget(label, row,0)
        self.zDimTxt = qtGui.QLineEdit()
        self.zDimTxt.setText("200")
        self.zDimValidator = qtGui.QIntValidator()
        self.zDimTxt.setValidator(self.zDimValidator)
        dataLayout.addWidget(self.zDimTxt, row,1)
        
        row += 1
        label = qtGui.QLabel("Output File")
        dataLayout.addWidget(label, row,0)
        self.outputFileName = ""
        self.outFileTxt = qtGui.QLineEdit()
        self.outFileTxt.setText(self.outputFileName)
        dataLayout.addWidget(self.outFileTxt, row,1)
        self.outputFileButton = qtGui.QPushButton(BROWSE_STR)
        dataLayout.addWidget(self.outputFileButton, row, 2)

        row += 1
        label = qtGui.QLabel("Output Type")
        dataLayout.addWidget(label, row, 0)
        self.outputTypeSelect = QComboBox()
        self.outputTypeSelect.addItem(BINARY_OUTPUT)
        self.outputTypeSelect.addItem(ASCII_OUTPUT)
        dataLayout.addWidget(self.outputTypeSelect, row, 2)
        
        self.outputFileButton.clicked.connect(self._browseForOutputFile)
#         self.connect(self.outputFileButton, \
#                      qtCore.SIGNAL(EDIT_FINISHED_SIGNAL), 
#                      self._editFinishedOutputFile)
        self.outFileTxt.editingFinished.connect(self._editFinishedOutputFile)
        self.setFileName[str].connect( self.setOutFileText)
        self.outputTypeSelect.currentIndexChanged[str]. \
            connect(self._selectedTypeChanged)
        logger.debug(METHOD_EXIT_STR)
        return dataBox
        
    @Slot()
    def _editFinishedOutputFile(self):
        '''
        When editing is finished the a check is done to make sure that the 
        directory exists and the file is writable
        '''
        logger.debug(METHOD_ENTER_STR)
        fileName = str(self.outFileTxt.text())
        if fileName != EMPTY_STR:
            if os.path.exists(os.path.dirname(fileName)):
                self.outputFileName = fileName
            else:
                if os.path.dirname(fileName) == EMPTY_STR:
                    curDir = os.path.realpath(os.path.curdir)
                    fileName = str(os.path.join(curDir, fileName))
                else:
                    message = qtGui.QMessageBox()
                    message.warning(self, \
                                 WARNING_STR, \
                                 "The specified directory \n" + \
                                 str(os.path.dirname(fileName)) + \
                                 "\ndoes not exist")
                
#               self.outputFileName = fileName
                self.setFileName.emit(fileName)
                
            if not os.access(os.path.dirname(fileName), os.W_OK):
                message = qtGui.QMessageBox()
                message.warning(self, \
                             WARNING_STR, \
                             "The specified file is not writable")
        else:
            self.outputFileName = EMPTY_STR
            self.setOutFileText.emit(EMPTY_STR)
        logger.debug(METHOD_EXIT_STR)
#     @Slot()
#     def _process(self):
#         '''
#         Emit a signal to trigger the start of processing.
#         '''
#         self.emit(qtCore.SIGNAL(PROCESS_SIGNAL))
        
    def runMapper(self, dataSource, transform, gridWriter=None):
        '''
        Run the selected mapper
        '''
        logger.debug(METHOD_ENTER_STR)
        self.dataSource = dataSource
        nx = int(self.xDimTxt.text())
        ny = int(self.yDimTxt.text())
        nz = int(self.zDimTxt.text())
        logger.debug( "nx,ny,nz %d,%d,%d" % (nx, ny, nz))
        outType = self.outputType
        if self.outputFileName == "":
            self.outputFileName = os.path.join(dataSource.projectDir,  \
                                               "%s.vti" %dataSource.projectName)
            self.setFileName[str].emit(self.outputFileName)
        if os.access(os.path.dirname(self.outputFileName), os.W_OK):
            self.mapper = QGridMapper(dataSource, \
                                     self.outputFileName, \
                                     outType, \
                                     nx=nx, ny=ny, nz=nz, \
                                     transform = transform, \
                                     gridWriter = gridWriter,
                                     appConfig=self.appConfig)
            self.mapper.setGridWriter(VTIGridWriter())
            self.mapper.setProgressUpdater(self._updateProgress)
            self.mapper.doMap()
        else:
            self.processError.emit("The specified directory \n" + \
                                   str(os.path.dirname(self.outputFileName)) + \
                                   "\nis not writable")
        logger.debug(METHOD_EXIT_STR)
        
    @Slot(str)
    def _selectedTypeChanged(self, typeStr):
        logger.debug(METHOD_ENTER_STR)
        self.outputType = str(typeStr)
        logger.debug(METHOD_EXIT_STR)
        
    @Slot(str)
    def setOutFileText(self, outFile):
        logger.debug(METHOD_ENTER_STR)
        self.outFileTxt.setText(outFile)
        self.outFileTxt.editingFinished.emit()
        
    def _stopMapper(self):
        '''
        Halt the mapping _process
        '''
        logger.debug(METHOD_ENTER_STR)
        self.mapper.stopMap()
        logger.debug(METHOD_EXIT_STR)
        
        
Esempio n. 8
0
maxImageMemory = appConfig.getMaxImageMemory()

for scans in scanList1:
    scanRange = srange(scans).list()
    print ("scanRange %s" % scanRange)
    print("specName, specExt: %s, %s" % (specName, specExt))
    ds = Sector33SpecDataSource(projectDir, specName, specExt,
                                instConfigName, detectorConfigName, roi=roi, 
                                pixelsToAverage=bin, scanList= scanRange, 
                                badPixelFile=badPixelFile, 
                                appConfig=appConfig)
    ds.setCurrentDetector(detectorName)
    ds.setProgressUpdater(updateDataSourceProgress)
    ds.loadSource(mapHKL=mapHKL)
    ds.setRangeBounds(ds.getOverallRanges())
    imageToBeUsed = ds.getImageToBeUsed()
    print("imageToBeUsed %s" % imageToBeUsed)
#    wavelen = ENERGY_WAVELENGTH_CONVERT_FACTOR/ds.getIncidentEnergy()[scans[0]]
    imageSize = np.prod(ds.getDetectorDimensions())
    
    gridMapper = QGridMapper(ds,
                             outputFileName, 
                             outputType=BINARY_OUTPUT,
                             transform=UnityTransform3D(),
                             gridWriter=VTIGridWriter(),
                             appConfig=appConfig)

    gridMapper.setProgressUpdater(updateMapperProgress)
    gridMapper.doMap()
    
        
Esempio n. 9
0
            print '  Q  : ', coords_x_min, ' .... ', coords_x_max
        print 'Resulting data set:'
        print '  x_min:  ', x_min
        print '  x_max:  ', x_max
        print '  x_step: ', x_step
        print '  nbins:  ', nbins
        print ''
        print 'Starting the gridding process'

    # Create a gridder object and initialize
    gridder = xu.Gridder1D(int(nbins))
    gridder.KeepData(True)
    gridder.dataRange(x_min, x_max)

    # Create the QGridMapper object
    gm = QGridMapper(ds, 'dummy.vti')

    # Run through all scans of the scan-set to perform the gridding
    for scan in ds.getAvailableScans():
        if verbose:
            print "Scan number: ", scan
        numImages = len(imageToBeUsed[scan])
        if imageSize * 4 * numImages <= max_image_memory:
            if verbose:
                print "  Only 1 pass required."
            qx, qy, qz, intensity = ds.rawmap((scan, ),
                                              mask=imageToBeUsed[scan])
            Q = np.sqrt(qx**2 + qy**2 + qz**2)
            if data_coordinate is 'tth':
                coords_x = np.rad2deg(
                    np.arcsin((Q * wavelen) / (4.0 * np.pi)) * 2.0)