intersection.cuh

/*
 * 版权所有 (c) 2022-2024, NVIDIA CORPORATION。
 *
 * 根据 Apache 许可证版本 2.0 ("许可证") 获得许可；
 * 除非符合许可证的规定，否则您不得使用此文件。
 * 您可以在以下网址获得许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发是基于"原样"原则的，
 * 不附带任何明示或暗示的保证或条件。
 * 有关管理权限和限制的特定语言，请参阅许可证。
 * 许可证下的限制。
 */
 
#pragma once
 
#include <cuspatial/geometry/segment.cuh>
#include <cuspatial/geometry/vec_2d.hpp>
 
#include <rmm/cuda_stream_view.hpp>
#include <rmm/device_uvector.hpp>
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
 
#include <thrust/pair.h>
 
namespace cuspatial {
 
enum IntersectionTypeCode : uint8_t { POINT = 0, LINESTRING = 1 };

template <typename T, typename OffsetType>
struct linestring_intersection_result {
 using point_t = vec_2d<T>;
 using segment_t = segment<T>;
 using types_t = uint8_t;
 using index_t = OffsetType;

 std::unique_ptr<rmm::device_uvector<index_t>> geometry_collection_offset;

 std::unique_ptr<rmm::device_uvector<types_t>> types_buffer;
 std::unique_ptr<rmm::device_uvector<index_t>> offset_buffer;

 std::unique_ptr<rmm::device_uvector<point_t>> points_coords;

 std::unique_ptr<rmm::device_uvector<segment_t>> segments_coords;

 std::unique_ptr<rmm::device_uvector<index_t>> lhs_linestring_id;
 std::unique_ptr<rmm::device_uvector<index_t>> lhs_segment_id;
 std::unique_ptr<rmm::device_uvector<index_t>> rhs_linestring_id;
 std::unique_ptr<rmm::device_uvector<index_t>> rhs_segment_id;
};

template <typename T,
 typename index_t,
 typename MultiLinestringRange1,
 typename MultiLinestringRange2>
linestring_intersection_result<T, index_t> pairwise_linestring_intersection(
 MultiLinestringRange1 multilinestrings1,
 MultiLinestringRange2 multilinestrings2,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource(),
 rmm::cuda_stream_view stream = rmm::cuda_stream_default);
 
} // namespace cuspatial
 
#include <cuspatial/detail/intersection/linestring_intersection.cuh>