TrackToTruthParticleAssociation.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
  */
#ifndef TRACKTOGENPARTICLEASSOCIATION_H
#define TRACKTOGENPARTICLEASSOCIATION_H
#include "boost/container/small_vector.hpp"
#include "xAODTruth/TruthParticle.h"
#include "xAODMeasurementBase/MeasurementDefs.h"
#include "ActsEvent/TrackContainer.h"
#include "AthLinks/DataLink.h"
#include <vector>
#include <array>
#include <cstdint>
#include <utility>
 
namespace ActsTrk
{
  // NHitCounter is the number of types, without the Other (i.e. unknown) type
   constexpr unsigned int NHitCounter = static_cast< std::underlying_type<xAOD::UncalibMeasType>::type >(xAOD::UncalibMeasType::nTypes);
   constexpr unsigned int NTruthParticlesPerTrack = 5;  // a tiny fraction of measurements will have more than
                                                        // 5 associated GenParticles
 
   // Wrapper around std::array to ensure that by default elements are initialised to zero
   class  HitCounterArray : public  std::array<uint8_t,  NHitCounter>
   {
   public:
      HitCounterArray() : std::array<uint8_t,  NHitCounter>{} {}
   };

   class HitCountsPerTrack {
   public:
      using container = boost::container::small_vector<std::pair<const xAOD::TruthParticle *, HitCounterArray >, NTruthParticlesPerTrack>;

      const container &countsPerTruthParticle() const { return m_counts; }

      container &countsPerTruthParticle()             { return m_counts; }

      HitCounterArray &totalCounts()               { return m_recoCounts[kTotal]; }

      const HitCounterArray &totalCounts() const   { return m_recoCounts[kTotal]; }

      HitCounterArray &noiseCounts()               { return m_recoCounts[kNoise]; }

      const HitCounterArray &noiseCounts() const   { return m_recoCounts[kNoise]; }
 
   private:
      container m_counts;
      enum ECounts {kTotal, kNoise, kNRecoCounts };
      std::array<HitCounterArray,kNRecoCounts> m_recoCounts;
   };
 
   class TrackToTruthParticleAssociation : public std::vector<HitCountsPerTrack> {
   public:
      using std::vector<HitCountsPerTrack>::vector ;
 
      void setSourceContainer(DataLink<ActsTrk::TrackContainer> &&source) {
         m_sourceTracks = std::move(source);
      }
      const ActsTrk::TrackContainer *sourceContainer() const {
         return  m_sourceTracks.getDataPtr();
      }
         
   private:
      DataLink<ActsTrk::TrackContainer> m_sourceTracks;
   };
}
 
#include "AthenaKernel/CLASS_DEF.h"
CLASS_DEF( ActsTrk::TrackToTruthParticleAssociation, 101405904, 1 )
 
#endif