def test(a):
from PyMca5.PyMcaGui import PyMcaQt as qt
from PyMca5.PyMcaMath.fitting import Specfit
from PyMca5.PyMcaGui.pymca import ScanWindow
#print dir(a)
x = numpy.arange(1000).astype(numpy.float)
p1 = numpy.array([1500,100.,50.0])
p2 = numpy.array([1500,700.,50.0])
y = a.gauss(p1,x)+1
y = y + a.gauss(p2,x)
app=qt.QApplication([])
fit=Specfit.Specfit(x,y)
fit.addtheory('Gaussians',a.gauss,['Height','Position','FWHM'],
a.estimate_gauss)
fit.settheory('Gaussians')
fit.setbackground('Linear')
fit.estimate()
fit.startfit()
yfit=fit.gendata(x=x,parameters=fit.paramlist)
print("I set an offset of 1 to see the difference in log scale :-)")
w = ScanWindow.ScanWindow()
w.addCurve(x, y + 1, "Data + 1")
w.addCurve(x, yfit, "Fit")
w.show()
app.exec_()
def __init__(self, parent=None, name="McaSimpleFit", specfit=None, fl=0):
qt.QWidget.__init__(self, parent)
self.setWindowTitle(name)
if specfit is None:
self.specfit = Specfit.Specfit()
else:
self.specfit = specfit
layout = qt.QVBoxLayout(self)
##############
self.headerlabel = qt.QLabel(self)
self.headerlabel.setAlignment(qt.Qt.AlignHCenter)
self.setheader('<b>Fit of XXXXXXXXXX from Channel XXXXX to XXXX<\b>')
##############
defaultFunctions = "SpecfitFunctions.py"
if not os.path.exists(defaultFunctions):
defaultFunctions = os.path.join(os.path.dirname(__file__),
defaultFunctions)
self.specfit.importfun(defaultFunctions)
self.specfit.settheory('Area Gaussians')
self.specfit.setbackground('Linear')
fitconfig = {}
fitconfig.update(self.specfit.fitconfig)
fitconfig['WeightFlag'] = 1
fitconfig['McaMode'] = 1
self.specfit.configure(**fitconfig)
self.specfitGui = SpecfitGui.SpecfitGui(self,
config=1,
status=1,
buttons=0,
specfit=self.specfit,
eh=self.specfit.eh)
layout.addWidget(self.headerlabel)
layout.addWidget(self.specfitGui)
hbox = qt.QWidget(self)
hboxLayout = qt.QHBoxLayout(hbox)
self.estimatebutton = qt.QPushButton(hbox)
self.estimatebutton.setText("Estimate")
hs1 = qt.HorizontalSpacer(hbox)
self.fitbutton = qt.QPushButton(hbox)
self.fitbutton.setText("Fit Again!")
self.dismissbutton = qt.QPushButton(hbox)
self.dismissbutton.setText("Dismiss")
self.estimatebutton.clicked.connect(self.estimate)
self.fitbutton.clicked.connect(self.fit)
self.dismissbutton.clicked.connect(self.dismiss)
self.specfitGui.sigSpecfitGuiSignal.connect(self.__anasignal)
hs2 = qt.HorizontalSpacer(hbox)
hboxLayout.addWidget(hs1)
hboxLayout.addWidget(self.estimatebutton)
hboxLayout.addWidget(self.fitbutton)
hboxLayout.addWidget(self.dismissbutton)
hboxLayout.addWidget(hs2)
layout.addWidget(hbox)
self.estimatebutton.hide()
def __init__(self,parent = None,name = None,fl = 0, specfit = None,
config = 0, status = 0, buttons = 0, eh = None):
if name == None:
name = "SpecfitGui"
qt.QWidget.__init__(self, parent)
self.setWindowTitle(name)
layout= qt.QVBoxLayout(self)
#layout.setAutoAdd(1)
if eh == None:
self.eh = EventHandler.EventHandler()
else:
self.eh = eh
if specfit is None:
self.specfit = Specfit.Specfit(eh=self.eh)
else:
self.specfit = specfit
#initialize the default fitting functions in case
#none is present
if not len(self.specfit.theorylist):
funsFile = "SpecfitFunctions.py"
if not os.path.exists(funsFile):
funsFile = os.path.join(os.path.dirname(Specfit.__file__),\
funsFile)
self.specfit.importfun(funsFile)
#copy specfit configure method for direct access
self.configure=self.specfit.configure
self.fitconfig=self.specfit.fitconfig
self.setdata=self.specfit.setdata
self.guiconfig=None
if config:
self.guiconfig = FitConfigGui.FitConfigGui(self)
self.guiconfig.MCACheckBox.stateChanged[int].connect(self.mcaevent)
self.guiconfig.WeightCheckBox.stateChanged[int].connect(self.weightevent)
self.guiconfig.AutoFWHMCheckBox.stateChanged[int].connect(self.autofwhmevent)
self.guiconfig.AutoScalingCheckBox.stateChanged[int].connect(self.autoscaleevent)
self.guiconfig.ConfigureButton.clicked.connect(self.__configureGuiSlot)
self.guiconfig.PrintPushButton.clicked.connect(self.printps)
self.guiconfig.BkgComBox.activated[str].connect(self.bkgevent)
self.guiconfig.FunComBox.activated[str].connect(self.funevent)
layout.addWidget(self.guiconfig)
self.guiparameters = MultiParameters.ParametersTab(self)
layout.addWidget(self.guiparameters)
self.guiparameters.sigMultiParametersSignal.connect(self.__forward)
if config:
for key in self.specfit.bkgdict.keys():
self.guiconfig.BkgComBox.addItem(str(key))
for key in self.specfit.theorylist:
self.guiconfig.FunComBox.addItem(str(key))
configuration={}
if specfit is not None:
configuration = specfit.configure()
if configuration['fittheory'] is None:
self.guiconfig.FunComBox.setCurrentIndex(1)
self.funevent(self.specfit.theorylist[0])
else:
self.funevent(configuration['fittheory'])
if configuration['fitbkg'] is None:
self.guiconfig.BkgComBox.setCurrentIndex(1)
self.bkgevent(list(self.specfit.bkgdict.keys())[0])
else:
self.bkgevent(configuration['fitbkg'])
else:
self.guiconfig.BkgComBox.setCurrentIndex(1)
self.guiconfig.FunComBox.setCurrentIndex(1)
self.funevent(self.specfit.theorylist[0])
self.bkgevent(list(self.specfit.bkgdict.keys())[0])
configuration.update(self.configure())
if configuration['McaMode']:
self.guiconfig.MCACheckBox.setChecked(1)
else:
self.guiconfig.MCACheckBox.setChecked(0)
if configuration['WeightFlag']:
self.guiconfig.WeightCheckBox.setChecked(1)
else:
self.guiconfig.WeightCheckBox.setChecked(0)
if configuration['AutoFwhm']:
self.guiconfig.AutoFWHMCheckBox.setChecked(1)
else:
self.guiconfig.AutoFWHMCheckBox.setChecked(0)
if configuration['AutoScaling']:
self.guiconfig.AutoScalingCheckBox.setChecked(1)
else:
self.guiconfig.AutoScalingCheckBox.setChecked(0)
if status:
self.guistatus = FitStatusGui.FitStatusGui(self)
self.eh.register('FitStatusChanged',self.fitstatus)
layout.addWidget(self.guistatus)
if buttons:
self.guibuttons = FitActionsGui.FitActionsGui(self)
self.guibuttons.EstimateButton.clicked.connect(self.estimate)
self.guibuttons.StartfitButton.clicked.connect(self.startfit)
self.guibuttons.DismissButton.clicked.connect(self.dismiss)
layout.addWidget(self.guibuttons)
def __init__(self, parent=None, name="ScanFit", specfit=None, fl=0):
#fl=qt.Qt.WDestructiveClose):
qt.QWidget.__init__(self, parent)
self.setWindowTitle(name)
if specfit is None:
self.specfit = Specfit.Specfit()
else:
self.specfit = specfit
self.info = None
layout = qt.QVBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(0)
##############
self.headerlabel = qt.QLabel(self)
self.headerlabel.setAlignment(qt.Qt.AlignHCenter)
self.setHeader('<b>Fit of XXXXXXXXXX from X XXXXX to XXXX<\b>')
##############
if not len(self.specfit.theorylist):
funsFile = "SpecfitFunctions.py"
if not os.path.exists(funsFile):
funsFile = os.path.join(os.path.dirname(Specfit.__file__),
funsFile)
self.specfit.importfun(funsFile)
if 'Area Gaussians' not in self.specfit.theorylist:
funsFile = "SpecfitFunctions.py"
if not os.path.exists(funsFile):
funsFile = os.path.join(os.path.dirname(Specfit.__file__),
funsFile)
self.specfit.importfun(funsFile)
self.specfit.settheory('Area Gaussians')
self.specfit.setbackground('Linear')
fitconfig = {}
fitconfig.update(self.specfit.fitconfig)
fitconfig['WeightFlag'] = 0
fitconfig['ForcePeakPresence'] = 1
fitconfig['McaMode'] = 0
self.specfit.configure(**fitconfig)
self.specfitGui = SpecfitGui.SpecfitGui(self,
config=1,
status=1,
buttons=0,
specfit=self.specfit,
eh=self.specfit.eh)
#self.specfitGui.updateGui(configuration=fitconfig)
#self.setdata = self.specfit.setdata
self.specfitGui.guiconfig.MCACheckBox.setEnabled(1)
palette = self.specfitGui.guiconfig.MCACheckBox.palette()
##############
hbox = qt.QWidget(self)
hboxlayout = qt.QHBoxLayout(hbox)
hboxlayout.setContentsMargins(0, 0, 0, 0)
hboxlayout.setSpacing(0)
self.estimatebutton = qt.QPushButton(hbox)
self.estimatebutton.setText("Estimate")
self.fitbutton = qt.QPushButton(hbox)
self.fitbutton.setText("Fit")
hboxlayout.addWidget(self.estimatebutton)
hboxlayout.addWidget(qt.HorizontalSpacer(hbox))
hboxlayout.addWidget(self.fitbutton)
self.dismissbutton = qt.QPushButton(hbox)
self.dismissbutton.setText("Dismiss")
self.estimatebutton.clicked.connect(self.estimate)
self.fitbutton.clicked.connect(self.fit)
self.dismissbutton.clicked.connect(self.dismiss)
self.specfitGui.sigSpecfitGuiSignal.connect(self._specfitGuiSignal)
hboxlayout.addWidget(qt.HorizontalSpacer(hbox))
hboxlayout.addWidget(self.dismissbutton)
layout.addWidget(self.headerlabel)
layout.addWidget(self.specfitGui)
layout.addWidget(hbox)