sampledemodlg.cpp

SampleDemo指纹模式识别预处理演示（特征点获取）

// SampleDemoDlg.cpp : implementation file
//

#include "stdafx.h"
#include "SampleDemo.h"
#include "SampleDemoDlg.h"
#include "FPA.H"
#include "resource.h"

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

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CSampleDemoDlg dialog

CSampleDemoDlg::CSampleDemoDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CSampleDemoDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CSampleDemoDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CSampleDemoDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CSampleDemoDlg)
	DDX_Control(pDX, IDC_STATIC_FEATURE, m_img_feature);
	DDX_Control(pDX, IDC_STATIC_IMG, m_img);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CSampleDemoDlg, CDialog)
	//{{AFX_MSG_MAP(CSampleDemoDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_BUTTON_OPEN, OnButtonOpen)
	ON_BN_CLICKED(IDC_BUTTON_SAVE_AS, OnButtonSaveAs)
	ON_BN_CLICKED(IDC_BUTTON_Smooth, OnBUTTONSmooth)
	ON_BN_CLICKED(IDC_BUTTON_CONVERGE, OnButtonConverge)
	ON_BN_CLICKED(IDC_BUTTON_GRAD, OnButtonGrad)
	ON_BN_CLICKED(IDC_BUTTON_ORIENTATION, OnButtonOrientation)
	ON_BN_CLICKED(IDC_BUTTON_ENHANCE, OnButtonEnhance)
	ON_BN_CLICKED(IDC_BUTTON_BINARY, OnButtonBinary)
	ON_BN_CLICKED(IDC_BUTTON_SEGMENT, OnButtonSegment)
	ON_BN_CLICKED(IDC_BUTTON_THIN, OnButtonThin)
	ON_BN_CLICKED(IDC_BUTTON_FEATURE, OnButtonFeature)
	ON_BN_CLICKED(IDC_BUTTON_EQUALIZATION, OnButtonEqualization)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CSampleDemoDlg message handlers

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

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// 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
	
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CSampleDemoDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 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 CSampleDemoDlg::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 CSampleDemoDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

BEGIN_EVENTSINK_MAP(CSampleDemoDlg, CDialog)
    //{{AFX_EVENTSINK_MAP(CSampleDemoDlg)
	//}}AFX_EVENTSINK_MAP
END_EVENTSINK_MAP()

void CSampleDemoDlg::OnOLEStartDragMschart1(LPDISPATCH FAR* Data, long FAR* AllowedEffects) 
{
	// TODO: Add your control notification handler code here
	
}

CString InputImagePath;
CString OutputImagePath;

LPVOID lpImage = new BYTE[SizeCDib];

BYTE *lpData;
int Width, Height, Size, r;
BYTE *lpGrad, *lpTemp, *lpOrient, *lpSeg;


void CSampleDemoDlg::OnButtonOpen() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE,"bmp",".bmp",OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"Bitmap (*.BMP)|*.BMP||");
	HBITMAP m_hImage;

	if (dlg.DoModal() != IDOK) return;

		InputImagePath = dlg.GetPathName();				
		
		m_hImage = (HBITMAP)LoadImage(AfxGetInstanceHandle(),
								dlg.GetPathName(),
								IMAGE_BITMAP,
								0,
								0,
								LR_LOADFROMFILE|LR_CREATEDIBSECTION);
		GetDlgItem(IDC_STATIC_IMG)->SendMessage(STM_SETIMAGE,IMAGE_BITMAP, (LPARAM)m_hImage);	
	

	FPA_CreateImage(lpImage);
	FPA_LoadImage(InputImagePath, lpImage, &lpData, &Width, &Height);	
	Size = Width * Height;
	lpGrad = new BYTE[Size];
	lpTemp = new BYTE[Size];
	lpOrient = new BYTE[Size];
	lpSeg = new BYTE[Size];
	r = 6;
}

void CSampleDemoDlg::OnButtonSaveAs() 
{
	// TODO: Add your control notification handler code here
	CFileDialog dlg(TRUE,"bmp",".bmp",OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,"Bitmap (*.BMP)|*.BMP||");
	HBITMAP m_hImage;
	
	if (dlg.DoModal() != IDOK) return;
		
		OutputImagePath = dlg.GetPathName();				

		FPA_SaveImage(OutputImagePath, lpImage);
		
		m_hImage = (HBITMAP)LoadImage(AfxGetInstanceHandle(),
								OutputImagePath,
								IMAGE_BITMAP,
								0,
								0,
								LR_LOADFROMFILE|LR_CREATEDIBSECTION);
		GetDlgItem(IDC_STATIC_IMG)->SendMessage(STM_SETIMAGE,IMAGE_BITMAP, (LPARAM)m_hImage);

	FPA_CloseImage(lpImage);
}




void CSampleDemoDlg::Process(CommandType command)
{
	BYTE *lpTmp = new BYTE[Width * Height];
	memcpy(lpTmp, lpData, Width * Height);
	
	switch (command)
	{
	case Smooth:
		FPA_Smooth(lpTmp, lpData, Width, Height, 0);		
		break;
	case Converge:
		FPA_GaussSmooth(lpTmp, lpData, Width, Height, 0.4);
		break;
	case Grad:
		FPA_Grads(lpTmp, lpData, Width, Height, 6);			
		return;
		break;
	case Orient:
		FPA_OrientationFields(lpTmp, lpData, Width, Height, 6);
		return;
		break;
	case Enhance:
		FPA_Enhance_Gabor(lpTmp, lpData, lpOrient, Width, Height);
		memcpy(lpTmp, lpData, Size);
		FPA_Enhance_Gabor(lpTmp, lpData, lpOrient, Width, Height);
		break;
	case Segmentation:		
		FPA_Segment(lpGrad, lpSeg, 6, 80, Width, Height);		
		FPA_Segment_Clear(lpTmp, lpData, lpOrient, lpSeg, Width, Height);
		break;
	case Binary:
		FPA_Binary(lpTmp, lpOrient, lpData, Width, Height);
		memcpy(lpTmp, lpData, Size);
		FPA_Binary_Clean(lpTmp, lpData, lpSeg, Width, Height);
		break;
	case Thin:
		FPA_Thin(lpTmp, lpData, Width, Height);
		memcpy(lpTmp, lpData, Size);
		FPA_Thin_Clean(lpTmp, lpData, 12, Width, Height);
		break;
	case Equalization:
		FPA_Equalize(lpTmp, lpData, Width, Height);
		break;
	}

	
	FPA_SaveImage("tmp.bmp", lpImage);

	HBITMAP m_hImage = (HBITMAP)LoadImage(AfxGetInstanceHandle(),
								"tmp.bmp",
								IMAGE_BITMAP,
								0,
								0,
								LR_LOADFROMFILE|LR_CREATEDIBSECTION);

	GetDlgItem(IDC_STATIC_IMG)->SendMessage(STM_SETIMAGE,IMAGE_BITMAP, (LPARAM)m_hImage);
}

void CSampleDemoDlg::OnBUTTONSmooth()
{	
	Process(Smooth);
}

void CSampleDemoDlg::OnButtonConverge() 
{
	Process(Converge);
}

void CSampleDemoDlg::OnButtonGrad() 
{	
	memcpy(lpTemp, lpData, Size);
	BYTE *lpZoom = new BYTE[Size];
	FPA_ZoomOut(lpData, lpZoom, Width, Height);
	FPA_Grads(lpZoom, lpData, Width, Height, r);

	memcpy(lpGrad, lpData, Size);	

	FPA_SaveImage("grads.bmp", lpImage);

	HBITMAP m_hImage = (HBITMAP)LoadImage(AfxGetInstanceHandle(),
								"grads.bmp",
								IMAGE_BITMAP,
								0,
								0,
								LR_LOADFROMFILE|LR_CREATEDIBSECTION);
	GetDlgItem(IDC_STATIC_IMG_GRAD)->SendMessage(STM_SETIMAGE,IMAGE_BITMAP, (LPARAM)m_hImage);
	
	memcpy(lpData, lpTemp, Size);
		
}

void CSampleDemoDlg::OnButtonOrientation() 
{
	memcpy(lpTemp, lpData, Size);
	BYTE *lpZoom = new BYTE[Size];
	FPA_ZoomOut(lpData, lpZoom, Width, Height);
	FPA_OrientationFields(lpZoom, lpData, Width, Height, r);

	FPA_SaveImage("orient.bmp", lpImage);

	HBITMAP m_hImage = (HBITMAP)LoadImage(AfxGetInstanceHandle(),
								"orient.bmp",
								IMAGE_BITMAP,
								0,
								0,
								LR_LOADFROMFILE|LR_CREATEDIBSECTION);
	GetDlgItem(IDC_STATIC_IMG_ORIENT)->SendMessage(STM_SETIMAGE,IMAGE_BITMAP, (LPARAM)m_hImage);

	memcpy(lpOrient, lpData, Size);
	memcpy(lpData, lpTemp, Size);
}

void CSampleDemoDlg::OnButtonEnhance() 
{
	Process(Enhance);
}

void CSampleDemoDlg::OnButtonBinary() 
{
	Process(Binary);
}

void CSampleDemoDlg::OnButtonSegment() 
{
	Process(Segmentation);	
}

void CSampleDemoDlg::OnButtonThin() 
{
	Process(Thin);
}


void ShowImg(CString FilePath, int x, int y, CDC *pDC)
{
	CBitmap bmpFinger;
	HBITMAP hBitmap;
	hBitmap = (HBITMAP)::LoadImage(AfxGetInstanceHandle(),
		FilePath, IMAGE_BITMAP, 
		0, 0, LR_LOADFROMFILE|LR_CREATEDIBSECTION); //装载指纹图片并得到句柄
	
	if(!hBitmap)
	{
		//AfxMessageBox("读取指纹图像不成功!", MB_ICONEXCLAMATION);
	}else
	{
		BITMAP		bm;
		bmpFinger.Attach(hBitmap);  
		bmpFinger.GetBitmap(&bm);
		if(bm.bmBitsPixel != 8 ) //验证图片正确性
		{
			//AfxMessageBox("错误的指纹图像!", MB_ICONEXCLAMATION);
			bmpFinger.DeleteObject();
			return;
		}else //图片正确则显示图片并使能功能按钮
		{
			CDC MemDC;
			MemDC.CreateCompatibleDC(pDC);
			MemDC.SelectObject(bmpFinger);
			//	dc->StretchBlt(x,y,width,height,&MemDC,0,0,204,212,SRCCOPY);
			pDC->BitBlt(x, y, bm.bmWidth , bm.bmHeight,  &MemDC, 0, 0, SRCCOPY);
			
			
			//	Invalidate(TRUE);
		}
	}
	
}


void DrawMinutiae(int dx, int dy, int x, int y, CDC *pDc, CPen *Pen)
{
	CPen *oldPen=pDc->SelectObject(Pen);
	CRect rc;
	POINT   p, pb;	
	
	p.x = dx+x;
	p.y = dy+360-y;
	pb.x=p.x;
	pb.y=p.y;

	rc.left=p.x-3;
	rc.top=p.y-3;
	rc.bottom=p.y+3;
	rc.right=p.x+3;
	pDc->Arc(&rc,p,p);
	pDc->SelectObject(&oldPen);	
}

void DrawCore(POINT core, int x, int y, CDC *pDC)
{
	CRect rc;

	rc.left = core.x + x-5;
	rc.right = core.x + x +5;
	rc.top = (360-core.y) + y-5;
	rc.bottom = (360-core.y) + y + 5;

	pDC->Draw3dRect(&rc,RGB(255,0,0),RGB(255,0,0));

}

#define     MAX_MINUTIANUM			60			// 最大特征点数
#define		MAX_SINGULARYNUM		30

typedef struct tagMinutiae { 
  int   x;				// 横坐标
  int    y;				// 纵坐标
  int    Direction;		// 方向
  int	Triangle[3];	// 特征点为中心外接圆半径为定值的正三角形三个顶点的方向
  int    Type;			// 类型
} MINUTIA, *MINUTIAPTR; 

// 指纹特征(模板)结构
typedef struct tagFeature{
	int		MinutiaNum;					// 特征点数
	MINUTIA		MinutiaArr[MAX_MINUTIANUM];	// 特征点数组
} FEATURE, *FEATUREPTR;

void CSampleDemoDlg::OnButtonFeature() 
{
	// TODO: Add your control notification handler code here
	POINT Core[MAX_SINGULARYNUM];
	int Num_Core = 0;
	POINT Delta[MAX_SINGULARYNUM];
	int Num_Delta = 0;

	int i;

	FEATURE feature;
	FPA_GetMinutia(lpData, lpOrient, (LPVOID*)&feature, Width, Height);

	FPA_GetSingulary(lpOrient, Width, Height, &Num_Core, Core, -1);

	FPA_GetSingulary(lpOrient, Width, Height, &Num_Delta, Delta, 1);


	CString OutImagePathName = "tmp.bmp";
	CDC *myDC = m_img_feature.GetDC();

	COLORREF red = RGB(255, 0, 0);
	COLORREF green = RGB(0, 255, 0);
	COLORREF blue = RGB(0, 0, 255);
	COLORREF yellow = RGB(255, 255, 0);
	COLORREF orange = RGB(255, 102, 0);
	COLORREF purple = RGB(204, 33, 255);
	COLORREF gray = RGB(192, 192, 192);

	CBrush br;
	br.CreateSolidBrush(red);

	CDC *dc = myDC;

	dc->SelectObject(&br);

	ShowImg(OutImagePathName, 0, 0, dc);

	for(i = 0; i < Num_Core; i++)
	{
		int x = Core[i].x;
		int y = Core[i].y;
		

		CRect rc;

		rc.left = x-5;
		rc.right = x +5;
		rc.top = 360 - (y-5);
		rc.bottom = 360 - (y + 5);

		dc->Draw3dRect(&rc, red, red);

		dc->DrawFocusRect(&rc);

	}

	for(i = 0; i < Num_Delta; i++)
	{
		int x = Delta[i].x;
		int y = Delta[i].y;

		CRect rc;

		rc.left = x-5;
		rc.right = x +5;
		rc.top = 360- (y-5);
		rc.bottom = 360 - (y + 5);

		dc->Draw3dRect(&rc, yellow, yellow);			
		dc->DrawFocusRect(&rc);

		
	}

	for(i = 0; i < feature.MinutiaNum; i++)
	{
		int x = feature.MinutiaArr[i].x;
		int y = feature.MinutiaArr[i].y;

		CRect rc;

		rc.left = x-5;
		rc.right = x +5;
		rc.top = 360- (y-5);
		rc.bottom = 360 - (y + 5);

	

		if (feature.MinutiaArr[i].Type == 1)
			dc->Draw3dRect(&rc, blue, blue);
		else
			dc->Draw3dRect(&rc, green, green);
		dc->DrawFocusRect(&rc);
	}
}

void CSampleDemoDlg::OnButtonEqualization() 
{	
	Process(Equalization);
}