class XYFitFrame(wx.Frame): _about = """Larch XYFit: XY Data Viewing & Curve Fitting Matt Newville <newville @ cars.uchicago.edu> """ def __init__(self, parent=None, size=(875, 550), _larch=None, **kws): wx.Frame.__init__(self, parent, -1, size=size, style=FRAMESTYLE) self.last_array_sel = {} title = "Larch XYFit (BETA!!): XY Data Viewing & Curve Fitting" self.larch_buffer = parent if not isinstance(parent, LarchFrame): self.larch_buffer = LarchFrame(_larch=_larch) self.larch_buffer.Show() self.larch_buffer.Raise() self.larch = self.larch_buffer._larch self.controller = XYFitController(wxparent=self, _larch=self.larch_buffer.larchshell) self.subframes = {} self.plotframe = None self.SetTitle(title) self.SetSize(size) self.SetFont(Font(10)) self.config = {'chdir_on_fileopen': True} self.createMainPanel() self.createMenus() self.statusbar = self.CreateStatusBar(2, 0) self.statusbar.SetStatusWidths([-3, -1]) statusbar_fields = ["Initializing....", " "] for i in range(len(statusbar_fields)): self.statusbar.SetStatusText(statusbar_fields[i], i) read_workdir('xyfit.dat') self.larch_buffer.Hide() 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 onSelAll(self, event=None): self.controller.filelist.SetCheckedStrings( self.controller.file_groups.keys()) def onSelNone(self, event=None): self.controller.filelist.SetCheckedStrings([]) def init_larch(self): self.SetStatusText('ready') self.title.SetLabel('') self.fit_panel.larch = self.controller.larch fico = self.controller.get_iconfile() plotframe = self.controller.get_display(stacked=False) xpos, ypos = self.GetPosition() xsiz, ysiz = self.GetSize() plotframe.SetPosition((xpos + xsiz, ypos)) self.Raise() def write_message(self, s, panel=0): """write a message to the Status Bar""" self.SetStatusText(s, panel) def onPlotOne(self, evt=None, groupname=None): if groupname is None: groupname = self.controller.groupname dgroup = self.controller.get_group(groupname) if dgroup is not None: self.controller.plot_group(groupname=groupname, new=True) def onPlotSel(self, evt=None): newplot = True group_ids = self.controller.filelist.GetCheckedStrings() for checked in group_ids: groupname = self.controller.file_groups[str(checked)] dgroup = self.controller.get_group(groupname) if dgroup is None: continue dgroup.plot_yarrays = [(dgroup.y, PLOTOPTS_1, dgroup.filename)] if dgroup.datatype == 'xas': dgroup.plot_xlabel = '$E\,\mathrm{(eV)}$' dgroup.plot_ymarkers = [] self.controller.plot_group(groupname=groupname, title='', new=newplot) newplot = False def plot_group(self, groupname=None, title=None, new=True, **kws): self.controller.plot_group(groupname=groupname, title=title, new=new, **kws) self.Raise() def ShowFile(self, evt=None, groupname=None, **kws): filename = None if evt is not None: filename = str(evt.GetString()) if groupname is None and filename is not None: groupname = self.controller.file_groups[filename] if not hasattr(self.larch.symtable, groupname): print('Error reading file ', groupname) return dgroup = self.controller.get_group(groupname) self.controller.group = dgroup self.controller.groupname = groupname self.nb.SetSelection(0) self.proc_panel.fill(dgroup) if filename is None: filename = dgroup.filename self.title.SetLabel(filename) def createMenus(self): # ppnl = self.plotpanel self.menubar = wx.MenuBar() # fmenu = wx.Menu() MenuItem(self, fmenu, "&Open Data File\tCtrl+O", "Open Data File", self.onReadDialog) fmenu.AppendSeparator() MenuItem(self, fmenu, 'Show Larch Buffer\tCtrl+L', 'Show Larch Programming Buffer', self.onShowLarchBuffer) MenuItem(self, fmenu, "debug wx\tCtrl+I", "", self.showInspectionTool) MenuItem(self, fmenu, "&Quit\tCtrl+Q", "Quit program", self.onClose) self.menubar.Append(fmenu, "&File") omenu = wx.Menu() self.menubar.Append(omenu, "Options") MenuItem(self, omenu, "Configure Data Processing", "Configure Data Processing", self.onConfigDataProcessing) MenuItem(self, omenu, "Configure Data Fitting", "Configure Data Fitting", self.onConfigDataFitting) self.SetMenuBar(self.menubar) self.Bind(wx.EVT_CLOSE, self.onClose) def onShowLarchBuffer(self, evt=None): if self.larch_buffer is None: self.larch_buffer = LarchFrame(_larch=self.larch) self.larch_buffer.Show() self.larch_buffer.Raise() def onConfigDataProcessing(self, event=None): pass def onConfigDataFitting(self, event=None): pass def showInspectionTool(self, event=None): app = wx.GetApp() app.ShowInspectionTool() def onAbout(self, evt): dlg = wx.MessageDialog(self, self._about, "About Larch XYFit", wx.OK | wx.ICON_INFORMATION) dlg.ShowModal() dlg.Destroy() def onClose(self, evt): save_workdir('xyfit.dat') self.proc_panel.proc_timer.Stop() for nam in dir(self.larch.symtable._plotter): obj = getattr(self.larch.symtable._plotter, nam) try: obj.Destroy() except: pass for nam in dir(self.larch.symtable._sys.wx): obj = getattr(self.larch.symtable._sys.wx, nam) del obj if self.larch_buffer is not None: try: self.larch_buffer.Destroy() except: pass self.Destroy() def show_subframe(self, name, frameclass, **opts): shown = False if name in self.subframes: try: self.subframes[name].Raise() shown = True except: del self.subframes[name] if not shown: self.subframes[name] = frameclass(self, **opts) def onSelectColumns(self, evt=None): dgroup = self.controller.get_group(self.controller.groupname) self.show_subframe('readfile', ColumnDataFileFrame, group=dgroup.raw, last_array_sel=self.last_array_sel, _larch=self.larch, read_ok_cb=partial(self.onRead_OK, overwrite=True)) def onReadDialog(self, evt=None): dlg = wx.FileDialog(self, message="Read Data File", defaultDir=os.getcwd(), wildcard=FILE_WILDCARDS, style=wx.FD_OPEN) if dlg.ShowModal() == wx.ID_OK: path = dlg.GetPath() path = path.replace('\\', '/') self.onRead(path) dlg.Destroy() def onRead(self, path): if path in self.controller.file_groups: if wx.ID_YES != popup(self, "Re-read file '%s'?" % path, 'Re-read file?'): return filedir, filename = os.path.split(path) pref = fix_varname(filename.replace('.', '_')) if len(pref) > 15: pref = pref[:15] groupname = pref count, maxcount = 0, 999 while hasattr(self.larch.symtable, groupname) and count < maxcount: count += 1 groupname = '%s_%2.2i' % (pref, count) if self.config['chdir_on_fileopen']: os.chdir(filedir) kwargs = dict(filename=path, _larch=self.larch_buffer.larchshell, last_array_sel=self.last_array_sel, read_ok_cb=self.onRead_OK) # check for athena projects if is_athena_project(path): self.show_subframe('athena_import', AthenaImporter, **kwargs) else: self.show_subframe('readfile', ColumnDataFileFrame, **kwargs) def onRead_OK(self, datagroup, array_sel=None, overwrite=False): """ called when column data has been selected and is ready to be used overwrite: whether to overwrite the current datagroup, as when editing a datagroup """ gname = datagroup.groupname fname = datagroup.filename path = datagroup.path datatype = getattr(datagroup, 'datatype', 'raw') if array_sel is not None: self.last_array_sel = array_sel # file /group may already exist in list if fname in self.controller.file_groups and not overwrite: for i in range(1, 101): ftest = "%s (%i)" % (fname, i) if ftest not in self.controller.file_groups: fname = ftest break if fname not in self.controller.file_groups: self.controller.filelist.Append(fname) self.controller.file_groups[fname] = gname self.nb.SetSelection(0) self.ShowFile(groupname=gname)
class XYFitFrame(wx.Frame): _about = """Larch XYFit: XY Data Viewing & Curve Fitting Matt Newville <newville @ cars.uchicago.edu> """ def __init__(self, parent=None, size=(875, 550), _larch=None, **kws): wx.Frame.__init__(self, parent, -1, size=size, style=FRAMESTYLE) self.last_array_sel = {} self.paths2read = [] title = "Larch XYFit: XY Data Viewing & Curve Fitting" self.larch_buffer = parent if not isinstance(parent, LarchFrame): self.larch_buffer = LarchFrame(_larch=_larch) self.larch_buffer.Show() self.larch_buffer.Raise() self.larch=self.larch_buffer.larchshell self.controller = XYFitController(wxparent=self, _larch=self.larch) self.subframes = {} self.plotframe = None self.SetTitle(title) self.SetSize(size) self.SetFont(Font(10)) self.larch_buffer.Hide() self.createMainPanel() self.createMenus() self.statusbar = self.CreateStatusBar(2, 0) self.statusbar.SetStatusWidths([-3, -1]) statusbar_fields = ["Initializing....", " "] for i in range(len(statusbar_fields)): self.statusbar.SetStatusText(statusbar_fields[i], i) 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, remove_action=self.RemoveFile) 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 onSelAll(self, event=None): self.controller.filelist.SetCheckedStrings(self.controller.file_groups.keys()) def onSelNone(self, event=None): self.controller.filelist.SetCheckedStrings([]) def init_larch(self): self.SetStatusText('initializing Larch') self.title.SetLabel('') self.fit_panel.larch = self.controller.larch self.controller.init_larch() plotframe = self.controller.get_display(stacked=False) xpos, ypos = self.GetPosition() xsiz, ysiz = self.GetSize() plotframe.SetPosition((xpos+xsiz, ypos)) self.SetStatusText('ready') self.Raise() def write_message(self, s, panel=0): """write a message to the Status Bar""" self.SetStatusText(s, panel) def onPlotOne(self, evt=None, groupname=None): if groupname is None: groupname = self.controller.groupname dgroup = self.controller.get_group(groupname) if dgroup is not None: self.controller.plot_group(groupname=groupname, new=True) def onPlotSel(self, evt=None): newplot = True group_ids = self.controller.filelist.GetCheckedStrings() for checked in group_ids: groupname = self.controller.file_groups[str(checked)] dgroup = self.controller.get_group(groupname) if dgroup is not None: self.controller.plot_group(groupname=groupname, title='', new=newplot, label=dgroup.filename) newplot=False def plot_group(self, groupname=None, title=None, new=True, **kws): self.controller.plot_group(groupname=groupname, title=title, new=new, **kws) self.Raise() def RemoveFile(self, fname=None, **kws): if fname is not None: s = str(fname) if s in self.controller.file_groups: group = self.controller.file_groups.pop(s) def ShowFile(self, evt=None, groupname=None, **kws): filename = None if evt is not None: filename = str(evt.GetString()) if groupname is None and filename is not None: groupname = self.controller.file_groups[filename] if not hasattr(self.larch.symtable, groupname): return dgroup = self.controller.get_group(groupname) self.controller.group = dgroup self.controller.groupname = groupname self.nb.SetSelection(0) self.proc_panel.fill(dgroup) if filename is None: filename = dgroup.filename self.title.SetLabel(filename) def createMenus(self): # ppnl = self.plotpanel self.menubar = wx.MenuBar() # fmenu = wx.Menu() MenuItem(self, fmenu, "&Open Data File\tCtrl+O", "Open Data File", self.onReadDialog) fmenu.AppendSeparator() MenuItem(self, fmenu, 'Show Larch Buffer\tCtrl+L', 'Show Larch Programming Buffer', self.onShowLarchBuffer) MenuItem(self, fmenu, "debug wx\tCtrl+I", "", self.showInspectionTool) MenuItem(self, fmenu, "&Quit\tCtrl+Q", "Quit program", self.onCloseNicely) self.menubar.Append(fmenu, "&File") omenu = wx.Menu() self.menubar.Append(omenu, "Options") MenuItem(self, omenu, "Configure Data Processing", "Configure Data Processing", self.onConfigDataProcessing) MenuItem(self, omenu, "Configure Data Fitting", "Configure Data Fitting", self.onConfigDataFitting) self.SetMenuBar(self.menubar) self.Bind(wx.EVT_CLOSE, self.onExit) def onShowLarchBuffer(self, evt=None): if self.larch_buffer is None: self.larch_buffer = LarchFrame(_larch=self.larch) self.larch_buffer.Show() self.larch_buffer.Raise() def onConfigDataProcessing(self, event=None): pass def onConfigDataFitting(self, event=None): pass def showInspectionTool(self, event=None): app = wx.GetApp() app.ShowInspectionTool() def onAbout(self,evt): dlg = wx.MessageDialog(self, self._about, "About Larch XYFit", wx.OK | wx.ICON_INFORMATION) dlg.ShowModal() dlg.Destroy() def onExit(self, evt): for nam in dir(self.larch.symtable._plotter): obj = getattr(self.larch.symtable._plotter, nam) time.sleep(0.05) try: obj.Destroy() except: pass for nam in dir(self.larch.symtable._sys.wx): obj = getattr(self.larch.symtable._sys.wx, nam) time.sleep(0.05) del obj if self.larch_buffer is not None: try: self.larch_buffer.Destroy() except: pass self.Destroy() def onCloseNicely(self, evt): dlg = wx.MessageDialog(None, 'Really Quit?', 'Question', wx.YES_NO | wx.NO_DEFAULT | wx.ICON_QUESTION) if wx.ID_YES != dlg.ShowModal(): return self.controller.save_config() self.proc_panel.proc_timer.Stop() time.sleep(0.05) self.onExit(evt) def show_subframe(self, name, frameclass, **opts): shown = False if name in self.subframes: try: self.subframes[name].Raise() shown = True except: del self.subframes[name] if not shown: self.subframes[name] = frameclass(self, **opts) def onSelectColumns(self, evt=None): dgroup = self.controller.get_group(self.controller.groupname) self.show_subframe('readfile', ColumnDataFileFrame, group=dgroup.raw, last_array_sel=self.last_array_sel, _larch=self.larch, read_ok_cb=partial(self.onRead_OK, overwrite=True)) def onReadDialog(self, evt=None): dlg = wx.FileDialog(self, message="Read Data File", defaultDir=os.getcwd(), wildcard=FILE_WILDCARDS, style=wx.FD_OPEN|wx.FD_MULTIPLE) self.paths2read = [] if dlg.ShowModal() == wx.ID_OK: self.paths2read = dlg.GetPaths() dlg.Destroy() if len(self.paths2read) < 1: return path = self.paths2read.pop(0) path = path.replace('\\', '/') do_read = True if path in self.controller.file_groups: do_read = (wx.ID_YES == popup(self, "Re-read file '%s'?" % path, 'Re-read file?')) if do_read: self.onRead(path) def onRead(self, path): filedir, filename = os.path.split(path) if self.controller.get_config('chdir_on_fileopen'): os.chdir(filedir) self.controller.set_workdir() # check for athena projects if is_athena_project(path): kwargs = dict(filename=path, _larch = self.controller.larch, read_ok_cb=self.onReadAthenaProject_OK) self.show_subframe('athena_import', AthenaImporter, **kwargs) else: kwargs = dict(filename=path, _larch=self.larch_buffer.larchshell, last_array_sel = self.last_array_sel, read_ok_cb=self.onRead_OK) self.show_subframe('readfile', ColumnDataFileFrame, **kwargs) def onReadAthenaProject_OK(self, path, namelist): """read groups from a list of groups from an athena project file""" self.larch.eval("_prj = read_athena('{path:s}', do_fft=False, do_bkg=False)".format(path=path)) s = """{group:s} = _prj.{group:s} {group:s}.datatype = 'xas' {group:s}.x = {group:s}.xdat = {group:s}.energy {group:s}.y = {group:s}.ydat = {group:s}.mu {group:s}.yerr = 1.0 {group:s}.plot_ylabel = 'mu' {group:s}.plot_xlabel = 'energy' """ for gname in namelist: self.larch.eval(s.format(group=gname)) self.install_group(gname, gname) self.larch.eval("del _prj") def onRead_OK(self, script, path, groupname=None, array_sel=None, overwrite=False): """ called when column data has been selected and is ready to be used overwrite: whether to overwrite the current datagroup, as when editing a datagroup """ abort_read = False filedir, filename = os.path.split(path) if not overwrite and hasattr(self.larch.symtable, groupname): newname = file2groupname(filename, symtable=self.larch.symtable) msg = """Warning: groupname '%s' already used. Will use groupname '%s' instead """ % (groupname, newname) dlg = wx.MessageDialog(self, msg, 'Warning', wx.OK | wx.CANCEL ) abort_read = (wx.ID_OK != dlg.ShowModal()) dlg.Destroy() groupname = newname if abort_read: return self.larch.eval(script.format(group=groupname, path=path)) if array_sel is not None: self.last_array_sel = array_sel self.install_group(groupname, filename) if len(self.paths2read) > 0 : path = self.paths2read.pop(0) path = path.replace('\\', '/') filedir, filename = os.path.split(path) gname = file2groupname(filename, symtable=self.larch.symtable) self.onRead_OK(script, path, groupname=gname) def install_group(self, groupname, filename): """add groupname / filename to list of available data groups""" thisgroup = getattr(self.larch.symtable, groupname) thisgroup.groupname = groupname thisgroup.filename = filename datatype = getattr(thisgroup, 'datatype', 'raw') # file /group may already exist in list if filename in self.controller.file_groups and not overwrite: for i in range(1, 101): ftest = "%s (%i)" % (filename, i) if ftest not in self.controller.file_groups: filename = ftest break if filename not in self.controller.file_groups: self.controller.filelist.Append(filename) self.controller.file_groups[filename] = groupname self.nb.SetSelection(0) self.ShowFile(groupname=groupname)
class XASFrame(wx.Frame): _about = """Larch XAS GUI: XAS Visualization and Analysis Matt Newville <newville @ cars.uchicago.edu> """ def __init__(self, parent=None, _larch=None, **kws): wx.Frame.__init__(self, parent, -1, size=XASVIEW_SIZE, style=FRAMESTYLE) self.last_array_sel = {} self.paths2read = [] title = "Larch XAS GUI: XAS Visualization and Analysis" self.larch_buffer = parent if not isinstance(parent, LarchFrame): self.larch_buffer = LarchFrame(_larch=_larch) self.larch_buffer.Show() self.larch_buffer.Raise() self.larch=self.larch_buffer.larchshell self.controller = XASController(wxparent=self, _larch=self.larch) self.current_filename = None self.subframes = {} self.plotframe = None self.SetTitle(title) self.SetSize(XASVIEW_SIZE) self.SetFont(Font(10)) self.larch_buffer.Hide() self.createMainPanel() self.createMenus() self.statusbar = self.CreateStatusBar(2, 0) self.statusbar.SetStatusWidths([-3, -1]) statusbar_fields = ["Initializing....", " "] for i in range(len(statusbar_fields)): self.statusbar.SetStatusText(statusbar_fields[i], i) 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 onNBChanged(self, event=None): idx = self.nb.GetSelection() pan = self.nb_panels[idx] callback = getattr(pan, 'onPanelExposed', None) if callable(callback): callback() def onSelAll(self, event=None): self.controller.filelist.SetCheckedStrings(self.controller.file_groups.keys()) def onSelNone(self, event=None): self.controller.filelist.SetCheckedStrings([]) def init_larch(self): self.SetStatusText('initializing Larch') self.title.SetLabel('') self.controller.init_larch() plotframe = self.controller.get_display(stacked=False) xpos, ypos = self.GetPosition() xsiz, ysiz = self.GetSize() plotframe.SetPosition((xpos+xsiz+5, ypos)) self.SetStatusText('ready') self.Raise() def write_message(self, s, panel=0): """write a message to the Status Bar""" self.SetStatusText(s, panel) def RemoveFile(self, fname=None, **kws): if fname is not None: s = str(fname) if s in self.controller.file_groups: group = self.controller.file_groups.pop(s) def ShowFile(self, evt=None, groupname=None, with_plot=True, **kws): filename = None if evt is not None: # filename = str(evt.GetString()) # could have clicking on filename add to "Selected": # checked = list(self.controller.filelist.GetCheckedStrings()) # if filename not in checked: # checked.append(filename) # self.controller.filelist.SetCheckedStrings(checked) if groupname is None and filename is not None: groupname = self.controller.file_groups[filename] if not hasattr(self.larch.symtable, groupname): return dgroup = self.controller.get_group(groupname) self.controller.group = dgroup self.controller.groupname = groupname current_nbpanel = self.nb_panels[self.nb.GetSelection()] self.nb_panels[0].fill_form(dgroup) if with_plot: self.nb_panels[0].plot(dgroup) if filename is None: filename = dgroup.filename self.title.SetLabel(filename) self.current_filename = filename def createMenus(self): # ppnl = self.plotpanel self.menubar = wx.MenuBar() # fmenu = wx.Menu() group_menu = wx.Menu() data_menu = wx.Menu() ppeak_menu = wx.Menu() m = {} MenuItem(self, fmenu, "&Open Data File\tCtrl+O", "Open Data File", self.onReadDialog) MenuItem(self, fmenu, "&Save Project\tCtrl+S", "Save Session to Project File", self.onSaveProject) MenuItem(self, fmenu, "Export Selected Groups to Athena Project", "Export Selected Groups to Athena Project", self.onExportAthena) MenuItem(self, fmenu, "Export Selected Groups to CSV", "Export Selected Groups to CSV", self.onExportCSV) fmenu.AppendSeparator() MenuItem(self, fmenu, 'Show Larch Buffer\tCtrl+L', 'Show Larch Programming Buffer', self.onShowLarchBuffer) # MenuItem(self, fmenu, "&Inspect \tCtrl+I", # "e", self.showInspectionTool) MenuItem(self, fmenu, "&Quit\tCtrl+Q", "Quit program", self.onClose) MenuItem(self, group_menu, "Copy This Group", "Copy This Group", self.onCopyGroup) MenuItem(self, group_menu, "Rename This Group", "Rename This Group", self.onRenameGroup) MenuItem(self, group_menu, "Remove Selected Groups", "Remove Selected Group", self.onRemoveGroups) MenuItem(self, group_menu, "Merge Selected Groups", "Merge Selected Groups", self.onMergeData) MenuItem(self, data_menu, "Deglitch Data", "Deglitch Data", self.onDeglitchData) MenuItem(self, data_menu, "Recalibrate Energy", "Recalibrate Energy", self.onEnergyCalibrateData) MenuItem(self, data_menu, "Smooth Data", "Smooth Data", self.onSmoothData) MenuItem(self, data_menu, "Rebin Data", "Rebin Data", self.onRebinData) MenuItem(self, data_menu, "Correct Over-absorption", "Correct Over-absorption", self.onCorrectOverAbsorptionData) MenuItem(self, ppeak_menu, "&Read Fit Model\tCtrl+R", "Read Fit Model from File", self.onLoadFitResult) fsave = MenuItem(self, ppeak_menu, "Save Fit Model", "Save Fit Model to File", self.onSaveFitResult) fexport = MenuItem(self, ppeak_menu, "Export Data and Fit", "Export Data and Fit", self.onExportFitResult) self.afterfit_menus = (fsave, fexport) for m in self.afterfit_menus: m.Enable(False) self.menubar.Append(fmenu, "&File") self.menubar.Append(group_menu, "Groups") self.menubar.Append(data_menu, "Data") self.menubar.Append(ppeak_menu, "PreEdgePeaks") self.SetMenuBar(self.menubar) self.Bind(wx.EVT_CLOSE, self.onClose) def onShowLarchBuffer(self, evt=None): if self.larch_buffer is None: self.larch_buffer = LarchFrame(_larch=self.larch) self.larch_buffer.Show() self.larch_buffer.Raise() def onExportCSV(self, evt=None): group_ids = self.controller.filelist.GetCheckedStrings() savegroups = [] groupnames = [] for checked in group_ids: groupname = self.controller.file_groups[str(checked)] dgroup = self.controller.get_group(groupname) savegroups.append(dgroup) groupnames.append(groupname) if len(savegroups) < 1: Popup(self, "No files selected to export to CSV", "No files selected") return deffile = "%s_%i.csv" % (groupname, len(groupnames)) wcards = 'CSV Files (*.csv)|*.cvs|All files (*.*)|*.*' outfile = FileSave(self, 'Export Selected Groups to CSV File', default_file=deffile, wildcard=wcards) if outfile is None: return groups2csv(savegroups, outfile, x='energy', y='norm', _larch=self.larch) def onExportAthena(self, evt=None): groups = [] for checked in self.controller.filelist.GetCheckedStrings(): groups.append(self.controller.file_groups[str(checked)]) if len(groups) < 1: Popup(self, "No files selected to export to Athena", "No files selected") return self.save_athena_project(groups[0], groups, prompt=True) def onSaveProject(self, evt=None): groups = [gname for gname in self.controller.file_groups] if len(groups) < 1: Popup(self, "No files selected to export to Athena", "No files selected") return self.save_athena_project(groups[0], groups, prompt=True) def save_athena_project(self, filename, grouplist, prompt=True): if len(grouplist) < 1: return savegroups = [self.controller.get_group(gname) for gname in grouplist] deffile = "%s_%i.prj" % (filename, len(grouplist)) wcards = 'Athena Projects (*.prj)|*.prj|All files (*.*)|*.*' outfile = FileSave(self, 'Save Groups to Athena Project File', default_file=deffile, wildcard=wcards) if outfile is None: return aprj = AthenaProject(filename=outfile, _larch=self.larch) for label, grp in zip(grouplist, savegroups): aprj.add_group(grp, label=label) aprj.save(use_gzip=True) def onConfigDataProcessing(self, event=None): pass def onCopyGroup(self, event=None): fname = self.current_filename if fname is None: fname = self.current_filename = self.controller.filelist.GetStringSelection() groupname = self.controller.file_groups[fname] if not hasattr(self.larch.symtable, groupname): return ogroup = self.controller.get_group(groupname) ngroup = Group(datatype=ogroup.datatype, energy=1.0*ogroup.energy, mu=1.0*ogroup.mu, xdat=1.0*ogroup.energy, ydat=1.0*ogroup.mu) for attr in dir(ogroup): if attr in ('i0', 'data' 'yerr'): val = getattr(ogroup, attr)*1.0 if attr in ('norm', 'flat', 'deriv', 'deconv', 'post_edge', 'pre_edge'): pass else: try: val = copy.deepcopy(getattr(ogroup, attr)) except ValueError: val = None setattr(ngroup, attr, val) new_fname = unique_name(fname, self.controller.file_groups.keys()) new_gname = unique_name(groupname, self.controller.file_groups.values()) setattr(self.larch.symtable, new_gname, ngroup) self.install_group(new_gname, new_fname, overwrite=False) self.nb_panels[0].process(ngroup) self.ShowFile(groupname=new_gname) def onRenameGroup(self, event=None): fname = self.current_filename = self.controller.filelist.GetStringSelection() if fname is None: return dlg = RenameDialog(self, fname) res = dlg.GetResponse() dlg.Destroy() if res.ok: groupname = self.controller.file_groups.pop(fname) self.controller.file_groups[res.newname] = groupname self.controller.filelist.rename_item(self.current_filename, res.newname) dgroup = self.controller.get_group(groupname) dgroup.filename = self.current_filename = res.newname self.controller.filelist.SetStringSelection(res.newname) def onRemoveGroups(self, event=None): groups = [] for checked in self.controller.filelist.GetCheckedStrings(): groups.append(self.controller.file_groups[str(checked)]) if len(groups) < 1: return dlg = RemoveDialog(self, groups) res = dlg.GetResponse() dlg.Destroy() if res.ok: filelist = self.controller.filelist all_fnames = filelist.GetItems() for fname in groups: gname = self.controller.file_groups.pop(fname) delattr(self.controller.symtable, gname) all_fnames.remove(fname) filelist.Clear() for name in all_fnames: filelist.Append(name) def onMergeData(self, event=None): groups = [] for checked in self.controller.filelist.GetCheckedStrings(): groups.append(self.controller.file_groups[str(checked)]) if len(groups) < 1: return outgroup = unique_name('merge', self.controller.file_groups) dlg = MergeDialog(self, groups, outgroup=outgroup) res = dlg.GetResponse() dlg.Destroy() if res.ok: fname = res.group gname = fix_varname(res.group.lower()) yname = 'norm' if res.ynorm else 'mu' self.controller.merge_groups(groups, master=res.master, yarray=yname, outgroup=gname) self.install_group(gname, fname, overwrite=False) self.controller.filelist.SetStringSelection(fname) def onDeglitchData(self, event=None): DeglitchDialog(self, self.controller).Show() def onSmoothData(self, event=None): SmoothDataDialog(self, self.controller).Show() def onRebinData(self, event=None): RebinDataDialog(self, self.controller).Show() def onCorrectOverAbsorptionData(self, event=None): OverAbsorptionDialog(self, self.controller).Show() def onEnergyCalibrateData(self, event=None): EnergyCalibrateDialog(self, self.controller).Show() def onConfigDataFitting(self, event=None): pass def showInspectionTool(self, event=None): app = wx.GetApp() app.ShowInspectionTool() def onAbout(self,evt): dlg = wx.MessageDialog(self, self._about, "About Larch XAS GUI", wx.OK | wx.ICON_INFORMATION) dlg.ShowModal() dlg.Destroy() def onClose(self, event): dlg = QuitDialog(self) res = dlg.GetResponse() dlg.Destroy() if not res.ok: return if res.save: groups = [gname for gname in self.controller.file_groups] if len(groups) > 0: self.save_athena_project(groups[0], groups, prompt=True) self.controller.save_config() self.controller.get_display().Destroy() if self.larch_buffer is not None: try: self.larch_buffer.Destroy() except: pass time.sleep(0.05) for nam in dir(self.larch.symtable._plotter): obj = getattr(self.larch.symtable._plotter, nam) time.sleep(0.05) try: obj.Destroy() except: pass for name, wid in self.subframes.items(): if wid is not None: try: wid.Destroy() except: pass for nam in dir(self.larch.symtable._sys.wx): obj = getattr(self.larch.symtable._sys.wx, nam) self.Destroy() def show_subframe(self, name, frameclass, **opts): shown = False if name in self.subframes: try: self.subframes[name].Raise() shown = True except: del self.subframes[name] if not shown: self.subframes[name] = frameclass(self, **opts) def onSelectColumns(self, event=None): dgroup = self.controller.get_group() self.show_subframe('readfile', ColumnDataFileFrame, group=dgroup.raw, last_array_sel=self.last_array_sel, _larch=self.larch, read_ok_cb=partial(self.onRead_OK, overwrite=True)) def onLoadFitResult(self, event=None): self.nb.SetSelection(1) self.nb_panels[1].onLoadFitResult(event=event) def onSaveFitResult(self, event=None): self.nb_panels[1].onSaveFitResult(event=event) def onExportFitResult(self, event=None): self.nb_panels[1].onExportFitResult(event=event) def onReadDialog(self, event=None): dlg = wx.FileDialog(self, message="Read Data File", defaultDir=os.getcwd(), wildcard=FILE_WILDCARDS, style=wx.FD_OPEN|wx.FD_MULTIPLE) self.paths2read = [] if dlg.ShowModal() == wx.ID_OK: self.paths2read = dlg.GetPaths() dlg.Destroy() if len(self.paths2read) < 1: return path = self.paths2read.pop(0) path = path.replace('\\', '/') do_read = True if path in self.controller.file_groups: do_read = (wx.ID_YES == Popup(self, "Re-read file '%s'?" % path, 'Re-read file?')) if do_read: self.onRead(path) def onRead(self, path): filedir, filename = os.path.split(path) if self.controller.get_config('chdir_on_fileopen'): os.chdir(filedir) self.controller.set_workdir() # check for athena projects if is_athena_project(path): kwargs = dict(filename=path, _larch = self.controller.larch, read_ok_cb=self.onReadAthenaProject_OK) self.show_subframe('athena_import', AthenaImporter, **kwargs) else: kwargs = dict(filename=path, _larch=self.larch_buffer.larchshell, last_array_sel = self.last_array_sel, read_ok_cb=self.onRead_OK) self.show_subframe('readfile', ColumnDataFileFrame, **kwargs) def onReadAthenaProject_OK(self, path, namelist): """read groups from a list of groups from an athena project file""" self.larch.eval("_prj = read_athena('{path:s}', do_fft=False, do_bkg=False)".format(path=path)) dgroup = None s = """ {group:s} = _prj.{group:s} {group:s}.datatype = 'xas' {group:s}.xdat = 1.0*{group:s}.energy {group:s}.ydat = 1.0*{group:s}.mu {group:s}.yerr = 1.0 {group:s}.plot_ylabel = 'mu' {group:s}.plot_xlabel = 'energy' """ for gname in namelist: self.larch.eval(s.format(group=gname)) dgroup = self.install_group(gname, gname, with_plot=False) # self.nb_panels[0].fill_form(dgroup) if dgroup is not None: self.nb_panels[0].plot(dgroup) self.larch.eval("del _prj") def onRead_OK(self, script, path, groupname=None, array_sel=None, overwrite=False): """ called when column data has been selected and is ready to be used overwrite: whether to overwrite the current datagroup, as when editing a datagroup """ abort_read = False filedir, filename = os.path.split(path) if not overwrite and hasattr(self.larch.symtable, groupname): newname = file2groupname(filename, symtable=self.larch.symtable) msg = """Warning: groupname '%s' already used. Will use groupname '%s' instead """ % (groupname, newname) dlg = wx.MessageDialog(self, msg, 'Warning', wx.OK | wx.CANCEL ) abort_read = (wx.ID_OK != dlg.ShowModal()) dlg.Destroy() groupname = newname if abort_read: return self.larch.eval(script.format(group=groupname, path=path)) if array_sel is not None: self.last_array_sel = array_sel self.install_group(groupname, filename, overwrite=overwrite) for path in self.paths2read: path = path.replace('\\', '/') filedir, filename = os.path.split(path) gname = file2groupname(filename, symtable=self.larch.symtable) self.larch.eval(script.format(group=gname, path=path)) self.install_group(gname, filename, overwrite=True) def install_group(self, groupname, filename, overwrite=False, with_plot=True): """add groupname / filename to list of available data groups""" thisgroup = getattr(self.larch.symtable, groupname) thisgroup.groupname = groupname thisgroup.filename = filename datatype = getattr(thisgroup, 'datatype', 'raw') # file /group may already exist in list if filename in self.controller.file_groups and not overwrite: for i in range(1, 101): ftest = "%s (%i)" % (filename, i) if ftest not in self.controller.file_groups: filename = ftest break if filename not in self.controller.file_groups: self.controller.filelist.Append(filename) self.controller.file_groups[filename] = groupname self.nb.SetSelection(0) self.ShowFile(groupname=groupname, with_plot=with_plot) self.controller.filelist.SetStringSelection(filename) return thisgroup