www.pudn.com > WhiteBalance.rar > WhiteBalanceDoc.cpp

// WhiteBalanceDoc.cpp : implementation of the CWhiteBalanceDoc class 
// 
 
#include "stdafx.h" 
#include "WhiteBalance.h" 
 
#include "WhiteBalanceDoc.h" 
 
#ifdef _DEBUG 
#define new DEBUG_NEW 
#undef THIS_FILE 
static char THIS_FILE[] = __FILE__; 
#endif 
 
///////////////////////////////////////////////////////////////////////////// 
// CWhiteBalanceDoc 
 
IMPLEMENT_DYNCREATE(CWhiteBalanceDoc, CDocument) 
 
BEGIN_MESSAGE_MAP(CWhiteBalanceDoc, CDocument) 
	//{{AFX_MSG_MAP(CWhiteBalanceDoc) 
	ON_COMMAND(ID_LOAD_IMAGE, OnLoadImage) 
	//}}AFX_MSG_MAP 
END_MESSAGE_MAP() 
 
///////////////////////////////////////////////////////////////////////////// 
// CWhiteBalanceDoc construction/destruction 
 
CWhiteBalanceDoc::CWhiteBalanceDoc() 
{ 
	// TODO: add one-time construction code here 
    m_refColorBKG=0x00808080; 
    m_hDIB=NULL; 
    m_palDIB=NULL; 
	m_WhCh=0; 
    m_sizeDoc=CSize(1,1); 
	m_ChooseFI=0; 
	m_Color=RGB(255,255,255); 
	m_ChangeFlag=0; 
} 
 
CWhiteBalanceDoc::~CWhiteBalanceDoc() 
{ 
		if(m_hDIB!=NULL) 
 
		::GlobalFree((HGLOBAL)m_hDIB); 
	 
} 
 
BOOL CWhiteBalanceDoc::OnNewDocument() 
{ 
	if (!CDocument::OnNewDocument()) 
		return FALSE; 
 
	// TODO: add reinitialization code here 
	// (SDI documents will reuse this document) 
 
	return TRUE; 
} 
 
 
 
///////////////////////////////////////////////////////////////////////////// 
// CWhiteBalanceDoc serialization 
 
void CWhiteBalanceDoc::Serialize(CArchive& ar) 
{ 
	if (ar.IsStoring()) 
	{ 
		// TODO: add storing code here 
	} 
	else 
	{ 
		// TODO: add loading code here 
	} 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CWhiteBalanceDoc diagnostics 
 
#ifdef _DEBUG 
void CWhiteBalanceDoc::AssertValid() const 
{ 
	CDocument::AssertValid(); 
} 
 
void CWhiteBalanceDoc::Dump(CDumpContext& dc) const 
{ 
	CDocument::Dump(dc); 
} 
#endif //_DEBUG 
 
HDIB CWhiteBalanceDoc::DibReturn(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(); 
	 
	// 更改光标形状 
	BeginWaitCursor(); 
	 
	// 尝试调用ReadDIBFile()读取图像 
	TRY 
	{ 
		m_hDIB = ::ReadDIBFile(file); 
	} 
	CATCH (CFileException, eLoad) 
	{ 
		// 读取失败 
		file.Abort(); 
		 
		// 恢复光标形状 
		EndWaitCursor(); 
		 
		// 报告失败 
		ReportSaveLoadException(lpszPathName, eLoad, 
			FALSE, AFX_IDP_FAILED_TO_OPEN_DOC); 
		 
		// 设置DIB为空 
		m_hDIB = NULL; 
		 
		// 返回FALSE 
		return FALSE; 
	} 
	END_CATCH 
	 
 
	 
	// 恢复光标形状 
	EndWaitCursor(); 
	 
	// 判断读取文件是否成功 
	if (m_hDIB == NULL) 
	{ 
		// 失败，可能非BMP格式 
		CString strMsg; 
		strMsg = "读取图像时出错！可能是不支持该类型的图像文件！"; 
		 
		// 提示出错 
		MessageBox(NULL, strMsg, "系统提示", MB_ICONINFORMATION | MB_OK); 
		 
		// 返回FALSE 
		return FALSE; 
	} 
	 
	// 设置文件名称 
	SetPathName(lpszPathName); 
	 
	// 初始化胀标记为FALSE 
	SetModifiedFlag(FALSE); 
	 
	// 返回 
	return m_hDIB; 
} 
 
BOOL CWhiteBalanceDoc::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; 
	} 
 
	// 尝试调用SaveDIB保存图像 
	BOOL bSuccess = FALSE; 
	TRY 
	{ 
		 
		// 更改光标形状 
		BeginWaitCursor(); 
 
		// 尝试保存图像 
		bSuccess = ::SaveDIB(m_hDIB, file); 
 
		// 关闭文件 
		file.Close(); 
	} 
	CATCH (CException, eSave) 
	{ 
		// 失败 
		file.Abort(); 
		 
		// 恢复光标形状 
		EndWaitCursor(); 
		ReportSaveLoadException(lpszPathName, eSave, 
			TRUE, AFX_IDP_FAILED_TO_SAVE_DOC); 
		 
		// 返回FALSE 
		return FALSE; 
	} 
	END_CATCH 
 
	// 恢复光标形状 
	EndWaitCursor(); 
	 
	// 重置胀标记为FALSE 
	SetModifiedFlag(FALSE); 
 
	if (!bSuccess) 
	{ 
		// 保存失败，可能是其它格式的DIB，可以读取但是不能保存 
		// 或者是SaveDIB函数有误 
 
		CString strMsg; 
		strMsg = "无法保存BMP图像！"; 
		 
		// 提示出错 
		MessageBox(NULL, strMsg, "系统提示", MB_ICONINFORMATION | MB_OK); 
	} 
	 
	return bSuccess; 
} 
 
void CWhiteBalanceDoc::ReplaceHDIB(HDIB hDIB) 
{ 
	// 替换DIB，在功能粘贴中用到该函数 
	 
	// 判断DIB是否为空 
	if (m_hDIB != NULL) 
	{ 
		// 非空，则清除 
		::GlobalFree((HGLOBAL) m_hDIB); 
	} 
 
	// 替换成新的DIB对象 
	m_hDIB = hDIB; 
	 
} 
 
void CWhiteBalanceDoc::InitDIBData() 
{ 
	// 初始化DIB对象 
 
	// 判断调色板是否为空 
	if (m_palDIB != NULL) 
	{ 
		// 删除调色板对象 
		delete m_palDIB; 
 
		// 重置调色板为空 
		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); 
		 
		// 设置DIB为空 
		m_hDIB = NULL; 
		 
		CString strMsg; 
		strMsg = "BMP图像太大！"; 
		 
		// 提示用户 
		MessageBox(NULL, strMsg, "系统提示", MB_ICONINFORMATION | MB_OK); 
		 
		// 返回 
		return; 
	} 
	 
	// 设置文档大小 
	m_sizeDoc = CSize((int) ::DIBWidth(lpDIB), (int) ::DIBHeight(lpDIB)); 
	 
	::GlobalUnlock((HGLOBAL) m_hDIB); 
	 
	// 创建新调色板 
	m_palDIB = new CPalette; 
	 
	// 判断是否创建成功 
	if (m_palDIB == NULL) 
	{ 
		// 失败，可能是内存不足 
		::GlobalFree((HGLOBAL) m_hDIB); 
		 
		// 设置DIB对象为空 
		m_hDIB = NULL; 
		 
		// 返回 
		return; 
	} 
	 
	// 调用CreateDIBPalette来创建调色板 
	if (::CreateDIBPalette(m_hDIB, m_palDIB) == NULL) 
	{ 
		// 返回空，可能该DIB对象没有调色板 
		 
		// 删除 
		delete m_palDIB; 
		 
		// 设置为空 
		m_palDIB = NULL; 
		 
		// 返回 
		return; 
	} 
} 
 
///////////////////////////////////////////////////////////////////////////// 
// CWhiteBalanceDoc commands 
 
void CWhiteBalanceDoc::OnLoadImage()  
{ 
	// TODO: Add your command handler code here 
		// TODO: Add your command handler code here 
	CString strOpenFile="BMP图像文件|*.bmp|All Files(*.*)|*.*|"; 
	CFileDialog dlg(TRUE,"*.bmp",NULL,OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,strOpenFile); 
	if(dlg.DoModal()==IDOK) 
	{ 
 
		CString strName = dlg.GetPathName(); 
        m_hDIB=DibReturn(strName); 
        m_WhCh=1; 
		UpdateAllViews(NULL); 
	} 
}