ATLAS Offline Software
Loading...
Searching...
No Matches
DerivationFramework::TauThinningTool Class Reference

#include <TauThinningTool.h>

Inheritance diagram for DerivationFramework::TauThinningTool:
Collaboration diagram for DerivationFramework::TauThinningTool:

Public Member Functions

 TauThinningTool (const std::string &t, const std::string &n, const IInterface *p)
virtual ~TauThinningTool ()=default
virtual StatusCode initialize () override
virtual StatusCode finalize () override
virtual StatusCode doThinning () const override

Private Attributes

std::atomic< unsigned int > m_ntot = 0
std::atomic< unsigned int > m_npass = 0
StringProperty m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
Gaudi::Property< std::string > m_selectionString { this, "SelectionString", "", "" }
SG::ThinningHandleKey< xAOD::TauJetContainerm_taus { this, "Taus", "TauJets", "" }
SG::ThinningHandleKey< xAOD::TauTrackContainerm_tauTracks { this, "TauTracks", "TauTracks", "" }
SG::ThinningHandleKey< xAOD::TrackParticleContainerm_trackParticles { this, "TrackParticles", "InDetTrackParticles", "" }
SG::ThinningHandleKey< xAOD::PFOContainerm_neutralPFOs { this, "TauNeutralPFOs", "TauNeutralParticleFlowObjects", "" }
SG::ThinningHandleKey< xAOD::VertexContainerm_secondaryVertices { this, "TauSecondaryVertices", "TauSecondaryVertices", "" }

Detailed Description

Definition at line 24 of file TauThinningTool.h.

Constructor & Destructor Documentation

◆ TauThinningTool()

DerivationFramework::TauThinningTool::TauThinningTool ( const std::string & t,
const std::string & n,
const IInterface * p )

Definition at line 12 of file TauThinningTool.cxx.

14 :
15 base_class(t,n,p)
16{
17}

◆ ~TauThinningTool()

virtual DerivationFramework::TauThinningTool::~TauThinningTool ( )
virtualdefault

Member Function Documentation

◆ doThinning()

StatusCode DerivationFramework::TauThinningTool::doThinning ( ) const
overridevirtual

Definition at line 45 of file TauThinningTool.cxx.

46{
47 const EventContext& ctx = Gaudi::Hive::currentContext();
48
49 // retrieve containers and thin them
50 SG::ThinningHandle<xAOD::TauJetContainer> taus(m_taus, ctx);
51 taus.thinAll();
52 size_t nTaus = taus->size();
53
54 SG::ThinningHandle<xAOD::TauTrackContainer> tauTracks(m_tauTracks, ctx);
55 tauTracks.thinAll();
56
57 SG::ThinningHandle<xAOD::TrackParticleContainer> trackParticles(m_trackParticles, ctx);
58 trackParticles.thinAll();
59
60 SG::ThinningHandle<xAOD::PFOContainer> neutralPFOs(m_neutralPFOs, ctx);
61 neutralPFOs.thinAll();
62
63 SG::ThinningHandle<xAOD::VertexContainer> secondaryVertices(m_secondaryVertices, ctx);
64 secondaryVertices.thinAll();
65
66
67 std::vector<const xAOD::TauJet*> tausToKeep;
68
69 // execute the text parser if requested
70 if (!m_selectionString.empty()) {
71 std::vector<int> entries = m_parser->evaluateAsVector();
72 size_t nEntries = entries.size();
73 if (nTaus != nEntries) {
74 ATH_MSG_ERROR("Incompatible sizes: " << nTaus << " vs " << nEntries << "! Please check your selection string uses the appropriate tau container.");
75 return StatusCode::FAILURE;
76 }
77 // identify which taus to keep
78 for (size_t i=0; i<nTaus; ++i) if (entries[i]==1) tausToKeep.push_back(taus->at(i));
79 }
80 // use all taus if no selection string is passed
81 else {
82 for (size_t i=0; i<nTaus; ++i) tausToKeep.push_back(taus->at(i));
83 }
84
85 // protection against duplicate taus -- built from different seed jets, but end up having same (eta,phi)
86 if( tausToKeep.size() > 0){
87 for(size_t i=0; i < tausToKeep.size()-1; i++){
88 const auto* aTau=tausToKeep[i];
89 auto it = std::remove_if(tausToKeep.begin()+i+1,tausToKeep.end(),[aTau](const xAOD::TauJet* bTau) {return aTau->p4().DeltaR(bTau->p4()) < 0.01;});
90 tausToKeep.erase (it, tausToKeep.end());
91 }
92 }
93
94 // keep the various tau-related objects for taus passing the selection
95 for (const auto* tau : tausToKeep) {
96 // tau
97 taus.keep(tau->index());
98
99 // classifiedCharged tau tracks
100 for (const xAOD::TauTrack* track : tau->tracks()) {
101 tauTracks.keep(track->index());
102
103 // associated ID track
104 trackParticles.keep(track->track()->index());
105 }
106
107 // neutral PFOs
108 for (size_t i=0; i<tau->nNeutralPFOs(); i++) {
109 neutralPFOs.keep(tau->neutralPFO(i)->index());
110 }
111
112 // secondary vertex
113 if (tau->secondaryVertex() != nullptr) {
114 secondaryVertices.keep(tau->secondaryVertex()->index());
115 }
116 }
117
118 // increment counters
119 m_npass += tausToKeep.size();
120 m_ntot += nTaus;
121
122 return StatusCode::SUCCESS;
123}
#define ATH_MSG_ERROR(x)
static Double_t taus
SG::ThinningHandleKey< xAOD::PFOContainer > m_neutralPFOs
std::atomic< unsigned int > m_npass
Gaudi::Property< std::string > m_selectionString
SG::ThinningHandleKey< xAOD::TauJetContainer > m_taus
std::atomic< unsigned int > m_ntot
SG::ThinningHandleKey< xAOD::TauTrackContainer > m_tauTracks
SG::ThinningHandleKey< xAOD::VertexContainer > m_secondaryVertices
SG::ThinningHandleKey< xAOD::TrackParticleContainer > m_trackParticles
double entries
Definition listroot.cxx:49
DataModel_detail::iterator< DVL > remove_if(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end, Predicate pred)
Specialization of remove_if for DataVector/List.
TauTrack_v1 TauTrack
Definition of the current version.
Definition TauTrack.h:16
TauJet_v3 TauJet
Definition of the current "tau version".

◆ finalize()

StatusCode DerivationFramework::TauThinningTool::finalize ( )
overridevirtual

Definition at line 37 of file TauThinningTool.cxx.

38{
39 ATH_MSG_INFO("Processed " << m_ntot << " taus, " << m_npass << " were kept");
40 ATH_CHECK( finalizeParser() );
41 return StatusCode::SUCCESS;
42}
#define ATH_CHECK
Evaluate an expression and check for errors.
#define ATH_MSG_INFO(x)

◆ initialize()

StatusCode DerivationFramework::TauThinningTool::initialize ( )
overridevirtual

Definition at line 20 of file TauThinningTool.cxx.

21{
22 ATH_CHECK( m_taus.initialize(m_streamName) );
23 ATH_CHECK( m_tauTracks.initialize(m_streamName) );
27
28 // set up the text-parsing machinery for selecting taus according to user cuts
29 if (!m_selectionString.empty()) {
30 ATH_MSG_INFO("Selection string for " << m_taus.key() << ": " << m_selectionString);
31 ATH_CHECK( initializeParser(m_selectionString) );
32 }
33 return StatusCode::SUCCESS;
34}

Member Data Documentation

◆ m_neutralPFOs

SG::ThinningHandleKey<xAOD::PFOContainer> DerivationFramework::TauThinningTool::m_neutralPFOs { this, "TauNeutralPFOs", "TauNeutralParticleFlowObjects", "" }
private

Definition at line 40 of file TauThinningTool.h.

40{ this, "TauNeutralPFOs", "TauNeutralParticleFlowObjects", "" };

◆ m_npass

std::atomic<unsigned int> DerivationFramework::TauThinningTool::m_npass = 0
mutableprivate

Definition at line 34 of file TauThinningTool.h.

◆ m_ntot

std::atomic<unsigned int> DerivationFramework::TauThinningTool::m_ntot = 0
mutableprivate

Definition at line 33 of file TauThinningTool.h.

◆ m_secondaryVertices

SG::ThinningHandleKey<xAOD::VertexContainer> DerivationFramework::TauThinningTool::m_secondaryVertices { this, "TauSecondaryVertices", "TauSecondaryVertices", "" }
private

Definition at line 41 of file TauThinningTool.h.

41{ this, "TauSecondaryVertices", "TauSecondaryVertices", "" };

◆ m_selectionString

Gaudi::Property<std::string> DerivationFramework::TauThinningTool::m_selectionString { this, "SelectionString", "", "" }
private

Definition at line 36 of file TauThinningTool.h.

36{ this, "SelectionString", "", "" };

◆ m_streamName

StringProperty DerivationFramework::TauThinningTool::m_streamName { this, "StreamName", "", "Name of the stream being thinned" }
private

Definition at line 35 of file TauThinningTool.h.

35{ this, "StreamName", "", "Name of the stream being thinned" };

◆ m_taus

SG::ThinningHandleKey<xAOD::TauJetContainer> DerivationFramework::TauThinningTool::m_taus { this, "Taus", "TauJets", "" }
private

Definition at line 37 of file TauThinningTool.h.

37{ this, "Taus", "TauJets", "" };

◆ m_tauTracks

SG::ThinningHandleKey<xAOD::TauTrackContainer> DerivationFramework::TauThinningTool::m_tauTracks { this, "TauTracks", "TauTracks", "" }
private

Definition at line 38 of file TauThinningTool.h.

38{ this, "TauTracks", "TauTracks", "" };

◆ m_trackParticles

SG::ThinningHandleKey<xAOD::TrackParticleContainer> DerivationFramework::TauThinningTool::m_trackParticles { this, "TrackParticles", "InDetTrackParticles", "" }
private

Definition at line 39 of file TauThinningTool.h.

39{ this, "TrackParticles", "InDetTrackParticles", "" };

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