mirror of
https://gitlab.com/Trygve/contour-creator.git
synced 2024-12-21 21:30:18 +00:00
Added georeferencing
This commit is contained in:
parent
e457adf244
commit
0485253893
@ -4,13 +4,13 @@
|
||||
#include "contour_creator.hh"
|
||||
|
||||
|
||||
CellMap::CellMap(int width, int height, uint8_t* cells, OGRSpatialReference reference_system)
|
||||
CellMap::CellMap(int width, int height, uint8_t* cells, OGRSpatialReference reference_system, double* geotransform)
|
||||
{
|
||||
this->width = width;
|
||||
this->height = height;
|
||||
this->cells = cells;
|
||||
this->reference_system = reference_system;
|
||||
|
||||
this->geotransform = geotransform;
|
||||
}
|
||||
int CellMap::get_cell(int x, int y)
|
||||
{
|
||||
|
@ -1,3 +1,4 @@
|
||||
#include <gdal/cpl_conv.h>
|
||||
#include <iostream>
|
||||
#include <stdexcept>
|
||||
|
||||
@ -28,7 +29,11 @@ HeightMap::HeightMap(const char* filepath)
|
||||
this->height = band->GetYSize();
|
||||
this->min = band->GetMinimum();
|
||||
this->max = band-> GetMaximum();
|
||||
//https://gdal.org/api/gdaldataset_cpp.html#_CPPv4N11GDALDataset15GetGeoTransformEPd
|
||||
this->geotransform = (double *) CPLMalloc(sizeof(double)*6);
|
||||
this->reference_system = *(file->GetSpatialRef());
|
||||
|
||||
file->GetGeoTransform(this->geotransform);
|
||||
|
||||
this->data = (float *) CPLMalloc(sizeof(float)*width*height);
|
||||
CPLErr error = band->RasterIO( GF_Read, 0, 0, width, height,
|
||||
|
@ -9,6 +9,8 @@ class HeightMap
|
||||
{
|
||||
public:
|
||||
float* data;
|
||||
double* geotransform; //!< Six double buffer for storing the affine transformations
|
||||
// https://gdal.org/api/gdaldataset_cpp.html#_CPPv4N11GDALDataset15GetGeoTransformEPd
|
||||
int width; //!< width of image in pixels
|
||||
int height; //!< height of image in pixels
|
||||
float min; //!< Minimum value in image
|
||||
@ -26,10 +28,10 @@ class CellMap
|
||||
{
|
||||
public:
|
||||
uint8_t* cells; //!< pointer to the first cell in the array. uint8_t is a 8 bit unsigned integer
|
||||
double* geotransform; //!< Six double buffer for storing the affine transformations
|
||||
int width; //!< width of image in cells
|
||||
int height; //!< height of image in cells
|
||||
OGRSpatialReference reference_system;
|
||||
|
||||
CellMap(int width, int height, uint8_t* cells, OGRSpatialReference reference_system);
|
||||
CellMap(int width, int height, uint8_t* cells, OGRSpatialReference reference_system, double* geotransform);
|
||||
int get_cell(int x,int y);
|
||||
};
|
23
src/main.cpp
23
src/main.cpp
@ -24,7 +24,7 @@ CellMap produce_cellmap(HeightMap* heightmap, float z)
|
||||
*(cells + i) = result;
|
||||
}
|
||||
|
||||
return CellMap(heightmap->width-1, heightmap->height-1, cells, heightmap->reference_system);
|
||||
return CellMap(heightmap->width-1, heightmap->height-1, cells, heightmap->reference_system, heightmap->geotransform);
|
||||
|
||||
|
||||
}
|
||||
@ -50,6 +50,16 @@ std::vector<CellMap> vector_cellmap(HeightMap* heightmap, int interval)
|
||||
return vector_contours;
|
||||
}
|
||||
|
||||
std::tuple<double, double> local_to_projected(double* geotransform, int x, int y)
|
||||
{
|
||||
return {
|
||||
geotransform[1] * x + geotransform[2] * y +
|
||||
geotransform[1] * 0.5 + geotransform[2] * 0.5 + geotransform[0],
|
||||
geotransform[4] * x + geotransform[5] * y +
|
||||
geotransform[4] * 0.5 + geotransform[5] * 0.5 + geotransform[3]
|
||||
};
|
||||
}
|
||||
|
||||
void write_output_file(std::vector<CellMap> cellmaps, const char *filepath)
|
||||
{
|
||||
|
||||
@ -98,13 +108,7 @@ void write_output_file(std::vector<CellMap> cellmaps, const char *filepath)
|
||||
for (int i = 0; i<cellmap->height*cellmap->width; i++) {
|
||||
if (*(cellmap->cells + i) != 0 && *(cellmap->cells + i) != 15)
|
||||
{
|
||||
int x_int = i%cellmap->width;
|
||||
int y_int = cellmap->height*cellmap->width - i/cellmap->width;
|
||||
double x = double(x_int);
|
||||
double y = double(y_int);
|
||||
|
||||
OGRFeature *poFeature;
|
||||
|
||||
poFeature = OGRFeature::CreateFeature( poLayer->GetLayerDefn() );
|
||||
poFeature->SetField( "Name", *(cellmap->cells + i) );
|
||||
|
||||
@ -117,6 +121,11 @@ void write_output_file(std::vector<CellMap> cellmaps, const char *filepath)
|
||||
printf( "Transformation failed.\n" );
|
||||
*/
|
||||
OGRPoint pt;
|
||||
|
||||
int x_raw = i%cellmap->width;
|
||||
int y_raw = i/cellmap->width;
|
||||
auto [x, y] = local_to_projected(cellmap->geotransform, x_raw, y_raw);
|
||||
|
||||
pt.setX( x );
|
||||
pt.setY( y );
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user