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

#include <EvtInclusiveDecay.h>

Inheritance diagram for EvtInclusiveDecay:
Collaboration diagram for EvtInclusiveDecay:

Classes

struct  ParticleIdCompare

Public Member Functions

 EvtInclusiveDecay (const std::string &name, ISvcLocator *pSvcLocator)
virtual ~EvtInclusiveDecay ()
StatusCode initialize ()
StatusCode execute (const EventContext &ctx)
 Execute method.
StatusCode finalize ()
std::string xmlpath (void)
virtual StatusCode sysInitialize () override
 Override sysInitialize.
virtual bool isClonable () const override
 Specify if the algorithm is clonable.
virtual StatusCode sysExecute (const EventContext &ctx) override
 Execute an algorithm.
virtual const DataObjIDColl & extraOutputDeps () const override
 Return the list of extra output dependencies.
virtual bool filterPassed (const EventContext &ctx) const
 Get filter decision:
virtual void setFilterPassed (bool state, const EventContext &ctx) const
 Set filter decision:
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 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).

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.

Private Types

typedef ServiceHandle< StoreGateSvcStoreGateSvc_t

Private Member Functions

Gaudi::Details::PropertyBase & declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
 specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes

DataObjIDColl m_extendedExtraObjects
 Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.
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

Features for derived classes to use internally

ServiceHandle< IAthRNGSvcm_rndmSvc {this, "RndmSvc", "AthRNGSvc"}
IntegerProperty m_dsid {this, "Dsid", 999999}
IntegerProperty m_randomSeed {this, "RandomSeed", 1234567, "Random seed for the built-in random engine"}
 Seed for random number engine.
McEventCollectionm_mcEvtColl {}
EvtInclusiveAtRndmGenm_evtAtRndmGen {}
EvtGen * m_myEvtGen {}
std::string m_pdtFile
std::string m_decayFile
std::string m_userDecayFile
std::string m_randomStreamName
std::string m_inputKeyName
std::string m_outputKeyName
bool m_readExisting
bool m_prohibitFinalStateDecay
bool m_prohibitReDecay
bool m_prohibitUnDecay
bool m_prohibitRemoveSelfDecay
std::vector< int > m_blackList
std::set< int > m_blackListSet
bool m_allowAllKnownDecays
bool m_allowDefaultBDecays
std::vector< int > m_whiteList
std::set< int > m_whiteListSet
bool m_printHepMCBeforeEvtGen
bool m_printHepMCAfterEvtGen
bool m_printHepMCHighlighted
bool m_printHepMCHighLightTopLevelDecays
bool m_checkDecayTree
bool m_checkDecayChannels
std::map< int, long > m_noDecayChannels
int m_nRepeatedDecays
int m_maxNRepeatedDecays
bool m_applyUserSelection
bool m_userSelRequireOppositeSignedMu
double m_userSelMu1MinPt
double m_userSelMu2MinPt
double m_userSelMu1MaxEta
double m_userSelMu2MaxEta
double m_userSelMinDimuMass
double m_userSelMaxDimuMass
bool m_isfHerwig
bool m_setVMtransversePol
void reseedRandomEngine (const std::string &streamName, const EventContext &ctx)
CLHEP::HepRandomEngine * getRandomEngine (const std::string &streamName, unsigned long int randomSeedOffset, const EventContext &ctx) const
CLHEP::HepRandomEngine * getRandomEngineDuringInitialize (const std::string &streamName, unsigned long int randomSeedOffset, unsigned int conditionsRun=1, unsigned int lbn=1) const
StatusCode traverseDecayTree (HepMC::GenParticlePtr p, bool isToBeRemoved, std::set< HepMC::GenVertexPtr > &visited, std::set< HepMC::GenParticlePtr, ParticleIdCompare > &toBeDecayed)
void removeDecayTree (HepMC::GenEvent *hepMC, HepMC::GenParticlePtr p)
void decayParticle (HepMC::GenEvent *hepMC, HepMC::GenParticlePtr p)
void addEvtGenDecayTree (HepMC::GenEvent *hepMC, HepMC::GenParticlePtr part, EvtParticle *evtPart, EvtVector4R treeStart, double momentumScaleFactor=1.0)
bool isToBeDecayed (HepMC::ConstGenParticlePtr p, bool doCrossChecks)
bool isDefaultB (const int pId) const
bool passesUserSelection (HepMC::GenEvent *hepMC)
double invMass (HepMC::ConstGenParticlePtr p1, HepMC::ConstGenParticlePtr p2)
void printHepMC (HepMC::GenEvent *hepMC, std::set< HepMC::GenParticlePtr, ParticleIdCompare > *particleSet=nullptr)
unsigned int printTree (HepMC::GenParticlePtr p, std::set< HepMC::GenVertexPtr > &visited, int level, std::set< HepMC::GenParticlePtr, ParticleIdCompare > *particleSet=nullptr)
std::string pdgName (HepMC::ConstGenParticlePtr p, bool statusHighlighting=false, std::set< HepMC::GenParticlePtr, ParticleIdCompare > *particleSet=nullptr)

Properties

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 54 of file EvtInclusiveDecay.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

◆ EvtInclusiveDecay()

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

Definition at line 54 of file EvtInclusiveDecay.cxx.

54 :
55 GenBase( name, pSvcLocator ),
57
58 // Basic EvtGen configuration: decay and particle definition files, random number stream
59 declareProperty("pdtFile", m_pdtFile = "inclusive.pdt");
60 declareProperty("decayFile", m_decayFile = "2014inclusive.dec");
61 declareProperty("userDecayFile", m_userDecayFile = "");
62 declareProperty("randomStreamName", m_randomStreamName = "EVTGEN");
63 declareProperty("inputKeyName", m_inputKeyName = "GEN_EVENT");
64 declareProperty("outputKeyName",m_outputKeyName = "GEN_EVENT_EVTGEN");
65 declareProperty("readExisting",m_readExisting=false);
66
67 // Selection of particles to be decayed
68 declareProperty("prohibitFinalStateDecay", m_prohibitFinalStateDecay=false);
69 declareProperty("prohibitReDecay", m_prohibitReDecay=false);
70 declareProperty("prohibitUnDecay", m_prohibitUnDecay=true);
71 declareProperty("prohibitRemoveSelfDecay", m_prohibitRemoveSelfDecay=false);
72 declareProperty("blackList",m_blackList);
73 declareProperty("allowAllKnownDecays", m_allowAllKnownDecays=true);
74 declareProperty("allowDefaultBDecays", m_allowDefaultBDecays=true);
75 declareProperty("whiteList",m_whiteList);
76
77 // Level of output
78 declareProperty("printHepMCBeforeEvtGen", m_printHepMCBeforeEvtGen=false);
79 declareProperty("printHepMCAfterEvtGen", m_printHepMCAfterEvtGen=false);
80 declareProperty("printHepMCHighlighted", m_printHepMCHighlighted=true);
81 declareProperty("printHepMCHighLightTopLevelDecays", m_printHepMCHighLightTopLevelDecays=true);
82
83 // Optional checks
84 declareProperty("checkDecayTree", m_checkDecayTree=false);
85 declareProperty("checkDecayChannels", m_checkDecayChannels=false);
86
87 // Repeated decays
88 declareProperty("maxNRepeatedDecays", m_maxNRepeatedDecays=1);
89
90 // User selection
91 declareProperty("applyUserSelection", m_applyUserSelection=false);
92 declareProperty("userSelRequireOppositeSignedMu", m_userSelRequireOppositeSignedMu=true);
93 declareProperty("userSelMu1MinPt", m_userSelMu1MinPt=0.);
94 declareProperty("userSelMu2MinPt", m_userSelMu2MinPt=0.);
95 declareProperty("userSelMu1MaxEta", m_userSelMu1MaxEta=102.5);
96 declareProperty("userSelMu2MaxEta", m_userSelMu2MaxEta=102.5);
97 declareProperty("userSelMinDimuMass", m_userSelMinDimuMass=0.);
98 declareProperty("userSelMaxDimuMass", m_userSelMaxDimuMass=-1.); // set to negative to not apply cut
99 declareProperty("isfHerwig", m_isfHerwig=false);
100 declareProperty("setVMtransversePol", m_setVMtransversePol=false);
101
102 // We have decided to blacklist Tau decays because we are not sure whether the polarization
103 // would be properly passed to EvtGen
104 m_blackList.push_back(15);
105}
Gaudi::Details::PropertyBase & declareProperty(Gaudi::Property< T, V, H > &t)
std::string m_randomStreamName
std::string m_userDecayFile
std::vector< int > m_blackList
std::string m_outputKeyName
std::vector< int > m_whiteList
GenBase(const std::string &name, ISvcLocator *pSvcLocator)
Constructor.
Definition GenBase.cxx:11

◆ ~EvtInclusiveDecay()

EvtInclusiveDecay::~EvtInclusiveDecay ( )
virtual

Definition at line 109 of file EvtInclusiveDecay.cxx.

109 {
110 delete m_myEvtGen;
111 delete m_evtAtRndmGen;
112}
EvtInclusiveAtRndmGen * m_evtAtRndmGen

Member Function Documentation

◆ addEvtGenDecayTree()

void EvtInclusiveDecay::addEvtGenDecayTree ( HepMC::GenEvent * hepMC,
HepMC::GenParticlePtr part,
EvtParticle * evtPart,
EvtVector4R treeStart,
double momentumScaleFactor = 1.0 )
private

Definition at line 448 of file EvtInclusiveDecay.cxx.

449 {
450 if(evtPart->getNDaug()!=0) {
451 // Add decay vertex, starting from production vertex of particle
452 double ct=(evtPart->getDaug(0)->get4Pos()).get(0)+treeStart.get(0);
453 double x=(evtPart->getDaug(0)->get4Pos()).get(1)+treeStart.get(1);
454 double y=(evtPart->getDaug(0)->get4Pos()).get(2)+treeStart.get(2);
455 double z=(evtPart->getDaug(0)->get4Pos()).get(3)+treeStart.get(3);
456
458
459 hepMC->add_vertex(end_vtx);
460 end_vtx->add_particle_in(std::move(part));
461
462 // Add decay daughter with their own decay trees
463 for(uint it=0; it<evtPart->getNDaug(); it++) {
464 double e=(evtPart->getDaug(it)->getP4Lab()).get(0) * momentumScaleFactor;
465 double px=(evtPart->getDaug(it)->getP4Lab()).get(1) * momentumScaleFactor;
466 double py=(evtPart->getDaug(it)->getP4Lab()).get(2) * momentumScaleFactor;
467 double pz=(evtPart->getDaug(it)->getP4Lab()).get(3) * momentumScaleFactor;
468 int id=EvtPDL::getStdHep(evtPart->getDaug(it)->getId());
469 int status=1;
470 if(evtPart->getDaug(it)->getNDaug() != 0) status=2;
471 HepMC::GenParticlePtr daughter = HepMC::newGenParticlePtr(HepMC::FourVector(px,py,pz,e),id,status);
472 end_vtx->add_particle_out(daughter);
473 addEvtGenDecayTree(hepMC, std::move(daughter), evtPart->getDaug(it), treeStart, momentumScaleFactor);
474 }
475 }
476}
unsigned int uint
#define y
#define x
#define z
void addEvtGenDecayTree(HepMC::GenEvent *hepMC, HepMC::GenParticlePtr part, EvtParticle *evtPart, EvtVector4R treeStart, double momentumScaleFactor=1.0)
HepMC3::FourVector FourVector
GenParticlePtr newGenParticlePtr(const HepMC3::FourVector &mom=HepMC3::FourVector::ZERO_VECTOR(), int pid=0, int status=0)
Definition GenParticle.h:21
HepMC3::GenParticlePtr GenParticlePtr
Definition GenParticle.h:19
GenVertexPtr newGenVertexPtr(const HepMC3::FourVector &pos=HepMC3::FourVector::ZERO_VECTOR(), const int i=0)
Definition GenVertex.h:25
HepMC3::GenVertexPtr GenVertexPtr
Definition GenVertex.h:23
status
Definition merge.py:16

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

72 {
73 if (events()->empty())
74 ATH_MSG_ERROR("McEventCollection is empty during first event access");
75 return *(events()->begin());
76 }
#define ATH_MSG_ERROR(x)
static const Attributes_t empty

◆ decayParticle()

void EvtInclusiveDecay::decayParticle ( HepMC::GenEvent * hepMC,
HepMC::GenParticlePtr p )
private

Definition at line 410 of file EvtInclusiveDecay.cxx.

410 {
411 ATH_MSG_DEBUG("Decaying particle " << pdgName(part) << " " << part);
412 if (msgLvl(MSG::VERBOSE)) HepMC::Print::line(std::cout,part);
413
414 // Remove existing decay tree, if any, and flag particle as being decayed by EvtGen
415 removeDecayTree(hepMC,part);
416 part->set_status(0);
417
418 // Create EvtGen version of part and have EvtGen decay it.
419 // Since EvtGen uses GeV, convert particles momentum from MeV to GeV.
420 int id = part->pdg_id();
421 EvtId evtId=EvtPDL::evtIdFromStdHep(id);
422 double en =(part->momentum()).e()/1000.;
423 double px=(part->momentum()).px()/1000.;
424 double py=(part->momentum()).py()/1000.;
425 double pz=(part->momentum()).pz()/1000.;
426 EvtVector4R evtP(en,px,py,pz);
427 EvtParticle* evtPart = EvtParticleFactory::particleFactory(evtId,evtP);
428
429 // set transverse polarization to vector mesons (relevant for coherent production of J/Psi etc in UPC)
430 if(m_setVMtransversePol && (id==113 || id== 443 || id==100443 || id==553 || id==100553 || id==200553) )evtPart->setVectorSpinDensity();
431
432 m_myEvtGen->generateDecay(evtPart);
433 if (msgLvl(MSG::VERBOSE)) evtPart->printTree();
434 double ct_s = part->production_vertex()->position().t();
435 double x_s = part->production_vertex()->position().x();
436 double y_s = part->production_vertex()->position().y();
437 double z_s = part->production_vertex()->position().z();
438
439 EvtVector4R treeStart(ct_s,x_s,y_s,z_s);
440 // Add new decay tree to hepMC, converting back from GeV to MeV.
441 addEvtGenDecayTree(hepMC, part, evtPart, treeStart, 1000.);
442 if(evtPart->getNDaug() !=0) part->set_status(2);
443 evtPart->deleteTree();
444}
#define ATH_MSG_DEBUG(x)
bool msgLvl(const MSG::Level lvl) const
void removeDecayTree(HepMC::GenEvent *hepMC, HepMC::GenParticlePtr p)
std::string pdgName(HepMC::ConstGenParticlePtr p, bool statusHighlighting=false, std::set< HepMC::GenParticlePtr, ParticleIdCompare > *particleSet=nullptr)

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

79 {
80 const McEventCollection* coll = events_const(ctx);
81 if (coll->empty())
82 ATH_MSG_ERROR("Const McEventCollection is empty during first event access");
83 return *(coll->begin());
84 }
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:93

◆ events_const()

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

Access the current event's McEventCollection (const).

Definition at line 93 of file GenBase.h.

93 {
94 SG::ReadHandle<McEventCollection> ret = SG::makeHandle(m_mcevents_const, ctx);
95 if (!ret.isValid())
96 ATH_MSG_ERROR("No McEventCollection found in StoreGate with key " << m_mcevents_const.key());
97 return ret.cptr();
98 }
SG::ReadHandleKey< McEventCollection > m_mcevents_const
Const handle to the MC event collection.
Definition GenBase.h:117
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 EvtInclusiveDecay::execute ( const EventContext & ctx)
virtual

Execute method.

Reimplemented from GenBase.

Definition at line 215 of file EvtInclusiveDecay.cxx.

215 {
216 ATH_MSG_DEBUG("EvtInclusiveDecay executing");
217
219
220 std::string key = m_inputKeyName;
221 // retrieve event from Transient Store (Storegate)
222
223 // Load HepMC info
224 // FIXME should be using Read/WriteHandles here
225 const McEventCollection* oldmcEvtColl{};
226 if(m_readExisting) {
227 CHECK(evtStore()->retrieve(oldmcEvtColl, key));
228 // Fill the new McEventCollection with a copy of the initial HepMC::GenEvent
229 m_mcEvtColl = new McEventCollection(*oldmcEvtColl);
230 }
231 else {CHECK(evtStore()->retrieve(m_mcEvtColl, key));}
232
233 if(m_readExisting) {
234 if(m_outputKeyName!=key) {
236 }
237 }
238
240 for( mcItr = m_mcEvtColl->begin(); mcItr != m_mcEvtColl->end(); ++mcItr ) {
241 HepMC::GenEvent* hepMC = *mcItr;
242
243 // Search HepMC record for particles to be decayed by EvtGen
244 // NOTE: In order to ensure repeatability, we use customized a std::set to obtain
245 // an ordered list of particles to be decayed by EvtGen.
246 std::set<HepMC::GenVertexPtr> visited;
247 std::set<HepMC::GenParticlePtr,ParticleIdCompare> toBeDecayed;
248 for (auto p: *hepMC) {
249 if ( (!p->production_vertex()) ||
250 (p->production_vertex()->particles_in_size() == 0) ) {
251 StatusCode sc = traverseDecayTree(std::move(p),false,visited,toBeDecayed);
252 if (sc.isFailure())
253 return StatusCode::FAILURE;
254 }
255 }
256 // Print HepMC in tree format if desired (before doing anything)
258 msg(MSG::INFO) << "Printing HepMC record at " << hepMC << " BEFORE running EvtGen:" << endmsg;
260 printHepMC(hepMC,&toBeDecayed);
261 else
262 printHepMC(hepMC);
263 }
264
265
266 // Decay selected particles
267 bool eventPassesCuts(false);
268 int loopCounter(0);
269 while( !eventPassesCuts && loopCounter < m_maxNRepeatedDecays ) {
270
271 for (auto p: toBeDecayed) {
272 if (p == 0) {
273 msg(MSG::ERROR ) << "Overlapping decay tree for particle" << p <<endmsg;
274 return StatusCode::FAILURE;
275 }
276 decayParticle(hepMC,std::move(p));
278 }
279
281 eventPassesCuts = passesUserSelection(hepMC);
282 else
283 eventPassesCuts = true;
284
286 loopCounter++;
287 }
288
289 // Store the number of decay attempts in event weights std::map, only if repeated decays enabled
290
291 if(m_maxNRepeatedDecays > 1) {
292 hepMC->weight("nEvtGenDecayAttempts") = loopCounter;
293 }
294 // Print HepMC in tree format if desired (after finishing all EvtGen decays)
296 msg(MSG::INFO) << "Printing HepMC record at " << hepMC << " AFTER running EvtGen:" << endmsg;
298 printHepMC(hepMC,&toBeDecayed);
299 else
300 printHepMC(hepMC);
301 }
302 }
303
304 if(m_readExisting && m_outputKeyName==key) {
306 }
307
308 return StatusCode::SUCCESS;
309}
#define endmsg
#define CHECK(...)
Evaluate an expression and check for errors.
static Double_t sc
ServiceHandle< StoreGateSvc > & evtStore()
DataModel_detail::iterator< DataVector > iterator
Definition DataVector.h:842
void printHepMC(HepMC::GenEvent *hepMC, std::set< HepMC::GenParticlePtr, ParticleIdCompare > *particleSet=nullptr)
void reseedRandomEngine(const std::string &streamName, const EventContext &ctx)
StatusCode traverseDecayTree(HepMC::GenParticlePtr p, bool isToBeRemoved, std::set< HepMC::GenVertexPtr > &visited, std::set< HepMC::GenParticlePtr, ParticleIdCompare > &toBeDecayed)
bool passesUserSelection(HepMC::GenEvent *hepMC)
McEventCollection * m_mcEvtColl
void decayParticle(HepMC::GenEvent *hepMC, HepMC::GenParticlePtr p)
::StatusCode StatusCode
StatusCode definition for legacy code.
void fillBarcodesAttribute(GenEvent *e)
Definition GenEvent.h:393
HepMC3::GenEvent GenEvent
Definition GenEvent.h:39

◆ 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 & AthCommonAlgorithm< Gaudi::Algorithm >::extraOutputDeps ( ) const
overridevirtualinherited

Return the list of extra output dependencies.

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

Definition at line 89 of file AthCommonAlgorithm.cxx.

54{
55 // If we didn't find any symlinks to add, just return the collection
56 // from the base class. Otherwise, return the extended collection.
57 if (!m_extendedExtraObjects.empty()) {
59 }
61}
Common base class for algorithms.

◆ filterPassed()

virtual bool AthCommonAlgorithm< Gaudi::Algorithm >::filterPassed ( const EventContext & ctx) const
inlinevirtualinherited

Get filter decision:

Definition at line 93 of file AthCommonAlgorithm.h.

93 {
94 return execState( ctx ).filterPassed();
95 }
virtual bool filterPassed(const EventContext &ctx) const
Get filter decision:

◆ finalize()

StatusCode EvtInclusiveDecay::finalize ( )

Definition at line 312 of file EvtInclusiveDecay.cxx.

312 {
313
315 ATH_MSG_INFO("The following particles were checked and didn't have any decay channels:");
316 if (msgLvl(MSG::INFO)) {
317 std::cout << std::endl;
318 std::cout << " Particle code # Occurences" << std::endl;
319 std::cout << "----------------------------------" << std::endl;
320 for (std::map<int,long>::iterator p = m_noDecayChannels.begin(); p!=m_noDecayChannels.end(); ++p) {
321 int id = p->first;
322 int count = p->second;
323 std::cout << std::setw(14) << id
324 << std::setw(20) << count
325 << std::endl;
326 }
327 std::cout << std::endl;
328 }
329 }
330 ATH_MSG_INFO("Total number of repeated decays: " << m_nRepeatedDecays);
331 ATH_MSG_INFO("EvtInclusiveDecay finalized");
332 return StatusCode::SUCCESS;
333}
#define ATH_MSG_INFO(x)
std::map< int, long > m_noDecayChannels
int count(std::string s, const std::string &regx)
count how many occurances of a regx are in a string
Definition hcg.cxx:148

◆ getRandomEngine()

CLHEP::HepRandomEngine * EvtInclusiveDecay::getRandomEngine ( const std::string & streamName,
unsigned long int randomSeedOffset,
const EventContext & ctx ) const
private

Definition at line 189 of file EvtInclusiveDecay.cxx.

191{
192 ATHRNG::RNGWrapper* rngWrapper = m_rndmSvc->getEngine(this, streamName);
193 rngWrapper->setSeed( streamName, ctx.slot(), randomSeedOffset, ctx.eventID().run_number() );
194 return rngWrapper->getEngine(ctx);
195}
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
ServiceHandle< IAthRNGSvc > m_rndmSvc

◆ getRandomEngineDuringInitialize()

CLHEP::HepRandomEngine * EvtInclusiveDecay::getRandomEngineDuringInitialize ( const std::string & streamName,
unsigned long int randomSeedOffset,
unsigned int conditionsRun = 1,
unsigned int lbn = 1 ) const
private

Definition at line 198 of file EvtInclusiveDecay.cxx.

199{
200 const size_t slot=0;
201 EventContext ctx;
202 ctx.setSlot( slot );
203 ctx.setEventID (EventIDBase (conditionsRun,
204 EventIDBase::UNDEFEVT, // event
205 EventIDBase::UNDEFNUM, // timestamp
206 EventIDBase::UNDEFNUM, // timestamp ns
207 lbn));
209 Atlas::ExtendedEventContext( evtStore()->hiveProxyDict(),
210 conditionsRun) );
211 return getRandomEngine(streamName, randomSeedOffset, ctx);
212}
CLHEP::HepRandomEngine * getRandomEngine(const std::string &streamName, unsigned long int randomSeedOffset, const EventContext &ctx) const
void setExtendedEventContext(EventContext &ctx, ExtendedEventContext &&ectx)
Move an extended context into a context object.

◆ initialize()

StatusCode EvtInclusiveDecay::initialize ( )
virtual

Reimplemented from GenBase.

Definition at line 116 of file EvtInclusiveDecay.cxx.

116 {
117
119 // Get the random number service
120 CHECK(m_rndmSvc.retrieve());
121
122 msg(MSG::INFO) << "EvtInclusiveDecay initialize" << endmsg;
123 msg(MSG::INFO) << "Particle properties definition file = " << m_pdtFile << endmsg;
124 msg(MSG::INFO) << "Main decay file = " << m_decayFile << endmsg;
125 msg(MSG::INFO) << "User decay file = " << m_userDecayFile << endmsg;
126 msg(MSG::INFO) << "Max number of repeated decays = " << m_maxNRepeatedDecays << endmsg;
127 msg(MSG::INFO) << "EvtInclusiveDecay selection parameters:" << endmsg;
128 msg(MSG::INFO) << "* prohibitFinalStateDecay = " << m_prohibitFinalStateDecay << endmsg;
129 msg(MSG::INFO) << "* prohibitReDecay = " << m_prohibitReDecay << endmsg;
130 msg(MSG::INFO) << "* prohibitUnDecay = " << m_prohibitUnDecay << endmsg;
131 msg(MSG::INFO) << "* prohibitRemoveSelfDecay = " << m_prohibitRemoveSelfDecay << endmsg;
132 msg(MSG::INFO) << "* allowAllKnownDecays = " << m_allowAllKnownDecays << endmsg;
133 msg(MSG::INFO) << "* allowDefaultBDecays = " << m_allowDefaultBDecays << endmsg;
134 msg(MSG::INFO) << "User selection parameters:" << endmsg;
135 msg(MSG::INFO) << "* applyUserSelection = " << m_applyUserSelection << endmsg;
136 msg(MSG::INFO) << "* userSelRequireOppositeSignedMu = " << m_userSelRequireOppositeSignedMu << endmsg;
137 msg(MSG::INFO) << "* userSelMu1MinPt = " << m_userSelMu1MinPt << endmsg;
138 msg(MSG::INFO) << "* userSelMu2MinPt = " << m_userSelMu2MinPt << endmsg;
139 msg(MSG::INFO) << "* userSelMu1MaxEta = " << m_userSelMu1MaxEta << endmsg;
140 msg(MSG::INFO) << "* userSelMu2MaxEta = " << m_userSelMu2MaxEta << endmsg;
141 msg(MSG::INFO) << "* userSelMinDimuMass = " << m_userSelMinDimuMass << endmsg;
142 msg(MSG::INFO) << "* userSelMaxDimuMass = " << m_userSelMaxDimuMass << endmsg;
143
144 // Initialize and print blackList
145 m_blackListSet.insert(m_blackList.begin(),m_blackList.end());
146 msg(MSG::INFO) << "* blackList; = ";
147 for (std::set<int>::iterator i = m_blackListSet.begin(); i!=m_blackListSet.end(); ++i)
148 msg(MSG::INFO) << (*i) << " ";
149 msg(MSG::INFO)<< endmsg;
150
151 // Initialize and print whiteList
152 m_whiteListSet.insert(m_whiteList.begin(),m_whiteList.end());
153 msg(MSG::INFO) << "* whiteList = ";
154 for (std::set<int>::iterator i = m_whiteListSet.begin(); i!=m_whiteListSet.end(); ++i)
155 msg(MSG::INFO) << (*i) << " ";
156 msg(MSG::INFO) << endmsg;
157
158 CLHEP::HepRandomEngine* rndmEngine = getRandomEngineDuringInitialize(m_randomStreamName, m_randomSeed, m_dsid);
159 // Obtain random number generator for EvtGen
160 m_evtAtRndmGen = new EvtInclusiveAtRndmGen(rndmEngine);
161
162 // Create an instance of EvtGen and read particle properties and decay files
163 EvtExternalGenList genList(true,xmlpath(),"gamma");
164 EvtAbsRadCorr* radCorrEngine = genList.getPhotosModel();
165 std::list<EvtDecayBase*> extraModels = genList.getListOfModels();
166
167 // Create the EvtGen generator object
168 // EvtGen myGenerator("decayFile.dec", "evt.pdl", randomEnginePointer,
169 // radCorrEngine, &extraModels);
170
171
172 m_myEvtGen = new EvtGen( m_decayFile.c_str(), m_pdtFile.c_str(), m_evtAtRndmGen, radCorrEngine, &extraModels);
173 if(!m_userDecayFile.empty())
174 m_myEvtGen->readUDecay(m_userDecayFile.c_str());
175
176 return StatusCode::SUCCESS;
177}
#define ATH_CHECK
Evaluate an expression and check for errors.
std::set< int > m_blackListSet
CLHEP::HepRandomEngine * getRandomEngineDuringInitialize(const std::string &streamName, unsigned long int randomSeedOffset, unsigned int conditionsRun=1, unsigned int lbn=1) const
IntegerProperty m_dsid
std::string xmlpath(void)
std::set< int > m_whiteListSet
IntegerProperty m_randomSeed
Seed for random number engine.
virtual StatusCode initialize() override
Definition GenBase.cxx:17

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

◆ invMass()

double EvtInclusiveDecay::invMass ( HepMC::ConstGenParticlePtr p1,
HepMC::ConstGenParticlePtr p2 )
private

Definition at line 607 of file EvtInclusiveDecay.cxx.

607 {
608 double p1Px = p1->momentum().px();
609 double p1Py = p1->momentum().py();
610 double p1Pz = p1->momentum().pz();
611 double p1E = p1->momentum().e();
612 double p2Px = p2->momentum().px();
613 double p2Py = p2->momentum().py();
614 double p2Pz = p2->momentum().pz();
615 double p2E = p2->momentum().e();
616 double dimuE = p2E + p1E;
617 double dimuPx = p2Px + p1Px;
618 double dimuPy = p2Py + p1Py;
619 double dimuPz = p2Pz + p1Pz;
620 double invMass = std::sqrt(dimuE*dimuE - dimuPx*dimuPx - dimuPy*dimuPy - dimuPz*dimuPz);
621
622 return invMass;
623}
double invMass(HepMC::ConstGenParticlePtr p1, HepMC::ConstGenParticlePtr p2)

◆ isClonable()

virtual bool AthCommonAlgorithm< Gaudi::Algorithm >::isClonable ( ) const
inlineoverridevirtualinherited

Specify if the algorithm is clonable.

Only relevant for non-reentrant algorithms. Actual number of clones needs to be set via the "Cardinality" property.

Reimplemented in AFP_DigiTop, AlgB, AlgT, BCM_Digitization, CscDigitBuilder, CscDigitToCscRDO, G4AtlasAlg, G4RunAlg, HGTD_Digitization, HiveAlgBase, InDet::GNNSeedingTrackMaker, InDet::SCT_Clusterization, InDet::SiSPGNNTrackMaker, InDet::SiSPSeededTrackFinder, InDet::SiTrackerSpacePointFinder, ISF::SimKernelMT, ITk::StripDigitization, ITkPixelCablingAlg, ITkStripCablingAlg, LArHitEMapMaker, LArTTL1Maker, LUCID_DigiTop, LVL1::L1TopoSimulation, MergeCalibHits, MergeGenericMuonSimHitColl, MergeHijingPars, MergeMcEventCollection, MergeTrackRecordCollection, MergeTruthJets, MergeTruthParticles, MuonDigitizer, PileUpMTAlg, PixelDigitization, RoIBResultToxAOD, SCT_ByteStreamErrorsTestAlg, SCT_CablingCondAlgFromCoraCool, SCT_CablingCondAlgFromText, SCT_ConditionsParameterTestAlg, SCT_ConditionsSummaryTestAlg, SCT_ConfigurationConditionsTestAlg, SCT_Digitization, SCT_FlaggedConditionTestAlg, SCT_LinkMaskingTestAlg, SCT_MajorityConditionsTestAlg, SCT_ModuleVetoTestAlg, SCT_MonitorConditionsTestAlg, SCT_PrepDataToxAOD, SCT_RawDataToxAOD, SCT_ReadCalibChipDataTestAlg, SCT_ReadCalibDataTestAlg, SCT_RODVetoTestAlg, SCT_SensorsTestAlg, SCT_SiliconConditionsTestAlg, SCT_StripVetoTestAlg, SCT_TdaqEnabledTestAlg, SCT_TestCablingAlg, SCTEventFlagWriter, SCTRawDataProvider, SCTSiLorentzAngleTestAlg, SCTSiPropertiesTestAlg, SGInputLoader, Simulation::BeamEffectsAlg, TileHitVecToCnt, TileMuonFitter, TilePulseForTileMuonReceiver, TileRawChannelMaker, TRTDigitization, and ZDC_DigiTop.

Definition at line 68 of file AthCommonAlgorithm.h.

68 {
69 return true;
70 }

◆ isDefaultB()

bool EvtInclusiveDecay::isDefaultB ( const int pId) const
private

Definition at line 555 of file EvtInclusiveDecay.cxx.

555 {
556 int id = std::abs(pId);
557 if ( id == 511 ||
558 id == 521 ||
559 id == 531 ||
560 id == 541 ||
561 id == 5122 ||
562 id == 5132 ||
563 id == 5232 ||
564 id == 5112 ||
565 id == 5212 ||
566 id == 5222 )
567 return true;
568 else
569 return false;
570}

◆ isReEntrant()

virtual bool AthAlgorithm::isReEntrant ( ) const
inlinefinaloverrideprotectedvirtualinherited

Legacy algorithms are not thread-safe.

Definition at line 47 of file AthAlgorithm.h.

47{ return false; }

◆ isToBeDecayed()

bool EvtInclusiveDecay::isToBeDecayed ( HepMC::ConstGenParticlePtr p,
bool doCrossChecks )
private

Definition at line 486 of file EvtInclusiveDecay.cxx.

486 {
487 int id = p->pdg_id();
488 int nDaughters = 0;
489 auto v = p->end_vertex();
490 if (v) nDaughters = v->particles_out_size();
491
492 // Ignore documentation lines
493 if (p->status() == 3) return false;
494 // And any particles that aren't stable or decayed
495 if(!m_isfHerwig && !MC::isPhysical(p)) return false;
496
497 // Particularly for Herwig, try to ignore particles that really should
498 // be flagged as documentation lines
499 double m2 = p->momentum().m2();
500 if (m2 < -1.0E-3) {
501 ATH_MSG_DEBUG("Ignoring particle " << pdgName(std::move(p)) << " with m^2 = " << m2);
502 return false;
503 }
504
505 // Check whether EvtGen has any decay channels defined for this particle
506 EvtId evtId = EvtPDL::evtIdFromStdHep(id);
507 // std::cout << "EVTID: " << evtId.getId() << " alias " << evtId.getAlias() << std::endl;
508 int nModes = 0;
509 if (evtId.getId()>=0)
510 // nModes = EvtDecayTable::getNMode(evtId.getAlias());
511 nModes = EvtDecayTable::getInstance()->getNMode(evtId.getAlias());
512 if (doCrossChecks) {
513 ATH_MSG_VERBOSE("Checking particle " << pdgName(p)
514 << " (status = " << p->status()
515 <<") -- " << nModes << " decay modes found");
516 if (m_checkDecayChannels && nModes==0) {
517 std::map<int,long>::iterator pos = m_noDecayChannels.find(id);
518 if (pos != m_noDecayChannels.end())
519 (pos->second)++;
520 else
522 }
523 }
524
525 // Check prohibit* settings
526 if (m_prohibitFinalStateDecay && MC::isStable(p)) return false;
527 if (m_prohibitReDecay && nDaughters>0) return false;
528 if (m_prohibitUnDecay && nModes==0) return false;
529 if (m_prohibitRemoveSelfDecay && nDaughters>0) {
530 // For now, check only children - this should be sufficient and checking all
531 // descendants would be very expensive.
532 for (auto itd: *v) {
533 if (std::abs(itd->pdg_id()) == std::abs(id)) return false;
534 }
535 }
536
537 // Check blackList
538 if (m_blackListSet.count(std::abs(id))>0) return false;
539
540 // Check allow* settings
541 if (m_allowAllKnownDecays && nModes>0) return true;
542 if (m_allowDefaultBDecays && isDefaultB(id)) return true;
543
544 // Check whiteList
545 if (m_whiteListSet.count(std::abs(id))>0) return true;
546
547 return false; // Default is NOT to decay through EvtGen
548}
#define ATH_MSG_VERBOSE(x)
bool isDefaultB(const int pId) const
bool isStable(const T &p)
Identify if the particle is stable, i.e. has not decayed.
bool isPhysical(const T &p)
Identify if the particle is physical, i.e. is stable or decayed.

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

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

◆ passesUserSelection()

bool EvtInclusiveDecay::passesUserSelection ( HepMC::GenEvent * hepMC)
private

Definition at line 577 of file EvtInclusiveDecay.cxx.

577 {
578 bool passed(false);
579 std::vector<HepMC::GenParticlePtr> *muons = new std::vector<HepMC::GenParticlePtr>;
580
581 for ( const auto& p: *hepMC) {
582 if( std::abs(p->pdg_id()) == 13 )
583 muons->push_back(p);
584 }
585
586 for (auto muItr1 = muons->begin(); muItr1 != muons->end(); ++muItr1) {
587 for (auto muItr2 = muItr1+1; muItr2 != muons->end(); ++muItr2) {
588 if( m_userSelRequireOppositeSignedMu && (*muItr1)->pdg_id() * (*muItr2)->pdg_id() > 0)
589 continue;
590 if( !( (*muItr1)->momentum().perp() > m_userSelMu1MinPt && std::abs((*muItr1)->momentum().pseudoRapidity()) < m_userSelMu1MaxEta &&
591 (*muItr2)->momentum().perp() > m_userSelMu2MinPt && std::abs((*muItr2)->momentum().pseudoRapidity()) < m_userSelMu2MaxEta ) &&
592 !( (*muItr2)->momentum().perp() > m_userSelMu1MinPt && std::abs((*muItr2)->momentum().pseudoRapidity()) < m_userSelMu1MaxEta &&
593 (*muItr1)->momentum().perp() > m_userSelMu2MinPt && std::abs((*muItr1)->momentum().pseudoRapidity()) < m_userSelMu2MaxEta ) )
594 continue;
595 double dimuMass = invMass((*muItr1),(*muItr2));
596 if( !( dimuMass > m_userSelMinDimuMass && (dimuMass < m_userSelMaxDimuMass || m_userSelMaxDimuMass < 0.) ) )
597 continue;
598 passed = true;
599 }
600 }
601
602 delete muons;
603
604 return passed;
605}
bool passed(DecisionID id, const DecisionIDContainer &)
checks if required decision ID is in the set of IDs in the container

◆ pdgName()

std::string EvtInclusiveDecay::pdgName ( HepMC::ConstGenParticlePtr p,
bool statusHighlighting = false,
std::set< HepMC::GenParticlePtr, ParticleIdCompare > * particleSet = nullptr )
private

Definition at line 678 of file EvtInclusiveDecay.cxx.

678 {
679 std::ostringstream buf;
680 bool inlist = false;
681 if (particleSet) for (const auto& pinl: *particleSet) if (pinl&&p) if (pinl.get() == p.get()) inlist=true;
682 if (statusHighlighting) {
683 if ( ((particleSet!=0) && (inlist)) ||
684 ((particleSet==0) && isToBeDecayed(p,false)) )
685 buf << "\033[7m"; // reverse
686 if (p and !MC::isStable(p)) {
687 if (MC::isDecayed(p))
688 buf << "\033[33m"; // yellow
689 else
690 buf << "\033[31m"; // red
691 }
692 }
693 if (p){
694 buf << p->pdg_id();
695 if (statusHighlighting) {
696 buf << "\033[0m"; // revert color attributes
697 }
698 }
699 return buf.str();
700}
bool isToBeDecayed(HepMC::ConstGenParticlePtr p, bool doCrossChecks)
bool isDecayed(const T &p)
Identify if the particle decayed.

◆ printHepMC()

void EvtInclusiveDecay::printHepMC ( HepMC::GenEvent * hepMC,
std::set< HepMC::GenParticlePtr, ParticleIdCompare > * particleSet = nullptr )
private

Definition at line 631 of file EvtInclusiveDecay.cxx.

631 {
632 std::set<HepMC::GenVertexPtr> visited;
633 unsigned int nParticlesFound = 0;
634 unsigned int nTreesFound = 0;
635 for (auto p: *hepMC) {
636 if ( (!p->production_vertex()) ||
637 (p->production_vertex()->particles_in_size() == 0) ) {
638 nTreesFound++;
639 std::cout << "\n Found new partial decay tree:\n" << std::endl;
640 unsigned int nParticlesVisited = printTree(std::move(p),visited,1,particleSet);
641 std::cout << "\n " << nParticlesVisited << " particles in this subtree" << std::endl;
642 nParticlesFound += nParticlesVisited;
643 }
644 }
645 std::cout << "\n Total of " << nParticlesFound << " particles found in "
646 << nTreesFound << " decay subtrees in HepMC event record\n" << std::endl;
647}
unsigned int printTree(HepMC::GenParticlePtr p, std::set< HepMC::GenVertexPtr > &visited, int level, std::set< HepMC::GenParticlePtr, ParticleIdCompare > *particleSet=nullptr)

◆ printTree()

unsigned int EvtInclusiveDecay::printTree ( HepMC::GenParticlePtr p,
std::set< HepMC::GenVertexPtr > & visited,
int level,
std::set< HepMC::GenParticlePtr, ParticleIdCompare > * particleSet = nullptr )
private

Definition at line 649 of file EvtInclusiveDecay.cxx.

649 {
650
651 unsigned int nParticlesVisited = 1;
652 for (int i=0; i<level; i++) std::cout << " ";
653 std::cout << pdgName(p,m_printHepMCHighlighted,particleSet);
654 auto v = p->end_vertex();
655 if (v) {
656 if (v->particles_in_size() > 1)
657 std::cout << " [interaction: " << v->particles_in_size() << " particles, vertex " << v << "] --> ";
658 else
659 std::cout << " --> ";
660 if (visited.insert(v).second) {
661 for (auto itp: *v) {
662 std::cout << pdgName(itp,m_printHepMCHighlighted,particleSet) << " ";
663 }
664 std::cout << std::endl;
665 for (auto itp: *v) {
666 if (itp->end_vertex())
667 nParticlesVisited += printTree(std::move(itp), visited, level+1, particleSet);
668 else
669 nParticlesVisited++;
670 }
671 } else
672 std::cout << "see above" << std::endl;
673 } else
674 std::cout << " no decay vertex\n" << std::endl;
675 return nParticlesVisited;
676}

◆ removeDecayTree()

void EvtInclusiveDecay::removeDecayTree ( HepMC::GenEvent * hepMC,
HepMC::GenParticlePtr p )
private

Definition at line 383 of file EvtInclusiveDecay.cxx.

383 {
384 auto v = p->end_vertex();
385 if (v) {
386 //This is recursive in HepMC3. But explicit deletion is allowed as well.
387 hepMC->remove_vertex(std::move(v));
388 p->set_status(1); // For now, flag particle as undecayed (stable)
389 ATH_MSG_DEBUG("Removed existing " << pdgName(p) << " " << p );
390 }
391}

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

◆ reseedRandomEngine()

void EvtInclusiveDecay::reseedRandomEngine ( const std::string & streamName,
const EventContext & ctx )
private

Definition at line 180 of file EvtInclusiveDecay.cxx.

182{
183 long seeds[7];
184 ATHRNG::calculateSeedsMC21(seeds, streamName, ctx.eventID().event_number(), m_dsid, m_randomSeed);
185 m_evtAtRndmGen->getEngine()->setSeeds(seeds, 0); // NOT THREAD-SAFE
186}
void calculateSeedsMC21(long *seeds, const std::string &algName, uint64_t ev, uint64_t run, uint64_t offset=0)
Set the random seed using a string (e.g.

◆ setFilterPassed()

virtual void AthCommonAlgorithm< Gaudi::Algorithm >::setFilterPassed ( bool state,
const EventContext & ctx ) const
inlinevirtualinherited

Set filter decision:

Reimplemented in AthFilterAlgorithm.

Definition at line 99 of file AthCommonAlgorithm.h.

99 {
101 }
virtual void setFilterPassed(bool state, const EventContext &ctx) const
Set filter decision:

◆ sysExecute()

StatusCode AthCommonAlgorithm< Gaudi::Algorithm >::sysExecute ( const EventContext & ctx)
overridevirtualinherited

Execute an algorithm.

We override this in order to work around an issue with the Algorithm base class storing the event context in a member variable that can cause crashes in MT jobs.

Reimplemented in AthAnalysisAlgorithm.

Definition at line 80 of file AthCommonAlgorithm.cxx.

41{
42 return BaseAlg::sysExecute (ctx);
43}

◆ sysInitialize()

StatusCode AthCommonAlgorithm< Gaudi::Algorithm >::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, HypoBase, InputMakerBase, and PyAthena::Alg.

Definition at line 60 of file AthCommonAlgorithm.cxx.

71 {
73
74 if (sc.isFailure()) {
75 return sc;
76 }
77
78 ServiceHandle<ICondSvc> cs("CondSvc",name());
79 for (auto h : outputHandles()) {
80 if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
81 // do this inside the loop so we don't create the CondSvc until needed
82 if ( cs.retrieve().isFailure() ) {
83 ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
85 }
86 if (cs->regHandle(this,*h).isFailure()) {
88 ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
89 << " with CondSvc");
90 }
91 }
92 }
93 return sc;
94}
#define ATH_MSG_WARNING(x)
virtual StatusCode sysInitialize() override
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.

◆ traverseDecayTree()

StatusCode EvtInclusiveDecay::traverseDecayTree ( HepMC::GenParticlePtr p,
bool isToBeRemoved,
std::set< HepMC::GenVertexPtr > & visited,
std::set< HepMC::GenParticlePtr, ParticleIdCompare > & toBeDecayed )
private

Definition at line 344 of file EvtInclusiveDecay.cxx.

347 {
348 ATH_MSG_VERBOSE("Inspecting: " << pdgName(p) << " " << p);
349 if (!isToBeRemoved) {
350 if (isToBeDecayed(p,true)) {
351 toBeDecayed.insert(p);
352 isToBeRemoved = true;
353 ATH_MSG_VERBOSE("Selected particle for decay: " << pdgName(p) << " " << p );
354
355 // In principle we could stop the recursion here. However, to prevent
356 // pathological cases in certain decay trees (in particular from Herwig),
357 // we continue in order to mark all descendants of this particle
358 // as visited. Thus none of these descendants can be flagged for further
359 // decay, even if it has several mothers.
360 }
361 }
362 auto v = p->end_vertex();
363 if (v) {
364 if (visited.insert(v).second) {
365 if ( isToBeRemoved && (v->particles_in_size()>1) && m_checkDecayTree ) {
366 ATH_MSG_WARNING("Found particle to be decayed with vertex with >1 incoming mother particles in decay tree");
367 ATH_MSG_WARNING( ([&p, &v](){ std::stringstream ss; HepMC::Print::line(ss,p); HepMC::Print::line(ss,v); return ss.str();})());
368 }
369 for (auto itp: *v) {
370 ATH_CHECK(traverseDecayTree(std::move(itp),isToBeRemoved,visited,toBeDecayed) );
371 }
372
373 }
374 }
375 return StatusCode::SUCCESS;
376}
static Double_t ss

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

◆ xmlpath()

std::string EvtInclusiveDecay::xmlpath ( void )

Definition at line 716 of file EvtInclusiveDecay.cxx.

716 {
717 return PathResolverFindCalibDirectory( "Pythia8/xmldoc" );
718}
std::string PathResolverFindCalibDirectory(const std::string &logical_file_name)

Member Data Documentation

◆ m_allowAllKnownDecays

bool EvtInclusiveDecay::m_allowAllKnownDecays
private

Definition at line 125 of file EvtInclusiveDecay.h.

◆ m_allowDefaultBDecays

bool EvtInclusiveDecay::m_allowDefaultBDecays
private

Definition at line 126 of file EvtInclusiveDecay.h.

◆ m_applyUserSelection

bool EvtInclusiveDecay::m_applyUserSelection
private

Definition at line 143 of file EvtInclusiveDecay.h.

◆ m_blackList

std::vector<int> EvtInclusiveDecay::m_blackList
private

Definition at line 122 of file EvtInclusiveDecay.h.

◆ m_blackListSet

std::set<int> EvtInclusiveDecay::m_blackListSet
private

Definition at line 123 of file EvtInclusiveDecay.h.

◆ m_checkDecayChannels

bool EvtInclusiveDecay::m_checkDecayChannels
private

Definition at line 136 of file EvtInclusiveDecay.h.

◆ m_checkDecayTree

bool EvtInclusiveDecay::m_checkDecayTree
private

Definition at line 135 of file EvtInclusiveDecay.h.

◆ m_decayFile

std::string EvtInclusiveDecay::m_decayFile
private

Definition at line 111 of file EvtInclusiveDecay.h.

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

IntegerProperty EvtInclusiveDecay::m_dsid {this, "Dsid", 999999}
private

Definition at line 98 of file EvtInclusiveDecay.h.

98{this, "Dsid", 999999};

◆ m_evtAtRndmGen

EvtInclusiveAtRndmGen* EvtInclusiveDecay::m_evtAtRndmGen {}
private

Definition at line 106 of file EvtInclusiveDecay.h.

106{};

◆ 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 AthCommonAlgorithm< Gaudi::Algorithm >::m_extendedExtraObjects
privateinherited

Extra output dependency collection, extended by AthAlgorithmDHUpdate to add symlinks.

Empty if no symlinks were found.

Definition at line 108 of file AthCommonAlgorithm.h.

◆ m_inputKeyName

std::string EvtInclusiveDecay::m_inputKeyName
private

Definition at line 114 of file EvtInclusiveDecay.h.

◆ m_isfHerwig

bool EvtInclusiveDecay::m_isfHerwig
private

Definition at line 151 of file EvtInclusiveDecay.h.

◆ m_maxNRepeatedDecays

int EvtInclusiveDecay::m_maxNRepeatedDecays
private

Definition at line 141 of file EvtInclusiveDecay.h.

◆ m_mcEventKey

std::string GenBase::m_mcEventKey {}
protectedinherited

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

Definition at line 108 of file GenBase.h.

108{};

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

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

◆ m_mcEvtColl

McEventCollection* EvtInclusiveDecay::m_mcEvtColl {}
private

Definition at line 103 of file EvtInclusiveDecay.h.

103{};

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

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

◆ m_myEvtGen

EvtGen* EvtInclusiveDecay::m_myEvtGen {}
private

Definition at line 107 of file EvtInclusiveDecay.h.

107{};

◆ m_noDecayChannels

std::map<int,long> EvtInclusiveDecay::m_noDecayChannels
private

Definition at line 137 of file EvtInclusiveDecay.h.

◆ m_nRepeatedDecays

int EvtInclusiveDecay::m_nRepeatedDecays
private

Definition at line 139 of file EvtInclusiveDecay.h.

◆ m_outputKeyName

std::string EvtInclusiveDecay::m_outputKeyName
private

Definition at line 115 of file EvtInclusiveDecay.h.

◆ m_pdtFile

std::string EvtInclusiveDecay::m_pdtFile
private

Definition at line 110 of file EvtInclusiveDecay.h.

◆ m_printHepMCAfterEvtGen

bool EvtInclusiveDecay::m_printHepMCAfterEvtGen
private

Definition at line 131 of file EvtInclusiveDecay.h.

◆ m_printHepMCBeforeEvtGen

bool EvtInclusiveDecay::m_printHepMCBeforeEvtGen
private

Definition at line 130 of file EvtInclusiveDecay.h.

◆ m_printHepMCHighlighted

bool EvtInclusiveDecay::m_printHepMCHighlighted
private

Definition at line 132 of file EvtInclusiveDecay.h.

◆ m_printHepMCHighLightTopLevelDecays

bool EvtInclusiveDecay::m_printHepMCHighLightTopLevelDecays
private

Definition at line 133 of file EvtInclusiveDecay.h.

◆ m_prohibitFinalStateDecay

bool EvtInclusiveDecay::m_prohibitFinalStateDecay
private

Definition at line 118 of file EvtInclusiveDecay.h.

◆ m_prohibitReDecay

bool EvtInclusiveDecay::m_prohibitReDecay
private

Definition at line 119 of file EvtInclusiveDecay.h.

◆ m_prohibitRemoveSelfDecay

bool EvtInclusiveDecay::m_prohibitRemoveSelfDecay
private

Definition at line 121 of file EvtInclusiveDecay.h.

◆ m_prohibitUnDecay

bool EvtInclusiveDecay::m_prohibitUnDecay
private

Definition at line 120 of file EvtInclusiveDecay.h.

◆ m_randomSeed

IntegerProperty EvtInclusiveDecay::m_randomSeed {this, "RandomSeed", 1234567, "Random seed for the built-in random engine"}
private

Seed for random number engine.

Definition at line 101 of file EvtInclusiveDecay.h.

101{this, "RandomSeed", 1234567, "Random seed for the built-in random engine"}; // FIXME make this into an unsigned long int?

◆ m_randomStreamName

std::string EvtInclusiveDecay::m_randomStreamName
private

Definition at line 113 of file EvtInclusiveDecay.h.

◆ m_readExisting

bool EvtInclusiveDecay::m_readExisting
private

Definition at line 117 of file EvtInclusiveDecay.h.

◆ m_rndmSvc

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

Definition at line 95 of file EvtInclusiveDecay.h.

95{this, "RndmSvc", "AthRNGSvc"};

◆ m_setVMtransversePol

bool EvtInclusiveDecay::m_setVMtransversePol
private

Definition at line 152 of file EvtInclusiveDecay.h.

◆ m_userDecayFile

std::string EvtInclusiveDecay::m_userDecayFile
private

Definition at line 112 of file EvtInclusiveDecay.h.

◆ m_userSelMaxDimuMass

double EvtInclusiveDecay::m_userSelMaxDimuMass
private

Definition at line 150 of file EvtInclusiveDecay.h.

◆ m_userSelMinDimuMass

double EvtInclusiveDecay::m_userSelMinDimuMass
private

Definition at line 149 of file EvtInclusiveDecay.h.

◆ m_userSelMu1MaxEta

double EvtInclusiveDecay::m_userSelMu1MaxEta
private

Definition at line 147 of file EvtInclusiveDecay.h.

◆ m_userSelMu1MinPt

double EvtInclusiveDecay::m_userSelMu1MinPt
private

Definition at line 145 of file EvtInclusiveDecay.h.

◆ m_userSelMu2MaxEta

double EvtInclusiveDecay::m_userSelMu2MaxEta
private

Definition at line 148 of file EvtInclusiveDecay.h.

◆ m_userSelMu2MinPt

double EvtInclusiveDecay::m_userSelMu2MinPt
private

Definition at line 146 of file EvtInclusiveDecay.h.

◆ m_userSelRequireOppositeSignedMu

bool EvtInclusiveDecay::m_userSelRequireOppositeSignedMu
private

Definition at line 144 of file EvtInclusiveDecay.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.

◆ m_whiteList

std::vector<int> EvtInclusiveDecay::m_whiteList
private

Definition at line 127 of file EvtInclusiveDecay.h.

◆ m_whiteListSet

std::set<int> EvtInclusiveDecay::m_whiteListSet
private

Definition at line 128 of file EvtInclusiveDecay.h.


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