def __init__(self, *args, **kwargs):
super(GlobalScene, self).__init__(*args, **kwargs)
# Sync self to base_app in several objects
self.opticstate = DielectricSlab(base_app=self)
self.simulations.append(
LayerSimulation(base_app=self, outname=config.SIMPREFIX + '0'))
def __init__(self, *args, **kwargs):
super(GlobalScene, self).__init__(*args, **kwargs)
# Sync self to base_app in several objects
self.opticstate = DielectricSlab(base_app=self)
self.simulations.append(LayerSimulation(base_app=self, outname=config.SIMPREFIX + "0"))
class GlobalScene(HasTraits):
"""Global class to define all view-based stuff"""
specparms = Instance(HasTraits, SHARED_SPECPARMS)
fiberparms = Instance(AngleParms)
lambdas = DelegatesTo("specparms") # Actually not used except for making it easy to run sims
plot_selector = Instance(PlotSelector)
fview = Instance(ViewMlab) # May want to pass specparms and fiberparms to this later if it requries them
stratastyle = Enum("Fiber", "Glass Slide")
# Dielectric state probably could be in SHARED, but don't want to take apart
current_state = Instance(IOptic)
opticstate = Instance(IOptic)
opticview = DelegatesTo("opticstate")
save = Button
load = Button
# Simulation
sim_outdir = Directory
simulations = List(ISim)
selected_sim = Instance(ISim)
configure_storage = Instance(SimConfigure) # <--- Want all sims to share this, right?
# Editors##
layereditor = Instance(HasTraits, SHARED_LAYEREDITOR)
stack = DelegatesTo(
"layereditor"
) # Variables are stored here just because they can be useful for future implementations
selected_layer = DelegatesTo("layereditor")
selected_material = DelegatesTo("layereditor")
selected_d = DelegatesTo("layereditor")
# angle_avg=DelegatesTo('current_state')
# Actions
showreflectance = Action(
name="Popout Optics Plot", action="popout_optics"
) # PHASE THIS OUT LATER WITH UNIFIED VIEW FRAMEWORK
showmaterial = Action(name="Popout Material Plot", action="popout_material")
appendsim = Action(name="Add Simulation", action="new_sim")
savesim = Action(name="Save Selected Simulation", action="save_sim") # action gets underscore
savesim_all = Action(name="Save All Simulation", action="save_allsims") # action gets underscore
# Make Menubar
mainmenu = MenuBar(
Menu(showmaterial, showreflectance, name="Plot Options"),
Menu(appendsim, savesim, savesim_all, name="Simulation Options"),
)
fibergroup = Group(
# Angle_avg depends on the DielectricSlab
# Item('angle_avg', label='Angle Averaging Method',show_label=False),
Item("fiberparms", editor=InstanceEditor(), style="custom", show_label=False),
Item("fview", style="custom", show_label=False),
label=globalparms.strataname,
)
layergroup = Group(
Item("layereditor", editor=InstanceEditor(), style="custom", show_label=False), label=globalparms.stackname
)
materialgroup = Group(
Item("selected_material", editor=InstanceEditor(), style="custom", show_label=False),
label=globalparms.materialname,
)
# Main Panel
# -----------
spectralgroup = Group(Item("specparms", show_label=False, style="custom"), label="Spectral Parameters")
stratagroup = Group(Item("stratastyle", show_label=False), label="Choose Substrate Type")
choosesimgroup = Group(
Item("simulations", editor=sims_editor, show_label=False),
# Can't remove this or program trips
Item("opticstate", editor=state_editor, show_label=False, visible_when="8==9"), # always vis
label="Choose Simulation",
)
simgroup = Group(
Item("selected_sim", style="custom", editor=InstanceEditor(), show_label=False), label="Simulations"
)
maingroup = Group(
Include("spectralgroup"),
Include("stratagroup"),
Include("choosesimgroup"), # simulation and summary
label=globalparms.globsname,
)
fullgroup = VSplit(
HSplit(
VGroup(Item("plot_selector", show_label=False, style="custom")),
# PLOT
VGroup(Item("opticview", style="custom", show_label=False)),
),
Tabbed(
Include("maingroup"),
Include("fibergroup"),
Include("layergroup"),
Include("materialgroup"),
Include("simgroup"),
),
)
Mainview = View(
Include("fullgroup"),
menubar=mainmenu,
resizable=True,
buttons=["Undo"],
title="Plasmonic Assay Modeling Environment",
)
def __init__(self, *args, **kwargs):
super(GlobalScene, self).__init__(*args, **kwargs)
# Sync self to base_app in several objects
self.opticstate = DielectricSlab(base_app=self)
self.simulations.append(LayerSimulation(base_app=self, outname=config.SIMPREFIX + "0"))
def _load_fired(self):
self.simulations = pickle.load(open("test.p", "rb"))
def _save_fired(self):
pickle.dump(self.simulations, open("test.p", "wb"))
def _plot_selector_default(self):
return PlotSelector()
def _configure_storage_default(self):
return SimConfigure()
# Where should this point? WHAT IF DOESN"T EXIST
def _sim_outdir_default(self):
if op.exists(config.SIMFOLDER):
return config.SIMFOLDER
else:
return op.abspath(".")
# Store copy of current simulation
def new_sim(self):
self.simulations.append(
LayerSimulation(
base_app=self, # <--- LayerSimulation. Base_app = self.copy?
outname=config.SIMPREFIX + str(len(self.simulations)),
)
)
def save_sim(self):
self.selected_sim.save() # <--- NEED TO CALL TO_JSON NOT OUTPUT_SIMULATION
def save_allsims(self):
""" Saves all stored simulations in the sims_editor. Checks for duplicate names and non-run/incomplete
simulations and prompts user accordingly."""
# Check to make sure all simulations have completed data
unrun = [s.outname for s in self.simulations if s._completed == False]
nrunstring = " ".join(unrun)
if len(unrun) > 0:
message("Cannot save simulations: %s. Results not found." % nrunstring, title="Warning")
# Can't save either way, so force exit instead of user being able to continue
return
# Check for duplicate runnames
rnames = [s.outname for s in self.simulations]
non_uniq = [r for r in rnames if rnames.count(r) > 1]
if len(non_uniq) > 0:
non_uniq = list(set(non_uniq))
nustring = " ".join(non_uniq)
message("Duplicate simulation outfile names found: %s." % nustring, title="Warning")
return
# Output completed simulations
outsims = [s for s in self.simulations if s not in unrun]
for s in outsims:
s.save(confirmwindow=False)
message(
'%s simulation(s) saved to directory: "%s"' % (len(outsims), op.split(self.sim_outdir)[1]), title="Success"
)
# Pop Reflectance/Material Plots
# ------------------------
def popout_optics(self):
""" Refresh and popup optics plot """
self.opticstate.opticview.edit_traits()
def popout_material(self):
# http://code.enthought.com/projects/files/ets_api/enthought.traits.ui.view.html
# what kind should these use
self.plot_selector.edit_traits() # kind='panel')
def _stratastyle_changed(self):
if self.stratastyle == "Fiber":
self.fiberparms = FiberParms()
self.fview = FiberView()
elif self.stratastyle == "Glass Slide":
self.fiberparms = EllipsometryParms()
self.fview = EllispometryView()
# Refresh plot because; otherwise can get misleading behavior
self.opticstate.update_opticview()
# Strata parmaeters
# -----------------
def _stratastyle_default(self):
return "Fiber"
def _fiberparms_default(self):
return FiberParms()
def _fview_default(self):
return FiberView()
class GlobalScene(HasTraits):
'''Global class to define all view-based stuff'''
specparms = Instance(HasTraits, SHARED_SPECPARMS)
fiberparms = Instance(AngleParms)
lambdas = DelegatesTo(
'specparms') #Actually not used except for making it easy to run sims
plot_selector = Instance(PlotSelector)
fview = Instance(
ViewMlab
) #May want to pass specparms and fiberparms to this later if it requries them
stratastyle = Enum('Fiber', 'Glass Slide')
# Dielectric state probably could be in SHARED, but don't want to take apart
current_state = Instance(IOptic)
opticstate = Instance(IOptic)
opticview = DelegatesTo('opticstate')
save = Button
load = Button
#Simulation
sim_outdir = Directory
simulations = List(ISim)
selected_sim = Instance(ISim)
configure_storage = Instance(
SimConfigure) #<--- Want all sims to share this, right?
#Editors##
layereditor = Instance(HasTraits, SHARED_LAYEREDITOR)
stack = DelegatesTo(
'layereditor'
) #Variables are stored here just because they can be useful for future implementations
selected_layer = DelegatesTo('layereditor')
selected_material = DelegatesTo('layereditor')
selected_d = DelegatesTo('layereditor')
# angle_avg=DelegatesTo('current_state')
#Actions
showreflectance = Action(
name="Popout Optics Plot", action="popout_optics"
) #PHASE THIS OUT LATER WITH UNIFIED VIEW FRAMEWORK
showmaterial = Action(name='Popout Material Plot',
action="popout_material")
appendsim = Action(name="Add Simulation", action="new_sim")
savesim = Action(name="Save Selected Simulation",
action="save_sim") #action gets underscore
savesim_all = Action(name="Save All Simulation",
action="save_allsims") #action gets underscore
# Make Menubar
mainmenu = MenuBar(
Menu(showmaterial, showreflectance, name='Plot Options'),
Menu(appendsim, savesim, savesim_all, name='Simulation Options'),
)
fibergroup = Group(
# Angle_avg depends on the DielectricSlab
# Item('angle_avg', label='Angle Averaging Method',show_label=False),
Item('fiberparms',
editor=InstanceEditor(),
style='custom',
show_label=False),
Item('fview', style='custom', show_label=False),
label=globalparms.strataname)
layergroup = Group(Item('layereditor',
editor=InstanceEditor(),
style='custom',
show_label=False),
label=globalparms.stackname)
materialgroup = Group(Item('selected_material',
editor=InstanceEditor(),
style='custom',
show_label=False),
label=globalparms.materialname)
# Main Panel
#-----------
spectralgroup = Group(Item('specparms', show_label=False, style='custom'),
label='Spectral Parameters')
stratagroup = Group(Item('stratastyle', show_label=False),
label='Choose Substrate Type')
choosesimgroup = Group(
Item('simulations', editor=sims_editor, show_label=False),
# Can't remove this or program trips
Item(
'opticstate',
editor=state_editor,
show_label=False,
visible_when='8==9' #always vis
),
label='Choose Simulation')
simgroup = Group(Item('selected_sim',
style='custom',
editor=InstanceEditor(),
show_label=False),
label='Simulations')
maingroup = Group(
Include('spectralgroup'),
Include('stratagroup'),
Include('choosesimgroup'), #simulation and summary
label=globalparms.globsname,
)
fullgroup = VSplit(
HSplit(
VGroup(Item('plot_selector', show_label=False, style='custom'), ),
# PLOT
VGroup(Item('opticview', style='custom', show_label=False), )),
Tabbed(
Include('maingroup'),
Include('fibergroup'),
Include('layergroup'),
Include('materialgroup'),
Include('simgroup'),
),
)
Mainview = View(Include('fullgroup'),
menubar=mainmenu,
resizable=True,
buttons=['Undo'],
title='Plasmonic Assay Modeling Environment')
def __init__(self, *args, **kwargs):
super(GlobalScene, self).__init__(*args, **kwargs)
# Sync self to base_app in several objects
self.opticstate = DielectricSlab(base_app=self)
self.simulations.append(
LayerSimulation(base_app=self, outname=config.SIMPREFIX + '0'))
def _load_fired(self):
self.simulations = pickle.load(open("test.p", "rb"))
def _save_fired(self):
pickle.dump(self.simulations, open("test.p", "wb"))
def _plot_selector_default(self):
return PlotSelector()
def _configure_storage_default(self):
return SimConfigure()
# Where should this point? WHAT IF DOESN"T EXIST
def _sim_outdir_default(self):
if op.exists(config.SIMFOLDER):
return config.SIMFOLDER
else:
return op.abspath('.')
# Store copy of current simulation
def new_sim(self):
self.simulations.append(
LayerSimulation(
base_app=self, #<--- LayerSimulation. Base_app = self.copy?
outname=config.SIMPREFIX + str(len(self.simulations))))
def save_sim(self):
self.selected_sim.save(
) #<--- NEED TO CALL TO_JSON NOT OUTPUT_SIMULATION
def save_allsims(self):
''' Saves all stored simulations in the sims_editor. Checks for duplicate names and non-run/incomplete
simulations and prompts user accordingly.'''
# Check to make sure all simulations have completed data
unrun = [s.outname for s in self.simulations if s._completed == False]
nrunstring = ' '.join(unrun)
if len(unrun) > 0:
message('Cannot save simulations: %s. Results not found.' %
nrunstring,
title='Warning')
# Can't save either way, so force exit instead of user being able to continue
return
# Check for duplicate runnames
rnames = [s.outname for s in self.simulations]
non_uniq = [r for r in rnames if rnames.count(r) > 1]
if len(non_uniq) > 0:
non_uniq = list(set(non_uniq))
nustring = ' '.join(non_uniq)
message('Duplicate simulation outfile names found: %s.' % nustring,
title='Warning')
return
# Output completed simulations
outsims = [s for s in self.simulations if s not in unrun]
for s in outsims:
s.save(confirmwindow=False)
message('%s simulation(s) saved to directory: "%s"' %
(len(outsims), op.split(self.sim_outdir)[1]),
title='Success')
# Pop Reflectance/Material Plots
# ------------------------
def popout_optics(self):
""" Refresh and popup optics plot """
self.opticstate.opticview.edit_traits()
def popout_material(self):
#http://code.enthought.com/projects/files/ets_api/enthought.traits.ui.view.html
# what kind should these use
self.plot_selector.edit_traits() #kind='panel')
def _stratastyle_changed(self):
if self.stratastyle == 'Fiber':
self.fiberparms = FiberParms()
self.fview = FiberView()
elif self.stratastyle == 'Glass Slide':
self.fiberparms = EllipsometryParms()
self.fview = EllispometryView()
# Refresh plot because; otherwise can get misleading behavior
self.opticstate.update_opticview()
# Strata parmaeters
# -----------------
def _stratastyle_default(self):
return 'Fiber'
def _fiberparms_default(self):
return FiberParms()
def _fview_default(self):
return FiberView()
