// Wavelet2DDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Wavelet2D.h"
#include "Wavelet2DDlg.h"
#include "ColorTable.h"
#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()

/////////////////////////////////////////////////////////////////////////////
// CWavelet2DDlg dialog

CWavelet2DDlg::CWavelet2DDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CWavelet2DDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CWavelet2DDlg)
		// NOTE: the ClassWizard will add member initialization here
	m_bWavelet = FALSE;
	m_strBmpFile=_T("Sample.bmp");
	m_nDecomposeStage = 1;
	m_nFilterLen      = 6;
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_strFilterFile[0]="h1.dat";
	m_strFilterFile[1]="h2.dat";	
	m_strFilterFile[2]="h3.dat";
	m_strFilterFile[3]="h4.dat";
	m_strFilterFile[4]="h5.dat";
	m_strFilterFile[5]="h6.dat";
	m_strFilterFile[6]="h7.dat";
	m_strFilterFile[7]="h8.dat";
	m_strFilterFile[8]="h9.dat";
	m_strFilterFile[9]="h10.dat";

	m_pBmpInfo = (BITMAPINFO *)malloc(2048);
	if(m_pBmpInfo)
	{
		m_pBmpInfo->bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
		m_pBmpInfo->bmiHeader.biWidth = m_nImageWidth;
		m_pBmpInfo->bmiHeader.biHeight = m_nImageHeight;
		m_pBmpInfo->bmiHeader.biPlanes = 1;
		m_pBmpInfo->bmiHeader.biBitCount = 8;
		m_pBmpInfo->bmiHeader.biCompression = BI_RGB;
		m_pBmpInfo->bmiHeader.biSizeImage = 0;
		m_pBmpInfo->bmiHeader.biXPelsPerMeter = 0;
		m_pBmpInfo->bmiHeader.biYPelsPerMeter = 0;
		m_pBmpInfo->bmiHeader.biClrUsed = 0;
		m_pBmpInfo->bmiHeader.biClrImportant = 0;
		for( int k = 0; k < 256; ++k){
			m_pBmpInfo->bmiColors[k].rgbBlue = (BYTE)k;
			m_pBmpInfo->bmiColors[k].rgbGreen = (BYTE)k;
			m_pBmpInfo->bmiColors[k].rgbRed = (BYTE)k;
			m_pBmpInfo->bmiColors[k].rgbReserved	= 0;
		}
	}

	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CWavelet2DDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CWavelet2DDlg)
	DDX_Control(pDX, IDC_EDIT_BMPFILE, m_editBmpFile);
	DDX_Control(pDX, IDC_COMBO_FILTERLEN, m_combFilterLen);
	DDX_Control(pDX, IDC_COMBO_DECOMPOSESTAGE, m_combDecomposeStage);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CWavelet2DDlg, CDialog)
	//{{AFX_MSG_MAP(CWavelet2DDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON_WAVELET2D, OnBtnWavelet2D)
	ON_BN_CLICKED(IDC_BUTTON_BMPFILE, OnBtnBmpFile)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWavelet2DDlg message handlers

BOOL CWavelet2DDlg::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
	
	// 初始化控件
	m_editBmpFile.SetWindowText(m_strBmpFile);
	CString strComb;
	int k=0;
	for(k=0;k<6;k++)
	{ 
	    strComb.Format("%d",k+1);
	    m_combDecomposeStage.InsertString(k,strComb);
        m_combDecomposeStage.SetItemData(k,k+1);
        m_combDecomposeStage.SetCurSel(0);
	}
	for(k=0;k<10;k++)
	{ 
	    strComb.Format("%d",2*(k+1));
	    m_combFilterLen.InsertString(k,strComb);
        m_combFilterLen.SetItemData(k,2*(k+1));
        m_combFilterLen.SetCurSel(2);
	}
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CWavelet2DDlg::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 CWavelet2DDlg::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
	{
		if(m_bWavelet)
		   OnBtnWavelet2D();
		CDialog::OnPaint();
	}
}

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

// 選擇原始數據圖像文件
void CWavelet2DDlg::OnBtnBmpFile() 
{
	CFileDialog BmpFileDlg(TRUE,"bmp","*.bmp");
	if(BmpFileDlg.DoModal()==IDOK)
	{
	   m_bWavelet=FALSE;
		m_strBmpFile=BmpFileDlg.GetPathName();
	   m_editBmpFile.SetWindowText(m_strBmpFile);
	}	
}

// “二維小波變換”按鈕消息處理函數
// 功能：打開文件，分配內存，進行變換，繪制結果
void CWavelet2DDlg::OnBtnWavelet2D() 
{
	CFile BmpFile;
	CFileException fe;
	// 打開原始圖像數據
	if(m_strBmpFile.IsEmpty())
	{	
		AfxMessageBox (_T("請選擇一個位圖（*.bmp）文件!"),MB_ICONINFORMATION|MB_OK,NULL);
	    return;
	}
	if(!BmpFile.Open(m_strBmpFile,CFile::modeRead,&fe))
	{	
		AfxMessageBox (_T("打開文件失敗!"),MB_ICONINFORMATION|MB_OK,NULL);
	    return;
	}
	// 讀取原始數據圖像文件
	TRY
	{
		DIB = ::ReadDIBFile(BmpFile);
	}
	CATCH (CFileException, eLoad)
	{		
		BmpFile.Abort();	// 讀取失敗			
		AfxMessageBox (_T("讀取數據失敗!"),MB_ICONINFORMATION|MB_OK,NULL);		
		DIB = NULL;         // 設置DIB為空
		return ;
	}
	END_CATCH	
    
	// 鎖定DIB
	lpDIB = (LPSTR)::GlobalLock((HGLOBAL)GetHDIB());
	// 得到圖像寬度和高度
	m_nImageWidth=::DIBWidth(lpDIB);
	m_nImageHeight=::DIBHeight(lpDIB);

	// 打開待處理圖像
	// 判斷是否是8-bpp位圖
	if (::DIBNumColors(lpDIB) > 256||m_nImageWidth>400||m_nImageHeight>400)
	{
		// 提示用戶
		AfxMessageBox(_T("不是256色位圖或者圖像過大!") , MB_ICONINFORMATION | MB_OK,NULL);
		// 解除鎖定
		::GlobalUnlock((HGLOBAL) GetHDIB());
		return;
	}

	// 更改光標形狀
	BeginWaitCursor();
	// 找到DIB圖像象素起始位置
	lpDIBBits = ::FindDIBBits(lpDIB);
	// 得到小波變換所需的層次、濾波器長度等參數
	GetParam();	
	// 獲取濾波器參數
	if(!GetHn())
		return;
    // 繪制原始圖像
    DrawOriginalBmp();
    // 給image數組分配內存
	if(!GetMemory())	
		return;
	// 將圖像數據讀入image數組
	for(int i=0;i<m_nImageHeight;i++)
		for(int j=0;j<m_nImageWidth;j++)
		    image[i][j]=(unsigned char)(*((unsigned char *)lpDIBBits+m_nImageWidth*i+j));
     
	// 對圖像進行二維小波變換
	Wavelet2D();
	// 繪制二維小波變換后的圖像
	DrawWavelet2D();
	// 釋放資源	
	ReleaseMemory();
 	// 解除鎖定
	::GlobalUnlock((HGLOBAL) GetHDIB());
	// 恢復光標
	EndWaitCursor();

    // 設置已進行過小波變換標志
	m_bWavelet=TRUE;
}

// 確定退出函數
void CWavelet2DDlg::OnOK() 
{
	if(m_pBmpInfo)
		free(m_pBmpInfo);	

	CDialog::OnOK();
}

// 分配小波變換所需內存
BOOL CWavelet2DDlg::GetMemory()
{
    BOOL flag=TRUE;
	
    image=new double *[m_nImageHeight];
	for(int n=0;n<m_nImageHeight;n++)
	{
		*(image+n)=new double[sizeof(double)*m_nImageWidth];
	    if(*(image+n)==NULL)
		   flag=FALSE;
	}
	tempRow=new double[sizeof(double)*m_nImageWidth];
	tempCol=new double[sizeof(double)*m_nImageHeight];
	if(tempRow==NULL||tempCol==NULL)
		flag=FALSE;
    return flag;
}

// 釋放資源
void CWavelet2DDlg::ReleaseMemory()
{
	for(int k=0;k<m_nImageHeight;k++)
	  delete *(image+k);
    delete []image; 
	delete tempCol;
	delete tempRow;
	image=NULL;
	tempCol=NULL;
	tempRow=NULL;
	if(DIB)
	{ 
	   ::GlobalFree((HGLOBAL) DIB);
	   DIB=NULL;
	}
}

// 得到濾波器參數
BOOL CWavelet2DDlg::GetHn()
{

     int i;
	 double temp1;
	 FILE *fp1;
	 if((fp1=fopen(m_strFilterFile[m_nFilterLen/2-1],"rb+"))==NULL)
	 {
	    AfxMessageBox("打開Hn數據文件失敗!",MB_OK|MB_ICONINFORMATION);
		return FALSE;
	 }
	 for(i=0;i<30;i++)
		 h[i]=0;
	 for(i=0;i<m_nFilterLen;i++)
	 {
	    fread(&temp1,sizeof(double),1,fp1);
		h[i]=temp1;
	 }
	 fclose(fp1);
     return TRUE;
}

//返回H算子周期化后得值
double CWavelet2DDlg::HH(int i,int m)
{
     return h[i+m];

}

//返回G算子周期化后的值
double CWavelet2DDlg::GG(int i, int m)
{
     char flag;
	 if(i%2)
		 flag=-1;