medianblurview.cpp

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

// MedianBlurView.cpp : implementation of the CMedianBlurView class
//

#include "stdafx.h"
#include "MedianBlur.h"

#include "MedianBlurDoc.h"
#include "MedianBlurView.h"

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

#include "MedianFilter.h"

/////////////////////////////////////////////////////////////////////////////
// CMedianBlurView

IMPLEMENT_DYNCREATE(CMedianBlurView,  CView)

BEGIN_MESSAGE_MAP(CMedianBlurView,  CView)
	//{{AFX_MSG_MAP(CMedianBlurView)
	ON_COMMAND(IDM_MEDIAN_FILTER,  OnMedianFilter)
	ON_UPDATE_COMMAND_UI(IDM_MEDIAN_FILTER,  OnUpdateMedianFilter)
	ON_COMMAND(IDM_IMAGE_RESTORE,  OnImageRestore)
	ON_UPDATE_COMMAND_UI(IDM_IMAGE_RESTORE,  OnUpdateImageRestore)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CMedianBlurView construction/destruction

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

CMedianBlurView::~CMedianBlurView()
{
}

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

	return CView::PreCreateWindow(cs);
}

/////////////////////////////////////////////////////////////////////////////
// CMedianBlurView drawing

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

}

/////////////////////////////////////////////////////////////////////////////
// CMedianBlurView diagnostics

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

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

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

/////////////////////////////////////////////////////////////////////////////
// CMedianBlurView message handlers

void CMedianBlurView::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);

	//
	CMedianFilter* mf = new CMedianFilter();
	
	int* pnMask = new int[25];

	// 十字形
	for(int i = 0;i < 25;i++)pnMask[i] = 0;
	pnMask[2] = pnMask[7] = pnMask[12] = pnMask[17] = pnMask[22] = 1;
	pnMask[10] = pnMask[11] = pnMask[13] = pnMask[14] = 1;


	mf->SetMask(pnMask, 5, 5);
	
	mf->Medianize(pbyBits32, x, y, w, h, m_nWidth, m_nHeight);

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

void CMedianBlurView::OnMedianFilter() 
{
	// TODO: Add your command handler code here
	m_dwOperation = IMAGE_BLUR_MEDIAN_FILTER;
	Invalidate();	
}

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

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

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