DerivationFramework::TauIDDecoratorWrapper Class Reference

#include <TauIDDecoratorWrapper.h>

Inheritance diagram for DerivationFramework::TauIDDecoratorWrapper:

Collaboration diagram for DerivationFramework::TauIDDecoratorWrapper:

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

Private Attributes
SG::ReadHandleKey< xAOD::TauJetContainer >	m_tauContainerKey { this, "TauContainerName", "TauJets", "Input tau container key" }
SG::ReadHandleKey< xAOD::MuonContainer >	m_muonContainerKey { this, "MuonContainerName", "Muons", "Input muon container key" }
SG::WriteDecorHandleKeyArray< xAOD::TauJetContainer >	m_scoreDecorKeys { this, "ScoreDecorationKeys", m_tauContainerKey, {}, "List of score decorations added to the tau"}
SG::WriteDecorHandleKeyArray< xAOD::TauJetContainer >	m_WPDecorKeys { this, "WPDecorationKeys", m_tauContainerKey, {}, "List of WP decorations added to the tau"}
SG::WriteDecorHandleKey< xAOD::TauJetContainer >	m_trackWidthKey { this, "TrackWidthKey", m_tauContainerKey, "trackWidth", "Track width decoration name"}
SG::WriteDecorHandleKey< xAOD::TauJetContainer >	m_passTATTauMuonOLRKey { this, "PassTATTauMuonOLRKey", m_tauContainerKey, "passTATTauMuonOLR", "Decoration name"}
ToolHandleArray< TauRecToolBase >	m_tauIDTools { this, "TauIDTools", {}, "" }
Gaudi::Property< bool >	m_doEvetoWP {this, "DoEvetoWP", false}

Detailed Description

Definition at line 26 of file TauIDDecoratorWrapper.h.

Member Function Documentation

addBranches()

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

finaloverridevirtual

Definition at line 29 of file TauIDDecoratorWrapper.cxx.

  {
 
    // retrieve tau container
    SG::ReadHandle<xAOD::TauJetContainer> tauJetsReadHandle(m_tauContainerKey, ctx);
    if (!tauJetsReadHandle.isValid()) {
      ATH_MSG_ERROR ("Could not retrieve TauJetContainer with key " << tauJetsReadHandle.key());
      return StatusCode::FAILURE;
    }
    const xAOD::TauJetContainer* tauContainer = tauJetsReadHandle.cptr();
 
    //Create accessors
    static const SG::Accessor<float> acc_absEtaLead("ABS_ETA_LEAD_TRACK");
 
    std::vector<SG::WriteDecorHandle<xAOD::TauJetContainer, float> > scoreDecors;
    scoreDecors.reserve (m_scoreDecorKeys.size());
    for (const SG::WriteDecorHandleKey<xAOD::TauJetContainer>& k : m_scoreDecorKeys) {
      scoreDecors.emplace_back (k, ctx);
    }
    std::vector<SG::WriteDecorHandle<xAOD::TauJetContainer, char> > WPDecors;
    WPDecors.reserve (m_WPDecorKeys.size());
    for (const SG::WriteDecorHandleKey<xAOD::TauJetContainer>& k : m_WPDecorKeys) {
      WPDecors.emplace_back (k, ctx);
    }
 
    SG::WriteDecorHandle<xAOD::TauJetContainer, float> dec_trackWidth (m_trackWidthKey, ctx);
    SG::WriteDecorHandle<xAOD::TauJetContainer, bool> dec_passTATTauMuonOLR (m_passTATTauMuonOLRKey, ctx);
    for (const auto tau : *tauContainer) {
      float tauTrackBasedWidth = 0.;
      // equivalent to tracks(xAOD::TauJetParameters::TauTrackFlag::classifiedCharged)
      std::vector<const xAOD::TauTrack *> tauTracks = tau->tracks();
      for (const xAOD::TauTrack *trk : tau->tracks(xAOD::TauJetParameters::TauTrackFlag::classifiedIsolation)) {
        tauTracks.push_back(trk);
      }
      double sumWeightedDR = 0.;
      double ptSum = 0.;
      for (const xAOD::TauTrack *track : tauTracks) {
        double deltaR = tau->p4().DeltaR(track->p4());
        sumWeightedDR += deltaR * track->pt();
        ptSum += track->pt();
      }
      if (ptSum > 0.) {
        tauTrackBasedWidth = sumWeightedDR / ptSum;
      }
 
      dec_trackWidth(*tau) = tauTrackBasedWidth;
    }
 
    // create shallow copy
    auto shallowCopy = xAOD::shallowCopyContainer (*tauContainer);
 
    for (auto tau : *shallowCopy.first) {
 
      // ABS_ETA_LEAD_TRACK is removed from the AOD content and must be redecorated when computing eVeto WPs
      // note: this redecoration is not robust against charged track thinning, but charged tracks should never be thinned
      if (m_doEvetoWP) {
        float absEtaLead = -1111.;
        if(tau->nTracks() > 0) {
          const xAOD::TrackParticle* track = tau->track(0)->track();
          absEtaLead = std::abs( track->eta() );
        }
        acc_absEtaLead(*tau) = absEtaLead;
      }
 
      // pass the shallow copy to the tools
      for (const auto& tool : m_tauIDTools) {
        ATH_CHECK( tool->execute(*tau) );
      }
 
      // copy over the relevant decorations (scores and working points)
      const xAOD::TauJet* xTau = tauContainer->at(tau->index());
      for (SG::WriteDecorHandle<xAOD::TauJetContainer, float>& dec : scoreDecors) {
        SG::ConstAccessor<float> scoreAcc (dec.auxid());
        dec(*xTau) = scoreAcc(*tau);
      }
      for (SG::WriteDecorHandle<xAOD::TauJetContainer, char>& dec : WPDecors) {
        SG::ConstAccessor<char> WPAcc (dec.auxid());
        dec(*xTau) = WPAcc(*tau);
      }
    }
 
    delete shallowCopy.first;
    delete shallowCopy.second;
 
    // add TauAnalysisTool MuonOLR
    SG::ReadHandle<xAOD::MuonContainer> muonReadHandle(m_muonContainerKey, ctx);
    if (!muonReadHandle.isValid()) {
      ATH_MSG_DEBUG ("Could not retrieve MuonContainer with key " << muonReadHandle.key() << " so won't add TAT MuonOLR flag");
      return StatusCode::SUCCESS;
    }
    const xAOD::MuonContainer* muonContainer = muonReadHandle.cptr();
 
    for (const auto tau : *tauContainer) {
      bool bTauMuonOLR = true;
      for (auto muon : *muonContainer){
        if(muon->pt() < 2000.) continue; // pt > 2 GeV
        if(muon->muonType() == xAOD::Muon::CaloTagged) continue; // not calo-tagged
        if(muon->p4().DeltaR( tau->p4() ) > 0.2 ) continue; // delta R < 0.2
        bTauMuonOLR = false; // muon-tau overlapped
        break;
      }
      dec_passTATTauMuonOLR(*tau) = bTauMuonOLR;
    }
 
    return StatusCode::SUCCESS;
  }

initialize()

StatusCode DerivationFramework::TauIDDecoratorWrapper::initialize ( )

finaloverridevirtual

Definition at line 12 of file TauIDDecoratorWrapper.cxx.

  {
    // initialize tauRecTools tools
    ATH_CHECK( m_tauIDTools.retrieve() );
 
    // initialize read/write handle keys
    ATH_CHECK( m_tauContainerKey.initialize() );
    ATH_CHECK( m_muonContainerKey.initialize() );
    ATH_CHECK( m_scoreDecorKeys.initialize() );
    ATH_CHECK( m_WPDecorKeys.initialize() );
    ATH_CHECK( m_trackWidthKey.initialize() );
    ATH_CHECK( m_passTATTauMuonOLRKey.initialize() );
 
    return StatusCode::SUCCESS;
  }

Member Data Documentation

m_doEvetoWP

Gaudi::Property<bool> DerivationFramework::TauIDDecoratorWrapper::m_doEvetoWP {this, "DoEvetoWP", false}

private

Definition at line 43 of file TauIDDecoratorWrapper.h.

{this, "DoEvetoWP", false};

m_muonContainerKey

SG::ReadHandleKey<xAOD::MuonContainer> DerivationFramework::TauIDDecoratorWrapper::m_muonContainerKey { this, "MuonContainerName", "Muons", "Input muon container key" }

private

Definition at line 36 of file TauIDDecoratorWrapper.h.

{ this, "MuonContainerName", "Muons", "Input muon container key" };

m_passTATTauMuonOLRKey

SG::WriteDecorHandleKey<xAOD::TauJetContainer> DerivationFramework::TauIDDecoratorWrapper::m_passTATTauMuonOLRKey { this, "PassTATTauMuonOLRKey", m_tauContainerKey, "passTATTauMuonOLR", "Decoration name"}

private

Definition at line 40 of file TauIDDecoratorWrapper.h.

{ this, "PassTATTauMuonOLRKey", m_tauContainerKey, "passTATTauMuonOLR", "Decoration name"};

m_scoreDecorKeys

SG::WriteDecorHandleKeyArray<xAOD::TauJetContainer> DerivationFramework::TauIDDecoratorWrapper::m_scoreDecorKeys { this, "ScoreDecorationKeys", m_tauContainerKey, {}, "List of score decorations added to the tau"}

private

Definition at line 37 of file TauIDDecoratorWrapper.h.

{ this, "ScoreDecorationKeys", m_tauContainerKey, {}, "List of score decorations added to the tau"};

m_tauContainerKey

SG::ReadHandleKey<xAOD::TauJetContainer> DerivationFramework::TauIDDecoratorWrapper::m_tauContainerKey { this, "TauContainerName", "TauJets", "Input tau container key" }

private

Definition at line 35 of file TauIDDecoratorWrapper.h.

{ this, "TauContainerName", "TauJets", "Input tau container key" };

m_tauIDTools

ToolHandleArray<TauRecToolBase> DerivationFramework::TauIDDecoratorWrapper::m_tauIDTools { this, "TauIDTools", {}, "" }

private

Definition at line 42 of file TauIDDecoratorWrapper.h.

{ this, "TauIDTools", {}, "" };

m_trackWidthKey

SG::WriteDecorHandleKey<xAOD::TauJetContainer> DerivationFramework::TauIDDecoratorWrapper::m_trackWidthKey { this, "TrackWidthKey", m_tauContainerKey, "trackWidth", "Track width decoration name"}

private

Definition at line 39 of file TauIDDecoratorWrapper.h.

{ this, "TrackWidthKey", m_tauContainerKey, "trackWidth", "Track width decoration name"};

m_WPDecorKeys

SG::WriteDecorHandleKeyArray<xAOD::TauJetContainer> DerivationFramework::TauIDDecoratorWrapper::m_WPDecorKeys { this, "WPDecorationKeys", m_tauContainerKey, {}, "List of WP decorations added to the tau"}

private

Definition at line 38 of file TauIDDecoratorWrapper.h.

{ this, "WPDecorationKeys", m_tauContainerKey, {}, "List of WP decorations added to the tau"};

---

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

• TauIDDecoratorWrapper.h
• TauIDDecoratorWrapper.cxx