// ImageProcessingView.cpp : implementation of the CImageProcessingView class
//

#include "stdafx.h"
#include "ImageProcessing.h"

#include "ImageProcessingDoc.h"
#include "ImageProcessingView.h"
#include  "GlobalApi.h"
#include  "DlgCoding.h"
#include <complex>
using namespace std;

#include "DlgHistShow1.h"
#include "DlgSmooth.h"
#include "DlgMedian.h"
#include "DlgEnhColor.h"
#include "DlgEhnLinTrans.h"
#include "DlgReg.h"
#include "DlgRecMatch.h"

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

/////////////////////////////////////////////////////////////////////////////
// CImageProcessingView

IMPLEMENT_DYNCREATE(CImageProcessingView, CScrollView)

BEGIN_MESSAGE_MAP(CImageProcessingView, CScrollView)
	//{{AFX_MSG_MAP(CImageProcessingView)
	ON_COMMAND(ID_FFT_2D, OnFft2d)
	ON_COMMAND(ID_DFT_2D, OnDft2d)
	ON_COMMAND(ID_VIEW_HISTOGRAM, OnViewHistogram)
	ON_COMMAND(ID_ENHANCE_SMOOTH, OnEnhanceSmooth)
	ON_COMMAND(ID_ENHANCE_MEDIAN, OnEnhanceMedian)
	ON_COMMAND(ID_ENHANCE_PSEUDCOLOR, OnEnhancePseudcolor)
	ON_COMMAND(ID_TRANS_DWT, OnTransDwt)
	ON_COMMAND(ID_TRANS_IDWT, OnTransIdwt)
	ON_COMMAND(IDC_ENHANCE_LINTRANS, OnEnhanceLintrans)
	ON_COMMAND(IDC_ENHANCE_HISTEQU, OnEnhanceHistequ)
	ON_COMMAND(ID_REG_REG, OnRegReg)
	ON_COMMAND(IDC_ENHANCE_SHARP, OnEnhanceSharp)
	ON_COMMAND(ID_ENHANCE_SMOOTH_FR, OnEnhanceSmoothFr)
	ON_COMMAND(IDC_ENHANCE_BUTT_LOW, OnEnhanceButtLow)
	ON_COMMAND(IDC_ENHANCE_SHARP_FREQ, OnEnhanceSharpFreq)
	ON_COMMAND(IDC_ENHANCE_BUTT_HIGHT, OnEnhanceButtHight)
	ON_COMMAND(ID_REGIONSEG_FIX, OnRegionsegFix)
	ON_COMMAND(ID_ADA_REGION_SEG, OnAdaRegionSeg)
	ON_COMMAND(ID_EDGE_ROBERTS, OnEdgeRoberts)
	ON_COMMAND(ID_EDGE_SOBEL, OnEdgeSobel)
	ON_COMMAND(ID_EDGE_PREWITT, OnEdgePrewitt)
	ON_COMMAND(ID_EDGE_LAPLACE, OnEdgeLaplace)
	ON_COMMAND(ID_EDGE_CANNY, OnEdgeCanny)
	ON_COMMAND(ID_EDGE_TRACK, OnEdgeTrack)
	ON_COMMAND(ID_REGION_GROW, OnRegionGrow)
	ON_COMMAND(ID_MOTION_BACKGROUND, OnMotionBackground)
	ON_COMMAND(ID_RECOG_MATCH, OnRecogMatch)
	ON_COMMAND(ID_CODING_SHANFINO, OnCodingShanfino)
	ON_COMMAND(ID_DEGENERATION_INVERSE, OnDegenerationInverse)
	ON_COMMAND(ID_DEGENERATION_MOTION, OnDegenerationMotion)
	ON_COMMAND(ID_DEGENERATION_Winner, OnDEGENERATIONWinner)
	ON_COMMAND(ID_RESTORE_INVERSE, OnRestoreInverse)
	ON_COMMAND(ID_RESTORE_MOTION, OnRestoreMotion)
	ON_COMMAND(ID_RESTORE_WINNER, OnRestoreWinner)
	ON_COMMAND(ID_STREET_FRAMEWORK, OnStreetFramework)
	ON_COMMAND(ID_STREET_TRANSFORM, OnStreetTransform)
	ON_COMMAND(ID_TRACE, OnTrace)
	ON_COMMAND(ID_VIEW_BAYER, OnViewBayer)
	ON_COMMAND(ID_VIEW_FloydSteinberg, OnVIEWFloydSteinberg)
	ON_COMMAND(ID_OUTLINE, OnOutline)
	ON_COMMAND(ID_FRAME_RESTORE, OnFrameRestore)
	ON_COMMAND(ID_MOMENT, OnMoment)
	ON_COMMAND(ID_BARYCENTERMOMENT, OnBarycentermoment)
	ON_COMMAND(ID_ANALYSIS_HOLENUM, OnAnalysisHolenum)
	ON_COMMAND(ID_FREQ_DCT, OnFreqDct)
	ON_COMMAND(ID_FREQ_HOTELLING, OnFreqHotelling)
	ON_COMMAND(ID_FREQ_WALSH, OnFreqWalsh)
	ON_COMMAND(ID_CODING_ARITH, OnCodingArith)
	ON_COMMAND(ID_CODING_BITPLANE, OnCodingBitplane)
	ON_COMMAND(ID_CODING_HUFFMAN, OnCodingHuffman)
	ON_COMMAND(ID_CODING_LOADIMG, OnCodingLoadimg)
	ON_COMMAND(ID_CODING_WRITEIMG, OnCodingWriteimg)
	//}}AFX_MSG_MAP
	// Standard printing commands
	ON_COMMAND(ID_FILE_PRINT, CScrollView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_DIRECT, CScrollView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_PREVIEW, CScrollView::OnFilePrintPreview)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CImageProcessingView construction/destruction

CImageProcessingView::CImageProcessingView()
{
	// 為小波變換設(shè)置的參數(shù)
	// 臨時(shí)存放小波變換系數(shù)內(nèi)存
	m_pDbImage = NULL;	
	
	// 設(shè)置當(dāng)前層數(shù)
	m_nDWTCurDepth = 0;

	// 設(shè)置小波基緊支集長(zhǎng)度
	m_nSupp = 1;
}

CImageProcessingView::~CImageProcessingView()
{
	// 釋放已分配內(nèi)存
	if(m_pDbImage){
		delete[]m_pDbImage;
		m_pDbImage = NULL;
	}
}

BOOL CImageProcessingView::PreCreateWindow(CREATESTRUCT& cs)
{
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return CScrollView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CImageProcessingView drawing

void CImageProcessingView::OnDraw(CDC* pDC)
{
	CImageProcessingDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	
	CSize sizeDibDisplay;		
	

	
	if(!pDoc->m_pDibInit->IsEmpty()){	
		sizeDibDisplay = pDoc->m_pDibInit->GetDimensions();
		pDoc->m_pDibInit->Draw(pDC,CPoint(0,0),sizeDibDisplay);	
	}	

}

void CImageProcessingView::OnInitialUpdate()
{
	CScrollView::OnInitialUpdate();

	CImageProcessingDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	
	CSize sizeTotal = pDoc->m_pDibInit->GetDimensions();
	SetScrollSizes(MM_TEXT, sizeTotal);

	GetParentFrame()->RecalcLayout();
	ResizeParentToFit();

}

/////////////////////////////////////////////////////////////////////////////
// CImageProcessingView printing

BOOL CImageProcessingView::OnPreparePrinting(CPrintInfo* pInfo)
{
	// default preparation
	return DoPreparePrinting(pInfo);
}

void CImageProcessingView::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add extra initialization before printing
}

void CImageProcessingView::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add cleanup after printing
}

/////////////////////////////////////////////////////////////////////////////
// CImageProcessingView diagnostics

#ifdef _DEBUG
void CImageProcessingView::AssertValid() const
{
	CScrollView::AssertValid();
}

void CImageProcessingView::Dump(CDumpContext& dc) const
{
	CScrollView::Dump(dc);
}

CImageProcessingDoc* CImageProcessingView::GetDocument() // non-debug version is inline
{
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CImageProcessingDoc)));
	return (CImageProcessingDoc*)m_pDocument;
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CImageProcessingView message handlers









/*************************************************************************
 *
 * \函數(shù)名稱：
 *   OnFft2d()
 *
 * \輸入?yún)?shù):
 *   無(wú)
 * 
 * \返回值:
 *   無(wú)
 *
 * \說(shuō)明:
 *   運(yùn)行二維快速傅立葉變換
 *
 *************************************************************************
 */
void CImageProcessingView::OnFft2d() 
{
	//圖象FFT變換

	// 更改光標(biāo)形狀
	BeginWaitCursor();
	
	// 循環(huán)控制變量
	int y;
	int x;
	
	// 獲得Doc類的指針
	CImageProcessingDoc * pDoc = (CImageProcessingDoc *)this->GetDocument();

	CDib * pDib = pDoc->m_pDibInit;

	// 獲得圖象的頭文件信息
	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;

	// 判斷是否是8-bpp位圖（這里為了方便，只處理8-bpp位圖的離散傅立葉變換）
	if (lpBMIH->biBitCount != 8)
	{
		// 提示用戶
		MessageBox("目前只支持256色位圖的離散傅立葉變換！", "系統(tǒng)提示" ,
			MB_ICONINFORMATION | MB_OK);
						
		// 返回
		return;
	}

	// 圖象的寬長(zhǎng)
	CSize sizeImage ;
	int nWidth ;
	int nHeight;

	// 獲得圖象的寬長(zhǎng)
	sizeImage = pDib->GetDimensions() ;
		
	nWidth = sizeImage.cx;
	nHeight= sizeImage.cy;

	// 臨時(shí)變量
	double	dTmpOne;
	double  dTmpTwo;
	
	// 傅立葉變換豎直方向點(diǎn)數(shù)
	int nTransHeight ;

	// 傅立葉變換水平方向點(diǎn)數(shù)
	int nTransWidth  ;	
	
	// 計(jì)算進(jìn)行傅立葉變換的點(diǎn)數(shù)	（2的整數(shù)次冪）
	dTmpOne = log(nWidth)/log(2);
	dTmpTwo = ceil(dTmpOne)		   ;
	dTmpTwo = pow(2,dTmpTwo)	   ;
	nTransWidth = (int) dTmpTwo	   ;
	
	// 計(jì)算進(jìn)行傅立葉變換的點(diǎn)數(shù) （2的整數(shù)次冪）
	dTmpOne = log(nHeight)/log(2);
	dTmpTwo = ceil(dTmpOne)		   ;
	dTmpTwo = pow(2,dTmpTwo)	   ;
	nTransHeight = (int) dTmpTwo   ;

	// 計(jì)算圖象數(shù)據(jù)存儲(chǔ)每行需要的字節(jié)數(shù)
	// BMP文件的每行數(shù)據(jù)存儲(chǔ)是DWORD對(duì)齊的
	int		nSaveWidth;
	nSaveWidth = ( (nWidth << 3) + 31)/32 * 4 ;

	// 指向圖象數(shù)據(jù)的指針
	LPBYTE lpImage ;
	lpImage = pDib->m_lpImage ;

	// 圖象象素值
	unsigned char unchValue;

	
	// 指向時(shí)域數(shù)據(jù)的指針
	complex<double> * pCTData ;

	// 指向頻域數(shù)據(jù)的指針
	complex<double> * pCFData ;

	// 分配內(nèi)存
	pCTData=new complex<double>[nTransWidth * nTransHeight];
	pCFData=new complex<double>[nTransWidth * nTransHeight];

	// 初始化
	// 圖象數(shù)據(jù)的寬和高不一定是2的整數(shù)次冪，所以pCTData
	// 有一部分?jǐn)?shù)據(jù)需要補(bǔ)0
	for(y=0; y<nTransHeight; y++)
	{
		for(x=0; x<nTransWidth; x++)
		{
			pCTData[y*nTransWidth + x]=complex<double>(0,0);
		}
	}

	// 把圖象數(shù)據(jù)傳給pCTData
	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			unchValue = lpImage[y*nSaveWidth +x];
			pCTData[y*nTransWidth + x]=complex<double>(unchValue,0);
		}
	}

	// 傅立葉正變換
	DIBFFT_2D(pCTData, nWidth, nHeight, pCFData) ;
	
	// 臨時(shí)變量
	double dTmp;

	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			dTmp = pCFData[y * nTransWidth + x].real() 
				   * pCFData[y * nTransWidth + x].real()
				 + pCFData[y * nTransWidth + x].imag() 
				   * pCFData[y * nTransWidth + x].imag();
			
			dTmp = sqrt(dTmp) ;

			// 為了顯示，需要對(duì)幅度的大小進(jìn)行伸縮
			dTmp /= 100        ;

			// 限制圖象數(shù)據(jù)的大小
			dTmp = min(dTmp, 255) ;

			lpImage[y*nSaveWidth +x] = (unsigned char)(int)dTmp;
		}
	}

	// 為了在屏幕上顯示，我們把幅度值大的部分用黑色顯示
	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			lpImage[y*nSaveWidth +x] = 255 - lpImage[y*nSaveWidth +x];
		}
	}
	
	// 刷新屏幕
	Invalidate();

	// 釋放內(nèi)存
	delete pCTData;
	delete pCFData;
	pCTData = NULL;
	pCFData = NULL;

	// 設(shè)置臟標(biāo)記
	pDoc->SetModifiedFlag(TRUE);
		
	// 更新視圖
	pDoc->UpdateAllViews(NULL);

	//  恢復(fù)光標(biāo)形狀
	EndWaitCursor();
}


/*************************************************************************
 *
 * \函數(shù)名稱：
 *   OnDft2d()
 *
 * \輸入?yún)?shù):
 *   無(wú)
 * 
 * \返回值:
 *   無(wú)
 *
 * \說(shuō)明:
 *   運(yùn)行二維傅立葉變換
 *
 *************************************************************************
 */
void CImageProcessingView::OnDft2d() 
{
	//圖象離散傅立葉變換
	
	//提示用戶，直接進(jìn)行離散傅立葉變換的時(shí)間很長(zhǎng)
	MessageBox("沒有使用FFT，時(shí)間可能很長(zhǎng)！", "作者提示" ,
			MB_ICONINFORMATION | MB_OK);
	
	//更改光標(biāo)形狀
	BeginWaitCursor(); 

	// 循環(huán)控制變量
	int y;
	int x;
	
	CImageProcessingDoc * pDoc = (CImageProcessingDoc *)this->GetDocument();
	CDib * pDib = pDoc->m_pDibInit;

	// 獲得圖象的頭文件信息
	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;

	// 判斷是否是8-bpp位圖（這里為了方便，只處理8-bpp位圖的離散傅立葉變換）
	if (lpBMIH->biBitCount != 8)
	{
		// 提示用戶
		MessageBox("目前只支持256色位圖的離散傅立葉變換！", "系統(tǒng)提示" ,
			MB_ICONINFORMATION | MB_OK);
						
		// 返回
		return;
	}

	//圖象的長(zhǎng)寬大小
	CSize sizeImage		= pDib->GetDimensions();
	int nWidth			= sizeImage.cx		;
	int nHeight			= sizeImage.cy		;

	// 計(jì)算圖象數(shù)據(jù)存儲(chǔ)每行需要的字節(jié)數(shù)
	// BMP文件的每行數(shù)據(jù)存儲(chǔ)是DWORD對(duì)齊的
	int		nSaveWidth;
	nSaveWidth = ( (nWidth << 3) + 31)/32 * 4 ;

	// 指向圖象數(shù)據(jù)的指針
	LPBYTE lpImage ;
	lpImage = pDib->m_lpImage ;

	double * pTrRstRpart  = new double [nWidth*nHeight];
	double * pTrRstIpart  = new double [nWidth*nHeight];	
	
	::DIBDFT_2D(pDib, pTrRstRpart,pTrRstIpart);

	// 臨時(shí)變量
	double dTmp;

	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			dTmp = pTrRstRpart[y*nWidth + x] * pTrRstRpart[y*nWidth + x]
				 + pTrRstIpart[y*nWidth + x] * pTrRstIpart[y*nWidth + x];
			
			dTmp = sqrt(dTmp) ;

			// 為了顯示，需要對(duì)幅度的大小進(jìn)行伸縮
			dTmp /= 100        ;

			// 限制圖象數(shù)據(jù)的大小
			dTmp = min(dTmp, 255) ;

			lpImage[y*nSaveWidth +x] = (unsigned char)(int)dTmp;
		}
	}

	// 為了在屏幕上顯示，我們把幅度值大的部分用黑色顯示
	for(y=0; y<nHeight; y++)
	{
		for(x=0; x<nWidth; x++)
		{
			lpImage[y*nSaveWidth +x] = 255 - lpImage[y*nSaveWidth +x];
		}
	}

	// 釋放內(nèi)存
	delete pTrRstRpart;
	pTrRstRpart=NULL  ;
	
	delete pTrRstIpart;
	pTrRstIpart=NULL  ;

	// 設(shè)置臟標(biāo)記
	pDoc->SetModifiedFlag(TRUE);
		
	// 更新視圖
	pDoc->UpdateAllViews(NULL);

	// 恢復(fù)光標(biāo)形狀
	EndWaitCursor(); 
	
	// 刷新屏幕
	Invalidate();
}

void CImageProcessingView::OnFreqDct() 
{
	// 圖象的離散余弦變換
		
	// 更改光標(biāo)形狀
	BeginWaitCursor();

	// 獲取文檔
	CImageProcessingDoc* pDoc = GetDocument();

	//  獲得圖象CDib類的指針
	CDib * pDib = pDoc->m_pDibInit;

	// 獲得圖象的頭文件信息
	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;

	// 判斷是否是8-bpp位圖（這里為了方便，只處理8-bpp位圖的離散余弦變換）
	if (lpBMIH->biBitCount != 8)
	{
		// 提示用戶
		MessageBox("目前只支持256色位圖的離散余弦變換！", "系統(tǒng)提示" ,
			MB_ICONINFORMATION | MB_OK);
						
		// 返回
		return;
	}
	
	::DIBDct(pDib);
	
	// 設(shè)置臟標(biāo)記
	pDoc->SetModifiedFlag(TRUE);
		
	// 更新視圖
	pDoc->UpdateAllViews(NULL);

    // 恢復(fù)光標(biāo)
	EndWaitCursor();
	
}

void CImageProcessingView::OnFreqHotelling() 
{
	// 圖象霍特林變換
	
	// 更改光標(biāo)形狀
	BeginWaitCursor();

	// 獲取文檔
	CImageProcessingDoc* pDoc = GetDocument();

	//  獲得圖象CDib類的指針
	CDib * pDib = pDoc->m_pDibInit;

	// 獲得圖象的頭文件信息
	LPBITMAPINFOHEADER lpBMIH=pDib->m_lpBMIH;

	// 判斷是否是8-bpp位圖（這里為了方便，只處理8-bpp位圖的離散霍特林變換）
	if (lpBMIH->biBitCount != 8)
	{
		// 提示用戶
		MessageBox("目前只支持256色位圖的離散霍特林變換！", "系統(tǒng)提示" ,
			MB_ICONINFORMATION | MB_OK);
						
		// 返回
		return;
	}

	//  圖象的霍特林變換
	DIBHOTELLING(pDib);
	
	// 設(shè)置臟標(biāo)記
	pDoc->SetModifiedFlag(TRUE);
		
	// 更新視圖
	pDoc->UpdateAllViews(NULL);

    // 恢復(fù)光標(biāo)
	EndWaitCursor();
	
}

void CImageProcessingView::OnFreqWalsh() 
{