-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathmainwindow.h
executable file
·194 lines (165 loc) · 6.89 KB
/
mainwindow.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
#ifndef MAINWINDOW_H
#define MAINWINDOW_H
#include <QMainWindow>
#include <QFileDialog>
#include <QDir>
#include <QStringList>
#include <QListWidgetItem>
#include <QMouseEvent>
#include <QGraphicsItem>
#include <QGraphicsRectItem>
#include <QGraphicsPixmapItem>
#include <QGraphicsView>
#include <QGraphicsPathItem>
#include <QPainterPath>
#include <QGraphicsSceneMouseEvent>
#include <QDebug>
#include <QRectF>
#include <QImage>
#include <QPainter>
#include <QStringList>
#include <QVector>
#include <QObject>
#include <QListWidget>
#include <QDrag>
#include <QPoint>
#include <QTransform>
#include <QThread>
#include <QMessageBox>
//#include <QWinTaskbarButton>
//#include <QWinTaskbarProgress>
#include <QShowEvent>
#include <QDirModel>
#include <QCustomPlot/qcustomplot.h>
#include <QGraphicsTextItem>
#undef ERROR
#include "graphics_view_zoom.h"
#include "minutiaemarker.h"
#include "networktrainer.h"
#include "minutiaechecker.h"
#include "extractiontester.h"
#include "databasetester.h"
#include "mousefingerprintscene.h"
namespace Ui {
class MainWindow;
}
class MainWindow : public QMainWindow
{
Q_OBJECT
public:
explicit MainWindow(QWidget *parent = 0);
~MainWindow();
private slots:
void on_pushButton_inputDirectory_clicked();
void on_listWidget_inputImages_itemClicked(QListWidgetItem *item);
void on_radioButton_ending_clicked();
void on_radioButton_bifurcation_clicked();
void on_radioButton_nothing_clicked();
void updateMinutiaeMarker(QString action);
void generateMinutiaeHeatmap();
void tableWidgetCellSelected(int row, int col);
void on_pushButton_marker_outputDirectory_clicked();
void on_pushButton_marker_saveBlocks_clicked();
void on_pushButton_netTrainer_input_clicked();
void on_pushButton_netTrainer_startTraining_clicked();
void on_pushButton_netTrainer_outputFolder_clicked();
void on_pushButton_inputDirectory_2_clicked();
void on_listWidget_inputImages_2_itemClicked(QListWidgetItem *item);
void on_tabWidget_main_currentChanged(int index);
void on_radioButton_standard_clicked();
void on_radioButton_heatmap_clicked();
void on_pushButton_inputModel_clicked();
void showPredictedMinutia(QPoint xy, const QVector<QPair<QString, float> > predictedData);
void predictMinutia(QPoint xy);
void updateProgressBar(QString barName, int value, QString text = "");
void showHeatmap(QImage &heatMap);
void drawCurrentRectanglePosition(QPoint xy);
void addLog(QString logFieldName, QString text);
void clearLog(QString logFieldName);
void drawGraph(DBTEST_RESULT results);
void markerBlocksSaved();
void questionMessageBox(QString question, QString subject);
void warningMessageBox(QString boxType, QString text, QString subject);
void on_pushButton_inputDatabase_clicked();
void on_pushButton_startTesting_clicked();
void on_tabWidget_testerSettings_currentChanged(int index);
void on_doubleSpinBox_marker_blur_valueChanged(double arg1);
void on_doubleSpinBox_marker_irisBlur_blur_valueChanged(double arg1);
void on_pushButton_exTester_inputDir_clicked();
void on_listWidget_exTester_inputImages_itemClicked(QListWidgetItem *item);
void showPreprocessingTestResults(cv::Mat imgOriginal, cv::Mat imgOMap, cv::Mat imgEnhanced, cv::Mat imgBinarized, cv::Mat imgSkeleton, cv::Mat imgSkeletonInv, cv::Mat imgMask, cv::Mat imgQMap, cv::Mat imgFMap);
void showExtractionTestResults(const cv::Mat &imgSkeleton, const QVector<MINUTIA> &crossingNumber, const QVector<MINUTIA> &fixedOrientations, const QVector<MINUTIA> &checkedOrientations);
void on_checkBox_exTester_invertedSkeleton_clicked(bool checked);
void on_comboBox_exTester_map_currentTextChanged(const QString &arg1);
void on_checkBox_exTester_contrast_clicked();
void on_spinBox_exTester_holeSize_valueChanged(int arg1);
void on_checkBox_exTester_holeRemover_clicked();
void on_checkBox_exTester_mask_clicked();
void on_checkBox_exTester_qualityMap_clicked();
void on_spinBox_exTester_threadNum_valueChanged(int arg1);
void on_checkBox_exTester_orientationFixer_clicked();
void on_checkBox_exTester_advancedOMap_clicked();
void on_radioButton_exTester_crossingNumber_clicked(bool checked);
void on_radioButton_exTester_fixedPredictedMinutiae_clicked(bool checked);
void on_radioButton_exTester_predictedMinutiae_clicked(bool checked);
void on_checkBox_exTester_varBlockSize_clicked(bool checked);
void on_tabWidget_exTester_settings_currentChanged(int index);
void on_doubleSpinBox_marker_irisBlur_radius_valueChanged(double arg1);
private:
Ui::MainWindow *ui;
//QWinTaskbarButton *winTaskBarButton;
//QWinTaskbarProgress *winTaskBarProgress;
QGraphicsScene *sceneTestOriginal;
QGraphicsScene *sceneTestMask;
QGraphicsScene *sceneTestMap;
QGraphicsScene *sceneTestEnhanced;
QGraphicsScene *sceneTestBinarized;
QGraphicsScene *sceneTestSkeleton;
QGraphicsScene *sceneTestExtraction;
QGraphicsScene *sceneTestBig;
Graphics_view_zoom *viewTestOriginal;
Graphics_view_zoom *viewTestMask;
Graphics_view_zoom *viewTestMap;
Graphics_view_zoom *viewTestEnhanced;
Graphics_view_zoom *viewTestBinarized;
Graphics_view_zoom *viewTestSkeleton;
Graphics_view_zoom *viewTestExtraction;
Graphics_view_zoom *viewTestBig;
Graphics_view_zoom *fpViewMarker;
Graphics_view_zoom *fpViewChecker;
MouseFingerprintScene *fpScene;
QCPItemText *eerLabel;
QGraphicsScene *sceneBlockOrig;
QGraphicsScene *sceneBlockBlur;
QGraphicsScene *sceneBlockIrisBlur;
MinutiaeMarker *minMarkerTh;
NetworkTrainer *netTrainerTh;
MinutiaeChecker *minCheckerTh;
ExtractionTester *exTesterTh;
DatabaseTester *dbTesterTh;
QString previousImgName;
QString actualImgName;
QImage fpImg;
QGraphicsRectItem *currentBlock;
QDirModel *trainerDirModel;
PLOT_PARAMS plotParams;
//void showEvent(QShowEvent *e);
void startMinutiaeMarker();
void startNetworkTrainer();
void startMinutiaeChecker();
void startExtractionTester();
void startDbTester();
void runExtractionTester();
void updateBlockPreviews(QPoint xy);
void drawCircles(const MINUTIA &minutia, bool difference);
signals:
void predictHeatmapSignal(QImage, int, bool);
void generateBlocksSignal(int, int, QString, bool, bool, double, bool, double, double);
void predictMinutiaSignal(QImage, QPoint, int, bool);
void updateMinutiaeSignal(QString, QString);
void questionBoxAnswerSignal(bool answer);
void startTrainSignal(bool rotations, bool blur, bool irisblur, int, int);
void startDBTestingSignal(MATCHER matcher, int numOfSubject, int imgPerSubject);
void startExTestingSignal();
};
#endif // MAINWINDOW_H