// 小波變換Doc.cpp : implementation of the CMyDoc class
//

#include "stdafx.h"
#include "小波變換.h"

#include "小波變換Doc.h"

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

/////////////////////////////////////////////////////////////////////////////
// CMyDoc

IMPLEMENT_DYNCREATE(CMyDoc, CDocument)

BEGIN_MESSAGE_MAP(CMyDoc, CDocument)
	//{{AFX_MSG_MAP(CMyDoc)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMyDoc construction/destruction

CMyDoc::CMyDoc()
{
	// TODO: add one-time construction code here
	m_hDIB = NULL;
	m_palDIB = NULL;
	m_sizeDoc = CSize(1,1);
}

CMyDoc::~CMyDoc()
{
	// 判斷DIB對象是否存在
	if (m_hDIB != NULL)
	{
		// 清除DIB對象
		::GlobalFree((HGLOBAL) m_hDIB);
	}
	
	// 判斷調(diào)色板是否存在
	if (m_palDIB != NULL)
	{
		// 清除調(diào)色板
		delete m_palDIB;
	}
}

BOOL CMyDoc::OnNewDocument()
{
	if (!CDocument::OnNewDocument())
		return FALSE;

	// TODO: add reinitialization code here
	// (SDI documents will reuse this document)

	return TRUE;
}



/////////////////////////////////////////////////////////////////////////////
// CMyDoc serialization

void CMyDoc::Serialize(CArchive& ar)
{
	if (ar.IsStoring())
	{
		// TODO: add storing code here
	}
	else
	{
		// TODO: add loading code here
	}
}

/////////////////////////////////////////////////////////////////////////////
// CMyDoc diagnostics

#ifdef _DEBUG
void CMyDoc::AssertValid() const
{
	CDocument::AssertValid();
}

void CMyDoc::Dump(CDumpContext& dc) const
{
	CDocument::Dump(dc);
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMyDoc commands

BOOL CMyDoc::OnOpenDocument(LPCTSTR lpszPathName) 
{
	CFile file;
	CFileException fe;
	// 打開文件
	if (!file.Open(lpszPathName, CFile::modeRead | CFile::shareDenyWrite, &fe))
	{
		// 失敗
		ReportSaveLoadException(lpszPathName, &fe,FALSE, AFX_IDP_FAILED_TO_OPEN_DOC);		
		// 返回FALSE
		return FALSE;
	}	
	DeleteContents();	
	// 更改光標(biāo)形狀
	BeginWaitCursor();	
	// 嘗試調(diào)用ReadDIBFile()讀取圖像
	TRY
	{
		m_hDIB = ::ReadDIBFile(file);
	}
	CATCH (CFileException, eLoad)
	{
		// 讀取失敗
		file.Abort();		
		// 恢復(fù)光標(biāo)形狀
		EndWaitCursor();		
		// 報(bào)告失敗
		ReportSaveLoadException(lpszPathName, eLoad,FALSE, AFX_IDP_FAILED_TO_OPEN_DOC);		
		// 設(shè)置DIB為空
		m_hDIB = NULL;		
		// 返回FALSE
		return FALSE;
	}
	END_CATCH	
	// 初始化DIB
	InitDIBData();	
	// 恢復(fù)光標(biāo)形狀
	EndWaitCursor();	
	// 判斷讀取文件是否成功
	if (m_hDIB == NULL)
	{
		// 失敗，可能非BMP格式
		CString strMsg;
		strMsg = "讀取圖像時(shí)出錯！可能是不支持該類型的圖像文件！";		
		// 提示出錯
		MessageBox(NULL, strMsg, "系統(tǒng)提示", MB_ICONINFORMATION | MB_OK);		
		// 返回FALSE
		return FALSE;
	}	
	// 設(shè)置文件名稱
	SetPathName(lpszPathName);	
	// 初始化脹標(biāo)記為FALSE
	SetModifiedFlag(FALSE);	
	// 返回TRUE
	return TRUE;
}

void CMyDoc::InitDIBData()
{
	// 初始化DIB對象
	// 判斷調(diào)色板是否為空
	if (m_palDIB != NULL)
	{
		// 刪除調(diào)色板對象
		delete m_palDIB;
		// 重置調(diào)色板為空
		m_palDIB = NULL;
	}
	// 如果DIB對象為空，直接返回
	if (m_hDIB == NULL)
	{
		// 返回
		return;
	}
	LPSTR lpDIB = (LPSTR) ::GlobalLock((HGLOBAL) m_hDIB);
	// 判斷圖像是否過大
	if (::DIBWidth(lpDIB) > INT_MAX ||::DIBHeight(lpDIB) > INT_MAX)
	{
		::GlobalUnlock((HGLOBAL) m_hDIB);
		// 釋放DIB對象
		::GlobalFree((HGLOBAL) m_hDIB);
		// 設(shè)置DIB為空
		m_hDIB = NULL;
		CString strMsg;
		strMsg = "BMP圖像太大！";
		// 提示用戶
		MessageBox(NULL, strMsg, "系統(tǒng)提示", MB_ICONINFORMATION | MB_OK);
		// 返回
		return;
	}
	// 設(shè)置文檔大小
	m_sizeDoc = CSize((int) ::DIBWidth(lpDIB), (int) ::DIBHeight(lpDIB));
	::GlobalUnlock((HGLOBAL) m_hDIB);
	// 創(chuàng)建新調(diào)色板
	m_palDIB = new CPalette;
	// 判斷是否創(chuàng)建成功
	if (m_palDIB == NULL)
	{
		// 失敗，可能是內(nèi)存不足
		::GlobalFree((HGLOBAL) m_hDIB);
		// 設(shè)置DIB對象為空
		m_hDIB = NULL;
		// 返回
		return;
	}
	// 調(diào)用CreateDIBPalette來創(chuàng)建調(diào)色板
	if (::CreateDIBPalette(m_hDIB, m_palDIB) == NULL)
	{
		// 返回空，可能該DIB對象沒有調(diào)色板
		// 刪除
		delete m_palDIB;
		// 設(shè)置為空
		m_palDIB = NULL;
		// 返回
		return;
	}

}
BOOL CMyDoc::OnSaveDocument(LPCTSTR lpszPathName) 
{
	CFile file;
	CFileException fe;
	// 打開文件
	if (!file.Open(lpszPathName, CFile::modeCreate |
	  CFile::modeReadWrite | CFile::shareExclusive, &fe))
	{
		// 失敗
		ReportSaveLoadException(lpszPathName, &fe,TRUE, AFX_IDP_INVALID_FILENAME);		
		// 返回FALSE
		return FALSE;
	}
	// 嘗試調(diào)用SaveDIB保存圖像
	BOOL bSuccess = FALSE;
	TRY
	{		
		// 更改光標(biāo)形狀
		BeginWaitCursor();
		// 嘗試保存圖像
		bSuccess = ::SaveDIB(m_hDIB, file);
		// 關(guān)閉文件
		file.Close();
	}
	CATCH (CException, eSave)
	{
		// 失敗
		file.Abort();		
		// 恢復(fù)光標(biāo)形狀
		EndWaitCursor();
		ReportSaveLoadException(lpszPathName, eSave,TRUE, AFX_IDP_FAILED_TO_SAVE_DOC);		
		// 返回FALSE
		return FALSE;
	}
	END_CATCH
	// 恢復(fù)光標(biāo)形狀
	EndWaitCursor();	
	// 重置脹標(biāo)記為FALSE
	SetModifiedFlag(FALSE);
	if (!bSuccess)
	{
		// 保存失敗，可能是其它格式的DIB，可以讀取但是不能保存
		// 或者是SaveDIB函數(shù)有誤
		CString strMsg;
		strMsg = "無法保存BMP圖像！";		
		// 提示出錯
		MessageBox(NULL, strMsg, "系統(tǒng)提示", MB_ICONINFORMATION | MB_OK);
	}	
	return bSuccess;
}