def __init__(self, parent, group, on_ok=None):
self.parent = parent
self.group = group
self.on_ok = on_ok
wx.Frame.__init__(self,
None,
-1,
'Edit Array Names',
style=wx.DEFAULT_FRAME_STYLE | wx.TAB_TRAVERSAL)
self.SetFont(Font(10))
sizer = wx.GridBagSizer(2, 2)
panel = scrolled.ScrolledPanel(self)
self.SetMinSize((675, 450))
self.wids = {}
ir = 0
sizer.Add(
Button(panel, 'Apply Changes', size=(200, -1), action=self.onOK),
(0, 1), (1, 2), LEFT, 3)
sizer.Add(
Button(panel,
'Use Column Number',
size=(200, -1),
action=self.onColNumber), (0, 3), (1, 2), LEFT, 3)
sizer.Add(HLine(panel, size=(550, 2)), (1, 1), (1, 5), LEFT, 3)
cind = SimpleText(panel, label='Column')
cold = SimpleText(panel, label='Current Name')
cnew = SimpleText(panel, label='Enter New Name')
cret = SimpleText(panel, label=' Result ', size=(150, -1))
cinfo = SimpleText(panel, label=' Data Range')
cplot = SimpleText(panel, label=' Plot')
ir = 2
sizer.Add(cind, (ir, 0), (1, 1), LEFT, 3)
sizer.Add(cold, (ir, 1), (1, 1), LEFT, 3)
sizer.Add(cnew, (ir, 2), (1, 1), LEFT, 3)
sizer.Add(cret, (ir, 3), (1, 1), LEFT, 3)
sizer.Add(cinfo, (ir, 4), (1, 1), LEFT, 3)
sizer.Add(cplot, (ir, 5), (1, 1), LEFT, 3)
for i, name in enumerate(group.array_labels):
ir += 1
cind = SimpleText(panel, label=' %i ' % (i + 1))
cold = SimpleText(panel, label=' %s ' % name)
cret = SimpleText(panel, label=fix_varname(name), size=(150, -1))
cnew = wx.TextCtrl(panel, value=name, size=(150, -1))
cnew.Bind(wx.EVT_KILL_FOCUS, partial(self.update, index=i))
cnew.Bind(wx.EVT_CHAR, partial(self.update_char, index=i))
cnew.Bind(wx.EVT_TEXT_ENTER, partial(self.update, index=i))
arr = group.data[i, :]
info_str = " [ %8g : %8g ] " % (arr.min(), arr.max())
cinfo = SimpleText(panel, label=info_str)
cplot = Button(panel, 'Plot', action=partial(self.onPlot, index=i))
self.wids["%d" % i] = cnew
self.wids["ret_%d" % i] = cret
sizer.Add(cind, (ir, 0), (1, 1), LEFT, 3)
sizer.Add(cold, (ir, 1), (1, 1), LEFT, 3)
sizer.Add(cnew, (ir, 2), (1, 1), LEFT, 3)
sizer.Add(cret, (ir, 3), (1, 1), LEFT, 3)
sizer.Add(cinfo, (ir, 4), (1, 1), LEFT, 3)
sizer.Add(cplot, (ir, 5), (1, 1), LEFT, 3)
pack(panel, sizer)
panel.SetupScrolling()
mainsizer = wx.BoxSizer(wx.VERTICAL)
mainsizer.Add(panel, 1, wx.GROW | wx.ALL, 1)
pack(self, mainsizer)
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,
filename=None,
read_ok_cb=None,
size=(725, 450),
_larch=None):
self.parent = parent
self.filename = filename
self.larch = _larch
self.read_ok_cb = read_ok_cb
self.colors = GUIColors()
wx.Frame.__init__(self, parent, -1, size=size, style=FRAMESTYLE)
splitter = wx.SplitterWindow(self, style=wx.SP_LIVE_UPDATE)
splitter.SetMinimumPaneSize(200)
leftpanel = wx.Panel(splitter)
ltop = wx.Panel(leftpanel)
sel_none = Button(ltop,
'Select None',
size=(100, 30),
action=self.onSelNone)
sel_all = Button(ltop,
'Select All',
size=(100, 30),
action=self.onSelAll)
sel_imp = Button(ltop,
'Import Selected Groups',
size=(200, 30),
action=self.onOK)
self.grouplist = FileCheckList(leftpanel,
select_action=self.onShowGroup)
self.grouplist.SetBackgroundColour(wx.Colour(255, 255, 255))
tsizer = wx.GridBagSizer(2, 2)
tsizer.Add(sel_all, (0, 0), (1, 1), LCEN, 0)
tsizer.Add(sel_none, (0, 1), (1, 1), LCEN, 0)
tsizer.Add(sel_imp, (1, 0), (1, 2), LCEN, 0)
pack(ltop, tsizer)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(ltop, 0, LCEN | wx.GROW, 1)
sizer.Add(self.grouplist, 1, LCEN | wx.GROW | wx.ALL, 1)
pack(leftpanel, sizer)
# right hand side
rightpanel = wx.Panel(splitter)
self.SetTitle("Reading Athena Project '%s'" % self.filename)
self.title = SimpleText(rightpanel,
self.filename,
font=Font(13),
colour=self.colors.title,
style=LCEN)
self.plotpanel = PlotPanel(rightpanel, messenger=self.plot_messages)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(self.title, 0, LCEN, 2)
sizer.Add(self.plotpanel, 0, LCEN, 2)
pack(rightpanel, sizer)
splitter.SplitVertically(leftpanel, rightpanel, 1)
self.statusbar = self.CreateStatusBar(2, 0)
self.statusbar.SetStatusWidths([-3, -1])
statusbar_fields = [self.filename, ""]
for i in range(len(statusbar_fields)):
self.statusbar.SetStatusText(statusbar_fields[i], i)
self.all = read_athena(self.filename,
do_bkg=False,
do_fft=False,
_larch=_larch)
for item in dir(self.all):
self.grouplist.Append(item)
self.Show()
self.Raise()
def build_display(self):
self.mod_nb = flat_nb.FlatNotebook(self, -1, agwStyle=FNB_STYLE)
self.mod_nb.SetTabAreaColour(wx.Colour(250,250,250))
self.mod_nb.SetActiveTabColour(wx.Colour(254,254,195))
self.mod_nb.SetNonActiveTabTextColour(wx.Colour(10,10,128))
self.mod_nb.SetActiveTabTextColour(wx.Colour(128,0,0))
self.mod_nb.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.onNBChanged)
pan = self.panel = GridPanel(self, ncols=4, nrows=4, pad=2, itemstyle=LCEN)
self.btns = {}
for name in ('ppeak_e0', 'ppeak_elo', 'ppeak_emin', 'ppeak_emax', 'ppeak_ehi'):
bb = BitmapButton(pan, get_icon('plus'),
action=partial(self.on_selpoint, opt=name),
tooltip='use last point selected from plot')
self.btns[name] = bb
opts = dict(size=(75, -1), gformat=True, precision=1)
self.ppeak_e0 = FloatCtrl(pan, value=0, **opts)
self.ppeak_emin = FloatCtrl(pan, value=-30, **opts)
self.ppeak_emax = FloatCtrl(pan, value=0, **opts)
self.ppeak_elo = FloatCtrl(pan, value=-15, **opts)
self.ppeak_ehi = FloatCtrl(pan, value=-5, **opts)
self.fitbline_btn = Button(pan,'Fit Baseline', action=self.onFitBaseline,
size=(150, 25))
self.fitmodel_btn = Button(pan, 'Fit Model',
action=self.onFitModel, size=(150, 25))
self.fitsel_btn = Button(pan, 'Fit Selected Groups',
action=self.onFitSelected, size=(150, 25))
self.fitmodel_btn.Disable()
self.fitsel_btn.Disable()
self.array_choice = Choice(pan, size=(125, -1),
choices=list(Array_Choices.keys()))
self.array_choice.SetSelection(1)
models_peaks = Choice(pan, size=(125, -1),
choices=ModelChoices['peaks'],
action=self.addModel)
models_other = Choice(pan, size=(125, -1),
choices=ModelChoices['other'],
action=self.addModel)
self.plot_choice = Choice(pan, size=(125, -1),
choices=PlotChoices,
action=self.onPlot)
self.message = SimpleText(pan,
'first fit baseline, then add peaks to fit model.')
self.msg_centroid = SimpleText(pan, '----')
opts = dict(default=True, size=(75, -1), action=self.onPlot)
self.show_centroid = Check(pan, label='show?', **opts)
self.show_peakrange = Check(pan, label='show?', **opts)
self.show_fitrange = Check(pan, label='show?', **opts)
self.show_e0 = Check(pan, label='show?', **opts)
opts = dict(default=False, size=(200, -1), action=self.onPlot)
self.plot_sub_bline = Check(pan, label='Subtract Baseline?', **opts)
titleopts = dict(font=Font(11), colour='#AA0000')
pan.Add(SimpleText(pan, ' Pre-edge Peak Fitting', **titleopts), dcol=7)
pan.Add(SimpleText(pan, ' Run Fit:'), style=LCEN)
pan.Add(SimpleText(pan, 'Array to fit: '), newrow=True)
pan.Add(self.array_choice, dcol=4)
pan.Add((10, 10), dcol=2)
pan.Add(self.fitbline_btn)
pan.Add(SimpleText(pan, 'E0: '), newrow=True)
pan.Add(self.btns['ppeak_e0'])
pan.Add(self.ppeak_e0)
pan.Add((10, 10), dcol=3)
pan.Add(self.show_e0)
pan.Add(self.fitmodel_btn)
pan.Add(SimpleText(pan, 'Fit Energy Range: '), newrow=True)
pan.Add(self.btns['ppeak_emin'])
pan.Add(self.ppeak_emin)
pan.Add(SimpleText(pan, ':'))
pan.Add(self.btns['ppeak_emax'])
pan.Add(self.ppeak_emax)
pan.Add(self.show_fitrange, dcol=1)
pan.Add(self.fitsel_btn)
t = SimpleText(pan, 'Pre-edge Peak Range: ')
t.SetToolTip('Range used as mask for background')
pan.Add(t, newrow=True)
pan.Add(self.btns['ppeak_elo'])
pan.Add(self.ppeak_elo)
pan.Add(SimpleText(pan, ':'))
pan.Add(self.btns['ppeak_ehi'])
pan.Add(self.ppeak_ehi)
pan.Add(self.show_peakrange, dcol=1)
pan.Add(SimpleText(pan, 'Peak Centroid: '), newrow=True)
pan.Add(self.msg_centroid, dcol=5)
pan.Add(self.show_centroid, dcol=1)
# plot buttons
ts = wx.BoxSizer(wx.HORIZONTAL)
ts.Add(self.plot_choice)
ts.Add(self.plot_sub_bline)
pan.Add(SimpleText(pan, 'Plot: '), newrow=True)
pan.Add(ts, dcol=7)
# add model
ts = wx.BoxSizer(wx.HORIZONTAL)
ts.Add(models_peaks)
ts.Add(models_other)
pan.Add(SimpleText(pan, 'Add Component: '), newrow=True)
pan.Add(ts, dcol=7)
pan.Add(SimpleText(pan, 'Messages: '), newrow=True)
pan.Add(self.message, dcol=7)
pan.pack()
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add((5,5), 0, LCEN, 3)
sizer.Add(HLine(self, size=(550, 2)), 0, LCEN, 3)
sizer.Add(pan, 0, LCEN, 3)
sizer.Add((5,5), 0, LCEN, 3)
sizer.Add(HLine(self, size=(550, 2)), 0, LCEN, 3)
sizer.Add((5,5), 0, LCEN, 3)
sizer.Add(self.mod_nb, 1, LCEN|wx.GROW, 10)
pack(self, sizer)
def scolor(txt, attr, **kws):
panel.AddText(txt, size=(SX, -1), style=LEFT, font=Font(11), **kws)
panel.Add(add_color(panel, attr), style=LEFT)
def __init__(self, _larch=None, parent=None, mca_file=None,
size=(725, 450), axissize=None, axisbg=None,
title='XRF Display', exit_callback=None,
output_title='XRF', **kws):
if size is None: size = (725, 450)
wx.Frame.__init__(self, parent=parent,
title=title, size=size, **kws)
self.conf = XRFDisplayConfig()
self.subframes = {}
self.data = None
self.title = title
self.plotframe = None
self.wids = {}
self.larch = _larch
if self.larch is None:
self.init_larch()
self._mcagroup = self.larch.symtable.new_group('_mcas')
self.exit_callback = exit_callback
self.roi_patch = None
self.selected_roi = None
self.roilist_sel = None
self.selected_elem = None
self.mca = None
self.mca2 = None
self.xdata = np.arange(2048)*0.015
self.ydata = np.ones(2048)*1.e-4
self.x2data = None
self.y2data = None
self.rois_shown = False
self.major_markers = []
self.minor_markers = []
self.hold_markers = []
self.hold_lines = None
self.saved_lines = None
self.energy_for_zoom = None
self.xview_range = None
self.show_yaxis = False
self.xmarker_left = None
self.xmarker_right = None
self.highlight_xrayline = None
self.highlight_xrayline = None
self.cursor_markers = [None, None]
self.ylog_scale = True
self.SetTitle("%s: %s " % (self.main_title, title))
self._menus = []
self.createMainPanel()
self.createMenus()
self.SetFont(Font(9, serif=True))
self.statusbar = self.CreateStatusBar(4)
self.statusbar.SetStatusWidths([-5, -3, -3, -4])
statusbar_fields = ["XRF Display", " ", " ", " "]
for i in range(len(statusbar_fields)):
self.statusbar.SetStatusText(statusbar_fields[i], i)
if mca_file is not None:
self.mca = gsemca_group(mca_file, _larch=self.larch)
self._mcagroup.mca1 = self.mca
self._mcagroup.mca2 = None
self.plotmca(self.mca, show_mca2=False)
def createEpicsPanel(self):
pane = wx.Panel(self, name='epics panel')
psizer = wx.GridBagSizer(4, 12) # wx.BoxSizer(wx.HORIZONTAL)
btnpanel = wx.Panel(pane, name='buttons')
nmca = self.nmca
NPERROW = 6
self.SetFont(Font(9))
if self.det_type.lower().startswith('me-4') and nmca < 5:
btnsizer = wx.GridBagSizer(2, 2)
else:
btnsizer = wx.GridBagSizer(
int((nmca + NPERROW - 2) / (1.0 * NPERROW)), NPERROW)
style = wx.ALIGN_LEFT | wx.ALIGN_CENTER_VERTICAL
rstyle = wx.ALIGN_RIGHT | wx.ALIGN_CENTER_VERTICAL
bkg_choices = ['None']
psizer.Add(SimpleText(pane, ' MCAs: '), (0, 0), (1, 1), style, 1)
for i in range(1, 1 + nmca):
bkg_choices.append("%i" % i)
b = Button(btnpanel,
'%i' % i,
size=(30, 25),
action=partial(self.onSelectDet, index=i))
self.wids['det%i' % i] = b
loc = divmod(i - 1, NPERROW)
if self.det_type.lower().startswith('me-4') and nmca < NPERROW - 1:
loc = self.me4_layout[i - 1]
btnsizer.Add(b, loc, (1, 1), style, 1)
pack(btnpanel, btnsizer)
nrows = 1 + loc[0]
if self.det_type.lower().startswith('me-4') and nmca < 5:
nrows = 2
psizer.Add(btnpanel, (0, 1), (nrows, 1), style, 1)
self.wids['det_status'] = SimpleText(pane,
' ',
size=(120, -1),
style=style)
self.wids['deadtime'] = SimpleText(pane,
' ',
size=(120, -1),
style=style)
self.wids['bkg_det'] = Choice(pane,
size=(75, -1),
choices=bkg_choices,
action=self.onSelectDet)
self.wids['dwelltime'] = FloatCtrl(pane,
value=0.0,
precision=1,
minval=0,
size=(80, -1),
act_on_losefocus=True,
action=self.onSetDwelltime)
self.wids['elapsed'] = SimpleText(pane,
' ',
size=(80, -1),
style=style)
b1 = Button(pane, 'Start', size=(90, 25), action=self.onStart)
b2 = Button(pane, 'Stop', size=(90, 25), action=self.onStop)
b3 = Button(pane, 'Erase', size=(90, 25), action=self.onErase)
b4 = Button(pane,
'Continuous',
size=(90, 25),
action=partial(self.onStart, dtime=0))
bkg_lab = SimpleText(pane, 'Background MCA:', size=(150, -1))
pre_lab = SimpleText(pane, 'Preset Time (s):', size=(125, -1))
ela_lab = SimpleText(pane, 'Elapsed Time (s):', size=(125, -1))
sta_lab = SimpleText(pane, 'Status :', size=(100, -1))
dea_lab = SimpleText(pane, '% Deadtime:', size=(100, -1))
psizer.Add(bkg_lab, (0, 2), (1, 1), style, 1)
psizer.Add(self.wids['bkg_det'], (1, 2), (1, 1), style, 1)
psizer.Add(pre_lab, (0, 3), (1, 1), style, 1)
psizer.Add(ela_lab, (1, 3), (1, 1), style, 1)
psizer.Add(self.wids['dwelltime'], (0, 4), (1, 1), style, 1)
psizer.Add(self.wids['elapsed'], (1, 4), (1, 1), style, 1)
psizer.Add(b1, (0, 5), (1, 1), style, 1)
psizer.Add(b4, (0, 6), (1, 1), style, 1)
psizer.Add(b2, (1, 5), (1, 1), style, 1)
psizer.Add(b3, (1, 6), (1, 1), style, 1)
psizer.Add(sta_lab, (0, 7), (1, 1), style, 1)
psizer.Add(self.wids['det_status'], (0, 8), (1, 1), style, 1)
psizer.Add(dea_lab, (1, 7), (1, 1), style, 1)
psizer.Add(self.wids['deadtime'], (1, 8), (1, 1), style, 1)
pack(pane, psizer)
# pane.SetMinSize((500, 53))
self.det.connect_displays(status=self.wids['det_status'],
elapsed=self.wids['elapsed'],
deadtime=self.wids['deadtime'])
wx.CallAfter(self.onSelectDet, index=1, init=True)
self.timer_counter = 0
self.mca_timer = wx.Timer(self)
self.Bind(wx.EVT_TIMER, self.UpdateData, self.mca_timer)
self.mca_timer.Start(100)
return pane
def createControlPanel(self):
ctrlpanel = wx.Panel(self, name='Ctrl Panel')
bkgpanel = self.createBkgdPanel()
searchpanel = self.createSearchPanel()
labstyle = wx.ALIGN_LEFT|wx.ALIGN_BOTTOM|wx.EXPAND
ctrlstyle = wx.ALIGN_LEFT|wx.ALIGN_BOTTOM
txtstyle=wx.ALIGN_LEFT|wx.ST_NO_AUTORESIZE|wx.TE_PROCESS_ENTER
Font9 = Font(9)
Font10 = Font(10)
Font11 = Font(11)
plttitle = txt(ctrlpanel, 'Plot Parameters', font=Font10, size=200)
# y scale
yscalepanel = wx.Panel(ctrlpanel, name='YScalePanel')
ysizer = wx.BoxSizer(wx.HORIZONTAL)
ytitle = txt(yscalepanel, ' Y Axis:', font=Font10, size=80)
yspace = txt(yscalepanel, ' ', font=Font10, size=20)
ylog = Choice(yscalepanel, size=(80, 30), choices=['linear', 'log'],
action=self.onLogLinear)
yaxis = Check(yscalepanel, ' Show Y Scale ', action=self.onYAxis,
default=True)
self.wids['show_yaxis'] = yaxis
ysizer.Add(ytitle, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 0)
ysizer.Add(ylog, 0, wx.EXPAND|wx.ALL, 0)
ysizer.Add(yspace, 0, wx.EXPAND|wx.ALL, 0)
ysizer.Add(yaxis, 0, wx.EXPAND|wx.ALL, 0)
pack(yscalepanel, ysizer)
# zoom buttons
zoompanel = wx.Panel(ctrlpanel, name='ZoomPanel')
zsizer = wx.BoxSizer(wx.HORIZONTAL)
z1 = Button(zoompanel, 'Zoom In', size=(80, 30), action=self.onZoomIn)
z2 = Button(zoompanel, 'Zoom Out', size=(80, 30), action=self.onZoomOut)
p1 = Button(zoompanel, 'Pan Lo', size=(75, 30), action=self.onPanLo)
p2 = Button(zoompanel, 'Pan Hi', size=(75, 30), action=self.onPanHi)
zsizer.Add(p1, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(p2, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(z1, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(z2, 0, wx.EXPAND|wx.ALL, 0)
pack(zoompanel, zsizer)
# x scale
xscalepanel = wx.Panel(ctrlpanel, name='XScalePanel')
xsizer = wx.BoxSizer(wx.HORIZONTAL)
xtitle = txt(xscalepanel, ' X Axis:', font=Font10, size=80)
xspace = txt(xscalepanel, ' ', font=Font10, size=20)
self.xaxis = wx.RadioBox(xscalepanel, -1, '',wx.DefaultPosition, wx.DefaultSize,
('q','2th','d'),
1, wx.RA_SPECIFY_ROWS)
self.xaxis.Bind(wx.EVT_RADIOBOX, self.onXaxis)
xsizer.Add(xtitle, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 0)
xsizer.Add(xspace, 0, wx.EXPAND|wx.ALL, 0)
xsizer.Add(self.xaxis, 0, wx.EXPAND|wx.ALL, 0)
pack(xscalepanel, xsizer)
###########################
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(plttitle, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(yscalepanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(xscalepanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(zoompanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(bkgpanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(searchpanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
pack(ctrlpanel, sizer)
return ctrlpanel
def __init__(self, parent, group, on_ok=None):
self.group = group
self.on_ok = on_ok
wx.Frame.__init__(self,
None,
-1,
'Edit Array Names',
style=wx.DEFAULT_FRAME_STYLE | wx.TAB_TRAVERSAL)
self.SetFont(Font(10))
sizer = wx.GridBagSizer(4, 4)
# if not hasattr(group, 'orig_array_labels'):
# group.orig_array_labels = group.array_labels[:]
# print("EDIT COLUMN NAMES ", group.array_labels)
self.SetMinSize((600, 600))
self.wids = {}
cind = SimpleText(self, label='Column')
cold = SimpleText(self, label='Current Name')
cnew = SimpleText(self, label='Enter New Name')
cret = SimpleText(self, label=' Result ', size=(150, -1))
cinfo = SimpleText(self, label=' Data Range')
ir = 0
sizer.Add(cind, (ir, 0), (1, 1), LCEN, 3)
sizer.Add(cold, (ir, 1), (1, 1), LCEN, 3)
sizer.Add(cnew, (ir, 2), (1, 1), LCEN, 3)
sizer.Add(cret, (ir, 3), (1, 1), LCEN, 3)
sizer.Add(cinfo, (ir, 4), (1, 1), LCEN, 3)
for i, name in enumerate(group.array_labels):
ir += 1
cind = SimpleText(self, label=' %i ' % (i + 1))
cold = SimpleText(self, label=' %s ' % name)
cret = SimpleText(self, label=fix_varname(name), size=(150, -1))
cnew = wx.TextCtrl(self, value=name, size=(150, -1))
cnew.Bind(wx.EVT_KILL_FOCUS, partial(self.update, index=i))
cnew.Bind(wx.EVT_CHAR, partial(self.update_char, index=i))
cnew.Bind(wx.EVT_TEXT_ENTER, partial(self.update, index=i))
# cnew.Bind(wx.EVT_TEXT, partial(self.update3, index=i))
arr = group.data[i, :]
info_str = " [ %8g : %8g ] " % (arr.min(), arr.max())
cinfo = SimpleText(self, label=info_str)
self.wids[i] = cnew
self.wids["ret_%i" % i] = cret
sizer.Add(cind, (ir, 0), (1, 1), LCEN, 3)
sizer.Add(cold, (ir, 1), (1, 1), LCEN, 3)
sizer.Add(cnew, (ir, 2), (1, 1), LCEN, 3)
sizer.Add(cret, (ir, 3), (1, 1), LCEN, 3)
sizer.Add(cinfo, (ir, 4), (1, 1), LCEN, 3)
sizer.Add(Button(self, 'OK', action=self.onOK), (ir + 1, 1), (1, 2),
LCEN, 3)
pack(self, sizer)
self.Show()
self.Raise()
def buildFrame(self):
self.SetFont(Font(11))
sbar = self.CreateStatusBar(3, wx.CAPTION)
self.SetStatusWidths([-1, -1, -1])
self.SetStatusText('', 0)
sizer = wx.GridBagSizer(3, 3)
panel = self.panel = wx.Panel(self)
pvpanel = PVConfigPanel(panel, self.prefix, self.config['controls'])
wsize = (100, -1)
lsize = (250, -1)
start_btn = wx.Button(panel, label='Start', size=wsize)
stop_btn = wx.Button(panel, label='Stop', size=wsize)
start_btn.Bind(wx.EVT_BUTTON, partial(self.onButton, key='start'))
stop_btn.Bind(wx.EVT_BUTTON, partial(self.onButton, key='stop'))
self.contrast = ContrastControl(panel,
callback=self.set_contrast_level)
self.imagesize = wx.StaticText(panel,
label='? x ?',
size=(150, 30),
style=txtstyle)
def lin(len=200, wid=2, style=wx.LI_HORIZONTAL):
return wx.StaticLine(panel, size=(len, wid), style=style)
irow = 0
sizer.Add(pvpanel, (irow, 0), (1, 3), labstyle)
irow += 1
sizer.Add(start_btn, (irow, 0), (1, 1), labstyle)
sizer.Add(stop_btn, (irow, 1), (1, 1), labstyle)
if self.config['general'].get('show_free_run', False):
free_btn = wx.Button(panel, label='Free Run', size=wsize)
free_btn.Bind(wx.EVT_BUTTON, partial(self.onButton, key='free'))
irow += 1
sizer.Add(free_btn, (irow, 0), (1, 2), labstyle)
irow += 1
sizer.Add(lin(200, wid=4), (irow, 0), (1, 3), labstyle)
irow += 1
sizer.Add(self.imagesize, (irow, 0), (1, 3), labstyle)
if self.colormode.startswith('mono'):
self.cmap_choice = wx.Choice(panel,
size=(80, -1),
choices=self.config['colormaps'])
self.cmap_choice.SetSelection(0)
self.cmap_choice.Bind(wx.EVT_CHOICE, self.onColorMap)
self.cmap_reverse = wx.CheckBox(panel,
label='Reverse',
size=(60, -1))
self.cmap_reverse.Bind(wx.EVT_CHECKBOX, self.onColorMap)
irow += 1
sizer.Add(wx.StaticText(panel, label='Color Map: '), (irow, 0),
(1, 1), labstyle)
sizer.Add(self.cmap_choice, (irow, 1), (1, 1), labstyle)
sizer.Add(self.cmap_reverse, (irow, 2), (1, 1), labstyle)
irow += 1
sizer.Add(self.contrast.label, (irow, 0), (1, 1), labstyle)
sizer.Add(self.contrast.choice, (irow, 1), (1, 1), labstyle)
if self.config['general']['show_1dintegration']:
self.show1d_btn = wx.Button(panel,
label='Show 1D Integration',
size=(200, -1))
self.show1d_btn.Bind(wx.EVT_BUTTON, self.onShowIntegration)
self.show1d_btn.Disable()
irow += 1
sizer.Add(self.show1d_btn, (irow, 0), (1, 2), labstyle)
if self.config['general']['show_thumbnail']:
t_size = self.config['general'].get('thumbnail_size', 100)
self.thumbnail = ThumbNailImagePanel(panel,
imgsize=t_size,
size=(350, 350),
motion_writer=partial(
self.write, panel=0))
label = wx.StaticText(panel,
label='Thumbnail size (pixels): ',
size=(200, -1),
style=txtstyle)
self.thumbsize = FloatSpin(panel,
value=100,
min_val=10,
increment=5,
action=self.onThumbSize,
size=(150, -1),
style=txtstyle)
irow += 1
sizer.Add(label, (irow, 0), (1, 1), labstyle)
sizer.Add(self.thumbsize, (irow, 1), (1, 1), labstyle)
irow += 1
sizer.Add(self.thumbnail, (irow, 0), (1, 2), labstyle)
panel.SetSizer(sizer)
sizer.Fit(panel)
# image panel
self.image = ADMonoImagePanel(
self,
prefix=self.prefix,
rot90=self.config['general']['default_rotation'],
size=(750, 750),
writer=partial(self.write, panel=1),
thumbnail=self.thumbnail,
motion_writer=partial(self.write, panel=2))
mainsizer = wx.BoxSizer(wx.HORIZONTAL)
mainsizer.Add(panel, 0, wx.LEFT | wx.GROW | wx.ALL)
mainsizer.Add(self.image, 1, wx.CENTER | wx.GROW | wx.ALL)
self.SetSizer(mainsizer)
mainsizer.Fit(self)
self.SetAutoLayout(True)
iconfile = self.config['general'].get('iconfile', None)
if iconfile is None or not os.path.exists(iconfile):
iconfile = DEFAULT_ICONFILE
try:
self.SetIcon(wx.Icon(iconfile, wx.BITMAP_TYPE_ICO))
except:
pass
self.connect_pvs()
def __init__(self, _larch=None, parent=None, xrd_file=None,
size=(725, 450), axissize=None, axisbg=None,
title='XRD Display', exit_callback=None,
output_title='XRD', **kws):
if size is None: size = (725, 450)
wx.Frame.__init__(self, parent=parent,
title=title, size=size, **kws)
self.marker_color = '#77BB99'
self.spectra_color = '#0000AA'
self.subframes = {}
self.data = None
self.title = title
self.plotframe = None
self.wids = {}
self.larch = _larch
if self.larch is None:
self.init_larch()
self._xrdgroup = self.larch.symtable.new_group('_xrds')
self.exit_callback = exit_callback
self.selected_elem = None
self.xrd = None
self.xrd2 = None
self.xdata = np.arange(2048)*0.015
self.ydata = np.ones(2048)*1.e-4
self.x2data = None
self.y2data = None
self.major_markers = []
self.minor_markers = []
self.hold_markers = []
self.xunit = 'q'
self.xlabel = None
self.hold_lines = None
self.saved_lines = None
self.x_for_zoom = None
self.xview_range = None
self.show_yaxis = True
self.xmarker_left = None
self.xmarker_right = None
self.highlight_xrayline = None
self.highlight_xrayline = None
self.cursor_markers = [None, None]
self.ylog_scale = False
self.SetTitle("%s: %s " % (self.main_title, title))
self._menus = []
self.createMainPanel()
self.createMenus()
self.SetFont(Font(9, serif=True))
self.statusbar = self.CreateStatusBar(4)
self.statusbar.SetStatusWidths([-5, -3, -3, -4])
statusbar_fields = ["XRD Display", " ", " ", " "]
for i in range(len(statusbar_fields)):
self.statusbar.SetStatusText(statusbar_fields[i], i)
if xrd_file is not None:
self.xrd = gsexrd_group(xrd_file, _larch=self.larch)
self._xrdgroup.xrd1 = self.xrd
self._xrdgroup.xrd2 = None
self.plot1Dxrd(self.xrd, show_xrd2=False)
def createMainPanel(self):
splitter = wx.SplitterWindow(self, style=wx.SP_LIVE_UPDATE)
splitter.SetMinimumPaneSize(250)
leftpanel = wx.Panel(splitter)
ltop = wx.Panel(leftpanel)
def Btn(msg, x, act):
b = Button(ltop, msg, size=(x, 30), action=act)
b.SetFont(Font(10))
return b
sel_none = Btn('Select None', 120, self.onSelNone)
sel_all = Btn('Select All', 120, self.onSelAll)
self.controller.filelist = FileCheckList(leftpanel,
# main=self,
select_action=self.ShowFile,
remove_action=self.RemoveFile)
self.controller.filelist.SetBackgroundColour(wx.Colour(255, 255, 255))
tsizer = wx.BoxSizer(wx.HORIZONTAL)
tsizer.Add(sel_all, 1, LCEN|wx.GROW, 1)
tsizer.Add(sel_none, 1, LCEN|wx.GROW, 1)
pack(ltop, tsizer)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(ltop, 0, LCEN|wx.GROW, 1)
sizer.Add(self.controller.filelist, 1, LCEN|wx.GROW|wx.ALL, 1)
pack(leftpanel, sizer)
# right hand side
panel = wx.Panel(splitter)
sizer = wx.BoxSizer(wx.VERTICAL)
self.title = SimpleText(panel, 'initializing...', size=(300, -1))
self.title.SetFont(Font(10))
ir = 0
sizer.Add(self.title, 0, LCEN|wx.GROW|wx.ALL, 1)
self.nb = flat_nb.FlatNotebook(panel, -1, agwStyle=FNB_STYLE)
self.nb.SetTabAreaColour(wx.Colour(250,250,250))
self.nb.SetActiveTabColour(wx.Colour(254,254,195))
self.nb.SetNonActiveTabTextColour(wx.Colour(10,10,128))
self.nb.SetActiveTabTextColour(wx.Colour(128,0,0))
self.nb_panels = []
for name, creator in NB_PANELS:
_panel = creator(parent=self, controller=self.controller)
self.nb.AddPage(_panel," %s " % name, True)
self.nb_panels.append(_panel)
sizer.Add(self.nb, 1, LCEN|wx.EXPAND, 2)
self.nb.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED, self.onNBChanged)
self.nb.SetSelection(0)
pack(panel, sizer)
splitter.SplitVertically(leftpanel, panel, 1)
wx.CallAfter(self.init_larch)
def __init__(self):
wx.Frame.__init__(self,
None,
-1,
'wxutil demo',
style=wx.DEFAULT_FRAME_STYLE | wx.RESIZE_BORDER
| wx.TAB_TRAVERSAL)
self.SetTitle('wxutil demo')
self.SetFont(Font(11))
self.set_menu()
self.statusbar = self.CreateStatusBar(2, 1)
self.statusbar.SetStatusWidths([-2, -1])
statusbar_fields = ['Initializing....', ' ']
for i in range(len(statusbar_fields)):
self.statusbar.SetStatusText(statusbar_fields[i], i)
self.Bind(wx.EVT_CLOSE, self.onExit)
panel = GridPanel(self, nrows=8, ncols=4)
tctrl_name = TextCtrl(panel,
value='',
action=self.onName,
size=(250, -1))
lctrl_addr = LabeledTextCtrl(self,
value='<>',
action=self.onAddr,
labeltext=' Address: ',
size=(250, -1))
lab3 = HyperText(panel,
' FloatCtrl: ',
size=(100, -1),
action=self.onHyperText)
val3 = FloatCtrl(panel,
'3',
action=self.onFloat1,
precision=2,
minval=0,
maxval=1000,
size=(250, -1))
lab4 = HyperText(panel,
' FloatSpin: ',
size=(100, -1),
action=self.onHyperText)
val4 = FloatSpin(panel,
'12.2',
action=self.onFloatSpin,
digits=2,
increment=0.1,
size=(250, -1))
labx = HyperText(panel,
' NumericCombo: ',
size=(100, -1),
action=self.onHyperText)
steps = make_steps(prec=1, tmin=0, tmax=100)
valx = NumericCombo(panel, steps, precision=1)
self.choice1 = Choice(panel, size=(200, -1), action=self.onChoice)
self.choice1.SetChoices(['Apple', 'Banana', 'Cherry'])
yesno = YesNo(panel)
check1 = Check(panel, label='enable? ', action=self.onCheck)
btn1 = Button(panel,
label='Start',
size=(100, -1),
action=self.onStart)
pinbtn = BitmapButton(panel,
get_icon('pin'),
size=(50, -1),
action=partial(self.onBMButton, opt='pin1'),
tooltip='use last point selected from plot')
togbtn = ToggleButton(panel,
'Press Me',
action=self.onToggleButton,
size=(100, -1),
tooltip='do it, do it now, you will like it')
browse_btn = Button(panel,
'Open File',
action=self.onFileOpen,
size=(150, -1))
okcancel = OkCancel(panel, onOK=self.onOK, onCancel=self.onCancel)
ptable_btn = Button(panel,
'Show Periodic Table',
action=self.onPTable,
size=(175, -1))
edlist_btn = Button(panel,
'Show Editable Listbox',
action=self.onEdList,
size=(175, -1))
filelist_btn = Button(panel,
'Show File CheckList',
action=self.onFileList,
size=(175, -1))
panel.AddText(' Name: ', style=LEFT)
panel.Add(tctrl_name, dcol=2)
panel.Add(lctrl_addr.label, newrow=True)
panel.Add(lctrl_addr, dcol=2)
panel.Add(lab3, newrow=True)
panel.Add(val3, dcol=3)
panel.Add(lab4, newrow=True)
panel.Add(val4, dcol=3)
panel.Add(labx, newrow=True)
panel.Add(valx, dcol=3)
panel.AddText(' Choice : ', newrow=True)
panel.Add(check1)
panel.Add(self.choice1)
panel.AddText(' Yes or No: ', newrow=True)
panel.Add(yesno)
panel.Add(HLine(panel, size=(500, -1)), dcol=3, newrow=True)
panel.Add(btn1, newrow=True)
panel.Add(pinbtn)
panel.Add(togbtn)
panel.Add(browse_btn, newrow=True)
panel.Add(ptable_btn)
panel.Add(edlist_btn, newrow=True)
panel.Add(filelist_btn)
panel.Add(okcancel, newrow=True)
panel.pack()
self.timer = wx.Timer(self)
self.last_time = time.time()
self.Bind(wx.EVT_TIMER, self.onTimer, self.timer)
fsizer = wx.BoxSizer(wx.VERTICAL)
fsizer.Add(panel, 0, LEFT | wx.EXPAND)
wx.CallAfter(self.init_timer)
psize = panel.GetBestSize()
self.SetSize((psize[0] + 5, psize[1] + 25))
pack(self, fsizer)
self.Refresh()
def __init__(self,
parent,
group=None,
last_array_sel=None,
read_ok_cb=None,
edit_groupname=True):
self.parent = parent
self.dgroup = group
if not hasattr(self.dgroup, 'is_xas'):
try:
self.dgroup.is_xas = 'energ' in self.dgroup.array_labels[
0].lower()
except:
self.dgroup.is_xas = False
self.read_ok_cb = read_ok_cb
self.array_sel = {
'xpop': '',
'xarr': None,
'ypop': '',
'yop': '/',
'yarr1': None,
'yarr2': None,
'use_deriv': False
}
if last_array_sel is not None:
self.array_sel.update(last_array_sel)
if self.array_sel['yarr2'] is None and 'i0' in self.dgroup.array_labels:
self.array_sel['yarr2'] = 'i0'
if self.array_sel['yarr1'] is None:
if 'itrans' in self.dgroup.array_labels:
self.array_sel['yarr1'] = 'itrans'
elif 'i1' in self.dgroup.array_labels:
self.array_sel['yarr1'] = 'i1'
message = "Build Arrys from Data Columns for %s" % self.dgroup.filename
wx.Frame.__init__(self,
None,
-1,
'Build Arrays from Data Columns for %s' %
self.dgroup.filename,
style=FRAMESTYLE)
self.SetFont(Font(10))
panel = scrolled.ScrolledPanel(self)
self.SetMinSize((600, 600))
self.colors = GUIColors()
self.plotframe = None
# title row
title = SimpleText(panel,
message,
font=Font(13),
colour=self.colors.title,
style=LCEN)
opts = dict(action=self.onColumnChoice, size=(120, -1))
arr_labels = self.dgroup.array_labels
yarr_labels = arr_labels + ['1.0', '0.0', '']
xarr_labels = arr_labels + ['<index>']
self.xarr = Choice(panel, choices=xarr_labels, **opts)
self.yarr1 = Choice(panel, choices=arr_labels, **opts)
self.yarr2 = Choice(panel, choices=yarr_labels, **opts)
opts['size'] = (90, -1)
self.xpop = Choice(panel, choices=XPRE_OPS, **opts)
self.ypop = Choice(panel, choices=YPRE_OPS, **opts)
opts['size'] = (50, -1)
self.yop = Choice(panel, choices=ARR_OPS, **opts)
ylab = SimpleText(panel, 'Y = ')
xlab = SimpleText(panel, 'X = ')
self.xsuf = SimpleText(panel, '')
self.ysuf = SimpleText(panel, '')
self.xpop.SetStringSelection(self.array_sel['xpop'])
self.ypop.SetStringSelection(self.array_sel['ypop'])
self.yop.SetStringSelection(self.array_sel['yop'])
if '(' in self.array_sel['ypop']:
self.ysuf.SetLabel(')')
ixsel, iysel, iy2sel = 0, 1, len(yarr_labels) - 1
if self.array_sel['xarr'] in xarr_labels:
ixsel = xarr_labels.index(self.array_sel['xarr'])
if self.array_sel['yarr1'] in arr_labels:
iysel = arr_labels.index(self.array_sel['yarr1'])
if self.array_sel['yarr2'] in yarr_labels:
iy2sel = yarr_labels.index(self.array_sel['yarr2'])
self.xarr.SetSelection(ixsel)
self.yarr1.SetSelection(iysel)
self.yarr2.SetSelection(iy2sel)
opts['size'] = (150, -1)
self.use_deriv = Check(panel,
label='use derivative',
default=self.array_sel['use_deriv'],
**opts)
self.is_xas = Check(panel,
label='use as XAS data',
default=self.dgroup.is_xas,
**opts)
bpanel = wx.Panel(panel)
bsizer = wx.BoxSizer(wx.HORIZONTAL)
bsizer.Add(Button(bpanel, 'Preview', action=self.onColumnChoice), 4)
bsizer.Add(Button(bpanel, 'OK', action=self.onOK), 4)
bsizer.Add(Button(bpanel, 'Cancel', action=self.onCancel), 4)
pack(bpanel, bsizer)
sizer = wx.GridBagSizer(4, 8)
sizer.Add(title, (0, 0), (1, 7), LCEN, 5)
ir = 1
sizer.Add(xlab, (ir, 0), (1, 1), CEN, 0)
sizer.Add(self.xpop, (ir, 1), (1, 1), CEN, 0)
sizer.Add(self.xarr, (ir, 2), (1, 1), CEN, 0)
sizer.Add(self.xsuf, (ir, 3), (1, 1), CEN, 0)
ir += 1
sizer.Add(ylab, (ir, 0), (1, 1), CEN, 0)
sizer.Add(self.ypop, (ir, 1), (1, 1), CEN, 0)
sizer.Add(self.yarr1, (ir, 2), (1, 1), CEN, 0)
sizer.Add(self.yop, (ir, 3), (1, 1), CEN, 0)
sizer.Add(self.yarr2, (ir, 4), (1, 1), CEN, 0)
sizer.Add(self.ysuf, (ir, 5), (1, 1), CEN, 0)
ir += 1
sizer.Add(self.use_deriv, (ir, 0), (1, 3), LCEN, 0)
ir += 1
sizer.Add(self.is_xas, (ir, 0), (1, 3), LCEN, 0)
self.wid_groupname = None
if edit_groupname:
wid_grouplab = SimpleText(panel, 'Use Group Name: ')
self.wid_groupname = wx.TextCtrl(panel,
value=self.dgroup._groupname,
size=(100, -1))
ir += 1
sizer.Add(wid_grouplab, (ir, 0), (1, 2), LCEN, 3)
sizer.Add(self.wid_groupname, (ir, 2), (1, 3), LCEN, 3)
ir += 1
sizer.Add(bpanel, (ir, 0), (1, 5), LCEN, 3)
pack(panel, sizer)
ftext = wx.TextCtrl(self,
style=wx.TE_MULTILINE | wx.TE_READONLY,
size=(-1, 150))
try:
m = open(self.dgroup.filename, 'r')
text = m.read()
m.close()
except:
text = "The file '%s'\n was not found" % self.dgroup.filename
ftext.SetValue(text)
ftext.SetFont(Font(9))
mainsizer = wx.BoxSizer(wx.VERTICAL)
mainsizer.Add(panel, 0, wx.GROW | wx.ALL, 2)
mainsizer.Add(ftext, 1, LCEN | wx.GROW, 2)
pack(self, mainsizer)
self.Show()
self.Raise()
def __init__(self, parent, group, on_ok=None):
self.parent = parent
self.group = group
self.on_ok = on_ok
wx.Frame.__init__(self,
None,
-1,
'Add Selected Columns',
style=wx.DEFAULT_FRAME_STYLE | wx.TAB_TRAVERSAL)
self.SetFont(Font(10))
sizer = wx.GridBagSizer(2, 2)
panel = scrolled.ScrolledPanel(self)
self.SetMinSize((550, 550))
self.wids = {}
lab_aname = SimpleText(panel, label=' Save Array Name:')
lab_range = SimpleText(panel, label=' Use column index:')
lab_regex = SimpleText(panel, label=' Use column label:')
wids = self.wids = {}
wids['arrayname'] = wx.TextCtrl(panel, value='sum', size=(175, -1))
wids['tc_nums'] = wx.TextCtrl(panel, value='1,3-10', size=(175, -1))
wids['tc_regex'] = wx.TextCtrl(panel, value='*fe*', size=(175, -1))
savebtn = Button(panel, 'Save', action=self.onOK)
plotbtn = Button(panel, 'Plot Sum', action=self.onPlot)
sel_nums = Button(panel, 'Select by Index', action=self.onSelColumns)
sel_re = Button(panel, 'Select by Pattern', action=self.onSelRegex)
sizer.Add(lab_aname, (0, 0), (1, 2), LEFT, 3)
sizer.Add(wids['arrayname'], (0, 2), (1, 1), LEFT, 3)
sizer.Add(plotbtn, (0, 3), (1, 1), LEFT, 3)
sizer.Add(savebtn, (0, 4), (1, 1), LEFT, 3)
sizer.Add(lab_range, (1, 0), (1, 2), LEFT, 3)
sizer.Add(wids['tc_nums'], (1, 2), (1, 1), LEFT, 3)
sizer.Add(sel_nums, (1, 3), (1, 2), LEFT, 3)
sizer.Add(lab_regex, (2, 0), (1, 2), LEFT, 3)
sizer.Add(wids['tc_regex'], (2, 2), (1, 1), LEFT, 3)
sizer.Add(sel_re, (2, 3), (1, 2), LEFT, 3)
sizer.Add(HLine(panel, size=(550, 2)), (3, 0), (1, 5), LEFT, 3)
ir = 4
cind = SimpleText(panel, label=' Index ')
csel = SimpleText(panel, label=' Select ')
cname = SimpleText(panel, label=' Array Name ')
sizer.Add(cind, (ir, 0), (1, 1), LEFT, 3)
sizer.Add(csel, (ir, 1), (1, 1), LEFT, 3)
sizer.Add(cname, (ir, 2), (1, 3), LEFT, 3)
for i, name in enumerate(group.array_labels):
ir += 1
cind = SimpleText(panel, label=' %i ' % (i + 1))
cname = SimpleText(panel, label=' %s ' % name)
csel = Check(panel, label='', default=False)
self.wids["col_%d" % i] = csel
sizer.Add(cind, (ir, 0), (1, 1), LEFT, 3)
sizer.Add(csel, (ir, 1), (1, 1), LEFT, 3)
sizer.Add(cname, (ir, 2), (1, 3), LEFT, 3)
pack(panel, sizer)
panel.SetupScrolling()
mainsizer = wx.BoxSizer(wx.VERTICAL)
mainsizer.Add(panel, 1, wx.GROW | wx.ALL, 1)
pack(self, mainsizer)
self.Show()
self.Raise()
def build_display(self):
self.SetFont(Font(10))
titleopts = dict(font=Font(11), colour='#AA0000')
gopts = dict(ncols=4, nrows=4, pad=2, itemstyle=LCEN)
xas = self.xaspanel = GridPanel(self, **gopts)
gen = self.genpanel = GridPanel(self, **gopts)
self.btns = {}
#gen
opts = dict(action=self.UpdatePlot, size=(65, -1), gformat=True)
self.xshift = FloatCtrl(gen, value=0.0, **opts)
self.xscale = FloatCtrl(gen, value=1.0, **opts)
self.yshift = FloatCtrl(gen, value=0.0, **opts)
self.yscale = FloatCtrl(gen, value=1.0, **opts)
self.btns['xshift'] = BitmapButton(
gen,
get_icon('plus'),
action=partial(self.on_selpoint, opt='xshift'),
tooltip='use last point selected from plot')
self.btns['yshift'] = BitmapButton(
gen,
get_icon('plus'),
action=partial(self.on_selpoint, opt='yshift'),
tooltip='use last point selected from plot')
opts = dict(action=self.onSmoothChoice, size=(30, -1))
sm_row1 = wx.Panel(gen)
sm_row2 = wx.Panel(gen)
sm_siz1 = wx.BoxSizer(wx.HORIZONTAL)
sm_siz2 = wx.BoxSizer(wx.HORIZONTAL)
self.smooth_c0 = FloatCtrl(sm_row1,
value=2,
precision=0,
minval=1,
**opts)
self.smooth_c1 = FloatCtrl(sm_row1,
value=1,
precision=0,
minval=1,
**opts)
self.smooth_msg = SimpleText(sm_row1,
label=' ',
size=(205, -1))
opts['size'] = (65, -1)
self.smooth_sig = FloatCtrl(sm_row2, value=1, gformat=True, **opts)
opts['size'] = (120, -1)
self.smooth_op = Choice(sm_row1, choices=SMOOTH_OPS, **opts)
self.smooth_op.SetSelection(0)
self.smooth_conv = Choice(sm_row2, choices=CONV_OPS, **opts)
self.smooth_c0.Disable()
self.smooth_c1.Disable()
self.smooth_sig.Disable()
self.smooth_conv.SetSelection(0)
self.smooth_conv.Disable()
sm_siz1.Add(self.smooth_op, 0, LCEN, 1)
sm_siz1.Add(SimpleText(sm_row1, ' n= '), 0, LCEN, 1)
sm_siz1.Add(self.smooth_c0, 0, LCEN, 1)
sm_siz1.Add(SimpleText(sm_row1, ' order= '), 0, LCEN, 1)
sm_siz1.Add(self.smooth_c1, 0, LCEN, 1)
sm_siz1.Add(self.smooth_msg, 0, LCEN, 1)
sm_siz2.Add(SimpleText(sm_row2, ' form= '), 0, LCEN, 1)
sm_siz2.Add(self.smooth_conv, 0, LCEN, 1)
sm_siz2.Add(SimpleText(sm_row2, ' sigma= '), 0, LCEN, 1)
sm_siz2.Add(self.smooth_sig, 0, LCEN, 1)
pack(sm_row1, sm_siz1)
pack(sm_row2, sm_siz2)
gen.Add(SimpleText(gen, ' General Data Processing', **titleopts),
dcol=8)
gen.Add(SimpleText(gen, ' X shift:'), newrow=True)
gen.Add(self.btns['xshift'])
gen.Add(self.xshift, dcol=2)
gen.Add(SimpleText(gen, ' X scale:'))
gen.Add(self.xscale, dcol=2)
gen.Add(SimpleText(gen, ' Y shift:'), newrow=True)
gen.Add(self.btns['yshift'])
gen.Add(self.yshift, dcol=2)
gen.Add(SimpleText(gen, ' Y scale:'))
gen.Add(self.yscale, dcol=2)
gen.Add(SimpleText(gen, ' Smoothing:'), newrow=True)
gen.Add(sm_row1, dcol=8)
gen.Add(sm_row2, icol=1, dcol=7, newrow=True)
gen.pack()
#xas
opts = {'action': self.UpdatePlot}
e0opts_panel = wx.Panel(xas)
self.xas_autoe0 = Check(e0opts_panel,
default=True,
label='auto?',
**opts)
self.xas_showe0 = Check(e0opts_panel,
default=True,
label='show?',
**opts)
sx = wx.BoxSizer(wx.HORIZONTAL)
sx.Add(self.xas_autoe0, 0, LCEN, 4)
sx.Add(self.xas_showe0, 0, LCEN, 4)
pack(e0opts_panel, sx)
self.xas_autostep = Check(xas, default=True, label='auto?', **opts)
opts['size'] = (250, -1)
self.xas_op = Choice(xas, choices=XASOPChoices, **opts)
for name in ('e0', 'pre1', 'pre2', 'nor1', 'nor2'):
bb = BitmapButton(xas,
get_icon('plus'),
action=partial(self.on_selpoint, opt=name),
tooltip='use last point selected from plot')
self.btns[name] = bb
opts = {'size': (65, -1), 'gformat': True, 'action': self.UpdatePlot}
self.xas_e0 = FloatCtrl(xas, value=0, **opts)
self.xas_step = FloatCtrl(xas, value=0, **opts)
opts['precision'] = 1
self.xas_pre1 = FloatCtrl(xas, value=-200, **opts)
self.xas_pre2 = FloatCtrl(xas, value=-30, **opts)
self.xas_nor1 = FloatCtrl(xas, value=50, **opts)
self.xas_nor2 = FloatCtrl(xas, value=-50, **opts)
opts = {
'size': (50, -1),
'choices': ('0', '1', '2', '3'),
'action': self.UpdatePlot
}
self.xas_vict = Choice(xas, **opts)
self.xas_nnor = Choice(xas, **opts)
self.xas_vict.SetSelection(1)
self.xas_nnor.SetSelection(2)
def CopyBtn(name):
return Button(xas,
'Copy',
size=(50, 30),
action=partial(self.onCopyParam, name))
xas.Add(SimpleText(xas, ' XAS Data Processing', **titleopts), dcol=6)
xas.Add(SimpleText(xas, ' Copy to Selected Groups?'),
style=RCEN,
dcol=3)
xas.Add(SimpleText(xas, 'Arrays to Plot: '), newrow=True)
xas.Add(self.xas_op, dcol=6)
xas.Add((10, 10))
xas.Add(CopyBtn('xas_op'), style=RCEN)
xas.Add(SimpleText(xas, 'E0 : '), newrow=True)
xas.Add(self.btns['e0'])
xas.Add(self.xas_e0)
xas.Add(e0opts_panel, dcol=4)
xas.Add((10, 1))
xas.Add(CopyBtn('xas_e0'), style=RCEN)
xas.Add(SimpleText(xas, 'Edge Step: '), newrow=True)
xas.Add((10, 1))
xas.Add(self.xas_step)
xas.Add(self.xas_autostep, dcol=3)
xas.Add((10, 1))
xas.Add((10, 1))
xas.Add(CopyBtn('xas_step'), style=RCEN)
xas.Add(SimpleText(xas, 'Pre-edge range: '), newrow=True)
xas.Add(self.btns['pre1'])
xas.Add(self.xas_pre1)
xas.Add(SimpleText(xas, ':'))
xas.Add(self.btns['pre2'])
xas.Add(self.xas_pre2)
xas.Add(SimpleText(xas, 'Victoreen:'))
xas.Add(self.xas_vict)
xas.Add(CopyBtn('xas_pre'), style=RCEN)
xas.Add(SimpleText(xas, 'Normalization range: '), newrow=True)
xas.Add(self.btns['nor1'])
xas.Add(self.xas_nor1)
xas.Add(SimpleText(xas, ':'))
xas.Add(self.btns['nor2'])
xas.Add(self.xas_nor2)
xas.Add(SimpleText(xas, 'PolyOrder:'))
xas.Add(self.xas_nnor)
xas.Add(CopyBtn('xas_norm'), style=RCEN)
xas.pack()
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.AddMany([((10, 10), 0, LCEN, 0), (gen, 0, LCEN, 10),
((10, 10), 0, LCEN, 0),
(HLine(self, size=(550, 2)), 0, LCEN, 10),
((10, 10), 0, LCEN, 0), (xas, 1, LCEN | wx.GROW, 10)])
xas.Disable()
pack(self, sizer)
def __init__(self,
parent,
filename=None,
groupname=None,
last_array_sel=None,
read_ok_cb=None,
edit_groupname=True,
_larch=None):
self.parent = parent
self._larch = _larch
self.path = filename
group = self.initgroup = self.read_column_file(self.path)
self.subframes = {}
self.workgroup = Group(raw=group)
for attr in ('path', 'filename', 'groupname', 'datatype',
'array_labels'):
setattr(self.workgroup, attr, getattr(group, attr, None))
arr_labels = [l.lower() for l in self.initgroup.array_labels]
if self.workgroup.datatype is None:
self.workgroup.datatype = 'raw'
if ('energ' in arr_labels[0] or 'energ' in arr_labels[1]):
self.workgroup.datatype = 'xas'
self.read_ok_cb = read_ok_cb
self.array_sel = {
'xpop': '',
'xarr': None,
'ypop': '',
'yop': '/',
'yarr1': None,
'yarr2': None,
'use_deriv': False
}
if last_array_sel is not None:
self.array_sel.update(last_array_sel)
if self.array_sel['yarr2'] is None and 'i0' in arr_labels:
self.array_sel['yarr2'] = 'i0'
if self.array_sel['yarr1'] is None:
if 'itrans' in arr_labels:
self.array_sel['yarr1'] = 'itrans'
elif 'i1' in arr_labels:
self.array_sel['yarr1'] = 'i1'
message = "Data Columns for %s" % group.filename
wx.Frame.__init__(self,
None,
-1,
'Build Arrays from Data Columns for %s' %
group.filename,
style=FRAMESTYLE)
self.SetFont(Font(10))
panel = wx.Panel(self)
self.SetMinSize((600, 600))
self.colors = GUIColors()
# title row
title = SimpleText(panel,
message,
font=Font(13),
colour=self.colors.title,
style=LCEN)
opts = dict(action=self.onUpdate, size=(120, -1))
yarr_labels = self.yarr_labels = arr_labels + ['1.0', '0.0', '']
xarr_labels = self.xarr_labels = arr_labels + ['_index']
self.xarr = Choice(panel, choices=xarr_labels, **opts)
self.yarr1 = Choice(panel, choices=arr_labels, **opts)
self.yarr2 = Choice(panel, choices=yarr_labels, **opts)
self.yerr_arr = Choice(panel, choices=yarr_labels, **opts)
self.yerr_arr.Disable()
self.datatype = Choice(panel, choices=DATATYPES, **opts)
self.datatype.SetStringSelection(self.workgroup.datatype)
opts['size'] = (50, -1)
self.yop = Choice(panel, choices=ARR_OPS, **opts)
opts['size'] = (120, -1)
self.use_deriv = Check(panel,
label='use derivative',
default=self.array_sel['use_deriv'],
**opts)
self.xpop = Choice(panel, choices=XPRE_OPS, **opts)
self.ypop = Choice(panel, choices=YPRE_OPS, **opts)
opts['action'] = self.onYerrChoice
self.yerr_op = Choice(panel, choices=YERR_OPS, **opts)
self.yerr_op.SetSelection(0)
self.yerr_const = FloatCtrl(panel, value=1, precision=4, size=(90, -1))
ylab = SimpleText(panel, 'Y = ')
xlab = SimpleText(panel, 'X = ')
yerr_lab = SimpleText(panel, 'Yerror = ')
self.xsuf = SimpleText(panel, '')
self.ysuf = SimpleText(panel, '')
self.xpop.SetStringSelection(self.array_sel['xpop'])
self.ypop.SetStringSelection(self.array_sel['ypop'])
self.yop.SetStringSelection(self.array_sel['yop'])
if '(' in self.array_sel['ypop']:
self.ysuf.SetLabel(')')
ixsel, iysel, iy2sel = 0, 1, len(yarr_labels) - 1
if self.array_sel['xarr'] in xarr_labels:
ixsel = xarr_labels.index(self.array_sel['xarr'])
if self.array_sel['yarr1'] in arr_labels:
iysel = arr_labels.index(self.array_sel['yarr1'])
if self.array_sel['yarr2'] in yarr_labels:
iy2sel = yarr_labels.index(self.array_sel['yarr2'])
self.xarr.SetSelection(ixsel)
self.yarr1.SetSelection(iysel)
self.yarr2.SetSelection(iy2sel)
bpanel = wx.Panel(panel)
bsizer = wx.BoxSizer(wx.HORIZONTAL)
_ok = Button(bpanel, 'OK', action=self.onOK)
_cancel = Button(bpanel, 'Cancel', action=self.onCancel)
_edit = Button(bpanel, 'Edit Array Names', action=self.onEditNames)
bsizer.Add(_ok)
bsizer.Add(_cancel)
bsizer.Add(_edit)
_ok.SetDefault()
pack(bpanel, bsizer)
sizer = wx.GridBagSizer(4, 8)
sizer.Add(title, (0, 0), (1, 7), LCEN, 5)
ir = 1
sizer.Add(xlab, (ir, 0), (1, 1), LCEN, 0)
sizer.Add(self.xpop, (ir, 1), (1, 1), CEN, 0)
sizer.Add(self.xarr, (ir, 2), (1, 1), CEN, 0)
sizer.Add(self.xsuf, (ir, 3), (1, 1), CEN, 0)
ir += 1
sizer.Add(ylab, (ir, 0), (1, 1), LCEN, 0)
sizer.Add(self.ypop, (ir, 1), (1, 1), CEN, 0)
sizer.Add(self.yarr1, (ir, 2), (1, 1), CEN, 0)
sizer.Add(self.yop, (ir, 3), (1, 1), CEN, 0)
sizer.Add(self.yarr2, (ir, 4), (1, 1), CEN, 0)
sizer.Add(self.ysuf, (ir, 5), (1, 1), CEN, 0)
sizer.Add(self.use_deriv, (ir, 6), (1, 1), LCEN, 0)
ir += 1
sizer.Add(yerr_lab, (ir, 0), (1, 1), LCEN, 0)
sizer.Add(self.yerr_op, (ir, 1), (1, 1), CEN, 0)
sizer.Add(self.yerr_arr, (ir, 2), (1, 1), CEN, 0)
sizer.Add(SimpleText(panel, 'Value:'), (ir, 3), (1, 1), CEN, 0)
sizer.Add(self.yerr_const, (ir, 4), (1, 2), CEN, 0)
ir += 1
sizer.Add(SimpleText(panel, 'Data Type:'), (ir, 0), (1, 1), LCEN, 0)
sizer.Add(self.datatype, (ir, 1), (1, 2), LCEN, 0)
ir += 1
self.wid_groupname = wx.TextCtrl(panel,
value=group.groupname,
size=(240, -1))
if not edit_groupname:
self.wid_groupname.Disable()
sizer.Add(SimpleText(panel, 'Group Name:'), (ir, 0), (1, 1), LCEN, 0)
sizer.Add(self.wid_groupname, (ir, 1), (1, 2), LCEN, 0)
ir += 1
sizer.Add(bpanel, (ir, 0), (1, 5), LCEN, 3)
pack(panel, sizer)
self.nb = fnb.FlatNotebook(self, -1, agwStyle=FNB_STYLE)
self.nb.SetTabAreaColour(wx.Colour(248, 248, 240))
self.nb.SetActiveTabColour(wx.Colour(254, 254, 195))
self.nb.SetNonActiveTabTextColour(wx.Colour(40, 40, 180))
self.nb.SetActiveTabTextColour(wx.Colour(80, 0, 0))
self.plotpanel = PlotPanel(self, messenger=self.plot_messages)
textpanel = wx.Panel(self)
ftext = wx.TextCtrl(textpanel,
style=wx.TE_MULTILINE | wx.TE_READONLY,
size=(400, 250))
ftext.SetValue(group.text)
ftext.SetFont(Font(10))
textsizer = wx.BoxSizer(wx.VERTICAL)
textsizer.Add(ftext, 1, LCEN | wx.GROW, 1)
pack(textpanel, textsizer)
self.nb.AddPage(textpanel, ' Text of Data File ', True)
self.nb.AddPage(self.plotpanel, ' Plot of Selected Arrays ', True)
mainsizer = wx.BoxSizer(wx.VERTICAL)
mainsizer.Add(panel, 0, wx.GROW | wx.ALL, 2)
mainsizer.Add(self.nb, 1, LCEN | wx.GROW, 2)
pack(self, mainsizer)
self.statusbar = self.CreateStatusBar(2, 0)
self.statusbar.SetStatusWidths([-1, -1])
statusbar_fields = [group.filename, ""]
for i in range(len(statusbar_fields)):
self.statusbar.SetStatusText(statusbar_fields[i], i)
self.Show()
self.Raise()
self.onUpdate(self)
def createMainPanel(self):
splitter = wx.SplitterWindow(self, style=wx.SP_LIVE_UPDATE)
splitter.SetMinimumPaneSize(250)
leftpanel = wx.Panel(splitter)
ltop = wx.Panel(leftpanel)
def Btn(msg, x, act):
b = Button(ltop, msg, size=(x, 30), action=act)
b.SetFont(Font(10))
return b
plot_one = Btn('Plot One', 120, self.onPlotOne)
plot_sel = Btn('Plot Selected', 120, self.onPlotSel)
sel_none = Btn('Select None', 120, self.onSelNone)
sel_all = Btn('Select All', 120, self.onSelAll)
self.controller.filelist = FileCheckList(leftpanel,
main=self,
select_action=self.ShowFile)
self.controller.filelist.SetBackgroundColour(wx.Colour(255, 255, 255))
tsizer = wx.GridBagSizer(1, 1)
tsizer.Add(plot_one, (0, 0), (1, 1), LCEN, 2)
tsizer.Add(plot_sel, (0, 1), (1, 1), LCEN, 2)
tsizer.Add(sel_all, (1, 0), (1, 1), LCEN, 2)
tsizer.Add(sel_none, (1, 1), (1, 1), LCEN, 2)
pack(ltop, tsizer)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(ltop, 0, LCEN | wx.GROW, 1)
sizer.Add(self.controller.filelist, 1, LCEN | wx.GROW | wx.ALL, 1)
pack(leftpanel, sizer)
# right hand side
panel = wx.Panel(splitter)
sizer = wx.BoxSizer(wx.VERTICAL)
self.title = SimpleText(panel, 'initializing...', size=(300, -1))
self.title.SetFont(Font(10))
ir = 0
sizer.Add(self.title, 0, LCEN | wx.GROW | wx.ALL, 1)
self.nb = flat_nb.FlatNotebook(panel, -1, agwStyle=FNB_STYLE)
self.nb.SetTabAreaColour(wx.Colour(250, 250, 250))
self.nb.SetActiveTabColour(wx.Colour(254, 254, 195))
self.nb.SetNonActiveTabTextColour(wx.Colour(10, 10, 128))
self.nb.SetActiveTabTextColour(wx.Colour(128, 0, 0))
panel_opts = dict(parent=self.nb, controller=self.controller)
self.proc_panel = ProcessPanel(**panel_opts)
self.fit_panel = XYFitPanel(**panel_opts)
self.nb.AddPage(self.proc_panel, ' Data Processing ', True)
self.nb.AddPage(self.fit_panel, ' Curve Fitting ', True)
sizer.Add(self.nb, 1, LCEN | wx.EXPAND, 2)
self.nb.SetSelection(0)
pack(panel, sizer)
splitter.SplitVertically(leftpanel, panel, 1)
wx.CallAfter(self.init_larch)
def createControlPanel(self):
ctrlpanel = wx.Panel(self, name='Ctrl Panel')
ptable = PeriodicTablePanel(ctrlpanel, onselect=self.onShowLines,
tooltip_msg='Select Element for KLM Lines')
self.wids['ptable'] = ptable
labstyle = wx.ALIGN_LEFT|wx.ALIGN_BOTTOM|wx.EXPAND
ctrlstyle = wx.ALIGN_LEFT|wx.ALIGN_BOTTOM
txtstyle=wx.ALIGN_LEFT|wx.ST_NO_AUTORESIZE|wx.TE_PROCESS_ENTER
Font9 = Font(9)
Font10 = Font(10)
Font11 = Font(11)
#
arrowpanel = wx.Panel(ctrlpanel)
ssizer = wx.BoxSizer(wx.HORIZONTAL)
for wname, dname in (('uparrow', 'up'),
('leftarrow', 'left'),
('rightarrow', 'right'),
('downarrow', 'down')):
self.wids[wname] = wx.BitmapButton(arrowpanel, -1,
get_icon(wname),
size=(25, 25),
style=wx.NO_BORDER)
self.wids[wname].Bind(wx.EVT_BUTTON,
partial(ptable.onKey, name=dname))
ssizer.Add(self.wids[wname], 0, wx.EXPAND|wx.ALL)
self.wids['kseries'] = Check(arrowpanel, ' K ', action=self.onKLM)
self.wids['lseries'] = Check(arrowpanel, ' L ', action=self.onKLM)
self.wids['mseries'] = Check(arrowpanel, ' M ', action=self.onKLM)
self.wids['holdbtn'] = wx.ToggleButton(arrowpanel, -1, 'Hold ', size=(65, 25))
self.wids['holdbtn'].Bind(wx.EVT_TOGGLEBUTTON, self.onToggleHold)
ssizer.Add(txt(arrowpanel, ' '), 0, wx.EXPAND|wx.ALL, 0)
ssizer.Add(self.wids['holdbtn'], 0, wx.EXPAND|wx.ALL, 2)
ssizer.Add(self.wids['kseries'], 0, wx.EXPAND|wx.ALL, 0)
ssizer.Add(self.wids['lseries'], 0, wx.EXPAND|wx.ALL, 0)
ssizer.Add(self.wids['mseries'], 0, wx.EXPAND|wx.ALL, 0)
pack(arrowpanel, ssizer)
# roi section...
rsizer = wx.GridBagSizer(4, 6)
roipanel = wx.Panel(ctrlpanel, name='ROI Panel')
self.wids['roilist'] = wx.ListBox(roipanel, size=(135, 120))
self.wids['roilist'].Bind(wx.EVT_LISTBOX, self.onROI)
self.wids['roilist'].SetMinSize((135, 120))
self.wids['roiname'] = wx.TextCtrl(roipanel, -1, '', size=(150, -1))
#
roibtns= wx.Panel(roipanel, name='ROIButtons')
zsizer = wx.BoxSizer(wx.HORIZONTAL)
z1 = Button(roibtns, 'Add', size=(70, 30), action=self.onNewROI)
z2 = Button(roibtns, 'Delete', size=(70, 30), action=self.onConfirmDelROI)
z3 = Button(roibtns, 'Rename', size=(70, 30), action=self.onRenameROI)
zsizer.Add(z1, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(z2, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(z3, 0, wx.EXPAND|wx.ALL, 0)
pack(roibtns, zsizer)
rt1 = txt(roipanel, ' Channels:', size=75, font=Font9)
rt2 = txt(roipanel, ' Energy:', size=75, font=Font9)
rt3 = txt(roipanel, ' Cen/Wid:', size=75, font=Font9)
m = ''
self.wids['roi_msg1'] = txt(roipanel, m, size=135, font=Font9)
self.wids['roi_msg2'] = txt(roipanel, m, size=135, font=Font9)
self.wids['roi_msg3'] = txt(roipanel, m, size=135, font=Font9)
rsizer.Add(txt(roipanel, ' Regions of Interest:', size=125, font=Font11),
(0, 0), (1, 3), labstyle)
rsizer.Add(self.wids['roiname'], (1, 0), (1, 3), labstyle)
rsizer.Add(roibtns, (2, 0), (1, 3), labstyle)
rsizer.Add(rt1, (3, 0), (1, 1), LEFT)
rsizer.Add(rt2, (4, 0), (1, 1), LEFT)
rsizer.Add(rt3, (5, 0), (1, 1), LEFT)
rsizer.Add(self.wids['roi_msg1'], (3, 1), (1, 2), labstyle)
rsizer.Add(self.wids['roi_msg2'], (4, 1), (1, 2), labstyle)
rsizer.Add(self.wids['roi_msg3'], (5, 1), (1, 2), labstyle)
rsizer.Add(self.wids['roilist'], (0, 3), (6, 1),
wx.EXPAND|wx.ALL|wx.ALIGN_RIGHT)
rsizer.SetHGap(1)
pack(roipanel, rsizer)
# end roi section
# y scale
yscalepanel = wx.Panel(ctrlpanel, name='YScalePanel')
ysizer = wx.BoxSizer(wx.HORIZONTAL)
ytitle = txt(yscalepanel, ' Y Axis:', font=Font10, size=80)
yspace = txt(yscalepanel, ' ', font=Font10, size=20)
ylog = Choice(yscalepanel, size=(80, 30), choices=['log', 'linear'],
action=self.onLogLinear)
yaxis = Check(yscalepanel, ' Show Y Scale ', action=self.onYAxis,
default=False)
self.wids['show_yaxis'] = yaxis
ysizer.Add(ytitle, 0, wx.ALIGN_CENTER_VERTICAL|wx.ALL, 0)
ysizer.Add(ylog, 0, wx.EXPAND|wx.ALL, 0)
ysizer.Add(yspace, 0, wx.EXPAND|wx.ALL, 0)
ysizer.Add(yaxis, 0, wx.EXPAND|wx.ALL, 0)
pack(yscalepanel, ysizer)
# zoom buttons
zoompanel = wx.Panel(ctrlpanel, name='ZoomPanel')
zsizer = wx.BoxSizer(wx.HORIZONTAL)
z1 = Button(zoompanel, 'Zoom In', size=(80, 30), action=self.onZoomIn)
z2 = Button(zoompanel, 'Zoom Out', size=(80, 30), action=self.onZoomOut)
p1 = Button(zoompanel, 'Pan Lo', size=(75, 30), action=self.onPanLo)
p2 = Button(zoompanel, 'Pan Hi', size=(75, 30), action=self.onPanHi)
zsizer.Add(p1, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(p2, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(z1, 0, wx.EXPAND|wx.ALL, 0)
zsizer.Add(z2, 0, wx.EXPAND|wx.ALL, 0)
pack(zoompanel, zsizer)
pileup_opt = Check(ctrlpanel, ' Show Pileup Prediction ',
action=self.onPileupPrediction, default=False)
self.wids['pileup'] = pileup_opt
#
self.wids['xray_lines'] = None
if HAS_DV:
dvstyle = dv.DV_SINGLE|dv.DV_VERT_RULES|dv.DV_ROW_LINES
xlines = dv.DataViewListCtrl(ctrlpanel, style=dvstyle)
self.wids['xray_lines'] = xlines
xlines.AppendTextColumn('Line ', width=55)
xlines.AppendTextColumn('Energy(keV) ', width=110)
xlines.AppendTextColumn('Strength ', width=85)
xlines.AppendTextColumn('Levels ', width=75)
for col in (0, 1, 2, 3):
this = xlines.Columns[col]
this.Sortable = True
align = RIGHT
if col in (0, 3): align = wx.ALIGN_LEFT
this.Alignment = this.Renderer.Alignment = align
xlines.SetMinSize((300, 240))
xlines.Bind(dv.EVT_DATAVIEW_SELECTION_CHANGED,
self.onSelectXrayLine)
store = xlines.GetStore()
# main layout
# may have to adjust comparison....
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(roipanel, 0, labstyle)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(yscalepanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(zoompanel, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
sizer.Add(ptable, 0, wx.ALIGN_RIGHT|wx.EXPAND|wx.ALL, 4)
sizer.Add(arrowpanel, 0, labstyle)
sizer.Add(pileup_opt, 0, labstyle)
sizer.Add(lin(ctrlpanel, 195), 0, labstyle)
if self.wids['xray_lines'] is not None:
sizer.Add(xlines, 0, wx.ALIGN_CENTER|wx.GROW|wx.ALL|wx.EXPAND)
pack(ctrlpanel, sizer)
return ctrlpanel
def Btn(msg, x, act):
b = Button(ltop, msg, size=(x, 30), action=act)
b.SetFont(Font(10))
return b
def __init__(self,
parent=None,
prefix=None,
det_type='ME-4',
is_xsp3=False,
nmca=4,
title='Select Epics MCA Detector'):
if prefix is None: prefix = self.def_prefix
if det_type not in self.det_types:
det_type = self.det_types[0]
wx.Dialog.__init__(self, parent, wx.ID_ANY, title=title)
self.SetBackgroundColour((240, 240, 230))
self.SetFont(Font(9))
if parent is not None:
self.SetFont(parent.GetFont())
self.is_xsp3 = Check(self, size=(120, -1), default=False)
self.dettype = Choice(self, size=(120, -1), choices=self.det_types)
self.dettype.SetStringSelection(det_type)
self.prefix = wx.TextCtrl(self, -1, prefix, size=(120, -1))
self.nelem = FloatCtrl(self,
value=nmca,
precision=0,
minval=1,
size=(120, -1))
btnsizer = wx.StdDialogButtonSizer()
if wx.Platform != "__WXMSW__":
btn = wx.ContextHelpButton(self)
btnsizer.AddButton(btn)
btn = wx.Button(self, wx.ID_OK)
btn.SetHelpText("Use this detector")
btn.SetDefault()
btnsizer.AddButton(btn)
btn = wx.Button(self, wx.ID_CANCEL)
btnsizer.AddButton(btn)
btnsizer.Realize()
hline = wx.StaticLine(self, size=(225, 3), style=wx.LI_HORIZONTAL)
sty = LEFT | wx.ALIGN_CENTER_VERTICAL
sizer = wx.GridBagSizer(5, 2)
def txt(label):
return SimpleText(self, label, size=(120, -1), style=LEFT)
sizer.Add(txt(' Detector Type'), (0, 0), (1, 1), sty, 2)
sizer.Add(txt(' Uses Xspress3?'), (1, 0), (1, 1), sty, 2)
sizer.Add(txt(' Epics Prefix'), (2, 0), (1, 1), sty, 2)
sizer.Add(txt(' # Elements'), (3, 0), (1, 1), sty, 2)
sizer.Add(self.dettype, (0, 1), (1, 1), sty, 2)
sizer.Add(self.is_xsp3, (1, 1), (1, 1), sty, 2)
sizer.Add(self.prefix, (2, 1), (1, 1), sty, 2)
sizer.Add(self.nelem, (3, 1), (1, 1), sty, 2)
sizer.Add(hline, (4, 0), (1, 2), sty, 2)
sizer.Add(btnsizer, (5, 0), (1, 2), sty, 2)
self.SetSizer(sizer)
sizer.Fit(self)
def build(self):
sizer = wx.GridBagSizer(10, 5)
sizer.SetVGap(2)
sizer.SetHGap(2)
panel = scrolled.ScrolledPanel(self)
self.SetMinSize((600, 450))
self.colors = GUIColors()
# title row
self.wids = wids = {}
title = SimpleText(panel, 'Fit Results', font=Font(12),
colour=self.colors.title, style=LCEN)
wids['data_title'] = SimpleText(panel, '< > ', font=Font(12),
colour=self.colors.title, style=LCEN)
wids['hist_info'] = SimpleText(panel, ' ___ ', font=Font(12),
colour=self.colors.title, style=LCEN)
sizer.Add(title, (0, 0), (1, 2), LCEN)
sizer.Add(wids['data_title'], (0, 2), (1, 2), LCEN)
sizer.Add(wids['hist_info'], (0, 4), (1, 2), LCEN)
irow = 1
wids['model_desc'] = SimpleText(panel, '<Model>', font=Font(12),
size=(550, 75), style=LCEN)
sizer.Add(wids['model_desc'], (irow, 0), (1, 6), LCEN)
irow += 1
sizer.Add(HLine(panel, size=(400, 3)), (irow, 0), (1, 5), LCEN)
irow += 1
title = SimpleText(panel, '[[Fit Statistics]]', font=Font(12),
colour=self.colors.title, style=LCEN)
sizer.Add(title, (irow, 0), (1, 4), LCEN)
for label, attr in (('Fit method', 'method'),
('# Fit Evaluations', 'nfev'),
('# Data Points', 'ndata'),
('# Fit Variables', 'nvarys'),
('# Free Points', 'nfree'),
('Chi-square', 'chisqr'),
('Reduced Chi-square', 'redchi'),
('Akaike Info Criteria', 'aic'),
('Bayesian Info Criteria', 'bic')):
irow += 1
wids[attr] = SimpleText(panel, '?')
sizer.Add(SimpleText(panel, " %s = " % label), (irow, 0), (1, 1), LCEN)
sizer.Add(wids[attr], (irow, 1), (1, 1), LCEN)
irow += 1
sizer.Add(HLine(panel, size=(400, 3)), (irow, 0), (1, 5), LCEN)
irow += 1
title = SimpleText(panel, '[[Variables]]', font=Font(12),
colour=self.colors.title, style=LCEN)
sizer.Add(title, (irow, 0), (1, 1), LCEN)
self.wids['copy_params'] = Button(panel, 'Update Model with Best Fit Values',
size=(250, -1), action=self.onCopyParams)
sizer.Add(self.wids['copy_params'], (irow, 1), (1, 3), LCEN)
dvstyle = dv.DV_SINGLE|dv.DV_VERT_RULES|dv.DV_ROW_LINES
pview = self.wids['params'] = dv.DataViewListCtrl(panel, style=dvstyle)
self.wids['paramsdata'] = []
pview.AppendTextColumn('Parameter', width=150)
pview.AppendTextColumn('Best-Fit Value', width=100)
pview.AppendTextColumn('Standard Error', width=100)
pview.AppendTextColumn('Info ', width=275)
for col in (0, 1, 2, 3):
this = pview.Columns[col]
isort, align = True, wx.ALIGN_LEFT
if col in (1, 2):
isort, align = False, wx.ALIGN_RIGHT
this.Sortable = isort
this.Alignment = this.Renderer.Alignment = align
pview.SetMinSize((650, 200))
pview.Bind(dv.EVT_DATAVIEW_SELECTION_CHANGED, self.onSelectParameter)
irow += 1
sizer.Add(pview, (irow, 0), (1, 5), LCEN)
irow += 1
sizer.Add(HLine(panel, size=(400, 3)), (irow, 0), (1, 5), LCEN)
irow += 1
title = SimpleText(panel, '[[Correlations]]', font=Font(12),
colour=self.colors.title, style=LCEN)
self.wids['all_correl'] = Button(panel, 'Show All',
size=(100, -1), action=self.onAllCorrel)
self.wids['min_correl'] = FloatCtrl(panel, value=MIN_CORREL,
minval=0, size=(60, -1), gformat=True)
ctitle = SimpleText(panel, 'minimum correlation: ')
sizer.Add(title, (irow, 0), (1, 1), LCEN)
sizer.Add(ctitle, (irow, 1), (1, 1), LCEN)
sizer.Add(self.wids['min_correl'], (irow, 2), (1, 1), LCEN)
sizer.Add(self.wids['all_correl'], (irow, 3), (1, 1), LCEN)
irow += 1
cview = self.wids['correl'] = dv.DataViewListCtrl(panel, style=dvstyle)
cview.AppendTextColumn('Parameter 1', width=150)
cview.AppendTextColumn('Parameter 2', width=150)
cview.AppendTextColumn('Correlation', width=100)
for col in (0, 1, 2):
this = cview.Columns[col]
isort, align = True, wx.ALIGN_LEFT
if col == 1:
isort = False
if col == 2:
align = wx.ALIGN_RIGHT
this.Sortable = isort
this.Alignment = this.Renderer.Alignment = align
cview.SetMinSize((450, 200))
irow += 1
sizer.Add(cview, (irow, 0), (1, 5), LCEN)
irow += 1
sizer.Add(HLine(panel, size=(400, 3)), (irow, 0), (1, 5), LCEN)
pack(panel, sizer)
panel.SetupScrolling()
mainsizer = wx.BoxSizer(wx.VERTICAL)
mainsizer.Add(panel, 1, wx.GROW|wx.ALL, 1)
pack(self, mainsizer)
self.Show()
self.Raise()
