299 строки
9.2 KiB
C++
299 строки
9.2 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Project: C++ Test Suite for GDAL/OGR
|
|
// Purpose: Test Arc/Info ASCII Grid support. Ported from gdrivers/aaigrid.py.
|
|
// Author: Mateusz Loskot <mateusz@loskot.net>
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (c) 2006, Mateusz Loskot <mateusz@loskot.net>
|
|
// Copyright (c) 2010, Even Rouault <even dot rouault at spatialys.com>
|
|
/*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
* copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included
|
|
* in all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
|
|
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
* DEALINGS IN THE SOFTWARE.
|
|
****************************************************************************/
|
|
|
|
#include "gdal_unit_test.h"
|
|
|
|
#include "cpl_string.h"
|
|
#include "gdal_alg.h"
|
|
#include "gdal_priv.h"
|
|
#include "gdal.h"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "gtest_include.h"
|
|
|
|
namespace
|
|
{
|
|
|
|
// Common fixture with test data
|
|
struct test_gdal_aaigrid : public ::testing::Test
|
|
{
|
|
struct raster_t
|
|
{
|
|
std::string file_;
|
|
int band_;
|
|
int checksum_;
|
|
raster_t(std::string const &f, int b, int c)
|
|
: file_(f), band_(b), checksum_(c)
|
|
{
|
|
}
|
|
};
|
|
typedef std::vector<raster_t> rasters_t;
|
|
|
|
GDALDriverH drv_;
|
|
std::string drv_name_;
|
|
std::string data_;
|
|
std::string data_tmp_;
|
|
rasters_t grids_;
|
|
rasters_t rasters_;
|
|
|
|
test_gdal_aaigrid() : drv_(nullptr), drv_name_("AAIGrid")
|
|
{
|
|
drv_ = GDALGetDriverByName(drv_name_.c_str());
|
|
|
|
// Compose data path for test group
|
|
data_ = tut::common::data_basedir;
|
|
data_tmp_ = tut::common::tmp_basedir;
|
|
|
|
// Collection of test AAIGrid grids
|
|
grids_.push_back(raster_t("byte.tif.grd", 1, 4672));
|
|
grids_.push_back(raster_t("pixel_per_line.asc", 1, 1123));
|
|
|
|
// Collection of non-AAIGrid rasters
|
|
rasters_.push_back(raster_t("byte.tif", 1, 4672));
|
|
}
|
|
|
|
void SetUp() override
|
|
{
|
|
if (drv_ == nullptr)
|
|
GTEST_SKIP() << "AAIGrid driver missing";
|
|
}
|
|
};
|
|
|
|
// Test open dataset
|
|
TEST_F(test_gdal_aaigrid, open)
|
|
{
|
|
for (const auto &raster : grids_)
|
|
{
|
|
std::string file(data_ + SEP);
|
|
file += raster.file_;
|
|
GDALDatasetH ds = GDALOpen(file.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != ds);
|
|
GDALClose(ds);
|
|
}
|
|
}
|
|
|
|
// Test dataset checksums
|
|
TEST_F(test_gdal_aaigrid, checksum)
|
|
{
|
|
for (const auto &raster : grids_)
|
|
{
|
|
std::string file(data_ + SEP);
|
|
file += raster.file_;
|
|
|
|
GDALDatasetH ds = GDALOpen(file.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != ds);
|
|
|
|
GDALRasterBandH band = GDALGetRasterBand(ds, raster.band_);
|
|
ASSERT_TRUE(nullptr != band);
|
|
|
|
const int xsize = GDALGetRasterXSize(ds);
|
|
const int ysize = GDALGetRasterYSize(ds);
|
|
const int checksum = GDALChecksumImage(band, 0, 0, xsize, ysize);
|
|
|
|
EXPECT_EQ(checksum, raster.checksum_);
|
|
|
|
GDALClose(ds);
|
|
}
|
|
}
|
|
|
|
// Test affine transformation coefficients
|
|
TEST_F(test_gdal_aaigrid, geotransform)
|
|
{
|
|
// Index of test file being tested
|
|
const std::size_t fileIdx = 1;
|
|
|
|
std::string file(data_ + SEP);
|
|
file += grids_.at(fileIdx).file_;
|
|
|
|
GDALDatasetH ds = GDALOpen(file.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != ds);
|
|
|
|
double geoTransform[6] = {0};
|
|
CPLErr err = GDALGetGeoTransform(ds, geoTransform);
|
|
ASSERT_EQ(err, CE_None);
|
|
|
|
// Test affine transformation coefficients
|
|
const double maxError = 0.000001;
|
|
const double expect[6] = {100000.0, 50, 0, 650600.0, 0, -50};
|
|
EXPECT_NEAR(expect[0], geoTransform[0], maxError);
|
|
EXPECT_NEAR(expect[1], geoTransform[1], maxError);
|
|
EXPECT_NEAR(expect[2], geoTransform[2], maxError);
|
|
EXPECT_NEAR(expect[3], geoTransform[3], maxError);
|
|
EXPECT_NEAR(expect[4], geoTransform[4], maxError);
|
|
EXPECT_NEAR(expect[5], geoTransform[5], maxError);
|
|
|
|
GDALClose(ds);
|
|
}
|
|
|
|
// Test projection definition
|
|
TEST_F(test_gdal_aaigrid, projection)
|
|
{
|
|
// Index of test file being tested
|
|
const std::size_t fileIdx = 1;
|
|
|
|
std::string file(data_ + SEP);
|
|
file += grids_.at(fileIdx).file_;
|
|
|
|
GDALDatasetH ds = GDALOpen(file.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != ds);
|
|
|
|
std::string proj(GDALGetProjectionRef(ds));
|
|
ASSERT_EQ(proj.empty(), false);
|
|
|
|
std::string expect(
|
|
"PROJCS[\"unnamed\",GEOGCS[\"NAD83\","
|
|
"DATUM[\"North_American_Datum_1983\",SPHEROID[\"GRS 1980\","
|
|
"6378137,298.257222101,AUTHORITY[\"EPSG\",\"7019\"]],"
|
|
"AUTHORITY[\"EPSG\",\"6269\"]],"
|
|
"PRIMEM[\"Greenwich\",0,AUTHORITY[\"EPSG\",\"8901\"]],"
|
|
"UNIT[\"degree\",0.0174532925199433,AUTHORITY[\"EPSG\",\"9122\"]],"
|
|
"AUTHORITY[\"EPSG\",\"4269\"]],"
|
|
"PROJECTION[\"Albers_Conic_Equal_Area\"],"
|
|
"PARAMETER[\"latitude_of_center\",59],"
|
|
"PARAMETER[\"longitude_of_center\",-132.5],"
|
|
"PARAMETER[\"standard_parallel_1\",61.6666666666667],"
|
|
"PARAMETER[\"standard_parallel_2\",68],"
|
|
"PARAMETER[\"false_easting\",500000],"
|
|
"PARAMETER[\"false_northing\",500000],"
|
|
"UNIT[\"METERS\",1],"
|
|
"AXIS[\"Easting\",EAST],AXIS[\"Northing\",NORTH]]");
|
|
|
|
EXPECT_EQ(proj, expect);
|
|
|
|
GDALClose(ds);
|
|
}
|
|
|
|
// Test band data type and NODATA value
|
|
TEST_F(test_gdal_aaigrid, nodata)
|
|
{
|
|
// Index of test file being tested
|
|
const std::size_t fileIdx = 1;
|
|
|
|
std::string file(data_ + SEP);
|
|
file += grids_.at(fileIdx).file_;
|
|
|
|
GDALDatasetH ds = GDALOpen(file.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != ds);
|
|
|
|
GDALRasterBandH band = GDALGetRasterBand(ds, grids_.at(fileIdx).band_);
|
|
ASSERT_TRUE(nullptr != band);
|
|
|
|
const double noData = GDALGetRasterNoDataValue(band, nullptr);
|
|
EXPECT_EQ(noData, -99999);
|
|
|
|
EXPECT_EQ(GDALGetRasterDataType(band), GDT_Float32);
|
|
|
|
GDALClose(ds);
|
|
}
|
|
|
|
// Create simple copy and check
|
|
TEST_F(test_gdal_aaigrid, copy)
|
|
{
|
|
// Index of test file being tested
|
|
const std::size_t fileIdx = 0;
|
|
|
|
std::string src(data_ + SEP);
|
|
src += rasters_.at(fileIdx).file_;
|
|
|
|
GDALDatasetH dsSrc = GDALOpen(src.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != dsSrc);
|
|
|
|
std::string dst(data_tmp_ + SEP);
|
|
dst += rasters_.at(fileIdx).file_;
|
|
dst += ".grd";
|
|
|
|
GDALDatasetH dsDst = nullptr;
|
|
dsDst = GDALCreateCopy(drv_, dst.c_str(), dsSrc, FALSE, nullptr, nullptr,
|
|
nullptr);
|
|
GDALClose(dsSrc);
|
|
ASSERT_TRUE(nullptr != dsDst);
|
|
|
|
std::string proj(GDALGetProjectionRef(dsDst));
|
|
ASSERT_EQ(proj.empty(), false);
|
|
|
|
std::string expect(
|
|
"PROJCS[\"NAD27 / UTM zone 11N\",GEOGCS[\"NAD27\","
|
|
"DATUM[\"North_American_Datum_1927\",SPHEROID[\"Clarke 1866\","
|
|
"6378206.4,294.978698213898,AUTHORITY[\"EPSG\",\"7008\"]],"
|
|
"AUTHORITY[\"EPSG\",\"6267\"]],PRIMEM[\"Greenwich\",0],"
|
|
"UNIT[\"Degree\",0.0174532925199433]],"
|
|
"PROJECTION[\"Transverse_Mercator\"],"
|
|
"PARAMETER[\"latitude_of_origin\",0],"
|
|
"PARAMETER[\"central_meridian\",-117],"
|
|
"PARAMETER[\"scale_factor\",0.9996],"
|
|
"PARAMETER[\"false_easting\",500000],"
|
|
"PARAMETER[\"false_northing\",0],"
|
|
"UNIT[\"metre\",1,AUTHORITY[\"EPSG\",\"9001\"]],"
|
|
"AXIS[\"Easting\",EAST],AXIS[\"Northing\",NORTH]]");
|
|
|
|
ASSERT_EQ(proj, expect);
|
|
|
|
GDALRasterBandH band = GDALGetRasterBand(dsDst, rasters_.at(fileIdx).band_);
|
|
ASSERT_TRUE(nullptr != band);
|
|
|
|
const int xsize = GDALGetRasterXSize(dsDst);
|
|
const int ysize = GDALGetRasterYSize(dsDst);
|
|
const int checksum = GDALChecksumImage(band, 0, 0, xsize, ysize);
|
|
|
|
ASSERT_EQ(checksum, rasters_.at(fileIdx).checksum_);
|
|
|
|
GDALClose(dsDst);
|
|
}
|
|
|
|
// Test subwindow read and the tail recursion problem.
|
|
TEST_F(test_gdal_aaigrid, subwindow_read)
|
|
{
|
|
// Index of test file being tested
|
|
const std::size_t fileIdx = 1;
|
|
|
|
std::string file(data_ + SEP);
|
|
file += grids_.at(fileIdx).file_;
|
|
|
|
GDALDatasetH ds = GDALOpen(file.c_str(), GA_ReadOnly);
|
|
ASSERT_TRUE(nullptr != ds);
|
|
|
|
GDALRasterBandH band = GDALGetRasterBand(ds, grids_.at(fileIdx).band_);
|
|
ASSERT_TRUE(nullptr != band);
|
|
|
|
// Sub-windows size
|
|
const int win[4] = {5, 5, 5, 5};
|
|
// subwindow checksum
|
|
const int winChecksum = 187;
|
|
const int checksum =
|
|
GDALChecksumImage(band, win[0], win[1], win[2], win[3]);
|
|
|
|
EXPECT_EQ(checksum, winChecksum);
|
|
|
|
GDALClose(ds);
|
|
}
|
|
|
|
} // namespace
|