?? matrixoper.cpp
字號(hào):
#include "stdafx.h"
#include "math.h"
COMPLEX CMatrixOper::Add(COMPLEX c1,COMPLEX c2)
{
COMPLEX c;
c.re=c1.re+c2.re;
c.im=c1.im+c2.im;
return c;
}
COMPLEX CMatrixOper::Sub(COMPLEX c1,COMPLEX c2)
{
COMPLEX c;
c.re=c1.re-c2.re;
c.im=c1.im-c2.im;
return c;
}
COMPLEX CMatrixOper::Mul(COMPLEX c1,COMPLEX c2)
{
COMPLEX c;
c.re=c1.re*c2.re-c1.im*c2.im;
c.im=c1.re*c2.im+c2.re*c1.im;
return c;
}
void CMatrixOper::Convert(COMPLEX*c1)
{
(*c1).im = -(*c1).im;
}
/* 參數(shù)：
FD為頻域值
TD為時(shí)域值
power為2的冪數(shù)
返回值：
無
說明：
本函數(shù)利用快速傅立葉變換實(shí)現(xiàn)傅立葉變換
****************************************************/
void CMatrixOper::FFT(COMPLEX*TD,COMPLEX*FD,int power)
{
int count;
int i,j,k,bfsize,p;
double angle;
COMPLEX *W,*X1,*X2,*X;
/*計(jì)算傅立葉變換點(diǎn)數(shù)*/
count=1<<power;
/*分配運(yùn)算所需存儲(chǔ)器*/
W = (COMPLEX *)malloc(sizeof(COMPLEX)*count/2);
X1 = (COMPLEX *)malloc(sizeof(COMPLEX)*count);
X2 = (COMPLEX *)malloc(sizeof(COMPLEX)*count);
/*計(jì)算加權(quán)系數(shù)*/
for(i=0;i<count/2;i++)
{
angle = -i*PI*2/count;
W[i].re = cos(angle);
W[i].im = sin(angle);
}
/*將時(shí)域點(diǎn)寫入存儲(chǔ)器*/
memcpy(X1,TD,sizeof(COMPLEX)*count);
/*蝶形運(yùn)算*/
for(k = 0;k<power;k++)
{
for(j = 0;j<1<<k;j++)
{
bfsize = 1<<(power-k);
for(i = 0;i<bfsize/2;i++)
{
p = j*bfsize;
X2[i+p] = Add(X1[i+p],X1[i+p+bfsize/2]);
X2[i+p+bfsize/2] = Mul(Sub(X1[i+p],X1[i+p+bfsize/2]),W[i*(1<<k)]);
}
}
X = X1;
X1 = X2;
X2 = X;
}
/*重新排序*/
for(j = 0;j < count;j++)
{
p = 0;
for(i = 0;i<power;i++)
{
if(j&(1<<i))
p+=1<<(power-i-1);
}
FD[j] = X1[p];
}
/*釋放存儲(chǔ)器*/
free(W);
free(X1);
free(X2);
}
/* 參數(shù)：
FD為頻域值
TD為時(shí)域值
power為2的冪數(shù)
返回值：
無
說明：
本函數(shù)利用快速傅立葉變換實(shí)現(xiàn)傅立葉反變換
****************************************************/
void CMatrixOper::IFFT(COMPLEX*FD,COMPLEX*TD,int power)
{
int i,count;
COMPLEX *x;
/*計(jì)算傅立葉反變換點(diǎn)數(shù)*/
count = 1<<power;
/*分配運(yùn)算所需存儲(chǔ)器*/
x = (COMPLEX *)malloc(sizeof(COMPLEX)*count);
/*將頻域點(diǎn)寫入存儲(chǔ)器*/
memcpy(x,FD,sizeof(COMPLEX)*count);
/*求頻域點(diǎn)的共軛*/
for(i = 0 ;i < count; i++)
x[i].im = -x[i].im;
/*調(diào)用FFT*/
FFT(x,TD,power);
/*求時(shí)域點(diǎn)的共軛*/
for(i = 0;i<count;i++)
{
TD[i].re /= count;
TD[i].im = -TD[i].im / count;
}
/*釋放存儲(chǔ)器*/
free(x);
}
void CMatrixOper::MatrCOMRePointMulMatr(double*Matr1,int row,int colum,COMPLEX*Matr2,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).re = *(Matr1 + i) * (*(Matr2 + i)).re;
}
}
void CMatrixOper::MatrCOMImPointMulMatr(double*Matr1,int row,int colum,COMPLEX*Matr2,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).im = *(Matr1 + i) * (*(Matr2 + i)).im;
}
}
void CMatrixOper::MatrPointMul(double*Matr1,int row,int colum,double*Matr2,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr1 + i) * *(Matr2 + i);
}
}
void CMatrixOper::MatrPointDiv(double*Matr1,int row,int colum,double*Matr2,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr1 + i) / *(Matr2 + i);
}
}
void CMatrixOper::MatrPointAdd(double*Matr1,int row,int colum,double*Matr2,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr1 + i) + *(Matr2 + i);
}
}
void CMatrixOper::MatrPointSub(double*Matr1,int row,int colum,double*Matr2,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr1 + i) - *(Matr2 + i);
}
}
void CMatrixOper::MatrreAddim(COMPLEX*Matr,int row,int colum)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).re = (*(Matr + i)).re + (*(Matr + i)).im;
}
}
void CMatrixOper::MatrPointAddNum(double*Matr,int row,int colum,double a)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr + i) + a;
}
}
void CMatrixOper::MatrPointMulNum(double*Matr,int row,int colum,double a)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr + i) * a;
}
}
void CMatrixOper::MatrPointMulNumre(COMPLEX*Matr,int row,int colum,double a)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).re = (*(Matr + i)).re * a;
}
}
void CMatrixOper::MatrPointMulNumim(COMPLEX*Matr,int row,int colum,double a)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).im = (*(Matr + i)).im * a;
}
}
void CMatrixOper::MatrPointMulNumretoim(COMPLEX*Matr1,int row,int colum,double a,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).im = (*(Matr1 + i)).re * a;
}
}
void CMatrixOper::MatrGetre(double*Matr1,int row,int colum,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).re = cos(*(Matr1 + i));
}
}
void CMatrixOper::MatrGetim(double*Matr1,int row,int colum,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).im = sin(*(Matr1 + i));
}
}
void CMatrixOper::MatrPointMulNumimtore(COMPLEX*Matr1,int row,int colum,double a,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).re = (*(Matr1 + i)).im * a;
}
}
void CMatrixOper::MatrPointPow2(double*Matr1,int row,int colum,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr1 + i) * *(Matr1 + i);
}
}
void CMatrixOper::MatrPointPow2re(COMPLEX*Matr1,int row,int colum,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).re = (*(Matr1 + i)).re * (*(Matr1 + i)).re;
}
}
void CMatrixOper::MatrCOMreEquMatr(COMPLEX*Matr1,int row,int colum,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = (*(Matr1 + i)).re;
}
}
void CMatrixOper::MatrCOMimEquMatr(COMPLEX*Matr1,int row,int colum,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = (*(Matr1 + i)).im;
}
}
void CMatrixOper::MatrEqu(double*Matr1,int row,int colum,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = *(Matr1 + i);
}
}
void CMatrixOper::MatrPointPow2im(COMPLEX*Matr1,int row,int colum,COMPLEX*Matr)
{
for(int i = 0;i < row*colum;i++)
{
(*(Matr + i)).im = (*(Matr1 + i)).im * (*(Matr1 + i)).im;
}
}
void CMatrixOper::MatrPointSqrt(double*Matr1,int row,int colum,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = sqrt(*(Matr1 + i));
}
}
void CMatrixOper::MatrPointcos(double*Matr,int row,int colum)
{
for(int i = 0;i < row*colum;i++)
{
*(Matr + i) = cos(*(Matr + i));
}
}
void CMatrixOper::MatrPoinJudge(double*Matr,int row,int colum,double range1,double range2)
{
for(int i = 0;i < row*colum;i++)
{
if ((*(Matr + i) >= range1)&&(*(Matr + i)<=range2))
{
*(Matr + i) = 1;
}
else
{
*(Matr + i) = 0;
}
}
}
void CMatrixOper::MatrJudge(double*Matr1,int row,int colum,double*Matr)
{
for(int i = 0;i < row*colum;i++)
{
if ((*(Matr1 + i)<=*(Matr + i)))
{
*(Matr + i) = 1;
}
else
{
*(Matr + i) = 0;
}
}
}
void CMatrixOper::MatrPoinJudgeSmal(double*Matr,int row,int colum,double range)
{
for(int i = 0;i < row*colum;i++)
{
if (*(Matr + i)<range)
{
*(Matr + i) = 1;
}
else
{
?? 快捷鍵說明
復(fù)制代碼
Ctrl + C
搜索代碼
Ctrl + F
全屏模式
F11
切換主題
Ctrl + Shift + D
顯示快捷鍵
?
增大字號(hào)
Ctrl + =
減小字號(hào)
Ctrl + -