Added gdalchop tool for chopping DEM data available in GDAL-supported raster files.
This commit is contained in:
Ralf Gerlich
James Turner
parent
5c5db25930
commit
f0a5c9ad45
8 changed files with 590 additions and 6 deletions
12
README.gpc
12
README.gpc
|
|
@ -55,11 +55,13 @@ by Curt Olson:
|
||||||
|
|
||||||
-------------------------- snip ------------------------------
|
-------------------------- snip ------------------------------
|
||||||
|
|
||||||
# Unix/Linux makefile for GPC 2.31
|
# Unix/Linux makefile for GPC 2.32
|
||||||
#
|
#
|
||||||
# Riley Rainey (riley.rainey@websimulations.com)
|
# Riley Rainey (riley.rainey@websimulations.com)
|
||||||
|
# Ralf Gerlich (ralf.gerlich (at) custom-scenery (dot) org)
|
||||||
|
|
||||||
CFLAGS = -O -g
|
CFLAGS = -O -g
|
||||||
|
PREFIX=/usr/local
|
||||||
|
|
||||||
libgenpolyclip.a: gpc.o
|
libgenpolyclip.a: gpc.o
|
||||||
rm -f $@
|
rm -f $@
|
||||||
|
|
@ -70,7 +72,7 @@ clean:
|
||||||
rm -f libgenpolyclip.a *.o core *~
|
rm -f libgenpolyclip.a *.o core *~
|
||||||
|
|
||||||
install: libgenpolyclip.a
|
install: libgenpolyclip.a
|
||||||
-mkdir -p /usr/local/lib
|
-mkdir -p $(PREFIX)/lib
|
||||||
-mkdir -p /usr/local/include
|
-mkdir -p $(PREFIX)/include
|
||||||
install libgenpolyclip.a /usr/local/lib/libgenpolyclip.a
|
install libgenpolyclip.a $(PREFIX)/lib/libgenpolyclip.a
|
||||||
install gpc.h /usr/local/include/gpc.h
|
install gpc.h $(PREFIX)/include/gpc.h
|
||||||
|
|
|
||||||
13
configure.ac
13
configure.ac
|
|
@ -91,6 +91,10 @@ AC_ARG_ENABLE(ogrdecode,
|
||||||
AC_HELP_STRING([--disable-ogrdecode],
|
AC_HELP_STRING([--disable-ogrdecode],
|
||||||
[enable ogrdecode tool (default depending on OGR availability)]))
|
[enable ogrdecode tool (default depending on OGR availability)]))
|
||||||
|
|
||||||
|
AC_ARG_ENABLE(gdalchop,
|
||||||
|
AC_HELP_STRING([--enable-gdalchop],
|
||||||
|
[enable gdalchop tool (default depending on GDAL availability)]))
|
||||||
|
|
||||||
AC_ARG_ENABLE(poly2ogr,
|
AC_ARG_ENABLE(poly2ogr,
|
||||||
AC_HELP_STRING([--disable-poly2ogr],
|
AC_HELP_STRING([--disable-poly2ogr],
|
||||||
[enable poly2ogr tool (default depending on OGR availability)]))
|
[enable poly2ogr tool (default depending on OGR availability)]))
|
||||||
|
|
@ -144,6 +148,13 @@ if test x$have_ogr != xyes -a x$enable_ogrdecode = xyes; then
|
||||||
using --with-gdal.])
|
using --with-gdal.])
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
if test x$have_gdal != xyes -a x$enable_gdalchop = xyes; then
|
||||||
|
AC_MSG_ERROR([gdalchop was forcefully enabled but GDAL was not found.
|
||||||
|
Please make sure that you have GDAL installed
|
||||||
|
and gdal-config in your PATH resp. the path to gdal-config provided
|
||||||
|
using --with-gdal.])
|
||||||
|
fi
|
||||||
|
|
||||||
if test x$have_ogr != xyes -a x$enable_poly2ogr = xyes; then
|
if test x$have_ogr != xyes -a x$enable_poly2ogr = xyes; then
|
||||||
AC_MSG_ERROR([poly2ogr was forcefully enabled but OGR was not found.
|
AC_MSG_ERROR([poly2ogr was forcefully enabled but OGR was not found.
|
||||||
Please make sure that you have GDAL installed with OGR enabled
|
Please make sure that you have GDAL installed with OGR enabled
|
||||||
|
|
@ -152,6 +163,7 @@ if test x$have_ogr != xyes -a x$enable_poly2ogr = xyes; then
|
||||||
fi
|
fi
|
||||||
|
|
||||||
AM_CONDITIONAL(WANT_OGRDECODE, test x$have_ogr = xyes -a x$enable_ogrdecode != xno)
|
AM_CONDITIONAL(WANT_OGRDECODE, test x$have_ogr = xyes -a x$enable_ogrdecode != xno)
|
||||||
|
AM_CONDITIONAL(WANT_GDALCHOP, test x$have_gdal = xyes -a x$enable_gdalchop != xno)
|
||||||
AM_CONDITIONAL(WANT_POLY2OGR, test x$have_ogr = xyes -a x$enable_poly2ogr != xno)
|
AM_CONDITIONAL(WANT_POLY2OGR, test x$have_ogr = xyes -a x$enable_poly2ogr != xno)
|
||||||
|
|
||||||
dnl Check for MS Windows environment
|
dnl Check for MS Windows environment
|
||||||
|
|
@ -456,6 +468,7 @@ AC_CONFIG_FILES([ \
|
||||||
src/Prep/DemInfo/Makefile \
|
src/Prep/DemInfo/Makefile \
|
||||||
src/Prep/DemRaw2ascii/Makefile \
|
src/Prep/DemRaw2ascii/Makefile \
|
||||||
src/Prep/E00Lines/Makefile \
|
src/Prep/E00Lines/Makefile \
|
||||||
|
src/Prep/GDALChop/Makefile \
|
||||||
src/Prep/GSHHS/Makefile \
|
src/Prep/GSHHS/Makefile \
|
||||||
src/Prep/MergerClipper/Makefile \
|
src/Prep/MergerClipper/Makefile \
|
||||||
src/Prep/Photo/Makefile \
|
src/Prep/Photo/Makefile \
|
||||||
|
|
|
||||||
|
|
@ -22,6 +22,7 @@
|
||||||
|
|
||||||
|
|
||||||
#include <simgear/compiler.h>
|
#include <simgear/compiler.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
|
|
|
||||||
|
|
@ -24,6 +24,7 @@
|
||||||
|
|
||||||
|
|
||||||
bin_PROGRAMS = demchop hgtchop fillvoids testassem
|
bin_PROGRAMS = demchop hgtchop fillvoids testassem
|
||||||
|
|
||||||
if HAVE_LIBTIFF
|
if HAVE_LIBTIFF
|
||||||
bin_PROGRAMS+=srtmchop
|
bin_PROGRAMS+=srtmchop
|
||||||
endif
|
endif
|
||||||
|
|
|
||||||
1
src/Prep/GDALChop/.gitignore
vendored
Normal file
1
src/Prep/GDALChop/.gitignore
vendored
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
gdalchop
|
||||||
38
src/Prep/GDALChop/Makefile.am
Normal file
38
src/Prep/GDALChop/Makefile.am
Normal file
|
|
@ -0,0 +1,38 @@
|
||||||
|
#---------------------------------------------------------------------------
|
||||||
|
# Makefile
|
||||||
|
#
|
||||||
|
# Written by Ralf Gerlich, started Nov 2007
|
||||||
|
#
|
||||||
|
# Copyright (C) 2007 Ralf Gerlich - http://www.custom-scenery.org/
|
||||||
|
#
|
||||||
|
# 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., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
#
|
||||||
|
# $Id$
|
||||||
|
#---------------------------------------------------------------------------
|
||||||
|
|
||||||
|
bin_PROGRAMS = gdalchop
|
||||||
|
|
||||||
|
gdalchop_SOURCES = \
|
||||||
|
gdalchop.cxx
|
||||||
|
|
||||||
|
gdalchop_CXXFLAGS=$(GDAL_CFLAGS)
|
||||||
|
|
||||||
|
gdalchop_LDADD = \
|
||||||
|
$(GDAL_LIBS) -lsgbucket -lsgmisc -lsgdebug -lsgxml -lz
|
||||||
|
$(base_LIBS)
|
||||||
|
|
||||||
|
INCLUDES = -I$(top_srcdir)/src -I$(top_srcdir)/src/Lib
|
||||||
|
|
||||||
|
#---------------------------------------------------------------------------
|
||||||
525
src/Prep/GDALChop/gdalchop.cxx
Normal file
525
src/Prep/GDALChop/gdalchop.cxx
Normal file
|
|
@ -0,0 +1,525 @@
|
||||||
|
// gdalchop.cxx -- chop up a GDAL-readable raster-file into it's corresponding pieces
|
||||||
|
// and stuff them into the workspace directory
|
||||||
|
//
|
||||||
|
// Written by Ralf Gerlich, started February 20th, 2007
|
||||||
|
//
|
||||||
|
// Copyright (C) 2007 Ralf Gerlich - ralf.gerlich@custom-scenery.org
|
||||||
|
//
|
||||||
|
// Loosely based on hgtchop by Curtis L. Olson - http://www.flightgear.org/~curt
|
||||||
|
//
|
||||||
|
// 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., 675 Mass Ave, Cambridge, MA 02139, USA.
|
||||||
|
//
|
||||||
|
// $Id$
|
||||||
|
|
||||||
|
#ifdef HAVE_CONFIG_H
|
||||||
|
# include <config.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <simgear/compiler.h>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <iostream>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
#include <simgear/bucket/newbucket.hxx>
|
||||||
|
#include <simgear/debug/logstream.hxx>
|
||||||
|
#include <simgear/io/lowlevel.hxx>
|
||||||
|
|
||||||
|
#ifdef _MSC_VER
|
||||||
|
# include <Win32/mkdir.hpp>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <gdal.h>
|
||||||
|
#include <gdal_priv.h>
|
||||||
|
#include <gdalwarper.h>
|
||||||
|
#include <ogr_spatialref.h>
|
||||||
|
|
||||||
|
using std::cout;
|
||||||
|
using std::string;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* A simple benchmark using a 5x5 degree package
|
||||||
|
* has shown that gdalchop takes only 80% of the time
|
||||||
|
* hgtchop requires for chopping that data.
|
||||||
|
*
|
||||||
|
* This includes unzipping of the files.
|
||||||
|
*
|
||||||
|
* The experiment was done with the srtm_38_03.zip tile of
|
||||||
|
* CGIAR data and the equivalent tiles of SRTM v2 data,
|
||||||
|
* using gdalchop for chopping the CGIAR data and hgtchop for
|
||||||
|
* the SRTM v2 data.
|
||||||
|
*
|
||||||
|
* CPU: Intel(R) Core(TM)2 CPU T5500 @ 1.66GHz
|
||||||
|
* (only one core is used during chopping)
|
||||||
|
*
|
||||||
|
* Three measurements were taken for each tool and each
|
||||||
|
* generated .arr.gz files for 800 buckets.
|
||||||
|
*
|
||||||
|
* hgtchop: 216s 211s 211s, mean: 212s, 0.266s/bucket
|
||||||
|
* gdalchop: 172s 167s 169s, mean: 169s, 0.212s/bucket
|
||||||
|
*
|
||||||
|
* - Ralf Gerlich
|
||||||
|
*/
|
||||||
|
|
||||||
|
struct SimpleRasterTransformerInfo {
|
||||||
|
GDALTransformerFunc pfnTransformer;
|
||||||
|
void* pTransformerArg;
|
||||||
|
double x0, y0;
|
||||||
|
double col_step, row_step;
|
||||||
|
};
|
||||||
|
|
||||||
|
int SimpleRasterTransformer(void *pTransformerArg,
|
||||||
|
int bDstToSrc, int nPointCount,
|
||||||
|
double *x, double *y, double *z, int *panSuccess )
|
||||||
|
{
|
||||||
|
SimpleRasterTransformerInfo* info=(SimpleRasterTransformerInfo*)pTransformerArg;
|
||||||
|
int success;
|
||||||
|
|
||||||
|
if (bDstToSrc) {
|
||||||
|
/* transform destination -> source */
|
||||||
|
for (int i=0;i<nPointCount;i++) {
|
||||||
|
x[i]=info->x0+info->col_step*x[i];
|
||||||
|
y[i]=info->y0+info->row_step*y[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
success=info->pfnTransformer(info->pTransformerArg,
|
||||||
|
bDstToSrc,nPointCount,
|
||||||
|
x,y,z,panSuccess);
|
||||||
|
} else {
|
||||||
|
success=info->pfnTransformer(info->pTransformerArg,
|
||||||
|
bDstToSrc,nPointCount,
|
||||||
|
x,y,z,panSuccess);
|
||||||
|
for (int i=0;i<nPointCount;i++) {
|
||||||
|
if (!panSuccess[i]) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
x[i]=(x[i]-info->x0)/info->col_step;
|
||||||
|
y[i]=(y[i]-info->y0)/info->row_step;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|
||||||
|
class ImageInfo {
|
||||||
|
public:
|
||||||
|
ImageInfo(GDALDataset *dataset);
|
||||||
|
|
||||||
|
void GetBounds(double &n, double &s, double &e, double &w) const {
|
||||||
|
n=north;
|
||||||
|
s=south;
|
||||||
|
e=east;
|
||||||
|
w=west;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* GetDescription() const {
|
||||||
|
return dataset->GetDescription();
|
||||||
|
}
|
||||||
|
|
||||||
|
void GetDataChunk(int *buffer,
|
||||||
|
double x, double y,
|
||||||
|
double colstep, double rowstep,
|
||||||
|
int w, int h,
|
||||||
|
int srcband=1, int nodata=-32768);
|
||||||
|
protected:
|
||||||
|
/* The dataset */
|
||||||
|
GDALDataset *dataset;
|
||||||
|
|
||||||
|
/* Source spatial reference system */
|
||||||
|
OGRSpatialReference srs;
|
||||||
|
|
||||||
|
/* Coordinate transformation pixel -> geographic */
|
||||||
|
double geoXfrm[6];
|
||||||
|
|
||||||
|
/* Coordinate transformation to WGS84 */
|
||||||
|
OGRCoordinateTransformation *wgs84xform;
|
||||||
|
|
||||||
|
/* geographical edge coordinates in CCW order, WGS84 */
|
||||||
|
double geoX[4],geoY[4];
|
||||||
|
|
||||||
|
/* bounding box in WGS84 */
|
||||||
|
double north,south,east,west;
|
||||||
|
};
|
||||||
|
|
||||||
|
ImageInfo::ImageInfo(GDALDataset *dataset):
|
||||||
|
dataset(dataset),
|
||||||
|
srs(dataset->GetProjectionRef())
|
||||||
|
{
|
||||||
|
OGRSpatialReference wgs84SRS;
|
||||||
|
wgs84SRS.SetWellKnownGeogCS( "EPSG:4326" );
|
||||||
|
|
||||||
|
|
||||||
|
/* Determine the bounds of the input file in WGS84 */
|
||||||
|
int w=dataset->GetRasterXSize();
|
||||||
|
int h=dataset->GetRasterYSize();
|
||||||
|
|
||||||
|
if (dataset->GetGeoTransform(geoXfrm)!=CE_None) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT,
|
||||||
|
"Could not determine transform matrix for dataset "
|
||||||
|
"'" << dataset->GetDescription() << "'"
|
||||||
|
":" << CPLGetLastErrorMsg());
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* create points in CCW order */
|
||||||
|
geoX[0]=geoXfrm[0]+0*geoXfrm[1]+0*geoXfrm[2];
|
||||||
|
geoX[1]=geoXfrm[0]+0*geoXfrm[1]+h*geoXfrm[2];
|
||||||
|
geoX[2]=geoXfrm[0]+w*geoXfrm[1]+h*geoXfrm[2];
|
||||||
|
geoX[3]=geoXfrm[0]+w*geoXfrm[1]+0*geoXfrm[2];
|
||||||
|
geoY[0]=geoXfrm[3]+0*geoXfrm[4]+0*geoXfrm[5];
|
||||||
|
geoY[1]=geoXfrm[3]+0*geoXfrm[4]+h*geoXfrm[5];
|
||||||
|
geoY[2]=geoXfrm[3]+w*geoXfrm[4]+h*geoXfrm[5];
|
||||||
|
geoY[3]=geoXfrm[3]+w*geoXfrm[4]+0*geoXfrm[5];
|
||||||
|
|
||||||
|
const char* projRef=dataset->GetProjectionRef();
|
||||||
|
|
||||||
|
srs=OGRSpatialReference(projRef);
|
||||||
|
|
||||||
|
wgs84xform = OGRCreateCoordinateTransformation( &srs, &wgs84SRS );
|
||||||
|
|
||||||
|
if (!wgs84xform->Transform(4,geoX,geoY)) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT,
|
||||||
|
"Could not transform edge points of dataset "
|
||||||
|
"'" << dataset->GetDescription() << "'"
|
||||||
|
":" << CPLGetLastErrorMsg());
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
east=west=geoX[0];
|
||||||
|
north=south=geoY[0];
|
||||||
|
|
||||||
|
for (int j=1;j<4;j++) {
|
||||||
|
north=std::max(north,geoY[j]);
|
||||||
|
south=std::min(south,geoY[j]);
|
||||||
|
east =std::max(east ,geoX[j]);
|
||||||
|
west =std::min(west ,geoX[j]);
|
||||||
|
}
|
||||||
|
|
||||||
|
SG_LOG(SG_GENERAL, SG_INFO,
|
||||||
|
"INFO: Bounds for '" << dataset->GetDescription() << "' are"
|
||||||
|
" n=" << north << " s=" << south <<
|
||||||
|
" e=" << east << " w=" << west);
|
||||||
|
}
|
||||||
|
|
||||||
|
void ImageInfo::GetDataChunk(int *buffer,
|
||||||
|
double x, double y,
|
||||||
|
double colstep, double rowstep,
|
||||||
|
int w, int h,
|
||||||
|
int srcband, int nodata) {
|
||||||
|
OGRSpatialReference wgs84SRS;
|
||||||
|
wgs84SRS.SetWellKnownGeogCS( "EPSG:4326" );
|
||||||
|
|
||||||
|
char* wgs84WKT;
|
||||||
|
|
||||||
|
wgs84SRS.exportToWkt(&wgs84WKT);
|
||||||
|
|
||||||
|
/* Setup a raster transformation from WGS84 to raster coordinates of the array files */
|
||||||
|
SimpleRasterTransformerInfo xformData;
|
||||||
|
xformData.pTransformerArg=GDALCreateGenImgProjTransformer(
|
||||||
|
dataset,NULL,
|
||||||
|
NULL,wgs84WKT,
|
||||||
|
FALSE,
|
||||||
|
0.0,
|
||||||
|
1);
|
||||||
|
xformData.pfnTransformer=GDALGenImgProjTransform;
|
||||||
|
xformData.x0=x;
|
||||||
|
xformData.y0=y;
|
||||||
|
xformData.col_step=colstep;
|
||||||
|
xformData.row_step=rowstep;
|
||||||
|
|
||||||
|
// TODO: check if this image can actually cover part of the chunk
|
||||||
|
|
||||||
|
/* establish the full source to target transformation */
|
||||||
|
GDALWarpOptions *psWarpOptions = GDALCreateWarpOptions();
|
||||||
|
|
||||||
|
int srcBandNumbers[]={srcband};
|
||||||
|
int dstBandNumbers[]={1};
|
||||||
|
double dstNodata[]={nodata};
|
||||||
|
double srcNodataReal[1];
|
||||||
|
double srcNodataImag[]={0.0};
|
||||||
|
int srcHasNodataValue;
|
||||||
|
|
||||||
|
srcNodataReal[0]=dataset->GetRasterBand(srcband)->GetNoDataValue(&srcHasNodataValue);
|
||||||
|
|
||||||
|
psWarpOptions->hSrcDS=dataset;
|
||||||
|
psWarpOptions->hDstDS=NULL;
|
||||||
|
psWarpOptions->nBandCount=1;
|
||||||
|
psWarpOptions->panSrcBands=srcBandNumbers;
|
||||||
|
psWarpOptions->panDstBands=dstBandNumbers;
|
||||||
|
psWarpOptions->nSrcAlphaBand=0;
|
||||||
|
psWarpOptions->nDstAlphaBand=0;
|
||||||
|
psWarpOptions->padfSrcNoDataReal=(srcHasNodataValue?srcNodataReal:NULL);
|
||||||
|
psWarpOptions->padfSrcNoDataImag=(srcHasNodataValue?srcNodataImag:NULL);
|
||||||
|
psWarpOptions->padfDstNoDataReal=dstNodata;
|
||||||
|
psWarpOptions->eResampleAlg = GRA_Bilinear;
|
||||||
|
psWarpOptions->eWorkingDataType = GDT_Int32;
|
||||||
|
|
||||||
|
psWarpOptions->pfnTransformer=SimpleRasterTransformer;
|
||||||
|
psWarpOptions->pTransformerArg=&xformData;
|
||||||
|
|
||||||
|
GDALWarpOperation oOperation;
|
||||||
|
|
||||||
|
oOperation.Initialize( psWarpOptions );
|
||||||
|
|
||||||
|
/* do the warp */
|
||||||
|
if (oOperation.WarpRegionToBuffer(0,0,w,h,buffer,GDT_Int32)!=CE_None) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT,
|
||||||
|
"Could not warp to buffer on dataset '" << GetDescription() << "'"
|
||||||
|
":" << CPLGetLastErrorMsg());
|
||||||
|
}
|
||||||
|
|
||||||
|
/* clean up */
|
||||||
|
psWarpOptions->panSrcBands=NULL;
|
||||||
|
psWarpOptions->panDstBands=NULL;
|
||||||
|
psWarpOptions->padfSrcNoDataReal=NULL;
|
||||||
|
psWarpOptions->padfSrcNoDataImag=NULL;
|
||||||
|
psWarpOptions->padfDstNoDataReal=NULL;
|
||||||
|
|
||||||
|
GDALDestroyGenImgProjTransformer( xformData.pTransformerArg );
|
||||||
|
GDALDestroyWarpOptions( psWarpOptions );
|
||||||
|
}
|
||||||
|
|
||||||
|
void write_bucket(const string& work_dir, SGBucket bucket,
|
||||||
|
int* buffer,
|
||||||
|
int min_x, int min_y,
|
||||||
|
int span_x, int span_y,
|
||||||
|
int col_step, int row_step,
|
||||||
|
bool compress=true) {
|
||||||
|
string base = bucket.gen_base_path();
|
||||||
|
string path = work_dir + "/" + base;
|
||||||
|
#ifdef _MSC_VER
|
||||||
|
fg_mkdir( path.c_str() );
|
||||||
|
#else
|
||||||
|
string command = "mkdir -p " + path;
|
||||||
|
system( command.c_str() );
|
||||||
|
#endif
|
||||||
|
string array_file = path + "/" + bucket.gen_index_str() + ".arr";
|
||||||
|
|
||||||
|
FILE *fp;
|
||||||
|
if ( (fp = fopen(array_file.c_str(), "w")) == NULL ) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT, "cannot open " << array_file << " for writing!");
|
||||||
|
exit(-1);
|
||||||
|
}
|
||||||
|
|
||||||
|
fprintf( fp, "%d %d\n", min_x, min_y );
|
||||||
|
fprintf( fp, "%d %d %d %d\n",
|
||||||
|
span_x + 1, col_step,
|
||||||
|
span_y + 1, row_step );
|
||||||
|
int k=0;
|
||||||
|
for ( int x = 0; x <= span_x; ++x ) {
|
||||||
|
for ( int y = 0; y <= span_y; ++y ) {
|
||||||
|
fprintf( fp, "%d ", buffer[ y * span_x + x ] );
|
||||||
|
}
|
||||||
|
fprintf( fp, "\n" );
|
||||||
|
}
|
||||||
|
fclose(fp);
|
||||||
|
|
||||||
|
if ( compress ) {
|
||||||
|
string command = "gzip --best -f " + array_file;
|
||||||
|
system( command.c_str() );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void process_bucket(const string& work_dir, SGBucket bucket,
|
||||||
|
ImageInfo* images[], int imagecount,
|
||||||
|
bool forceWrite=false) {
|
||||||
|
double clat,clon;
|
||||||
|
double bnorth,bsouth,beast,bwest;
|
||||||
|
|
||||||
|
clat=bucket.get_center_lat();
|
||||||
|
clon=bucket.get_center_lon();
|
||||||
|
|
||||||
|
bnorth=clat+bucket.get_height()/2.0;
|
||||||
|
bsouth=clat-bucket.get_height()/2.0;
|
||||||
|
beast =clon+bucket.get_width()/2.0;
|
||||||
|
bwest =clon-bucket.get_width()/2.0;
|
||||||
|
|
||||||
|
SG_LOG(SG_GENERAL, SG_INFO, "processing bucket " << bucket << "(" << bucket.gen_index() << ") n=" << bnorth << " s=" << bsouth << " e=" << beast << " w=" << bwest);
|
||||||
|
|
||||||
|
/* Get the data from the input datasets... */
|
||||||
|
int min_x, min_y, span_x, span_y;
|
||||||
|
|
||||||
|
min_x = (int)(bwest*3600.0);
|
||||||
|
min_y = (int)(bsouth*3600.0);
|
||||||
|
|
||||||
|
// TODO: Make other resolutions possible as well
|
||||||
|
int col_step=3, row_step=3;
|
||||||
|
span_x=bucket.get_width()*3600/col_step;
|
||||||
|
span_y=bucket.get_height()*3600/row_step;
|
||||||
|
|
||||||
|
int cellcount=(span_x+1)*(span_y+1);
|
||||||
|
int buffer[cellcount];
|
||||||
|
|
||||||
|
::memset(buffer,-1,(span_x+1)*(span_y+1)*sizeof(int));
|
||||||
|
|
||||||
|
for (int i=0;i<imagecount;i++) {
|
||||||
|
double inorth,isouth,ieast,iwest;
|
||||||
|
images[i]->GetBounds(inorth,isouth,ieast,iwest);
|
||||||
|
|
||||||
|
images[i]->GetDataChunk(buffer,
|
||||||
|
bwest, bsouth,
|
||||||
|
col_step/3600.0, row_step/3600.0,
|
||||||
|
span_x+1, span_y+1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ...check the amount of undefined cells... */
|
||||||
|
int nodataCellCount=0;
|
||||||
|
|
||||||
|
for (int i=0;i<cellcount;i++) {
|
||||||
|
if (buffer[i]==-32768) {
|
||||||
|
nodataCellCount++;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const double nodataPercLimit=5.0;
|
||||||
|
double nodataPerc=100.0*nodataCellCount/cellcount;
|
||||||
|
|
||||||
|
if (nodataCellCount>0) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_INFO, " " << nodataCellCount << " cells are not covered with data (" << nodataPerc << "% of cells)");
|
||||||
|
}
|
||||||
|
if (nodataPerc>nodataPercLimit) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_INFO, " there is not enough data available to cover this cell (limit for non-covered cells is " << nodataPercLimit <<"%)");
|
||||||
|
/* don't write out if not forced to */
|
||||||
|
if (!forceWrite)
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* ...and write it out */
|
||||||
|
write_bucket(work_dir, bucket,
|
||||||
|
buffer,
|
||||||
|
min_x, min_y,
|
||||||
|
span_x, span_y,
|
||||||
|
col_step, row_step);
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, const char **argv) {
|
||||||
|
sglog().setLogLevels( SG_ALL, SG_INFO );
|
||||||
|
|
||||||
|
if ( argc < 3 ) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT,
|
||||||
|
"Usage " << argv[0] << " <work_dir> <datasetname...> [-- <bucket-idx> ...]");
|
||||||
|
exit(-1);
|
||||||
|
}
|
||||||
|
|
||||||
|
string work_dir = argv[1];
|
||||||
|
|
||||||
|
#ifdef _MSC_VER
|
||||||
|
fg_mkdir( work_dir.c_str() );
|
||||||
|
#else
|
||||||
|
string command = "mkdir -p " + work_dir;
|
||||||
|
system( command.c_str() );
|
||||||
|
#endif
|
||||||
|
|
||||||
|
GDALAllRegister();
|
||||||
|
|
||||||
|
int datasetcount=0,tilecount=0;
|
||||||
|
int dashpos;
|
||||||
|
|
||||||
|
for (dashpos=2;dashpos<argc;dashpos++) {
|
||||||
|
if (!strcmp(argv[dashpos],"--")) {
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
datasetcount=dashpos-2;
|
||||||
|
tilecount=(dashpos==argc?0:argc-dashpos-1);
|
||||||
|
|
||||||
|
if (datasetcount==0) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT,
|
||||||
|
"No data sets supplied. Must provide at least one dataset.");
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
const char** tilenames=argv+dashpos+1;
|
||||||
|
const char** datasetnames=argv+2;
|
||||||
|
|
||||||
|
ImageInfo *images[datasetcount];
|
||||||
|
|
||||||
|
double north=-1000,south=1000,east=-1000,west=1000;
|
||||||
|
|
||||||
|
// TODO: allow specification of bounds by user
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Step 1: Open all provided datasets and determine their bounds in WGS84.
|
||||||
|
*/
|
||||||
|
for (int i=0;i<datasetcount;i++) {
|
||||||
|
GDALDataset* dataset;
|
||||||
|
|
||||||
|
dataset=(GDALDataset *)GDALOpenShared(datasetnames[i],GA_ReadOnly);
|
||||||
|
|
||||||
|
if (dataset==NULL) {
|
||||||
|
SG_LOG(SG_GENERAL, SG_ALERT,
|
||||||
|
"Could not open dataset '" << datasetnames[i] << "'"
|
||||||
|
":" << CPLGetLastErrorMsg());
|
||||||
|
exit(1);
|
||||||
|
}
|
||||||
|
|
||||||
|
images[i]=new ImageInfo(dataset);
|
||||||
|
|
||||||
|
double inorth,isouth,ieast,iwest;
|
||||||
|
images[i]->GetBounds(inorth,isouth,ieast,iwest);
|
||||||
|
|
||||||
|
north=std::max(north,inorth);
|
||||||
|
south=std::min(south,isouth);
|
||||||
|
east =std::max(east ,ieast );
|
||||||
|
west =std::min(west ,iwest );
|
||||||
|
}
|
||||||
|
|
||||||
|
SG_LOG(SG_GENERAL, SG_INFO, "Complete bounds n=" << north << " s=" << south << " e=" << east << " w=" << west);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Step 2: If no tiles were specified, go through all tiles contained in
|
||||||
|
* the common bounds of all datasets and find those which have
|
||||||
|
* sufficient coverage (non-null pixels). Create output for these.
|
||||||
|
* It tiles were specified, go through them and create output for
|
||||||
|
* all of them. Warn if no sufficient coverage (non-null pixels) is
|
||||||
|
* available.
|
||||||
|
*/
|
||||||
|
if (tilecount==0) {
|
||||||
|
/*
|
||||||
|
* No tiles were specified, so we determine the common bounds of all
|
||||||
|
* specified datasets and check all the tiles contained in them.
|
||||||
|
*/
|
||||||
|
|
||||||
|
SGBucket start(west,south),end(east,north);
|
||||||
|
|
||||||
|
int dx,dy;
|
||||||
|
|
||||||
|
sgBucketDiff(start, end, &dx, &dy);
|
||||||
|
|
||||||
|
SG_LOG(SG_GENERAL, SG_INFO, "dx=" << dx << " dy=" << dy);
|
||||||
|
|
||||||
|
for (int x=0;x<=dx;x++) {
|
||||||
|
for (int y=0;y<=dy;y++) {
|
||||||
|
SGBucket bucket=sgBucketOffset(west,south,x,y);
|
||||||
|
|
||||||
|
process_bucket(work_dir,bucket,images,datasetcount);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
/*
|
||||||
|
* Tiles were specified, so process them and warn if not enough
|
||||||
|
* data is available, but write them in any case.
|
||||||
|
*/
|
||||||
|
for (int i=0;i<tilecount;i++) {
|
||||||
|
SGBucket bucket(atol(tilenames[i]));
|
||||||
|
|
||||||
|
process_bucket(work_dir,bucket,images,datasetcount,true);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// :mode=c++:indentSize=4:
|
||||||
|
|
@ -16,4 +16,7 @@ if WANT_OGRDECODE
|
||||||
SUBDIRS+=OGRDecode
|
SUBDIRS+=OGRDecode
|
||||||
endif
|
endif
|
||||||
|
|
||||||
EXTRA_SUBDIRS=OGRDecode
|
if WANT_GDALCHOP
|
||||||
|
SUBDIRS+=GDALChop
|
||||||
|
endif
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue