#include <TrackTruthMatchingBaseAlg.h>

Inheritance diagram for ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >:

Collaboration diagram for ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >:

Public Types
using	TruthParticleSet = std::conditional< DetailEnabled, std::unordered_set< const xAOD::TruthParticle * >, typename BaseStat< DetailEnabled >::Empty >::type

using	CounterArrayVec = std::conditional< DetailEnabled, std::vector< std::array< std::size_t, kNCategorisedCounter > >, Empty >::type

using	StatArrayVec = std::conditional< DetailEnabled, std::vector< std::array< ActsUtils::Stat, kNCategorisedStat > >, Empty >::type

Public Member Functions
	EventStatBase (const IAthSelectionTool &truth_selection_tool, std::size_t per_eta_size, std::size_t per_pdg_size, [[maybe_unused]] std::size_t track_to_truth_size)

void	fill ([[maybe_unused]] unsigned int eta_category_i, [[maybe_unused]] unsigned int pdg_id_category_i, [[maybe_unused]] float hit_efficiency, [[maybe_unused]] float hit_purity, [[maybe_unused]] float match_prob, [[maybe_unused]] const xAOD::TruthParticle *best_match)

void	reset (const IAthSelectionTool &truth_selection_tool, [[maybe_unused]] std::size_t per_eta_size, [[maybe_unused]] std::size_t per_pdg_size)

void	fill ([[maybe_unused]] unsigned int eta_category_i, [[maybe_unused]] unsigned int pdg_id_category_i, [[maybe_unused]] float hit_efficiency, [[maybe_unused]] float hit_purity, [[maybe_unused]] float match_prob)

void	incrementTotal ([[maybe_unused]] unsigned int eta_category_i, [[maybe_unused]] unsigned int pdg_id_category_i)

BaseStat< DetailEnabled > &	operator+= (const BaseStat< DetailEnabled > &event_stat)

void	printStatTables (const TrackTruthMatchingBaseAlg &parent, const std::vector< float > &statPtBins, const std::vector< float > &statEtaBins, std::vector< int > &pdgId, bool printDetails, bool pdgIdCategorisation, bool useAbsEtaForStat)

std::size_t	perEtaSize () const

std::size_t	perPdgIdSize () const

Public Attributes
TruthParticleSet	m_truthParticlesWithAssociatedTrack

unsigned int	m_nTruthParticleWithoutAssociatedCounts =0u

unsigned int	m_nTracksWithoutAssociatedTruthParticle =0u

unsigned int	m_nTracksWithoutSelectedTruthParticle =0u

unsigned int	m_nTruthParticleNonoiseMismatches =0u

unsigned int	m_nTruthCuts

ActsUtils::StatHist	m_truthSelectionCuts

CounterArrayVec	m_counterPerEta

CounterArrayVec	m_counterPerPdgId

StatArrayVec	m_statPerEta

StatArrayVec	m_statPerPdgId

Static Public Attributes
static constexpr bool	doDetail = DetailEnabled

Detailed Description

template<bool DetailEnabled>
struct ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >

Definition at line 79 of file TrackTruthMatchingBaseAlg.h.

Member Typedef Documentation

◆ CounterArrayVec

template<bool DetailEnabled>

using ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::CounterArrayVec = std::conditional< DetailEnabled, std::vector< std::array< std::size_t, kNCategorisedCounter> >, Empty >::type

inherited

Definition at line 345 of file TrackTruthMatchingBaseAlg.h.

◆ StatArrayVec

template<bool DetailEnabled>

using ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::StatArrayVec = std::conditional< DetailEnabled, std::vector< std::array<ActsUtils::Stat, kNCategorisedStat> >, Empty >::type

inherited

Definition at line 348 of file TrackTruthMatchingBaseAlg.h.

◆ TruthParticleSet

template<bool DetailEnabled>

using ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >::TruthParticleSet = std::conditional< DetailEnabled, std::unordered_set<const xAOD::TruthParticle *>, typename BaseStat<DetailEnabled>::Empty >::type

Definition at line 117 of file TrackTruthMatchingBaseAlg.h.

Constructor & Destructor Documentation

◆ EventStatBase()

template<bool DetailEnabled>

ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >::EventStatBase	(	const IAthSelectionTool &	truth_selection_tool,
		std::size_t	per_eta_size,
		std::size_t	per_pdg_size,
		[[maybe_unused] ] std::size_t	track_to_truth_size
	)

inline

Definition at line 82 of file TrackTruthMatchingBaseAlg.h.

            : BaseStat<DetailEnabled>(truth_selection_tool, per_eta_size, per_pdg_size),
              m_nTruthCuts(truth_selection_tool.nCuts())
         {
            if constexpr(DetailEnabled) {
               m_truthParticlesWithAssociatedTrack.reserve(track_to_truth_size);
            }
  
         };

Member Function Documentation

◆ fill() [1/2]

template<bool DetailEnabled>

void ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::fill	(	[[maybe_unused] ] unsigned int	eta_category_i,
		[[maybe_unused] ] unsigned int	pdg_id_category_i,
		[[maybe_unused] ] float	hit_efficiency,
		[[maybe_unused] ] float	hit_purity,
		[[maybe_unused] ] float	match_prob
	)

inlineinherited

Definition at line 295 of file TrackTruthMatchingBaseAlg.h.

                                                      {
            if (DetailEnabled) {
               assert( eta_category_i <m_statPerEta.size());
               m_statPerEta[eta_category_i][kHitEfficiency].add( hit_efficiency);
               m_statPerEta[eta_category_i][kHitPurity].add( hit_purity);
               m_statPerEta[eta_category_i][kMatchProbability].add( match_prob);
               assert( pdg_id_category_i <m_statPerPdgId.size());
               m_statPerPdgId[pdg_id_category_i][kHitEfficiency].add( hit_efficiency);
               m_statPerPdgId[pdg_id_category_i][kHitPurity].add( hit_purity);
               m_statPerPdgId[pdg_id_category_i][kMatchProbability].add( match_prob);
               assert( eta_category_i < m_counterPerEta.size());
               assert( pdg_id_category_i <m_counterPerPdgId.size());
               ++m_counterPerEta[eta_category_i][kNTotalTracks];
               ++m_counterPerPdgId[pdg_id_category_i][kNTotalTracks];
            }
         }

◆ fill() [2/2]

template<bool DetailEnabled>

void ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >::fill	(	[[maybe_unused] ] unsigned int	eta_category_i,
		[[maybe_unused] ] unsigned int	pdg_id_category_i,
		[[maybe_unused] ] float	hit_efficiency,
		[[maybe_unused] ] float	hit_purity,
		[[maybe_unused] ] float	match_prob,
		[[maybe_unused] ] const xAOD::TruthParticle *	best_match
	)

inline

Definition at line 94 of file TrackTruthMatchingBaseAlg.h.

                                                                         {
            BaseStat<DetailEnabled>::fill(eta_category_i,
                                          pdg_id_category_i,
                                          hit_efficiency,
                                          hit_purity,
                                          match_prob);
            if constexpr(DetailEnabled) {
               if (!m_truthParticlesWithAssociatedTrack.insert(best_match).second) {
                  // truth particle already had a best match
                  ++this->m_counterPerEta[eta_category_i][kNParticleWithMultipleAssociatedTracks];
                  ++this->m_counterPerPdgId[pdg_id_category_i][kNParticleWithMultipleAssociatedTracks];
               }
               else {
                  ++this->m_counterPerEta[eta_category_i][kNParticleWithAssociatedTrack];
                  ++this->m_counterPerPdgId[pdg_id_category_i][kNParticleWithAssociatedTrack];
               }
            }
         }

◆ incrementTotal()

template<bool DetailEnabled>

void ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::incrementTotal	(	[[maybe_unused] ] unsigned int	eta_category_i,
		[[maybe_unused] ] unsigned int	pdg_id_category_i
	)

inlineinherited

Definition at line 315 of file TrackTruthMatchingBaseAlg.h.

                                                                              {
            if constexpr(DetailEnabled) {
               ++m_counterPerEta[eta_category_i][kNTotalParticles];
               ++m_counterPerPdgId[pdg_id_category_i][kNTotalParticles];
            }
         }

◆ operator+=()

template<bool DetailEnabled>

TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled > & ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::operator+= ( const BaseStat< DetailEnabled > & event_stat )

inlineinherited

Definition at line 440 of file TrackTruthMatchingBaseAlg.cxx.

                                                                                                                                  {
      if constexpr(DetailEnabled) {
         addStat(event_stat.m_counterPerEta,m_counterPerEta);
         addStat(event_stat.m_statPerEta, m_statPerEta);
         addStat(event_stat.m_counterPerPdgId,m_counterPerPdgId);
         addStat(event_stat.m_statPerPdgId, m_statPerPdgId);
      }
      m_truthSelectionCuts += event_stat.m_truthSelectionCuts;
      return *this;
   }

◆ perEtaSize()

template<bool DetailEnabled>

std::size_t ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::perEtaSize ( ) const

inlineinherited

Definition at line 334 of file TrackTruthMatchingBaseAlg.h.

                                       {
            if constexpr(DetailEnabled) { return m_counterPerEta.size(); }
            else { return 0u; }
         }

◆ perPdgIdSize()

template<bool DetailEnabled>

std::size_t ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::perPdgIdSize ( ) const

inlineinherited

Definition at line 338 of file TrackTruthMatchingBaseAlg.h.

                                        {
            if constexpr(DetailEnabled) { return m_counterPerPdgId.size(); }
            else { return 0u; }
         }

◆ printStatTables()

template<bool DetailEnabled>

void ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::printStatTables	(	const TrackTruthMatchingBaseAlg &	parent,
		const std::vector< float > &	statPtBins,
		const std::vector< float > &	statEtaBins,
		std::vector< int > &	pdgId,
		bool	printDetails,
		bool	pdgIdCategorisation,
		bool	useAbsEtaForStat
	)

inherited

Definition at line 452 of file TrackTruthMatchingBaseAlg.cxx.

                                                                                                   {
     if constexpr(DetailEnabled) {
        static constexpr bool rotate=true;// row : eta/PDG ID; column: pt
        std::vector<std::string> counter_labels { std::string("Truth particles"),
                                                  std::string("with asso. track"),
                                                  std::string("with >1 asso. tracks"),
                                                  std::string("total tracks")};
        std::vector<std::string> pt_labels;
        pt_labels.reserve(statPtBins.size() + 2);
        unsigned int pt_precision=0;
        for (float pt : statPtBins) {
           if (pt<1.) {
              pt_precision=1;
              break;
           }
        }
        for (std::size_t bin_i = 0; bin_i < statPtBins.size() + 2; ++bin_i) {
           pt_labels.push_back(TableUtils::makeBinLabel("pt",statPtBins, bin_i, true, pt_precision));
        }
        // statistics eta-bins
        {
           std::vector<std::string> eta_labels;
           eta_labels.reserve(statEtaBins.size() + 2);
           for (std::size_t eta_bin_i = 0; eta_bin_i < statEtaBins.size() + 2; ++eta_bin_i) {
              eta_labels.push_back(TableUtils::makeEtaBinLabel(statEtaBins, eta_bin_i, useAbsEtaForStat));
           }
  
           accumulateToLastColumnRow(statPtBins.size()+2,statEtaBins.size()+2, m_statPerEta);
           accumulateToLastColumnRow(statPtBins.size()+2,statEtaBins.size()+2, m_counterPerEta);
  
           if (statPtBins.empty() || printDetails) {
              parent.printCategories(pt_labels, eta_labels, counter_labels, m_statPerEta, m_counterPerEta,
                                     (!statPtBins.empty()
                                      ? hfill("pt ",
                                              "eta",
                                              TableUtils::maxLabelWidth(pt_labels)
                                              +TableUtils::maxLabelWidth(eta_labels))
                                      : std::string("eta") ),
                                     !statPtBins.empty());
           }
           if (!statPtBins.empty()) {
              parent.printData2D(pt_labels, eta_labels,
                                 rotate
                                 ? hfill("eta","\\ pt", TableUtils::maxLabelWidth(eta_labels))
                                 : hfill("pt","\\ eta", TableUtils::maxLabelWidth(pt_labels)),
                                 m_statPerEta,
                                 m_counterPerEta,
                                 rotate);
           }
        }
  
        // statistics in PDG ID bins.
        if (pdgIdCategorisation) {
           std::vector<std::string> pdg_id_labels;
           pdg_id_labels.reserve( pdgId.size());
           pdg_id_labels.push_back("Other");
           for (unsigned int pdg_i=1; pdg_i < pdgId.size(); ++pdg_i) {
              std::stringstream a_label;
              a_label << HepPID::particleName(pdgId[pdg_i])  << " [" << pdgId[pdg_i] << "]";
              pdg_id_labels.push_back(  a_label.str() );
           }
           unsigned int max_pdg_id_slots=m_statPerPdgId.size()/(statPtBins.size()+2);
           assert( m_statPerPdgId.size() % (statPtBins.size()+2) == 0 );
           // also the unused columns are projected, but that does not harm :
           accumulateToLastRow(statPtBins.size()+2,max_pdg_id_slots, m_statPerPdgId);
           accumulateToLastRow(statPtBins.size()+2,max_pdg_id_slots, m_counterPerPdgId);
  
           if (statPtBins.empty() || printDetails) {
              parent.printCategories(pt_labels, pdg_id_labels, counter_labels, m_statPerPdgId, m_counterPerPdgId,
                              (!statPtBins.empty()
                               ? hfill("pt ",
                                       "PDG-id",
                                       TableUtils::maxLabelWidth(pt_labels)
                                       +TableUtils::maxLabelWidth(pdg_id_labels))
                               : std::string("eta")),
                              !statPtBins.empty());
           }
           if (!statPtBins.empty()) {
              parent.printData2D(pt_labels, pdg_id_labels,
                                 rotate
                                 ? hfill("PDG ID","\\ pt", TableUtils::maxLabelWidth(pdg_id_labels))
                                 : hfill("pt","\\ PDG ID", TableUtils::maxLabelWidth(pt_labels)),
                                 m_statPerPdgId,
                                 m_counterPerPdgId,
                                 rotate);
           }
        }
     }
   }

◆ reset()

template<bool DetailEnabled>

void ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::reset	(	const IAthSelectionTool &	truth_selection_tool,
		[[maybe_unused] ] std::size_t	per_eta_size,
		[[maybe_unused] ] std::size_t	per_pdg_size
	)

inlineinherited

Definition at line 279 of file TrackTruthMatchingBaseAlg.h.

         {
            m_truthSelectionCuts.setBinning(truth_selection_tool.nCuts()+1, -0.5,truth_selection_tool.nCuts()+.5);
            if constexpr(DetailEnabled) {
               m_counterPerEta.clear();
               m_counterPerPdgId.clear();
               m_statPerEta.clear();
               m_statPerPdgId.clear();
               m_counterPerEta.resize(per_eta_size);
               m_counterPerPdgId.resize( per_pdg_size);
               m_statPerEta.resize( per_eta_size );
               m_statPerPdgId.resize( per_pdg_size );
            }
         }