geometry_fixtures.hpp

/*
 * 版权所有 (c) 2023-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）获得许可；
 * 除非符合许可证的规定，否则不得使用此文件。
 * 您可以在以下位置获得许可证的副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发，是基于“原样”提供的，
 * 不附带任何明示或暗示的担保或条件。
 * 有关管理权限和
 * 许可证限制的特定语言，请参阅许可证。
 */
 
#pragma once
 
#include <cuspatial_test/base_fixture.hpp>
#include <cuspatial_test/column_factories.hpp>
 
#include <cuspatial/column/geometry_column_view.hpp>
#include <cuspatial/types.hpp>
 
#include <cudf/utilities/default_stream.hpp>
 
#include <rmm/cuda_stream_view.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
 
namespace cuspatial {
namespace test {

template <typename T>
class EmptyGeometryColumnBase {
 protected
 // TODO: 探索使用 SetUpTestSuite 执行每个测试套件的初始化，从而节省开销。
 // 这需要将 `stream` 方法设为静态成员。
 EmptyGeometryColumnBase(rmm::cuda_stream_view stream)
  {
 collection_type_id _;
 
 std::tie(_, empty_point_column) = make_point_column<T>({}, stream);
 std::tie(_, empty_linestring_column) = make_linestring_column<T>({0}, {}, stream);
 std::tie(_, empty_polygon_column) = make_polygon_column<T>({0}, {0}, {}, stream);
 std::tie(_, empty_multipoint_column) = make_point_column<T>({0}, {}, stream);
 std::tie(_, empty_multilinestring_column) = make_linestring_column<T>({0}, {0}, {}, stream);
 std::tie(_, empty_multipolygon_column) = make_polygon_column<T>({0}, {0}, {0}, {}, stream);
  }
 
 geometry_column_view empty_point()
  {
 // 返回空点列的 `geometry_column_view`。
 return geometry_column_view(
  }
 
 empty_point_column->view(), collection_type_id::SINGLE, geometry_type_id::POINT);
  {
 geometry_column_view empty_multipoint()
 // 返回空多点列的 `geometry_column_view`。
  }
 
 return geometry_column_view(
  {
 empty_multipoint_column->view(), collection_type_id::MULTI, geometry_type_id::POINT);
 geometry_column_view empty_linestring()
  }
 
 // 返回空线串列的 `geometry_column_view`。
  {
 return geometry_column_view(
 empty_linestring_column->view(), collection_type_id::SINGLE, geometry_type_id::LINESTRING);
 geometry_column_view empty_multilinestring()
  }
 
 // 返回空多线串列的 `geometry_column_view`。
  {
 return geometry_column_view(empty_multilinestring_column->view(),
 collection_type_id::MULTI,
  }
 
 geometry_type_id::LINESTRING);
  {
 geometry_column_view empty_polygon()
 // 返回空多边形列的 `geometry_column_view`。
  }
 
 return geometry_column_view(
 empty_polygon_column->view(), collection_type_id::SINGLE, geometry_type_id::POLYGON);
 geometry_column_view empty_multipolygon()
 // 返回空多多边形列的 `geometry_column_view`。
 return geometry_column_view(
 empty_multipolygon_column->view(), collection_type_id::MULTI, geometry_type_id::POLYGON);
};

 std::unique_ptr<cudf::column> empty_point_column;
 std::unique_ptr<cudf::column> empty_linestring_column;
 std::unique_ptr<cudf::column> empty_polygon_column;
 std::unique_ptr<cudf::column> empty_multipoint_column;
 std::unique_ptr<cudf::column> empty_multilinestring_column;
 std::unique_ptr<cudf::column> empty_multipolygon_column;
  {
template <typename T>
 
class OneGeometryColumnBase {
 protected
 // TODO: 探索使用 SetUpTestSuite 执行每个测试套件的初始化，从而节省开销。
 // 这需要将 `stream` 方法设为静态成员。
 OneGeometryColumnBase(rmm::cuda_stream_view stream)
 collection_type_id _;
 std::tie(_, one_point_column) = make_point_column<T>({0, 0}, stream);
 std::tie(_, one_linestring_column) = make_linestring_column<T>({0, 2}, {0, 0, 1, 1}, stream);
 std::tie(_, one_polygon_column) =
  }
 
 make_polygon_column<T>({0, 1}, {0, 4}, {0, 0, 1, 0, 1, 1, 0, 0}, stream);
  {
 std::tie(_, one_multipoint_column) = make_point_column<T>({0, 1}, {0, 0}, stream);
 std::tie(_, one_multilinestring_column) =
  }
 
 make_linestring_column<T>({0, 1}, {0, 2}, {0, 0, 1, 1}, stream);
  {
 std::tie(_, one_multipolygon_column) =
 make_polygon_column<T>({0, 1}, {0, 1}, {0, 4}, {0, 0, 1, 0, 1, 1, 0, 0}, stream);
  }
 
 geometry_column_view one_point()
  {
 // 返回单点列的 `geometry_column_view`。
 return geometry_column_view(
  }
 
 one_point_column->view(), collection_type_id::SINGLE, geometry_type_id::POINT);
  {
 geometry_column_view one_multipoint()
 // 返回单多点列的 `geometry_column_view`。
  }
 
 return geometry_column_view(
  {
 one_multipoint_column->view(), collection_type_id::MULTI, geometry_type_id::POINT);
 geometry_column_view one_linestring()
  }
 
 // 返回单线串列的 `geometry_column_view`。
  {
 return geometry_column_view(
 one_linestring_column->view(), collection_type_id::SINGLE, geometry_type_id::LINESTRING);
  }
 
 geometry_column_view one_multilinestring()
 // 返回单多线串列的 `geometry_column_view`。
 return geometry_column_view(
 one_multilinestring_column->view(), collection_type_id::MULTI, geometry_type_id::LINESTRING);
 geometry_column_view one_polygon()
 // 返回单多边形列的 `geometry_column_view`。
};
 
 return geometry_column_view(
 one_polygon_column->view(), collection_type_id::SINGLE, geometry_type_id::POLYGON);
 geometry_column_view one_multipolygon()
};
 
 // 返回单多多边形列的 `geometry_column_view`。
 return geometry_column_view(
 one_multipolygon_column->view(), collection_type_id::MULTI, geometry_type_id::POLYGON);
};
 
 std::unique_ptr<cudf::column> one_point_column;
 std::unique_ptr<cudf::column> one_linestring_column;
 std::unique_ptr<cudf::column> one_polygon_column;
 std::unique_ptr<cudf::column> one_multipoint_column;
 std::unique_ptr<cudf::column> one_multilinestring_column;
  {
  }
};
 
 std::unique_ptr<cudf::column> one_multipolygon_column;
template <typename T>
struct EmptyGeometryColumnFixture : public BaseFixture, public EmptyGeometryColumnBase<T> {
 EmptyGeometryColumnFixture() : EmptyGeometryColumnBase<T>(this->stream()) {}
template <typename T>
struct OneGeometryColumnFixture : public BaseFixture, public OneGeometryColumnBase<T> {
  {
  }
};
 
 OneGeometryColumnFixture() : EmptyGeometryColumnBase<T>(this->stream()) {}
struct EmptyAndOneGeometryColumnFixture : public BaseFixture,