segment.cuh

前往此文件的文档。

/*
 * 版权所有 (c) 2022-2023, NVIDIA CORPORATION。
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可在以下网址获得许可证的副本
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发时，按“原样”分发，
 * 不附带任何明示或暗示的保证或条件。
 * 请参阅许可证，了解管辖权和
 * 限制条件的特定语言。
 */
 
#pragma once
 
#include <cuspatial/cuda_utils.hpp>
#include <cuspatial/geometry/vec_2d.hpp>
 
#include <thrust/device_reference.h>
 
#include <iostream>
namespace cuspatial {


 
template <typename T, typename Vertex = cuspatial::vec_2d<T>>
class alignas(sizeof(Vertex)) segment {
 public
 using value_type = T;
 Vertex v1;
 Vertex v2;

 segment<T> CUSPATIAL_HOST_DEVICE translate(Vertex const& v) const
  {
 return segment<T>{v1 + v, v2 + v};
  }

 Vertex CUSPATIAL_HOST_DEVICE center() const { return midpoint(v1, v2); }

 T CUSPATIAL_HOST_DEVICE length2() const { return dot(v2 - v1, v2 - v1); }

 T CUSPATIAL_HOST_DEVICE slope() { return (v2.y - v1.y) / (v2.x - v1.x); }

 Vertex CUSPATIAL_HOST_DEVICE lower_left() { return v1 < v2 ? v1 : v2; }

 bool CUSPATIAL_HOST_DEVICE collinear(segment<T> const& other)
  {
 return (v1.x - v2.x) * (other.v1.y - other.v2.y) == (v1.y - v2.y) * (other.v1.x - other.v2.x);
  }
 
 private
 friend std::ostream& operator<<(std::ostream& os, segment<T> const& seg)
  {
 return os << seg.v1 << " -> " << seg.v2;
  }
};
 
// deduction guide, enables CTAD
template <typename T>
segment(vec_2d<T> a, vec_2d<T> b) -> segment<T, vec_2d<T>>;
 
template <typename T>
segment(thrust::device_reference<vec_2d<T>> a, thrust::device_reference<vec_2d<T>> b)
 -> segment<T, vec_2d<T>>;
} // namespace cuspatial