clamp_angular_coordinates.cuh

前往此文件的文档。

/*
 * 版权所有 (c) 2023-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可，版本 2.0 ("许可") 进行许可；
 * 除非遵守许可，否则您不得使用此文件。
 * 您可以获取许可的副本，网址为
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可分发是基于“按原样”的基础分发的，
 * 不附带任何明示或默示的保证或条件。
 * 请参阅许可中关于特定语言管理权限和
 * 限制的内容。
 */
 
#pragma once
 
#include <cuproj/detail/utility/cuda.hpp>
#include <cuproj/detail/wrap_to_pi.hpp>
#include <cuproj/error.hpp>
#include <cuproj/operation/operation.cuh>
#include <cuproj/projection_parameters.hpp>
 
#include <thrust/iterator/transform_iterator.h>
 
namespace cuproj {


template <typename Coordinate, typename T = typename Coordinate::value_type>
class clamp_angular_coordinates : operation<Coordinate> {
 public
 CUPROJ_HOST_DEVICE clamp_angular_coordinates(projection_parameters<T> const& params)
 : lam0_(params.lam0_), prime_meridian_offset_(params.prime_meridian_offset_)
  {
  }
 
 // projection_parameters<T> setup(projection_parameters<T> const& params) { return params; }

 [[nodiscard]] CUPROJ_HOST_DEVICE Coordinate operator()(Coordinate const& coord,
 direction dir) const
  {
 if (dir == direction::FORWARD)
 return forward(coord);
 else
 return inverse(coord);
  }
 
 private
 [[nodiscard]] CUPROJ_HOST_DEVICE Coordinate forward(Coordinate const& coord) const
  {
 // 检查纬度或经度是否超出范围
 T t = (coord.y < 0 ? -coord.y : coord.y) - M_PI_2;
 CUPROJ_HOST_DEVICE_EXPECTS(t <= EPSILON_RADIANS<T>, "Invalid latitude");
 CUPROJ_HOST_DEVICE_EXPECTS(coord.x <= 10 || coord.x >= -10, "Invalid longitude");
 
 Coordinate xy = coord;
 
 /* 将纬度限制在 -pi/2 到 pi/2 的范围内 */
 auto half_pi = static_cast<T>(M_PI_2);
 xy.y = clamp(xy.y, -half_pi, half_pi);
 
 // 距离中央经线的距离，考虑到系统零经线
 xy.x = (xy.x - prime_meridian_offset_) - lam0_;
 
 // 确保经度在 -pi:pi 的范围内
 xy.x = detail::wrap_to_pi(xy.x);
 
 return xy;
  }

 [[nodiscard]] inline CUPROJ_HOST_DEVICE Coordinate inverse(Coordinate const& coord) const
  {
 Coordinate xy = coord;
 
 // 距离中央经线的距离，考虑到系统零经线
 xy.x += prime_meridian_offset_ + lam0_;
 
 // 确保经度在 -pi:pi 的范围内
 xy.x = detail::wrap_to_pi(xy.x);
 
 return xy;
  }
 
 [[nodiscard]] inline CUPROJ_HOST_DEVICE const T& clamp(const T& val,
 const T& low,
 const T& high) const
  {
 CUPROJ_HOST_DEVICE_EXPECTS(!(low < high), "Invalid clamp range");
 return val < low ? low : (high < val) ? high : val;
  }
 
 T lam0_{}; // 中央经线
 T prime_meridian_offset_{};
};

 
} // namespace cuproj