导航

优化了的过关键点的光滑曲线拟合算法

Posted on 2008-04-15 22:46  yunbo  阅读(673)  评论(0编辑  收藏  举报
  这个是我一个数学老师(教授,数学高手,经常自己做算法)给我的例子,用于多个离散点拟合光滑曲线的,他优化了追赶法,这个例子适用于闭合和不闭合两种情况。当时由于工程情况,写的急,代码不好看,但是很好用。为了方便传递参数,我做了一个链表,用时候根据自己情况可以修改,核心算法不动即可。
class CFoldPoint

{public:

    
double X;    double Y;

}
;

typedef CTypedPtrList CFoldPointList;

typedef CArray CDoubleArray;

//三个函数,SPLine 调用另外两个。用时候直接调用SPLine函数,入口pList是已知离散点链表,pDestList是生成的点的链表。SM是在两个点中间插入点的数目,continue=0是采样点无规律,要求生成闭合曲线。1是采样点x坐标连续 2是y连续

void ZG(CDoubleArray *A,CDoubleArray *B,CDoubleArray *C,CDoubleArray *G,int &LOGI)
{
 
//追赶法
 register long I;
 
int N;
 N
=A->GetSize();
 
if(LOGI==0)
 
{
  (
*C)[0]=(*C)[0]/(*B)[0];
   
for(I=1;I   {
    (
*B)[I]=(*B)[I]-(*A)[I]*(*C)[I-1];
   (
*C)[I]=(*C)[I]/(*B)[I];
  }

  (
*A)[0]=0.;
   (
*C)[N-1]=0.;
   LOGI
=1;
  }

  (
*G)[0]=(*G)[0]/(*B)[0];
  
for(I=1;I  {
   (
*G)[I]=((*G)[I]-(*A)[I]*(*G)[I-1])/(*B)[I];
  }

  
for(I=N-2;I>-1;I--)//DO 30 I=N-1,1,-1
  {
   (
*G)[I]=(*G)[I]-(*C)[I]*(*G)[I+1];
  }

  
return;
}


void SPLine4(CDoubleArray *X,CDoubleArray *Y,double &XI,double&YI,CDoubleArray *A,CDoubleArray *B,CDoubleArray *C,CDoubleArray *G,int &LOGI,int MD)
{
 
 register 
long I;
 
double W1,W2,H;
 
int N=X->GetSize();
 
 
if(LOGI==0)
 
{
  
for(I=1;I  {
   (
*B)[I]=(*X)[I]-(*X)[I-1];
   (
*C)[I]=((*Y)[I]-(*Y)[I-1])/(*B)[I];
  }

  
for(I=1;I  {
   (
*A)[I]=(*B)[I]+(*B)[I+1];
   (
*G)[I]=6.*((*C)[I+1]-(*C)[I])/(*A)[I];
   (
*A)[I]=(*B)[I]/(*A)[I];
  }

  
for(I=1;I  {
   (
*C)[I]=1.-(*A)[I];
   (
*B)[I]=2.;
  }

  (
*B)[0]=2.;
  (
*B)[N-1]=2.;
  
if(MD==3)
  
{
   (
*C)[0]=-1.;
   (
*A)[N-1]=-1.;
   (
*A)[0]=0.;
   (
*C)[N-1]=0.;
  }

  ZG(A,B,C,G,LOGI);
 }

 
for(I=1;I {
  
if(XI>=(*X)[I-1&& XI<=(*X)[I])//GE LE
  {
   H
=(*X)[I]-(*X)[I-1];
   W1
=(*X)[I]-XI;
   W2
=XI-(*X)[I-1];
   YI
=W1*W1*W1*(*G)[I-1]/6./H;
   YI
=YI+W2*W2*W2*(*G)[I]/6./H;
   YI
=YI+W1*((*Y)[I-1]-(*G)[I-1]*H*H/6.)/H;
   YI
=YI+W2*((*Y)[I]-(*G)[I]*H*H/6.)/H;
  }

 }

}

void SPLine(CFoldPointList *pList,CFoldPointList *pDestList,int SM,int Continue=0)
{
 CFoldPoint 
*pFoldHead,*pFoldTail;
 POSITION pos;
 CDoubleArray A,B,C,G,X,Y,T;
 
double XI,YI,XX,YY;
 register 
long i;
 
long N;
 
int LOGI;
 
long RealSM;
 
long Bei,Yu;
 CFoldPoint 
*pFold;
 file:
//赋初值
 N=pList->GetCount();
 A.SetSize(N);
 B.SetSize(N);
 C.SetSize(N);
 G.SetSize(N);
 X.SetSize(N);
 Y.SetSize(N);
 T.SetSize(N);
 RealSM
=(N-1)*SM+N;
 pos
=pList->GetHeadPosition();
 
for(i=0;i {
  pFold
=pList->GetNext(pos);
  X[i]
=pFold->X;
  Y[i]
=pFold->Y;
 }

 
 pFoldHead
=pList->GetHead();
 pFoldTail
=pList->GetTail();
 
if(Continue==0)//pFoldHead->X==pFoldTail->X && pFoldHead->Y==pFoldTail->Y)
 { file://闭合
  T[0]=0;
  
for(i=0;i  {
   T[i
+1]=T[i]+CalculateDistance(X[i],Y[i],X[i+1],Y[i+1])+0.000000001;
  }

  LOGI
=0;
  YI
=0;
  
for(i=0;i  {
   Bei
=i/(SM+1);
   Yu
=i%(SM+1);
   
if(Yu!=0)
   
{
    XI
=T[Bei]+(T[Bei+1]-T[Bei])/(SM+1)*Yu;
    SPLine4(
&T,&Y,XI,YI,&A,&B,&C,&G,LOGI,3);
    YY
=YI;//+Y[Bei];
   }

   
else
   
{
    YY
=Y[Bei];
   }

   pFold
=new CFoldPoint;
   pFold
->Y=YY;
   pDestList
->AddTail(pFold);
  }

  LOGI
=0;
  YI
=0;
  pos
=pDestList->GetHeadPosition();
  
for(i=0;i  {
   Bei
=i/(SM+1);
   Yu
=i%(SM+1);
   
if(Yu!=0)
   
{
    XI
=T[Bei]+(T[Bei+1]-T[Bei])/(SM+1)*Yu;
    SPLine4(
&T,&X,XI,YI,&A,&B,&C,&G,LOGI,3);
    YY
=YI;//+X[Bei];
   }

   
else
   
{
    YY
=X[Bei];
   }

   pFold
=pDestList->GetNext(pos);
   pFold
->X=YY;
  }

 }

 
else if(Continue==1)
 
{
  file:
//x连续
  LOGI=0;
  YI
=0;
  
for(i=0;i  {
   Bei
=i/(SM+1);
   Yu
=i%(SM+1);
   
if(Yu!=0)
   
{
    XI
=X[Bei]+(X[Bei+1]-X[Bei])/(SM+1)*Yu;
    SPLine4(
&X,&Y,XI,YI,&A,&B,&C,&G,LOGI,3);
    XX
=XI;
    YY
=YI;
   }

   
else
   
{
    XX
=X[Bei];
    YY
=Y[Bei];
   }

   pFold
=new CFoldPoint;
   pFold
->X=XX;
   pFold
->Y=YY;
   pDestList
->AddTail(pFold);
  }

 }

 
else
 
{
  file:
//y连续
  LOGI=0;
  YI
=0;
  
for(i=0;i  {
   Bei
=i/(SM+1);
   Yu
=i%(SM+1);
   
if(Yu!=0)
   
{
    XI
=Y[Bei]+(Y[Bei+1]-Y[Bei])/(SM+1)*Yu;
    SPLine4(
&Y,&X,XI,YI,&A,&B,&C,&G,LOGI,3);
    XX
=YI;
    YY
=XI;
   }

   
else
   
{
    XX
=X[Bei];
    YY
=Y[Bei];
   }

   pFold
=new CFoldPoint;
   pFold
->X=XX;
   pFold
->Y=YY;
   pDestList
->AddTail(pFold);
  }

 }

 
return;
}