def settings_page(self):
"create fit and background settings"
mca = self.parent.mca
conf = self.parent.conf
wids = self.wids
width = getattr(mca, 'bgr_width', 4)
compr = getattr(mca, 'bgr_compress', 2)
expon = getattr(mca, 'bgr_exponent', 2)
p = GridPanel(self, itemstyle=LEFT)
wids.bgr_use = Check(p, label='Fit Background-Subtracted Spectra',
default=True)
wids.bgr_width = FloatCtrl(p, value=width, minval=0, maxval=10,
precision=1, size=(70, -1))
wids.bgr_compress = Choice(p, choices=['1', '2', '4', '8', '16'],
size=(70, -1), default=1)
wids.bgr_exponent = Choice(p, choices=['2', '4', '6'],
size=(70, -1), default=0)
sopts = {'vary': True, 'precision': 5}
sig_offset = Parameter(value=0.050, name=self.gsig_offset, vary=True)
sig_slope = Parameter(value=0.005, name=self.gsig_slope, vary=True)
sig_quad = Parameter(value=0.000, name=self.gsig_quad, vary=False)
setattr(self.paramgroup, self.gsig_offset, sig_offset)
setattr(self.paramgroup, self.gsig_slope, sig_slope)
setattr(self.paramgroup, self.gsig_quad, sig_quad)
wids.sig_offset = ParameterPanel(p, sig_offset, vary=True, precision=5)
wids.sig_slope = ParameterPanel(p, sig_slope, vary=True, precision=5)
wids.sig_quad = ParameterPanel(p, sig_quad, vary=False, precision=5)
wids.xray_en = FloatCtrl(p, value=20.0, size=(70, -1),
minval=0, maxval=1000, precision=3)
wids.fit_emin = FloatCtrl(p, value=conf.e_min, size=(70, -1),
minval=0, maxval=1000, precision=3)
wids.fit_emax = FloatCtrl(p, value=conf.e_max, size=(70, -1),
minval=0, maxval=1000, precision=3)
wids.flyield_use = Check(p,
label='Account for Absorption, Fluorescence Efficiency')
p.AddText(' General Settings ', colour='#880000', dcol=3)
p.AddText(' X-ray Energy (keV): ', newrow=True)
p.Add(wids.xray_en)
p.AddText(' Min Energy (keV): ', newrow=True)
p.Add(wids.fit_emin)
p.AddText(' Max Energy (keV): ', newrow=False)
p.Add(wids.fit_emax)
wids.det_mat = Choice(p, choices=self.Detector_Materials,
size=(55, -1), default=0)
wids.det_thk = FloatCtrl(p, value=0.40, size=(70, -1),
minval=0, maxval=100, precision=3)
wids.det_use = Check(p, label='Account for Detector Thickness')
p.AddText(' Detector Material: ', newrow=True)
p.Add(wids.det_mat)
p.AddText(' Thickness (mm): ', newrow=False)
p.Add(wids.det_thk)
p.Add(wids.det_use, dcol=4, newrow=True)
p.Add(wids.flyield_use, dcol=4, newrow=True)
p.Add(HLine(p, size=(600, 3)), dcol=5, newrow=True)
p.AddText(' Energy Dependence of Peak Width:',
colour='#880000', newrow=True, dcol=2)
# p.AddText(' ', newrow=True)
p.AddText(' sigma = offset + slope * Energy + quad * Energy^2 (keV)',
dcol=3)
p.AddText(' Offset: ', newrow=True)
p.Add(wids.sig_offset, dcol=3)
p.AddText(' Slope: ', newrow=True)
p.Add(wids.sig_slope, dcol=3)
p.AddText(' Quad: ', newrow=True)
p.Add(wids.sig_quad, dcol=3)
p.Add(HLine(p, size=(600, 3)), dcol=5, newrow=True)
p.AddText(" Background Parameters: ", colour='#880000', dcol=2,
newrow=True)
p.Add(wids.bgr_use, dcol=2)
p.AddText(" Exponent:", newrow=True)
p.Add(wids.bgr_exponent)
p.AddText(" Energy Width (keV): ", newrow=False)
p.Add(wids.bgr_width)
p.AddText(" Compression: ", newrow=True)
p.Add(wids.bgr_compress)
p.Add(Button(p, 'Show Background', size=(130, -1),
action=self.onShowBgr), dcol=2)
p.Add(HLine(p, size=(600, 3)), dcol=5, newrow=True)
p.pack()
return p
def __init__(self, parent, param, precision=4, vary=None, **kws):
self.param = param
title = " Parameter: %s " % (param.name)
wx.Dialog.__init__(self, parent, wx.ID_ANY, title=title)
panel = GridPanel(self)
self.SetFont(parent.GetFont())
if vary is None:
vary = 0
if param.vary:
vary = 1
elif param.expr is not None:
vary = 2
minval, maxval = param.min, param.max
stderr, expr = param.stderr, param.expr
sminval = "%s" % minval
smaxval = "%s" % maxval
if minval in (None, 'None', -np.inf): minval = -np.inf
if maxval in (None, 'None', np.inf): maxval = np.inf
if stderr is None: stderr = ''
if expr is None: expr = ''
self.wids = Empty()
self.wids.vary = Choice(panel,
choices=VARY_CHOICES,
action=self.onVaryChoice,
size=(110, -1))
self.wids.vary.SetSelection(vary)
self.wids.val = FloatCtrl(panel,
value=param.value,
size=(100, -1),
precision=precision,
minval=minval,
maxval=maxval)
self.wids.min = FloatCtrl(panel, value=minval, size=(100, -1))
self.wids.max = FloatCtrl(panel, value=maxval, size=(100, -1))
self.wids.expr = wx.TextCtrl(panel, value=expr, size=(300, -1))
self.wids.err = wx.StaticText(panel, label="%s" % stderr)
SetTip(self.wids.expr, "Mathematical expression to calcutate value")
btnsizer = wx.StdDialogButtonSizer()
ok_btn = wx.Button(panel, wx.ID_OK)
ok_btn.SetDefault()
btnsizer.AddButton(ok_btn)
btnsizer.AddButton(wx.Button(panel, wx.ID_CANCEL))
btnsizer.Realize()
panel.AddText(' Name:', style=LEFT)
panel.AddText(param.name, style=LEFT)
panel.AddText(' Type:', style=LEFT)
panel.Add(self.wids.vary, style=LEFT)
panel.AddText(' Value:', style=LEFT, newrow=True)
panel.Add(self.wids.val, style=LEFT)
panel.AddText(' Std Error:', style=LEFT)
panel.Add(self.wids.err, style=LEFT)
panel.AddText(' Min Value:', style=LEFT, newrow=True)
panel.Add(self.wids.min, style=LEFT)
panel.AddText(' Max Value:', style=LEFT)
panel.Add(self.wids.max, style=LEFT)
panel.AddText(' Constraint:', style=LEFT, newrow=True)
panel.Add(self.wids.expr, style=LEFT, dcol=3)
panel.Add(HLine(panel, size=(375, 2)), dcol=4, newrow=True)
panel.Add(btnsizer, dcol=4, newrow=True, style=LEFT)
panel.pack()
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(panel, 0, 0, 25)
self.onVaryChoice()
pack(self, sizer)
bsize = self.GetBestSize()
self.SetSize((bsize[0] + 10, bsize[1] + 10))
def __init__(self, parent, size=(475, 325), **kws):
self.parent = parent
conf = parent.conf
kws['style'] = wx.DEFAULT_FRAME_STYLE
wx.Frame.__init__(self,
parent,
-1,
size=size,
title='XRF Line Selection',
**kws)
panel = GridPanel(self)
self.checkbox = {}
def add_elines(panel, lines, checked):
for i in lines:
cb = Check(panel,
'%s ' % i,
default=i in checked,
action=partial(self.onLine, label=i, lines=checked))
self.checkbox[i] = cb
panel.Add(cb, style=LEFT)
hopts = {'size': (125, -1), 'bgcolour': (250, 250, 200)}
labopts = {'newrow': True, 'style': LEFT}
self.linedata = {
'Major K Lines:': self.k1lines,
'Minor K Lines:': self.k2lines,
'Major L Lines:': self.l1lines,
'Minor L Lines:': self.l2lines + self.l3lines,
'Major M Lines:': self.mlines
}
panel.AddText(' Select X-ray Emission Lines', dcol=4, colour='#880000')
panel.Add(HLine(panel, size=(450, 3)), dcol=5, newrow=True)
panel.Add(
HyperText(panel,
'Major K Lines:',
action=partial(self.ToggleLines, lines=conf.K_major),
**hopts), **labopts)
add_elines(panel, self.k1lines, conf.K_major)
panel.Add(
HyperText(panel,
'Minor K Lines:',
action=partial(self.ToggleLines, lines=conf.K_minor),
**hopts), **labopts)
add_elines(panel, self.k2lines, conf.K_minor)
panel.Add(
HyperText(panel,
'Major L Lines:',
action=partial(self.ToggleLines, lines=conf.L_major),
**hopts), **labopts)
add_elines(panel, self.l1lines, conf.L_major)
panel.Add(
HyperText(panel,
'Minor L Lines:',
action=partial(self.ToggleLines, lines=conf.L_minor),
**hopts), **labopts)
add_elines(panel, self.l2lines, conf.L_minor)
panel.AddText(' ', **labopts)
add_elines(panel, self.l3lines, conf.L_minor)
panel.Add(
HyperText(panel,
'Major M Lines:',
action=partial(self.ToggleLines, lines=conf.M_major),
**hopts), **labopts)
add_elines(panel, self.mlines, conf.M_major)
panel.AddText('Energy Range (keV): ', **labopts)
fopts = {'minval': 0, 'maxval': 1000, 'precision': 2, 'size': (75, -1)}
panel.Add(FloatCtrl(panel,
value=conf.e_min,
action=partial(self.onErange, is_max=False),
**fopts),
dcol=2,
style=LEFT)
panel.AddText(' : ')
panel.Add(FloatCtrl(panel,
value=conf.e_max,
action=partial(self.onErange, is_max=True),
**fopts),
dcol=2,
style=LEFT)
panel.Add(HLine(panel, size=(450, 3)), dcol=5, newrow=True)
panel.Add(Button(panel, 'Done', size=(80, -1), action=self.onDone),
newrow=True,
style=LEFT)
panel.pack()
self.Show()
self.Raise()
def __init__(self, parent, mca, larch=None, size=(-1, -1), callback=None):
self.mca = mca
self.larch = larch
self.callback = callback
wx.Frame.__init__(self,
parent,
-1,
'Calibrate MCA',
size=size,
style=wx.DEFAULT_FRAME_STYLE)
self.SetFont(Font(8))
panel = GridPanel(self)
self.calib_updated = False
panel.AddText("Calibrate MCA Energy (Energies in eV)",
colour='#880000',
dcol=7)
panel.AddText("ROI", newrow=True, style=CEN)
panel.AddText("Predicted", style=CEN)
panel.AddText("Peaks with Current Calibration", dcol=3, style=CEN)
panel.AddText("Peaks with Refined Calibration", dcol=3, style=CEN)
panel.AddText("Name", newrow=True, style=CEN)
panel.AddText("Energy", style=CEN)
panel.AddText("Center", style=CEN)
panel.AddText("Difference", style=CEN)
panel.AddText("FWHM", style=CEN)
panel.AddText("Center", style=CEN)
panel.AddText("Difference", style=CEN)
panel.AddText("FWHM", style=CEN)
panel.AddText("Use? ", style=CEN)
panel.Add(HLine(panel, size=(700, 3)), dcol=9, newrow=True)
self.wids = []
# find ROI peak positions
xrf_calib_fitrois(mca, _larch=self.larch)
for roi in self.mca.rois:
try:
eknown, ecen, fwhm, amp, fit = mca.init_calib[roi.name]
except:
continue
diff = ecen - eknown
name = (' ' + roi.name + ' ' * 10)[:10]
opts = {'style': CEN, 'size': (75, -1)}
w_name = SimpleText(panel, name, **opts)
w_pred = SimpleText(panel, "% .1f" % (1000 * eknown), **opts)
w_ccen = SimpleText(panel, "% .1f" % (1000 * ecen), **opts)
w_cdif = SimpleText(panel, "% .1f" % (1000 * diff), **opts)
w_cwid = SimpleText(panel, "% .1f" % (1000 * fwhm), **opts)
w_ncen = SimpleText(panel, "-----", **opts)
w_ndif = SimpleText(panel, "-----", **opts)
w_nwid = SimpleText(panel, "-----", **opts)
w_use = Check(panel,
label='use?',
size=(40, -1),
default=fwhm < 0.50)
panel.Add(w_name, style=LEFT, newrow=True)
panel.AddMany((w_pred, w_ccen, w_cdif, w_cwid, w_ncen, w_ndif,
w_nwid, w_use))
self.wids.append(
(roi.name, eknown, ecen, w_ncen, w_ndif, w_nwid, w_use))
panel.Add(HLine(panel, size=(700, 3)), dcol=9, newrow=True)
offset = 1000.0 * self.mca.offset
slope = 1000.0 * self.mca.slope
panel.AddText("Current Calibration:", dcol=2, newrow=True)
panel.AddText("offset(eV):")
panel.AddText("%.3f" % (offset), dcol=1, style=RIGHT)
panel.AddText("slope(eV/chan):")
panel.AddText("%.3f" % (slope), dcol=1, style=RIGHT)
panel.AddText("Refined Calibration:", dcol=2, newrow=True)
self.new_offset = FloatCtrl(panel,
value=offset,
precision=3,
size=(80, -1))
self.new_slope = FloatCtrl(panel,
value=slope,
precision=3,
size=(80, -1))
panel.AddText("offset(eV):")
panel.Add(self.new_offset, dcol=1, style=RIGHT)
panel.AddText("slope(eV/chan):")
panel.Add(self.new_slope, dcol=1, style=RIGHT)
panel.Add(Button(panel,
'Compute Calibration',
size=(125, -1),
action=self.onCalibrate),
dcol=2,
newrow=True)
panel.Add(Button(panel,
'Use New Calibration',
size=(125, -1),
action=self.onUseCalib),
dcol=2,
style=RIGHT)
panel.Add(Button(panel, 'Done', size=(125, -1), action=self.onClose),
dcol=2,
style=RIGHT)
panel.pack()
a = panel.GetBestSize()
self.SetSize((a[0] + 25, a[1] + 50))
self.Show()
self.Raise()
def __init__(self, parent, mca, size=(-1, -1), callback=None):
self.mca = mca
self.callback = callback
wx.Frame.__init__(self,
parent,
-1,
'Calibrate MCA',
size=size,
style=wx.DEFAULT_FRAME_STYLE)
panel = GridPanel(self)
self.calib_updated = False
panel.AddText("Calibrate MCA Energy (Energies in eV)",
colour='#880000',
dcol=7)
panel.AddText("ROI", newrow=True, style=CEN)
panel.AddText("Predicted", style=CEN)
panel.AddText("Peaks with Current Calibration", dcol=3, style=CEN)
panel.AddText("Peaks with Refined Calibration", dcol=3, style=CEN)
panel.AddText("Name", newrow=True, style=CEN)
panel.AddText("Energy", style=CEN)
panel.AddText("Center", style=CEN)
panel.AddText("Difference", style=CEN)
panel.AddText("FWHM", style=CEN)
panel.AddText("Center", style=CEN)
panel.AddText("Difference", style=CEN)
panel.AddText("FWHM", style=CEN)
panel.AddText("Use? ", style=CEN)
panel.Add(HLine(panel, size=(700, 3)), dcol=9, newrow=True)
self.wids = []
# find ROI peak positions
self.init_wids = {}
for roi in self.mca.rois:
eknown, ecen, fwhm = 1, 1, 1
words = roi.name.split()
elem = words[0].title()
family = 'ka'
if len(words) > 1:
family = words[1]
try:
eknown = xray_line(elem, family).energy / 1000.0
except (AttributeError, ValueError):
eknown = 0.0001
mid = (roi.right + roi.left) / 2
wid = (roi.right - roi.left) / 2
ecen = mid * mca.slope + mca.offset
fwhm = 2.354820 * wid * mca.slope
diff = ecen - eknown
name = (' ' + roi.name + ' ' * 10)[:10]
opts = {'style': CEN, 'size': (75, -1)}
w_name = SimpleText(panel, name, **opts)
w_pred = SimpleText(panel, "% .1f" % (1000 * eknown), **opts)
w_ccen = SimpleText(panel, "% .1f" % (1000 * ecen), **opts)
w_cdif = SimpleText(panel, "% .1f" % (1000 * diff), **opts)
w_cwid = SimpleText(panel, "% .1f" % (1000 * fwhm), **opts)
w_ncen = SimpleText(panel, "-----", **opts)
w_ndif = SimpleText(panel, "-----", **opts)
w_nwid = SimpleText(panel, "-----", **opts)
w_use = Check(panel, label='', size=(40, -1), default=0)
panel.Add(w_name, style=LEFT, newrow=True)
panel.AddMany((w_pred, w_ccen, w_cdif, w_cwid, w_ncen, w_ndif,
w_nwid, w_use))
self.init_wids[roi.name] = [
False, w_pred, w_ccen, w_cdif, w_cwid, w_use
]
self.wids.append(
(roi.name, eknown, ecen, w_ncen, w_ndif, w_nwid, w_use))
panel.Add(HLine(panel, size=(700, 3)), dcol=9, newrow=True)
offset = 1000.0 * self.mca.offset
slope = 1000.0 * self.mca.slope
panel.AddText("Current Calibration:", dcol=2, newrow=True)
panel.AddText("offset(eV):")
panel.AddText("%.3f" % (offset), dcol=1, style=RIGHT)
panel.AddText("slope(eV/chan):")
panel.AddText("%.3f" % (slope), dcol=1, style=RIGHT)
panel.AddText("Refined Calibration:", dcol=2, newrow=True)
self.new_offset = FloatCtrl(panel,
value=offset,
precision=3,
size=(80, -1))
self.new_slope = FloatCtrl(panel,
value=slope,
precision=3,
size=(80, -1))
panel.AddText("offset(eV):")
panel.Add(self.new_offset, dcol=1, style=RIGHT)
panel.AddText("slope(eV/chan):")
panel.Add(self.new_slope, dcol=1, style=RIGHT)
self.calib_btn = Button(panel,
'Compute Calibration',
size=(175, -1),
action=self.onCalibrate)
self.calib_btn.Disable()
panel.Add(self.calib_btn, dcol=3, newrow=True)
panel.Add(Button(panel,
'Use New Calibration',
size=(175, -1),
action=self.onUseCalib),
dcol=3,
style=RIGHT)
panel.Add(Button(panel, 'Done', size=(125, -1), action=self.onClose),
dcol=2,
style=RIGHT)
panel.pack()
a = panel.GetBestSize()
self.SetSize((a[0] + 25, a[1] + 50))
self.Show()
self.Raise()
self.init_proc = False
self.init_t0 = time.time()
self.init_timer = wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.onInitTimer, self.init_timer)
self.init_timer.Start(2)
def addModel(self, event=None, model=None, prefix=None, isbkg=False):
if model is None and event is not None:
model = event.GetString()
if model is None or model.startswith('<'):
return
if prefix is None:
p = model[:5].lower()
curmodels = ["%s%i_" % (p, i+1) for i in range(1+len(self.fit_components))]
for comp in self.fit_components:
if comp in curmodels:
curmodels.remove(comp)
prefix = curmodels[0]
label = "%s(prefix='%s')" % (model, prefix)
title = "%s: %s " % (prefix[:-1], model)
title = prefix[:-1]
mclass_kws = {'prefix': prefix}
if 'step' in model.lower():
form = model.lower().replace('step', '').strip()
if form.startswith('err'): form = 'erf'
label = "Step(form='%s', prefix='%s')" % (form, prefix)
title = "%s: Step %s" % (prefix[:-1], form[:3])
mclass = lm_models.StepModel
mclass_kws['form'] = form
minst = mclass(form=form, prefix=prefix)
else:
if model in ModelFuncs:
mclass = getattr(lm_models, ModelFuncs[model])
else:
mclass = getattr(lm_models, model+'Model')
minst = mclass(prefix=prefix)
panel = GridPanel(self.mod_nb, ncols=1, nrows=1, pad=1, itemstyle=CEN)
def SLabel(label, size=(80, -1), **kws):
return SimpleText(panel, label,
size=size, style=wx.ALIGN_LEFT, **kws)
usebox = Check(panel, default=True, label='Use in Fit?', size=(100, -1))
bkgbox = Check(panel, default=False, label='Is Baseline?', size=(125, -1))
if isbkg:
bkgbox.SetValue(1)
delbtn = Button(panel, 'Delete Component', size=(125, -1),
action=partial(self.onDeleteComponent, prefix=prefix))
pick2msg = SimpleText(panel, " ", size=(125, -1))
pick2btn = Button(panel, 'Pick Values from Data', size=(150, -1),
action=partial(self.onPick2Points, prefix=prefix))
# SetTip(mname, 'Label for the model component')
SetTip(usebox, 'Use this component in fit?')
SetTip(bkgbox, 'Label this component as "background" when plotting?')
SetTip(delbtn, 'Delete this model component')
SetTip(pick2btn, 'Select X range on Plot to Guess Initial Values')
panel.Add(SLabel(label, size=(275, -1), colour='#0000AA'),
dcol=3, style=wx.ALIGN_LEFT, newrow=True)
panel.Add(usebox, dcol=1)
panel.Add(bkgbox, dcol=2, style=LCEN)
panel.Add(delbtn, dcol=1, style=wx.ALIGN_LEFT)
panel.Add(pick2btn, dcol=2, style=wx.ALIGN_LEFT, newrow=True)
panel.Add(pick2msg, dcol=2, style=wx.ALIGN_RIGHT)
# panel.Add((10, 10), newrow=True)
# panel.Add(HLine(panel, size=(150, 3)), dcol=4, style=wx.ALIGN_CENTER)
panel.Add(SLabel("Parameter "), style=wx.ALIGN_LEFT, newrow=True)
panel.AddMany((SLabel(" Value"), SLabel(" Type"), SLabel(' Bounds'),
SLabel(" Min", size=(60, -1)),
SLabel(" Max", size=(60, -1)), SLabel(" Expression")))
parwids = OrderedDict()
parnames = sorted(minst.param_names)
for a in minst._func_allargs:
pname = "%s%s" % (prefix, a)
if (pname not in parnames and
a in minst.param_hints and
a not in minst.independent_vars):
parnames.append(pname)
for pname in parnames:
sname = pname[len(prefix):]
hints = minst.param_hints.get(sname, {})
par = Parameter(name=pname, value=0, vary=True)
if 'min' in hints:
par.min = hints['min']
if 'max' in hints:
par.max = hints['max']
if 'value' in hints:
par.value = hints['value']
if 'expr' in hints:
par.expr = hints['expr']
pwids = ParameterWidgets(panel, par, name_size=100, expr_size=125,
float_size=80, prefix=prefix,
widgets=('name', 'value', 'minval',
'maxval', 'vary', 'expr'))
parwids[par.name] = pwids
panel.Add(pwids.name, newrow=True)
panel.AddMany((pwids.value, pwids.vary, pwids.bounds,
pwids.minval, pwids.maxval, pwids.expr))
for sname, hint in minst.param_hints.items():
pname = "%s%s" % (prefix, sname)
if 'expr' in hint and pname not in parnames:
par = Parameter(name=pname, value=0, expr=hint['expr'])
pwids = ParameterWidgets(panel, par, name_size=100, expr_size=225,
float_size=80, prefix=prefix,
widgets=('name', 'value', 'expr'))
parwids[par.name] = pwids
panel.Add(pwids.name, newrow=True)
panel.Add(pwids.value)
panel.Add(pwids.expr, dcol=4, style=wx.ALIGN_RIGHT)
pwids.value.Disable()
fgroup = Group(prefix=prefix, title=title, mclass=mclass,
mclass_kws=mclass_kws, usebox=usebox, panel=panel,
parwids=parwids, float_size=65, expr_size=150,
pick2_msg=pick2msg, bkgbox=bkgbox)
self.fit_components[prefix] = fgroup
panel.pack()
self.mod_nb.AddPage(panel, title, True)
sx,sy = self.GetSize()
self.SetSize((sx, sy+1))
self.SetSize((sx, sy))
