代ç å¦ä¸ï¼
// DWT.cpp : å®ä¹æ§å¶å°åºç¨ç¨åºçå
¥å£ç¹ã
//
#include "stdafx.h"
#include "cv.h"
#include "highgui.h"
// äºç»´ç¦»æ£å°æ³¢åæ¢ï¼åééæµ®ç¹å¾åï¼
void DWT(IplImage *pImage, int nLayer)
{
// æ§è¡æ¡ä»¶
if (pImage)
{
if (pImage->nChannels == 1 &&
pImage->depth == IPL_DEPTH_32F &&
((pImage->width >> nLayer) << nLayer) == pImage->width &&
((pImage->height >> nLayer) << nLayer) == pImage->height)
{
int i, x, y, n;
float fValue = 0;
float fRadius = sqrt(2.0f);
int nWidth = pImage->width;
int nHeight = pImage->height;
int nHalfW = nWidth / 2;
int nHalfH = nHeight / 2;
float **pData = new float*[pImage->height];
float *pRow = new float[pImage->width];
float *pColumn = new float[pImage->height];
for (i = 0; i < pImage->height; i++)
{
pData[i] = (float*) (pImage->imageData + pImage->widthStep * i);
}
// å¤å±å°æ³¢åæ¢
for (n = 0; n < nLayer; n++, nWidth /= 2, nHeight /= 2, nHalfW /= 2, nHalfH /= 2)
{
// æ°´å¹³åæ¢
for (y = 0; y < nHeight; y++)
{
// å¥å¶å离
memcpy(pRow, pData[y], sizeof(float) * nWidth);
for (i = 0; i < nHalfW; i++)
{
x = i * 2;
pData[y][i] = pRow[x];
pData[y][nHalfW + i] = pRow[x + 1];
}
// æåå°æ³¢åæ¢
for (i = 0; i < nHalfW - 1; i++)
{
fValue = (pData[y][i] + pData[y][i + 1]) / 2;
pData[y][nHalfW + i] -= fValue;
}
fValue = (pData[y][nHalfW - 1] + pData[y][nHalfW - 2]) / 2;
pData[y][nWidth - 1] -= fValue;
fValue = (pData[y][nHalfW] + pData[y][nHalfW + 1]) / 4;
pData[y][0] += fValue;
for (i = 1; i < nHalfW; i++)
{
fValue = (pData[y][nHalfW + i] + pData[y][nHalfW + i - 1]) / 4;
pData[y][i] += fValue;
}
// é¢å¸¦ç³»æ°
for (i = 0; i < nHalfW; i++)
{
pData[y][i] *= fRadius;
pData[y][nHalfW + i] /= fRadius;
}
}
// åç´åæ¢
for (x = 0; x < nWidth; x++)
{
// å¥å¶å离
for (i = 0; i < nHalfH; i++)
{
y = i * 2;
pColumn[i] = pData[y][x];
pColumn[nHalfH + i] = pData[y + 1][x];
}
for (i = 0; i < nHeight; i++)
{
pData[i][x] = pColumn[i];
}
// æåå°æ³¢åæ¢
for (i = 0; i < nHalfH - 1; i++)
{
fValue = (pData[i][x] + pData[i + 1][x]) / 2;
pData[nHalfH + i][x] -= fValue;
}
fValue = (pData[nHalfH - 1][x] + pData[nHalfH - 2][x]) / 2;
pData[nHeight - 1][x] -= fValue;
fValue = (pData[nHalfH][x] + pData[nHalfH + 1][x]) / 4;
pData[0][x] += fValue;
for (i = 1; i < nHalfH; i++)
{
fValue = (pData[nHalfH + i][x] + pData[nHalfH + i - 1][x]) / 4;
pData[i][x] += fValue;
}
// é¢å¸¦ç³»æ°
for (i = 0; i < nHalfH; i++)
{
pData[i][x] *= fRadius;
pData[nHalfH + i][x] /= fRadius;
}
}
}
delete[] pData;
delete[] pRow;
delete[] pColumn;
}
}
}
int _tmain(int argc, _TCHAR* argv[])
{
// å°æ³¢åæ¢å±æ°
int nLayer = 2;
// è¾å
¥å½©è²å¾å
IplImage *pSrc = cvLoadImage("lena.jpg", 1);
// 计ç®å°æ³¢å¾è±¡å¤§å°ï¼ä½¿å
¶widthåheighté½æ¯2çåæ°
CvSize size = cvGetSize(pSrc);
if ((pSrc->width >> nLayer) << nLayer != pSrc->width)
{
size.width = ((pSrc->width >> nLayer) + 1) << nLayer;
}
if ((pSrc->height >> nLayer) << nLayer != pSrc->height)
{
size.height = ((pSrc->height >> nLayer) + 1) << nLayer;
}
// å建å°æ³¢å¾è±¡
IplImage *pWavelet = cvCreateImage(size, IPL_DEPTH_32F, pSrc->nChannels);
if (pWavelet)
{
// å°æ³¢å¾è±¡èµå¼
cvSetImageROI(pWavelet, cvRect(0, 0, pSrc->width, pSrc->height));
cvConvertScale(pSrc, pWavelet, 1, -128);//使ç¨çº¿æ§åæ¢è½¬æ¢æ°ç»ï¼pWavelet = pSrc*1-128
cvResetImageROI(pWavelet);
// 彩è²å¾åå°æ³¢åæ¢
IplImage *pImage = cvCreateImage(cvGetSize(pWavelet), IPL_DEPTH_32F, 1);
if (pImage)
{
for (int i = 1; i <= pWavelet->nChannels; i++)
{
cvSetImageCOI(pWavelet, i);//设置æå
´è¶£ééchannel
cvCopy(pWavelet, pImage, NULL);//pImage为ç°åº¦å¾åï¼å°pWaveletçæ¯ä¸ªééæ°æ®copyå°pImageä¸
// äºç»´ç¦»æ£å°æ³¢åæ¢
DWT(pImage, nLayer);//对æ¯ä¸ªééè¿è¡DWT
// äºç»´ç¦»æ£å°æ³¢æ¢å¤
// IDWT(pImage, nLayer);
cvCopy(pImage, pWavelet, NULL);//å°æ¯ä¸ªééåæ¢åçæ°æ®åå
¥pWaveletç对åºééä¸
}
cvSetImageCOI(pWavelet, 0);
cvReleaseImage(&pImage);
}
// å°æ³¢åæ¢å¾è±¡
cvSetImageROI(pWavelet, cvRect(0, 0, pSrc->width, pSrc->height));
cvConvertScale(pWavelet, pSrc, 1, 128);
cvResetImageROI(pWavelet); // æ¬è¡ä»£ç æç¹å¤ä½ï¼ä½æå©ç¨å
»æè¯å¥½çç¼ç¨ä¹ æ¯
cvReleaseImage(&pWavelet);
}
// æ¾ç¤ºå¾åpSrc
cvNamedWindow("dwt",1);
cvShowImage("dwt",pSrc);
cvWaitKey(0);
cvDestroyWindow("dwt");
// ...
cvReleaseImage(&pSrc);
cvReleaseImage(&pSrc0);
return 0;
}
温馨提示:答案为网友推荐,仅供参考