B-Spline曲線擬合

B-Spline曲線擬合 – Python實現

1.曲線定義

定義曲線爲p階樣條曲線

給定n+1個控制點 P0,P1,...Pn

節點向量 U = {u0,u1....um}，且m = n+p+1

B-Spline曲線定義如下：

N(i,p)爲樣條基函數

P(i)爲控制頂點

參數u爲參數節點，一般取0 ≤ u ≤ 1

上式爲樣條基函數的遞推公式

2.De Boor 算法

參考文章：
https://pages.mtu.edu/~shene/COURSES/cs3621/NOTES/spline/B-spline/single-insertion.html

根據鏈接中的描述，B-Spline公式可以簡化爲以下公式：

Pi爲控制頂點

ai爲比例係數，由當前節點t和節點u(i) u(i+p) 計算得到

3.算法實現流程 – 3次B-Spline曲線

第一步：設置曲線次數爲p=3，讀取控制頂點P0,P1,...Pn，根據控制頂點得到節點向量U = {u0,u1....um}，注意m = n+p+1.爲了保證擬合的曲線經過第一個、最後一個控制點，節點向量首末重複度要設置爲爲p+1，即U = {0,0,0,0,u(4)...u(n),1,1,1,1}

第二步：按照本文描述的De Boor算法完成遞推函數

第三步：在0 ≤ u ≤ 1取值，即可得到u對應的P(u)位置

4.Python 實現

下面貼出完整python代碼，僅供參考

import matplotlib.pyplot as plt
import numpy as nmp
import math
#degree. 3次樣條
knotType = 1   #knot的計算方式 0：平均取值  1：根據長度取值
CPointX = []
CPointY = []
#控制輪廓頂點

with open('data.txt', 'r') as f1:
    list1 = f1.readlines()

for i in range(len(list1)):
    str = list1[i]
    str1 = str.split(",")
    CPointX.append(float(str1[0]))
    CPointY.append(float(str1[1]))

n = len(CPointX)
knot = list(nmp.arange(n+3+1))
for i in range(0,4):
    knot[i] = 0.0
    knot[n+i] = 1.0

if knotType:
    L = list(nmp.arange(n-1))
    S = 0
    for i in range(n-1):
        L[i] = nmp.sqrt(pow(CPointX[i+1]- CPointX[i],2) + pow(CPointY[i+1]- CPointY[i],2))
        S = S + L[i]

    tmp = L[0]
    for i in range(4,n):
        tmp = tmp + L[i-3]
        knot[i] = tmp / S
else:
    tmp = 1 / (n-3)
    tmpS = tmp
    for i in range(4,n):
        knot[i] = tmpS
        tmpS = tmpS + tmp
print(knot)
#節點向量
#knot = [0,0,0,0,0.25,0.4,0.8,1,1,1,1]

def de_boor_cal(u,knot,X, Y):
    #判斷u在哪個區間
    m = len(knot) #節點個數 且 m = n + k +1
    n = len(X)
    i = 3
    for i in range(3,m-3):
        if u>=knot[i] and u<knot[i+1]:
            break
    #遞歸計算
    if knot[i+3] - knot[i] == 0:
        a41 = 0
    else:
        a41 = (u - knot[i]) / (knot[i+3] - knot[i])
    if knot[i-1+3] - knot[i-1] == 0:
        a31 = 0
    else:
        a31 = (u - knot[i-1]) / (knot[i-1+3] - knot[i-1])
    if knot[i-2+3] - knot[i-2] == 0:
        a21 = 0
    else:
        a21 = (u - knot[i-2]) / (knot[i-2+3] - knot[i-2])

    XP40 = X[i]
    YP40 = Y[i]
    XP30 = X[i-1]
    YP30 = Y[i - 1]
    XP20 = X[i-2]
    YP20 = Y[i - 2]
    XP10 = X[i-3]
    YP10 = Y[i - 3]

    XP41 = (1 - a41) * XP30 + a41 * XP40
    YP41 = (1 - a41) * YP30 + a41 * YP40

    XP31 = (1 - a31) * XP20 + a31 * XP30
    YP31 = (1 - a31) * YP20 + a31 * YP30

    XP21 = (1 - a21) * XP10 + a21 * XP20
    YP21 = (1 - a21) * YP10 + a21 * YP20
    if knot[i+3-1] - knot[i] == 0:
        a42 = 0
    else:
        a42 = (u - knot[i]) / (knot[i+3-1] - knot[i])
    if knot[i-1+3-1] - knot[i-1] == 0:
        a32 = 0
    else:
        a32 = (u - knot[i-1]) / (knot[i-1+3-1] - knot[i-1])


    XP42 = (1 - a42) * XP31 + a42 * XP41
    YP42 = (1 - a42) * YP31 + a42 * YP41

    XP32 = (1 - a32) * XP21 + a32 * XP31
    YP32 = (1 - a32) * YP21 + a32 * YP31

    if knot[i+3-2] - knot[i] == 0:
        a43 = 0
    else:
        a43 = (u - knot[i]) / (knot[i+3-2] - knot[i])

    P43 = [(1 - a43) * XP32 + a43 * XP42,(1 - a43) * YP32 + a43 * YP42]
    return P43


plt.plot(CPointX,CPointY, '-ob', label="Waypoints")
plt.grid(True)
plt.legend()
plt.axis("equal")

Bspline = [[],[]]
for u in nmp.arange(0.0,1.0,0.0005):
    P = de_boor_cal(u,knot,CPointX,CPointY)
    Bspline[0].append(P[0])
    Bspline[1].append(P[1])

Bspline[0].append(CPointX[-1])
Bspline[1].append(CPointY[-1])
plt.plot(Bspline[0],Bspline[1], 'r')
plt.show()

運行效果：