full stage tests

This commit is contained in:
2017-07-21 14:25:09 +02:00
parent 7794f23c5f
commit ede6671ff0
5 changed files with 99 additions and 7 deletions

View File

@@ -51,16 +51,42 @@ class MPLCanvasImg(FigureCanvas):
fig=self.fig
ax=self.ax
#res=rot.ActPos,x.ActPos,y.ActPos,rot.DesPos,x.DesPos,y.DesPos
#idx 0 1 2 3 4 5
#idx=np.ndarray(shape=len(pts),dtype=np.int32)
#for i in range(len(pts)):
# l=rec[:,4:6]-pts[i,:]
# l2=l[:,0]**2+l[:,1]**2
# idx[i]=np.argmin(l2)
idx=[]
for i in range(len(pts)):
l=rec[:,4:6]-pts[i,:]
l2=l[:,0]**2+l[:,1]**2
idx.extend(np.where(l2<1)[0].tolist())
#print idx
recPts=rec[idx,:]
f2 = plt.figure()
a2 = f2.add_axes([0.075,0.075,0.85,0.85])
a2.plot(rec[:,1],'r-',label='ptsDot') #,picker=5 default value
a2.plot(rec[:,2],'g-',label='ptsDot') #,picker=5 default value
a2.plot(rec[:,4],'r--',label='ptsDot') #,picker=5 default value
a2.plot(rec[:,5],'g--',label='ptsDot') #,picker=5 default value
plt.show()
hl=[]
hl+=ax.plot(pts[:,0],pts[:,1],'r.',label='ptsDot') #,picker=5 default value
hl+=ax.plot(pts[:,0],pts[:,1],'y--',label='ptsLine')
ec = mpl.collections.EllipseCollection(1, 1, 0, units='xy', offsets=pts,transOffset=ax.transData,edgecolors='g',facecolors=(1,1,0,0.3))
ax.add_collection(ec)
hl+=ax.plot(rec[:, 4], rec[:, 5], 'b-',label='recDesPos')
hl+=ax.plot(rec[:,1],rec[:,2],'g-',label='recActPos')
hl+=ax.plot(recPts[:,1],recPts[:,2],'g.',label='recDot')
ax.xaxis.set_label_text('x-pos um')
ax.yaxis.set_label_text('y-pos um')