spatial_join.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/detail/join/intersection.cuh>
#include <cuspatial/detail/join/traversal.cuh>
#include <cuspatial/iterator_factory.cuh>
#include <cuspatial/point_quadtree.cuh>
#include <cuspatial/traits.hpp>
 
#include <rmm/device_uvector.hpp>
#include <rmm/resource_ref.hpp>
 
#include <iterator>
#include <utility>
 
namespace cuspatial {
 
template <class BoundingBoxIterator,
          class T = typename cuspatial::iterator_vec_base_type<BoundingBoxIterator>>
std::pair<rmm::device_uvector<uint32_t>, rmm::device_uvector<uint32_t>>
join_quadtree_and_bounding_boxes(
  point_quadtree_ref quadtree,
  BoundingBoxIterator bounding_boxes_first,
  BoundingBoxIterator bounding_boxes_last,
  vec_2d<T> const& v_min,
  T scale,
  int8_t max_depth,
  rmm::cuda_stream_view stream      = rmm::cuda_stream_default,
  rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
 
template <class PolyIndexIterator,
          class QuadIndexIterator,
          class PointIndexIterator,
          class PointIterator,
          class MultiPolygonRange,
          class IndexType = iterator_value_type<PointIndexIterator>>
std::pair<rmm::device_uvector<IndexType>, rmm::device_uvector<IndexType>> quadtree_point_in_polygon(
  PolyIndexIterator poly_indices_first,
  PolyIndexIterator poly_indices_last,
  QuadIndexIterator quad_indices_first,
  point_quadtree_ref quadtree,
  PointIndexIterator point_indices_first,
  PointIndexIterator point_indices_last,
  PointIterator points_first,
  MultiPolygonRange polygons,
  rmm::cuda_stream_view stream      = rmm::cuda_stream_default,
  rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
 
template <class LinestringIndexIterator,
          class QuadIndexIterator,
          class PointIndexIterator,
          class PointIterator,
          class MultiLinestringRange,
          typename IndexType = iterator_value_type<PointIndexIterator>,
          typename T         = iterator_vec_base_type<PointIterator>>
std::tuple<rmm::device_uvector<IndexType>, rmm::device_uvector<IndexType>, rmm::device_uvector<T>>
quadtree_point_to_nearest_linestring(
  LinestringIndexIterator linestring_indices_first,
  LinestringIndexIterator linestring_indices_last,
  QuadIndexIterator quad_indices_first,
  point_quadtree_ref quadtree,
  PointIndexIterator point_indices_first,
  PointIndexIterator point_indices_last,
  PointIterator points_first,
  MultiLinestringRange linestrings,
  rmm::cuda_stream_view stream      = rmm::cuda_stream_default,
  rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());
 
}  // namespace cuspatial
 
#include <cuspatial/detail/join/quadtree_bbox_filtering.cuh>
#include <cuspatial/detail/join/quadtree_point_in_polygon.cuh>
#include <cuspatial/detail/join/quadtree_point_to_nearest_linestring.cuh>