intersection.cuh

/*
 * Copyright (c) 2022-2024, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#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>