granulateview.cpp

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

// GranulateView.cpp : implementation of the CGranulateView class
//

#include "stdafx.h"
#include "Granulate.h"

#include "GranulateDoc.h"
#include "GranulateView.h"

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

#include "GranulationFilter.h"

/////////////////////////////////////////////////////////////////////////////
// CGranulateView

IMPLEMENT_DYNCREATE(CGranulateView,  CView)

BEGIN_MESSAGE_MAP(CGranulateView,  CView)
	//{{AFX_MSG_MAP(CGranulateView)
	ON_COMMAND(IDM_GRANULATION_FILTER,  OnGranulationFilter)
	ON_UPDATE_COMMAND_UI(IDM_GRANULATION_FILTER,  OnUpdateGranulationFilter)
	ON_COMMAND(IDM_IMAGE_RESTORE,  OnImageRestore)
	ON_UPDATE_COMMAND_UI(IDM_IMAGE_RESTORE,  OnUpdateImageRestore)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CGranulateView construction/destruction

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

CGranulateView::~CGranulateView()
{
}

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

	return CView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CGranulateView drawing

void CGranulateView::OnDraw(CDC* pDC)
{
	CGranulateDoc* 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();
	}

}

/////////////////////////////////////////////////////////////////////////////
// CGranulateView diagnostics

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

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

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

/////////////////////////////////////////////////////////////////////////////
// CGranulateView message handlers

void CGranulateView::OnGranulationFilter() 
{
	// TODO: Add your command handler code here
	m_dwOperation = IMAGE_GRANUALATION_PLOT_FILTER;
	Invalidate();
}

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

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

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

void CGranulateView::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 == 0) return;

	memset(pbyBits32, 0, dwSize);


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


	//粒化图像
	CGranulationFilter* gf = new CGranulationFilter();
	
	gf->SetGranulationSize(5, 5);
	gf->Filter(pbyBits32, x, y, w, h, m_nWidth, m_nHeight);

	CClientDC dc(this);
	CDC memDC;
	memDC.CreateCompatibleDC(&dc);
	HBITMAP hBitmap = gf->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 gf;
	delete[] pbyBits32;
	ReleaseDC(&dc);
}