// FingerAnalyserDlg.cpp : implementation file
//#pragma once
#include "stdafx.h"
#include "FingerAnalyser.h"
#include "FingerAnalyserDlg.h"
#include "math.h"
#include <list>
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// CFingerAnalyserDlg dialog
CFingerAnalyserDlg::CFingerAnalyserDlg(CWnd* pParent /*=NULL*/)
: CDialog(CFingerAnalyserDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
// Инициализация переменных
m_WorkFile = _T(""); // рабочий файл
m_ShowBase = FALSE; // Просмотр выключен
mouseX = 0; // Координаты точки
mouseY = 0;
m_DotsCount = 0; // Количество точек
m_ScanTime = 0; // Время анализа
picture = NULL;
}
// Деструктор
CFingerAnalyserDlg::~CFingerAnalyserDlg()
{
delete(fp);
delete(picture);
if(compareResult)
{
for(list<CCompareFing>::iterator i = compareResult->begin();i != compareResult->end();i++)
{
list<CPairSur>::iterator j;
for(j=i->surdots.begin(); j!=i->surdots.end(); j++)
{
j->first->clear();
delete(j->first);
j->second->clear();
delete(j->second);
}
}
compareResult->clear();
delete(compareResult);
compareResult = NULL;
}
}
void CFingerAnalyserDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
DDX_Check(pDX, IDC_SHOW, m_ShowBase);
DDX_Control(pDX, IDC_PROGRESS_COMPARE, m_ProgressCompare);
DDX_Control(pDX, IDC_PROGRESS_SAVE, m_ProgressSave);
DDX_Text(pDX, IDC_POINTS, m_DotsCount);
DDX_Text(pDX, IDC_TIME, m_ScanTime);
DDX_Text(pDX, IDC_FILE, m_WorkFile);
}
BEGIN_MESSAGE_MAP(CFingerAnalyserDlg, CDialog)
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_BN_CLICKED(IDC_OPEN, &CFingerAnalyserDlg::OnBnClickedOpen)
ON_BN_CLICKED(IDC_ANALYSE, &CFingerAnalyserDlg::OnBnClickedAnalyse)
ON_BN_CLICKED(IDC_COMPARE, &CFingerAnalyserDlg::OnBnClickedCompare)
ON_BN_CLICKED(IDC_SAVE, &CFingerAnalyserDlg::OnBnClickedSave)
ON_BN_CLICKED(IDC_BACK, &CFingerAnalyserDlg::OnBnClickedBack)
ON_BN_CLICKED(IDC_NEXT, &CFingerAnalyserDlg::OnBnClickedNext)
ON_BN_CLICKED(IDC_SHOW,&CFingerAnalyserDlg::OnBnClickedShowBase)
ON_WM_LBUTTONDOWN()
ON_WM_TIMER()
END_MESSAGE_MAP()
// CFingerAnalyserDlg message handlers
BOOL CFingerAnalyserDlg::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
fp = new CFingPicture(this->GetDC());
char fullpath[200];
_fullpath(fullpath, NULL, 200);
pathToSAV = fullpath;
pathToSAV += "\\sav\\";
dbFile = pathToSAV + "fingbase.bse";
compareResult = NULL;
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 CFingerAnalyserDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<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
{
CPaintDC dc(this);
if(m_ShowBase)
{
ShowBase(true,false);
}
else
{//режим просмотра открытого образа
if (picture != NULL)
{
picture->GetPic1()->Show(140, 70);
picture->GetPic2()->Show(545, 70);
}
m_DotsCount = (int)fingA.size();
UpdateData(false);
}
CDialog::OnPaint();
}
}
// The system calls this function to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CFingerAnalyserDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
// Открытие файла
void CFingerAnalyserDlg::OnBnClickedOpen()
{
char szFilters[]= "Образы (*.bmp)|*.bmp|All Files (*.*)|*.*||";
CFileDialog dlg(TRUE, "bmp", "*.bmp", OFN_FILEMUSTEXIST| OFN_HIDEREADONLY, szFilters, this);
if(dlg.DoModal() != IDOK)
return; //никаких файлов не открыли
if(dlg.GetFileExt().CompareNoCase("bmp"))
return; //открытый файл не имеет расширеня .bmp
CString fileName = dlg.GetFileName();
if(picture != NULL)
delete(picture);
picture = new CAnalysePicture(fileName, this->GetDC());
m_WorkFile = fileName;
if(compareResult)
{
for(list<CCompareFing>::iterator i = compareResult->begin(); i != compareResult->end(); i++)
{
list<CPairSur>::iterator j;
for(j=i->surdots.begin(); j!=i->surdots.end(); j++)
{
j->first->clear();
delete(j->first);
j->second->clear();
delete(j->second);
}
}
compareResult->clear();
}
m_ShowBase = false;
Invalidate();
}
// Анализ изображения
void CFingerAnalyserDlg::OnBnClickedAnalyse()
{
if(picture == NULL)
return;
LPSYSTEMTIME mTime;
mTime = new SYSTEMTIME;
GetSystemTime(mTime);
long int workTime;
workTime = mTime->wSecond*1000+mTime->wMilliseconds;
fingA = picture->AnalysePicture();
fingA.SaveFing(GetSAV(picture->getPathSrc()));
GetSystemTime(mTime);
workTime = mTime->wSecond*1000+mTime->wMilliseconds - workTime;
workTime = (workTime<0)?60000+workTime:workTime;
delete(mTime);
m_ScanTime = workTime;
Invalidate();
}
// Сравнение
void CFingerAnalyserDlg::OnBnClickedCompare()
{
if(fingA.size() == 0)
{
MessageBox("Отпечаток не обработан", "Ошибка");
return;
}
fingR.Convert(fingA);
if(compareResult)
{
for(list<CCompareFing>::iterator i = compareResult->begin(); i != compareResult->end(); i++)
{
list<CPairSur>::iterator j;
for(j=i->surdots.begin(); j!=i->surdots.end(); j++)
{
j->first->clear(); delete(j->first);
j->second->clear(); delete(j->second);
}
}
compareResult->clear();
delete(compareResult);
compareResult = NULL;
showIter = NULL;
}
compareResult = CompareWithBase();
if(compareResult->size() == 0)
return;
CString sOut="";
for(list<CCompareFing>::iterator i = compareResult->begin(); i != compareResult->end(); i++)
{
CString s="";
int mlevel = min(i->nfng,m_DotsCount);
int percent;
if(mlevel > 10)
mlevel = 10;
if(i->cDot > mlevel)
percent = 100;
else
percent = (int)(100.0*i->cDot/(double)mlevel + 0.5);
if(percent == 0)
continue;
s.Format("%d %d %% %s\n", i->cDot, percent, i->name);
sOut += s;
}
if(sOut.GetLength()==0)
sOut = "Ни одного отпечатка не найдено!\n";
CString kol;
kol.Format("\n Всего в базе: %d", compareResult->size());
sOut += kol;
PrintReport(picture->getPathSrc(), sOut);
MessageBox(sOut, picture->getPathSrc());
}
// Сохранить в БД
void CFingerAnalyserDlg::OnBnClickedSave()
{
char szFilters[] = "Образы (*.bmp)|*.bmp|All Files (*.*)|*.*||";
CFileDialog dlg( TRUE, "bmp", "*.bmp", OFN_FILEMUSTEXIST| OFN_HIDEREADONLY| OFN_ALLOWMULTISELECT, szFilters, this);
if(dlg.DoModal() == IDOK)
{
listCInfo *fingDB = LoadDB(dbFile);
FILE *fbse = fopen(dbFile, "wb");
if(fbse == NULL)
{
MessageBox("Невозможно создать базу данных с отпечатками", "Ошибка создания БД", MB_OK);
return;
}
POSITION pos, posStart;
CInfo newFingInDB;
pos = posStart = dlg.GetStartPosition();
int kolFile = 0;
while(pos)
{
dlg.GetNextPathName(pos);
kolFile++;
}
pos = posStart;
m_ProgressSave.SetRange(0, kolFile);
int progressPos = 1;
while(pos)
{
CString fileName = dlg.GetNextPathName(pos).MakeLower();
if(fileName.Find(".bmp") == -1)
continue;
CAnalysePicture *loadingPic;
loadingPic = new CAnalysePicture(fileName, this->GetDC());
m_WorkFile = fileName;
fingA = loadingPic->AnalysePicture();
if(fingA.size() < MIN_SIZE)
{
MessageBox("Отпечаток не пригоден для сохраниения в базу!", fileName);
continue;
}
if(fingA.size() > MAX_SIZE)
{
MessageBox("Отпечаток не пригоден для сохраниения в базу!", fileName);
continue;
}
fingA.SaveFing(GetSAV(fileName));
newFingInDB.src = fileName;
fingDB->remove(newFingInDB);
fingDB->push_back(newFingInDB);
m_ProgressSave.SetPos(progressPos);
progressPos++;
Invalidate();
delete(loadingPic);
}
m_ProgressSave.SetPos(0);
int count = 0;
fwrite((void*)&count, sizeof(count), 1, fbse);
for(list<CInfo>::iterator iter = fingDB->begin(); iter != fingDB->end(); iter++)
{
iter->Printf(fbse);
count++;
}
fseek(fbse, 0, SEEK_SET);
fwrite((void*)&count, sizeof(count), 1, fbse);
fingDB->clear();
delete(fingDB);
fclose(fbse);
}
}
void CFingerAnalyserDlg::OnBnClickedBack()
{
ShowBase(false);
}
void CFingerAnalyserDlg::OnBnClickedNext()
{
ShowBase(true);
}
// Нажатие кнопки мыши над формой
void CFingerAnalyserDlg::OnLButtonDown(UINT nFlags, CPoint point)
{
if(!m_ShowBase)
return;
mousePos = point;
mousePos.x -= 110;
mousePos.y -= 45;
ShowBase(true, false);
CDialog::OnLButtonDown(nFlags, point);
}
// обновлять при событии от таймера
void CFingerAnalyserDlg::OnTimer(UINT_PTR nIDEvent)
{
Invalidate();
CDialog::OnTimer(nIDEvent);
}
void CFingerAnalyserDlg::OnBnClickedShowBase()
{
m_ShowBase =! m_ShowBase;
UpdateData(false);
if(m_ShowBase)
{
ShowBase(true, false);
}
else
{
OnPaint();
}
}
//key - направление перемотки по базе (влево, вправо)
//next - нужно ли переходить к следующему отпечатку
void CFingerAnalyserDlg::ShowBase(bool key, bool next)
{
if(!compareResult)
return;
if(compareResult->size() == 0)
{
MessageBox("База данных отпечатков пуста", "Сообщение", MB_OK);
return;
}
if (showIter == NULL)
showIter = compareResult->begin();
else
{
if(next)
if(key)
{
showIter++;
if(showIter == compareResult->end())
showIter = compareResult->begin();
}
else
{
if(showIter == compareResult->begin())
showIter = compareResult->end();
showIter--;
}
}
CFingPicture *pic;
pic = new CFingPicture(this->GetDC());
if(!pic->Load(BLANK))
return;
CPaintDC dc(this); // device context for painting
list<CPairSur>::iterator is = showIter->surdots.begin();
list<CPairAbsDot>::iterator id = showIter->dots.begin();
for(; id != showIter->dots.end(); id++, is++)
{
COLORREF col;
if(is->first->empty())
col = 0xBBBBBB;
else
col = (id->first.type)?0xff0000:0x000000;
pic->Line(id->first.coord, id->first.coord, 5, col);
pic->Line(id->first.coord,
CPoint(id->first.coord.x+(int)(10.0*cos(id->first.alpha)),id->first.coord.y-(int)(10.0*sin(id->first.alpha))),2, col);
if(is->first->empty())
continue; //окружения для этой точки нет
//проверка, что "мышь" находится над точкой
if( abs(mousePos.x-id->first.coord.x)<6 && abs(mousePos.y-id->first.coord.y)<6 )
{
CFingPicture pic2(this->GetDC());
if(!pic2.Load(BLANK))
return;
pic2.Copy(*picture->GetPic2());
for(listCRelDot::iterator ii = is->first->begin(); ii != is->first->end(); ii++)
{
COLORREF cl = 0x554444;
CPoint cd;
cd.x = (long)(id->first.coord.x - ii->l * cos(ii->a1*M_PI/180.0 - id->first.alpha));
cd.y = (long)(id->first.coord.y - ii->l * sin(ii->a1*M_PI/180.0 - id->first.alpha));
pic->Line(id->first.coord, cd, 1, cl);
}
for(listCRelDot::iterator ii = is->second->begin(); ii != is->second->end(); ii++)
{
COLORREF cl = 0x554444;
CPoint cd;
cd.x = (long)(id->second.coord.x - ii->l * cos(ii->a1*M_PI/180.0 - id->second.alpha));
cd.y = (long)(id->second.coord.y - ii->l * sin(ii->a1*M_PI/180.0 - id->second.alpha));
pic2.Line(id->second.coord, cd, 1, cl);
}
pic2.Show(545, 45);
}
}
if (pic != NULL)
{
pic->Show(110, 45);
m_WorkFile = showIter->name;
}
UpdateData(false);
delete(pic);
}
void CFingerAnalyserDlg::PrintReport(CString file, CString report)
{
FILE *outf = fopen("report.txt", "a");
CString msg = "\n------ "+file+" ------\n"+report;
fprintf(outf, msg);
fclose(outf);
}
CString CFingerAnalyserDlg::GetSAV(CString srcName)
{
CString fsav = srcName.Left(srcName.GetLength() - 3) + "sav";
while(fsav.Find("\\") != -1){ fsav = fsav.Right(fsav.GetLength() - fsav.Find("\\")-1); }
return pathToSAV + fsav;
}
//загрузить точки из БД
listCInfo *CFingerAnalyserDlg::LoadDB(CString dbFile)
{
listCInfo *bse = new listCInfo();
CInfo finf; //данные по отпечатку
FILE *fbse = fopen(dbFile, "rb");
if(fbse == NULL)
{
// MessageBox("Невозможно загрузить базу данных с отпечатками", "Ошибка загрузки БД", MB_OK);
return bse;
}
int count = 0;
fread((void*)&count, sizeof(count), 1, fbse);
for(;count > 0; count--)
{
finf.Scanf(fbse);
bse->push_back(finf);
}
fclose(fbse);
return bse;
}
//сравнить точку с точками в БД
list<CCompareFing> *CFingerAnalyserDlg::CompareWithBase()
{
listCInfo *bse;
list<CCompareFing> *cFng;
cFng = new list<CCompareFing>;
bse = LoadDB(dbFile);
if(bse->empty())
{
MessageBox("База данных отпечатков пуста", "Сообщение", MB_OK);
return cFng;
}
CAbsFing aFng;
CRelFing baseFng;
m_ProgressCompare.SetRange(0, (short)bse->size());
for(list<CInfo>::iterator ibse = bse->begin();ibse != bse->end(); ibse++)
{
if(!aFng.LoadFing(GetSAV(ibse->src)))
continue;
baseFng.Convert(aFng);
CCompareFing compareRes = fingR.Compare(baseFng);
compareRes.name = ibse->src;
cFng->push_back(compareRes);
m_ProgressCompare.SetPos((int)cFng->size());
}
bse->clear();
m_ProgressCompare.SetPos(0);
delete(bse);
return cFng;
}