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 ()
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.
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 ()
StatusCode finalize ()
Counters
int nPassed () const
int nFailed () const
int nNeeded () 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::GenEvent * event_const () const
 Access the current signal event (const)
const McEventCollectionevents_const () const
 Access the current event's McEventCollection (const)
const McEventCollectionevents_const (const EventContext &ctx) 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.

Protected Member Functions

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
Properties
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.

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

Utility event-mangling functions

Todo
Replace with HepMC units when available
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.
void GeVToMeV (HepMC::GenEvent *evt)
 Scale event energies/momenta by x 1000.
void MeVToGeV (HepMC::GenEvent *evt)
 Scale event energies/momenta by x 1/1000.
void cmTomm (HepMC::GenEvent *evt)
 Scale event lengths by x 10.
void mmTocm (HepMC::GenEvent *evt)
 Scale event lengths by x 1/10.

Detailed Description

Definition at line 20 of file TauFilter.h.

Member Typedef Documentation

◆ StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< 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]

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

◆ ATLAS_NOT_CONST_THREAD_SAFE() [2/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.

◆ cmTomm()

void GenBase::cmTomm ( HepMC::GenEvent * evt)
protectedinherited

Scale event lengths by x 10.

Definition at line 78 of file GenBase.cxx.

78 {
79 for (HepMC::GenEvent::vertex_iterator vtx = evt->vertices_begin(); vtx != evt->vertices_end(); ++vtx) {
80 const HepMC::FourVector fv((*vtx)->position().x() * 10,
81 (*vtx)->position().y() * 10,
82 (*vtx)->position().z() * 10,
83 (*vtx)->position().t() * 10);
84 (*vtx)->set_position(fv);
85 }
86}

◆ declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< 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< 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< 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
inlineinherited

Access the current signal event (const)

Definition at line 83 of file GenBase.h.

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

◆ events_const() [1/2]

const McEventCollection * GenBase::events_const ( ) const
inlineinherited

Access the current event's McEventCollection (const)

Definition at line 96 of file GenBase.h.

96 {
97 return events_const( getContext() );
98 }

◆ events_const() [2/2]

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

Definition at line 99 of file GenBase.h.

99 {
100 SG::ReadHandle<McEventCollection> ret = SG::makeHandle(m_mcevents_const, ctx);
101 if (!ret.isValid())
102 ATH_MSG_ERROR("No McEventCollection found in StoreGate with key " << m_mcevents_const.key());
103 return ret.cptr();
104 }
SG::ReadHandleKey< McEventCollection > m_mcevents_const
Const handle to the MC event collection.
Definition GenBase.h:163
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< 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 ( )
virtualinherited
Todo
Probably the filter should only look at the first event... right?

Reimplemented from GenBase.

Definition at line 29 of file GenFilter.cxx.

29 {
30 if (events_const()->empty()) {
31 ATH_MSG_ERROR("No events found in McEventCollection");
32 return StatusCode::FAILURE;
33 } else if (events_const()->size() > 1) {
35 ATH_MSG_WARNING("More than one event in current McEventCollection -- which is valid?");
36 }
38#ifdef HEPMC3
39 if (filterPassed() || m_keepAll ) {
40#else
41 if (filterPassed() ) {
42#endif
43 ATH_MSG_DEBUG("Event passed filter");
44 m_nPass += 1;
45 } else {
46 ATH_MSG_DEBUG("Event failed filter");
47 m_nFail += 1;
48 }
49 // Bail out once we have enough events
50 if (m_nPass >= m_nNeeded && m_nNeeded > 0)
51 sc = StatusCode::FAILURE;
52 return sc;
53}
#define ATH_MSG_WARNING(x)
#define ATH_MSG_DEBUG(x)
static Double_t sc
int m_nPass
Definition GenFilter.h:65
int m_nNeeded
Definition GenFilter.h:67
int m_nFail
Definition GenFilter.h:66
virtual StatusCode filterEvent()=0
::StatusCode StatusCode
StatusCode definition for legacy code.

◆ extraDeps_update_handler()

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

◆ 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

◆ finalize()

StatusCode GenFilter::finalize ( )
inherited

Definition at line 56 of file GenFilter.cxx.

56 {
57 ATH_MSG_INFO("Events passed = " << m_nPass << " Events failed = " << m_nFail);
59 return StatusCode::SUCCESS;
60}
virtual StatusCode filterFinalize()
Definition GenFilter.h:47

◆ getRandomEngine()

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

Definition at line 340 of file TauFilter.cxx.

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

◆ GeVToMeV()

void GenBase::GeVToMeV ( HepMC::GenEvent * evt)
protectedinherited

Scale event energies/momenta by x 1000.

Todo
Add HepMC units awareness and do it differently when HepMC provides this functionality directly (and reference-based FourVector accessors)

Definition at line 58 of file GenBase.cxx.

58 {
59 for (HepMC::GenEvent::particle_iterator p = evt->particles_begin(); p != evt->particles_end(); ++p) {
60 const HepMC::FourVector fv((*p)->momentum().px() * 1000,
61 (*p)->momentum().py() * 1000,
62 (*p)->momentum().pz() * 1000,
63 (*p)->momentum().e() * 1000);
64 (*p)->set_momentum(fv);
65 (*p)->set_generated_mass(1000 * (*p)->generated_mass());
66 }
67}

◆ initialize()

StatusCode GenFilter::initialize ( )
virtualinherited

Reimplemented from GenBase.

Definition at line 20 of file GenFilter.cxx.

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

◆ inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< 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.

◆ MeVToGeV()

void GenBase::MeVToGeV ( HepMC::GenEvent * evt)
protectedinherited

Scale event energies/momenta by x 1/1000.

Definition at line 68 of file GenBase.cxx.

68 {
69 for (HepMC::GenEvent::particle_iterator p = evt->particles_begin(); p != evt->particles_end(); ++p) {
70 const HepMC::FourVector fv((*p)->momentum().px() / 1000,
71 (*p)->momentum().py() / 1000,
72 (*p)->momentum().pz() / 1000,
73 (*p)->momentum().e() / 1000);
74 (*p)->set_momentum(fv);
75 (*p)->set_generated_mass((*p)->generated_mass() / 1000);
76 }
77}

◆ mmTocm()

void GenBase::mmTocm ( HepMC::GenEvent * evt)
protectedinherited

Scale event lengths by x 1/10.

Definition at line 87 of file GenBase.cxx.

87 {
88 for (HepMC::GenEvent::vertex_iterator vtx = evt->vertices_begin(); vtx != evt->vertices_end(); ++vtx) {
89 const HepMC::FourVector fv((*vtx)->position().x() / 10,
90 (*vtx)->position().y() / 10,
91 (*vtx)->position().z() / 10,
92 (*vtx)->position().t() / 10);
93 (*vtx)->set_position(fv);
94 }
95}

◆ msg()

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

Definition at line 24 of file AthCommonMsg.h.

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

◆ msgLvl()

bool AthCommonMsg< 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< 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 126 of file GenBase.h.

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

◆ particleTable()

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

Get a particle data table.

Definition at line 118 of file GenBase.h.

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

◆ partPropSvc()

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

Access the particle property service.

Definition at line 113 of file GenBase.h.

113 {
114 return m_ppSvc;
115 }

◆ pdt()

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

Shorter alias to get a particle data table.

Definition at line 123 of file GenBase.h.

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

◆ 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< 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< Algorithm > >::renounceArray ( SG::VarHandleKeyArray & handlesArray)
inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

364 {
366 }

◆ 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 ( ( (std::abs( part->pdg_id() ) == 12 ) || ( std::abs( part->pdg_id() ) == 14 ) || ( std::abs( part->pdg_id() ) == 16 ))
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}

◆ 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< 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< 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< 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 }
std::vector< SG::VarHandleKeyArray * > m_vhka

Member Data Documentation

◆ m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< 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< 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 79 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_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 137 of file GenBase.h.

137{};

◆ 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 163 of file GenBase.h.

163{ 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 139 of file GenBase.h.

139{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 66 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 67 of file GenFilter.h.

◆ m_nPass

int GenFilter::m_nPass
protectedinherited

Definition at line 65 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 160 of file GenBase.h.

160{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< Algorithm > >::m_varHandleArraysDeclared
privateinherited

Definition at line 399 of file AthCommonDataStore.h.

◆ m_vhka

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

Definition at line 398 of file AthCommonDataStore.h.


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