diffuseview.cpp

visual c++数字图像与图形处理中的光盘内容

// DiffuseView.cpp : implementation of the CDiffuseView class
//

#include "stdafx.h"
#include "Diffuse.h"

#include "DiffuseDoc.h"
#include "DiffuseView.h"

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

#include "DiffuseFilter.h"

/////////////////////////////////////////////////////////////////////////////
// CDiffuseView

IMPLEMENT_DYNCREATE(CDiffuseView,  CView)

BEGIN_MESSAGE_MAP(CDiffuseView,  CView)
	//{{AFX_MSG_MAP(CDiffuseView)
	ON_COMMAND(IDM_DIFFUSE_FILTER,  OnDiffuseFilter)
	ON_UPDATE_COMMAND_UI(IDM_DIFFUSE_FILTER,  OnUpdateDiffuseFilter)
	ON_COMMAND(IDM_IMAGE_RESTORE,  OnImageRestore)
	ON_UPDATE_COMMAND_UI(IDM_IMAGE_RESTORE,  OnUpdateImageRestore)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CDiffuseView construction/destruction

CDiffuseView::CDiffuseView()
{
	// TODO: add construction code here
	m_nWidth = 800;
	m_nHeight = 600;
	m_dwOperation = 0L;
}

CDiffuseView::~CDiffuseView()
{
}

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

	return CView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CDiffuseView drawing

void CDiffuseView::OnDraw(CDC* pDC)
{
	CDiffuseDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	// TODO: add draw code for native data here
	CDib* pDib = pDoc->m_pDib;
	if(pDib)
	{
		BeginWaitCursor();
		m_nWidth = (int)pDib->GetWidth();
		m_nHeight = (int)pDib->GetHeight(); 

		CDC memDC;
		memDC.CreateCompatibleDC(pDC);
		CBitmap ddb;
		ddb.CreateCompatibleBitmap(pDC, m_nWidth, m_nHeight);
		CBitmap* pOldBitmap = memDC.SelectObject(&ddb);

		pDib->Draw(memDC.m_hDC, 0, 0, m_nWidth, m_nHeight, 
					0, 0, m_nWidth, m_nHeight, DIB_RGB_COLORS, SRCCOPY);
		
		if(m_dwOperation == 0L)
			pDC->BitBlt(0, 0, m_nWidth, m_nHeight, &memDC, 0, 0, SRCCOPY);
		else
			Filter(pDC, pDib, 0, 0, m_nWidth, m_nHeight);
		
		memDC.SelectObject(pOldBitmap);
		ddb.DeleteObject();
		EndWaitCursor();
	}

}

/////////////////////////////////////////////////////////////////////////////
// CDiffuseView diagnostics

#ifdef _DEBUG
void CDiffuseView::AssertValid() const
{
	CView::AssertValid();
}

void CDiffuseView::Dump(CDumpContext& dc) const
{
	CView::Dump(dc);
}

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

/////////////////////////////////////////////////////////////////////////////
// CDiffuseView message handlers

void CDiffuseView::OnDiffuseFilter() 
{
	// TODO: Add your command handler code here
	m_dwOperation = IMAGE_DIFFUSE_PLOT_FILTER;
	Invalidate();	
}

void CDiffuseView::OnUpdateDiffuseFilter(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	if(m_dwOperation == IMAGE_DIFFUSE_PLOT_FILTER)
		pCmdUI ->SetCheck(TRUE);
	else
		pCmdUI ->SetCheck(FALSE);	
}

void CDiffuseView::OnImageRestore() 
{
	// TODO: Add your command handler code here
	m_dwOperation = 0L;
	Invalidate();	
}

void CDiffuseView::OnUpdateImageRestore(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	if(m_dwOperation == 0L)
		pCmdUI->SetCheck(TRUE);
	else
		pCmdUI->SetCheck(FALSE);	
}

void CDiffuseView::Filter(CDC *pDC,  CDib *pDib,  int x,  int y,  int nWidth,  int nHeight)
{
	ASSERT(pDib);
		//限制处理的区域----防止任意填写数据而使用访问无效
	if((x > (m_nWidth - 1)) || (y > (m_nHeight - 1))) return ;
	//实际处理的宽度和高度
	int w = min(nWidth, m_nWidth - x);
	int h = min(nHeight, m_nHeight - y);
	
	// 32位颜色数据
	DWORD dwSize = m_nWidth * m_nHeight * 4;

	//分配全局内存 32位源数据
	BYTE* pbyBits32 =  new BYTE[dwSize];
	if(pbyBits32 == NULL) return;
	memset(pbyBits32, 0, dwSize);



	//CDib类只提供了一个全部数据的函数
	pDib->GetDdbData32(pbyBits32);


	//扩散图像, 引入随机性
	CDiffuseFilter* df = new CDiffuseFilter();
	
	df->SetDiffuseSize(5, 5);
	df->Filter(pbyBits32, x, y, w, h, m_nWidth, m_nHeight);

	CClientDC dc(this);
	CDC memDC;
	memDC.CreateCompatibleDC(&dc);
	HBITMAP hBitmap = df->CreateDdb(pDC->m_hDC, m_nWidth, m_nHeight, pbyBits32);
	HBITMAP hOldBitmap = (HBITMAP)memDC.SelectObject(hBitmap);
	pDC->BitBlt(0, 0, m_nWidth, m_nHeight, &memDC, 0, 0, SRCCOPY);
	::DeleteObject(hBitmap);
	memDC.SelectObject(hOldBitmap);

	delete df;
	delete[] pbyBits32;
	ReleaseDC(&dc);
}