#include <TrackTruthMatchingBaseAlg.h>

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

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

Public Types
using	TruthParticleSet
using	CounterArrayVec
using	StatArrayVec

Public Member Functions
	EventStatBase (const IAthSelectionTool &truth_selection_tool, std::size_t per_eta_size, std::size_t per_pdg_size, std::size_t track_to_truth_size)
void	fill (unsigned int eta_category_i, unsigned int pdg_id_category_i, float hit_efficiency, float hit_purity, float match_prob, const xAOD::TruthParticle *best_match)
void	reset (const IAthSelectionTool &truth_selection_tool, std::size_t per_eta_size, std::size_t per_pdg_size)
void	fill (unsigned int eta_category_i, unsigned int pdg_id_category_i, float hit_efficiency, float hit_purity, float match_prob)
void	incrementTotal (unsigned int eta_category_i, 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 {0u}
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 74 of file TrackTruthMatchingBaseAlg.h.

Member Typedef Documentation

◆ CounterArrayVec

template<bool DetailEnabled>

using ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::CounterArrayVec

inherited

Initial value:

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

Definition at line 322 of file TrackTruthMatchingBaseAlg.h.

◆ StatArrayVec

template<bool DetailEnabled>

using ActsTrk::TrackTruthMatchingBaseAlg::BaseStat< DetailEnabled >::StatArrayVec

inherited

Initial value:

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

Definition at line 325 of file TrackTruthMatchingBaseAlg.h.

◆ TruthParticleSet

template<bool DetailEnabled>

using ActsTrk::TrackTruthMatchingBaseAlg::EventStatBase< DetailEnabled >::TruthParticleSet

Initial value:

std::conditional_t< DetailEnabled,
                                                  std::unordered_set<const xAOD::TruthParticle *>,
                                                  typename BaseStat<DetailEnabled>::Empty >

Definition at line 112 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,
		std::size_t	track_to_truth_size )

inline

Definition at line 77 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	(	unsigned int	eta_category_i,
		unsigned int	pdg_id_category_i,
		float	hit_efficiency,
		float	hit_purity,
		float	match_prob )

inlineinherited

Definition at line 272 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	(	unsigned int	eta_category_i,
		unsigned int	pdg_id_category_i,
		float	hit_efficiency,
		float	hit_purity,
		float	match_prob,
		const xAOD::TruthParticle *	best_match )

inline

Definition at line 89 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	(	unsigned int	eta_category_i,
		unsigned int	pdg_id_category_i )

inlineinherited

Definition at line 292 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 434 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 311 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 315 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 446 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;
          }
       }
       const std::string labelStr{"pt"};
       for (std::size_t bin_i = 0; bin_i < statPtBins.size() + 2; ++bin_i) {
          pt_labels.push_back(TableUtils::makeBinLabel(labelStr,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");
          auto gendata = std::make_unique<GenData>();
          for (unsigned int pdg_i=1; pdg_i < pdgId.size(); ++pdg_i) {
             std::stringstream a_label;
             a_label << gendata->particleName(pdgId[pdg_i]).value()  << " [" << 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,
		std::size_t	per_eta_size,
		std::size_t	per_pdg_size )

inlineinherited

Definition at line 256 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 );
           }
        }