host.ipp

/* VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2023-2025 CERN for the benefit of the ACTS project
 *
 * Mozilla Public License Version 2.0
 */
#pragma once
 
// Local include(s).
#include "vecmem/edm/details/host_traits.hpp"
#include "vecmem/edm/details/schema_traits.hpp"
 
// System include(s).
#include <string>
#include <tuple>
#include <type_traits>
 
namespace vecmem {
namespace edm {
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST host<schema<VARTYPES...>, INTERFACE>::host(
    memory_resource& resource)
    : m_data{details::host_alloc<VARTYPES>::make(resource)...},
      m_resource{resource} {}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST std::size_t host<schema<VARTYPES...>, INTERFACE>::size() const {
 
    // Make sure that there are some (jagged) vector types in the container.
    static_assert(
        std::disjunction_v<type::details::is_vector<VARTYPES>...>,
        "This function requires at least one (jagged) vector variable.");
 
    // Get the size of the vector(s).
    return details::get_host_size<VARTYPES...>(
        m_data, std::index_sequence_for<VARTYPES...>{});
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST void host<schema<VARTYPES...>, INTERFACE>::resize(
    std::size_t size) {
 
    // Make sure that there are some (jagged) vector types in the container.
    static_assert(
        std::disjunction_v<type::details::is_vector<VARTYPES>...>,
        "This function requires at least one (jagged) vector variable.");
 
    // Resize the vector(s).
    details::host_resize<VARTYPES...>(m_data, size,
                                      std::index_sequence_for<VARTYPES...>{});
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST void host<schema<VARTYPES...>, INTERFACE>::reserve(
    std::size_t size) {
 
    // Make sure that there are some (jagged) vector types in the container.
    static_assert(
        std::disjunction_v<type::details::is_vector<VARTYPES>...>,
        "This function requires at least one (jagged) vector variable.");
 
    // Resize the vector(s).
    details::host_reserve<VARTYPES...>(m_data, size,
                                       std::index_sequence_for<VARTYPES...>{});
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST void host<schema<VARTYPES...>, INTERFACE>::push_back(
    const object_type& element) {
 
    // Make sure that there are some (jagged) vector types in the container.
    static_assert(
        std::disjunction_v<type::details::is_vector<VARTYPES>...>,
        "This function requires at least one (jagged) vector variable.");
 
    // Add a new element to the container.
    const size_type index = size();
    resize(index + 1);
    // Set the new element.
    at(index) = element;
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
template <std::size_t INDEX>
VECMEM_HOST typename details::host_type_at<INDEX, VARTYPES...>::return_type
host<schema<VARTYPES...>, INTERFACE>::get() {
 
    // For scalar types we don't return the array, but rather a
    // reference to the single scalar held by the array.
    if constexpr (type::details::is_scalar_v<typename std::tuple_element<
                      INDEX, std::tuple<VARTYPES...>>::type>) {
        return std::get<INDEX>(m_data)[0];
    } else {
        return std::get<INDEX>(m_data);
    }
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
template <std::size_t INDEX>
VECMEM_HOST
    typename details::host_type_at<INDEX, VARTYPES...>::const_return_type
    host<schema<VARTYPES...>, INTERFACE>::get() const {
 
    // For scalar types we don't return the array, but rather a
    // reference to the single scalar held by the array.
    if constexpr (type::details::is_scalar_v<typename std::tuple_element<
                      INDEX, std::tuple<VARTYPES...>>::type>) {
        return std::get<INDEX>(m_data)[0];
    } else {
        return std::get<INDEX>(m_data);
    }
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST typename host<schema<VARTYPES...>, INTERFACE>::proxy_type
host<schema<VARTYPES...>, INTERFACE>::at(size_type index) {
 
    // Make sure that the index is within bounds.
    if (index >= size()) {
        throw std::out_of_range("index (" + std::to_string(index) +
                                ") >= size (" + std::to_string(size()) + ")");
    }
 
    // Use the unprotected function.
    return this->operator[](index);
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST typename host<schema<VARTYPES...>, INTERFACE>::const_proxy_type
host<schema<VARTYPES...>, INTERFACE>::at(size_type index) const {
 
    // Make sure that the index is within bounds.
    if (index >= size()) {
        throw std::out_of_range("index (" + std::to_string(index) +
                                ") >= size (" + std::to_string(size()) + ")");
    }
 
    // Use the unprotected function.
    return this->operator[](index);
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST typename host<schema<VARTYPES...>, INTERFACE>::proxy_type
host<schema<VARTYPES...>, INTERFACE>::operator[](size_type index) {
 
    // Create the proxy.
    return proxy_type{*this, index};
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST typename host<schema<VARTYPES...>, INTERFACE>::const_proxy_type
host<schema<VARTYPES...>, INTERFACE>::operator[](size_type index) const {
 
    // Create the proxy.
    return const_proxy_type{*this, index};
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST typename host<schema<VARTYPES...>, INTERFACE>::tuple_type&
host<schema<VARTYPES...>, INTERFACE>::variables() {
 
    return m_data;
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST const typename host<schema<VARTYPES...>, INTERFACE>::tuple_type&
host<schema<VARTYPES...>, INTERFACE>::variables() const {
 
    return m_data;
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST memory_resource& host<schema<VARTYPES...>, INTERFACE>::resource()
    const {
 
    return m_resource;
}
 
}  // namespace edm
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST void get_data_impl(edm::host<edm::schema<VARTYPES...>, INTERFACE>&,
                               edm::data<edm::schema<VARTYPES...>>&,
                               memory_resource&, std::index_sequence<>) {
    // For SonarCloud: The terminal node doesn't need to do anything.
}
 
template <typename... VARTYPES, template <typename> class INTERFACE,
          std::size_t I, std::size_t... Is>
VECMEM_HOST void get_data_impl(
    edm::host<edm::schema<VARTYPES...>, INTERFACE>& host,
    edm::data<edm::schema<VARTYPES...>>& data, memory_resource& mr,
    std::index_sequence<I, Is...>) {
 
    if constexpr (edm::type::details::is_jagged_vector_v<
                      typename std::tuple_element<
                          I, std::tuple<VARTYPES...>>::type>) {
        // Make the @c vecmem::edm::data object hold on to the
        // @c vecmem::data::jagged_vector_data object. Notice that this is a
        // move assignment here.
        std::get<I>(data.variables()) =
            ::vecmem::get_data(host.template get<I>(), &mr);
        // Set up the @c vecmem::edm::view object to point at the
        // @c vecmem::data::jagged_vector_data object.
        data.template get<I>() =
            ::vecmem::get_data(std::get<I>(data.variables()));
    } else if constexpr (edm::type::details::is_scalar<
                             typename std::tuple_element<
                                 I, std::tuple<VARTYPES...>>::type>::value) {
        // For scalar variables we just make @c vecmem::edm::view remember
        // a pointer.
        data.template get<I>() = &(host.template get<I>());
    } else {
        // For 1D vectors it's enough to make @c vecmem::edm::view have a
        // correct @c vecmem::data::vector_view object.
        data.template get<I>() = ::vecmem::get_data(host.template get<I>());
    }
    // Continue the recursion.
    get_data_impl(host, data, mr, std::index_sequence<Is...>{});
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST edm::data<edm::schema<VARTYPES...>> get_data(
    edm::host<edm::schema<VARTYPES...>, INTERFACE>& host,
    memory_resource* resource) {
 
    // Create the result object.
    edm::data<edm::schema<VARTYPES...>> result;
    // Decide what memory resource to use for setting it up.
    memory_resource& mr = (resource != nullptr ? *resource : host.resource());
    // Set its size, if that's available. Note that if there are no vector
    // variables in the container, then @c vecmem::edm::data also doesn't need
    // a memory resource.
    if constexpr (std::disjunction_v<
                      edm::type::details::is_vector<VARTYPES>...>) {
        result = {static_cast<
                      typename edm::data<edm::schema<VARTYPES...>>::size_type>(
                      host.size()),
                  mr};
    }
    // Fill it with the helper function.
    get_data_impl<VARTYPES...>(host, result, mr,
                               std::index_sequence_for<VARTYPES...>{});
    // Return the filled object.
    return result;
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST void get_data_impl(
    const edm::host<edm::schema<VARTYPES...>, INTERFACE>&,
    edm::data<edm::details::add_const_t<edm::schema<VARTYPES...>>>&,
    memory_resource&, std::index_sequence<>) {
    // For SonarCloud: The terminal node doesn't need to do anything.
}
 
template <typename... VARTYPES, template <typename> class INTERFACE,
          std::size_t I, std::size_t... Is>
VECMEM_HOST void get_data_impl(
    const edm::host<edm::schema<VARTYPES...>, INTERFACE>& host,
    edm::data<edm::details::add_const_t<edm::schema<VARTYPES...>>>& data,
    memory_resource& mr, std::index_sequence<I, Is...>) {
 
    if constexpr (edm::type::details::is_jagged_vector<
                      typename std::tuple_element<
                          I, std::tuple<VARTYPES...>>::type>::value) {
        // Make the @c vecmem::edm::data object hold on to the
        // @c vecmem::data::jagged_vector_data object. Notice that this is a
        // move assignment here.
        std::get<I>(data.variables()) =
            ::vecmem::get_data(host.template get<I>(), &mr);
        // Set up the @c vecmem::edm::view object to point at the
        // @c vecmem::data::jagged_vector_data object.
        data.template get<I>() =
            ::vecmem::get_data(std::get<I>(data.variables()));
    } else if constexpr (edm::type::details::is_scalar<
                             typename std::tuple_element<
                                 I, std::tuple<VARTYPES...>>::type>::value) {
        // For scalar variables we just make @c vecmem::edm::view remember
        // a pointer.
        data.template get<I>() = &(host.template get<I>());
    } else {
        // For 1D vectors it's enough to make @c vecmem::edm::view have a
        // correct @c vecmem::data::vector_view object.
        data.template get<I>() = ::vecmem::get_data(host.template get<I>());
    }
    // Continue the recursion.
    get_data_impl(host, data, mr, std::index_sequence<Is...>{});
}
 
template <typename... VARTYPES, template <typename> class INTERFACE>
VECMEM_HOST edm::data<edm::details::add_const_t<edm::schema<VARTYPES...>>>
get_data(const edm::host<edm::schema<VARTYPES...>, INTERFACE>& host,
         memory_resource* resource) {
 
    // Create the result object.
    edm::data<edm::details::add_const_t<edm::schema<VARTYPES...>>> result;
    // Decide what memory resource to use for setting it up.
    memory_resource& mr = (resource != nullptr ? *resource : host.resource());
    // Set its size, if that's available. Note that if there are no vector
    // variables in the container, then @c vecmem::edm::data also doesn't need
    // a memory resource.
    if constexpr (std::disjunction_v<
                      edm::type::details::is_vector<VARTYPES>...>) {
        result = {static_cast<
                      typename edm::view<edm::schema<VARTYPES...>>::size_type>(
                      host.size()),
                  mr};
    }
    // Fill it with the helper function.
    get_data_impl<VARTYPES...>(host, result, mr,
                               std::index_sequence_for<VARTYPES...>{});
    // Return the filled object.
    return result;
}
 
}  // namespace vecmem