DeviceMeasurementContainers.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ACTSGPUEVENT_DEVICEMEASUREMENTCOLLECTION_H
#define ACTSGPUEVENT_DEVICEMEASUREMENTCOLLECTION_H
 
 
#include "traccc/definitions/qualifiers.hpp"
#include <vecmem/edm/container.hpp>
 
#include <array>
#include <compare>
#include <ostream>
 
namespace ActsTrk {

template <typename BASE>
class DevicePixelMeasurement : public BASE {
 
    public:

    using BASE::BASE;
    using BASE::operator=;



    TRACCC_HOST_DEVICE
    auto& identifier() { return BASE::template get<0>(); }
    TRACCC_HOST_DEVICE
    const auto& identifier() const { return BASE::template get<0>(); }

    TRACCC_HOST_DEVICE
    auto& local_position() { return BASE::template get<1>(); }
    TRACCC_HOST_DEVICE
    const auto& local_position() const { return BASE::template get<1>(); }

    TRACCC_HOST_DEVICE
    auto& local_variance() { return BASE::template get<2>(); }
    TRACCC_HOST_DEVICE
    const auto& local_variance() const { return BASE::template get<2>(); }

    TRACCC_HOST_DEVICE
    auto& global_position() { return BASE::template get<3>(); }
    TRACCC_HOST_DEVICE
    const auto& global_position() const { return BASE::template get<3>(); }

    TRACCC_HOST_DEVICE
    auto& module_id_hash() { return BASE::template get<4>(); }
    TRACCC_HOST_DEVICE
    const auto& module_id_hash() const { return BASE::template get<4>(); }



    template <typename T>
    TRACCC_HOST_DEVICE bool operator==(const DevicePixelMeasurement<T>& other) const;

    template <typename T>
    TRACCC_HOST_DEVICE std::partial_ordering operator<=>(
        const DevicePixelMeasurement<T>& other) const;

 
    private:
    TRACCC_HOST
    friend std::ostream& operator<<(std::ostream& os, const DevicePixelMeasurement& m) {
        os << "id = " << m.identifier()
            << ", ";
        os << "module_hash = " << m.module_id_hash()
            << ", ";
        os << "local_pos = [" << m.local_position()[0]
            << ", " << m.local_position()[1]
            << "], ";
        os << "local_var = [" << m.local_variance()[0]
            << ", " << m.local_variance()[1]
            << "], ";
        os << "global_pos = [" << m.global_position()[0]
            << ", " << m.global_position()[1]
            << ", " << m.global_position()[2]
            << "]";
        return os;
    }
};  // class DevicePixelmeasurement

template <typename BASE>
class DeviceStripMeasurement : public BASE {
 
    public:

    using BASE::BASE;
    using BASE::operator=;



    TRACCC_HOST_DEVICE
    auto& identifier() { return BASE::template get<0>(); }
    TRACCC_HOST_DEVICE
    const auto& identifier() const { return BASE::template get<0>(); }

    TRACCC_HOST_DEVICE
    auto& local_position() { return BASE::template get<1>(); }
    TRACCC_HOST_DEVICE
    const auto& local_position() const { return BASE::template get<1>(); }

    TRACCC_HOST_DEVICE
    auto& local_variance() { return BASE::template get<2>(); }
    TRACCC_HOST_DEVICE
    const auto& local_variance() const { return BASE::template get<2>(); }

    TRACCC_HOST_DEVICE
    auto& module_id_hash() { return BASE::template get<3>(); }
    TRACCC_HOST_DEVICE
    const auto& module_id_hash() const { return BASE::template get<3>(); }



    template <typename T>
    TRACCC_HOST_DEVICE bool operator==(const DeviceStripMeasurement<T>& other) const;

    template <typename T>
    TRACCC_HOST_DEVICE std::partial_ordering operator<=>(
        const DeviceStripMeasurement<T>& other) const;

 
    private:
    TRACCC_HOST
    friend std::ostream& operator<<(std::ostream& os, const DeviceStripMeasurement& m) {
        os << "id = " << m.identifier()
            << ", ";
        os << "module_hash = " << m.module_id_hash()
            << ", ";
        os << "local_pos = " << m.local_position()[0]
            << ", ";
        os << "local_var = " << m.local_variance()[0]
            << ", ";
        return os;
    }
 
};  // class DeviceStripmeasurement

using DevicePixelMeasurementContainer = vecmem::edm::container<
    DevicePixelMeasurement
    // identifier
    , vecmem::edm::type::vector<long unsigned int>
    // local_position
    , vecmem::edm::type::vector<std::array<float, 2u>>
    // local_variance
    , vecmem::edm::type::vector<std::array<float, 4u>>
    // global_position
    , vecmem::edm::type::vector<std::array<float, 3u>>
    // module hash
    , vecmem::edm::type::vector<unsigned int>
    >;

using DeviceStripMeasurementContainer = vecmem::edm::container<
    DeviceStripMeasurement
    // identifier
    , vecmem::edm::type::vector<long unsigned int>
    // local_position
    , vecmem::edm::type::vector<float>
    // local_variance
    , vecmem::edm::type::vector<float>
    // module hash
    , vecmem::edm::type::vector<unsigned int>
    >;
 
}  // namespace ActsTrk
 
#endif // ACTSGPUEVENT_DEVICEMEASUREMENTCOLLECTION_H