def calcCoeff():
ret, nRow, nCol, ptx, pty = ptfile.loadPoints('data/all_pts.txt')
if ret != 0:
print 'calcCoeff() >> 读取数据失败'
return
nPoly = 2 # 抛物线
rowCoeff = []
colCoeff = []
for i in xrange(nRow):
coeff = fitPoly(ptx[i], pty[i], nPoly)
plotFitResult(ptx[i], pty[i], coeff)
rowCoeff.append(coeff)
print i
dfile.saveMatrix(rowCoeff, 'data/row_coeff.txt')
ptx1 = -np.transpose(pty)
pty1 = np.transpose(ptx)
for i in xrange(nCol):
coeff = fitPoly(ptx1[i], pty1[i], nPoly)
plotFitResult(ptx1[i], pty1[i], coeff, 1)
colCoeff.append(coeff)
print i
dfile.saveMatrix(colCoeff, 'data/col_coeff.txt')
#plt.xlim(50, 650)
#plt.ylim(-400, -150)
plt.show()
def buildMap():
ret, nRow, nCol, srcx, srcy = ptfile.loadPoints('data/true_pts.txt')
if ret != 0:
print 'buildMap() >> 读取数据失败'
return
scale = 50
dstx = [ [j*scale for j in xrange(nCol)] for i in xrange(nRow) ]
dsty = [ [i*scale for j in xrange(nCol)] for i in xrange(nRow) ]
srcx = dim2To1(srcx)
srcy = dim2To1(srcy)
dstx = dim2To1(dstx)
dsty = dim2To1(dsty)
tfin = merge(dstx, dsty)
tfout = merge(srcx, srcy)
tfln(tfin, tfout)
