concentric_cylindrical2D.hpp

// This file is part of the ACTS project.
//
// Copyright (C) 2016 CERN for the benefit of the ACTS project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
#pragma once
 
// Project include(s)
#include "detray/definitions/algebra.hpp"
#include "detray/definitions/detail/qualifiers.hpp"
#include "detray/definitions/math.hpp"
 
namespace detray {

template <concepts::algebra algebra_t>
struct concentric_cylindrical2D {
  using algebra_type = algebra_t;
  using scalar_type = dscalar<algebra_t>;
  using point2_type = dpoint2D<algebra_t>;
  using point3_type = dpoint3D<algebra_t>;
  using vector3_type = dvector3D<algebra_t>;
  using transform3_type = dtransform3D<algebra_t>;

  using loc_point = point2_type;

  DETRAY_HOST_DEVICE
  static inline point3_type global_to_local_3D(const transform3_type & /*trf*/,
                                               const point3_type &p,
                                               const vector3_type & /*dir*/) {
    return {vector::phi(p), p[2], vector::perp(p)};
  }

  DETRAY_HOST_DEVICE
  static inline loc_point global_to_local(const transform3_type & /*trf*/,
                                          const point3_type &p,
                                          const vector3_type & /*dir*/) {
    return {vector::phi(p), p[2]};
  }

  DETRAY_HOST_DEVICE static inline point3_type local_to_global(
      const transform3_type & /*trf*/, const point3_type &p) {
    const scalar_type x{p[2] * math::cos(p[0])};
    const scalar_type y{p[2] * math::sin(p[0])};
    const scalar_type z{p[1]};
 
    return point3_type{x, y, z};
  }

  template <typename mask_t>
  DETRAY_HOST_DEVICE static inline point3_type local_to_global(
      const transform3_type & /*trf*/, const mask_t &mask, const loc_point &p,
      const vector3_type & /*dir*/) {
    const scalar_type r{mask[mask_t::shape::e_r]};
    const scalar_type x{r * math::cos(p[0])};
    const scalar_type y{r * math::sin(p[0])};
    const scalar_type z{p[1]};
 
    return point3_type{x, y, z};
  }

  template <typename mask_t>
  DETRAY_HOST_DEVICE static inline vector3_type normal(
      const transform3_type & /*trf*/, const point2_type &p,
      const mask_t & /*mask*/) {
    // normal vector in global coordinates (concentric cylinders have no
    // rotation)
    return {math::cos(p[0]), math::sin(p[0]), static_cast<scalar_type>(0)};
  }

  DETRAY_HOST_DEVICE static inline vector3_type normal(
      const transform3_type & /*trf*/, const point3_type &p) {
    // normal vector in global coordinates (concentric cylinders have no
    // rotation)
    return {math::cos(p[0]), math::sin(p[0]), static_cast<scalar_type>(0)};
  }
 
};  // struct concentric_cylindrical2D
 
}  // namespace detray