GDAL-correlator/main.cpp at master · wwwonekey/GDAL-correlator · 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
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
/**
 * @file
 * @author Andrew Migal migal.drew@gmail.com
 * @brief Main demonstration function
 *
 * This program is free software and
 * is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY
 */

#include "gdal.h"
#include "gdal_priv.h"
#include "cpl_conv.h"
#include "cpl_vsi.h"

#include "GDALSimpleSURF.h"
#include "GDALFeaturePoint.h"
#include "GDALFeaturePointsCollection.h"
#include "GDALMatchedPointsCollection.h"
#include "GDALCorrelator.h"

#include <fstream>

using namespace std;

/**
 * Main function, which invokes correlator methods
 */
int main(int argc, char* argv[])
{
	const char* USAGE = "Usage: filename, filename, lowest octave, highest octave\n";

    GDALAllRegister();

    GDALDataset  *poDataset_1;
    GDALDataset  *poDataset_2;

    if (argc < 5)
    {
    	printf("Parameters are missing!\n");
    	printf(USAGE);
    	return -1;
    }

    //poDataset_1 = (GDALDataset *) GDALOpen(
    //		"/home/andrew/workspace/GDAL-correlator/Debug/1.jpg", GA_ReadOnly );
    poDataset_1 = (GDALDataset *) GDALOpen(argv[1], GA_ReadOnly );
	if( poDataset_1 == NULL )
       return 0;

	//poDataset_2 = (GDALDataset *) GDALOpen(
	//		"/home/andrew/workspace/GDAL-correlator/Debug/2.jpg", GA_ReadOnly );
	poDataset_2 = (GDALDataset *) GDALOpen(argv[2], GA_ReadOnly );
	if( poDataset_2 == NULL )
		return 0;

	GDALFeaturePointsCollection *poFPCollection_1 =
			new GDALFeaturePointsCollection(poDataset_1);
	GDALFeaturePointsCollection *poFPCollection_2 =
			new GDALFeaturePointsCollection(poDataset_2);

	GDALMatchedPointsCollection *poMatched = new GDALMatchedPointsCollection();

	int nOctStart = atoi(argv[3]);
    int nOctEnd = atoi(argv[4]);
	double dfSURFTreshold = 0.001;
	double dfMatchingThreshold = 0.015;

    printf("Initial octave %d\n", nOctStart);
    printf("Ending octave %d\n", nOctEnd);

	int* panBands = new int[3];
	for (int i = 0; i < 3; i++)
		panBands[i] = i + 1;

	// Find feature points on both images
    printf("Finding feature points on 1 image... ");
	GatherFeaturePoints(poDataset_1, panBands,
			poFPCollection_1, nOctStart, nOctEnd, dfSURFTreshold);
	printf("Found: %d points \n", poFPCollection_1->GetSize());
    printf("Finding feature points on 2 image... ");
	GatherFeaturePoints(poDataset_2, panBands,
			poFPCollection_2, nOctStart, nOctEnd, dfSURFTreshold);
	printf("Found: %d points \n", poFPCollection_2->GetSize());
	// Use gathered points to find correspondences
    printf("Matching... ");
	MatchFeaturePoints(poMatched,
			poFPCollection_1, poFPCollection_2, dfMatchingThreshold);
	printf("Pairs found: %d \n", poMatched->GetSize());

/* -------------------------------------------------------------------- */
/*      Printing parameters for demonstration                           */
/* -------------------------------------------------------------------- */
	ofstream out;
    printf("Writing results...\n");

	out.open("points_1.txt");
	for (int i = 0; i < poFPCollection_1->GetSize(); i++)
		out << poFPCollection_1->GetPoint(i)->GetX() << " "
			<< poFPCollection_1->GetPoint(i)->GetY() << endl;
	out.close();

	out.open("points_2.txt");
	for (int i = 0; i < poFPCollection_2->GetSize(); i++)
		out << poFPCollection_2->GetPoint(i)->GetX() << " "
			<< poFPCollection_2->GetPoint(i)->GetY() << endl;

	out.close();

	GDALFeaturePoint *point_1 = new GDALFeaturePoint();
	GDALFeaturePoint *point_2 = new GDALFeaturePoint();

	out.open("matched_points.txt");
	for (int i = 0; i < poMatched->GetSize(); i++)
	{
		poMatched->GetPoints(i, point_1, point_2);

		out << point_1->GetX() << " " << point_1->GetY();
		out << " ";
		out << point_2->GetX() << " " << point_2->GetY();
		out << endl;
	}
	out.close();

	delete poDataset_1;
	delete poDataset_2;
	delete poFPCollection_1;
	delete poFPCollection_2;
	delete poMatched;
	delete point_1;
	delete point_2;

	delete[] panBands;

	printf("Everything looks good...\n");

	return 0;
}