www.pudn.com > Hough.rar > HoughDlg.cpp
// HoughDlg.cpp : implementation file
//
#include "stdafx.h"
#include "Hough.h"
#include "HoughDlg.h"
#include "HoughTransform.h"
#include "Bmp.h"
#include "CfgDlg.h"
#include "GlobalVariable.h"
#include "afx.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CHoughDlg dialog
CHoughDlg::CHoughDlg(CWnd* pParent /*=NULL*/)
: CDialog(CHoughDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CHoughDlg)
m_sCentersCoord = _T("");
//}}AFX_DATA_INIT
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CHoughDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CHoughDlg)
DDX_Control(pDX, ID_SOURCE_BMP, m_picSourceImg);
DDX_Text(pDX, IDC_Edit_CircleCenters, m_sCentersCoord);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CHoughDlg, CDialog)
//{{AFX_MSG_MAP(CHoughDlg)
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDOPEN, OnOpen)
ON_BN_CLICKED(IDCONFIG, OnConfig)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CHoughDlg message handlers
BOOL CHoughDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
//m_sCentersCoord = "";
// TODO: Add extra initialization here
m_FullFilename = "c:\\source.bmp";
return TRUE; // return TRUE unless you set the focus to a control
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CHoughDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CHoughDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
void CHoughDlg::OnOK()
{
// 清空圆心图和圆图的显示
((CStatic*)GetDlgItem(ID_CENTER_BMP))->SetBitmap(NULL);
((CStatic*)GetDlgItem(ID_TARGET_BMP))->SetBitmap(NULL);
// 删除圆心图文件和圆图文件
CFileStatus fs;
if (CFile::GetStatus(_T("c:\\centers.bmp"), fs)) // centers.bmp文件存在
{
CFile::Remove("c:\\centers.bmp");
}
if (CFile::GetStatus(_T("c:\\target.bmp"), fs)) // target.bmp文件存在
{
CFile::Remove("c:\\target.bmp");
}
// 清空圆心坐标编辑框
m_sCentersCoord = "";
// TODO: Add extra validation here
BYTE* pPaletteData;
BYTE* pImgData;
//BYTE* pLeftData;
long paletteDataCount,imgDataCount;
//long leftDataCount;
BmpHead bmpHead;
InfoHead infoHead;
CString errorMsg="", info="";
// 读BMP图像到内存
BOOL result = ::ReadBinaryBMP(m_FullFilename, bmpHead, infoHead,
pPaletteData, paletteDataCount,
pImgData, imgDataCount, /*pLeftData, leftDataCount,*/
errorMsg, info);
CString showMsg = result?info:errorMsg;
AfxMessageBox(showMsg);
// 生成两份内存块的拷贝，后面要用
BYTE* pPaletteData2;
BYTE* pImgData2;
BYTE* pPaletteData3;
BYTE* pImgData3;
pPaletteData2 = new BYTE[paletteDataCount];
pPaletteData3 = new BYTE[paletteDataCount];
pImgData2 = new BYTE[imgDataCount];
pImgData3 = new BYTE[imgDataCount];
memcpy(pPaletteData2, pPaletteData, paletteDataCount);
memcpy(pPaletteData3, pPaletteData, paletteDataCount);
memcpy(pImgData2, pImgData, imgDataCount);
memcpy(pImgData3, pImgData, imgDataCount);
CString tmpInfo;
info = "";
errorMsg = "";
int threshold = 85;
int* pDetectedCenterPos;
// int detectedCentersCount;
// // Hough变换检测圆心, 用设阈值法
// ::HoughTransform((unsigned char *)pImgData, infoHead.width, infoHead.height, threshold,
// pDetectedCenterPos, detectedCentersCount);
// info.Format("共检测出%d个圆心", detectedCentersCount);
int circleNum = g_nCircleNum;
int nMaxRadius = g_nMaxRadius;
int edgeThick = g_nEdgeThick;
int nDistBetweenCenters = g_nDistBetweenCenters;
pDetectedCenterPos = new int[circleNum];
// Hough变换检测圆心, 用设置圆个数的方法
::HoughTransform((unsigned char *)pImgData, infoHead.width, infoHead.height, circleNum,
nMaxRadius, edgeThick, nDistBetweenCenters, pDetectedCenterPos);
// 将圆心的坐标写在圆心坐标编辑框
int xPos, yPos;
CString sTemp;
for (int i = 0; i < circleNum; i++)
{
yPos = pDetectedCenterPos[i] / infoHead.width;
xPos = pDetectedCenterPos[i] % infoHead.width;
sTemp.Format("(%d,%d) ", xPos, yPos);
m_sCentersCoord += sTemp;
}
UpdateData(FALSE);
// 将检测出的圆心点画在内存块中
info = "";
errorMsg = "";
BOOL isCircleCenter;
for(int m=0; mSetBitmap(hsource);
// 显示检测出的圆图像
HBITMAP hsource2=(HBITMAP)LoadImage(AfxGetInstanceHandle(),"c:\\target.bmp",IMAGE_BITMAP,0,0,LR_LOADFROMFILE);
int k2 = GetLastError();
HBITMAP hbmp2 = ((CStatic*)GetDlgItem(ID_TARGET_BMP))->SetBitmap(hsource2);
//CDialog::OnOK();
}
void CHoughDlg::OnCancel()
{
// TODO: Add extra cleanup here
CDialog::OnCancel();
}
void CHoughDlg::OnOpen()
{
// TODO: Add your control notification handler code here
AfxMessageBox("本软件只能针对bmp格式的二值（像素值为0或255）图像, \n 且背景为黑色，前景为白色!");
CFileDialog dlg(TRUE);
if(dlg.DoModal() == IDOK)
{
m_FullFilename = dlg.GetPathName();
}
// 显示原图
HBITMAP hsource=(HBITMAP)LoadImage(AfxGetInstanceHandle(),m_FullFilename,IMAGE_BITMAP,0,0,LR_LOADFROMFILE);
int i = GetLastError();
HBITMAP hbmp = ((CStatic*)GetDlgItem(ID_SOURCE_BMP))->SetBitmap(hsource);
// 清空圆心图和圆图的显示
((CStatic*)GetDlgItem(ID_CENTER_BMP))->SetBitmap(NULL);
((CStatic*)GetDlgItem(ID_TARGET_BMP))->SetBitmap(NULL);
// 删除圆心图文件和圆图文件
CFileStatus fs;
if (CFile::GetStatus(_T("c:\\centers.bmp"), fs)) // centers.bmp文件存在
{
CFile::Remove("c:\\centers.bmp");
}
if (CFile::GetStatus(_T("c:\\target.bmp"), fs)) // target.bmp文件存在
{
CFile::Remove("c:\\target.bmp");
}
m_sCentersCoord = "";
UpdateData(FALSE);
}
void CHoughDlg::OnConfig()
{
// TODO: Add your control notification handler code here
CCfgDlg cfgDlg;
cfgDlg.DoModal();
}