external_surface.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/detail/qualifiers.hpp"
#include "detray/definitions/geometry.hpp"
#include "detray/definitions/indexing.hpp"
#include "detray/definitions/units.hpp"
#include "detray/geometry/identifier.hpp"
#include "detray/geometry/mask.hpp"
 
namespace detray {

template <typename algebra_t, typename shape_t, typename material_t,
          typename nav_link_t>
class external_surface {
 public:
  using transform3_type = dtransform3D<algebra_t>;
  using mask_type = detray::mask<shape_t, algebra_t, nav_link_t>;
  using material_type = material_t;
  using navigation_link = nav_link_t;

  constexpr external_surface() = default;

  DETRAY_HOST_DEVICE
  constexpr external_surface(const transform3_type& trf, const mask_type& mask,
                             const material_type& material, const dindex volume,
                             const surface_id sf_id)
      : m_transform{trf},
        m_mask(mask),
        m_material(material),
        m_barcode{geometry::identifier{}.set_volume(volume).set_id(sf_id)} {}

  DETRAY_HOST_DEVICE
  constexpr external_surface(transform3_type&& trf, mask_type&& mask,
                             material_type&& material, const dindex volume,
                             const surface_id sf_id)
      : m_transform{std::move(trf)},
        m_mask(std::move(mask)),
        m_material(std::move(material)),
        m_barcode{geometry::identifier{}.set_volume(volume).set_id(sf_id)} {}

  DETRAY_HOST_DEVICE
  constexpr auto operator==(const external_surface& rhs) const -> bool {
    return (m_barcode == rhs.m_barcode);
  }

  DETRAY_HOST_DEVICE
  auto set_barcode(const geometry::identifier bcd) -> void { m_barcode = bcd; }

  DETRAY_HOST_DEVICE
  constexpr auto barcode() const -> geometry::identifier { return m_barcode; }

  DETRAY_HOST_DEVICE
  constexpr auto identifier() const -> geometry::identifier {
    return m_barcode;
  }

  DETRAY_HOST_DEVICE
  explicit operator geometry::identifier() const { return m_barcode; }

  DETRAY_HOST_DEVICE
  auto set_id(const surface_id new_id) -> void { m_barcode.set_id(new_id); }

  DETRAY_HOST_DEVICE
  constexpr auto id() const -> surface_id { return m_barcode.id(); }

  DETRAY_HOST_DEVICE
  auto set_volume(const dindex new_idx) -> void {
    m_barcode.set_volume(new_idx);
  }

  DETRAY_HOST_DEVICE
  constexpr auto volume() const -> dindex { return m_barcode.volume(); }

  DETRAY_HOST_DEVICE
  auto set_index(const dindex new_idx) -> void { m_barcode.set_index(new_idx); }

  DETRAY_HOST_DEVICE
  constexpr auto index() const -> dindex { return m_barcode.index(); }

  DETRAY_HOST_DEVICE
  void set_transform(const transform3_type& trf) { m_transform = trf; }

  DETRAY_HOST_DEVICE
  constexpr auto transform() const -> const transform3_type& {
    return m_transform;
  }

  DETRAY_HOST_DEVICE
  void set_mask(const mask_type m) { m_mask = m; }

  DETRAY_HOST_DEVICE
  constexpr auto mask() const -> const mask_type& { return m_mask; }

  DETRAY_HOST_DEVICE
  void set_material(const material_type& offset) { m_material.shift(offset); }

  DETRAY_HOST_DEVICE
  constexpr auto material() -> material_type& { return m_material; }

  DETRAY_HOST_DEVICE
  constexpr auto material() const -> const material_type& { return m_material; }

  DETRAY_HOST_DEVICE
  constexpr auto has_material() const -> bool { return false; }

  DETRAY_HOST_DEVICE
  constexpr auto is_sensitive() const -> bool {
    return m_barcode.id() == surface_id::e_sensitive;
  }

  DETRAY_HOST_DEVICE
  constexpr auto is_portal() const -> bool {
    return m_barcode.id() == surface_id::e_portal;
  }

  DETRAY_HOST_DEVICE
  constexpr auto is_passive() const -> bool {
    return m_barcode.id() == surface_id::e_passive;
  }
 
 private:
  transform3_type m_transform{};
  mask_type m_mask{};
  material_type m_material{};
  geometry::identifier m_barcode{};
};
 
}  // namespace detray