dynsplitview2.cpp

< VC++图像处理程程序设计>>配套光盘,这是第4章的灰度变换的程序

// DynSplitView2.cpp : implementation file
//

#include "stdafx.h"
#include "DSplit.h"
#include "DynSplitView2.h"
#include "DlgZhiFangTu.h"
#include "DlgChuangkoubianhuan.h" 
#include "DlgZhexianbianhuan.h" 
#include "DlgGuDingFaZhi.h"
#include "DialogShuangYu.h" 
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CDynSplitView2

IMPLEMENT_DYNCREATE(CDynSplitView2, CView)

CDynSplitView2::CDynSplitView2()
{
	state2=0;
}


/*初始化图像数据*/
void CDynSplitView2::clearmem()
{
	CDSplitDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	pDoc ->statedoc=0;
    state2=1;
	CDibNew1=&pDoc->CDibNew;
    CDib1=&pDoc->CDib;
    long int  size=CDib1->GetHeight()*CDib1->GetWidth();
    memcpy(CDibNew1->m_pData,CDib1->m_pData,size);
}

CPalette * CDynSplitView2::CreateBitmapPalette(HuiDuBianHuanDib * 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);
	
	return &hPalette;
}

CDynSplitView2::~CDynSplitView2()
{
}
CDSplitDoc* CDynSplitView2::GetDocument() // non-debug version is inline
{
	ASSERT(m_pDocument->IsKindOf(RUNTIME_CLASS(CDSplitDoc)));
	return (CDSplitDoc*)m_pDocument;
}


BEGIN_MESSAGE_MAP(CDynSplitView2, CView)
	//{{AFX_MSG_MAP(CDynSplitView2)
	ON_WM_ERASEBKGND()
	ON_COMMAND(ID_FILESAVE, OnFilesave)
	ON_COMMAND(ID_Zhifangtutongji, OnZhifangtutongji)
	ON_COMMAND(ID_FanSeBianHuan, OnFanSeBianHuan)
	ON_COMMAND(ID_Chuangkoubianhuan, OnChuangkoubianhuan)
	ON_COMMAND(ID_ZheXianBianHuan, OnZheXianBianHuan)
	ON_COMMAND(ID_Fenbujunhenghua, OnFenbujunhenghua)
	ON_COMMAND(ID_Pipeibianhuan, OnPipeibianhuan)
	ON_COMMAND(ID_Yuantuzhifangtu, OnYuantuzhifangtu)
	ON_COMMAND(ID_Shuipingtouying, OnShuipingtouying)
	ON_COMMAND(ID_Chuizhitouying, OnChuizhitouying)
	ON_COMMAND(ID_Fei0, OnFei0)
	ON_COMMAND(ID_GUDING, OnGuding)
	ON_COMMAND(ID_ShuanGuDingFa, OnShuanGuDingFa)
	ON_COMMAND(ID_RedZhiFangtu, OnRedZhiFangtu)
	ON_COMMAND(ID_BlueZhiFangtu, OnBlueZhiFangtu)
	ON_COMMAND(ID_GreenZhifangtu, OnGreenZhifangtu)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CDynSplitView2 drawing

void CDynSplitView2::OnDraw(CDC* pDC)
{	
	CDSplitDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	if(!pDoc ->statedoc&&state2==1)
	{
        BYTE* pBitmapData = CDibNew1->GetData();
        LPBITMAPINFO pBitmapInfo = CDibNew1->GetInfo();
		int m_scale=1;
        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);
		}
	}
}


/////////////////////////////////////////////////////////////////////////////
// CDynSplitView2 diagnostics

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

void CDynSplitView2::Dump(CDumpContext& dc) const
{
	CView::Dump(dc);
}
#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CDynSplitView2 message handlers

BOOL CDynSplitView2::OnEraseBkgnd(CDC* pDC) 
{
	CRect rect;
	GetClientRect(&rect);
	pDC->FillSolidRect(&rect,::GetSysColor(COLOR_WINDOW));
	return TRUE;
}

void CDynSplitView2::OnFilesave() 
{
	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();
}

//处理后直方图显示
void CDynSplitView2::OnZhifangtutongji() 
{
	int i;
	// 各颜色分量的灰度分布密度
    float midu[256];
	
	// 计算灰度分布密度
	CDSplitDoc* pDoc = GetDocument();
	ASSERT_VALID(pDoc);
	pDoc ->statedoc=0;
    state2=1;
	
	CDibNew1->ZhiFangTu(midu);             //调用灰度统计处理函数
	
	// 创建对话框
	CDlgZhiFangTu*  dlg;	 
    dlg=new CDlgZhiFangTu(this);
    dlg->Create(IDD_DLG_zhifangtufenbu);
	// 传递灰度分布密度数据给面板类
	for (i = 0; i <256; i++)
		dlg->m_fIntensity[i] = midu[i];
	
	// 显示处理后的直方图
	dlg->ShowWindow(SW_RESTORE);
}

//反色变换消息映射
void CDynSplitView2::OnFanSeBianHuan() 
{
    clearmem();
	int i;
	state3=0;
	
	// 计算灰度分布密度
	CDibNew1->ZhiFangTu(midu2);									//调用灰度统计处理函数
	
	// 创建对话框
	CDlgZhiFangTu*  dlg2; 
    dlg2=new CDlgZhiFangTu(this);
    dlg2->Create(IDD_DLG_zhifangtufenbu);
	
	// 传递灰度分布密度数据给面板类
	for (i = 0; i <256; i++)
		dlg2->m_fIntensity[i] = midu2[i];
	if(CDibNew1->m_pBitmapInfoHeader->biBitCount<9)		//灰度图像
	{
		clearmem();
		CDibNew1->FanSeBianHuan();								//调用反色变换处理函数
		Invalidate();											//调用刷新函数
	}
	else		//24位彩色
	{
		clearmem2();
		CDibNew1->FanSeBianHuan();								//调用反色变换处理函数
		Invalidate();											//调用刷新函数
	}
}

//灰度窗口变换消息映射
void CDynSplitView2::OnChuangkoubianhuan() 
{
	clearmem();
	int i;
	state3=0;
	// 计算灰度分布密度
	CDibNew1->ZhiFangTu(midu2);									//调用灰度统计处理函数
	
	// 创建对话框
	CDlgZhiFangTu*  dlg2;	 
	dlg2=new CDlgZhiFangTu(this);
	dlg2->Create(IDD_DLG_zhifangtufenbu);
	
	// 传递灰度分布密度数据给面板类
	for (i = 0; i <256; i++)
		dlg2->m_fIntensity[i] = midu2[i];
	// 创建对话框
	CDlgChuangkoubianhuan  dlg;
	
	// 窗口下限
	BYTE	bLow;	
	// 窗口上限
	BYTE	bUp;
	
	// 初始化变量值
	dlg.m_bLow = 50;
	dlg.m_bUp = 200;
	
	// 显示对话框，提示用户设定窗口上下限
	if (dlg.DoModal() != IDOK)
	{
		// 返回
		return;
	}	
	// 获取用户设定的窗口上下限
	bLow = dlg.m_bLow;
	bUp = dlg.m_bUp;
	
	// 删除对话框
	delete dlg;			
	// 调用WindowTrans()函数进行窗口变换
	if(CDibNew1->m_pBitmapInfoHeader->biBitCount<9)		//灰度图像
	{
		clearmem();
		CDibNew1->Chuangkoubianhuan(  bLow,   bUp);             //调用窗口变换处理函数
		Invalidate();											//调用刷新函数 
	}
	else		//24位彩色
	{
		clearmem2();
		CDibNew1->Chuangkoubianhuan(  bLow,   bUp);             //调用窗口变换处理函数
		Invalidate();											//调用刷新函数 
	}
	
}


//折线变换消息映射
void CDynSplitView2::OnZheXianBianHuan() 
{
	clearmem();
	int i;
	state3=0;
	
	// 计算灰度分布密度
	CDibNew1->ZhiFangTu(midu2);             //调用灰度统计处理函数
	
	// 创建对话框
	CDlgZhiFangTu*  dlg2;
    dlg2=new CDlgZhiFangTu(this);
    dlg2->Create(IDD_DLG_zhifangtufenbu);
	
	// 传递灰度分布密度数据给面板类
	for (i = 0; i <256; i++)
		dlg2->m_fIntensity[i] = midu2[i];
	
	// 创建对话框
	CDlgZhexianbianhuan dlgPara;
	
	// 点1坐标
	BYTE	bX1;
	BYTE	bY1;	
	// 点2坐标
	BYTE	bX2;
	BYTE	bY2;
	
	// 初始化变量值
	dlgPara.m_bX1 = 50;
	dlgPara.m_bY1 = 30;
	dlgPara.m_bX2 = 200;
	dlgPara.m_bY2 = 220;
	
	// 显示对话框，提示用户设定拉伸位置
	if (dlgPara.DoModal() != IDOK)
	{
		// 返回
		return;
	}
	
	// 获取用户的设定
	bX1 = dlgPara.m_bX1;
	bY1 = dlgPara.m_bY1;
	bX2 = dlgPara.m_bX2;
	bY2 = dlgPara.m_bY2;
	
	// 删除对话框
	delete dlgPara;	
	if  (CDibNew1->m_pBitmapInfoHeader->biBitCount<9)		//灰度图像
	{
		// 调用Zhexianbianhuan()函数进行灰度拉伸	 
		clearmem();
		CDibNew1->Zhexianbianhuan( bX1,bY1,bX2,bY2);             //调用折线变换处理函数
		Invalidate();           //调用刷新函数
	}
	else		//24位彩色
	{
		// 调用Zhexianbianhuan()函数进行灰度拉伸	 
		clearmem2();
		CDibNew1->Zhexianbianhuan( bX1,bY1,bX2,bY2);             //调用折线变换处理函数
		Invalidate();           //调用刷新函数