dwt.cpp

来自「自己做的小波与神经网络组合方法实现负荷预测程序」· C++ 代码 · 共 30 行

/////小波分解

#include <stdio.h>
void dwt(double g[],double h[],int wlen,double c[],double d[],int m,int sca[])
{
  int i,j,k,mid,flag[20];
  double p,q;
    for(flag[0]=0,i=0;i<m;i++)
     {flag[i+1]=flag[i]+sca[i];}
  printf("\n doing decomposition");
  for(j=1;j<=m;j++)
    {
	  printf("*******");
	  for(i=0;i<sca[j];i++)
	    {
		  p=0;
	      q=0;
		  for(k=0;k<wlen;k++)
		    {
			  mid=k+2*i;
			  if(mid>=sca[j-1]) mid=mid-sca[j-1];
			  p=p+h[k]*c[flag[j-1]+mid];
			  q=q+g[k]*c[flag[j-1]+mid];
            }
           c[flag[j]+i]=p;
		   d[flag[j]+i]=q;
		  
        }
    }
}