Skip to content

Commit

Permalink
Merge pull request #10896 from rouault/gdal_typetraits
Browse files Browse the repository at this point in the history 
Add public gdal_typetraits.h header with gdal::CXXTypeTraits<T> and gdal::GDALDataTypeTraits<T>
  • Loading branch information
@rouault
rouault authored Oct 23, 2024
2 parents 38dd322 + 591bed8 commit 62513a3
Show file tree
Hide file tree
Showing 4 changed files with 709 additions and 0 deletions.
1 change: 1 addition & 0 deletions autotest/cpp/CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@ add_executable(
test_gdal_dted.cpp
test_gdal_gtiff.cpp
test_gdal_pixelfn.cpp
test_gdal_typetraits.cpp
test_ogr.cpp
test_ogr_organize_polygons.cpp
test_ogr_geometry_stealing.cpp
Expand Down
197 changes: 197 additions & 0 deletions autotest/cpp/test_gdal_typetraits.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,197 @@
// SPDX-License-Identifier: MIT
// Copyright 2024, Even Rouault <even.rouault at spatialys.com>

#include "gdal_unit_test.h"

#include "gdal_typetraits.h"

#include "gtest_include.h"

namespace
{

struct test_gdal_typetraits : public ::testing::Test
{
};

TEST_F(test_gdal_typetraits, CXXTypeTraits)
{
static_assert(gdal::CXXTypeTraits<int8_t>::gdal_type == GDT_Int8);
static_assert(gdal::CXXTypeTraits<int8_t>::size == 1);
EXPECT_EQ(
gdal::CXXTypeTraits<int8_t>::GetExtendedDataType().GetNumericDataType(),
GDT_Int8);
static_assert(gdal::CXXTypeTraits<uint8_t>::gdal_type == GDT_Byte);
static_assert(gdal::CXXTypeTraits<uint8_t>::size == 1);
EXPECT_EQ(gdal::CXXTypeTraits<uint8_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Byte);
static_assert(gdal::CXXTypeTraits<int16_t>::gdal_type == GDT_Int16);
static_assert(gdal::CXXTypeTraits<int16_t>::size == 2);
EXPECT_EQ(gdal::CXXTypeTraits<int16_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int16);
static_assert(gdal::CXXTypeTraits<uint16_t>::gdal_type == GDT_UInt16);
static_assert(gdal::CXXTypeTraits<uint16_t>::size == 2);
EXPECT_EQ(gdal::CXXTypeTraits<uint16_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_UInt16);
static_assert(gdal::CXXTypeTraits<int32_t>::gdal_type == GDT_Int32);
static_assert(gdal::CXXTypeTraits<int32_t>::size == 4);
EXPECT_EQ(gdal::CXXTypeTraits<int32_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int32);
static_assert(gdal::CXXTypeTraits<uint32_t>::gdal_type == GDT_UInt32);
static_assert(gdal::CXXTypeTraits<uint32_t>::size == 4);
EXPECT_EQ(gdal::CXXTypeTraits<uint32_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_UInt32);
static_assert(gdal::CXXTypeTraits<int64_t>::gdal_type == GDT_Int64);
static_assert(gdal::CXXTypeTraits<int64_t>::size == 8);
EXPECT_EQ(gdal::CXXTypeTraits<int64_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int64);
static_assert(gdal::CXXTypeTraits<uint64_t>::gdal_type == GDT_UInt64);
static_assert(gdal::CXXTypeTraits<uint64_t>::size == 8);
EXPECT_EQ(gdal::CXXTypeTraits<uint64_t>::GetExtendedDataType()
.GetNumericDataType(),
GDT_UInt64);
static_assert(gdal::CXXTypeTraits<float>::gdal_type == GDT_Float32);
static_assert(gdal::CXXTypeTraits<float>::size == 4);
EXPECT_EQ(
gdal::CXXTypeTraits<float>::GetExtendedDataType().GetNumericDataType(),
GDT_Float32);
static_assert(gdal::CXXTypeTraits<double>::gdal_type == GDT_Float64);
static_assert(gdal::CXXTypeTraits<double>::size == 8);
EXPECT_EQ(
gdal::CXXTypeTraits<double>::GetExtendedDataType().GetNumericDataType(),
GDT_Float64);
static_assert(gdal::CXXTypeTraits<std::complex<float>>::gdal_type ==
GDT_CFloat32);
static_assert(gdal::CXXTypeTraits<std::complex<float>>::size == 8);
EXPECT_EQ(gdal::CXXTypeTraits<std::complex<float>>::GetExtendedDataType()
.GetNumericDataType(),
GDT_CFloat32);
static_assert(gdal::CXXTypeTraits<std::complex<double>>::gdal_type ==
GDT_CFloat64);
static_assert(gdal::CXXTypeTraits<std::complex<double>>::size == 16);
EXPECT_EQ(gdal::CXXTypeTraits<std::complex<double>>::GetExtendedDataType()
.GetNumericDataType(),
GDT_CFloat64);
static_assert(gdal::CXXTypeTraits<std::string>::size == 0);
EXPECT_EQ(
gdal::CXXTypeTraits<std::string>::GetExtendedDataType().GetClass(),
GEDTC_STRING);
}

TEST_F(test_gdal_typetraits, GDALDataTypeTraits)
{
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Byte>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Byte);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Byte>::type, uint8_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_Byte>::size == 1);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Int8>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int8);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Int8>::type, int8_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_Int8>::size == 1);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Int16>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int16);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Int16>::type, int16_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_Int16>::size == 2);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_UInt16>::GetExtendedDataType()
.GetNumericDataType(),
GDT_UInt16);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_UInt16>::type, uint16_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_UInt16>::size == 2);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Int32>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int32);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Int32>::type, int32_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_Int32>::size == 4);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_UInt32>::GetExtendedDataType()
.GetNumericDataType(),
GDT_UInt32);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_UInt32>::type, uint32_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_UInt32>::size == 4);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Int64>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Int64);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Int64>::type, int64_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_Int64>::size == 8);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_UInt64>::GetExtendedDataType()
.GetNumericDataType(),
GDT_UInt64);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_UInt64>::type, uint64_t>);
static_assert(gdal::GDALDataTypeTraits<GDT_UInt64>::size == 8);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Float32>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Float32);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Float32>::type, float>);
static_assert(gdal::GDALDataTypeTraits<GDT_Float32>::size == 4);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_Float64>::GetExtendedDataType()
.GetNumericDataType(),
GDT_Float64);
static_assert(
std::is_same_v<gdal::GDALDataTypeTraits<GDT_Float64>::type, double>);
static_assert(gdal::GDALDataTypeTraits<GDT_Float64>::size == 8);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_CInt16>::GetExtendedDataType()
.GetNumericDataType(),
GDT_CInt16);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_CInt32>::GetExtendedDataType()
.GetNumericDataType(),
GDT_CInt32);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_CFloat32>::GetExtendedDataType()
.GetNumericDataType(),
GDT_CFloat32);
static_assert(std::is_same_v<gdal::GDALDataTypeTraits<GDT_CFloat32>::type,
std::complex<float>>);
static_assert(gdal::GDALDataTypeTraits<GDT_CFloat32>::size == 8);
EXPECT_EQ(gdal::GDALDataTypeTraits<GDT_CFloat64>::GetExtendedDataType()
.GetNumericDataType(),
GDT_CFloat64);
static_assert(std::is_same_v<gdal::GDALDataTypeTraits<GDT_CFloat64>::type,
std::complex<double>>);
static_assert(gdal::GDALDataTypeTraits<GDT_CFloat64>::size == 16);
}

TEST_F(test_gdal_typetraits, GetOGRFieldType)
{
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Byte), OFTInteger);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Int8), OFTInteger);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Int16), OFTInteger);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Int32), OFTInteger);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_UInt16), OFTInteger);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_UInt32), OFTInteger64);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Int64), OFTInteger64);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_UInt64), OFTReal);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Float32), OFTReal);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Float64), OFTReal);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_CInt16), OFTMaxType);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_CInt32), OFTMaxType);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_CFloat32), OFTMaxType);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_CFloat64), OFTMaxType);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_Unknown), OFTMaxType);
EXPECT_EQ(gdal::GetOGRFieldType(GDT_TypeCount), OFTMaxType);

EXPECT_EQ(gdal::GetOGRFieldType(GDALExtendedDataType::Create(GDT_Byte)),
OFTInteger);
EXPECT_EQ(gdal::GetOGRFieldType(GDALExtendedDataType::CreateString()),
OFTString);
EXPECT_EQ(
gdal::GetOGRFieldType(GDALExtendedDataType::Create("compound", 0, {})),
OFTMaxType);
}

} // namespace
1 change: 1 addition & 0 deletions gcore/CMakeLists.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -194,6 +194,7 @@ target_public_header(
gdalgeorefpamdataset.h
gdal_mdreader.h
gdalsubdatasetinfo.h
gdal_typetraits.h
)

set(GDAL_DATA_FILES
Expand Down
Loading

0 comments on commit 62513a3

Please sign in to comment.