特别长序列的快速卷积

Posted liam-ji

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了特别长序列的快速卷积相关的知识,希望对你有一定的参考价值。

一、功能

用重叠保留法和快速傅里叶变换计算一个特别长序列和一个短序列的快速卷积。它通常用于数字滤波。

二、方法简介

设序列(x(n))的长度为(L),序列(h(n))的长度为(M),序列(x(n))(y(n))的线性卷积定义为
[ y(n)=sum_{i=0}^{M-1}x(i)h(n-i) ]
用重叠保留法和快速傅里叶变换计算线性卷积的算法如下:

1、将序列(h(n))按如下方式补零,形成长度为(N=2^{gamma })的序列
[ egin{matrix} h(n)=left{egin{matrix} egin{align*} h(n) &, n=0,1,...,M-1 \\ 0 &, n=M,M+1,...,N-1 end{align*} end{matrix} ight.\\ end{matrix} ]

2、用FFT算法计算序列(h(n))的离散傅里叶变换(H(k))
[ H(k)=sum_{n=0}^{N-1}h(n)e^{-j2pi nk/N} ]
3、将长序列(x(n))分为长度为(N)的小段(x_i(n)),相邻段间重叠(M-1)点,即
[ egin{matrix} x_i(n)=left{egin{matrix} egin{align*} x[n+i(N-M+1)-(M-1)] &, 0 leqslant n leqslant N-1 \\ 0 &, others end{align*} end{matrix} ight.\\ i=0,1,... end{matrix} ]
注意:对于第一段的(x_0(n)),由于没有前一段的保留信号,因此要在其前面填补(M-1)个零。

4、用FFT算法计算序列(x_i(n))的离散傅里叶变换(X_i(k))
[ X_i(k)=sum_{n=0}^{N-1}x_i(n)e^{-j2pi nk/N} ]
5、计算(X_i(k))(H(k))的乘积
[ Y_i(k)=X_i(k)H(K) ]
6、用FFT算法计算(Y_i(k))的离散傅里叶反变换,得到序列(x_i(n))(h(n))的卷积(y_i(n))
[ y_i(n)=frac{1}{N}sum_{k=0}^{N-1}Y_i(k)e^{j2pi nk/N}, n=0,1,...,N-1 ]
7、将(y_i(n))的前(M-1)点舍去,仅保留后面的(N-M+1)个样本。
[ egin{matrix} ar{y}_i(n)=left{egin{matrix} egin{align*} y(n) &, M-1 leqslant n leqslant N-1 \\ 0 &, others end{align*} end{matrix} ight.\\ end{matrix} ]
8、重复步骤3~7,直到所有分段算完为止。

9、考虑到(x(n))分段时,(x_i(n))(M-1)点与前一段重叠,新添的样本只有(N-M+1)个,所以(y(n))(ar{y}_i(n))首尾衔接而成,即
[ y(n)=sum_{i=0}^{infty}ar{y}_i[n-i(N-M+1)+(M-1)] ]

由于数据量很大,因此序列(h(n))和特别长序列(x(n))要存储到磁盘中,计算完毕后,卷积序列(y(n))也由程序自动存储到磁盘中。

三、使用说明

快速卷积的C语言实现方式如下

/************************************
    hfname      ----字符指针。它指向短序列h(i)的文件名。
    xfname      ----字符指针。它指向长序列x(i)的文件名。
    yfname      ----字符指针。它指向卷积y(i)的文件名。
    n           ----整型变量。对长序列x(i)进行分段的长度。一般选取n大于短序列h(i)长度m的两倍以上,
                且必须是2的整数次幂,即n=2^gamma。  
************************************/
#include "math.h"
#include "stdio.h"
#include "stdlib.h"
#include "rfft.c"
#include "irfft.c"
void convold(char *hfname, char *xfname, char *yfname, int n)
{
    int i, j, m, i1, i2, ls0, len, num, nblks;
    double *h, *r, *s;
    FILE *fp, *fp1, *fp2;
    r = malloc(n * sizeof(double));
    h = malloc(n * sizeof(double));
    n2 = n / 2;
    fopen(hfname, "r");
    i = 0;
    while(!fopen(fp)){
        fscanf(fp, "%lf", &h[i++]);
    }
    fclose(fp);
    m = i - 1;
    for(i = m; i < n; i++){
        h[i] = 0.0;
    }   
    rfft(h, n);
    s = malloc((n - m + 1) * sizeof(double));
    num = n - m + 1;
    fp1 = fopen(xfname, "r");
    fp2 = fopen(yfname, "r");
    len = 5000;
    ls0 = 0;
    do{
        for(i = ls0; i < 5000; i++){
            if(feof(fp1)){
                len = i;
                break;
            }
            fscanf(fp1, "%lf", &x[i]);
        }
        nblks = floor((len - n + m) / (double)num) + 1;
        for(j = 0; j < nblks; j++){
            if(j == 0){
                for(i = 0; i < (m - 1); i++)
                    r[i] = 0.0;
                for(i = m - 1; i < n; i++){
                    i1 = i - m + 1;
                    r[i] = x[i1];
                }
            } else {
                for(i = 0; i < n; i++){
                    i1 = i + j * num - m + 1;
                    r[i] = x[i1];
                }
                for(i = 0; i < num; i++){
                    i1 = i + (j - 1) * num;
                    x[i1] = s[i];
                }
            }
            rfft(r, n);
            r[0] = r[0] * h[0];
            r[n2] = r[n2] * h[n2];
            for(i = 1; i < n2; i++){
                t = h[i] * r[i] - h[n -i] * r[n - i];
                r[n - i] = h[i] * r[n - i] + h[n - i] * r[i];
                r[i] = t;
            }
            irfft(r, n);
            for(i = (m - 1); i < m; i++){
                i1 = i - m + 1;
                s[i1] = r[i];
            }
        }
        for(i = 0; i < num; i++){
            i1 = i + (j - 1) * num;
            x[i1] = s[i];
        }
        i1 = j * num;
        for(i = 0; i < i1; i++)
            fprintf(fp2, "%lf
", x[i]);

        for(i = i1; i < len; i++){
            i2 = i - i1;
            x[i2] = x[i];
        }
        ls0 = len - i1;
    }while(!feof(fp1))
    fclose(fp);
    fclose(fp1);
    fclose(fp2);
    free(r);
    free(h);
    free(s);
}

其中rfft.c文件请参考实序列快速傅里叶变换(一)

irfft.c在rfft.c的基础上添加系数长度的倒数。

以上是关于特别长序列的快速卷积的主要内容,如果未能解决你的问题,请参考以下文章

应用Matlab计算两有限长序列的线性卷积

[SDOI2015]序列统计

SSE图像算法优化系列十一:使用FFT变换实现图像卷积。

bam文件截取指定长度的序列

卷积神经网络手写数字识别数据集中的train_y和test_y是人为定义的吗?

过滤函数的简单示例,特别是递归选项