bin_fillers.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/builders/detail/bin_association.hpp"
#include "detray/core/concepts.hpp"
#include "detray/utils/grid/axis.hpp"
#include "detray/utils/grid/concepts.hpp"
#include "detray/utils/grid/populators.hpp"
 
// System include(s)
#include <cassert>
#include <vector>
 
namespace detray {

struct fill_by_bin {
  template <std::size_t DIM, typename value_t>
  struct bin_data {
    detray::axis::multi_bin<DIM> local_bin_idx;
    value_t single_element;
  };

  template <concepts::grid grid_t>
  using bin_data_type = bin_data<grid_t::dim, typename grid_t::value_type>;

  template <concepts::grid grid_t, typename volume_t,
            typename surface_container_t, typename mask_container,
            typename transform_container, typename context_t, typename... Args>
  DETRAY_HOST auto operator()(grid_t &grid, const volume_t & /*unused*/,
                              const surface_container_t & /*unused*/,
                              const mask_container & /*unused*/,
                              const context_t /*unused*/,
                              std::vector<bin_data_type<grid_t>> &bins) const
      -> void {
    for (const bin_data_type<grid_t> &bd : bins) {
      grid.template populate<attach<>>(bd.local_bin_idx, bd.single_element);
    }
  }
};

struct fill_by_pos {
  template <concepts::surface_grid grid_t, typename volume_t,
            typename surface_container_t, typename mask_container,
            typename transform_container, typename context_t, typename... Args>
  DETRAY_HOST auto operator()(grid_t &grid, const volume_t &vol,
                              const surface_container_t &surfaces,
                              const transform_container &transforms,
                              const mask_container & /*masks*/,
                              const context_t ctx, Args &&.../*unused*/) const
      -> void {
    // Fill the volumes surfaces into the grid
    for (const auto &sf : surfaces) {
      // no portals in grids allowed
      if (sf.volume() == vol.index() && sf.is_sensitive()) {
        const auto &sf_trf = transforms.at(sf.transform(), ctx);
        const auto &t = sf_trf.translation();
 
        // transform to axis coordinate system
        const auto loc_pos = grid.project(vol.transform(), t, t);
 
        // Populate
        grid.template populate<attach<>>(loc_pos, sf);
      }
    }
  }
};

struct bin_associator {
  template <concepts::detector detector_t, typename volume_type,
            concepts::surface_grid grid_t, typename... Args>
  DETRAY_HOST auto operator()(grid_t &grid, detector_t &det,
                              const volume_type &vol,
                              const typename detector_t::geometry_context ctx,
                              Args &&.../*unused*/) const -> void {
    this->operator()(grid, det.surfaces(vol), det.mask_store(),
                     det.transform_store(), ctx);
  }

  template <concepts::surface_grid grid_t, typename volume_t,
            typename surface_container_t, typename mask_container,
            typename transform_container, typename context_t, typename... Args>
  DETRAY_HOST auto operator()(grid_t &grid, const volume_t & /*unused*/,
                              const surface_container_t &surfaces,
                              const transform_container &transforms,
                              const mask_container &masks, const context_t ctx,
                              Args &&.../*unused*/) const -> void {
    using scalar_t = dscalar<typename grid_t::algebra_type>;
 
    // Fill the surfaces into the grid by matching their contour onto the
    // grid bins
    bin_association(ctx, surfaces, transforms, masks, grid,
                    darray<scalar_t, 2>{0.1f, 0.1f}, false);
  }
};
 
}  // namespace detray