homework3view.cpp

图像分割迭代、峰谷、半阈值算法。

// homework3View.cpp : implementation of the CHomework3View class
//

#include "stdafx.h"
#include "homework3.h"

#include "homework3Doc.h"
#include "homework3View.h"
#include "DlgYuZhiFenGe1.h"
#include "DlgZhiFangTu.h"

#include "CDIB.h"
#include "BingXingBianJieDib.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CHomework3View

IMPLEMENT_DYNCREATE(CHomework3View, CView)

BEGIN_MESSAGE_MAP(CHomework3View, CView)
	//{{AFX_MSG_MAP(CHomework3View)
	ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
	ON_COMMAND(ID_YUANTU, OnYuantu)
	ON_COMMAND(ID_Diedaifazhi, OnDiedaifazhi)
	ON_COMMAND(ID_Yuzhifenge, OnYuzhifenge)
	ON_COMMAND(ID_BanYuZhi, OnBanYuZhi)
	ON_COMMAND(ID_LunKuoTiQu, OnLunKuoTiQu)
	ON_COMMAND(ID_FILE_SAVE, OnFileSave)
	//}}AFX_MSG_MAP
	// Standard printing commands
	ON_COMMAND(ID_FILE_PRINT, CView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_DIRECT, CView::OnFilePrint)
	ON_COMMAND(ID_FILE_PRINT_PREVIEW, CView::OnFilePrintPreview)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CHomework3View construction/destruction

CHomework3View::CHomework3View()
{
	// TODO: add construction code here
    state2=0;
	state=0;
	state1=0;
}

CHomework3View::~CHomework3View()
{
}

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

	return CView::PreCreateWindow(cs);
}

/*初始化图像数据*/
void CHomework3View::clearmem()
{
	CHomework3Doc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
//	pDoc ->statedoc=0;
    state2=1;
	CDibNew1=&CDibNew;
    CDib1=&CDib;
    long int  size=CDib1->GetHeight()*CDib1->GetDibWidthBytes();
	memcpy(CDibNew1->m_pData,CDib1->m_pData,size);//复制原图像到处理区
}
/////////////////////////////////////////////////////////////////////////////
// CHomework3View drawing

void CHomework3View::OnDraw(CDC* pDC)
{
	CHomework3Doc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	// TODO: add draw code for native data here
	if(state1==1)
	{
		CBitmap m_bitmap;
		HBITMAP hBitmap=(HBITMAP)LoadImage(NULL,_T(filename),IMAGE_BITMAP,
			0,0,LR_CREATEDIBSECTION|LR_DEFAULTSIZE|LR_LOADFROMFILE);
        m_bitmap.Attach (hBitmap);
		CDC dcImage;
		if(!dcImage.CreateCompatibleDC (pDC))
			return;
		BITMAP bm;
		m_bitmap.GetBitmap (&bm);
        dcImage.SelectObject (&m_bitmap);
		pDC->BitBlt (0,0,bm.bmWidth ,bm.bmHeight ,&dcImage,0,0,SRCCOPY);
	}
	if(state2==1)//!pDoc ->statedoc&&
	{
        int m_scale=1;
		BYTE* pBitmapData = CDibNew1->GetData();
        LPBITMAPINFO pBitmapInfo = CDibNew1->GetInfo();
        int bitmapHeight = CDibNew1->GetHeight();
        int bitmapWidth = CDibNew1->GetWidth();
		int scaledWidth = (int)(bitmapWidth * m_scale);
		int scaledHeight = (int)(bitmapHeight * m_scale);
		if (CDibNew1->GetRGB()) // Has a color table
		{
			CPalette * hPalette=CreateBitmapPalette(CDibNew1);
            CPalette * hOldPalette =
                pDC->SelectPalette(hPalette, true);
            pDC->RealizePalette();
			::StretchDIBits(pDC->GetSafeHdc(),0, 0, bitmapWidth, bitmapHeight,
				0, 0, bitmapWidth, bitmapHeight,
				pBitmapData, pBitmapInfo,
				DIB_RGB_COLORS, SRCCOPY);
            pDC->SelectPalette(hOldPalette, true);
            ::DeleteObject(hPalette);
		}
		else
		{
            StretchDIBits(pDC->GetSafeHdc(),
                0, 0, scaledWidth, scaledHeight,
                0, 0, bitmapWidth, bitmapHeight,
				pBitmapData, pBitmapInfo,
				DIB_RGB_COLORS, SRCCOPY);
		}
	}
}


CPalette * CHomework3View::CreateBitmapPalette(BingXingBianJieDib * pBitmap)
{
	struct
	{
		WORD Version;
		WORD NumberOfEntries;
		PALETTEENTRY aEntries[256];
	} palette = { 0x300, 256 };
	
	LPRGBQUAD pRGBTable = pBitmap->GetRGB();
	UINT numberOfColors = pBitmap->GetNumberOfColors();
	
	for(UINT x=0; x<numberOfColors; ++x)
	{
		palette.aEntries[x].peRed =
			pRGBTable[x].rgbRed;
		palette.aEntries[x].peGreen =
			pRGBTable[x].rgbGreen;
		palette.aEntries[x].peBlue =
			pRGBTable[x].rgbBlue;
		palette.aEntries[x].peFlags = 0;
	}
	
	hPalette.CreatePalette((LPLOGPALETTE)&palette);
	//CPalette hPale=&hPalette;
	return &hPalette;
}
/////////////////////////////////////////////////////////////////////////////
// CHomework3View printing

BOOL CHomework3View::OnPreparePrinting(CPrintInfo* pInfo)
{
	// default preparation
	return DoPreparePrinting(pInfo);
}

void CHomework3View::OnBeginPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add extra initialization before printing
}

void CHomework3View::OnEndPrinting(CDC* /*pDC*/, CPrintInfo* /*pInfo*/)
{
	// TODO: add cleanup after printing
}

/////////////////////////////////////////////////////////////////////////////
// CHomework3View diagnostics

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

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

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

/////////////////////////////////////////////////////////////////////////////
// CHomework3View message handlers

void CHomework3View::OnFileOpen() 
{
	// TODO: Add your command handler code here
	CFileDialog dlg(TRUE,_T("BMP"),_T("*.BMP"),OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,_T("位图文件(*.BMP)|*.BMP|"));	
    if(IDOK==dlg.DoModal ())
    filename.Format ("%s",dlg.GetPathName() );    
	CDib.LoadFile(filename);
    CDibNew.LoadFile(filename);
	state1=1;
	Invalidate();
}

void CHomework3View::OnYuantu() 
{
	// TODO: Add your command handler code here
	CHomework3Doc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	clearmem();
 	state1=1;
	Invalidate();
}

void CHomework3View::OnDiedaifazhi() 
{
	// TODO: Add your command handler code here
	clearmem();
	// 各颜色分量的灰度分布密度
	int tongji[256];
	//初始为0
	memset(tongji, 0, sizeof(tongji));
    CDibNew1->Fenbutongji(tongji);      //统计直方图灰度值
	CDibNew1->Diedaifazhi(tongji);             //调用迭代法值检测处理函数
	Invalidate();
}
/*阈值分割消息映射*/ 
void CHomework3View::OnYuzhifenge() 
{
	// TODO: Add your command handler code here
	int i;
	int Yuzhi;   //阈值变量
	// 各颜色分量的灰度分布密度
    float midu[256];
	
	// 计算灰度分布密度	 	  
	clearmem();
	CDibNew1->Zhifangtu(midu);             //调用灰度统计处理函数
	
	// 创建灰度直方图对话框
	CDlgZhiFangTu*    dlg;
	dlg=new CDlgZhiFangTu(this);
	dlg->Create(IDD_DIALOG_ZhiFangTu);
	// 传递灰度分布密度数据给面板类
	for (i = 0; i <256; i++)
		dlg->m_fIntensity[i] = midu[i];
	
	// 显示对话框，由用户进行灰度折线变换
	dlg->ShowWindow(SW_RESTORE);
	
    //创建阈值选择对话框
	CDlgYuZhiFenGe  dlg1;
	dlg1.m_Yuzhi=0;
	
	// 显示对话框，提示用户输入阈值
	if (dlg1.DoModal() != IDOK)
	{
		// 返回
		return;
	}
    Yuzhi=dlg1.m_Yuzhi;
	
	// 删除对话框
	delete dlg1;
	delete dlg;
	
	clearmem();
	CDibNew1->Yuzhifenge(Yuzhi);             //调用阈值分割处理函数
	Invalidate();
}

/*半阈值分割消息映射*/ 
void CHomework3View::OnBanYuZhi() 
{
	// TODO: Add your command handler code here
		int i;
	int Yuzhi;   //阈值变量
	// 各颜色分量的灰度分布密度
    float midu[256];
	
	// 计算灰度分布密度	 	  
	clearmem();
	CDibNew1->Zhifangtu(midu);             //调用灰度统计处理函数
	
	// 创建灰度直方图对话框
	CDlgZhiFangTu*    dlg;
	dlg=new CDlgZhiFangTu(this);
	dlg->Create(IDD_DIALOG_ZhiFangTu);
	// 传递灰度分布密度数据给面板类
	for (i = 0; i <256; i++)
		dlg->m_fIntensity[i] = midu[i];
	
	// 显示对话框，由用户进行灰度折线变换
	dlg->ShowWindow(SW_RESTORE);
	
    //创建阈值选择对话框
	CDlgYuZhiFenGe  dlg1;
	dlg1.m_Yuzhi=0;
	
	// 显示对话框，提示用户输入阈值
	if (dlg1.DoModal() != IDOK)
	{
		// 返回
		return;
	}
    Yuzhi=dlg1.m_Yuzhi;
	
	// 删除对话框
	delete dlg1;
	delete dlg;
	
	clearmem();
	CDibNew1->BanYuZhi(Yuzhi);             //调用半阈值分割处理函数
	Invalidate();
}

/*轮廓提取消息映射*/
void CHomework3View::OnLunKuoTiQu() 
{
	// TODO: Add your command handler code here
	clearmem();
	if(CDibNew1->m_pBitmapInfoHeader->biBitCount<9)
	{
		
		CDibNew1->Lunkuotiqu();             //调用轮廓提取处理函数
		Invalidate();
	}
	else
	{
		Invalidate();
		MessageBox("在右图单击鼠标左键选取种子点");
		state=3;
	}
}

void CHomework3View::OnFileSave() 
{
	// TODO: Add your command handler code here
	CFileDialog dlg(FALSE,_T("BMP"),_T("*.BMP"),OFN_HIDEREADONLY|OFN_OVERWRITEPROMPT,_T("位图文件(*.BMP)|*.BMP|"));	
    if(IDOK==dlg.DoModal())
	CString  filename;
    filename.Format ("%s",dlg.GetPathName() );    
    CDibNew1->SaveFile(filename);
	state2=1;
	Invalidate();
}