full stage tests
This commit is contained in:
@@ -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')
|
||||
|
||||
|
||||
Reference in New Issue
Block a user