DerivationFramework::TruthDressingTool Class Reference

#include <TruthDressingTool.h>

Inheritance diagram for DerivationFramework::TruthDressingTool:

Collaboration diagram for DerivationFramework::TruthDressingTool:

Public Member Functions
virtual StatusCode	initialize () override final
virtual StatusCode	addBranches (const EventContext &ctx) const override final

Private Attributes
SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_particlesKey {this, "particlesKey", "TruthParticles", "ReadHandleKey for TruthParticles for photon list input"}
	ReadHandleKey input collection key. More...
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorationKey {this, "decorationName", m_particlesKey, "unusedPhotonDecoration", "Name of the decoration for photons that were used in dressing"}
	WriteDecorHandleKeys for decorations for particlesKey. More...
SG::ReadHandleKey< xAOD::TruthParticleContainer >	m_dressParticlesKey {this, "dressParticlesKey", "", "ReadHandleKey for input particles to be dressed. If taus are selected, everything in this input key will be used"}
	ReadHandleKey for particles to be dressed. More...
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_eKey {this, "e_dressed", m_dressParticlesKey, "e_dressed", "e_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_ptKey {this, "pt_dressed", m_dressParticlesKey, "pt_dressed", "pt_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_etaKey {this, "eta_dressed", m_dressParticlesKey, "eta_dressed", "eta_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_phiKey {this, "phi_dressed", m_dressParticlesKey, "phi_dressed", "phi_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_pt_visKey {this, "pt_vis_dressed", m_dressParticlesKey, "pt_vis_dressed", "pt_vis_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_eta_visKey {this, "eta_vis_dressed", m_dressParticlesKey, "eta_vis_dressed", "eta_vis_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_phi_visKey {this, "phi_vis_dressed", m_dressParticlesKey, "phi_vis_dressed", "phi_vis_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_m_visKey {this, "m_vis_dressed", m_dressParticlesKey, "m_vis_dressed", "m_vis_dressed decoration"}
SG::WriteDecorHandleKey< xAOD::TruthParticleContainer >	m_decorator_nphotonKey {this, "nPhotons_dressed", m_dressParticlesKey, "nPhotons_dressed", "nPhotons_dressed decoration"}
SG::ReadDecorHandleKey< xAOD::TruthParticleContainer >	m_truthClassKey {this, "truthClassifierKey", m_dressParticlesKey, "Classification"}
	To ensure that the algorithm is scheduled after the truth classifier. More...
Gaudi::Property< bool >	m_usePhotonsFromHadrons {this, "usePhotonsFromHadrons", false, "Add photons coming from hadron decays while dressing"}
	Parameter: Use photons from hadron decays? More...
Gaudi::Property< bool >	m_useLeptonsFromHadrons {this, "useLeptonsFromHadrons", false, "Consider leptons coming from hadron decays?"}
	Parameter: Use leptons from hadron decays? More...
Gaudi::Property< float >	m_coneSize {this, "dressingConeSize", 0.1, "Size of dR cone in which to include FSR photons in dressing"}
	Parameter: Cone size for dressing. More...
Gaudi::Property< std::vector< int > >	m_listOfPIDs {this, "particleIDsToDress", {11,13}, "List of the pdgID's of particles to be dressed (usually 11,13). Special treatment for taus (15)"}
	Parameter: List of pdgIDs of particles to dress. More...
Gaudi::Property< bool >	m_useAntiKt {this, "useAntiKt", false, "use anti-k_T instead of fixed-cone dressing"}
	Parameter: Use antikT algorithm for dressing? More...
Gaudi::Property< bool >	m_decoratePhotons {this, "decoratePhotons", true, "Do we want to decorate the photons used for dressing?"}
	Parameter: Do we want to decorate the photons used for dressing? More...

Detailed Description

Definition at line 22 of file TruthDressingTool.h.

Member Function Documentation

addBranches()

StatusCode DerivationFramework::TruthDressingTool::addBranches ( const EventContext &  ctx ) const

finaloverridevirtual

Definition at line 63 of file TruthDressingTool.cxx.

{
    // Get the event context
 
    // Retrieve the truth collections
    SG::ReadHandle<xAOD::TruthParticleContainer> truthParticles(m_particlesKey,ctx);
    if (!truthParticles.isValid()) {
        ATH_MSG_ERROR("Couldn't retrieve TruthParticle collection with name " << m_particlesKey);
        return StatusCode::FAILURE;
    }
 
    SG::ReadHandle<xAOD::TruthParticleContainer> dressTruthParticles(m_dressParticlesKey,ctx);
    if (!dressTruthParticles.isValid()) {
        ATH_MSG_ERROR("Couldn't retrieve TruthParticle collection with name " << m_dressParticlesKey);
        return StatusCode::FAILURE;
    }
 
    // Decorators
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_e(m_decorator_eKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_pt(m_decorator_ptKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_eta(m_decorator_etaKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_phi(m_decorator_phiKey, ctx);
    // for truth taus, use 'vis' in the decoration name to avoid prompt/visible tau momentum ambiguity
    // use (pt,eta,phi,m) for taus, for consistency with other TauAnalysisTools decorations
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_pt_vis(m_decorator_pt_visKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_eta_vis(m_decorator_eta_visKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_phi_vis(m_decorator_phi_visKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,float > decorator_m_vis(m_decorator_m_visKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,int > decorator_nphoton(m_decorator_nphotonKey, ctx);
    SG::WriteDecorHandle< xAOD::TruthParticleContainer,char > dressDec(m_decorationKey, ctx);
    // If we want to decorate, then we need to decorate everything with false to begin with
    if (m_decoratePhotons){
      for (const auto * particle : *truthParticles){
        dressDec(*particle) = 0;
      }
    } // We are using the decoration
 
    // accessors for truth tau visible momentum
    static const SG::ConstAccessor<double> pt_visAcc("pt_vis");
    static const SG::ConstAccessor<double> eta_visAcc("eta_vis");
    static const SG::ConstAccessor<double> phi_visAcc("phi_vis");
    static const SG::ConstAccessor<double> mvisAcc("m_vis");
 
    //get struct of helper functions
    DerivationFramework::DecayGraphHelper decayHelper;
 
    std::vector<const xAOD::TruthParticle*> listOfParticlesToDress;
    std::vector<xAOD::TruthParticle::FourMom_t> listOfDressedP4;
    std::vector<xAOD::TruthParticle::FourMom_t> listOfBareP4;
 
    if (m_listOfPIDs.size()==1 && std::abs(m_listOfPIDs[0])==MC::TAU) {
      // when dressing truth taus, it is assumed that the truth tau container has been built beforehand and is used as input
      for (auto *pItr : *dressTruthParticles) {
        if (!pItr->isTau()) {
          ATH_MSG_ERROR("Input particles should be truth taus.");
          return StatusCode::FAILURE;
        }
        if (!pt_visAcc.isAvailable(*pItr) || !eta_visAcc.isAvailable(*pItr) || !phi_visAcc.isAvailable(*pItr) || !mvisAcc.isAvailable(*pItr)) {
          ATH_MSG_ERROR("Visible momentum not available for truth taus, cannot perform dressing!");
          return StatusCode::FAILURE;
        }
 
        listOfParticlesToDress.push_back(pItr);
 
        // we dresss the visible 4-momentum
        xAOD::TruthParticle::FourMom_t bare_part;
        bare_part.SetPtEtaPhiM(pt_visAcc(*pItr), eta_visAcc(*pItr), phi_visAcc(*pItr), mvisAcc(*pItr));
        listOfDressedP4.push_back(bare_part);
      }
      // in the cone-based approach, make a copy of bare P4 to avoid recomputing it
      if (!m_useAntiKt) {
        listOfBareP4 = listOfDressedP4;
      }
    }
    else {
      // non-prompt particles are still included here to ensure all particles
      // will get the decoration; however further down only the prompt particles
      // are actually dressed depending on the value of m_useLeptonsFromHadrons
      decayHelper.constructListOfFinalParticles(dressTruthParticles.ptr(), listOfParticlesToDress, m_listOfPIDs, true);
 
      for (const auto* part : listOfParticlesToDress) {
        listOfDressedP4.push_back(part->p4());
      }
    }
 
    std::vector<int> dressedParticlesNPhot(listOfParticlesToDress.size(), 0);
 
    //fill the photon list
    std::vector<const xAOD::TruthParticle*>  photonsFSRList;
    std::vector<int> photonPID{22};
    const bool pass = decayHelper.constructListOfFinalParticles(truthParticles.ptr(), photonsFSRList,
                                                                photonPID, m_usePhotonsFromHadrons);
    if (!pass) {
      ATH_MSG_WARNING("Cannot construct the list of final state particles "<<m_truthClassKey.fullKey());
    }
 
    // Do dR-based photon dressing (default)
    if (!m_useAntiKt){
      //loop over photons, uniquely associate each to nearest bare particle
      for (const auto* phot : photonsFSRList) {
        double dRmin = m_coneSize;
        int idx = -1;
 
        for (size_t i = 0; i < listOfParticlesToDress.size(); ++i) {
          if (!m_useLeptonsFromHadrons) {
            if (!acc_origin.isAvailable(*listOfParticlesToDress[i])) {
              ATH_MSG_WARNING("MCTruthClassifier "<<m_truthClassKey.fullKey() <<" not available, cannot apply notFromHadron veto!");
            }
            unsigned int result = acc_origin(*listOfParticlesToDress[i]);
            const bool isPrompt = MCTruthPartClassifier::isPrompt(result, true);
            if (!isPrompt)  continue;
          }
          xAOD::TruthParticle::FourMom_t bare_part;
          if (listOfParticlesToDress[i]->isTau()) {
            bare_part = listOfBareP4[i];
          }
          else {
            bare_part = listOfParticlesToDress[i]->p4();
          }
 
          double dR = bare_part.DeltaR(phot->p4());
          if (dR < dRmin) {
            dRmin = dR;
            idx = i;
          }
        }
 
        if(idx > -1) {
          listOfDressedP4[idx] += phot->p4();
          dressedParticlesNPhot[idx]++;
          if (m_decoratePhotons){
            dressDec(*phot) = 1;
          }
        }
      }
 
      //loop over particles and add decorations
      for (size_t i = 0; i < listOfParticlesToDress.size(); ++i) {
        const xAOD::TruthParticle* part = listOfParticlesToDress[i];
        const xAOD::TruthParticle::FourMom_t& dressedVec = listOfDressedP4[i];
 
        if(part->isTau()) {
          decorator_pt_vis(*part)      = dressedVec.Pt();
          decorator_eta_vis(*part)     = dressedVec.Eta();
          decorator_phi_vis(*part)     = dressedVec.Phi();
          decorator_m_vis(*part)       = dressedVec.M();
        }
        else {
          decorator_e(*part)       = dressedVec.E();
          decorator_pt(*part)      = dressedVec.Pt();
          decorator_eta(*part)     = dressedVec.Eta();
          decorator_phi(*part)     = dressedVec.Phi();
        }
        decorator_nphoton(*part) = dressedParticlesNPhot[i];
      }
    } // end of the dR matching part
 
    //build the anti-kt jet list
    if (m_useAntiKt) {
      std::vector<fastjet::PseudoJet> sorted_jets;
      std::vector<fastjet::PseudoJet> fj_particles;
      for (const auto* part : listOfParticlesToDress) {
        if(part->isTau()) {
          const xAOD::TruthParticle::FourMom_t& tauvis = listOfDressedP4[part->index()];
          fj_particles.emplace_back(tauvis.Px(), tauvis.Py(), tauvis.Pz(), tauvis.E());
        }
        else {
          fj_particles.emplace_back(part->px(), part->py(), part->pz(), part->e());
        }
 
        fj_particles.back().set_user_index(HepMC::uniqueID(part));
      }
      for (const auto* part : photonsFSRList) {
        fj_particles.emplace_back(part->px(), part->py(), part->pz(), part->e());
        fj_particles.back().set_user_index(HepMC::uniqueID(part));
      }
 
      //run the clustering
      fastjet::JetAlgorithm alg=fastjet::antikt_algorithm;
      const fastjet::JetDefinition jet_def(alg, m_coneSize);
      fastjet::ClusterSequence cseq(fj_particles, jet_def);
      sorted_jets = sorted_by_pt(cseq.inclusive_jets(0));
      //associate clustered jets back to bare particles
      std::vector<int> photon_uniqueIDs(50);
      photon_uniqueIDs.clear();
      for (const auto* part : listOfParticlesToDress) {
        //loop over fastjet pseudojets and associate one with this particle
        bool found=false;
        auto pjItr=sorted_jets.begin();
        while(!found && pjItr!=sorted_jets.end()) {
          std::vector<fastjet::PseudoJet> constituents = pjItr->constituents();
          for(const auto& constit : constituents) {
            if (HepMC::uniqueID(part)==constit.user_index()) {
 
              // shall we count the number of photons among the pseudojet constituents
              // to decorate leptons with the number of dressing photons found by the anti-kt algorithm?
 
              if(part->isTau()) {
                decorator_pt_vis(*part)      = pjItr->pt();
                decorator_eta_vis(*part)     = pjItr->pseudorapidity();
                decorator_phi_vis(*part)     = pjItr->phi_std(); //returns phi in [-pi,pi]
                decorator_m_vis(*part)       = pjItr->m();
              }
              else {
                decorator_e(*part)       = pjItr->e();
                decorator_pt(*part)      = pjItr->pt();
                decorator_eta(*part)     = pjItr->pseudorapidity();
                decorator_phi(*part)     = pjItr->phi_std(); //returns phi in [-pi,pi]
              }
              found=true;
              break;
            } // Found the matching unique ID
          } // Loop over the jet constituents
          if (found){
            for(const auto& constit : constituents) {
              photon_uniqueIDs.push_back(constit.user_index());
            } // Loop over the constituents
          } // Found one of the key leptons in this jet
          ++pjItr;
        }
        if (!found) {
          if(part->isTau()) {
            decorator_pt_vis(*part)      = 0.;
            decorator_eta_vis(*part)     = 0.;
            decorator_phi_vis(*part)     = 0.;
            decorator_m_vis(*part)       = 0.;
          }
          else {
            decorator_e(*part)       = 0;
            decorator_pt(*part)      = 0;
            decorator_eta(*part)     = 0;
            decorator_phi(*part)     = 0;
          }
          ATH_MSG_WARNING("Bare particle not found in constituents ");
        }
      }
      // Check if we wanted to decorate photons used for dressing
      if (m_decoratePhotons){
        //loop over photons, uniquely associate each to nearest bare particle
        for (const auto* phot : photonsFSRList ) {
          bool found = std::find(photon_uniqueIDs.begin(), photon_uniqueIDs.end(), HepMC::uniqueID(phot)) != photon_uniqueIDs.end();
          if (found) {
            dressDec(*phot) = 1;
          }
        } // End of loop over photons
      } // End of decoration of photons used in dressing
    } // End of anti-kT dressing
 
    return StatusCode::SUCCESS;
}

initialize()

StatusCode DerivationFramework::TruthDressingTool::initialize ( )

finaloverridevirtual

Definition at line 32 of file TruthDressingTool.cxx.

{
    // Initialise handle keys
    ATH_CHECK(m_particlesKey.initialize());
    ATH_CHECK(m_decorationKey.initialize(SG::AllowEmpty));
 
    ATH_CHECK(m_dressParticlesKey.initialize());
    ATH_CHECK(m_decorator_eKey.initialize());
    ATH_CHECK(m_decorator_ptKey.initialize());
    ATH_CHECK(m_decorator_etaKey.initialize());
    ATH_CHECK(m_decorator_phiKey.initialize());
    ATH_CHECK(m_decorator_pt_visKey.initialize());
    ATH_CHECK(m_decorator_eta_visKey.initialize());
    ATH_CHECK(m_decorator_phi_visKey.initialize());
    ATH_CHECK(m_decorator_m_visKey.initialize());
    ATH_CHECK(m_decorator_nphotonKey.initialize());
 
    ATH_CHECK(m_truthClassKey.initialize());
 
    // ensure we are not mixing truth taus with other truth particles
    if (m_listOfPIDs.size() > 1) {
      if (std::find(m_listOfPIDs.begin(), m_listOfPIDs.end(), MC::TAU) != m_listOfPIDs.end()) {
        ATH_MSG_ERROR("Truth taus must be dressed separately from other truth particles");
        return StatusCode::FAILURE;
      }
    }
 
    return StatusCode::SUCCESS;
}

Member Data Documentation

m_coneSize

Gaudi::Property<float> DerivationFramework::TruthDressingTool::m_coneSize {this, "dressingConeSize", 0.1, "Size of dR cone in which to include FSR photons in dressing"}

private

Parameter: Cone size for dressing.

Definition at line 69 of file TruthDressingTool.h.

m_decoratePhotons

Gaudi::Property<bool> DerivationFramework::TruthDressingTool::m_decoratePhotons {this, "decoratePhotons", true, "Do we want to decorate the photons used for dressing?"}

private

Parameter: Do we want to decorate the photons used for dressing?

Definition at line 78 of file TruthDressingTool.h.

m_decorationKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorationKey {this, "decorationName", m_particlesKey, "unusedPhotonDecoration", "Name of the decoration for photons that were used in dressing"}

private

WriteDecorHandleKeys for decorations for particlesKey.

Definition at line 35 of file TruthDressingTool.h.

m_decorator_eKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_eKey {this, "e_dressed", m_dressParticlesKey, "e_dressed", "e_dressed decoration"}

private

Definition at line 41 of file TruthDressingTool.h.

m_decorator_eta_visKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_eta_visKey {this, "eta_vis_dressed", m_dressParticlesKey, "eta_vis_dressed", "eta_vis_dressed decoration"}

private

Definition at line 51 of file TruthDressingTool.h.

m_decorator_etaKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_etaKey {this, "eta_dressed", m_dressParticlesKey, "eta_dressed", "eta_dressed decoration"}

private

Definition at line 45 of file TruthDressingTool.h.

m_decorator_m_visKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_m_visKey {this, "m_vis_dressed", m_dressParticlesKey, "m_vis_dressed", "m_vis_dressed decoration"}

private

Definition at line 55 of file TruthDressingTool.h.

m_decorator_nphotonKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_nphotonKey {this, "nPhotons_dressed", m_dressParticlesKey, "nPhotons_dressed", "nPhotons_dressed decoration"}

private

Definition at line 57 of file TruthDressingTool.h.

m_decorator_phi_visKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_phi_visKey {this, "phi_vis_dressed", m_dressParticlesKey, "phi_vis_dressed", "phi_vis_dressed decoration"}

private

Definition at line 53 of file TruthDressingTool.h.

m_decorator_phiKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_phiKey {this, "phi_dressed", m_dressParticlesKey, "phi_dressed", "phi_dressed decoration"}

private

Definition at line 47 of file TruthDressingTool.h.

m_decorator_pt_visKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_pt_visKey {this, "pt_vis_dressed", m_dressParticlesKey, "pt_vis_dressed", "pt_vis_dressed decoration"}

private

Definition at line 49 of file TruthDressingTool.h.

m_decorator_ptKey

SG::WriteDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_decorator_ptKey {this, "pt_dressed", m_dressParticlesKey, "pt_dressed", "pt_dressed decoration"}

private

Definition at line 43 of file TruthDressingTool.h.

m_dressParticlesKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_dressParticlesKey {this, "dressParticlesKey", "", "ReadHandleKey for input particles to be dressed. If taus are selected, everything in this input key will be used"}

private

ReadHandleKey for particles to be dressed.

Definition at line 38 of file TruthDressingTool.h.

m_listOfPIDs

Gaudi::Property< std::vector<int> > DerivationFramework::TruthDressingTool::m_listOfPIDs {this, "particleIDsToDress", {11,13}, "List of the pdgID's of particles to be dressed (usually 11,13). Special treatment for taus (15)"}

private

Parameter: List of pdgIDs of particles to dress.

Definition at line 72 of file TruthDressingTool.h.

m_particlesKey

SG::ReadHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_particlesKey {this, "particlesKey", "TruthParticles", "ReadHandleKey for TruthParticles for photon list input"}

private

ReadHandleKey input collection key.

Definition at line 32 of file TruthDressingTool.h.

m_truthClassKey

SG::ReadDecorHandleKey<xAOD::TruthParticleContainer> DerivationFramework::TruthDressingTool::m_truthClassKey {this, "truthClassifierKey", m_dressParticlesKey, "Classification"}

private

To ensure that the algorithm is scheduled after the truth classifier.

Definition at line 60 of file TruthDressingTool.h.

m_useAntiKt

Gaudi::Property<bool> DerivationFramework::TruthDressingTool::m_useAntiKt {this, "useAntiKt", false, "use anti-k_T instead of fixed-cone dressing"}

private

Parameter: Use antikT algorithm for dressing?

Definition at line 75 of file TruthDressingTool.h.

m_useLeptonsFromHadrons

Gaudi::Property<bool> DerivationFramework::TruthDressingTool::m_useLeptonsFromHadrons {this, "useLeptonsFromHadrons", false, "Consider leptons coming from hadron decays?"}

private

Parameter: Use leptons from hadron decays?

Definition at line 66 of file TruthDressingTool.h.

m_usePhotonsFromHadrons

Gaudi::Property<bool> DerivationFramework::TruthDressingTool::m_usePhotonsFromHadrons {this, "usePhotonsFromHadrons", false, "Add photons coming from hadron decays while dressing"}

private

Parameter: Use photons from hadron decays?

Definition at line 63 of file TruthDressingTool.h.

---

The documentation for this class was generated from the following files:

• TruthDressingTool.h
• TruthDressingTool.cxx