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

#include <VBFMjjIntervalFilter.h>

Inheritance diagram for VBFMjjIntervalFilter:
Collaboration diagram for VBFMjjIntervalFilter:

Public Member Functions

 VBFMjjIntervalFilter (const std::string &name, ISvcLocator *pSvcLocator)
virtual StatusCode filterInitialize ()
virtual StatusCode filterEvent ()
bool ApplyMassDphi (const xAOD::JetContainer *jets) const
double getEventWeight (const xAOD::JetContainer *jets) const
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 ()
Gen-specific event loop methods: to be overloaded!
virtual StatusCode filterFinalize ()
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
bool checkOverlap (double, double, const std::vector< HepMC::ConstGenParticlePtr > &) const
bool checkOverlap (double, double, const std::vector< TLorentzVector > &) const
TLorentzVector sumDaughterNeutrinos (const HepMC::ConstGenParticlePtr &) const
Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

double m_olapPt
double m_yMax
double m_pTavgMin
std::string m_TruthJetContainerName
ServiceHandle< IAthRNGSvcm_rndmSvc {this, "RndmSvc", "AthRNGSvc"}
double m_norm
double m_prob0
double m_prob1
double m_prob2low
double m_prob2high
double m_mjjlow
bool m_truncatelowmjj
double m_mjjhigh
bool m_truncatehighmjj
bool m_photonjetoverlap
bool m_electronjetoverlap
bool m_taujetoverlap
double m_alpha
bool m_ApplyNjet
unsigned int m_NJetsMin
unsigned int m_NJetsMax
bool m_ApplyWeighting
bool m_applyDphi
double m_dphijj
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 19 of file VBFMjjIntervalFilter.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

◆ VBFMjjIntervalFilter()

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

Definition at line 23 of file VBFMjjIntervalFilter.cxx.

24 : GenFilter(name, pSvcLocator),
25 m_norm(1.) //< @todo Scalefactor always set to 1.0! Remove?
26{
27 declareProperty("RapidityAcceptance", m_yMax = 5.0);
28 declareProperty("MinSecondJetPT", m_pTavgMin = 15.0*Gaudi::Units::GeV);
29 declareProperty("MinOverlapPT", m_olapPt = 15.0*Gaudi::Units::GeV);
30 declareProperty("TruthJetContainerName", m_TruthJetContainerName = "AntiKt4TruthJets");
31 //declareProperty("DoShape", m_doShape = true);
32 declareProperty("NoJetProbability", m_prob0 = 0.0002);
33 declareProperty("OneJetProbability", m_prob1 = 0.001);
34 declareProperty("LowMjjProbability", m_prob2low = 0.005);
35 declareProperty("HighMjjProbability", m_prob2high = 1.0);
36 declareProperty("LowMjj", m_mjjlow=100.0*Gaudi::Units::GeV);
37 declareProperty("TruncateAtLowMjj", m_truncatelowmjj = false);
38 declareProperty("HighMjj", m_mjjhigh=800.0*Gaudi::Units::GeV);
39 declareProperty("TruncateAtHighMjj", m_truncatehighmjj = false);
40 declareProperty("PhotonJetOverlapRemoval", m_photonjetoverlap = false);
41 declareProperty("ElectronJetOverlapRemoval", m_electronjetoverlap = true);
42 declareProperty("TauJetOverlapRemoval", m_taujetoverlap = false);
44 declareProperty("ApplyNjet",m_ApplyNjet=false);
45 declareProperty("Njets", m_NJetsMin=2);
46 declareProperty("NjetsMax",m_NJetsMax=-1);
47 declareProperty("ApplyWeighting",m_ApplyWeighting=true);
48 declareProperty("ApplyDphi",m_applyDphi=false);
49 declareProperty("dphijjMax",m_dphijj=2.5);
50
51}
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
GenFilter(const std::string &name, ISvcLocator *pSvcLocator)
Definition GenFilter.cxx:8

Member Function Documentation

◆ ApplyMassDphi()

bool VBFMjjIntervalFilter::ApplyMassDphi ( const xAOD::JetContainer * jets) const

Definition at line 252 of file VBFMjjIntervalFilter.cxx.

252 {
253 if(jets->size()<2) return false;
254 double mjj = (jets->at(0)->p4() + jets->at(1)->p4()).M();
255 double dphi = std::abs(jets->at(0)->p4().DeltaPhi(jets->at(1)->p4()));
256 ATH_MSG_INFO("mjj " << mjj << " dphi " << dphi);
257 bool pass = true;
258 if(mjj<m_mjjlow)pass=false;
259 if(mjj>m_mjjhigh)pass=false;
260 if(m_applyDphi && dphi > m_dphijj) pass = false;
261
262 return pass;
263}
#define ATH_MSG_INFO(x)

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

◆ checkOverlap() [1/2]

bool VBFMjjIntervalFilter::checkOverlap ( double eta,
double phi,
const std::vector< HepMC::ConstGenParticlePtr > & list ) const
private
Todo
Provide a helper function for this (and similar)

Definition at line 218 of file VBFMjjIntervalFilter.cxx.

218 {
219 for (size_t i = 0; i < list.size(); ++i) {
220 double pt = list.at(i)->momentum().perp();
221 if (pt > m_olapPt) {
223 double dphi = phi-list[i]->momentum().phi();
224 double deta = eta-list[i]->momentum().pseudoRapidity();
225 if (dphi > M_PI) { dphi -= 2.*M_PI; }
226 if (dphi < -M_PI) { dphi += 2.*M_PI; }
227 double dr = std::sqrt(deta*deta+dphi*dphi);
228 if (dr < 0.3) return true;
229 }
230 }
231 return false;
232}
#define M_PI
Scalar eta() const
pseudorapidity method
Scalar phi() const
phi method
list(name, path='/')
Definition histSizes.py:38

◆ checkOverlap() [2/2]

bool VBFMjjIntervalFilter::checkOverlap ( double eta,
double phi,
const std::vector< TLorentzVector > & list ) const
private
Todo
Provide a helper function for this (and similar)

Definition at line 236 of file VBFMjjIntervalFilter.cxx.

236 {
237 for (size_t i = 0; i < list.size(); ++i) {
238 double pt = list[i].Vect().Perp();
239 if (pt > m_olapPt) {
241 double dphi = phi-list[i].Phi();
242 double deta = eta-list[i].Vect().PseudoRapidity();
243 if (dphi > M_PI) { dphi -= 2.*M_PI; }
244 if (dphi < -M_PI) { dphi += 2.*M_PI; }
245 double dr = std::sqrt(deta*deta+dphi*dphi);
246 if (dr < 0.3) return true;
247 }
248 }
249 return false;
250}

◆ 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 VBFMjjIntervalFilter::filterEvent ( )
virtual

Implements GenFilter.

Definition at line 65 of file VBFMjjIntervalFilter.cxx.

65 {
66 // Get random number engine
67 const EventContext& ctx = Gaudi::Hive::currentContext();
68 CLHEP::HepRandomEngine* rndm = this->getRandomEngine(name(), ctx);
69 if (!rndm) {
70 ATH_MSG_ERROR("Failed to retrieve random number engine VBFMjjIntervalFilter");
71 setFilterPassed(false);
72 return StatusCode::SUCCESS;
73 }
74
75 // Retrieve jet container
76 const xAOD::JetContainer* truthJetCollection = 0;
78 evtStore()->retrieve( truthJetCollection, m_TruthJetContainerName).isFailure() ||
79 !truthJetCollection ) {
80 ATH_MSG_ERROR("No xAOD::JetContainer found in StoreGate with key " << m_TruthJetContainerName);
81 setFilterPassed(false);
82 return StatusCode::SUCCESS;
83 }
84
85 // Find overlap objects
86 std::vector<HepMC::ConstGenParticlePtr> MCTruthPhotonList;
87 std::vector<HepMC::ConstGenParticlePtr> MCTruthElectronList;
88 std::vector<TLorentzVector> MCTruthTauList;
89 for (McEventCollection::const_iterator itr = events()->begin(); itr != events()->end(); ++itr) {
90 const HepMC::GenEvent* genEvt = (*itr);
91 for (const auto& pitr: *genEvt) {
92 if (m_photonjetoverlap==true) {
93 // photon - copied from VBFForwardJetsFilter.cxx
94 if ( MC::isPhoton(pitr) && MC::isStable(pitr) &&
95 pitr->momentum().perp() >= m_olapPt &&
96 std::abs(pitr->momentum().pseudoRapidity()) <= m_yMax) {
97 MCTruthPhotonList.push_back(pitr);
98 }
99 }
100 if (m_electronjetoverlap==true) {
101 // electron
102 if (MC::isElectron(pitr) && MC::isStable(pitr) &&
103 pitr->momentum().perp() >= m_olapPt &&
104 std::abs(pitr->momentum().pseudoRapidity()) <= m_yMax) {
105 MCTruthElectronList.push_back(pitr);
106 }
107 }
108 if (m_taujetoverlap==true) {
109 // tau - copied from VBFForwardJetsFilter.cxx
110 if ( MC::isTau(pitr) && MC::isPhysical(pitr) ) {
111 auto tau = pitr;
112 int leptonic = 0;
113 for (const auto& beg: *(tau->end_vertex()) ) {
114 if ( beg->production_vertex() != tau->end_vertex() ) continue;
115 if ( std::abs( beg->pdg_id() ) == 12 ) leptonic = 1;
116 if ( std::abs( beg->pdg_id() ) == 14 ) leptonic = 2;
117 if ( std::abs( beg->pdg_id() ) == 15 ) leptonic = 11;
118 }
119
120 if (leptonic == 0) {
121 TLorentzVector nutau = sumDaughterNeutrinos( tau );
122 TLorentzVector tauvis = TLorentzVector(tau->momentum().px()-nutau.Px(),
123 tau->momentum().py()-nutau.Py(),
124 tau->momentum().pz()-nutau.Pz(),
125 tau->momentum().e()-nutau.E());
126 if (tauvis.Vect().Perp() >= m_olapPt && std::abs(tauvis.Vect().PseudoRapidity()) <= m_yMax) {
127 MCTruthTauList.push_back(tauvis);
128 }
129 }
130 }
131 }
132 }
133 }
134
135 // Filter based on rapidity acceptance and sort
136 ConstDataVector<xAOD::JetContainer> filteredJets(SG::VIEW_ELEMENTS);
137 for (xAOD::JetContainer::const_iterator jitr = truthJetCollection->begin(); jitr != truthJetCollection->end(); ++jitr) {
138 if (std::abs( (*jitr)->rapidity() ) < m_yMax && (*jitr)->pt() >= m_olapPt) {
139 bool JetOverlapsWithPhoton = false;
140 bool JetOverlapsWithElectron = false;
141 bool JetOverlapsWithTau = false;
142
143 if (m_photonjetoverlap==true) {
144 JetOverlapsWithPhoton = checkOverlap((*jitr)->rapidity(), (*jitr)->phi(), MCTruthPhotonList);
145 }
146 if (m_electronjetoverlap==true) {
147 JetOverlapsWithElectron = checkOverlap((*jitr)->rapidity(), (*jitr)->phi(), MCTruthElectronList);
148 }
149 if (m_taujetoverlap==true) {
150 JetOverlapsWithTau = checkOverlap((*jitr)->rapidity(), (*jitr)->phi(), MCTruthTauList);
151 }
152
153 if (!JetOverlapsWithPhoton && !JetOverlapsWithElectron && !JetOverlapsWithTau ) {
154 filteredJets.push_back(*jitr);
155 }
156 }
157 }
158 filteredJets.sort(High2LowByJetClassPt());
159
161
162 double eventWeight = 1.0;
163 eventWeight = getEventWeight(filteredJets.asDataVector());
164 double rnd = rndm->flat();
165 if (1.0/eventWeight < rnd) {
166 setFilterPassed(false);
167 ATH_MSG_DEBUG("Event failed weighting. Weight is " << eventWeight);
168 return StatusCode::SUCCESS;
169 }
170
171 // Get MC event collection for setting weight
172 const McEventCollection* mecc = 0;
173 if ( evtStore()->retrieve( mecc ).isFailure() || !mecc ){
174 setFilterPassed(false);
175 ATH_MSG_ERROR("Could not retrieve MC Event Collection - weight might not work");
176 return StatusCode::SUCCESS;
177 }
178
179 ATH_MSG_INFO("Event passed. Will weight events " << eventWeight*m_norm);
180 McEventCollection* mec = const_cast<McEventCollection*> (&(*mecc));
181 for (unsigned int i = 0; i < mec->size(); ++i) {
182 if (!(*mec)[i]) continue;
183 double existingWeight = (*mec)[i]->weights().size()>0 ? (*mec)[i]->weights()[0] : 1.;
184 if ((*mec)[i]->weights().size()>0) {
185 (*mec)[i]->weights()[0] = existingWeight*eventWeight*m_norm;
186 } else {
187 (*mec)[i]->weights().push_back( eventWeight*m_norm*existingWeight );
188 }
189 }
190 }//Apply weighting
191 else {
192 //just compute mjj, dphi etc
193 bool pass = ApplyMassDphi(filteredJets.asDataVector());
194 if(!pass){
195 setFilterPassed(false);
196 ATH_MSG_DEBUG("Event failed filter");
197 return StatusCode::SUCCESS;
198 }
199 if(m_ApplyNjet){
200 if(filteredJets.size()<m_NJetsMin){
201 setFilterPassed(false);
202 return StatusCode::SUCCESS;
203 }
204 if(m_NJetsMax>0){
205 if(filteredJets.size()>m_NJetsMax){
206 setFilterPassed(false);
207 return StatusCode::SUCCESS;
208 }
209 }//Njets <
210 }//Apply Njets filter
211 }
212 // Made it to the end - success!
213 setFilterPassed(true);
214 return StatusCode::SUCCESS;
215}
static TRandom * rnd
DataModel_detail::const_iterator< DataVector > const_iterator
Definition DataVector.h:838
const_iterator end() const noexcept
Return a const_iterator pointing past the end of the collection.
size_type size() const noexcept
Returns the number of elements in the collection.
bool ApplyMassDphi(const xAOD::JetContainer *jets) const
double getEventWeight(const xAOD::JetContainer *jets) const
CLHEP::HepRandomEngine * getRandomEngine(const std::string &streamName, const EventContext &ctx) const
bool checkOverlap(double, double, const std::vector< HepMC::ConstGenParticlePtr > &) const
TLorentzVector sumDaughterNeutrinos(const HepMC::ConstGenParticlePtr &) const
bool contains(const std::string &s, const std::string &regx)
does a string contain the substring
Definition hcg.cxx:114
bool isPhoton(const T &p)
bool isElectron(const T &p)
bool isStable(const T &p)
Identify if the particle is stable, i.e. has not decayed.
bool isTau(const T &p)
bool isPhysical(const T &p)
Identify if the particle is physical, i.e. is stable or decayed.
@ VIEW_ELEMENTS
this data object is a view, it does not own its elmts
retrieve(aClass, aKey=None)
Definition PyKernel.py:110
JetContainer_v1 JetContainer
Definition of the current "jet container version".

◆ filterFinalize()

◆ filterInitialize()

StatusCode VBFMjjIntervalFilter::filterInitialize ( )
virtual

Reimplemented from GenFilter.

Definition at line 54 of file VBFMjjIntervalFilter.cxx.

54 {
55 ATH_MSG_INFO("Configured for jets in " << m_TruthJetContainerName << " inside |y|<" << m_yMax);
56
57 CHECK(m_rndmSvc.retrieve());
58
60 ATH_MSG_INFO("m_alpha set to" << m_alpha);
61 return StatusCode::SUCCESS;
62}
#define CHECK(...)
Evaluate an expression and check for errors.
ServiceHandle< IAthRNGSvc > m_rndmSvc

◆ 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

◆ getEventWeight()

double VBFMjjIntervalFilter::getEventWeight ( const xAOD::JetContainer * jets) const

Definition at line 265 of file VBFMjjIntervalFilter.cxx.

265 {
266 double weight = 1.0;
267 if (jets->size() == 0) {
268 weight /= m_prob0;
269 ATH_MSG_DEBUG("Event in 0-jet weighting. Weight is " << weight);
270 } else if (jets->size() == 1) {
271 weight /= m_prob1;
272 ATH_MSG_DEBUG("Event in 1-jet weighting. Weight is " << weight);
273 } else {
274 double mjj = (jets->at(0)->p4() + jets->at(1)->p4()).M();
275 if (mjj < m_mjjlow) {
276 if (m_truncatelowmjj==false) {
278 }
279 else {
280 weight = -1.0;
281 }
282 } else if (mjj > m_mjjhigh) {
283 if (m_truncatehighmjj==false) {
285 }
286 else {
287 weight = -1.0;
288 }
289 } else {
290 weight = weight * std::pow(m_mjjlow/mjj, m_alpha) / m_prob2low;
291 ATH_MSG_DEBUG("WEIGHTING:: " << mjj << "\t" << weight);
292 }
293 }
294 return weight;
295}

◆ getRandomEngine()

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

Definition at line 316 of file VBFMjjIntervalFilter.cxx.

318{
319 ATHRNG::RNGWrapper* rngWrapper = m_rndmSvc->getEngine(this, streamName);
320 std::string rngName = name()+streamName;
321 rngWrapper->setSeed( rngName, ctx );
322 return rngWrapper->getEngine(ctx);
323}
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()

TLorentzVector VBFMjjIntervalFilter::sumDaughterNeutrinos ( const HepMC::ConstGenParticlePtr & part) const
private

Definition at line 298 of file VBFMjjIntervalFilter.cxx.

298 {
299 TLorentzVector nu( 0, 0, 0, 0);
300
301 if ( ( std::abs( part->pdg_id() ) == 12 ) || ( std::abs( part->pdg_id() ) == 14 ) || ( std::abs( part->pdg_id() ) == 16 ) ) {
302 nu.SetPx(part->momentum().px());
303 nu.SetPy(part->momentum().py());
304 nu.SetPz(part->momentum().pz());
305 nu.SetE(part->momentum().e());
306 return nu;
307 }
308
309 if ( !part->end_vertex() ) return nu;
310
311 for (const auto& beg: *(part->end_vertex()) ) nu += sumDaughterNeutrinos( beg );
312 return nu;
313}

◆ 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_alpha

double VBFMjjIntervalFilter::m_alpha
private

Definition at line 56 of file VBFMjjIntervalFilter.h.

◆ m_applyDphi

bool VBFMjjIntervalFilter::m_applyDphi
private

Definition at line 61 of file VBFMjjIntervalFilter.h.

◆ m_ApplyNjet

bool VBFMjjIntervalFilter::m_ApplyNjet
private

Definition at line 57 of file VBFMjjIntervalFilter.h.

◆ m_ApplyWeighting

bool VBFMjjIntervalFilter::m_ApplyWeighting
private

Definition at line 60 of file VBFMjjIntervalFilter.h.

◆ 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_dphijj

double VBFMjjIntervalFilter::m_dphijj
private

Definition at line 62 of file VBFMjjIntervalFilter.h.

◆ m_electronjetoverlap

bool VBFMjjIntervalFilter::m_electronjetoverlap
private

Definition at line 54 of file VBFMjjIntervalFilter.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_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_mjjhigh

double VBFMjjIntervalFilter::m_mjjhigh
private

Definition at line 51 of file VBFMjjIntervalFilter.h.

◆ m_mjjlow

double VBFMjjIntervalFilter::m_mjjlow
private

Definition at line 49 of file VBFMjjIntervalFilter.h.

◆ 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_nFail

int GenFilter::m_nFail
protectedinherited

Definition at line 66 of file GenFilter.h.

◆ m_NJetsMax

unsigned int VBFMjjIntervalFilter::m_NJetsMax
private

Definition at line 59 of file VBFMjjIntervalFilter.h.

◆ m_NJetsMin

unsigned int VBFMjjIntervalFilter::m_NJetsMin
private

Definition at line 58 of file VBFMjjIntervalFilter.h.

◆ m_nNeeded

int GenFilter::m_nNeeded
protectedinherited

Definition at line 67 of file GenFilter.h.

◆ m_norm

double VBFMjjIntervalFilter::m_norm
private

Definition at line 42 of file VBFMjjIntervalFilter.h.

◆ m_nPass

int GenFilter::m_nPass
protectedinherited

Definition at line 65 of file GenFilter.h.

◆ m_olapPt

double VBFMjjIntervalFilter::m_olapPt
private

Definition at line 31 of file VBFMjjIntervalFilter.h.

◆ m_photonjetoverlap

bool VBFMjjIntervalFilter::m_photonjetoverlap
private

Definition at line 53 of file VBFMjjIntervalFilter.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_prob0

double VBFMjjIntervalFilter::m_prob0
private

Definition at line 45 of file VBFMjjIntervalFilter.h.

◆ m_prob1

double VBFMjjIntervalFilter::m_prob1
private

Definition at line 46 of file VBFMjjIntervalFilter.h.

◆ m_prob2high

double VBFMjjIntervalFilter::m_prob2high
private

Definition at line 48 of file VBFMjjIntervalFilter.h.

◆ m_prob2low

double VBFMjjIntervalFilter::m_prob2low
private

Definition at line 47 of file VBFMjjIntervalFilter.h.

◆ m_pTavgMin

double VBFMjjIntervalFilter::m_pTavgMin
private

Definition at line 33 of file VBFMjjIntervalFilter.h.

◆ m_rndmSvc

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

Definition at line 36 of file VBFMjjIntervalFilter.h.

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

◆ m_taujetoverlap

bool VBFMjjIntervalFilter::m_taujetoverlap
private

Definition at line 55 of file VBFMjjIntervalFilter.h.

◆ m_truncatehighmjj

bool VBFMjjIntervalFilter::m_truncatehighmjj
private

Definition at line 52 of file VBFMjjIntervalFilter.h.

◆ m_truncatelowmjj

bool VBFMjjIntervalFilter::m_truncatelowmjj
private

Definition at line 50 of file VBFMjjIntervalFilter.h.

◆ m_TruthJetContainerName

std::string VBFMjjIntervalFilter::m_TruthJetContainerName
private

Definition at line 34 of file VBFMjjIntervalFilter.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.

◆ m_yMax

double VBFMjjIntervalFilter::m_yMax
private

Definition at line 32 of file VBFMjjIntervalFilter.h.


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