distance.hpp

转到此文件的文档。

/*
 * 版权所有 (c) 2020-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非符合许可证，否则不得使用此文件。
 * 您可在以下网址获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件
 * 均按“原样”提供，
 * 不附带任何明示或暗示的担保或条件。
 * 请参阅许可证以了解特定的语言管理权限和
 * 限制。
 */
 
#pragma once
 
#include <cuspatial/column/geometry_column_view.hpp>
#include <cuspatial/constants.hpp>
 
#include <cudf/column/column.hpp>
#include <cudf/column/column_view.hpp>
#include <cudf/table/table_view.hpp>
#include <cudf/utilities/span.hpp>
 
#include <rmm/mr/device/per_device_resource.hpp>
#include <rmm/resource_ref.hpp>
 
#include <optional>
 
namespace cuspatial {


std::unique_ptr<cudf::column> haversine_distance(
 cudf::column_view const& a_lon,
 cudf::column_view const& a_lat,
 cudf::column_view const& b_lon,
 cudf::column_view const& b_lat,
 double const radius = EARTH_RADIUS_KM,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

std::pair<std::unique_ptr<cudf::column>, cudf::table_view> directed_hausdorff_distance(
 cudf::column_view const& xs,
 cudf::column_view const& ys,
 cudf::column_view const& space_offsets,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

std::unique_ptr<cudf::column> pairwise_point_distance(
 geometry_column_view const& multipoints1,
 geometry_column_view const& multipoints2,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

std::unique_ptr<cudf::column> pairwise_point_linestring_distance(
 geometry_column_view const& multipoints,
 geometry_column_view const& multilinestrings,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

 
std::unique_ptr<cudf::column> pairwise_point_polygon_distance(
 geometry_column_view const& multipoints,
 geometry_column_view const& multipolygons,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

std::unique_ptr<cudf::column> pairwise_linestring_distance(
 geometry_column_view const& multilinestrings1,
 geometry_column_view const& multilinestrings2,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

std::unique_ptr<cudf::column> pairwise_linestring_polygon_distance(
 geometry_column_view const& multilinestrings,
 geometry_column_view const& multipolygons,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

std::unique_ptr<cudf::column> pairwise_polygon_distance(
 geometry_column_view const& multipolygons1,
 geometry_column_view const& multipolygons2,
 rmm::device_async_resource_ref mr = rmm::mr::get_current_device_resource());

 
} // namespace cuspatial