TruthParticleHitCountAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ACTSTRK_TRUTHPARTICLEHITCOUNT_H
#define ACTSTRK_TRUTHPARTICLEHITCOUNT_H 1
 
// Base Class
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
 
 
// Handle Keys
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
 
#include "ActsEvent/TruthParticleHitCounts.h"
#include "ActsEvent/MeasurementToTruthParticleAssociation.h"
#include "ActsEvent/TrackContainer.h"
#include "ActsEvent/ContextUtility.h"
#include "ActsTruth/ElasticDecayUtil.h"
 
#include <mutex>
#include "ActsInterop/StatUtils.h"
 
#include <string>
#include <memory>
#include <array>
#include <atomic>
#include <type_traits>
 
#include <cmath>
#include <iomanip>
#include <ostream>
#include <string>
#include <sstream>
#include <vector>
 
namespace ActsTrk
{
  constexpr bool TruthParticleHitCountDebugHists = false;
 
  class TruthParticleHitCountAlg : public AthReentrantAlgorithm
  {
  public:
    using AthReentrantAlgorithm::AthReentrantAlgorithm;
 
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
    virtual StatusCode execute(const EventContext &ctx) const override;
 
  private:
     ContextUtility m_ctxProvider{this};
      
     SG::ReadHandleKeyArray<MeasurementToTruthParticleAssociation>  m_measAssocKeys
        {this, "MeasurementAssocKeys", {}, "Association map from the measurements to generator particles." };
     SG::WriteHandleKey<TruthParticleHitCounts>  m_truthHitCountsOut
        {this, "TruthParticleHitCountsOut", "", "Map from truth particle to hit counts." };
 
     Gaudi::Property<float> m_maxEnergyLoss
        {this, "MaxEnergyLoss", 10e12, "Stop moving up the decay chain if the energy loss is above  this value." };
     Gaudi::Property<unsigned int> m_nHitsMin
        {this, "NHitsMin", 1, "Minimum number of hits associated to a truth particle to consider the particle." };
 
     ElasticDecayUtil<TruthParticleHitCountDebugHists> m_elasticDecayUtil;
     static constexpr float s_unitGeV = 1e3;
  
     Gaudi::Property<std::vector<float>> m_energyLossBinning
        {this, "EnergyLossBinning", {20.,0.,5.*s_unitGeV}, "Binning to be used for the energy loss histograms." };
 
     template <bool IsDebug>
     struct AssociationCounter {
        struct Empty {
           template <typename... T_Args>
           Empty(T_Args... ) {}
        };
        mutable typename std::conditional<IsDebug,
                                          std::mutex,
                                          Empty>::type m_mutex ATLAS_THREAD_SAFE;
        mutable typename std::conditional<IsDebug,
                                          ActsUtils::StatHist,
                                          Empty>::type m_measPerTruthParticle ATLAS_THREAD_SAFE {20,-.5,40.-.5};
 
        template <class T_OutStream>
        void dumpStatistics(T_OutStream &out) const;
        void fillStatistics(unsigned int n_measurements) const;
     };
     AssociationCounter<TruthParticleHitCountDebugHists> m_associationCounter;
     mutable std::atomic<std::size_t> m_nTruthParticlesWithHits ATLAS_THREAD_SAFE {};
  };
 
} // namespace
 
#endif