points_in_range.cuh

/*
 * Copyright (c) 2022, 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/vec_2d.hpp>
 
#include <rmm/cuda_stream_view.hpp>
 
#include <thrust/iterator/iterator_traits.h>
 
namespace cuspatial {
 
template <class InputIt, class T>
typename thrust::iterator_traits<InputIt>::difference_type count_points_in_range(
  vec_2d<T> vertex_1,
  vec_2d<T> vertex_2,
  InputIt points_first,
  InputIt points_last,
  rmm::cuda_stream_view stream = rmm::cuda_stream_default);
 
template <class InputIt, class OutputIt, class T>
OutputIt copy_points_in_range(vec_2d<T> vertex_1,
                              vec_2d<T> vertex_2,
                              InputIt points_first,
                              InputIt points_last,
                              OutputIt output_points_first,
                              rmm::cuda_stream_view stream = rmm::cuda_stream_default);
 
}  // namespace cuspatial
 
#include <cuspatial/detail/points_in_range.cuh>