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&amt; 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&amt; dc) const
{
CDocument::Dump(dc);
}
#endif //_DEBUG

HDIB CWhiteBalanceDoc::DibReturn(LPCTSTR lpszPathName)
{

CFile file;
CFileException fe;

// 打开文件
if (!file.Open(lpszPathName, CFile::modeRead | CFile::shareDenyWrite, &amt;fe))
{
// 失败
ReportSaveLoadException(lpszPathName, &amt;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, &amt;fe))
{
// 失败
ReportSaveLoadException(lpszPathName, &amt;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);
}
}