histo.cpp

KL变换的实现

// Histo.cpp : implementation file
//

#include "stdafx.h"
#include "ClusterKLTransformation.h"
#include "Histo.h"

#include "MainFrm.h"
#include "ClusterKLTransformationView.h"

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

/////////////////////////////////////////////////////////////////////////////
// CHisto dialog


CHisto::CHisto(CWnd* pParent /*=NULL*/)
	: CDialog(CHisto::IDD, pParent)
{
	//{{AFX_DATA_INIT(CHisto)
	//}}AFX_DATA_INIT
}


void CHisto::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CHisto)
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CHisto, CDialog)
	//{{AFX_MSG_MAP(CHisto)
	ON_WM_PAINT()
	ON_BN_CLICKED(IDC_RADIO_BLUE, OnRadioBlue)
	ON_BN_CLICKED(IDC_RADIO_GREEN, OnRadioGreen)
	ON_BN_CLICKED(IDC_RADIO_RED, OnRadioRed)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CHisto message handlers

BOOL CHisto::OnInitDialog() 
{
	CDialog::OnInitDialog();
	

	
	unsigned char * lpSrc;
    CDib *m_pDib = pDoc->pDib;
	// 循环变量
	int i,j;


	// 获取绘制直方图的标签
	CWnd* pWnd = GetDlgItem(IDC_COORD);

	// 计算得到直方图
	// 图象的高度和宽度
	CSize sizeImage;

	sizeImage = m_pDib->GetDimensions();


	// 获得图象数据存储的高度和宽度
	CSize sizeSaveImage;
 	sizeSaveImage = m_pDib->GetDibSaveDim();

	// 重置计数为0
	for (i = 0; i < 256; i ++)
	{
		// 清零
		m_bcount[i] = 0;
		m_gcount[i] = 0;
		m_rcount[i] = 0;
	}
	
	// 计算各个灰度值的计数，即得到直方图
	for (i = 0; i < sizeImage.cy; i ++)
	{
		for (j = 0; j < sizeImage.cx; j ++)
		{
			if(m_pDib->NumberColors()==8)
			{
			lpSrc = (unsigned char *)m_pDib->m_lpImage + sizeSaveImage.cx * i + j;
			
			// 计数加1
			m_bcount[*(lpSrc)]++;
			m_gcount[*(lpSrc)]++;
			m_rcount[*(lpSrc)]++;
			}
			else if(m_pDib->NumberColors()==24)
			{
			lpSrc = (unsigned char *)m_pDib->m_lpImage + sizeSaveImage.cx * i + j*3;
			
			// 计数加1
			m_bcount[*(lpSrc)]++;
			m_gcount[*(lpSrc+1)]++;
            m_rcount[*(lpSrc+2)]++;
			}
			
		}
	}
	

	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CHisto::OnPaint() 
{
	CPaintDC dc(this); // device context for painting

	int i;
	// 获取绘制直方图文本框的标签
	CWnd* pWnd = GetDlgItem(IDC_COORD);

	// 获取设备上下文
	CDC* pDC = pWnd->GetDC();
	pWnd->Invalidate();
	pWnd->UpdateWindow();

	pDC->Rectangle(0, 0, 330, 300);

	// 创建画笔对象
	CPen* pPenBlack = new CPen;

	// 创建红色画笔（用于绘制坐标轴）
	pPenBlack->CreatePen(PS_SOLID, 1, RGB(0,0,0));

	// 选入红色画笔，并保存以前的画笔
	CPen* pOldPen = pDC->SelectObject(pPenBlack);

	// 绘制坐标轴
	pDC->MoveTo(10,10);
	
	// 绘制垂直轴
	pDC->LineTo(10, 280);

	// 绘制水平轴
	pDC->LineTo(320, 280);

	// 绘制X轴刻度值
	CString strTemp;
	strTemp.Format("0");
	pDC->TextOut(10, 283, strTemp);
	strTemp.Format("50");
	pDC->TextOut(60, 283, strTemp);
	strTemp.Format("100");
	pDC->TextOut(110, 283, strTemp);
	strTemp.Format("150");
	pDC->TextOut(160, 283, strTemp);
	strTemp.Format("200");
	pDC->TextOut(210, 283, strTemp);
	strTemp.Format("255");
	pDC->TextOut(265, 283, strTemp);
	
	// 绘制X轴刻度
	for (i = 0; i < 256; i += 5)
	{
		if ((i & 1) == 0)
		{
			// 10的倍数
			pDC->MoveTo(i + 10, 280);
			pDC->LineTo(i + 10, 284);
		}
		else
		{
			// 5的奇数倍数
			pDC->MoveTo(i + 10, 280);
			pDC->LineTo(i + 10, 282);
		}
	}
	
	// 绘制X轴箭头
	pDC->MoveTo(315,275);
	pDC->LineTo(320,280);
	pDC->LineTo(315,285);
	
	// 绘制Y轴箭头
	pDC->MoveTo(10,10);
	pDC->LineTo(5,15);
	pDC->MoveTo(10,10);
	pDC->LineTo(15,15);
	// TODO: Add your message handler code here
	
	// Do not call CDialog::OnPaint() for painting messages
}

void CHisto::OnRadioBlue() 
{
	// TODO: Add your control notification handler code here
	CPaintDC dc(this); 
	CWnd* pWnd = GetDlgItem(IDC_COORD);
	CDC* pDC = pWnd->GetDC();
	int i;
    pWnd->Invalidate();
	pWnd->UpdateWindow();
	OnPaint();
	// 直方图中最大计数值
	LONG lMaxCount = 0;

	// 计算最大计数值
	for (i = 0; i <= 255; i ++)
	{
		// 判断是否大于当前最大值
		if (m_bcount[i] > lMaxCount)
		{
			// 更新最大值
			lMaxCount = m_bcount[i];
		}
	}
	CString strTemp;
	// 输出最大计数值
	pDC->MoveTo(10, 25);
	pDC->LineTo(14, 25);
	strTemp.Format("%d", lMaxCount);
	pDC->TextOut(11, 26, strTemp);
	
	// 声名画笔对象
	CPen* pPenBlue = new CPen;

	// 创建蓝色画笔（用于绘制直方图）
	pPenBlue->CreatePen(PS_SOLID, 1, RGB(0,0,255));

	// 选入蓝色画笔
	pDC->SelectObject(pPenBlue);

	// 判断是否存在计数值
	if(lMaxCount > 0){
		// 绘制直方图
		for (i = 0; i <= 255; i ++)
		{
			pDC->MoveTo(i + 10, 280);
			pDC->LineTo(i + 10, 281 - (int) (m_gcount[i] * 256 / lMaxCount));
		}
	}
	delete pPenBlue;
}

void CHisto::OnRadioGreen() 
{
	// TODO: Add your control notification handler code here
	CPaintDC dc(this); 
	CWnd* pWnd = GetDlgItem(IDC_COORD);
	CDC* pDC = pWnd->GetDC();	
	int i;
    pWnd->Invalidate();
	pWnd->UpdateWindow();
	OnPaint();
	// 直方图中最大计数值
	LONG lMaxCount = 0;

	// 计算最大计数值
	for (i = 0; i <= 255; i ++)
	{
		// 判断是否大于当前最大值
		if (m_gcount[i] > lMaxCount)
		{
			// 更新最大值
			lMaxCount = m_gcount[i];
		}
	}
	CString strTemp;
	// 输出最大计数值
	pDC->MoveTo(10, 25);
	pDC->LineTo(14, 25);
	strTemp.Format("%d", lMaxCount);
	pDC->TextOut(11, 26, strTemp);
	
	// 声名画笔对象
	CPen* pPenGreen = new CPen;

	// 创建蓝色画笔（用于绘制直方图）
	pPenGreen->CreatePen(PS_SOLID, 1, RGB(0,255,0));

	// 选入蓝色画笔
	pDC->SelectObject(pPenGreen);

	// 判断是否存在计数值
	if(lMaxCount > 0){
		// 绘制直方图
		for (i = 0; i <= 255; i ++)
		{
			pDC->MoveTo(i + 10, 280);
			pDC->LineTo(i + 10, 281 - (int) (m_gcount[i] * 256 / lMaxCount));
		}
	}
    
	delete pPenGreen;
}

void CHisto::OnRadioRed() 
{
	// TODO: Add your control notification handler code here
		CPaintDC dc(this); 
	CWnd* pWnd = GetDlgItem(IDC_COORD);
	CDC* pDC = pWnd->GetDC();
	
	int i;
        pWnd->Invalidate();
	pWnd->UpdateWindow();
	OnPaint();
	// 直方图中最大计数值
	LONG lMaxCount = 0;

	// 计算最大计数值
	for (i = 0; i <= 255; i ++)
	{
		// 判断是否大于当前最大值
		if (m_rcount[i] > lMaxCount)
		{
			// 更新最大值
			lMaxCount = m_rcount[i];
		}
	}
	CString strTemp;
	// 输出最大计数值
	pDC->MoveTo(10, 25);
	pDC->LineTo(14, 25);
	strTemp.Format("%d", lMaxCount);
	pDC->TextOut(11, 16, strTemp);
	
	// 声名画笔对象
	CPen* pPenRed = new CPen;

	// 创建蓝色画笔（用于绘制直方图）
	pPenRed->CreatePen(PS_SOLID, 1, RGB(255,0,0));

	// 选入蓝色画笔
	pDC->SelectObject(pPenRed);

	// 判断是否存在计数值
	if(lMaxCount != 0){
		// 绘制直方图
		for (i = 0; i <= 255; i ++)
		{
			pDC->MoveTo(i + 10, 280);
			pDC->LineTo(i + 10, 281 - m_rcount[i] * 256 / lMaxCount);
		}
	}
	delete pPenRed;
}

void CHisto::OnOK() 
{
//  	if(m_Blue == -1) 
// 		AfxMessageBox("请选择特定类像素!");
// 
	CDialog::EndDialog(IDOK);
}

void CHisto::GetpDoc(CClusterKLTransformationDoc* p)
{
	pDoc = p;
}