GDAL

gdal_simplesurf.h
Go to the documentation of this file.
1 /******************************************************************************
2 * $Id$
3 * Project: GDAL
4 * Purpose: Correlator
5 * Author: Andrew Migal, migal.drew@gmail.com
6 *
7 ******************************************************************************
8 * Copyright (c) 2012, Andrew Migal
9 *
10 * Permission is hereby granted, free of charge, to any person obtaining a
11 * copy of this software and associated documentation files (the "Software"),
12 * to deal in the Software without restriction, including without limitation
13 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
14 * and/or sell copies of the Software, and to permit persons to whom the
15 * Software is furnished to do so, subject to the following conditions:
16 *
17 * The above copyright notice and this permission notice shall be included
18 * in all copies or substantial portions of the Software.
19 *
20 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
21 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
23 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
24 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
25 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
26 * DEALINGS IN THE SOFTWARE.
27 ****************************************************************************/
28
35 #ifndef GDALSIMPLESURF_H_
36 #define GDALSIMPLESURF_H_
37
38 #include "gdal_priv.h"
39 #include "cpl_conv.h"
40 #include <list>
41
50 class GDALFeaturePoint
51 {
52 public:
57 GDALFeaturePoint();
58
63 GDALFeaturePoint(const GDALFeaturePoint& fp);
64
77 GDALFeaturePoint(int nX, int nY, int nScale, int nRadius, int nSign);
78 virtual ~GDALFeaturePoint();
79
81 GDALFeaturePoint& operator=(const GDALFeaturePoint& point);
82
92 double& operator[](int nIndex);
93
95 static const int DESC_SIZE = 64;
96
102 int GetX() const;
103
109 void SetX(int nX);
110
116 int GetY() const;
117
123 void SetY(int nY);
124
130 int GetScale() const ;
131
137 void SetScale(int nScale);
138
144 int GetRadius() const;
145
151 void SetRadius(int nRadius);
152
158 int GetSign() const;
159
165 void SetSign(int nSign);
166
167 private:
168 // Coordinates of point in image
169 int nX;
170 int nY;
171 // --------------------
172 int nScale;
173 int nRadius;
174 int nSign;
175 // Descriptor array
176 double *padfDescriptor;
177 };
178
188 class GDALIntegralImage
189 {
190 CPL_DISALLOW_COPY_ASSIGN(GDALIntegralImage)
191
192 public:
193 GDALIntegralImage();
194 virtual ~GDALIntegralImage();
195
203 void Initialize(const double **padfImg, int nHeight, int nWidth);
204
213 double GetValue(int nRow, int nCol);
214
226 double GetRectangleSum(int nRow, int nCol, int nWidth, int nHeight);
227
237 double HaarWavelet_X(int nRow, int nCol, int nSize);
238
248 double HaarWavelet_Y(int nRow, int nCol, int nSize);
249
255 int GetHeight();
256
262 int GetWidth();
263
264 private:
265 double **pMatrix = nullptr;
266 int nWidth = 0;
267 int nHeight = 0;
268 };
269
278 class GDALOctaveLayer
279 {
280 CPL_DISALLOW_COPY_ASSIGN(GDALOctaveLayer)
281
282 public:
283 GDALOctaveLayer();
284
293 GDALOctaveLayer(int nOctave, int nInterval);
294 virtual ~GDALOctaveLayer();
295
305 void ComputeLayer(GDALIntegralImage *poImg);
306
310 int octaveNum;
314 int filterSize;
318 int radius;
322 int scale;
326 int width;
330 int height;
334 double **detHessians;
338 int **signs;
339 };
340
347 class GDALOctaveMap
348 {
349 CPL_DISALLOW_COPY_ASSIGN( GDALOctaveMap )
350
351 public:
358 GDALOctaveMap(int nOctaveStart, int nOctaveEnd);
359 virtual ~GDALOctaveMap();
360
367 void ComputeMap(GDALIntegralImage *poImg);
368
386 static bool PointIsExtremum(int row, int col, GDALOctaveLayer *bot,
387 GDALOctaveLayer *mid, GDALOctaveLayer *top, double threshold);
388
392 GDALOctaveLayer ***pMap;
393
397 static const int INTERVALS = 4;
398
402 int octaveStart;
403
407 int octaveEnd;
408 };
409
421 class GDALSimpleSURF
422 {
423 private:
428 class MatchedPointPairInfo
429 {
430 public:
431 MatchedPointPairInfo(int nInd_1, int nInd_2, double dfDist):
432 ind_1(nInd_1), ind_2(nInd_2), euclideanDist(dfDist) {}
433
434 int ind_1;
435 int ind_2;
436 double euclideanDist;
437 };
438
439 CPL_DISALLOW_COPY_ASSIGN( GDALSimpleSURF )
440
441 public:
463 GDALSimpleSURF(int nOctaveStart, int nOctaveEnd);
464 virtual ~GDALSimpleSURF();
465
482 static CPLErr ConvertRGBToLuminosity(
483 GDALRasterBand *red,
484 GDALRasterBand *green,
485 GDALRasterBand *blue,
486 int nXSize, int nYSize,
487 double **padfImg, int nHeight, int nWidth);
488
503 std::vector<GDALFeaturePoint>*
504 ExtractFeaturePoints(GDALIntegralImage *poImg, double dfThreshold);
505
526 static CPLErr MatchFeaturePoints(
527 std::vector<GDALFeaturePoint*> *poMatchPairs,
528 std::vector<GDALFeaturePoint> *poFirstCollect,
529 std::vector<GDALFeaturePoint> *poSecondCollect,
530 double dfThreshold);
531
532 private:
542 static double GetEuclideanDistance(
543 GDALFeaturePoint &firstPoint, GDALFeaturePoint &secondPoint);
544
550 static void NormalizeDistances(std::list<MatchedPointPairInfo> *poList);
551
558 static void SetDescriptor(GDALFeaturePoint *poPoint, GDALIntegralImage *poImg);
559
560 private:
561 int octaveStart;
562 int octaveEnd;
563 GDALOctaveMap *poOctMap;
564 };
565
566 #endif /* GDALSIMPLESURF_H_ */
GDALFeaturePoint
Class of "feature point" in raster.
Definition: gdal_simplesurf.h:51
GDALFeaturePoint::GDALFeaturePoint
GDALFeaturePoint()
Standard constructor.
Definition: gdal_simplesurf.cpp:40
GDALFeaturePoint::SetY
void SetY(int nY)
Set Y coordinate of point.
Definition: gdal_simplesurf.cpp:97
GDALFeaturePoint::operator=
GDALFeaturePoint & operator=(const GDALFeaturePoint &point)
Assignment operator.
Definition: gdal_simplesurf.cpp:71
GDALFeaturePoint::GetSign
int GetSign() const
Fetch sign of Hessian determinant of point.
Definition: gdal_simplesurf.cpp:105
GDALFeaturePoint::SetX
void SetX(int nX)
Set X coordinate of point.
Definition: gdal_simplesurf.cpp:94
GDALFeaturePoint::DESC_SIZE
static const int DESC_SIZE
Descriptor length.
Definition: gdal_simplesurf.h:95
GDALFeaturePoint::GetY
int GetY() const
Fetch Y-coordinate (line) of point.
Definition: gdal_simplesurf.cpp:96
GDALFeaturePoint::GetRadius
int GetRadius() const
Fetch radius of point.
Definition: gdal_simplesurf.cpp:102
GDALFeaturePoint::GetScale
int GetScale() const
Fetch scale of point.
Definition: gdal_simplesurf.cpp:99
GDALFeaturePoint::GetX
int GetX() const
Fetch X-coordinate (pixel) of point.
Definition: gdal_simplesurf.cpp:93
GDALFeaturePoint::SetScale
void SetScale(int nScale)
Set scale of point.
Definition: gdal_simplesurf.cpp:100
GDALFeaturePoint::SetRadius
void SetRadius(int nRadius)
Set radius of point.
Definition: gdal_simplesurf.cpp:103
GDALFeaturePoint::SetSign
void SetSign(int nSign)
Set sign of point.
Definition: gdal_simplesurf.cpp:106
GDALFeaturePoint::operator[]
double & operator[](int nIndex)
Provide access to point's descriptor.
Definition: gdal_simplesurf.cpp:108
GDALIntegralImage
Integral image class (summed area table).
Definition: gdal_simplesurf.h:189
GDALIntegralImage::GetHeight
int GetHeight()
Fetch height of integral image.
Definition: gdal_octave.cpp:40
GDALIntegralImage::HaarWavelet_Y
double HaarWavelet_Y(int nRow, int nCol, int nSize)
Get value of vertical Haar wavelet in specified square grid.
Definition: gdal_octave.cpp:139
GDALIntegralImage::GetValue
double GetValue(int nRow, int nCol)
Fetch value of specified position in integral image.
Definition: gdal_octave.cpp:86
GDALIntegralImage::HaarWavelet_X
double HaarWavelet_X(int nRow, int nCol, int nSize)
Get value of horizontal Haar wavelet in specified square grid.
Definition: gdal_octave.cpp:133
GDALIntegralImage::Initialize
void Initialize(const double **padfImg, int nHeight, int nWidth)
Compute integral image for specified array.
Definition: gdal_octave.cpp:44
GDALIntegralImage::GetRectangleSum
double GetRectangleSum(int nRow, int nCol, int nWidth, int nHeight)
Get sum of values in specified rectangular grid.
Definition: gdal_octave.cpp:94
GDALIntegralImage::GetWidth
int GetWidth()
Fetch width of integral image.
Definition: gdal_octave.cpp:42
GDALOctaveLayer
Class for computation and storage of Hessian values in SURF-based algorithm.
Definition: gdal_simplesurf.h:279
GDALOctaveLayer::height
int height
Image height in pixels.
Definition: gdal_simplesurf.h:330
GDALOctaveLayer::signs
int ** signs
Hessian signs for speeded matching.
Definition: gdal_simplesurf.h:338
GDALOctaveLayer::detHessians
double ** detHessians
Hessian values for image pixels.
Definition: gdal_simplesurf.h:334
GDALOctaveLayer::filterSize
int filterSize
Length of the side of filter.
Definition: gdal_simplesurf.h:314
GDALOctaveLayer::radius
int radius
Length of the border.
Definition: gdal_simplesurf.h:318
GDALOctaveLayer::octaveNum
int octaveNum
Octave which contains this layer (1,2,3...)
Definition: gdal_simplesurf.h:310
GDALOctaveLayer::width
int width
Image width in pixels.
Definition: gdal_simplesurf.h:326
GDALOctaveLayer::scale
int scale
Scale for this layer.
Definition: gdal_simplesurf.h:322
GDALOctaveLayer::ComputeLayer
void ComputeLayer(GDALIntegralImage *poImg)
Perform calculation of Hessian determinants and their signs for specified integral image.
Definition: gdal_octave.cpp:171
GDALOctaveMap
Class for handling octave layers in SURF-based algorithm.
Definition: gdal_simplesurf.h:348
GDALOctaveMap::INTERVALS
static const int INTERVALS
Value for constructing internal octave space.
Definition: gdal_simplesurf.h:397
GDALOctaveMap::octaveStart
int octaveStart
Number of bottom octave.
Definition: gdal_simplesurf.h:402
GDALOctaveMap::pMap
GDALOctaveLayer *** pMap
2-dimensional array of octave layers
Definition: gdal_simplesurf.h:392
GDALOctaveMap::GDALOctaveMap
GDALOctaveMap(int nOctaveStart, int nOctaveEnd)
Create octave space.
Definition: gdal_octave.cpp:241
GDALOctaveMap::ComputeMap
void ComputeMap(GDALIntegralImage *poImg)
Calculate Hessian values for octave space (for all stored octave layers) using specified integral ima...
Definition: gdal_octave.cpp:254
GDALOctaveMap::octaveEnd
int octaveEnd
Number of top octave.
Definition: gdal_simplesurf.h:407
GDALOctaveMap::PointIsExtremum
static bool PointIsExtremum(int row, int col, GDALOctaveLayer *bot, GDALOctaveLayer *mid, GDALOctaveLayer *top, double threshold)
Method makes decision that specified point in middle octave layer is maximum among all points from 3x...
Definition: gdal_octave.cpp:261
GDALRasterBand
A single raster band (or channel).
Definition: gdal_priv.h:1127
GDALSimpleSURF
Class for searching corresponding points on images.
Definition: gdal_simplesurf.h:422
GDALSimpleSURF::ExtractFeaturePoints
std::vector< GDALFeaturePoint > * ExtractFeaturePoints(GDALIntegralImage *poImg, double dfThreshold)
Find feature points using specified integral image.
Definition: gdal_simplesurf.cpp:213
GDALSimpleSURF::GDALSimpleSURF
GDALSimpleSURF(int nOctaveStart, int nOctaveEnd)
Prepare class according to specified parameters.
Definition: gdal_simplesurf.cpp:129
GDALSimpleSURF::MatchFeaturePoints
static CPLErr MatchFeaturePoints(std::vector< GDALFeaturePoint * > *poMatchPairs, std::vector< GDALFeaturePoint > *poFirstCollect, std::vector< GDALFeaturePoint > *poSecondCollect, double dfThreshold)
Find corresponding points (equal points in two collections).
Definition: gdal_simplesurf.cpp:345
GDALSimpleSURF::ConvertRGBToLuminosity
static CPLErr ConvertRGBToLuminosity(GDALRasterBand *red, GDALRasterBand *green, GDALRasterBand *blue, int nXSize, int nYSize, double **padfImg, int nHeight, int nWidth)
Convert image with RGB channels to grayscale using "luminosity" method.
Definition: gdal_simplesurf.cpp:137
cpl_conv.h
Various convenience functions for CPL.
CPLErr
CPLErr
Error category.
Definition: cpl_error.h:53
CPL_DISALLOW_COPY_ASSIGN
#define CPL_DISALLOW_COPY_ASSIGN(ClassName)
Helper to remove the copy and assignment constructors so that the compiler will not generate the defa...
Definition: cpl_port.h:1007
gdal_priv.h
C++ GDAL entry points.