#/*##########################################################################
#
# The PyMca X-Ray Fluorescence Toolkit
#
# Copyright (c) 2004-2014 European Synchrotron Radiation Facility
#
# This file is part of the PyMca X-ray Fluorescence Toolkit developed at
# the ESRF by the Software group.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#
#############################################################################*/
__author__ = "V. Armando Sole - ESRF Data Analysis"
__contact__ = "sole@esrf.fr"
__license__ = "MIT"
__copyright__ = "European Synchrotron Radiation Facility, Grenoble, France"
import sys
import numpy
from PyMca5.PyMcaGui import PyMcaQt as qt
from PyMca5.PyMcaGui import PyMca_Icons
IconDict = PyMca_Icons.IconDict
from PyMca5.PyMcaGui.pymca import PyMcaFileDialogs
DEBUG = 0
[docs]class FastXRFLinearFitWindow(qt.QWidget):
def __init__(self, parent=None):
qt.QWidget.__init__(self, parent)
self.setWindowTitle("FastXRFLinearFitWindow")
self.setWindowIcon(qt.QIcon(qt.QPixmap(IconDict['gioconda16'])))
self.mainLayout = qt.QGridLayout(self)
self.mainLayout.setContentsMargins(0, 0, 0, 0)
# configuration file
configLabel = qt.QLabel(self)
configLabel.setText("Fit Configuration File:")
self._configLine = qt.QLineEdit(self)
self._configLine.setReadOnly(True)
self._configButton = qt.QPushButton(self)
self._configButton.setText("Browse")
self._configButton.setAutoDefault(False)
self._configButton.clicked.connect(self.browseConfigurationFile)
# output directory
outLabel = qt.QLabel(self)
outLabel.setText("Output dir:")
self._outLine = qt.QLineEdit(self)
self._outLine.setReadOnly(True)
self._outButton = qt.QPushButton(self)
self._outButton.setText('Browse')
self._outButton.setAutoDefault(False)
self._outButton.clicked.connect(self.browseOutputDir)
# output file name
fileLabel = qt.QLabel(self)
fileLabel.setText("Output file root:")
self._fileLine = qt.QLineEdit(self)
self._fileLine.setReadOnly(False)
self._fileLine.setText("images")
# concentrations
self._concentrationsBox = qt.QCheckBox(self)
self._concentrationsBox.setText("calculate concentrations")
self._concentrationsBox.setChecked(False)
self._concentrationsBox.setEnabled(True)
# weight method
self._weightWidget = qt.QWidget(self)
self._weightWidget.mainLayout = qt.QHBoxLayout(self._weightWidget)
self._weightWidget.mainLayout.setContentsMargins(0, 0, 0, 0)
self._weightWidget.mainLayout.setSpacing(0)
self._weightButtonGroup = qt.QButtonGroup(self._weightWidget)
i = 0
weightLabel = qt.QLabel(self)
weightLabel.setText("Weight policy: ")
for txt in ["No Weight (Fastest)", "Average Weight (Fast)", "Individual Weights (slow)"]:
button = qt.QRadioButton(self._weightWidget)
button.setText(txt)
self._weightButtonGroup.addButton(button)
self._weightButtonGroup.setId(button, i)
self._weightWidget.mainLayout.addWidget(button)
i += 1
self._weightButtonGroup.buttons()[0].setChecked(True)
#self._weightWidget.mainLayout.addWidget(qt.HorizontalSpacer(self._weightWidget))
self.mainLayout.addWidget(configLabel, 0, 0)
self.mainLayout.addWidget(self._configLine, 0, 1)
self.mainLayout.addWidget(self._configButton, 0, 2)
self.mainLayout.addWidget(outLabel, 1, 0)
self.mainLayout.addWidget(self._outLine, 1, 1)
self.mainLayout.addWidget(self._outButton, 1, 2)
self.mainLayout.addWidget(fileLabel, 2, 0)
self.mainLayout.addWidget(self._fileLine, 2, 1)
self.mainLayout.addWidget(self._concentrationsBox, 4, 0, 1, 2)
self.mainLayout.addWidget(weightLabel, 3, 0)
self.mainLayout.addWidget(self._weightWidget, 3, 1, 1, 1)
[docs] def sizeHint(self):
return qt.QSize(int(1.8 * qt.QWidget.sizeHint(self).width()),
qt.QWidget.sizeHint(self).height())
[docs] def browseConfigurationFile(self):
f = PyMcaFileDialogs.getFileList(parent=self,
filetypelist=["Configuration files (*.cfg)"],
message="Open a new fit config file",
mode="OPEN",
single=True)
if len(f):
self._configLine.setText(f[0])
[docs] def browseOutputDir(self):
f = PyMcaFileDialogs.getExistingDirectory(parent=self,
message="Please select output directory",
mode="OPEN")
if len(f):
self._outLine.setText(f)
[docs] def getParameters(self):
ddict = {}
ddict['configuration'] = qt.safe_str(self._configLine.text())
ddict['output_dir'] = qt.safe_str(self._outLine.text())
ddict['file_root'] = qt.safe_str(self._fileLine.text())
if self._concentrationsBox.isChecked():
ddict['concentrations'] = 1
else:
ddict['concentrations'] = 0
ddict['weight_policy'] = self._weightButtonGroup.checkedId()
return ddict
[docs]class FastXRFLinearFitDialog(qt.QDialog):
def __init__(self, parent=None):
qt.QDialog.__init__(self, parent)
self.setWindowTitle("Fast XRF Linear Fit Dialog")
self.setWindowIcon(qt.QIcon(qt.QPixmap(IconDict['gioconda16'])))
self.mainLayout = qt.QGridLayout(self)
self.mainLayout.setContentsMargins(10, 10, 10, 10)
self.parametersWidget = FastXRFLinearFitWindow(self)
self.rejectButton= qt.QPushButton(self)
self.rejectButton.setAutoDefault(False)
self.rejectButton.setText("Cancel")
self.acceptButton= qt.QPushButton(self)
self.acceptButton.setAutoDefault(False)
self.acceptButton.setText("OK")
self.rejectButton.clicked.connect(self.reject)
self.acceptButton.clicked.connect(self.accept)
self.mainLayout.addWidget(self.parametersWidget, 0, 0, 5, 4)
self.mainLayout.addWidget(self.rejectButton, 6, 1)
self.mainLayout.addWidget(self.acceptButton, 6, 2)
[docs] def getParameters(self):
return self.parametersWidget.getParameters()
if __name__ == "__main__":
app = qt.QApplication([])
w = FastXRFLinearFitDialog()
ret = w.exec_()
if ret:
print(w.getParameters())
#app.exec_()