populators.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/algorithms.hpp"
#include "detray/definitions/detail/qualifiers.hpp"
#include "detray/definitions/indexing.hpp"
#include "detray/utils/invalid_values.hpp"
 
namespace detray {

template <bool kSORT = false>
struct replace {
  template <typename bin_t, typename content_t>
  DETRAY_HOST_DEVICE void operator()(bin_t &&bin, content_t &&entry) const {
    std::forward<bin_t>(bin).init(std::forward<content_t>(entry));
 
    // Optionally sort the bin content
    if constexpr (kSORT) {
      detray::sequential_sort(std::forward<bin_t>(bin).begin(),
                              std::forward<bin_t>(bin).end());
    }
  }
};

template <bool kSORT = false>
struct attach {
  template <typename bin_t, typename entry_t>
  DETRAY_HOST_DEVICE void operator()(bin_t &&bin, entry_t &&entry) const {
    std::forward<bin_t>(bin).push_back(std::forward<entry_t>(entry));
 
    // Optionally sort the bin content
    if constexpr (kSORT) {
      detray::sequential_sort(std::forward<bin_t>(bin).begin(),
                              std::forward<bin_t>(bin).end());
    }
  }
};

template <bool kSORT = false>
struct complete {
  template <typename bin_t, typename entry_t>
  DETRAY_HOST_DEVICE void operator()(bin_t &&bin, entry_t &&entry) const {
    for (dindex i{std::forward<bin_t>(bin).size()};
         i < std::forward<bin_t>(bin).capacity(); ++i) {
      std::forward<bin_t>(bin).push_back(std::forward<entry_t>(entry));
    }
 
    // Optionally sort the bin content
    if constexpr (kSORT) {
      detray::sequential_sort(std::forward<bin_t>(bin).begin(),
                              std::forward<bin_t>(bin).end());
    }
  }
};
 
}  // namespace detray