Control/AthenaExamples/AthExCUDA/src/TrackParticleContainer.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
//
// Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHEXCUDA_TRACKPARTICLECONTAINER_H
#define ATHEXCUDA_TRACKPARTICLECONTAINER_H
 
// VecMem include(s).
#include <vecmem/edm/container.hpp>
 
namespace AthCUDAExamples {

template <typename BASE>
struct TrackParticleInterface : public BASE {

  using BASE::BASE;

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

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

  VECMEM_HOST_AND_DEVICE
  const auto& qOverP() const { return BASE::template get<2>(); }
  VECMEM_HOST_AND_DEVICE
  auto& qOverP() { return BASE::template get<2>(); }
 
};  // struct TrackParticleInterface

using TrackParticleContainer = vecmem::edm::container<
    TrackParticleInterface, vecmem::edm::type::vector<float>,
    vecmem::edm::type::vector<float>, vecmem::edm::type::vector<float>>;
 
}  // namespace AthCUDAExamples
 
#endif  // ATHEXCUDA_TRACKPARTICLECONTAINER_H