wip

python/PBMotionAnalyzer/MAError.py

@@ -8,7 +8,7 @@
 '''
 implements an image view to show a colored image of a hdf5 dataset.
 '''
-
+from __future__ import print_function
 if __name__ == '__main__':
   #Used to guarantee to use at least Wx2.8
   import wxversion
@@ -45,15 +45,17 @@ class MPLCanvasErr(FigureCanvas):
     self.fig=fig
     self.ax=ax
 
-  def InitChild(self,meta,err):
+  def InitChild(self,meta,err,trig):
 
     fig=self.fig
     ax=self.ax
     errx,erry,err=err
     hl = []
+    hl += ax.plot(trig, 'k-',label='trigger')
     hl += ax.plot(errx, 'b-',label='x-error')
     hl += ax.plot(erry, 'g-',label='y-error')
     hl += ax.plot(err, 'r-',label='error')
+
     ax.xaxis.set_label_text('datapoint (timebase: %g ms per data point)'%meta['timebase'])
     ax.yaxis.set_label_text('pos-error um')
     legend = ax.legend(loc='upper right', shadow=True)
@@ -89,12 +91,12 @@ class MPLCanvasErr(FigureCanvas):
 
   def OnBtnPress(self, event):
     """on button press we will see if the mouse is over us and store some data"""
-    print dir(event.guiEvent)
+    print(dir(event.guiEvent))
     return
 
   def OnBtnRelease(self, event):
     """on release we reset the press data"""
-    print dir(event.guiEvent)
+    print(dir(event.guiEvent))
     #self.OnMouse(event)
     return
 
@@ -129,15 +131,15 @@ class MAErrorFrame(wx.Frame):
     toolbar=ut.AddToolbar(canvas,sizer)
 
     meta = doc.fh['meta'].item()
+    rec = doc.fh['rec']
     try:
       err=doc.err
     except AttributeError:
-      rec=doc.fh['rec']
       errx = (rec[:, 1] - rec[:, 3])
       erry = (rec[:, 0] - rec[:, 2])
       errxy = np.sqrt(errx ** 2 + erry ** 2)
       doc.err = err = (errx, erry, errxy)
-    canvas.InitChild(meta,err)
+    canvas.InitChild(meta,err,rec[:,4])
 
     self.Centre()