File: opencvalgorithms.h

package info (click to toggle)
actionaz 3.8.0-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 5,608 kB
  • ctags: 6,390
  • sloc: cpp: 44,713; ansic: 82; xml: 17; makefile: 14; sh: 10
file content (184 lines) | stat: -rw-r--r-- 6,547 bytes parent folder | download
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
/*
	Actionaz
	Copyright (C) 2008-2014 Jonathan Mercier-Ganady

	Actionaz is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	Actionaz is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program. If not, see <http://www.gnu.org/licenses/>.

	Contact : jmgr@jmgr.info
*/

/***************************************************************************
 *            FastMatchTemplate.h
 *
 *
 *  Copyright  2010  Tristen Georgiou
 *  tristen_georgiou@hotmail.com
 ****************************************************************************/

/*
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.


	How fast match template works:
	1. Both target and source image are down sampled numDownPyrs times
	2. cvMatchTemplate() function is called on shrunken images
	   (uses CCORR_NORMED algorithm)
	3. The numMaxima best locations are found
	4. For each point where a maxima was located:
	   a) Original source image is searched at point +/- searchExpansion
		  pixels in both x and y direction
	5. If match score is above matchPercentage then the location and score is
	   saved in the foundPointsList and confidencesList, respectively
	6. If findMultipleTargets is true, an attempt will be made to find up to
	   numMaxima targets
	7. (Optional) The targets can be drawn to a color version of the source
	   image using the DrawFoundTargets() function
 */
/*
  Adapted to the Actionaz project by Jonathan Mercier-Ganady
  Original files: http://opencv.willowgarage.com/wiki/FastMatchTemplate
*/

#ifndef OPENCVALGORITHMS_H
#define OPENCVALGORITHMS_H

#include "actiontools_global.h"
#include "matchingpointlist.h"

#include <QObject>
#include <QList>
#include <QImage>
#include <QVector>
#include <QFuture>
#include <QFutureWatcher>
#include <QMetaType>

namespace cv
{
	class Mat;
}

namespace ActionTools
{
	class ACTIONTOOLSSHARED_EXPORT OpenCVAlgorithms : public QObject
	{
		Q_OBJECT

	public:
		enum AlgorithmError
		{
			NoError,
			AlreadyRunningError,
			SourceImageSmallerThanTargerImageError,
			NotSameDepthError,
			NotSameChannelCountError,
			OpenCVException
		};

        enum AlgorithmMethod
        {
            CorrelationCoefficientMethod,
            CrossCorrelationMethod,
            SquaredDifferenceMethod
        };

		explicit OpenCVAlgorithms(QObject *parent = 0);

        bool findSubImageAsync(const QList<QImage> &sources,
						  const QImage &target,
						  int matchPercentage = 70,
						  int maximumMatches = 10,
						  int downPyrs = 2,
                          int searchExpansion = 15,
                          AlgorithmMethod method = CorrelationCoefficientMethod);
        bool findSubImage(const QList<QImage> &sources,
						  const QImage &target,
						  MatchingPointList &matchingPoints,
						  int matchPercentage = 70,
						  int maximumMatches = 10,
						  int downPyrs = 2,
                          int searchExpansion = 15,
                          AlgorithmMethod method = CorrelationCoefficientMethod);
        void cancelSearch();

		AlgorithmError error() const { return mError; }
		const QString &errorString() const { return mErrorString; }

	signals:
		void finished(const ActionTools::MatchingPointList &matchingPointList);

	private slots:
		void finished();

	private:
        bool checkInputImages(const QList<cv::Mat> &sources, const cv::Mat &target);

		/*=============================================================================
		  FastMatchTemplate
		  Performs a fast match template
		  Returns: true on success, false on failure
		  Parameters:
			source - source image (where we are searching)
			target - target image (what we are searching for)
			foundPointsList - contains a list of the points where the target was found
			confidencesList - contains a list of the confidence value (0-100) for each
							  found target
			matchPercentage - the minimum required match score to consider the target
							  found
			numMaxima - the maximum number of search locations to try before exiting
						(i.e. when image is down-sampled and searched, we collect the
						best numMaxima locations - those with the highest confidence -
						and search the original image at these locations)
			numDownPyrs - the number of times to down-sample the image (only increase
						  this number if your images are really large)
			searchExpansion - The original source image is searched at the top locations
							  with +/- searchExpansion pixels in both the x and y
							  directions
		*/
        MatchingPointList fastMatchTemplate(const QList<cv::Mat> &sources,
                                            const cv::Mat &target,
                                            int matchPercentage,
                                            int maximumMatches,
                                            int downPyrs,
                                            int searchExpansion,
                                            AlgorithmMethod method);

        static QVector<QPoint> multipleMinMaxLoc(const cv::Mat &image, int maximumMatches, AlgorithmMethod method);

        static QImage toQImage(const cv::Mat &image);
        static cv::Mat toCVMat(const QImage &image);
        static int toOpenCVMethod(AlgorithmMethod method);

		AlgorithmError mError;
		QString mErrorString;
		QFuture<MatchingPointList> mFuture;
		QFutureWatcher<MatchingPointList> mFutureWatcher;
	};
}

Q_DECLARE_METATYPE(ActionTools::MatchingPointList)

#endif // OPENCVALGORITHMS_H