qr-code-reader/QRCode.cpp at master · matheusrmorgado/qr-code-reader · GitHub

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
#include <cekeikon.h>

//double time1; double time2; double time3; double time4;
double minVal; double maxVal;
Point minLoc; Point maxLoc; Point matchLoc;

Mat_<COR> qrCode;
Mat_<FLT> image;
Mat_<FLT> templ;
Mat_<FLT> result;

int achaTamanho( Mat_<FLT> image, Mat_<FLT> templ, Mat_<FLT> result ){

	int tamanho; float aux = 0.0;

	for ( int s = 30; s < 120; s++){
		resize( templ, templ, Size(s,s), 0, 0, INTER_AREA );
		templ = somaAbsDois( dcReject( templ ) );
		matchTemplate( image, templ, result, CV_TM_CCORR );
		minMaxLoc( result, &minVal, &maxVal, &minLoc, &maxLoc );
		if ( maxVal > aux){
			aux = maxVal;
			tamanho = s;
		}
	}
	return tamanho;
}

Mat_<FLT> pinta( Mat_<FLT> a, int ls, int cs ) {

	Mat_<FLT> b = a.clone();
	queue<int> q;
	q.push( ls ); q.push( cs );

	for( int i = 0; i < 150; i++ ) {
		int l = q.front(); q.pop();
		int c = q.front(); q.pop();
		b(l,c) = 0;
		q.push( l-1 ); q.push( c ); //N
		q.push( l+1 ); q.push( c ); //S
		q.push( l ); q.push( c+1 ); //E
		q.push( l ); q.push( c-1 ); //W
	}
	return b;
}

void localizaQR( Mat_<COR> qrCode, Mat_<FLT> templ, Mat_<FLT> result ){

	string str;

	for ( int i = 0; i < 3; i++ ){
		minMaxLoc( result, &minVal, &maxVal, &minLoc, &maxLoc );
		matchLoc = maxLoc;

		line( qrCode, Point( matchLoc.x + templ.cols/2 , matchLoc.y + templ.rows/2 + 0.12*templ.rows ),
			Point( matchLoc.x + templ.cols/2 , matchLoc.y + templ.rows/2 - 0.12*templ.rows ), Scalar(0,0,255), 1, 8);
		line( qrCode, Point( matchLoc.x + templ.cols/2 + 0.12*templ.rows , matchLoc.y + templ.rows/2 ),
			Point( matchLoc.x + templ.cols/2 - 0.12*templ.rows , matchLoc.y + templ.rows/2 ), Scalar(0,0,255), 1, 8);
		rectangle( qrCode, matchLoc, Point( matchLoc.x + templ.cols , matchLoc.y + templ.rows ), Scalar(0,0,255), 1.5, 8);

		str = to_string(i);
		putTxt( qrCode, matchLoc.y + templ.rows/2 + 2, matchLoc.x + templ.cols/2 + 2,
			str, COR(0,255,0) , 1, true, COR(0,0,0));

		str = to_string(maxVal);
		putTxt( qrCode, matchLoc.y + templ.rows/2 + 35, matchLoc.x + templ.cols/2 + 2,
			str, COR(255,0,0) , 1, true, COR(0,0,0));

		//time1 = (double)getTickCount();
		result = pinta( result, matchLoc.y, matchLoc.x );
		//time1 = ((double)getTickCount() - time1)/getTickFrequency();
		//cout << " Tempo pinta" << i << ": " << time1 << endl;
		cout << " MatchLoc" << i << ": " << maxVal << endl;
	}

}

int main( int argc, char* argv[] ){

	//time4 = (double)getTickCount();
	le( qrCode, argv[1] );
	le( image, argv[1] );
	le( templ, "padrao.png" );

	int result_cols =  image.cols - templ.cols + 1;
	int result_rows = image.rows - templ.rows + 1;
	result.create( result_rows, result_cols );

	//time2 = (double)getTickCount();
	int s = achaTamanho(image,templ,result);
	//time2 = ((double)getTickCount() - time2)/getTickFrequency();
	//cout << " Tempo achaTamanho: " << time2 << endl;
	resize( templ, templ, Size(s,s), 0, 0, INTER_AREA );
	templ = somaAbsDois( dcReject( templ ) );
	matchTemplate( image, templ, result, CV_TM_CCORR );

	mostra( result );

	//time3 = (double)getTickCount();
	localizaQR( qrCode, templ, result );
	//time3 = ((double)getTickCount() - time3)/getTickFrequency();
	//cout << " Tempo localizaQR: " << time3 << endl;

	mostra( qrCode );
	imp( qrCode, argv[2] );
	cout << "\n Imagem " << argv[2] << " gerada com sucesso!" << endl;
	//time4 = ((double)getTickCount() - time4)/getTickFrequency();
	//cout << " Tempo total: " << time4 << endl;
}