fftcg.cpp

选自<gpu gemes 2>,用gpu实现快速傅立叶变换

/***************************************************************************
*        FILE NAME:  FFTCg.cpp
*
* ONE LINE SUMMARY:
*        This file contains Cg-related member functions for the FFT class.
*        
*        Thilaka Sumanaweera
*        Siemens Medical Solutions USA, Inc.
*        1230 Shorebird Way
*        Mountain View, CA 94039
*        USA
*        [e-mail]
*
* DESCRIPTION:
*
*****************************************************************************
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
****************************************************************************/
#include <math.h>
#include <iostream>
#include <fstream>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include "FFT.h"

// The shaders
#include "FFTfragmentX_CG1.h"
#include "FFTfragmentX_CG2.h"
#include "FFTfragmentY_CG1.h"
#include "FFTfragmentY_CG2.h"
#include "DispFragment_CG.h"

using namespace std;

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::InitCg
*
* DESCRIPTION:
*	Initialize Cg variables
*
* FORMAL PARAMETERS:
*	none
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::InitCg(void)
{	
	Context = NULL;
	dfProgram = NULL;
	DispTexR1 = NULL;
	DispTexI1 = NULL;
	DispTexR2 = NULL;
	DispTexI2 = NULL;
	DispInvEnergy = NULL;
	dfProfile = CG_PROFILE_FP40;

	xfProgram1 = NULL;
	xBaseTexR1_1 = NULL;
	xBaseTexI1_1 = NULL;
	xBaseTexR2_1 = NULL;
	xBaseTexI2_1 = NULL;
	xButterflyLookupI_1 = NULL;
	xButterflyLookupWR_1 = NULL;
	xButterflyLookupWI_1 = NULL;
	xfProgram2 = NULL;
	xBaseTexR1_2 = NULL;
	xBaseTexI1_2 = NULL;
	xBaseTexR2_2 = NULL;
	xBaseTexI2_2 = NULL;
	xfProfile = CG_PROFILE_FP40;
	
	yfProgram1 = NULL;
	yBaseTexR1_1 = NULL;
	yBaseTexI1_1 = NULL;
	yBaseTexR2_1 = NULL;
	yBaseTexI2_1 = NULL;
	yButterflyLookupI_1 = NULL;
	yButterflyLookupWR_1 = NULL;
	yButterflyLookupWI_1 = NULL;
	
	yfProgram2 = NULL;
	yBaseTexR1_2 = NULL;
	yBaseTexI1_2 = NULL;
	yBaseTexR2_2 = NULL;
	yBaseTexI2_2 = NULL;
	yfProfile = CG_PROFILE_FP40;
}

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::WriteObject
*
* DESCRIPTION:
*	Writes the object file of a shader into a file
*
* FORMAL PARAMETERS:
*	ofilename:	file name
*	oProgram:	Cg program
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::WriteObject(char *ofilename, CGprogram oProgram)
{
	ofstream ofile;
	ofile.open(ofilename, ios::out);
	ofile << cgGetProgramString(oProgram, CG_COMPILED_PROGRAM) << endl;
	ofile.close();
}

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::InitFFTFragmentCgX1
*
* DESCRIPTION:
*	Initializes the fragment shader for FFT in x for method 1
*
* FORMAL PARAMETERS:
*	none
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::InitFFTFragmentCgX1(void)
{
	//cerr << "*** Creating fragment program for X ***" << endl;
	xfProfile = cgGLGetLatestProfile(CG_GL_FRAGMENT);
	cgGLSetOptimalOptions(xfProfile);
	
	xfProgram1 = cgCreateProgram(Context, CG_SOURCE, FFTfragmentX_CG1, xfProfile, "FragmentProgram", 0);
	CheckCgError();

	/*
	cerr << "----PROGRAM BEGIN----" << endl;
	cerr << cgGetProgramString(xfProgram1, CG_COMPILED_PROGRAM) << endl;
	cerr << "----PROGRAM END  ----" << endl;
	*/

	//char file[200] = "FFTfragmentX_CG1.ocg";
	//WriteObject(file, xfProgram);

	if (xfProgram1 != NULL) {
		cgGLLoadProgram(xfProgram1);
		CheckCgError();
		xBaseTexR1_1 = cgGetNamedParameter(xfProgram1, "Real1");
		CheckCgError();
		xBaseTexI1_1 = cgGetNamedParameter(xfProgram1, "Imag1");
		CheckCgError();
		xBaseTexR2_1 = cgGetNamedParameter(xfProgram1, "Real2");
		CheckCgError();
		xBaseTexI2_1 = cgGetNamedParameter(xfProgram1, "Imag2");
		CheckCgError();
		xButterflyLookupI_1 = cgGetNamedParameter(xfProgram1, "ButterflyLookupI");
		CheckCgError();
		xButterflyLookupWR_1 = cgGetNamedParameter(xfProgram1, "ButterflyLookupWR");
		CheckCgError();
		xButterflyLookupWI_1 = cgGetNamedParameter(xfProgram1, "ButterflyLookupWI");
		CheckCgError();
	}
}

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::InitFFTFragmentCgX2
*
* DESCRIPTION:
*	Initializes the fragment shader for FFT in x for method 2
*
* FORMAL PARAMETERS:
*	none
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::InitFFTFragmentCgX2(void)
{
	//cerr << "*** Creating fragment program for X ***" << endl;
	xfProfile = cgGLGetLatestProfile(CG_GL_FRAGMENT);
	cgGLSetOptimalOptions(xfProfile);

	xfProgram2 = cgCreateProgram(Context, CG_SOURCE, FFTfragmentX_CG2, xfProfile, "FragmentProgram", 0);
	CheckCgError();

	/*
	cerr << "----PROGRAM BEGIN----" << endl;
	cerr << cgGetProgramString(xfProgram2, CG_COMPILED_PROGRAM) << endl;
	cerr << "----PROGRAM END  ----" << endl;
	*/

	//char file[200] = "FFTfragmentX_CG2.ocg";
	//WriteObject(file, xfProgram2);

	if (xfProgram2 != NULL) {
		cgGLLoadProgram(xfProgram2);
		CheckCgError();
		xBaseTexR1_2 = cgGetNamedParameter(xfProgram2, "Real1");
		CheckCgError();
		xBaseTexI1_2 = cgGetNamedParameter(xfProgram2, "Imag1");
		CheckCgError();
		xBaseTexR2_2 = cgGetNamedParameter(xfProgram2, "Real2");
		CheckCgError();
		xBaseTexI2_2 = cgGetNamedParameter(xfProgram2, "Imag2");
		CheckCgError();
	}
}

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::InitFFTFragmentCgY1
*
* DESCRIPTION:
*	Initializes the fragment shader for FFT in y for method 1
*
* FORMAL PARAMETERS:
*	none
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::InitFFTFragmentCgY1(void)
{
	//cerr << "*** Creating fragment program for Y ***" << endl;
	yfProfile = cgGLGetLatestProfile(CG_GL_FRAGMENT);
	cgGLSetOptimalOptions(yfProfile);
	
	yfProgram1 = cgCreateProgram(Context, CG_SOURCE, FFTfragmentY_CG1, yfProfile, "FragmentProgram", 0);
	CheckCgError();
	
	/*
	cerr << "----PROGRAM BEGIN----" << endl;
	cerr << cgGetProgramString(yfProgram1, CG_COMPILED_PROGRAM) << endl;
	cerr << "----PROGRAM END  ----" << endl;
	*/

	//char file[200] = "FFTfragmentY_CG1.ocg";
	//WriteObject(file, yfProgram);
	
	if (yfProgram1 != NULL) {
		cgGLLoadProgram(yfProgram1);
		CheckCgError();
		yBaseTexR1_1 = cgGetNamedParameter(yfProgram1, "Real1");
		CheckCgError();
		yBaseTexI1_1 = cgGetNamedParameter(yfProgram1, "Imag1");
		CheckCgError();
		yBaseTexR2_1 = cgGetNamedParameter(yfProgram1, "Real2");
		CheckCgError();
		yBaseTexI2_1 = cgGetNamedParameter(yfProgram1, "Imag2");
		CheckCgError();
		yButterflyLookupI_1 = cgGetNamedParameter(yfProgram1, "ButterflyLookupI");
		CheckCgError();
		yButterflyLookupWR_1 = cgGetNamedParameter(yfProgram1, "ButterflyLookupWR");
		CheckCgError();
		yButterflyLookupWI_1 = cgGetNamedParameter(yfProgram1, "ButterflyLookupWI");
		CheckCgError();
	}
}

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::InitFFTFragmentCgY2
*
* DESCRIPTION:
*	Initializes the fragment shader for FFT in y for method 2
*
* FORMAL PARAMETERS:
*	none
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::InitFFTFragmentCgY2(void)
{
	//cerr << "*** Creating fragment program for Y ***" << endl;
	yfProfile = cgGLGetLatestProfile(CG_GL_FRAGMENT);
	cgGLSetOptimalOptions(yfProfile);
	
	yfProgram2 = cgCreateProgram(Context, CG_SOURCE, FFTfragmentY_CG2, yfProfile, "FragmentProgram", 0);
	CheckCgError();
	
	/*
	cerr << "----PROGRAM BEGIN----" << endl;
	cerr << cgGetProgramString(yfProgram2, CG_COMPILED_PROGRAM) << endl;
	cerr << "----PROGRAM END  ----" << endl;
	*/

	//char file[200] = "FFTfragmentY_CG2.ocg";
	//WriteObject(file, yfProgram2);
	
	if (yfProgram2 != NULL) {
		cgGLLoadProgram(yfProgram2);
		CheckCgError();
		yBaseTexR1_2 = cgGetNamedParameter(yfProgram2, "Real1");
		CheckCgError();
		yBaseTexI1_2 = cgGetNamedParameter(yfProgram2, "Imag1");
		CheckCgError();
		yBaseTexR2_2 = cgGetNamedParameter(yfProgram2, "Real2");
		CheckCgError();
		yBaseTexI2_2 = cgGetNamedParameter(yfProgram2, "Imag2");
		CheckCgError();
	}
}

/*************************[MAN-BEG]*****************************************
*
* NAME:
*	FFT::InitDispFragmentCg
*
* DESCRIPTION:
*	Initializes the fragment shader displaying the result.
*
* FORMAL PARAMETERS:
*	none
*
* RETURNS:
*	none
*
* REVISION HISTORY:
* Rev     When      Who         What
* V1      15Dec2004 Thilaka     Created.
**************************[MAN-END]*****************************************/
void FFT::InitDispFragmentCg(void)
{
	//cerr << "*** Creating fragment program for display ***" << endl;
	dfProfile = cgGLGetLatestProfile(CG_GL_FRAGMENT);
	cgGLSetOptimalOptions(dfProfile);
	
	dfProgram = cgCreateProgram(Context, CG_SOURCE, DispFragment_CG, dfProfile, "FragmentProgram", 0);
	CheckCgError();
	
	/*
	cerr << "----PROGRAM BEGIN----" << endl;
	cerr << cgGetProgramString(dfProgram, CG_COMPILED_PROGRAM) << endl;
	cerr << "----PROGRAM END  ----" << endl;
	*/

	//char file[200] = "Dispfragment_CG.ocg";
	//WriteObject(file, dfProgram);

	if (dfProgram != NULL) {
		cgGLLoadProgram(dfProgram);
		CheckCgError();