fftmethoddlg.cpp

嵌入式系统中FFT算法研究,看看对要使用VC++进行数字信号处理的朋友还是很实用的

// FftMethodDlg.cpp : implementation file
//
#include "math.h"
#include "string.h"
#include "stdafx.h"
#include "FftMethod.h"
#include "FftMethodDlg.h"
#define ORIGINAL_X 70
#define ORIGINAL_Y 280
#define FFT_ORIGINAL_Y 440
#define TERMINAL_X 600
#define UPPER_Y 90
#define BOTTOM_Y 480
#define Y_TIMES 60
#define X_TIMES 80
#define FFT_X_TIMES 15
#define FFT_Y_TIMES 10


#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CFftMethodDlg dialog

CFftMethodDlg::CFftMethodDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CFftMethodDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CFftMethodDlg)
	m_fre = _T("");
	m_number = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_scoordvisual=0;
	m_fftcoordvisual=0;
	m_f=0;
	m_n=0;
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
int CFftMethodDlg::ReversePow(int n)
{
	int count=0;
	while(n/2!=0)
	{
		count++;
		n=n/2;
	}
	return count;
}
void CFftMethodDlg::CreateSignal()
{
	int i;
	for(i=0;i<m_n;i++)
	{
		Xi[i]=0;
		Xr[i]=cos(4.0*PI2*i/m_f)+cos(8.0*PI2*i/m_f)+cos(10.0*PI2*i/m_f);
	}
}
double CFftMethodDlg::CalCulateSignal(double t)
{
	return cos(PI2*4.0*t)+cos(PI2*8.0*t)+cos(PI2*1.0*t);
}
void CFftMethodDlg::ChangeOrder()
{
	int LH,N1,I,J,K;
	double T;
	LH=m_n/2;
	J=LH;
	N1=m_n-2;
	for(I=1;I<=N1;I++)
	{
		if(I<J)
		{
			T=Xr[I];  Xr[I]=Xr[J];  Xr[J]=T;
			T=Xi[I];  Xi[I]=Xi[J];  Xi[J]=T;
		}
		K=LH;
		while(J>=K)
		{
			J=J-K;
			K=(int)(K/2+0.5);
		}
		J=J+K;
	}
}
void CFftMethodDlg::FftTransfer()
{
	int L,B,J,P,k,i;
	double rPartKB,iPartKB;
	double *rcf,*icf;
	rcf=new double [m_n];
	icf=new double [m_n];
	for(i=0;i<m_n;i++)
	{
		rcf[i]=cos(i*PI2/m_n);
		icf[i]=sin(i*PI2/m_n);
	}
	ChangeOrder();
	for(L=1;L<=ReversePow(m_n);L++)  //M=4
	{
		B=(int)(pow(2,(L-1))+0.5);
		for(J=0;J<=B-1;J++)
		{
			P=J*((int)(pow(2,(ReversePow(m_n)-L))+0.5));
			for(k=J;k<=m_n-1;k+=(int)(pow(2,L)+0.5))
			{
				rPartKB=Xr[k+B]*rcf[P]-Xi[k+B]*icf[P];
				iPartKB=Xi[k+B]*rcf[P]+Xr[k+B]*icf[P];
				Xr[k+B]=Xr[k]-rPartKB;
				Xi[k+B]=Xi[k]-iPartKB;
				Xr[k]=Xr[k]+rPartKB;
				Xi[k]=Xi[k]+iPartKB;
			}
		}
	}
}
CFftMethodDlg::~CFftMethodDlg()
{
	
}
void CFftMethodDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CFftMethodDlg)
	DDX_Text(pDX, IDC_EDIT1, m_fre);
	DDX_Text(pDX, IDC_EDIT2, m_number);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CFftMethodDlg, CDialog)
	//{{AFX_MSG_MAP(CFftMethodDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON1, OnButton1)
	ON_BN_CLICKED(IDC_BUTTON2, OnButton2)
	ON_BN_CLICKED(IDC_BUTTON4, OnButton4)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CFftMethodDlg message handlers
void CFftMethodDlg::paint()
{
	if(m_scoordvisual==1)
	{
		CDC *dc;
	    dc=GetDC();
	    dc->TextOut(ORIGINAL_X+3,UPPER_Y+10,"x(t)");
	    dc->TextOut(TERMINAL_X-15,ORIGINAL_Y-15,"t");
		dc->TextOut(ORIGINAL_X-8,ORIGINAL_Y-Y_TIMES,"1");
		dc->TextOut(ORIGINAL_X-8,ORIGINAL_Y-2*Y_TIMES,"2");
		dc->TextOut(ORIGINAL_X-8,ORIGINAL_Y-3*Y_TIMES,"3");
	    dc->MoveTo(ORIGINAL_X,ORIGINAL_Y);
	    dc->LineTo(TERMINAL_X,ORIGINAL_Y);
	    dc->LineTo(TERMINAL_X-10,ORIGINAL_Y-7);
	    dc->MoveTo(TERMINAL_X,ORIGINAL_Y);
	    dc->LineTo(TERMINAL_X-10,ORIGINAL_Y+7);
	    dc->MoveTo(ORIGINAL_X,BOTTOM_Y);
	    dc->LineTo(ORIGINAL_X,UPPER_Y);
	    dc->LineTo(ORIGINAL_X-7,UPPER_Y+10);
	    dc->MoveTo(ORIGINAL_X,UPPER_Y);
	    dc->LineTo(ORIGINAL_X+7,UPPER_Y+10);
	
	}
	if(m_signalvisual==1)
	{
		CDC *dc;
	    dc=GetDC();
		CPen pen;
		pen.CreatePen(PS_SOLID,1,RGB(255,0,0));
		dc->SelectObject(&pen);
		dc->MoveTo(ORIGINAL_X,ORIGINAL_Y);
		for(int i=0;i<=520;i++)
		{
			dc->LineTo(i+ORIGINAL_X,ORIGINAL_Y-(int)CalCulateSignal((double)i/X_TIMES)*Y_TIMES);
		}
	}
	if(m_fftcoordvisual==1)
	{
		CDC *dc;
	    dc=GetDC();
		char index[5];
		char n[5];
		char tip[30];
        strcpy(tip,"FFT变换区间 N = ");
		sprintf(n,"%d",m_n);
		strcat(tip,n);
		dc->TextOut(250,100,tip);
	    dc->TextOut(ORIGINAL_X+3,UPPER_Y+10,"|x(k)|");
	    dc->TextOut(TERMINAL_X-15,FFT_ORIGINAL_Y-15,"k");
		
		for(int w=2;w<=(FFT_ORIGINAL_Y-UPPER_Y)/FFT_Y_TIMES;w+=2)
		{
			sprintf(index,"%d",w);
			dc->TextOut(ORIGINAL_X-18,FFT_ORIGINAL_Y-w*FFT_Y_TIMES,index);
		}
	    dc->MoveTo(ORIGINAL_X,FFT_ORIGINAL_Y);
	    dc->LineTo(TERMINAL_X,FFT_ORIGINAL_Y);
	    dc->LineTo(TERMINAL_X-10,FFT_ORIGINAL_Y-7);
	    dc->MoveTo(TERMINAL_X,FFT_ORIGINAL_Y);
	    dc->LineTo(TERMINAL_X-10,FFT_ORIGINAL_Y+7);
	    dc->MoveTo(ORIGINAL_X,BOTTOM_Y);
	    dc->LineTo(ORIGINAL_X,UPPER_Y);
	    dc->LineTo(ORIGINAL_X-7,UPPER_Y+10);
	    dc->MoveTo(ORIGINAL_X,UPPER_Y);
	    dc->LineTo(ORIGINAL_X+7,UPPER_Y+10);
	}
	if(m_fftvisual==1)
	{
		CDC *dc;
	    dc=GetDC();
		CPen pen;
		pen.CreatePen(PS_SOLID,3,RGB(255,0,255));
		dc->SelectObject(&pen);
		char k[5];
		for(int j=0;j<m_n;j++)
		{
			
			sprintf(k,"%d",j);
			dc->TextOut(ORIGINAL_X+j*FFT_X_TIMES-3,FFT_ORIGINAL_Y+3,k);
			dc->SetPixel(ORIGINAL_X+j*FFT_X_TIMES,FFT_ORIGINAL_Y-(int)(sqrt(Xr[j]*Xr[j]+Xi[j]*Xi[j])*FFT_Y_TIMES),RGB(255,0,0));
		    dc->MoveTo(ORIGINAL_X+j*FFT_X_TIMES,FFT_ORIGINAL_Y);
			dc->LineTo(ORIGINAL_X+j*FFT_X_TIMES,FFT_ORIGINAL_Y-(int)(sqrt(Xr[j]*Xr[j]+Xi[j]*Xi[j])*FFT_Y_TIMES));
		}
	}
}
BOOL CFftMethodDlg::OnInitDialog()
{
	CDialog::OnInitDialog();
    
	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CFftMethodDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CFftMethodDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CFftMethodDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CFftMethodDlg::OnButton1() 
{
	// TODO: Add your control notification handler code here
	UpdateData(true);
	m_scoordvisual=1;
	m_fftcoordvisual=0;
	m_signalvisual=1;
	m_fftvisual=0;
	m_f=atoi(m_fre);
	m_n=atoi(m_number);
	if((m_f==0)||(m_n==0))
	{
		MessageBox("请在上面对话框内输入合适的频率和变换区间长度","提示");
		return;
	}
    PI2=8.0*atan(1.0);
	paint();
	
	
	
}

void CFftMethodDlg::OnButton2() 
{
	// TODO: Add your control notification handler code here
	
    UpdateData(true);
	m_fftcoordvisual=1;
	m_scoordvisual=0;
	m_signalvisual=0;
	m_fftvisual=1;
	m_f=atoi(m_fre);
	m_n=atoi(m_number);
	if((m_f==0)||(m_n==0))
	{
		MessageBox("请在上面对话框内输入合适的频率和变换区间长度","提示");
		return;
	}
	Rcf= new double [m_n];
	Icf= new double [m_n];
	Xr= new double [m_n];
	Xi= new double [m_n];
    PI2=8.0*atan(1.0);
	CreateSignal();
	FftTransfer();
	paint();
}

void CFftMethodDlg::OnButton4() 
{
	// TODO: Add your control notification handler code here
	Invalidate(TRUE);
	
}