def __init__(self, parent, pv=None,
font=None, fg=None, bg=None, precision=None, **kw):
self.pv = None
FloatCtrl.__init__(self, parent, value=0,
precision=precision, **kw)
pvCtrlMixin.__init__(self, pv=pv, font=font, fg=None, bg=None)
def __init__(self, parent, pv=None,
font=None, fg=None, bg=None, precision=None, **kw):
self.pv = None
FloatCtrl.__init__(self, parent, value=0,
precision=precision, **kw)
PVCtrlMixin.__init__(self, pv=pv, font=font, fg=None, bg=None)
def build_rowpanel(self, parent):
"build right hand panel"
panel = wx.Panel(parent)
sizer = wx.GridBagSizer(5, 2)
row = -1
for attr, dtype, extra in self.attrs:
row += 1
label = None
if dtype is None:
wid = wx.TextCtrl(panel, value="", size=(TWID, -1))
elif dtype == 'float':
prec, xmin, xmax = extra
wid = FloatCtrl(panel,
value="",
precision=prec,
minval=xmin,
maxval=xmax,
size=(TWID, -1))
elif dtype == 'multi':
height = extra
wid = wx.TextCtrl(panel,
value="",
size=(TWID, height),
style=wx.TE_MULTILINE)
elif dtype == 'foreign':
choiceclass, manager, llabel = extra
_wid = HyperChoice(panel,
choiceclass,
label='%s:' % llabel.title(),
manager=manager,
db=self.db)
wid = _wid.choice
label = _wid.link
setattr(self, attr, wid)
if label is None:
label = wx.StaticText(panel, label="%s:" % attr.title())
sizer.Add(label, (row, 0), (1, 1), wx.ALIGN_LEFT | wx.ALL, 2)
sizer.Add(wid, (row, 1), (1, 1), wx.ALIGN_LEFT | wx.ALL, 2)
pack(panel, sizer)
return panel
def buildDataPanel(self):
"build Data / Column selection panel"
panel = self.columns_panel = wx.Panel(self.nb)
self.nb.AddPage(panel, 'Data')
sizer = wx.GridBagSizer(10, 6)
def stext(label):
return wx.StaticText(panel, label=label)
self.elem = ElementChoice(panel, db=self.db)
self.edge = EdgeChoice(panel, db=self.db)
self.name = wx.TextCtrl(panel, value='', size=(220, -1))
self.name_ok = stext('')
self.column_en = ColumnChoice(panel)
self.column_i0 = ColumnChoice(panel)
self.column_it = ColumnChoice(panel)
self.column_if = ColumnChoice(panel)
self.column_ir = ColumnChoice(panel)
self.type_en = TypeChoice(panel, self.energy_choices)
self.type_i0 = TypeChoice(panel, (
'not collected',
'intensity',
))
self.type_it = TypeChoice(panel, ('not collected', 'intensity', 'mu'))
self.type_if = TypeChoice(panel, ('not collected', 'intensity', 'mu'))
self.type_ir = TypeChoice(panel, ('not collected', 'intensity', 'mu'))
self.column_en.SetSelection(0)
self.type_en.SetSelection(0)
self.type_i0.SetSelection(0)
self.type_it.SetSelection(0)
self.type_ir.SetSelection(0)
self.type_if.SetSelection(0)
_mono = HyperChoice(panel,
MonochromatorChoice,
label='Monochromator:',
manager=MonochromatorManager,
db=self.db)
self.mono_choice = _mono.choice
self.mono_dspacing = FloatCtrl(panel, precision=6, value=0, minval=0)
self.mono_steps = FloatCtrl(panel, precision=2, value=0, minval=0)
self.type_en.Bind(wx.EVT_CHOICE, self.onEnergyChoice)
sizer.Add(stext('Spectra Name:'), (0, 0), (1, 1), labstyle, 1)
sizer.Add(self.name, (0, 1), (1, 4), namestyle, 1)
sizer.Add(self.name_ok, (0, 5), (1, 1), clabstyle, 1)
sizer.Add(stext('Element:'), (1, 0), (1, 1), labstyle, 1)
sizer.Add(self.elem, (1, 1), (1, 1), choicestyle, 1)
sizer.Add(stext('Edge:'), (1, 2), (1, 1), rlabstyle, 1)
sizer.Add(self.edge, (1, 3), (1, 1), choicestyle, 1)
sizer.Add(wx.StaticLine(panel, size=(350, 2), style=wx.LI_HORIZONTAL),
(2, 0), (1, 6), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
irow = 3
sizer.Add(stext('Data:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(stext(' Type:'), (irow, 1), (1, 1), labstyle, 1)
sizer.Add(stext('Column #:'), (irow, 2), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext('Energy:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_en, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_en, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('I0:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_i0, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_i0, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('Transmission:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_it, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_it, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('Fluorescence:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_if, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_if, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('Reference:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_ir, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_ir, (irow, 2), (1, 1), choicestyle, 1)
irow = 4
sizer.Add(_mono.link, (irow, 3), (1, 1), labstyle, 1)
sizer.Add(self.mono_choice, (irow, 4), (1, 2), choicestyle, 1)
irow += 1
sizer.Add(stext(' d spacing ='), (irow, 3), (1, 1), rlabstyle, 1)
sizer.Add(self.mono_dspacing, (irow, 4), (1, 1), namestyle, 1)
sizer.Add(stext('Angtroms'), (irow, 5), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext(' resolution ='), (irow, 3), (1, 1), rlabstyle, 1)
sizer.Add(self.mono_steps, (irow, 4), (1, 1), namestyle, 1)
sizer.Add(stext('steps/degree'), (irow, 5), (1, 1), labstyle, 1)
#self.mono_dspacing.DisableEntry()
#self.mono_steps.DisableEntry()
pack(panel, sizer)
class FileImporter(wx.Frame):
title = 'ASCII File Importer'
energy_choices = ('energy (eV)', 'energy (keV)', 'angle (degrees)')
def __init__(self, fname=None, db=None):
wx.Frame.__init__(self, parent=None, title=self.title, size=(650, 650))
self.fname = fname
self.db = db
menuBar = wx.MenuBar()
fmenu = wx.Menu()
add_menu(self,
fmenu,
"E&xit\tAlt-X",
"Exit this Program",
action=self.onClose)
menuBar.Append(fmenu, "&File")
self.SetMenuBar(menuBar)
self.CreateStatusBar(1, wx.CAPTION | wx.THICK_FRAME)
splitter = wx.SplitterWindow(self, style=wx.SP_LIVE_UPDATE)
splitter.SetMinimumPaneSize(150)
self.top_panel = wx.Panel(splitter) # top
self.bot_panel = wx.Panel(splitter) # bottom
self.filename = wx.StaticText(self.bot_panel, -1, label='')
self.fileview = wx.TextCtrl(self.bot_panel,
size=(175, 175),
style=wx.TE_READONLY | wx.TE_MULTILINE
| wx.HSCROLL)
self.fileview.SetBackgroundColour(wx.Colour(250, 250, 250))
bot_sizer = wx.BoxSizer(wx.VERTICAL)
bot_sizer.Add(self.filename, 0,
wx.ALIGN_TOP | wx.ALIGN_CENTER_VERTICAL, 5)
bot_sizer.Add(self.fileview, 1, wx.ALIGN_TOP | wx.GROW | wx.ALL, 5)
pack(self.bot_panel, bot_sizer)
top_sizer = wx.BoxSizer(wx.VERTICAL)
title_row = wx.Panel(self.top_panel)
title_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.top_label = Title(title_row,
'Data from File:',
colour=(200, 20, 20))
self.save_btn = add_btn(title_row, "Import Data", action=self.onImport)
title_sizer.Add(self.top_label, 1, wx.ALIGN_LEFT | wx.ALL, 3)
title_sizer.Add(self.save_btn, 0, wx.ALIGN_RIGHT | wx.GROW | wx.ALL, 3)
pack(title_row, title_sizer)
self.nb = wx.Notebook(self.top_panel, size=(600, 275))
self.buildDataPanel()
self.buildBeamlinePanel()
self.buildSamplePanel()
top_sizer.Add(title_row, 0, wx.ALIGN_LEFT | wx.GROW | wx.ALL, 1)
top_sizer.Add(self.nb, 1, wx.ALIGN_CENTER | wx.GROW | wx.ALL, 1)
pack(self.top_panel, top_sizer)
splitter.SplitHorizontally(self.top_panel, self.bot_panel, -1)
framesizer = wx.BoxSizer(wx.VERTICAL)
framesizer.Add(splitter, 1, wx.GROW | wx.ALL, 5)
self.SetSizer(framesizer)
if fname is not None:
self.ShowFile(fname)
self.Show()
self.Raise()
def buildBeamlinePanel(self):
"build Beamline info panel"
# beamline
# facility
# monochromator
# energy units
panel = self.beamline_panel = wx.Panel(self.nb)
self.nb.AddPage(panel, 'Data Collection')
sizer = wx.GridBagSizer(10, 6)
def stext(label):
return wx.StaticText(panel, label=label)
_person = HyperChoice(panel,
PersonChoice,
label='Person:',
manager=PersonManager,
db=self.db)
self.person = _person.choice
_beamline = HyperChoice(panel,
BeamlineChoice,
label='Beamline:',
manager=BeamlineManager,
db=self.db)
self.beamline = _beamline.choice
_cite = HyperChoice(panel,
CitationChoice,
label='Citation:',
db=self.db)
self.citation = _cite.choice
self.collection_datetime = DateTimeCtrl(panel, name="collection_time")
self.submission_datetime = DateTimeCtrl(panel,
name="collection_time",
use_now=True)
sizer.Add(Title(panel, 'Data Collection Information:'), (0, 0), (1, 3),
titlestyle, 1)
irow = 1
sizer.Add(_person.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.person, (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(stext('Collection Date:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.collection_datetime[0], (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(stext('Submission Date:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.submission_datetime[0], (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(_beamline.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.beamline, (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(_cite.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.citation, (irow, 1), (1, 1), namestyle, 1)
## RHS of Beamline Panel
sizer.Add(wx.StaticLine(panel, size=(1, 160), style=wx.LI_VERTICAL),
(1, 2), (6, 1), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
irow = 1
self.ring_current = FloatCtrl(panel,
precision=2,
value=0,
minval=0,
maxval=1e6)
self.ring_energy = FloatCtrl(panel,
precision=3,
value=0,
minval=0,
maxval=1e3)
self.xray_source = wx.TextCtrl(panel, value='', size=(180, -1))
self.focusing = wx.TextCtrl(panel, value='', size=(180, -1))
self.collimation = wx.TextCtrl(panel, value='', size=(180, -1))
self.harmonic_rej = wx.TextCtrl(panel, value='', size=(180, -1))
sizer.Add(stext('Ring Current:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.ring_current, (irow, 5), (1, 1), namestyle, 1)
sizer.Add(stext('mA'), (irow, 6), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext('Ring Energy:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.ring_energy, (irow, 5), (1, 1), namestyle, 1)
sizer.Add(stext('GeV'), (irow, 6), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext('X-ray Source:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.xray_source, (irow, 5), (1, 2), namestyle, 1)
irow += 1
sizer.Add(stext('Focusing:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.focusing, (irow, 5), (1, 2), namestyle, 1)
irow += 1
sizer.Add(stext('Collimation:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.collimation, (irow, 5), (1, 2), namestyle, 1)
irow += 1
sizer.Add(stext('Harmonic\nRejection:'), (irow, 4), (1, 1), labstyle,
1)
sizer.Add(self.harmonic_rej, (irow, 5), (1, 2), namestyle, 1)
pack(panel, sizer)
# def buildCitationPanel(self):
# "build Citation / Misc panel"
# panel = self.refer_panel = wx.Panel(self.nb)
# self.nb.AddPage(panel, 'Literature Citation')
#
# sizer = wx.GridBagSizer(10, 6)
#
# def stext(label):
# return wx.StaticText(panel, label=label)
#
# self.cite_authors = wx.TextCtrl(panel, value='', size=(225, 85),
# style=wx.TE_MULTILINE)
# self.cite_journal = wx.TextCtrl(panel, value='', size=(225, -1))
# self.cite_title = wx.TextCtrl(panel, value='', size=(225, -1))
# self.cite_volume = wx.TextCtrl(panel, value='', size=(120, -1))
# self.cite_pages = wx.TextCtrl(panel, value='', size=(120, -1))
# self.cite_year = wx.TextCtrl(panel, value='', size=(120, -1))
# self.cite_doi = wx.TextCtrl(panel, value='', size=(225, -1))
#
# _cite = HyperChoice(panel, CitationChoice,
# label='Citation:', db=self.db)
# self.citation = _cite.choice
#
# sizer.Add(_cite.link, (0, 0), (1, 1), labstyle, 1)
# sizer.Add(self.citation, (0, 1), (1, 1), choicestyle, 1)
#
# irow = 1
# sizer.Add(stext('Journal:'), (irow, 0), (1, 1), labstyle, 1)
# sizer.Add(self.cite_journal, (irow, 1), (1, 1), namestyle, 1)
#
# irow += 1
# sizer.Add(stext('Title:'), (irow, 0), (1, 1), labstyle, 1)
# sizer.Add(self.cite_title, (irow, 1), (1, 1), namestyle, 1)
# irow += 1
# sizer.Add(stext('Volume:'), (irow, 0), (1, 1), labstyle, 1)
# sizer.Add(self.cite_volume, (irow, 1), (1, 1), namestyle, 1)
#
# irow += 1
# sizer.Add(stext('Pages:'), (irow, 0), (1, 1), labstyle, 1)
# sizer.Add(self.cite_pages, (irow, 1), (1, 1), namestyle, 1)
#
# irow += 1
# sizer.Add(stext('Year:'), (irow, 0), (1, 1), labstyle, 1)
# sizer.Add(self.cite_year, (irow, 1), (1, 1), namestyle, 1)
#
# sizer.Add(stext('Authors:'), (1, 3), (1, 1), labstyle, 1)
# sizer.Add(self.cite_authors, (1, 4), (3, 1), namestyle, 1)
# sizer.Add(stext('DOI:'), (4, 3), (1, 1), labstyle, 1)
# sizer.Add(self.cite_doi, (4, 4), (1, 1), namestyle, 1)
#
# sizer.Add(wx.StaticLine(panel, size=(-1, 150),
# style=wx.LI_VERTICAL),
# (1, 2), (5, 1), wx.ALIGN_CENTER|wx.GROW|wx.ALL, 0)
#
#
# pack(panel, sizer)
def buildSamplePanel(self):
"build Sample panel"
# temperature
# sample crystal structure data, format
# ref sample formula
# ref sample material source
# ref sample crystal structure data, format
panel = self.sample_panel = wx.Panel(self.nb)
self.nb.AddPage(panel, 'Sample')
sizer = wx.GridBagSizer(10, 6)
def stext(label):
return wx.StaticText(panel, label=label)
# self.sample_formula = wx.TextCtrl(panel, value='', size=(220, -1))
# self.refer_formula = wx.TextCtrl(panel, value='', size=(220, -1))
# self.sample_xtal = add_btn(panel, "Add Crystal Structure Data",
# action=self.onSampleXTAL)
# self.sample_source = wx.TextCtrl(panel, value='', size=(220, 75),
# style=wx.TE_MULTILINE)
self.user_notes = wx.TextCtrl(panel,
value='',
size=(450, 90),
style=wx.TE_MULTILINE)
irow = 1
sizer.Add(Title(panel, 'User Notes from\nData Collection:'), (0, 0),
(1, 1), titlestyle, 1)
sizer.Add(self.user_notes, (0, 1), (1, 3), namestyle, 1)
sizer.Add(wx.StaticLine(panel, size=(150, 1), style=wx.LI_HORIZONTAL),
(1, 0), (1, 5), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
_sample = HyperChoice(panel,
SampleChoice,
label='Sample:',
manager=SampleManager,
db=self.db)
self.sample = _sample.choice
_refsample = HyperChoice(panel,
SampleChoice,
label='Reference Sample:',
manager=SampleManager,
db=self.db)
self.refsample = _refsample.choice
sizer.Add(_sample.link, (2, 0), (1, 1), labstyle, 1)
sizer.Add(self.sample, (2, 1), (1, 1), namestyle, 1)
irow = 3
tpanel = wx.Panel(panel)
tsizer = wx.BoxSizer(wx.HORIZONTAL)
self.sample_temperature = FloatCtrl(tpanel,
precision=3,
size=(100, -1),
value=300,
minval=-500)
self.temperature_units = TypeChoice(tpanel,
choices=('K', 'C', 'F'),
size=(50, -1))
self.temperature_units.SetSelection(0)
tsizer.Add(self.sample_temperature, 0, wx.ALIGN_CENTER | wx.ALL, 1)
tsizer.Add(self.temperature_units, 0, wx.ALIGN_CENTER | wx.ALL, 1)
pack(tpanel, tsizer)
sizer.Add(stext('Temperature:'), (irow, 0), (1, 1), tlabstyle, 1)
sizer.Add(tpanel, (irow, 1), (1, 2), tnamestyle, 1)
irow += 1
sizer.Add(wx.StaticLine(panel, size=(150, 1), style=wx.LI_HORIZONTAL),
(irow, 0), (1, 5), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
irow += 1
sizer.Add(_refsample.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.refsample, (irow, 1), (1, 1), namestyle, 1)
pack(panel, sizer)
def buildDataPanel(self):
"build Data / Column selection panel"
panel = self.columns_panel = wx.Panel(self.nb)
self.nb.AddPage(panel, 'Data')
sizer = wx.GridBagSizer(10, 6)
def stext(label):
return wx.StaticText(panel, label=label)
self.elem = ElementChoice(panel, db=self.db)
self.edge = EdgeChoice(panel, db=self.db)
self.name = wx.TextCtrl(panel, value='', size=(220, -1))
self.name_ok = stext('')
self.column_en = ColumnChoice(panel)
self.column_i0 = ColumnChoice(panel)
self.column_it = ColumnChoice(panel)
self.column_if = ColumnChoice(panel)
self.column_ir = ColumnChoice(panel)
self.type_en = TypeChoice(panel, self.energy_choices)
self.type_i0 = TypeChoice(panel, (
'not collected',
'intensity',
))
self.type_it = TypeChoice(panel, ('not collected', 'intensity', 'mu'))
self.type_if = TypeChoice(panel, ('not collected', 'intensity', 'mu'))
self.type_ir = TypeChoice(panel, ('not collected', 'intensity', 'mu'))
self.column_en.SetSelection(0)
self.type_en.SetSelection(0)
self.type_i0.SetSelection(0)
self.type_it.SetSelection(0)
self.type_ir.SetSelection(0)
self.type_if.SetSelection(0)
_mono = HyperChoice(panel,
MonochromatorChoice,
label='Monochromator:',
manager=MonochromatorManager,
db=self.db)
self.mono_choice = _mono.choice
self.mono_dspacing = FloatCtrl(panel, precision=6, value=0, minval=0)
self.mono_steps = FloatCtrl(panel, precision=2, value=0, minval=0)
self.type_en.Bind(wx.EVT_CHOICE, self.onEnergyChoice)
sizer.Add(stext('Spectra Name:'), (0, 0), (1, 1), labstyle, 1)
sizer.Add(self.name, (0, 1), (1, 4), namestyle, 1)
sizer.Add(self.name_ok, (0, 5), (1, 1), clabstyle, 1)
sizer.Add(stext('Element:'), (1, 0), (1, 1), labstyle, 1)
sizer.Add(self.elem, (1, 1), (1, 1), choicestyle, 1)
sizer.Add(stext('Edge:'), (1, 2), (1, 1), rlabstyle, 1)
sizer.Add(self.edge, (1, 3), (1, 1), choicestyle, 1)
sizer.Add(wx.StaticLine(panel, size=(350, 2), style=wx.LI_HORIZONTAL),
(2, 0), (1, 6), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
irow = 3
sizer.Add(stext('Data:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(stext(' Type:'), (irow, 1), (1, 1), labstyle, 1)
sizer.Add(stext('Column #:'), (irow, 2), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext('Energy:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_en, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_en, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('I0:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_i0, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_i0, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('Transmission:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_it, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_it, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('Fluorescence:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_if, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_if, (irow, 2), (1, 1), choicestyle, 1)
irow += 1
sizer.Add(stext('Reference:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.type_ir, (irow, 1), (1, 1), choicestyle, 1)
sizer.Add(self.column_ir, (irow, 2), (1, 1), choicestyle, 1)
irow = 4
sizer.Add(_mono.link, (irow, 3), (1, 1), labstyle, 1)
sizer.Add(self.mono_choice, (irow, 4), (1, 2), choicestyle, 1)
irow += 1
sizer.Add(stext(' d spacing ='), (irow, 3), (1, 1), rlabstyle, 1)
sizer.Add(self.mono_dspacing, (irow, 4), (1, 1), namestyle, 1)
sizer.Add(stext('Angtroms'), (irow, 5), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext(' resolution ='), (irow, 3), (1, 1), rlabstyle, 1)
sizer.Add(self.mono_steps, (irow, 4), (1, 1), namestyle, 1)
sizer.Add(stext('steps/degree'), (irow, 5), (1, 1), labstyle, 1)
#self.mono_dspacing.DisableEntry()
#self.mono_steps.DisableEntry()
pack(panel, sizer)
def onImport(self, event=None):
print "Import Data to db! ", self.db
kws = {}
for attr in ('energy', 'i0', 'itrans', 'ifluor', 'irefer', 'mutrans',
'mufluor', 'murefer', 'k', 'chi'):
if hasattr(self.xdifile, attr):
kws[attr] = getattr(self.xdifile, attr)
for name, choice_widget in (('edge', self.edge), ('element',
self.elem)):
val = choice_widget.GetStringSelection()
if val not in (None, '', 'None'):
kws[name] = val
sdate = self.collection_datetime[1].GetValue()
stime = self.collection_datetime[2].GetValue()
#print '--> db.add_spectra ', kws.keys()
#print 'sdata ', type(sdate), sdate
print dir(sdate)
print sdate.GetDateOnly()
print sdate.ToUTC()
# self.db.add_spectra(self.name.GetValue(), **kws)
# name, notes='', attributes='', file_link='',
# notes_i0='', notes_itrans='', notes_ifluor='',
# notes_irefer='', temperature='', submission_date=None,
# collection_date=None, reference_used='',
# energy_units=None, monochromator=None, person=None,
# sample=None, beamline=None,
# data_format=None, citation=None, reference=None, **kws):
def onEnergyChoice(self, evt=None):
""" on selection of energy type """
if 'angle' in evt.GetString().lower():
self.mono_dspacing.EnableEntry()
else:
self.mono_dspacing.DisableEntry()
if 'step' in evt.GetString().lower():
self.mono_steps.EnableEntry()
else:
self.mono_steps.DisableEntry()
def ShowFile(self, fname=None):
if fname is not None:
text = open(fname, 'r').read()
short_fname = os.path.basename(fname)
self.name.SetValue(short_fname.replace('/', '_'))
self.filename.SetLabel(fname)
self.top_label.SetLabel('Data from File: %s' % (short_fname))
self.fileview.Clear()
self.fileview.SetValue(text)
self.fileview.ShowPosition(0)
if True:
self.xdifile = XDIFile(fname)
self.fill_from_xdifile(self.xdifile)
else:
print 'Not an XDI File'
def fill_from_xdifile(self, xfile):
"fill in form from XDIFile"
def get_attr(name):
if '.' in name:
family, member = name.lower().split('.', 1)
if family in xfile.attrs:
return xfile.attrs[family].get(member, None)
return None
self.user_notes.SetValue(xfile.comments)
self.edge.Select(get_attr('scan.edge'))
self.elem.Select(get_attr('scan.element'))
for entry in (('beamline.collimation', self.collimation,
str), ('beamline.focusing', self.focusing, str),
('beamline.harmonic_rejection', self.harmonic_rej,
str), ('mono.d_spacing', self.mono_dspacing, float),
('facility.xray_source', self.xray_source,
str), ('facility.ring_current', self.ring_current, str),
('facility.energy', self.ring_energy, float)):
attr, widget, cast = entry
value = get_attr(attr)
if value not in (None, 'none', 'None'):
widget.SetValue(cast(value))
# beamline, facility, mono settings --- tricky
beamline = get_attr('beamline.name')
if beamline not in (None, 'none', 'None'):
self.beamline.Select(beamline)
m_xtal = get_attr('mono.name')
if m_xtal not in (None, 'none', 'None'):
try:
m_name = self.monochromator.Select(m_xtal)
if m_name is not None:
mono = self.db.query(xasdb.Monochromator).filter(
xasdb.Monochromator.name == m_name).one()
self.mono_dspacing.SetValue(float(mono.dspacing))
self.mono_steps.SetValue(float(mono.steps_per_degree))
self.mono_dspacing.Disable()
self.mono_stepsd.Disable()
except:
pass
# set date/time
isotime = get_attr('scan.end_time')
if isotime is None:
isotime = get_attr('scan.start_time')
if isotime is not None:
dtime = isotime2datetime(isotime)
wxdate = wx.DateTime()
wxdate.Set(dtime.day, dtime.month - 1, dtime.year, dtime.hour,
dtime.minute, dtime.second, dtime.microsecond / 1000.0)
self.collection_datetime[1].SetValue(wxdate)
self.collection_datetime[2].SetValue(wxdate)
nrow, ncol = xfile.rawdata.shape
col_choices = ['%i' % (i + 1) for i in range(ncol)]
self.column_en.SetChoices(col_choices)
self.column_i0.SetChoices(col_choices)
self.column_it.SetChoices(col_choices)
self.column_if.SetChoices(col_choices)
self.column_ir.SetChoices(col_choices)
for num, name in xfile.column_labels.items():
if name == 'energy':
self.column_en.Select(num)
extra = xfile.column_attrs.get(num, 'eV')
if extra in ('eV', 'keV'):
self.type_it.Select('energy (%s)' % extra)
elif name == 'angle':
self.column_en.Select(num)
extra = xfile.column_attrs.get(num, 'degrees')
if extra.startswith('deg'): extra = 'degrees'
if extra.startswith('rad'): extra = 'radians'
if extra in ('degrees', 'radians'):
self.type_it.Select('angle (%s)' % extra)
elif name == 'i0':
self.column_i0.Select(num)
self.type_i0.Select('intensity')
elif name == 'mutrans':
self.column_it.Select(num)
self.type_it.Select('mu')
elif name == 'itrans':
self.column_it.Select(num)
self.type_it.Select('intensity')
elif name == 'mufluor':
self.column_if.Select(num)
self.type_if.Select('mu')
elif name == 'ifluor':
self.column_if.Select(num)
self.type_if.Select('intensity')
elif name == 'murefer':
self.column_ir.Select(num)
self.type_ir.Select('mu')
elif name == 'irefer':
self.column_ir.Select(num)
self.type_ir.Select('intensity')
def onClose(self, evt=None):
"quit frame"
self.Destroy()
def buildSamplePanel(self):
"build Sample panel"
# temperature
# sample crystal structure data, format
# ref sample formula
# ref sample material source
# ref sample crystal structure data, format
panel = self.sample_panel = wx.Panel(self.nb)
self.nb.AddPage(panel, 'Sample')
sizer = wx.GridBagSizer(10, 6)
def stext(label):
return wx.StaticText(panel, label=label)
# self.sample_formula = wx.TextCtrl(panel, value='', size=(220, -1))
# self.refer_formula = wx.TextCtrl(panel, value='', size=(220, -1))
# self.sample_xtal = add_btn(panel, "Add Crystal Structure Data",
# action=self.onSampleXTAL)
# self.sample_source = wx.TextCtrl(panel, value='', size=(220, 75),
# style=wx.TE_MULTILINE)
self.user_notes = wx.TextCtrl(panel,
value='',
size=(450, 90),
style=wx.TE_MULTILINE)
irow = 1
sizer.Add(Title(panel, 'User Notes from\nData Collection:'), (0, 0),
(1, 1), titlestyle, 1)
sizer.Add(self.user_notes, (0, 1), (1, 3), namestyle, 1)
sizer.Add(wx.StaticLine(panel, size=(150, 1), style=wx.LI_HORIZONTAL),
(1, 0), (1, 5), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
_sample = HyperChoice(panel,
SampleChoice,
label='Sample:',
manager=SampleManager,
db=self.db)
self.sample = _sample.choice
_refsample = HyperChoice(panel,
SampleChoice,
label='Reference Sample:',
manager=SampleManager,
db=self.db)
self.refsample = _refsample.choice
sizer.Add(_sample.link, (2, 0), (1, 1), labstyle, 1)
sizer.Add(self.sample, (2, 1), (1, 1), namestyle, 1)
irow = 3
tpanel = wx.Panel(panel)
tsizer = wx.BoxSizer(wx.HORIZONTAL)
self.sample_temperature = FloatCtrl(tpanel,
precision=3,
size=(100, -1),
value=300,
minval=-500)
self.temperature_units = TypeChoice(tpanel,
choices=('K', 'C', 'F'),
size=(50, -1))
self.temperature_units.SetSelection(0)
tsizer.Add(self.sample_temperature, 0, wx.ALIGN_CENTER | wx.ALL, 1)
tsizer.Add(self.temperature_units, 0, wx.ALIGN_CENTER | wx.ALL, 1)
pack(tpanel, tsizer)
sizer.Add(stext('Temperature:'), (irow, 0), (1, 1), tlabstyle, 1)
sizer.Add(tpanel, (irow, 1), (1, 2), tnamestyle, 1)
irow += 1
sizer.Add(wx.StaticLine(panel, size=(150, 1), style=wx.LI_HORIZONTAL),
(irow, 0), (1, 5), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
irow += 1
sizer.Add(_refsample.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.refsample, (irow, 1), (1, 1), namestyle, 1)
pack(panel, sizer)
def buildBeamlinePanel(self):
"build Beamline info panel"
# beamline
# facility
# monochromator
# energy units
panel = self.beamline_panel = wx.Panel(self.nb)
self.nb.AddPage(panel, 'Data Collection')
sizer = wx.GridBagSizer(10, 6)
def stext(label):
return wx.StaticText(panel, label=label)
_person = HyperChoice(panel,
PersonChoice,
label='Person:',
manager=PersonManager,
db=self.db)
self.person = _person.choice
_beamline = HyperChoice(panel,
BeamlineChoice,
label='Beamline:',
manager=BeamlineManager,
db=self.db)
self.beamline = _beamline.choice
_cite = HyperChoice(panel,
CitationChoice,
label='Citation:',
db=self.db)
self.citation = _cite.choice
self.collection_datetime = DateTimeCtrl(panel, name="collection_time")
self.submission_datetime = DateTimeCtrl(panel,
name="collection_time",
use_now=True)
sizer.Add(Title(panel, 'Data Collection Information:'), (0, 0), (1, 3),
titlestyle, 1)
irow = 1
sizer.Add(_person.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.person, (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(stext('Collection Date:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.collection_datetime[0], (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(stext('Submission Date:'), (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.submission_datetime[0], (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(_beamline.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.beamline, (irow, 1), (1, 1), namestyle, 1)
irow += 1
sizer.Add(_cite.link, (irow, 0), (1, 1), labstyle, 1)
sizer.Add(self.citation, (irow, 1), (1, 1), namestyle, 1)
## RHS of Beamline Panel
sizer.Add(wx.StaticLine(panel, size=(1, 160), style=wx.LI_VERTICAL),
(1, 2), (6, 1), wx.ALIGN_CENTER | wx.GROW | wx.ALL, 0)
irow = 1
self.ring_current = FloatCtrl(panel,
precision=2,
value=0,
minval=0,
maxval=1e6)
self.ring_energy = FloatCtrl(panel,
precision=3,
value=0,
minval=0,
maxval=1e3)
self.xray_source = wx.TextCtrl(panel, value='', size=(180, -1))
self.focusing = wx.TextCtrl(panel, value='', size=(180, -1))
self.collimation = wx.TextCtrl(panel, value='', size=(180, -1))
self.harmonic_rej = wx.TextCtrl(panel, value='', size=(180, -1))
sizer.Add(stext('Ring Current:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.ring_current, (irow, 5), (1, 1), namestyle, 1)
sizer.Add(stext('mA'), (irow, 6), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext('Ring Energy:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.ring_energy, (irow, 5), (1, 1), namestyle, 1)
sizer.Add(stext('GeV'), (irow, 6), (1, 1), labstyle, 1)
irow += 1
sizer.Add(stext('X-ray Source:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.xray_source, (irow, 5), (1, 2), namestyle, 1)
irow += 1
sizer.Add(stext('Focusing:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.focusing, (irow, 5), (1, 2), namestyle, 1)
irow += 1
sizer.Add(stext('Collimation:'), (irow, 4), (1, 1), labstyle, 1)
sizer.Add(self.collimation, (irow, 5), (1, 2), namestyle, 1)
irow += 1
sizer.Add(stext('Harmonic\nRejection:'), (irow, 4), (1, 1), labstyle,
1)
sizer.Add(self.harmonic_rej, (irow, 5), (1, 2), namestyle, 1)
pack(panel, sizer)
def _SetValue(self, value):
"set widget value"
FloatCtrl.SetValue(self, value, act=False)