📄 cxdxt.cpp
字号:
if( inv )
{
// conjugate the vector - i.e. invert sign of the imaginary part
int* idst = (int*)dst;
for( i = 0; i <= n - 2; i += 2 )
{
int t0 = idst[i*2+1] ^ 0x80000000;
int t1 = idst[i*2+3] ^ 0x80000000;
idst[i*2+1] = t0; idst[i*2+3] = t1;
}
if( i < n )
idst[2*i+1] ^= 0x80000000;
}
}
n = 1;
// 1. power-2 transforms
if( (factors[0] & 1) == 0 )
{
// radix-4 transform
for( ; n*4 <= factors[0]; )
{
nx = n;
n *= 4;
dw0 /= 4;
for( i = 0; i < n0; i += n )
{
CvComplex32f* v0;
CvComplex32f* v1;
double r0, i0, r1, i1, r2, i2, r3, i3, r4, i4;
v0 = dst + i;
v1 = v0 + nx*2;
r2 = v0[0].re; i2 = v0[0].im;
r1 = v0[nx].re; i1 = v0[nx].im;
r0 = r1 + r2; i0 = i1 + i2;
r2 -= r1; i2 -= i1;
i3 = v1[nx].re; r3 = v1[nx].im;
i4 = v1[0].re; r4 = v1[0].im;
r1 = i4 + i3; i1 = r4 + r3;
r3 = r4 - r3; i3 = i3 - i4;
v0[0].re = (float)(r0 + r1); v0[0].im = (float)(i0 + i1);
v1[0].re = (float)(r0 - r1); v1[0].im = (float)(i0 - i1);
v0[nx].re = (float)(r2 + r3); v0[nx].im = (float)(i2 + i3);
v1[nx].re = (float)(r2 - r3); v1[nx].im = (float)(i2 - i3);
for( j = 1, dw = dw0; j < nx; j++, dw += dw0 )
{
v0 = dst + i + j;
v1 = v0 + nx*2;
r2 = v0[nx].re*wave[dw*2].re - v0[nx].im*wave[dw*2].im;
i2 = v0[nx].re*wave[dw*2].im + v0[nx].im*wave[dw*2].re;
r0 = v1[0].re*wave[dw].im + v1[0].im*wave[dw].re;
i0 = v1[0].re*wave[dw].re - v1[0].im*wave[dw].im;
r3 = v1[nx].re*wave[dw*3].im + v1[nx].im*wave[dw*3].re;
i3 = v1[nx].re*wave[dw*3].re - v1[nx].im*wave[dw*3].im;
r1 = i0 + i3; i1 = r0 + r3;
r3 = r0 - r3; i3 = i3 - i0;
r4 = v0[0].re; i4 = v0[0].im;
r0 = r4 + r2; i0 = i4 + i2;
r2 = r4 - r2; i2 = i4 - i2;
v0[0].re = (float)(r0 + r1); v0[0].im = (float)(i0 + i1);
v1[0].re = (float)(r0 - r1); v1[0].im = (float)(i0 - i1);
v0[nx].re = (float)(r2 + r3); v0[nx].im = (float)(i2 + i3);
v1[nx].re = (float)(r2 - r3); v1[nx].im = (float)(i2 - i3);
}
}
}
for( ; n < factors[0]; )
{
// do the remaining radix-2 transform
nx = n;
n *= 2;
dw0 /= 2;
for( i = 0; i < n0; i += n )
{
CvComplex32f* v = dst + i;
double r0 = v[0].re + v[nx].re;
double i0 = v[0].im + v[nx].im;
double r1 = v[0].re - v[nx].re;
double i1 = v[0].im - v[nx].im;
v[0].re = (float)r0; v[0].im = (float)i0;
v[nx].re = (float)r1; v[nx].im = (float)i1;
for( j = 1, dw = dw0; j < nx; j++, dw += dw0 )
{
v = dst + i + j;
r1 = v[nx].re*wave[dw].re - v[nx].im*wave[dw].im;
i1 = v[nx].im*wave[dw].re + v[nx].re*wave[dw].im;
r0 = v[0].re; i0 = v[0].im;
v[0].re = (float)(r0 + r1); v[0].im = (float)(i0 + i1);
v[nx].re = (float)(r0 - r1); v[nx].im = (float)(i0 - i1);
}
}
}
}
// 2. all the other transforms
for( f_idx = (factors[0]&1) ? 0 : 1; f_idx < nf; f_idx++ )
{
int factor = factors[f_idx];
nx = n;
n *= factor;
dw0 /= factor;
if( factor == 3 )
{
// radix-3
for( i = 0; i < n0; i += n )
{
CvComplex32f* v = dst + i;
double r1 = v[nx].re + v[nx*2].re;
double i1 = v[nx].im + v[nx*2].im;
double r0 = v[0].re;
double i0 = v[0].im;
double r2 = icv_sin_120*(v[nx].im - v[nx*2].im);
double i2 = icv_sin_120*(v[nx*2].re - v[nx].re);
v[0].re = (float)(r0 + r1); v[0].im = (float)(i0 + i1);
r0 -= 0.5*r1; i0 -= 0.5*i1;
v[nx].re = (float)(r0 + r2); v[nx].im = (float)(i0 + i2);
v[nx*2].re = (float)(r0 - r2); v[nx*2].im = (float)(i0 - i2);
for( j = 1, dw = dw0; j < nx; j++, dw += dw0 )
{
v = dst + i + j;
r0 = v[nx].re*wave[dw].re - v[nx].im*wave[dw].im;
i0 = v[nx].re*wave[dw].im + v[nx].im*wave[dw].re;
i2 = v[nx*2].re*wave[dw*2].re - v[nx*2].im*wave[dw*2].im;
r2 = v[nx*2].re*wave[dw*2].im + v[nx*2].im*wave[dw*2].re;
r1 = r0 + i2; i1 = i0 + r2;
r2 = icv_sin_120*(i0 - r2); i2 = icv_sin_120*(i2 - r0);
r0 = v[0].re; i0 = v[0].im;
v[0].re = (float)(r0 + r1); v[0].im = (float)(i0 + i1);
r0 -= 0.5*r1; i0 -= 0.5*i1;
v[nx].re = (float)(r0 + r2); v[nx].im = (float)(i0 + i2);
v[nx*2].re = (float)(r0 - r2); v[nx*2].im = (float)(i0 - i2);
}
}
}
else if( factor == 5 )
{
// radix-5
for( i = 0; i < n0; i += n )
{
for( j = 0, dw = 0; j < nx; j++, dw += dw0 )
{
CvComplex32f* v0 = dst + i + j;
CvComplex32f* v1 = v0 + nx*2;
CvComplex32f* v2 = v1 + nx*2;
double r0, i0, r1, i1, r2, i2, r3, i3, r4, i4, r5, i5;
r3 = v0[nx].re*wave[dw].re - v0[nx].im*wave[dw].im;
i3 = v0[nx].re*wave[dw].im + v0[nx].im*wave[dw].re;
r2 = v2[0].re*wave[dw*4].re - v2[0].im*wave[dw*4].im;
i2 = v2[0].re*wave[dw*4].im + v2[0].im*wave[dw*4].re;
r1 = r3 + r2; i1 = i3 + i2;
r3 -= r2; i3 -= i2;
r4 = v1[nx].re*wave[dw*3].re - v1[nx].im*wave[dw*3].im;
i4 = v1[nx].re*wave[dw*3].im + v1[nx].im*wave[dw*3].re;
r0 = v1[0].re*wave[dw*2].re - v1[0].im*wave[dw*2].im;
i0 = v1[0].re*wave[dw*2].im + v1[0].im*wave[dw*2].re;
r2 = r4 + r0; i2 = i4 + i0;
r4 -= r0; i4 -= i0;
r0 = v0[0].re; i0 = v0[0].im;
r5 = r1 + r2; i5 = i1 + i2;
v0[0].re = (float)(r0 + r5); v0[0].im = (float)(i0 + i5);
r0 -= 0.25*r5; i0 -= 0.25*i5;
r1 = icv_fft5_2*(r1 - r2); i1 = icv_fft5_2*(i1 - i2);
r2 = -icv_fft5_3*(i3 + i4); i2 = icv_fft5_3*(r3 + r4);
i3 *= -icv_fft5_5; r3 *= icv_fft5_5;
i4 *= -icv_fft5_4; r4 *= icv_fft5_4;
r5 = r2 + i3; i5 = i2 + r3;
r2 -= i4; i2 -= r4;
r3 = r0 + r1; i3 = i0 + i1;
r0 -= r1; i0 -= i1;
v0[nx].re = (float)(r3 + r2); v0[nx].im = (float)(i3 + i2);
v2[0].re = (float)(r3 - r2); v2[0].im = (float)(i3 - i2);
v1[0].re = (float)(r0 + r5); v1[0].im = (float)(i0 + i5);
v1[nx].re = (float)(r0 - r5); v1[nx].im = (float)(i0 - i5);
}
}
}
else
{
// radix-"factor" - an odd number
int p, q, factor2 = (factor - 1)/2;
int d, dd, dw_f = tab_size/factor;
CvComplex32f* a = buf;
CvComplex32f* b = buf + factor2;
for( i = 0; i < n0; i += n )
{
for( j = 0, dw = 0; j < nx; j++, dw += dw0 )
{
CvComplex32f* v = dst + i + j;
CvComplex32f v_0 = v[0];
CvComplex64f vn_0(v_0);
if( j == 0 )
{
for( p = 1, k = nx; p <= factor2; p++, k += nx )
{
double r0 = v[k].re + v[n-k].re;
double i0 = v[k].im - v[n-k].im;
double r1 = v[k].re - v[n-k].re;
double i1 = v[k].im + v[n-k].im;
vn_0.re += r0; vn_0.im += i1;
a[p-1].re = (float)r0; a[p-1].im = (float)i0;
b[p-1].re = (float)r1; b[p-1].im = (float)i1;
}
}
else
{
const CvComplex32f* wave_ = wave + dw*factor;
d = dw;
for( p = 1, k = nx; p <= factor2; p++, k += nx, d += dw )
{
double r2 = v[k].re*wave[d].re - v[k].im*wave[d].im;
double i2 = v[k].re*wave[d].im + v[k].im*wave[d].re;
double r1 = v[n-k].re*wave_[-d].re - v[n-k].im*wave_[-d].im;
double i1 = v[n-k].re*wave_[-d].im + v[n-k].im*wave_[-d].re;
double r0 = r2 + r1;
double i0 = i2 - i1;
r1 = r2 - r1;
i1 = i2 + i1;
vn_0.re += r0; vn_0.im += i1;
a[p-1].re = (float)r0; a[p-1].im = (float)i0;
b[p-1].re = (float)r1; b[p-1].im = (float)i1;
}
}
v[0].re = (float)vn_0.re;
v[0].im = (float)vn_0.im;
for( p = 1, k = nx; p <= factor2; p++, k += nx )
{
CvComplex64f s0(v_0), s1 = s0;
d = dd = dw_f*p;
for( q = 0; q < factor2; q++ )
{
double r0 = wave[d].re * a[q].re;
double i0 = wave[d].im * a[q].im;
double r1 = wave[d].re * b[q].im;
double i1 = wave[d].im * b[q].re;
s1.re += r0 + i0; s0.re += r0 - i0;
s1.im += r1 - i1; s0.im += r1 + i1;
d += dd;
d -= -(d >= tab_size) & tab_size;
}
v[k].re = (float)s0.re;
v[k].im = (float)s0.im;
v[n-k].re = (float)s1.re;
v[n-k].im = (float)s1.im;
}
}
}
}
}
if( fabs(scale - 1.) > DBL_EPSILON )
{
double re_scale = scale, im_scale = scale;
if( inv )
im_scale = -im_scale;
for( i = 0; i < n0; i++ )
{
double t0 = dst[i].re*re_scale;
double t1 = dst[i].im*im_scale;
dst[i].re = (float)t0;
dst[i].im = (float)t1;
}
}
else if( inv )
{
for( i = 0; i <= n0 - 2; i += 2 )
{
double t0 = -dst[i].im;
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -