def __init__(self, master=None, model=None, **kwargs):
BaseGui.__init__(self, master, title="Navigational Solver UI", menus=["File", "Build", "Run", "Display", "Debug"], pycommand=not bool(master),height=4, showframe=False)
if master:
self.document=master.document
if model:
self.alg=model
else:
self.alg=None
controls=[
["Build", "Select/Create Model", self.setModel],
["Build", "Set Bias Field", self.setBias],
["Build", "Configure Bias Field Conditioning", self.confBias],
["Build", "Save Conditioned Field as Base", self.swapBias],
["Build", "Set Start and Target Points", self.setGoal],
["Build", "Choose Algorithm", self.setAlg],
["Build", "Configure Algorithm", self.config],
["Run", "Full Solve", self.runAll],
["Display", "Show Raw Bias Field", self.rawBias],
["Display", "Show Conditioned Bias Field", self.condBias],
["Debug", "Reload Blocks", self.bounceDSP],
["Debug", "Launch Debugger", self.debug],
]
if not master:
controls.append(["Build", 'Launch Data Editor', self.launchEditor])
self.fillMenus(controls)
self.stdFileMenu()
self.biasmode='raw'
self.displaytype='bias'
self.mainSizer = wx.BoxSizer(wx.VERTICAL)
self.main.SetSizer(self.mainSizer)
self.main.SetAutoLayout(True)
self.graph=SpaceGraph(self.main)
self.graph.report=self.report
# self.graph.OnShiftClick=self.selectPoint
self.mainSizer.Add(self.graph, 1, wx.GROW|wx.ALIGN_CENTRE|wx.ALL, 5)
self.graph.Show(True)
self.graph.SetDropTarget(self.dropLoader)
self.mainSizer.Fit(self.main)
self.SetSize(wx.Size(600,700))
if not self.alg:
try:
self.setModel()
except:
self.report('you may need to build or debug the model')
class NavGui(BaseGui):
def __init__(self, master=None, model=None, **kwargs):
BaseGui.__init__(self, master, title="Navigational Solver UI", menus=["File", "Build", "Run", "Display", "Debug"], pycommand=not bool(master),height=4, showframe=False)
if master:
self.document=master.document
if model:
self.alg=model
else:
self.alg=None
controls=[
["Build", "Select/Create Model", self.setModel],
["Build", "Set Bias Field", self.setBias],
["Build", "Configure Bias Field Conditioning", self.confBias],
["Build", "Save Conditioned Field as Base", self.swapBias],
["Build", "Set Start and Target Points", self.setGoal],
["Build", "Choose Algorithm", self.setAlg],
["Build", "Configure Algorithm", self.config],
["Run", "Full Solve", self.runAll],
["Display", "Show Raw Bias Field", self.rawBias],
["Display", "Show Conditioned Bias Field", self.condBias],
["Debug", "Reload Blocks", self.bounceDSP],
["Debug", "Launch Debugger", self.debug],
]
if not master:
controls.append(["Build", 'Launch Data Editor', self.launchEditor])
self.fillMenus(controls)
self.stdFileMenu()
self.biasmode='raw'
self.displaytype='bias'
self.mainSizer = wx.BoxSizer(wx.VERTICAL)
self.main.SetSizer(self.mainSizer)
self.main.SetAutoLayout(True)
self.graph=SpaceGraph(self.main)
self.graph.report=self.report
# self.graph.OnShiftClick=self.selectPoint
self.mainSizer.Add(self.graph, 1, wx.GROW|wx.ALIGN_CENTRE|wx.ALL, 5)
self.graph.Show(True)
self.graph.SetDropTarget(self.dropLoader)
self.mainSizer.Fit(self.main)
self.SetSize(wx.Size(600,700))
if not self.alg:
try:
self.setModel()
except:
self.report('you may need to build or debug the model')
def launchEditor(self, event=None):
from mien.interface.main import MienGui
d=MienGui(self)
d.newDoc(self.document)
def bounceDSP(self, event):
import mien.dsp.modules
fl=mien.dsp.modules.refresh()
if fl:
self.report('Reload generated some errors: %s' % (str(fl),))
else:
self.report("Reload complete")
def setModel(self, event=None):
if not self.document:
self.report('no document (create one with file->New)')
return
df=self.document.getElements('AbstractModel', depth=1)
if not df:
self.report("There are no navigational models in this document. Creating blank model")
abst=self.createElement('AbstractModel', {'Name':'NavModel'})
self.document.newElement(abst)
self.alg=abst
elif len(df)==1:
self.alg=df[0]
else:
amp={}
for a in df:
amp[a.upath()]=a
d=self.askParam([{'Name':'Which Model?', 'Type':'List', 'Value':amp.keys()}])
if d:
self.model=amp[d[0]]
else:
abst=self.createElement('AbstractModel', {'Name':'NavModel'})
self.document.newElement(abst)
self.alg=abst
self.scrubModel()
def _verify_component(self, index, function):
if not len(self.alg.elements)>index:
obj=self.createElement('MienBlock', {'Function':function[0]})
self.alg.newElement(obj)
return True
o=self.alg.elements[index]
if o.__tag__=='MienBlock' and o.attrib('Function') in function:
return True
print index, function, o.attributes
a=self.askUsr('Component %i is not appropriate for a nav problem' % index, ['Rebuild it', 'Select a different model', 'Create a new model', 'Abort'])
if a.startswith("R"):
if o.__tag__=='MienBlock':
for e in o.elements[:]:
e.sever()
o.setAttrib('Function', function[0])
else:
o.sever()
obj=self.createElement('MienBlock', {'Function':function[0]})
self.alg.newElement(obj)
self.arg.reorderElement(obj, index)
self.report('Model element has been rebuilt. You will need to configure the new element')
return True
if a.startswith('S'):
self.setModel()
elif a.startswith('C'):
self.report("Creating blank model")
abst=self.createElement('AbstractModel', {'Name':'NavModel'})
self.document.newElement(abst)
self.alg=abst
self.scrubModel()
self.update_all(object=self.alg, event='create')
self.report('New model created. You will need to set the bias, conditions, and algorithm now')
else:
self.alg=None
self.report("No model is selected. All operations other than loading a new document will fail!!")
return False
def scrubModel(self):
l=[['mien.dsp.nmpml.receiveData'], ['gicmapseek.dsp.setGoals'] ,['gicmapseek.dsp.conditionBias'], self.getAlgList()]
for i, fl in enumerate(l):
if not self._verify_component(i, fl):
break
def onNewDoc(self):
self.setModel()
try:
self.drawBias(True)
except:
self.report('Not able to show problem (model is not complete?)')
def setBias(self, event):
if not self.alg:
self.setModel()
g=self.alg.elements[0]
d = self.load(returndoc=True)
d=d.getElements('Data', depth=1)[0]
if d.stype()=='image':
self.report('converting image to 2D bias')
dat = d.getData()
dat = dat.mean(3)
dat = dat.mean(2)
h = d.header()
h['SampleType']='sfield'
d.datinit(dat, h)
de=g.getElements('Data')
if de:
de[0].mirror(d, True)
else:
g.newElement(d)
dp=g.getElements('Data')[0].upath()
editBlock(g, self, args={'upath':dp, 'dpath':'/', 'recurse':False})
self.biasmode='raw'
self.drawBias()
def setGoal(self, event):
EVT_RETURN_PT(self, self.continueSetGoal)
if self.biasmode!='raw':
self.biasmode='raw'
self.drawBias()
elif not self.graph.plots.has_key('bias'):
self.drawBias()
sd, td = self.getSandT()
# if sd.shape[0]!=0:
# a=self.askUsr("The current source is %s." % (str(sd[0],),), ['Keep', 'Change'])
# if a == 'Keep':
# self.continueSetGoal('keep')
# return
self.report('Select (click on) source point')
self.graph.getAPoint(self, 'source')
def continueSetGoal(self, event):
if type(event)==str:
mode=event
else:
mode=event.mode
g=self.alg.elements[1]
if mode=='source':
print event.pt
source=g.getElements('Data', {'Name':'Source'}, depth=1)[0]
source.datinit( array([[event.pt[0], event.pt[1], 0]]), {"SampleType":'locus'})
targ=g.getElements('Data', {'Name':'Target'}, depth=1)[0]
if mode!='target':
td=targ.getData()
# if td.shape[0]!=0:
# a=self.askUsr("The current target is %s." % (str(td[0],),), ['Keep', 'Change'])
# if a == 'Keep':
# return
self.report('Select (click on) target point')
self.graph.getAPoint(self, 'target')
if mode=='target':
print event.pt
targ.datinit( array([[event.pt[0], event.pt[1], 0]]), {"SampleType":'locus'})
self.drawBias(True)
def getSandT(self):
g=self.alg.elements[1]
par=g.getElements("Parameters")
if par:
par=par[0]
else:
par=self.createElement("Parameters", {})
g.newElement(par)
par.setValue({'upathSource':g.upath()+"Data:Source", 'upathTarget':g.upath()+"Data:Target"})
source=g.getElements('Data', {'Name':'Source'}, depth=1)
if source:
source=source[0].getData()
else:
source=zeros((0,3))
s=self.createElement('Data', {'Name':'Source'})
g.newElement(s)
source=zeros((0,3))
s.datinit(source, {"SampleType":'locus'})
targ=g.getElements('Data', {'Name':'Target'}, depth=1)
if targ:
targ=targ[0].getData()
else:
s=self.createElement('Data', {'Name':'Target'})
g.newElement(s)
targ=zeros((0,3))
s.datinit(targ, {"SampleType":'locus'})
return (source, targ)
def getAlgList(self, withmod=True):
import mien.blocks as mb
dsp=mb.functionIndex('gicmapseek.dsp')
algs=[]
for f in dsp.keys():
fnt=f.split('.')[-1]
if not fnt in ['noop', 'setGoals', 'conditionBias']:
if withmod:
algs.append(f)
else:
algs.append(fnt)
return algs
def setAlg(self, event):
algs=self.getAlgList(False)
a=self.askUsr('select algorithm', algs)
a="%s.%s" % ('gicmapseek.dsp', a)
o=self.alg.elements[3]
o.setAttrib('Function', a)
self.report('selected %s' % a)
self.config(None)
def config(self, event):
editBlock(self.alg.elements[3], self)
self.report('edit finished')
def rawBias(self, event):
self.biasmode='raw'
self._getBiasData(True)
self.drawBias()
def condBias(self, event):
self.biasmode='conditioned'
self._getBiasData(True)
self.drawBias()
def _getBiasData(self, rerun=False):
bd=self.document.getElements('Data', 'BiasField', depth=1)
if bd and not rerun:
return bd[0]
if bd:
bd=bd[0]
bd.datinit(None, {'SampleType':'group'})
else:
bd=newData(None, {'Name':'BiasField', 'SampleType':'group'})
self.document.newElement(bd)
els=[0,1]
if self.biasmode=='conditioned':
els.append(2)
try:
bd=self.alg.run(bd, els)
except:
raise
bd=self.alg.run(bd, [0])
return bd
def drawBias(self, recalc=False):
self.graph.plots={}
try:
bd=self._getBiasData(recalc)
except:
self.report('can not run model')
self.graph.DrawAll()
return
try:
self.graph.plotNavData(bd)
except:
self.report('bias cant be plotted')
raise
sd = self.document.getElements('Data', {'NavSolution':1}, depth=1)
if sd:
sd=sd[0].getData()
self.graph.addPlot(sd, style='line', width=4, order=1)
self.graph.DrawAll()
def runAll(self, event=None):
st=time.time()
self.alg.run()
self.report('Run completed in %.2f sec' % (time.time()-st),)
self.drawBias()
def confBias(self, event):
editBlock(self.alg.elements[2], self)
self.condBias(None)
def swapBias(self, event):
self.biasmode='conditioned'
bd=self._getBiasData(True)
g=self.alg.elements[0]
d=g.getElements('Data', depth=1)[0]
#print bd.data.shape, bd.attributes
#print d.data.shape, d.attributes
d.mirror(bd, False)
#print d.data.shape, d.attributes
editBlock(self.alg.elements[2], self, args={'width':0, 'hexagonal':False, 'maptorange':(), 'invert':False, 'power':0, 'minBelow':0, 'maxAbove':0.0})
self.biasmode='raw'
bd=self._getBiasData(True)
base=bd.getSubData('/base')
if base:
base.sever()
self.drawBias()
def debug(self, event):
reload(debuggers)
mod=self.alg.elements[3]
args=mod.getArguments()
args['debug']=True
mod.setArguments(args)
ds=self.alg.run()
dbc=debuggers.DEBUGGERS[ds.navalg.__class__.__name__]
dbv=dbc(self, ds.navalg)
dbv.Show(True)
args['debug']=False
mod.setArguments(args)
self.db = dbv