波变换

开发平台：
Visual C++

Diproc.cpp：源码内容
							// Diproc.cpp: implementation of the CDiproc class.
//
//////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "小波变换.h"
#include "Diproc.h"
#include "WvltTrans.h"
#include <math.h>
#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CDiproc::CDiproc()
{
}
CDiproc::~CDiproc()
{
}
/********************************************************************************
*函数描述：	DIP_WvltRevers完成图像小波系数的复原，恢复出原始的图像数据			*
*函数参数：	short **spData		：二维指针，指向原始的图像数据					*
*			short **spTransData0：小波变换系数，存放一次水平变换后的小波系数	*
*			short **spTransData1：小波变换系数，存放一次数值变换后的小波系数	*
*			int   nHeight		：图像属性参数，数值为原始图像的高度值			*
*			int	  nHeight_H		：图像属性参数，数值为原始图像高度值的一半		*
*			int   nWidth		：图像属性参数，数值为原始图像的宽度值			*
*			int	  nWidth_H		：图像属性参数，数值为原始图像宽度值的一半		*
*			int   layer			：小波变换的层数,数值为3层						*
*			float fRadius		：小波变换因子，在调用时候已指定数值为1.414		*
********************************************************************************/
void CDiproc::DIP_WvltRevers(short **spData, short **spTransData0, short **spTransData1, int nHeight, int nHeight_H, int nWidth, int nWidth_H, int layer, float fRadius)
{
	short **spOriginData, **spTransData, **spWvltData;
	int iHeight = (int)nHeight /pow(2,layer-1), iWidth =(int)nWidth / pow(2,layer-1);
	int iHeight_H =(int) nHeight_H / pow(2,layer-1), iWidth_H = (int)nWidth_H/ pow(2,layer-1);
	//分配图像复原所需的内存空间
	spOriginData = spData;
	spTransData = spTransData0;
	spWvltData = spTransData1;
	//完成图像小波变换的逆变换
	CWvltTrans *WTrans;
	for(int i = layer; i >= 1; i--)
	{
		WTrans->DWTi_Once(spOriginData, spTransData, spWvltData, iHeight, iHeight_H, iWidth, iWidth_H, i, 1.414);
		iHeight <<= 1;		iWidth <<= 1;
		iHeight_H <<= 1;	iWidth_H <<= 1;
	}
		
}
/********************************************************************************
*函数描述：	DIP_ImageFusion完成两幅图像的融合，恢复出原始的图像					*
*函数参数：	short **spImgData0	：二维指针，存放其中一幅原始图像的数据			*
*			short **spImgData1	：二维指针，存放其中另外一幅原始图像的数据		*
*			int   nHeight		：图像属性参数，数值为原始图像的高度值			*
*			int   nWidth		：图像属性参数，数值为原始图像的宽度值			*
********************************************************************************/
void CDiproc::DIP_ImageFusion(short **spImgData0, short **spImgData1, int nHeight, int nWidth)
{
	//获取图像的属性参数
	int iHeight = nHeight, iWidth = nWidth;
	//图像融合所用到的数据空间及数据指针
	short **spOriginData, **spTransData, **spWvltData0, **spWvltData1;
	//分配数据空间
	spTransData = new short *[iHeight];
	spWvltData0 = new short *[iHeight];
	spWvltData1 = new short *[iHeight];
	for(int i = 0; i < iWidth; i ++)
	{
		spTransData[i] = new short [iWidth];
		spWvltData0[i] = new short [iWidth];
		spWvltData1[i] = new short [iWidth];
	}
	//创建小波变换类，完成图像的小波变换
	CWvltTrans *pTrans;
	//获得图像数据空间的指针，完成小波变换
	spOriginData = spImgData0;
	//三层小波变换
	pTrans->DWT_TriLayers(spOriginData, spTransData, spWvltData0, iHeight, iHeight / 2, iWidth, iWidth / 2, 1, 1.414);
	//获得图像数据空间的指针，完成另一幅图像的小波变换
	spOriginData = spImgData1;
	//三层小波变换
	pTrans->DWT_TriLayers(spOriginData, spTransData, spWvltData1, iHeight,iHeight / 2, iWidth, iWidth / 2, 1, 1.414);
	//小波系数的融合处理：频带有LL3,LH3,HL3,HH3,LH2,HL2,HH2,LH1,HL1,HH1
	//融合处理将分频带进行，处理方法采用的是3*3的窗口
	//LL3频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, 0, 0, iHeight / 8, iWidth / 8);
	//HL3频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, 0, iWidth / 8, iHeight / 8, iWidth / 4);
	//LH3频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, iHeight / 8, 0, iHeight / 4, iWidth / 8);
	//HH3频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, iHeight / 8, iWidth / 8, iHeight / 4, iWidth / 4);
	//HL2频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, 0, iWidth / 4, iHeight / 4, iWidth / 2);
	//LH2频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, iHeight / 4, 0, iHeight / 2, iWidth / 4);
	//HH2频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, iHeight / 4, iWidth / 4, iHeight / 2, iWidth / 2);
	//HL1频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, 0, iWidth / 2, iHeight / 2, iWidth);
	//LH1频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, iHeight / 2, 0, iHeight, iWidth / 2);
	//HH1频带小波系数的融合
	Window_WvltFusion(spWvltData0, spWvltData1, iHeight / 2, iWidth / 2, iHeight, iWidth);
	//将融合后的小波系数复原，完成图像的融合
	DIP_WvltRevers(spOriginData, spTransData, spWvltData0, iHeight, iHeight / 2, iWidth, iWidth / 2, 1, 1.414);
	//释放临时的数据空间
	delete spTransData;
	delete spWvltData0;
	delete spWvltData1;
}
/********************************************************************************
*函数描述：	Wvlt_Normalize完成图像小波数据与逆变换数据的正则化处理				*
*函数参数：	short **spWvltNormData：二维指针，存放正则化后的小波系数			*
*			int   nHeight		：图像属性参数，数值为原始图像的高度值			*
*			int	  nHeight_H		：图像属性参数，数值为原始图像高度值的一半		*
*			int   nWidth		：图像属性参数，数值为原始图像的宽度值			*
*			float *NormWvltRng	：存放图像小波的正则化参数以及逆变换的正则化参数*
********************************************************************************/
void CDiproc::Wvlt_Normalize(short **spWvltNormData, int nHeight, int nWidth, float *nWvltRng)
{
	int MaxPixVal, MinPixVal, Diff;
	int x, y;
	float NormCoeff;
	MaxPixVal=spWvltNormData[0][0];
	MinPixVal=spWvltNormData[0][0];
	for( y=0; y < nHeight; y++)
	{
		for( x=0; x < nWidth; x++)
		{
			if(MaxPixVal<spWvltNormData[y][x])
				MaxPixVal=spWvltNormData[y][x];
			if(MinPixVal>spWvltNormData[y][x])
				MinPixVal=spWvltNormData[y][x];
			//spWvltData[y][x] = spWvltNormData[y][x];
		}
	}
	Diff=MaxPixVal-MinPixVal;
	nWvltRng[1] = MaxPixVal;
	nWvltRng[0] = MinPixVal;
	for(y=0; y < nHeight; y++)
	{
		for(x=0; x < nWidth; x++)
		{
		//因为小波变换后的小波系数有可能超过255甚至更多，那么就将
		//小波系数的范围映射到0~255区间内，以后出现类似的处理，目的都是一样的
			NormCoeff = spWvltNormData[y][x];
			NormCoeff -= MinPixVal;
			NormCoeff *= 255;
			NormCoeff /= (float) Diff;
			spWvltNormData[y][x] = NormCoeff;
		}
	}
}
/*********************************************************************************
*函数描述：	Window_WvltFusion完成图像小波系数的融合操作，得到各频带的小波融合数据*
*函数参数：	short **spWvltData0 ：二维指针，存放其中一幅图像的原始小波系数		 *
*			short **spWvltData1 ：二维指针，存放其中另外一幅图像的原始小波系数	 *
*			int   Scan_y		：扫描线起始横坐标								 *
*			int   Scan_x		：扫描线起始纵坐标								 *
*			int   End_y			：扫描线终止横坐标								 *
*			int   End_x			：扫描线终止纵坐标								 *
*********************************************************************************/
void CDiproc::Window_WvltFusion(short **spWvltData0, short **spWvltData1, int Scan_y, int Scan_x, int End_y, int End_x)
{
	int y,x;
	short WndSum0, WndSum1;
	for(y = Scan_y; y < End_y; y ++)
	{
		for(x = Scan_x; x < End_x; x ++)
		{
			//初始化窗口中小波系数的和
			WndSum0 = 0;	WndSum1 = 0;
			//计算窗口中小波系数的和
			for(int i = -1; i <= 1; i++)
			{
				for(int j = -1; j <= 1; j++)
				{
					if( (y+i) < Scan_y || (x+j) < Scan_x || (y+i) >= End_y || (x+j) >= End_x)
					{
						WndSum0 += 0;
						WndSum1 += 0;
					}
					else
					{
						WndSum0 += abs((int)spWvltData0[y + i][x + j]);
						WndSum1 += abs((int)spWvltData1[y + i][x + j]);
					}
				}
			}
			if(WndSum0 < WndSum1)
				spWvltData0[y][x] = spWvltData1[y][x];
		}
	}
}
float CDiproc::Search_BandMax(short **spWvltData, int Scan_y, int Scan_x, int End_y, int End_x)
{
	int x, y;
	float Band_max;
	Band_max = 0;
	for(y = Scan_y; y < End_y; y ++)
	{
		for(x = Scan_x; x < End_x; x ++)
		{
			if(Band_max < spWvltData[y][x])
				Band_max = spWvltData[y][x];
		}
	}
	return Band_max;
}
void CDiproc::Band_Enhance(float **fpNormGradient, float FilterCoeff, int Scan_y, int Scan_x, int End_y, int End_x)
{
	for(int y = Scan_y; y < End_y; y ++)
		for(int x = Scan_x; x < End_x; x ++)
			fpNormGradient[y][x] *= (float)(255.0 * FilterCoeff);	
}

						