397 lines
14 KiB
Python
Executable File
397 lines
14 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# *-----------------------------------------------------------------------*
|
|
# | |
|
|
# | Copyright (c) 2016 by Paul Scherrer Institute (http://www.psi.ch) |
|
|
# | |
|
|
# | Author Thierry Zamofing (thierry.zamofing@psi.ch) |
|
|
# *-----------------------------------------------------------------------*
|
|
from __future__ import print_function
|
|
import os,sys
|
|
import wx
|
|
import wx.py
|
|
import numpy as np
|
|
from MAxyPlot import *
|
|
from MAError import *
|
|
from MAVelocity import *
|
|
#from hdfTree import *
|
|
#from hdfGrid import *
|
|
#from hdfAttrib import *
|
|
#from hdfImage import *
|
|
|
|
import wxutils as ut
|
|
class MADoc():
|
|
'''
|
|
lenRec=fh['rec']
|
|
lenPts=fh['pts']
|
|
pts= X,Y array
|
|
|
|
idx 0 1 2 3 4 5 6
|
|
OLD Motor[3].ActPos Motor[2].ActPos Motor[1].ActPos Motor[3].DesPos Motor[2].DesPos Motor[1].DesPos Gate3[1].Chan[1].UserFlag
|
|
NEW Motor[1].ActPos Motor[2].ActPos Motor[1].DesPos Motor[2].DesPos Gate3[1].Chan[1].UserFlag
|
|
NEW y.ActPos x.ActPos y.DesPos x.DesPos Gate3[1].Chan[1].UserFlag
|
|
OLD->NEW
|
|
0->none
|
|
1->1
|
|
2->0
|
|
3->none
|
|
4->3
|
|
5->2
|
|
|
|
Implemented messages
|
|
0: The time slider has changed. usrData=index
|
|
1: The position slider has changed. usrData=index
|
|
'''
|
|
|
|
def __init__(self):
|
|
self.view=[] #list of object that have view on this document
|
|
pass
|
|
|
|
def Update(self,skipView,msg,usrData):
|
|
for v in self.view:
|
|
if v==skipView: continue
|
|
v.OnUpdate(msg,usrData)
|
|
|
|
|
|
class AboutFrame(wx.Frame):
|
|
def __init__(self,parent):
|
|
wx.Frame.__init__(self,parent,-1,'About MotionAnalyzer',size=(300,330))
|
|
imgDir=ut.Path.GetImage()
|
|
icon = wx.Icon(os.path.join(imgDir,'PBMA.ico'), wx.BITMAP_TYPE_ICO)
|
|
self.SetIcon(icon)
|
|
self.Centre()
|
|
panel=wx.Panel(self,-1)
|
|
#import pkg_resources
|
|
#v=pkg_resources.get_distribution("h5pyViewer")
|
|
v='my version info'
|
|
s='Version:'+str(v)+'\n(c) www.psi.ch\n Author: Thierry Zamofing\n thierry.zamofing@psi.ch'
|
|
|
|
st0=wx.StaticText(panel,-1,s,(30,10))
|
|
bmp = wx.StaticBitmap(panel,-1,wx.Bitmap(os.path.join(imgDir,'PBMA.png'), wx.BITMAP_TYPE_ANY ), (30,st0.Position[1]+st0.Size[1]+10))
|
|
for k,v in os.environ.iteritems():
|
|
print(k,'=',v)
|
|
|
|
class MAMainFrame(wx.Frame):
|
|
def __init__(self, parent, title,doc):
|
|
wx.Frame.__init__(self, parent, title=title, size=wx.Size(650, 230),style = wx.DEFAULT_FRAME_STYLE | wx.STAY_ON_TOP)
|
|
self.doc=doc;doc.view.append(self)
|
|
imgDir=ut.Path.GetImage()
|
|
icon = wx.Icon(os.path.join(imgDir,'PBMA.ico'), wx.BITMAP_TYPE_ICO)
|
|
self.SetIcon(icon)
|
|
#wxSplt = wx.SplitterWindow(self, -1)
|
|
#wxTree = HdfTreeCtrl(wxSplt, 1, wx.DefaultPosition, (-1,-1), wx.TR_HAS_BUTTONS)
|
|
#wxTree.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnSelChanged, id=1)
|
|
#wxTree.Bind(wx.EVT_TREE_ITEM_MENU, self.OnMenu, id=1)
|
|
#wx.EVT_TREE_ITEM_MENU(id, func)
|
|
#wxTxt = wx.StaticText(wxSplt, -1, '',(10,10) )#, style=wx.ALIGN_CENTRE)
|
|
|
|
|
|
sizer = wx.BoxSizer(wx.VERTICAL)
|
|
self.SetSizer(sizer)
|
|
wxTxt = wx.StaticText(self, wx.ID_ANY, "MyLabel", wx.DefaultPosition, wx.DefaultSize, 0 )
|
|
|
|
sizer.Add(wxTxt, 1, wx.LEFT | wx.TOP | wx.GROW)
|
|
|
|
wxTimeCtrl=ut.SliderGroup(self, label='DataPoint', range=(0, 1000))
|
|
wxTimeCtrl.SetCallback(MAMainFrame.OnSetTime, wxTimeCtrl)
|
|
sizer.Add(wxTimeCtrl.sizer, 0, wx.EXPAND | wx.ALIGN_CENTER | wx.ALL, border=5) #wxSplt.SplitVertically(wxTree, wxTxt)
|
|
wxPosCtrl=ut.SliderGroup(self, label='IdxInPos', range=(0, 1000))
|
|
wxPosCtrl.SetCallback(MAMainFrame.OnSetIdxInPos, wxPosCtrl)
|
|
sizer.Add(wxPosCtrl.sizer, 0, wx.EXPAND | wx.ALIGN_CENTER | wx.ALL, border=5) #wxSplt.SplitVertically(wxTree, wxTxt)
|
|
wxTrigCtrl=ut.SliderGroup(self, label='IdxTrigger', range=(0, 1000))
|
|
wxTrigCtrl.SetCallback(MAMainFrame.OnSetIdxTrigger, wxTrigCtrl)
|
|
sizer.Add(wxTrigCtrl.sizer, 0, wx.EXPAND | wx.ALIGN_CENTER | wx.ALL, border=5) #wxSplt.SplitVertically(wxTree, wxTxt)
|
|
#wxSplt.SetMinimumPaneSize(320)
|
|
#wxLstCtrl=HdfAttrListCtrl(wxSplt)
|
|
#wxSplt.SplitVertically(wxTree, wxLstCtrl)
|
|
self.BuildMenu()
|
|
|
|
self.Centre()
|
|
self.wxTimeCtrl=wxTimeCtrl
|
|
self.wxPosCtrl=wxPosCtrl
|
|
self.wxTrigCtrl=wxTrigCtrl
|
|
self.wxTxt=wxTxt
|
|
#self.wxTree=wxTree
|
|
#self.display=wxTxt
|
|
|
|
def __del__(self):
|
|
self.CloseFile()
|
|
|
|
def OpenFile(self,fn_npz):
|
|
doc=self.doc
|
|
try:
|
|
doc.fh=fh=np.load(fn_npz)
|
|
except IOError as e:
|
|
sys.stderr.write('Unable to open File: '+fn_npz+'\n')
|
|
else:
|
|
pass
|
|
s='content of numpy file: '+fn_npz+'\n'
|
|
for k,v in fh.iteritems():
|
|
s+=' '+k+': '+str(v.dtype)+' '+str(v.shape)+'\n'
|
|
self.wxTxt.SetLabel(s)
|
|
|
|
rec=fh['rec']
|
|
pts=fh['pts']
|
|
lenRec=rec.shape[0]
|
|
lenPts=pts.shape[0]
|
|
#idxTrigger=np.hstack(([0],rec[:,4]))
|
|
#doc.idxTrigger=idxTrigger=np.where(np.diff(idxTrigger)==1)[0]
|
|
idxTrigger=rec[:,4]
|
|
doc.idxTrigger=idxTrigger=np.where(np.diff(idxTrigger)==1)[0]+1
|
|
idxInPos=[] #first point at idx 0
|
|
try:# find approximate distance of 2 points
|
|
rng =idxTrigger[2]-idxTrigger[1]
|
|
except IndexError:
|
|
rng = int(lenRec / lenPts)
|
|
idx=rng/2
|
|
|
|
for i in range(lenPts):
|
|
l=rec[idx:idx+rng,(3,2)]-pts[i,:]
|
|
l2=l[:,0]**2+l[:,1]**2
|
|
try:
|
|
ofs=l2.argmin()
|
|
except ValueError:
|
|
break#print(l2[ofs])
|
|
idx+=ofs
|
|
idxInPos.append(idx)
|
|
idx+=rng/2
|
|
|
|
doc.idxInPos=idxInPos=np.array(idxInPos)
|
|
idx=min(idxInPos.shape[0],idxTrigger.shape[0])-1
|
|
self.SetStatusText('scaling of DesTimeBase: %f'%(float(idxInPos[idx])/idxTrigger[idx]))
|
|
#self.PlotJitter()
|
|
|
|
self.wxTimeCtrl.slider.SetRange(0,lenRec-1)
|
|
self.wxTimeCtrl.slider.SetPageSize(idxInPos[1]-idxInPos[0])
|
|
self.wxPosCtrl.slider.SetRange(0,lenPts-1)
|
|
self.wxPosCtrl.slider.SetPageSize(10)
|
|
self.wxTrigCtrl.slider.SetRange(0,idxTrigger.shape[0]-1)
|
|
self.wxTrigCtrl.slider.SetPageSize(10)
|
|
|
|
def CloseFile(self):
|
|
#http://docs.wxwidgets.org/2.8/wx_windowdeletionoverview.html#windowdeletionoverview
|
|
#print 'CloseFile'
|
|
try:
|
|
self.fh.close()
|
|
del self.fh
|
|
except AttributeError as e:
|
|
pass
|
|
|
|
def PlotJitter(self):
|
|
doc=self.doc
|
|
fh=doc.fh
|
|
idxInPos=doc.idxInPos
|
|
idxTrigger=doc.idxTrigger
|
|
#display the jitter of inPos->trigger
|
|
ts=fh['meta'].item()['timebase']
|
|
jitter=idxTrigger[:idxInPos.shape[0]]-idxInPos
|
|
import matplotlib.pyplot as plt # used for the colormaps
|
|
plt.figure('jitter inPos -> trigger: scaling of DesTimeBase: %f'%(float(idxInPos[-1])/idxTrigger[idxInPos.shape[0]-1]))
|
|
plt.plot(jitter*ts)
|
|
plt.ylabel('jitter in ms')
|
|
plt.ylabel('point idx')
|
|
plt.show()
|
|
|
|
def DispJitter(self,idxInPos,idxTrigger):
|
|
doc=self.doc
|
|
ts=doc.fh['meta'].item()['timebase']
|
|
self.SetStatusText("Jitter: %.2gms scaling: %.6g"%((idxTrigger-idxInPos)*ts,float(idxInPos)/idxTrigger))
|
|
|
|
def OnOpen(self, event):
|
|
dlg = wx.FileDialog(self, "Choose a file", os.getcwd(), '','numpy files (*.npz;*.npy)|*.npz;*.npy|all (*.*)|*.*', wx.OPEN|wx.FD_CHANGE_DIR)
|
|
if dlg.ShowModal() == wx.ID_OK:
|
|
path = dlg.GetPath()
|
|
#mypath = os.path.basename(path)
|
|
#self.SetStatusText("You selected: %s" % mypath)
|
|
self.CloseFile()
|
|
self.OpenFile(path)
|
|
#print 'OnOpen',path
|
|
dlg.Destroy()
|
|
|
|
def OnCloseWindow(self, event):
|
|
#print 'OnCloseWindow'
|
|
self.Destroy()
|
|
|
|
def OnAbout(self,event):
|
|
frame=AboutFrame(self)
|
|
frame.Show()
|
|
|
|
def BuildMenu(self):
|
|
#http://wiki.wxpython.org/AnotherTutorial#wx.MenuBar
|
|
mnBar = wx.MenuBar()
|
|
|
|
#-------- File Menu --------
|
|
mn = wx.Menu()
|
|
mnItem=mn.Append(wx.ID_OPEN, '&Open', 'Open a new document');self.Bind(wx.EVT_MENU, self.OnOpen, mnItem)
|
|
#mnSub = wx.Menu()
|
|
#mnItem=mnSub.Append(wx.ID_ANY, 'SubMenuEntry', 'My SubMenuEntry')
|
|
#mn.AppendMenu(wx.ID_ANY, 'SubMenu', mnSub)
|
|
mn.AppendSeparator()
|
|
mnItem=mn.Append(wx.ID_EXIT, '&Quit', 'Quit the Application');self.Bind(wx.EVT_MENU, self.OnCloseWindow, mnItem)
|
|
mnBar.Append(mn, '&File')
|
|
|
|
self.Bind(wx.EVT_CLOSE, self.OnCloseWindow)
|
|
|
|
#-------- Edit Menu --------
|
|
mn = wx.Menu()
|
|
mnItem = mn.Append(wx.ID_ANY, 'Show &XY-Path', 'Show XY-path motion path');self.Bind(wx.EVT_MENU, self.OnShowXYPath, mnItem)
|
|
mnItem = mn.Append(wx.ID_ANY, 'Show &Error', 'Show error of motion path');self.Bind(wx.EVT_MENU, self.OnShowError, mnItem)
|
|
mnItem = mn.Append(wx.ID_ANY, 'Show &Velocity', 'Show velocity of motion path');self.Bind(wx.EVT_MENU, self.OnShowVelocity, mnItem)
|
|
mnItem = mn.Append(wx.ID_ANY, '&Python Shell', 'Opens an interactive python shell"');self.Bind(wx.EVT_MENU, self.OnShell, mnItem)
|
|
|
|
mnBar.Append(mn, '&Window')
|
|
|
|
#-------- Help Menu --------
|
|
mn = wx.Menu()
|
|
#mnItem=mn.Append(wx.ID_HELP,'Help','Application Help')
|
|
mnItem=mn.Append(wx.ID_ABOUT,'About','Application About');self.Bind(wx.EVT_MENU, self.OnAbout, mnItem)
|
|
mnBar.Append(mn, '&Help')
|
|
|
|
#mn.AppendSeparator()
|
|
#mnItem = wx.MenuItem(mn, 105, '&Quit\tCtrl+Q', 'Quit the Application')
|
|
#mnItem.SetBitmap(wx.Image('stock_exit-16.png', wx.BITMAP_TYPE_PNG).ConvertToBitmap())
|
|
#mn.AppendItem(mnItem)
|
|
self.SetMenuBar(mnBar)
|
|
self.CreateStatusBar()
|
|
self.SetStatusText("This is the status message")
|
|
|
|
def OnShowXYPath(self, event):
|
|
frame=MAxyPlotFrame(self,self.doc)
|
|
frame.Show(True)
|
|
|
|
def OnShowError(self, event):
|
|
frame=MAErrorFrame(self,self.doc)
|
|
frame.Show(True)
|
|
|
|
def OnShowVelocity(self, event):
|
|
frame=MAVelocityFrame(self,self.doc)
|
|
frame.Show(True)
|
|
|
|
def OnShell(self, event):
|
|
frame = wx.Frame(self, -1, "wxPyShell",size=wx.Size(800, 500))
|
|
imgDir=Path.GetImage()
|
|
icon = wx.Icon(os.path.join(imgDir,'PBMA.ico'), wx.BITMAP_TYPE_ICO)
|
|
frame.SetIcon(icon)
|
|
frame.Centre()
|
|
fh=app.GetTopWindow().fh
|
|
wnd=app.GetTopWindow()
|
|
loc={'app' :app,
|
|
'fh' :fh,
|
|
'pts' :fh['pts'],
|
|
'rec' :fh['rec']
|
|
}
|
|
introText='''Shell to the HDF5 objects
|
|
app: application object
|
|
fh: numpy file
|
|
pts: desired motion points
|
|
rec: recorded data
|
|
|
|
#Examples:
|
|
pts
|
|
rec
|
|
|
|
#using user defined modules
|
|
#import userSample as us;reload(us);us.test1(hid)
|
|
'''
|
|
shell=wx.py.shell.Shell(frame, introText=introText,locals=loc)
|
|
frame.Show(True)
|
|
#if loc is None, all variables are visible. the context is global
|
|
#shell.push('wnd=app.GetTopWindow()')
|
|
#for cmd in [
|
|
# 'wnd=app.GetTopWindow();wxTree=wnd.wxTree',
|
|
# 'wxNode=wnd.wxTree.GetSelection()',
|
|
# 'print wnd.fid',
|
|
# 'lbl=wxTree.GetItemText(wxNode)',
|
|
# 'hid=wxTree.GetPyData(wxNode)']:
|
|
# shell.run(cmd, prompt=False)
|
|
|
|
def OnUpdate(self,msg,usrData):
|
|
#this is the model-view-control update function
|
|
#print self,msg,usrData
|
|
pass
|
|
|
|
@staticmethod
|
|
def OnSetTime(usrData, idxRec, msg):
|
|
'called when the time slider has been changed'
|
|
#print('OnSetTime', usrData, value, msg)
|
|
view=usrData.slider.Parent
|
|
doc=view.doc
|
|
idxInPos=np.argmin(abs(doc.idxInPos-idxRec))
|
|
view.wxPosCtrl.SetValue(idxInPos)
|
|
idxTrigger=np.argmin(abs(doc.idxTrigger-idxRec))
|
|
view.wxTrigCtrl.SetValue(idxTrigger)
|
|
view.DispJitter(doc.idxInPos[idxInPos],doc.idxTrigger[idxTrigger])
|
|
doc.Update(view,0,idxRec)
|
|
|
|
@staticmethod
|
|
def OnSetIdxInPos(usrData, idxInPos, msg):
|
|
'called when the idxInPos slider has been changed'
|
|
#print('OnSetPosIdx', usrData, value, msg)
|
|
view = usrData.slider.Parent
|
|
doc = view.doc
|
|
idxRec=doc.idxInPos[idxInPos]
|
|
view.wxTimeCtrl.SetValue(idxRec)
|
|
idxTrigger=np.argmin(abs(doc.idxTrigger-idxRec))
|
|
view.wxTrigCtrl.SetValue(idxTrigger)
|
|
view.DispJitter(doc.idxInPos[idxInPos],doc.idxTrigger[idxTrigger])
|
|
doc.Update(view, 0, idxRec)
|
|
|
|
@staticmethod
|
|
def OnSetIdxTrigger(usrData, idxTrigger, msg):
|
|
'called when the idxTrigger slider has been changed'
|
|
#print('OnSetIdxTrigger', usrData, value, msg)
|
|
view = usrData.slider.Parent
|
|
doc = view.doc
|
|
idxRec=doc.idxTrigger[idxTrigger]
|
|
view.wxTimeCtrl.SetValue(idxRec)
|
|
idxInPos=np.argmin(abs(doc.idxInPos-idxRec))
|
|
view.wxPosCtrl.SetValue(idxInPos)
|
|
view.DispJitter(doc.idxInPos[idxInPos],doc.idxTrigger[idxTrigger])
|
|
doc.Update(view, 0, idxRec)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
def GetArgs():
|
|
import sys,argparse #since python 2.7
|
|
exampleCmd='/tmp/shapepath.npz'
|
|
parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
description=__doc__,
|
|
epilog='Example:\n'+os.path.basename(sys.argv[0])+' '+exampleCmd+'\n ')
|
|
parser.add_argument('npzFile', nargs='?', help='the npz file with motion path data to show',default='/tmp/shapepath.npz')
|
|
|
|
args = parser.parse_args()
|
|
return args
|
|
|
|
|
|
class MyApp(wx.App):
|
|
|
|
def OnInit(self):
|
|
args=GetArgs()
|
|
doc=MADoc()
|
|
frame = MAMainFrame(None,'PBMotionAnalyzer',doc)
|
|
if args.npzFile:
|
|
frame.OpenFile(args.npzFile)
|
|
frame.Show(True)
|
|
|
|
f1 = MAxyPlotFrame(frame, doc)
|
|
f1.Show(True)
|
|
#f2= MAErrorFrame(frame, doc)
|
|
#f2.Show(True)
|
|
#f3= MAVelocityFrame(frame, doc)
|
|
#f3.Show(True)
|
|
|
|
self.SetTopWindow(frame)
|
|
return True
|
|
|
|
#------------------ Main Code ----------------------------------
|
|
#redirect stdout/stderr:
|
|
#http://www.blog.pythonlibrary.org/2009/01/01/wxpython-redirecting-stdout-stderr/
|
|
#https://groups.google.com/forum/#!topic/wxpython-users/S9uSKIYdYoo
|
|
#https://17677433047266577941.googlegroups.com/attach/e4d343dc6a751906/REDIRECT.PY?part=2&view=1&vt=ANaJVrFeyCjCMydKnkyfFbYJM7ip07mE-ozUIBxJ5A1QuK1GhycJYJsPTxpAaNk5L2LpXvGhzRPInxDt8_WUcUyK2Ois28Dq8LNebfYoWG9Yxr-tujf5Jk4
|
|
#http://www.wxpython.org/docs/api/wx.PyOnDemandOutputWindow-class.html
|
|
rd=not sys.stdout.isatty()#have a redirect window, if there is no console
|
|
#rd=True #force to open a redirect window
|
|
rd=False #avoid a redirect window
|
|
app = MyApp(redirect=rd)
|
|
app.MainLoop()
|
|
|