ATLAS Offline Software
Loading...
Searching...
No Matches
TauFilter Class Reference

#include <TauFilter.h>

Inheritance diagram for TauFilter:
Collaboration diagram for TauFilter:

Public Member Functions

 TauFilter (const std::string &name, ISvcLocator *pSvcLocator)
StatusCode filterInitialize ()
StatusCode filterFinalize ()
StatusCode filterEvent (const EventContext &ctx)
CLHEP::HepLorentzVector sumDaughterNeutrinos (const HepMC::ConstGenParticlePtr &tau)
virtual StatusCode sysInitialize () override
 Override sysInitialize.
virtual const DataObjIDColl & extraOutputDeps () const override
 Return the list of extra output dependencies.
bool filterPassed (const EventContext &ctx) const
void setFilterPassed (bool state, const EventContext &ctx) const
ServiceHandle< StoreGateSvc > & evtStore ()
 The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > & detStore () const
 The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode sysStart () override
 Handle START transition.
virtual std::vector< Gaudi::DataHandle * > inputHandles () const override
 Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * > outputHandles () const override
 Return this algorithm's output handles.
Gaudi::Details::PropertyBase & declareProperty (Gaudi::Property< T, V, H > &t)
void updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream & msg () const
bool msgLvl (const MSG::Level lvl) const
Event loop algorithm methods: not to be overloaded
StatusCode initialize ()
StatusCode execute (const EventContext &ctx)
 Execute method.
StatusCode finalize ()
Counters
int nPassed () const
int nFailed () const
int nNeeded () const
bool keepAll () const
Event collection accessors (const and non-const)
HepMC::GenEvent *event ATLAS_NOT_CONST_THREAD_SAFE ()
 Access the current signal event (first in the McEventCollection).
McEventCollection *events ATLAS_NOT_CONST_THREAD_SAFE ()
 Access the current event's McEventCollection.
const HepMC::GenEventevent_const (const EventContext &ctx) const
 Access the current signal event (const).
const McEventCollectionevents_const (const EventContext &ctx) const
 Access the current event's McEventCollection (const).
Particle data accessors
const ServiceHandle< IPartPropSvc > partPropSvc () const
 Access the particle property service.
const HepPDT::ParticleDataTable & particleTable () const
 Get a particle data table.
const HepPDT::ParticleDataTable & pdt () const
 Shorter alias to get a particle data table.
const HepPDT::ParticleData * particleData (int pid) const
 Access an element in the particle data table.
const EventContext & getContext () const
 Deprecated methods (use the ones with EventContext).
bool filterPassed () const
void setFilterPassed (bool state) const

Protected Member Functions

virtual bool isReEntrant () const override final
 Legacy algorithms are not thread-safe.
void renounceArray (SG::VarHandleKeyArray &handlesArray)
 remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce (T &h)
void extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
 Add StoreName to extra input/output deps as needed.

Protected Attributes

Counters and requirements
int m_nPass
int m_nFail
int m_nNeeded
bool m_keepAll

Private Types

typedef ServiceHandle< StoreGateSvcStoreGateSvc_t

Private Member Functions

CLHEP::HepRandomEngine * getRandomEngine (const std::string &streamName, const EventContext &ctx) const
Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

ServiceHandle< IAthRNGSvcm_rndmSvc {this, "RndmSvc", "AthRNGSvc"}
int m_Ntau
double m_etaMaxe
double m_etaMaxmu
double m_etaMaxhad
double m_pTmine
double m_pTminmu
double m_pTminhad
bool m_NewOpt
int m_Nleptau
int m_Nhadtau
double m_etaMaxlep
double m_pTminlep
double m_pTminlep_lead
double m_pTminhad_lead
bool m_ReverseFilter
bool m_HasTightRegion
double m_LooseRejectionFactor
double m_pTminlep_tight
double m_pTminlep_tight_lead
double m_pTminhad_tight
double m_pTminhad_tight_lead
int m_filterEventNumber
bool m_useMaxNTaus
int m_maxNtau
int m_maxNhadtau
int m_maxNleptau
double m_events [6]
double m_events_sel [6]
double m_eventse
double m_eventsmu
double m_eventshad
double m_eventseacc
double m_eventsmuacc
double m_eventshadacc
DataObjIDColl m_extendedExtraObjects
StoreGateSvc_t m_evtStore
 Pointer to StoreGate (event store by default).
StoreGateSvc_t m_detStore
 Pointer to StoreGate (detector store by default).
std::vector< SG::VarHandleKeyArray * > m_vhka
bool m_varHandleArraysDeclared

Properties

ServiceHandle< IPartPropSvc > m_ppSvc {this, "PartPropSvc", "PartPropSvc"}
 Handle on the particle property service.
SG::ReadHandleKey< McEventCollectionm_mcevents_const { this, "McEventKey", "GEN_EVENT", "StoreGate key of the MC event collection" }
 Const handle to the MC event collection.
std::string m_mcEventKey {}
 StoreGate key for the MC event collection (defaults to GEN_EVENT).
BooleanProperty m_mkMcEvent {this, "MakeMcEvent", false, "Create a new MC event collection if it doesn't exist"}
 Flag to determine if a new MC event collection should be made if it doesn't exist.

Detailed Description

Definition at line 20 of file TauFilter.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::StoreGateSvc_t
privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

◆ TauFilter()

TauFilter::TauFilter ( const std::string & name,
ISvcLocator * pSvcLocator )

Definition at line 10 of file TauFilter.cxx.

11 : GenFilter( name,pSvcLocator ),
15{
16 declareProperty( "Ntaus", m_Ntau = 1 );
17 declareProperty( "MaxNtaus", m_maxNtau = 100 );
18 declareProperty( "EtaMaxe", m_etaMaxe = 2.5 );
19 declareProperty( "EtaMaxmu", m_etaMaxmu = 2.5 );
20 declareProperty( "EtaMaxhad", m_etaMaxhad = 2.5 );
21
22 declareProperty( "Ptcute", m_pTmine = 12000.0 );
23 declareProperty( "Ptcutmu", m_pTminmu = 12000.0 );
24 declareProperty( "Ptcuthad", m_pTminhad = 12000.0 );
25
26 // new options:
27 declareProperty( "UseNewOptions", m_NewOpt = false );
28 declareProperty( "UseMaxNTaus", m_useMaxNTaus = false );
29 declareProperty( "Nhadtaus", m_Nhadtau = 0 );
30 declareProperty( "MaxNhadtaus", m_maxNhadtau = 100 );
31 declareProperty( "Nleptaus", m_Nleptau = 0 );
32 declareProperty( "MaxNleptaus", m_maxNleptau = 100 );
33 declareProperty( "EtaMaxlep", m_etaMaxlep = 2.6 );
34 declareProperty( "Ptcutlep", m_pTminlep = 7000.0 );
35 declareProperty( "Ptcutlep_lead", m_pTminlep_lead = 7000.0);
36 declareProperty( "Ptcuthad_lead", m_pTminhad_lead = 12000.0 );
37 declareProperty( "ReverseFilter", m_ReverseFilter = false);
38
39 declareProperty( "HasTightRegion", m_HasTightRegion = false);
40 declareProperty( "LooseRejectionFactor", m_LooseRejectionFactor = 1);
41 declareProperty( "Ptcutlep_tight", m_pTminlep_tight = 7000.0 );
42 declareProperty( "Ptcutlep_tight_lead", m_pTminlep_tight_lead = 7000.0 );
43 declareProperty( "Ptcuthad_tight", m_pTminhad_tight = 12000.0 );
44 declareProperty( "Ptcuthad_tight_lead", m_pTminhad_tight_lead = 12000.0 );
45
46 declareProperty( "filterEventNumber", m_filterEventNumber = 0 );
47}
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
GenFilter(const std::string &name, ISvcLocator *pSvcLocator)
Definition GenFilter.cxx:8
double m_etaMaxhad
Definition TauFilter.h:40
bool m_NewOpt
Definition TauFilter.h:47
double m_pTmine
Definition TauFilter.h:42
double m_pTminhad
Definition TauFilter.h:44
int m_maxNtau
Definition TauFilter.h:65
double m_pTminmu
Definition TauFilter.h:43
bool m_useMaxNTaus
Definition TauFilter.h:64
bool m_ReverseFilter
Definition TauFilter.h:54
double m_pTminlep_tight_lead
Definition TauFilter.h:58
double m_eventshad
Definition TauFilter.h:77
double m_eventseacc
Definition TauFilter.h:79
double m_eventsmuacc
Definition TauFilter.h:80
double m_pTminhad_tight
Definition TauFilter.h:59
double m_LooseRejectionFactor
Definition TauFilter.h:56
double m_pTminlep_tight
Definition TauFilter.h:57
int m_maxNhadtau
Definition TauFilter.h:66
double m_pTminhad_lead
Definition TauFilter.h:53
double m_etaMaxmu
Definition TauFilter.h:39
double m_pTminhad_tight_lead
Definition TauFilter.h:60
double m_pTminlep
Definition TauFilter.h:51
double m_events_sel[6]
Definition TauFilter.h:73
int m_Ntau
Definition TauFilter.h:37
double m_events[6]
Definition TauFilter.h:72
double m_pTminlep_lead
Definition TauFilter.h:52
int m_Nleptau
Definition TauFilter.h:48
int m_filterEventNumber
Definition TauFilter.h:61
double m_etaMaxlep
Definition TauFilter.h:50
double m_etaMaxe
Definition TauFilter.h:38
double m_eventshadacc
Definition TauFilter.h:81
double m_eventse
Definition TauFilter.h:75
int m_Nhadtau
Definition TauFilter.h:49
double m_eventsmu
Definition TauFilter.h:76
bool m_HasTightRegion
Definition TauFilter.h:55
int m_maxNleptau
Definition TauFilter.h:67

Member Function Documentation

◆ ATLAS_NOT_CONST_THREAD_SAFE() [1/2]

McEventCollection *events GenBase::ATLAS_NOT_CONST_THREAD_SAFE ( )
inherited

Access the current event's McEventCollection.

Note
This function will make a new McEventCollection if there is not already a valid one and MakeMcEvent=True.

◆ ATLAS_NOT_CONST_THREAD_SAFE() [2/2]

HepMC::GenEvent *event GenBase::ATLAS_NOT_CONST_THREAD_SAFE ( )
inlineinherited

Access the current signal event (first in the McEventCollection).

Note
This function will make a new McEventCollection if there is not already a valid one and MakeMcEvent=True.

Definition at line 76 of file GenBase.h.

76 {
77 if (events()->empty())
78 ATH_MSG_ERROR("McEventCollection is empty during first event access");
79 return *(events()->begin());
80 }
#define ATH_MSG_ERROR(x)
static const Attributes_t empty

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareGaudiProperty ( Gaudi::Property< T, V, H > & hndl,
const SG::VarHandleKeyType &  )
inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

158 {
160 hndl.value(),
161 hndl.documentation());
162
163 }

◆ declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::declareProperty ( Gaudi::Property< T, V, H > & t)
inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

145 {
146 typedef typename SG::HandleClassifier<T>::type htype;
148 }
Gaudi::Details::PropertyBase & declareGaudiProperty(Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
specialization for handling Gaudi::Property<SG::VarHandleKey>

◆ detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::detStore ( ) const
inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

◆ event_const()

const HepMC::GenEvent * GenBase::event_const ( const EventContext & ctx) const
inlineinherited

Access the current signal event (const).

Definition at line 83 of file GenBase.h.

83 {
84 const McEventCollection* coll = events_const(ctx);
85 if (coll->empty())
86 ATH_MSG_ERROR("Const McEventCollection is empty during first event access");
87 return *(coll->begin());
88 }
const_iterator begin() const noexcept
Return a const_iterator pointing at the beginning of the collection.
bool empty() const noexcept
Returns true if the collection is empty.
const McEventCollection * events_const(const EventContext &ctx) const
Access the current event's McEventCollection (const).
Definition GenBase.h:97

◆ events_const()

const McEventCollection * GenBase::events_const ( const EventContext & ctx) const
inlineinherited

Access the current event's McEventCollection (const).

Definition at line 97 of file GenBase.h.

97 {
98 SG::ReadHandle<McEventCollection> ret = SG::makeHandle(m_mcevents_const, ctx);
99 if (!ret.isValid())
100 ATH_MSG_ERROR("No McEventCollection found in StoreGate with key " << m_mcevents_const.key());
101 return ret.cptr();
102 }
SG::ReadHandleKey< McEventCollection > m_mcevents_const
Const handle to the MC event collection.
Definition GenBase.h:145
virtual bool isValid() override final
Can the handle be successfully dereferenced?
const_pointer_type cptr()
Dereference the pointer.
SG::ReadCondHandle< T > makeHandle(const SG::ReadCondHandleKey< T > &key, const EventContext &ctx=Gaudi::Hive::currentContext())

◆ evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::evtStore ( )
inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

◆ execute()

StatusCode GenFilter::execute ( const EventContext & ctx)
virtualinherited

Execute method.

Provides access to the EventContext if needed but is non-const as opposed to AthReentrantAlgorithm.

Todo
Probably the filter should only look at the first event... right?

Reimplemented from GenBase.

Definition at line 27 of file GenFilter.cxx.

27 {
28 if (events_const(ctx)->empty()) {
29 ATH_MSG_ERROR("No events found in McEventCollection");
30 return StatusCode::FAILURE;
31 } else if (events_const(ctx)->size() > 1) {
33 ATH_MSG_WARNING("More than one event in current McEventCollection -- which is valid?");
34 }
36 if (filterPassed(ctx) || m_keepAll ) {
37 ATH_MSG_DEBUG("Event passed filter");
38 m_nPass += 1;
39 } else {
40 ATH_MSG_DEBUG("Event failed filter");
41 m_nFail += 1;
42 }
43 // Bail out once we have enough events
44 if (m_nPass >= m_nNeeded && m_nNeeded > 0)
45 sc = StatusCode::FAILURE;
46 return sc;
47}
#define ATH_MSG_WARNING(x)
#define ATH_MSG_DEBUG(x)
static Double_t sc
size_t size() const
Number of registered mappings.
bool filterPassed() const
bool m_keepAll
Definition GenFilter.h:66
int m_nPass
Definition GenFilter.h:63
int m_nNeeded
Definition GenFilter.h:65
virtual StatusCode filterEvent(const EventContext &ctx)=0
int m_nFail
Definition GenFilter.h:64
::StatusCode StatusCode
StatusCode definition for legacy code.

◆ extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps)
protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

◆ extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const
overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 50 of file AthAlgorithm.cxx.

51{
52 // If we didn't find any symlinks to add, just return the collection
53 // from the base class. Otherwise, return the extended collection.
54 if (!m_extendedExtraObjects.empty()) {
56 }
57 return Algorithm::extraOutputDeps();
58}
DataObjIDColl m_extendedExtraObjects

◆ filterEvent()

StatusCode TauFilter::filterEvent ( const EventContext & ctx)
virtual

Implements GenFilter.

Definition at line 105 of file TauFilter.cxx.

105 {
106 // Get random number engine
107 CLHEP::HepRandomEngine* rndm{};
108 if(m_HasTightRegion) {
109 rndm = this->getRandomEngine(name(), ctx);
110 if (!rndm) {
111 ATH_MSG_ERROR("Failed to retrieve random number engine for TauFilter");
112 setFilterPassed(false, ctx);
113 return StatusCode::SUCCESS;
114 }
115 }
116
117 HepMC::ConstGenParticlePtr tau = nullptr;
118 CLHEP::HepLorentzVector mom_tauprod; // will contain the momentum of the products of the tau decay
119 CLHEP::HepLorentzVector tauvis;
120 CLHEP::HepLorentzVector nutau;
121 tau = nullptr;
122 int ntau = 0;
123
124 double ptlep_max = 0;
125 double pthad_max = 0;
126 int ntaulep = 0;
127 int ntauhad = 0;
128 int ntaulep_tight = 0;
129 int ntauhad_tight = 0;
130 double weight = 1;
131
132 for (const HepMC::GenEvent* genEvt : *events_const(ctx)) {
133 int eventNumber = genEvt->event_number();
134
135 if(m_filterEventNumber==1 && (eventNumber%2)==0) {
136 setFilterPassed(false, ctx);
137 return StatusCode::SUCCESS;
138 }
139 else if(m_filterEventNumber==2 && (eventNumber%2)==1) {
140 setFilterPassed(false, ctx);
141 return StatusCode::SUCCESS;
142 }
143
144 auto wgtsC = genEvt->weights();
145 weight = wgtsC.size() > 0 ? wgtsC[0] : 1;
146
147 for (const auto& pitr: *genEvt) {
148 // Look for the first decayed/stable tau
149 if (MC::isTau(pitr) && MC::isPhysical(pitr)) {
150 tau = pitr;
151 ATH_MSG_DEBUG("found tau " << tau );
152 ATH_MSG_DEBUG("pT\t\teta\tphi\tid");
153 ATH_MSG_DEBUG(tau->momentum().perp() << "\t" <<
154 tau->momentum().eta() << "\t" <<
155 tau->momentum().phi() << "\t" <<
156 tau->pdg_id() << "\t");
157
158 int tauType = 0;
159 //TauType initialized as 0. tauType = 1 is an Tau_el, tauType = 2 is an Tau_mu, tauType = 0 is an Tau_had, tauType = 11 is a tau with tau parent, e.g a photon radiation event.
160 // Sherpa has some status code 20 particles without decays
161 if(!tau->end_vertex()) continue;
162 for ( const auto& beg: *(tau->end_vertex()) ) {
163 if ( (beg)->production_vertex() != tau->end_vertex() ) continue;
164
165 else if ( std::abs( (beg)->pdg_id() ) == MC::NU_E ) tauType = 1; //Tau decays into an electron
166
167 else if ( std::abs( (beg)->pdg_id() ) == MC::NU_MU ) tauType = 2; //Tau decays into an muon
168
169 else if ( MC::isTau(beg) ) tauType = 11; //Tau radiates a particle and decays into another tau
170
171 }
172
173 if (tauType == 11) {
174 ATH_MSG_DEBUG("tau has a tau as daughter - skipping");
175 continue;
176 }
177 nutau = sumDaughterNeutrinos(tau);
178
179 ATH_MSG_DEBUG("pT\t\teta\tphi\tlh");
180 ATH_MSG_DEBUG(nutau.perp() << "\t" << nutau.eta() << "\t" << nutau.phi() << "\t" << tauType);
181
182 tauvis.setPx(tau->momentum().px()-nutau.px());
183 tauvis.setPy(tau->momentum().py()-nutau.py());
184 tauvis.setPz(tau->momentum().pz()-nutau.pz());
185 tauvis.setE(tau->momentum().e()-nutau.e());
186
187 ATH_MSG_DEBUG(tauvis.perp() << "\t" << tauvis.eta() << "\t" << tauvis.phi() << "\t" << tauType);
188
189 if ( tauType == 1 ) {
190 if(!m_NewOpt) {
191 m_eventse++;
192 if ( tauvis.perp() < m_pTmine ) continue;
193 if ( std::abs( tauvis.eta() ) > m_etaMaxe ) continue;
194 ntau++;
195 m_eventseacc++;
196 }
197 else {
198 if ( tauvis.perp() < m_pTminlep ) continue;
199 if ( std::abs( tauvis.eta() ) > m_etaMaxlep ) continue;
200 ntaulep++;
201 if ( tauvis.perp() >= ptlep_max ) ptlep_max = tauvis.perp();
202 if ( tauvis.perp() >= m_pTminlep_tight ) ntaulep_tight++;
203 }
204 }
205
206 else if ( tauType == 2 ) {
207 if(!m_NewOpt) {
208 m_eventsmu++;
209 if ( tauvis.perp() < m_pTminmu ) continue;
210 if ( std::abs( tauvis.eta() ) > m_etaMaxmu ) continue;
211 ntau++;
213 }
214 else {
215 if ( tauvis.perp() < m_pTminlep ) continue;
216 if ( std::abs( tauvis.eta() ) > m_etaMaxlep ) continue;
217 ntaulep++;
218 if ( tauvis.perp() >= ptlep_max ) ptlep_max = tauvis.perp();
219 if ( tauvis.perp() >= m_pTminlep_tight ) ntaulep_tight++;
220 }
221 }
222
223 else if ( tauType == 0 ) {
224 m_eventshad++;
225 if ( tauvis.perp() < m_pTminhad ) continue;
226 if ( std::abs( tauvis.eta() ) > m_etaMaxhad ) continue;
227 ntau++;
229 ntauhad++;
230 if ( tauvis.perp() >= pthad_max ) pthad_max = tauvis.perp();
231 if ( tauvis.perp() >= m_pTminhad_tight ) ntauhad_tight++;
232 }
233
234 else {
235 ATH_MSG_DEBUG("Could not find a tauType! Something went wrong.");
236 std::cout << std::endl << std::endl <<"************ COULD NOT FIND A TAU TYPE *****************" << std::endl << std::endl;
237 }
238 }
239 }
240 }
241
242 bool pass1 = ( ntaulep + ntauhad >= m_Ntau //For use with UseNewOptions
243 && ntaulep >= m_Nleptau
244 && ntauhad >= m_Nhadtau
245 && (ntaulep<2 || ptlep_max>=m_pTminlep_lead)
246 && (ntauhad<2 || pthad_max>=m_pTminhad_lead)
247 );
248 bool pass2 = ( ntaulep_tight + ntauhad_tight >= m_Ntau //For use with UseNewOptions + HssTightRegion
249 && ntaulep_tight >= m_Nleptau
250 && ntauhad_tight >= m_Nhadtau
251 && (ntaulep_tight<2 || ptlep_max>=m_pTminlep_tight_lead)
252 && (ntauhad_tight<2 || pthad_max>=m_pTminhad_tight_lead)
253 );
254
255 bool pass3 = (ntaulep + ntauhad >= m_Ntau //For use with UseNewOptions + UseMaxNTaus
256 && ntaulep + ntauhad <= m_maxNtau
257 && ntaulep >= m_Nleptau
258 && ntauhad >= m_Nhadtau
259 && ntaulep <= m_maxNleptau
260 && ntauhad <= m_maxNhadtau
261 );
262
263 bool pass = false; //Initializing the pass variable that will be checked to se if the filter is passed.
264
265 if (m_NewOpt) pass = pass1;
266 if (m_useMaxNTaus) pass = pass3;
267
268 double extra_weight = 1;
270 if (pass2) {
271 pass = pass2;
272 extra_weight = 1/m_LooseRejectionFactor;
273 weight *= extra_weight;
274 }
275 else if (pass1) {
276 double rnd = rndm->flat(); // a random number between (0,1)
277 if(rnd > 1/m_LooseRejectionFactor) pass = false;
278 else pass = true;
279 }
280 else pass = false;
281 }
282
283 m_events[0]++;
284 m_events[3] += weight;
285 if(weight>=0) {
286 m_events[1]++;
287 m_events[4] += weight;
288 }
289 else {
290 m_events[2]++;
291 m_events[5] += weight;
292 }
293
294 if(pass) {
295 m_events_sel[0]++;
296 m_events_sel[3] += weight;
297 if(weight>=0) {
298 m_events_sel[1]++;
299 m_events_sel[4] += weight;
300 }
301 else {
302 m_events_sel[2]++;
303 m_events_sel[5] += weight;
304 }
305 }
306
308 // Get MC event collection for setting weight
309 const McEventCollection* mecc = 0;
310 if ( evtStore()->retrieve( mecc ).isFailure() || !mecc ){
311 setFilterPassed(false, ctx);
312 ATH_MSG_ERROR("Could not retrieve MC Event Collection - weight might not work");
313 return StatusCode::SUCCESS;
314 }
315
316 // Event passed. Will weight events
317 McEventCollection* mec = const_cast<McEventCollection*> (&(*mecc));
318 for (unsigned int i = 0; i < mec->size(); ++i) {
319 if (!(*mec)[i]) continue;
320 double existingWeight = (*mec)[i]->weights().size()>0 ? (*mec)[i]->weights()[0] : 1.;
321 if ((*mec)[i]->weights().size()>0) {
322 (*mec)[i]->weights()[0] = existingWeight*extra_weight;
323 } else {
324 (*mec)[i]->weights().push_back( existingWeight*extra_weight );
325 }
326 }
327 }
328
329 if(!m_NewOpt) pass = (ntau >= m_Ntau); //If UseNewOptions is not enabled, only look at total number of taus present.
330
331 if (m_ReverseFilter) pass = !pass; //If reverse filter is active, flip the truth value of pass
332
333 setFilterPassed(pass, ctx);
334
335 return StatusCode::SUCCESS;
336}
static TRandom * rnd
void setFilterPassed(bool state) const
size_type size() const noexcept
Returns the number of elements in the collection.
CLHEP::HepLorentzVector sumDaughterNeutrinos(const HepMC::ConstGenParticlePtr &tau)
Definition TauFilter.cxx:86
CLHEP::HepRandomEngine * getRandomEngine(const std::string &streamName, const EventContext &ctx) const
const std::string weights
HepMC3::ConstGenParticlePtr ConstGenParticlePtr
Definition GenParticle.h:20
HepMC3::GenEvent GenEvent
Definition GenEvent.h:39
static const int NU_MU
static const int NU_E
bool isTau(const T &p)
bool isPhysical(const T &p)
Identify if the particle is physical, i.e. is stable or decayed.

◆ filterFinalize()

StatusCode TauFilter::filterFinalize ( )
virtual

Reimplemented from GenFilter.

Definition at line 68 of file TauFilter.cxx.

68 {
69 if(m_NewOpt) {
70 ATH_MSG_INFO("Sum of Events total: " << m_events[0] << " , selected: " << m_events_sel[0]);
71 ATH_MSG_INFO("Sum of Events with pos. weights total: " << m_events[1] << " , selected: " << m_events_sel[1]);
72 ATH_MSG_INFO("Sum of Events with neg. weights total: " << m_events[2] << " , selected: " << m_events_sel[2]);
73 ATH_MSG_INFO("Sum of weights total: " << m_events[3] << " , selected: " << m_events_sel[3]);
74 ATH_MSG_INFO("Sum of pos. weights total: " << m_events[4] << " , selected: " << m_events_sel[4]);
75 ATH_MSG_INFO("Sum of neg. weights total: " << m_events[5] << " , selected: " << m_events_sel[5]);
76 }
77 else {
78 ATH_MSG_INFO(" , e: " << m_eventse << " , mu: " << m_eventsmu << " , had: " << m_eventshad <<
79 " , eacc: " << m_eventseacc << " , muacc: " << m_eventsmuacc << " , hadacc: " << m_eventshadacc);
80 }
81
82 return StatusCode::SUCCESS;
83}
#define ATH_MSG_INFO(x)

◆ filterInitialize()

StatusCode TauFilter::filterInitialize ( )
virtual

Reimplemented from GenFilter.

Definition at line 50 of file TauFilter.cxx.

50 {
51 m_eventse = 0;
52 m_eventsmu = 0;
53 m_eventshad = 0;
54 m_eventseacc = 0;
55 m_eventsmuacc = 0;
57
58 for(int i=0; i<6; i++) {
59 m_events[i] = 0; m_events_sel[i] = 0;
60 }
61
62 CHECK(m_rndmSvc.retrieve());
63
64 return StatusCode::SUCCESS;
65}
#define CHECK(...)
Evaluate an expression and check for errors.
ServiceHandle< IAthRNGSvc > m_rndmSvc
Definition TauFilter.h:35

◆ filterPassed() [1/2]

bool AthAlgorithm::filterPassed ( ) const
inherited

Definition at line 94 of file AthAlgorithm.cxx.

94 {
95 return filterPassed( Gaudi::Hive::currentContext() );
96}

◆ filterPassed() [2/2]

bool AthAlgorithm::filterPassed ( const EventContext & ctx) const
inherited

Definition at line 98 of file AthAlgorithm.cxx.

98 {
99 return execState( ctx ).filterPassed();
100}

◆ finalize()

StatusCode GenFilter::finalize ( )
inherited

Definition at line 50 of file GenFilter.cxx.

50 {
51 ATH_MSG_INFO("Events passed = " << m_nPass << " Events failed = " << m_nFail);
53 return StatusCode::SUCCESS;
54}
virtual StatusCode filterFinalize()
Definition GenFilter.h:47

◆ getContext()

const EventContext & AthAlgorithm::getContext ( ) const
inherited

Deprecated methods (use the ones with EventContext).

Definition at line 90 of file AthAlgorithm.cxx.

90 {
91 return Gaudi::Hive::currentContext();
92}

◆ getRandomEngine()

CLHEP::HepRandomEngine * TauFilter::getRandomEngine ( const std::string & streamName,
const EventContext & ctx ) const
private

Definition at line 339 of file TauFilter.cxx.

341{
342 ATHRNG::RNGWrapper* rngWrapper = m_rndmSvc->getEngine(this, streamName);
343 std::string rngName = name()+streamName;
344 rngWrapper->setSeed( rngName, ctx );
345 return rngWrapper->getEngine(ctx);
346}
void setSeed(const std::string &algName, const EventContext &ctx)
Set the random seed using a string (e.g.
Definition RNGWrapper.h:154
CLHEP::HepRandomEngine * getEngine(const EventContext &ctx) const
Retrieve the random engine corresponding to the provided EventContext.
Definition RNGWrapper.h:108

◆ initialize()

StatusCode GenFilter::initialize ( )
virtualinherited

Reimplemented from GenBase.

Definition at line 18 of file GenFilter.cxx.

18 {
20 m_nPass = 0;
21 m_nFail = 0;
23 return StatusCode::SUCCESS;
24}
virtual StatusCode initialize() override
Definition GenBase.cxx:17
virtual StatusCode filterInitialize()
Definition GenFilter.h:45

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::inputHandles ( ) const
overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ isReEntrant()

virtual bool AthAlgorithm::isReEntrant ( ) const
inlinefinaloverrideprotectedvirtualinherited

Legacy algorithms are not thread-safe.

Definition at line 111 of file AthAlgorithm.h.

111{ return false; }

◆ keepAll()

bool GenFilter::keepAll ( ) const
inlineinherited

Definition at line 55 of file GenFilter.h.

55{return m_keepAll; }

◆ msg()

MsgStream & AthCommonMsg< Gaudi::Algorithm >::msg ( ) const
inlineinherited

Definition at line 24 of file AthCommonMsg.h.

24 {
25 return this->msgStream();
26 }

◆ msgLvl()

bool AthCommonMsg< Gaudi::Algorithm >::msgLvl ( const MSG::Level lvl) const
inlineinherited

Definition at line 30 of file AthCommonMsg.h.

30 {
31 return this->msgLevel(lvl);
32 }

◆ nFailed()

int GenFilter::nFailed ( ) const
inlineinherited

Definition at line 53 of file GenFilter.h.

53{ return m_nFail; }

◆ nNeeded()

int GenFilter::nNeeded ( ) const
inlineinherited

Definition at line 54 of file GenFilter.h.

54{ return m_nNeeded; }

◆ nPassed()

int GenFilter::nPassed ( ) const
inlineinherited

Definition at line 52 of file GenFilter.h.

52{ return m_nPass; }

◆ outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::outputHandles ( ) const
overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

◆ particleData()

const HepPDT::ParticleData * GenBase::particleData ( int pid) const
inlineinherited

Access an element in the particle data table.

Definition at line 124 of file GenBase.h.

124 {
125 return pdt().particle(HepPDT::ParticleID(std::abs(pid)));
126 }
const HepPDT::ParticleDataTable & pdt() const
Shorter alias to get a particle data table.
Definition GenBase.h:121

◆ particleTable()

const HepPDT::ParticleDataTable & GenBase::particleTable ( ) const
inlineinherited

Get a particle data table.

Definition at line 116 of file GenBase.h.

116 {
117 return *(m_ppSvc->PDT());
118 }
ServiceHandle< IPartPropSvc > m_ppSvc
Handle on the particle property service.
Definition GenBase.h:142

◆ partPropSvc()

const ServiceHandle< IPartPropSvc > GenBase::partPropSvc ( ) const
inlineinherited

Access the particle property service.

Definition at line 111 of file GenBase.h.

111 {
112 return m_ppSvc;
113 }

◆ pdt()

const HepPDT::ParticleDataTable & GenBase::pdt ( ) const
inlineinherited

Shorter alias to get a particle data table.

Definition at line 121 of file GenBase.h.

121{ return particleTable(); }
const HepPDT::ParticleDataTable & particleTable() const
Get a particle data table.
Definition GenBase.h:116

◆ renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::renounce ( T & h)
inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

381 {
382 h.renounce();
384 }
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > renounce(T &h)

◆ renounceArray()

void AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::renounceArray ( SG::VarHandleKeyArray & handlesArray)
inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364 {
366 }

◆ setFilterPassed() [1/2]

void AthAlgorithm::setFilterPassed ( bool state) const
inherited

Definition at line 102 of file AthAlgorithm.cxx.

102 {
103 setFilterPassed( state, Gaudi::Hive::currentContext() );
104}

◆ setFilterPassed() [2/2]

void AthAlgorithm::setFilterPassed ( bool state,
const EventContext & ctx ) const
inherited

Definition at line 106 of file AthAlgorithm.cxx.

106 {
107 execState( ctx ).setFilterPassed(state);
108}

◆ sumDaughterNeutrinos()

CLHEP::HepLorentzVector TauFilter::sumDaughterNeutrinos ( const HepMC::ConstGenParticlePtr & tau)

Definition at line 86 of file TauFilter.cxx.

86 {
87 CLHEP::HepLorentzVector nu( 0, 0, 0, 0);
88 if ( MC::isSMNeutrino(part)
89 && MC::isPhysical(part) ) {
90 nu.setPx(part->momentum().px());
91 nu.setPy(part->momentum().py());
92 nu.setPz(part->momentum().pz());
93 nu.setE(part->momentum().e());
94 }
95
96 if (part->end_vertex() == 0) return nu;
97
98
99 for (const auto& beg: *(part->end_vertex())) nu += sumDaughterNeutrinos( beg );
100
101 return nu;
102}
bool isSMNeutrino(const T &p)

◆ sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )
overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

Loop through all output handles, and if they're WriteCondHandles, automatically register them and this Algorithm with the CondSvc.

Scan through all outputHandles, and if they're WriteCondHandles, register them with the CondSvc

Reimplemented from AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >.

Reimplemented in AthAnalysisAlgorithm, AthFilterAlgorithm, AthHistogramAlgorithm, and PyAthena::Alg.

Definition at line 66 of file AthAlgorithm.cxx.

66 {
68
69 if (sc.isFailure()) {
70 return sc;
71 }
72 ServiceHandle<ICondSvc> cs("CondSvc",name());
73 for (auto h : outputHandles()) {
74 if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
75 // do this inside the loop so we don't create the CondSvc until needed
76 if ( cs.retrieve().isFailure() ) {
77 ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
78 return StatusCode::SUCCESS;
79 }
80 if (cs->regHandle(this,*h).isFailure()) {
81 sc = StatusCode::FAILURE;
82 ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
83 << " with CondSvc");
84 }
85 }
86 }
87 return sc;
88}
virtual StatusCode sysInitialize() override
Override sysInitialize.
AthCommonDataStore(const std::string &name, T... args)
virtual std::vector< Gaudi::DataHandle * > outputHandles() const override

◆ sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::sysStart ( )
overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

◆ updateVHKA()

void AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::updateVHKA ( Gaudi::Details::PropertyBase & )
inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

308 {
309 // debug() << "updateVHKA for property " << p.name() << " " << p.toString()
310 // << " size: " << m_vhka.size() << endmsg;
311 for (auto &a : m_vhka) {
313 for (auto k : keys) {
314 k->setOwner(this);
315 }
316 }
317 }

Member Data Documentation

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_detStore
privateinherited

Pointer to StoreGate (detector store by default).

Definition at line 393 of file AthCommonDataStore.h.

◆ m_etaMaxe

double TauFilter::m_etaMaxe
private

Definition at line 38 of file TauFilter.h.

◆ m_etaMaxhad

double TauFilter::m_etaMaxhad
private

Definition at line 40 of file TauFilter.h.

◆ m_etaMaxlep

double TauFilter::m_etaMaxlep
private

Definition at line 50 of file TauFilter.h.

◆ m_etaMaxmu

double TauFilter::m_etaMaxmu
private

Definition at line 39 of file TauFilter.h.

◆ m_events

double TauFilter::m_events[6]
private

Definition at line 72 of file TauFilter.h.

◆ m_events_sel

double TauFilter::m_events_sel[6]
private

Definition at line 73 of file TauFilter.h.

◆ m_eventse

double TauFilter::m_eventse
private

Definition at line 75 of file TauFilter.h.

◆ m_eventseacc

double TauFilter::m_eventseacc
private

Definition at line 79 of file TauFilter.h.

◆ m_eventshad

double TauFilter::m_eventshad
private

Definition at line 77 of file TauFilter.h.

◆ m_eventshadacc

double TauFilter::m_eventshadacc
private

Definition at line 81 of file TauFilter.h.

◆ m_eventsmu

double TauFilter::m_eventsmu
private

Definition at line 76 of file TauFilter.h.

◆ m_eventsmuacc

double TauFilter::m_eventsmuacc
private

Definition at line 80 of file TauFilter.h.

◆ m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_evtStore
privateinherited

Pointer to StoreGate (event store by default).

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects
privateinherited

Definition at line 114 of file AthAlgorithm.h.

◆ m_filterEventNumber

int TauFilter::m_filterEventNumber
private

Definition at line 61 of file TauFilter.h.

◆ m_HasTightRegion

bool TauFilter::m_HasTightRegion
private

Definition at line 55 of file TauFilter.h.

◆ m_keepAll

bool GenFilter::m_keepAll
protectedinherited

Definition at line 66 of file GenFilter.h.

◆ m_LooseRejectionFactor

double TauFilter::m_LooseRejectionFactor
private

Definition at line 56 of file TauFilter.h.

◆ m_maxNhadtau

int TauFilter::m_maxNhadtau
private

Definition at line 66 of file TauFilter.h.

◆ m_maxNleptau

int TauFilter::m_maxNleptau
private

Definition at line 67 of file TauFilter.h.

◆ m_maxNtau

int TauFilter::m_maxNtau
private

Definition at line 65 of file TauFilter.h.

◆ m_mcEventKey

std::string GenBase::m_mcEventKey {}
protectedinherited

StoreGate key for the MC event collection (defaults to GEN_EVENT).

Definition at line 134 of file GenBase.h.

134{};

◆ m_mcevents_const

SG::ReadHandleKey<McEventCollection> GenBase::m_mcevents_const { this, "McEventKey", "GEN_EVENT", "StoreGate key of the MC event collection" }
privateinherited

Const handle to the MC event collection.

Definition at line 145 of file GenBase.h.

145{ this, "McEventKey", "GEN_EVENT", "StoreGate key of the MC event collection" };

◆ m_mkMcEvent

BooleanProperty GenBase::m_mkMcEvent {this, "MakeMcEvent", false, "Create a new MC event collection if it doesn't exist"}
protectedinherited

Flag to determine if a new MC event collection should be made if it doesn't exist.

Definition at line 136 of file GenBase.h.

136{this, "MakeMcEvent", false, "Create a new MC event collection if it doesn't exist"};

◆ m_NewOpt

bool TauFilter::m_NewOpt
private

Definition at line 47 of file TauFilter.h.

◆ m_nFail

int GenFilter::m_nFail
protectedinherited

Definition at line 64 of file GenFilter.h.

◆ m_Nhadtau

int TauFilter::m_Nhadtau
private

Definition at line 49 of file TauFilter.h.

◆ m_Nleptau

int TauFilter::m_Nleptau
private

Definition at line 48 of file TauFilter.h.

◆ m_nNeeded

int GenFilter::m_nNeeded
protectedinherited

Definition at line 65 of file GenFilter.h.

◆ m_nPass

int GenFilter::m_nPass
protectedinherited

Definition at line 63 of file GenFilter.h.

◆ m_Ntau

int TauFilter::m_Ntau
private

Definition at line 37 of file TauFilter.h.

◆ m_ppSvc

ServiceHandle<IPartPropSvc> GenBase::m_ppSvc {this, "PartPropSvc", "PartPropSvc"}
privateinherited

Handle on the particle property service.

Definition at line 142 of file GenBase.h.

142{this, "PartPropSvc", "PartPropSvc"};

◆ m_pTmine

double TauFilter::m_pTmine
private

Definition at line 42 of file TauFilter.h.

◆ m_pTminhad

double TauFilter::m_pTminhad
private

Definition at line 44 of file TauFilter.h.

◆ m_pTminhad_lead

double TauFilter::m_pTminhad_lead
private

Definition at line 53 of file TauFilter.h.

◆ m_pTminhad_tight

double TauFilter::m_pTminhad_tight
private

Definition at line 59 of file TauFilter.h.

◆ m_pTminhad_tight_lead

double TauFilter::m_pTminhad_tight_lead
private

Definition at line 60 of file TauFilter.h.

◆ m_pTminlep

double TauFilter::m_pTminlep
private

Definition at line 51 of file TauFilter.h.

◆ m_pTminlep_lead

double TauFilter::m_pTminlep_lead
private

Definition at line 52 of file TauFilter.h.

◆ m_pTminlep_tight

double TauFilter::m_pTminlep_tight
private

Definition at line 57 of file TauFilter.h.

◆ m_pTminlep_tight_lead

double TauFilter::m_pTminlep_tight_lead
private

Definition at line 58 of file TauFilter.h.

◆ m_pTminmu

double TauFilter::m_pTminmu
private

Definition at line 43 of file TauFilter.h.

◆ m_ReverseFilter

bool TauFilter::m_ReverseFilter
private

Definition at line 54 of file TauFilter.h.

◆ m_rndmSvc

ServiceHandle<IAthRNGSvc> TauFilter::m_rndmSvc {this, "RndmSvc", "AthRNGSvc"}
private

Definition at line 35 of file TauFilter.h.

35{this, "RndmSvc", "AthRNGSvc"};// Random number generator

◆ m_useMaxNTaus

bool TauFilter::m_useMaxNTaus
private

Definition at line 64 of file TauFilter.h.

◆ m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< Gaudi::Algorithm > >::m_vhka
privateinherited

Definition at line 398 of file AthCommonDataStore.h.


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