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 const DataObjIDColl & extraOutputDeps () const override
 Return the list of extra output dependencies.
bool filterPassed (const EventContext &ctx) const
void setFilterPassed (bool state, const EventContext &ctx) const
ServiceHandle< StoreGateSvc > & evtStore ()
 The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > & detStore () const
 The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode sysStart () override
 Handle START transition.
virtual std::vector< Gaudi::DataHandle * > inputHandles () const override
 Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * > outputHandles () const override
 Return this algorithm's output handles.
Gaudi::Details::PropertyBase & declareProperty (Gaudi::Property< T, V, H > &t)
void updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream & msg () const
bool msgLvl (const MSG::Level lvl) const
Event collection accessors (const and non-const)
HepMC::GenEvent *event ATLAS_NOT_CONST_THREAD_SAFE ()
 Access the current signal event (first in the McEventCollection).
McEventCollection *events ATLAS_NOT_CONST_THREAD_SAFE ()
 Access the current event's McEventCollection.
const HepMC::GenEventevent_const (const EventContext &ctx) const
 Access the current signal event (const).
const McEventCollectionevents_const (const EventContext &ctx) const
 Access the current event's McEventCollection (const).
Particle data accessors
const ServiceHandle< IPartPropSvc > partPropSvc () const
 Access the particle property service.
const HepPDT::ParticleDataTable & particleTable () const
 Get a particle data table.
const HepPDT::ParticleDataTable & pdt () const
 Shorter alias to get a particle data table.
const HepPDT::ParticleData * particleData (int pid) const
 Access an element in the particle data table.
const EventContext & getContext () const
 Deprecated methods (use the ones with EventContext).
bool filterPassed () const
void setFilterPassed (bool state) const

Protected Member Functions

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

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

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

Detailed Description

Definition at line 55 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 56 of file EvtInclusiveDecay.cxx.

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

111 {
112 delete m_myEvtGen;
113 delete m_evtAtRndmGen;
114}
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 451 of file EvtInclusiveDecay.cxx.

452 {
453 if(evtPart->getNDaug()!=0) {
454 // Add decay vertex, starting from production vertex of particle
455 double ct=(evtPart->getDaug(0)->get4Pos()).get(0)+treeStart.get(0);
456 double x=(evtPart->getDaug(0)->get4Pos()).get(1)+treeStart.get(1);
457 double y=(evtPart->getDaug(0)->get4Pos()).get(2)+treeStart.get(2);
458 double z=(evtPart->getDaug(0)->get4Pos()).get(3)+treeStart.get(3);
459
461
462 hepMC->add_vertex(end_vtx);
463 end_vtx->add_particle_in(std::move(part));
464
465 // Add decay daughter with their own decay trees
466 for(uint it=0; it<evtPart->getNDaug(); it++) {
467 double e=(evtPart->getDaug(it)->getP4Lab()).get(0) * momentumScaleFactor;
468 double px=(evtPart->getDaug(it)->getP4Lab()).get(1) * momentumScaleFactor;
469 double py=(evtPart->getDaug(it)->getP4Lab()).get(2) * momentumScaleFactor;
470 double pz=(evtPart->getDaug(it)->getP4Lab()).get(3) * momentumScaleFactor;
471 int id=EvtPDL::getStdHep(evtPart->getDaug(it)->getId());
472 int status=1;
473 if(evtPart->getDaug(it)->getNDaug() != 0) status=2;
474 HepMC::GenParticlePtr daughter = HepMC::newGenParticlePtr(HepMC::FourVector(px,py,pz,e),id,status);
475 end_vtx->add_particle_out(daughter);
476 addEvtGenDecayTree(hepMC, std::move(daughter), evtPart->getDaug(it), treeStart, momentumScaleFactor);
477 }
478 }
479}
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 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

◆ decayParticle()

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

Definition at line 413 of file EvtInclusiveDecay.cxx.

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

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

◆ events_const()

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

Access the current event's McEventCollection (const).

Definition at line 97 of file GenBase.h.

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

◆ evtStore()

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

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

Definition at line 85 of file AthCommonDataStore.h.

◆ execute()

StatusCode EvtInclusiveDecay::execute ( const EventContext & ctx)
virtual

Execute method.

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

Reimplemented from GenBase.

Definition at line 217 of file EvtInclusiveDecay.cxx.

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

◆ extraDeps_update_handler()

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

Add StoreName to extra input/output deps as needed.

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

◆ extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const
overridevirtualinherited

Return the list of extra output dependencies.

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

Definition at line 50 of file AthAlgorithm.cxx.

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

◆ filterPassed() [1/2]

bool AthAlgorithm::filterPassed ( ) const
inherited

Definition at line 94 of file AthAlgorithm.cxx.

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

◆ filterPassed() [2/2]

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

Definition at line 98 of file AthAlgorithm.cxx.

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

◆ finalize()

StatusCode EvtInclusiveDecay::finalize ( )

Definition at line 314 of file EvtInclusiveDecay.cxx.

314 {
315
317 ATH_MSG_INFO("The following particles were checked and didn't have any decay channels:");
318 if (msgLvl(MSG::INFO)) {
319 std::cout << std::endl;
320 std::cout << " Particle code Name from HepPDT # Occurences" << std::endl;
321 std::cout << "------------------------------------------------------" << std::endl;
322 for (std::map<int,long>::iterator p = m_noDecayChannels.begin(); p!=m_noDecayChannels.end(); ++p) {
323 int id = p->first;
324 int count = p->second;
325 std::cout << std::setw(14) << id
326 << std::setw(20) << HepPID::particleName(id)
327 << std::setw(20) << count
328 << std::endl;
329 }
330 std::cout << std::endl;
331 }
332 }
333 ATH_MSG_INFO("Total number of repeated decays: " << m_nRepeatedDecays);
334 ATH_MSG_INFO("EvtInclusiveDecay finalized");
335 return StatusCode::SUCCESS;
336}
#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

◆ getContext()

const EventContext & AthAlgorithm::getContext ( ) const
inherited

Deprecated methods (use the ones with EventContext).

Definition at line 90 of file AthAlgorithm.cxx.

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

◆ getRandomEngine()

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

Definition at line 191 of file EvtInclusiveDecay.cxx.

193{
194 ATHRNG::RNGWrapper* rngWrapper = m_rndmSvc->getEngine(this, streamName);
195 rngWrapper->setSeed( streamName, ctx.slot(), randomSeedOffset, ctx.eventID().run_number() );
196 return rngWrapper->getEngine(ctx);
197}
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 200 of file EvtInclusiveDecay.cxx.

201{
202 const size_t slot=0;
203 EventContext ctx;
204 ctx.setSlot( slot );
205 ctx.setEventID (EventIDBase (conditionsRun,
206 EventIDBase::UNDEFEVT, // event
207 EventIDBase::UNDEFNUM, // timestamp
208 EventIDBase::UNDEFNUM, // timestamp ns
209 lbn));
211 Atlas::ExtendedEventContext( evtStore()->hiveProxyDict(),
212 conditionsRun) );
213 return getRandomEngine(streamName, randomSeedOffset, ctx);
214}
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 118 of file EvtInclusiveDecay.cxx.

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

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

◆ isDefaultB()

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

Definition at line 558 of file EvtInclusiveDecay.cxx.

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

◆ isReEntrant()

virtual bool AthAlgorithm::isReEntrant ( ) const
inlinefinaloverrideprotectedvirtualinherited

Legacy algorithms are not thread-safe.

Definition at line 111 of file AthAlgorithm.h.

111{ return false; }

◆ isToBeDecayed()

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

Definition at line 489 of file EvtInclusiveDecay.cxx.

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

◆ particleData()

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

Access an element in the particle data table.

Definition at line 124 of file GenBase.h.

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

◆ particleTable()

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

Get a particle data table.

Definition at line 116 of file GenBase.h.

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

◆ partPropSvc()

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

Access the particle property service.

Definition at line 111 of file GenBase.h.

111 {
112 return m_ppSvc;
113 }

◆ passesUserSelection()

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

Definition at line 580 of file EvtInclusiveDecay.cxx.

580 {
581 bool passed(false);
582 std::vector<HepMC::GenParticlePtr> *muons = new std::vector<HepMC::GenParticlePtr>;
583
584 for ( const auto& p: *hepMC) {
585 if( std::abs(p->pdg_id()) == 13 )
586 muons->push_back(p);
587 }
588
589 for (auto muItr1 = muons->begin(); muItr1 != muons->end(); ++muItr1) {
590 for (auto muItr2 = muItr1+1; muItr2 != muons->end(); ++muItr2) {
591 if( m_userSelRequireOppositeSignedMu && (*muItr1)->pdg_id() * (*muItr2)->pdg_id() > 0)
592 continue;
593 if( !( (*muItr1)->momentum().perp() > m_userSelMu1MinPt && std::abs((*muItr1)->momentum().pseudoRapidity()) < m_userSelMu1MaxEta &&
594 (*muItr2)->momentum().perp() > m_userSelMu2MinPt && std::abs((*muItr2)->momentum().pseudoRapidity()) < m_userSelMu2MaxEta ) &&
595 !( (*muItr2)->momentum().perp() > m_userSelMu1MinPt && std::abs((*muItr2)->momentum().pseudoRapidity()) < m_userSelMu1MaxEta &&
596 (*muItr1)->momentum().perp() > m_userSelMu2MinPt && std::abs((*muItr1)->momentum().pseudoRapidity()) < m_userSelMu2MaxEta ) )
597 continue;
598 double dimuMass = invMass((*muItr1),(*muItr2));
599 if( !( dimuMass > m_userSelMinDimuMass && (dimuMass < m_userSelMaxDimuMass || m_userSelMaxDimuMass < 0.) ) )
600 continue;
601 passed = true;
602 }
603 }
604
605 delete muons;
606
607 return passed;
608}
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 681 of file EvtInclusiveDecay.cxx.

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

◆ pdt()

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

Shorter alias to get a particle data table.

Definition at line 121 of file GenBase.h.

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

◆ printHepMC()

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

Definition at line 634 of file EvtInclusiveDecay.cxx.

634 {
635 std::set<HepMC::GenVertexPtr> visited;
636 unsigned int nParticlesFound = 0;
637 unsigned int nTreesFound = 0;
638 for (auto p: *hepMC) {
639 if ( (!p->production_vertex()) ||
640 (p->production_vertex()->particles_in_size() == 0) ) {
641 nTreesFound++;
642 std::cout << "\n Found new partial decay tree:\n" << std::endl;
643 unsigned int nParticlesVisited = printTree(std::move(p),visited,1,particleSet);
644 std::cout << "\n " << nParticlesVisited << " particles in this subtree" << std::endl;
645 nParticlesFound += nParticlesVisited;
646 }
647 }
648 std::cout << "\n Total of " << nParticlesFound << " particles found in "
649 << nTreesFound << " decay subtrees in HepMC event record\n" << std::endl;
650}
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 652 of file EvtInclusiveDecay.cxx.

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

◆ removeDecayTree()

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

Definition at line 386 of file EvtInclusiveDecay.cxx.

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

◆ 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 182 of file EvtInclusiveDecay.cxx.

184{
185 long seeds[7];
186 ATHRNG::calculateSeedsMC21(seeds, streamName, ctx.eventID().event_number(), m_dsid, m_randomSeed);
187 m_evtAtRndmGen->getEngine()->setSeeds(seeds, 0); // NOT THREAD-SAFE
188}
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() [1/2]

void AthAlgorithm::setFilterPassed ( bool state) const
inherited

Definition at line 102 of file AthAlgorithm.cxx.

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

◆ setFilterPassed() [2/2]

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

Definition at line 106 of file AthAlgorithm.cxx.

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

◆ sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )
overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

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

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

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

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

Definition at line 66 of file AthAlgorithm.cxx.

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

◆ sysStart()

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

Handle START transition.

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

◆ traverseDecayTree()

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

Definition at line 347 of file EvtInclusiveDecay.cxx.

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

720 {
721 return PathResolverFindCalibDirectory( "Pythia8/xmldoc" );
722}
std::string PathResolverFindCalibDirectory(const std::string &logical_file_name)

Member Data Documentation

◆ m_allowAllKnownDecays

bool EvtInclusiveDecay::m_allowAllKnownDecays
private

Definition at line 126 of file EvtInclusiveDecay.h.

◆ m_allowDefaultBDecays

bool EvtInclusiveDecay::m_allowDefaultBDecays
private

Definition at line 127 of file EvtInclusiveDecay.h.

◆ m_applyUserSelection

bool EvtInclusiveDecay::m_applyUserSelection
private

Definition at line 144 of file EvtInclusiveDecay.h.

◆ m_blackList

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

Definition at line 123 of file EvtInclusiveDecay.h.

◆ m_blackListSet

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

Definition at line 124 of file EvtInclusiveDecay.h.

◆ m_checkDecayChannels

bool EvtInclusiveDecay::m_checkDecayChannels
private

Definition at line 137 of file EvtInclusiveDecay.h.

◆ m_checkDecayTree

bool EvtInclusiveDecay::m_checkDecayTree
private

Definition at line 136 of file EvtInclusiveDecay.h.

◆ m_decayFile

std::string EvtInclusiveDecay::m_decayFile
private

Definition at line 112 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 99 of file EvtInclusiveDecay.h.

99{this, "Dsid", 999999};

◆ m_evtAtRndmGen

EvtInclusiveAtRndmGen* EvtInclusiveDecay::m_evtAtRndmGen {}
private

Definition at line 107 of file EvtInclusiveDecay.h.

107{};

◆ m_evtStore

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

Pointer to StoreGate (event store by default).

Definition at line 390 of file AthCommonDataStore.h.

◆ m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects
privateinherited

Definition at line 114 of file AthAlgorithm.h.

◆ m_inputKeyName

std::string EvtInclusiveDecay::m_inputKeyName
private

Definition at line 115 of file EvtInclusiveDecay.h.

◆ m_isfHerwig

bool EvtInclusiveDecay::m_isfHerwig
private

Definition at line 152 of file EvtInclusiveDecay.h.

◆ m_maxNRepeatedDecays

int EvtInclusiveDecay::m_maxNRepeatedDecays
private

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

134{};

◆ m_mcevents_const

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

Const handle to the MC event collection.

Definition at line 145 of file GenBase.h.

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

◆ m_mcEvtColl

McEventCollection* EvtInclusiveDecay::m_mcEvtColl {}
private

Definition at line 104 of file EvtInclusiveDecay.h.

104{};

◆ m_mkMcEvent

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

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

Definition at line 136 of file GenBase.h.

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

◆ m_myEvtGen

EvtGen* EvtInclusiveDecay::m_myEvtGen {}
private

Definition at line 108 of file EvtInclusiveDecay.h.

108{};

◆ m_noDecayChannels

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

Definition at line 138 of file EvtInclusiveDecay.h.

◆ m_nRepeatedDecays

int EvtInclusiveDecay::m_nRepeatedDecays
private

Definition at line 140 of file EvtInclusiveDecay.h.

◆ m_outputKeyName

std::string EvtInclusiveDecay::m_outputKeyName
private

Definition at line 116 of file EvtInclusiveDecay.h.

◆ m_pdtFile

std::string EvtInclusiveDecay::m_pdtFile
private

Definition at line 111 of file EvtInclusiveDecay.h.

◆ m_ppSvc

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

Handle on the particle property service.

Definition at line 142 of file GenBase.h.

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

◆ m_printHepMCAfterEvtGen

bool EvtInclusiveDecay::m_printHepMCAfterEvtGen
private

Definition at line 132 of file EvtInclusiveDecay.h.

◆ m_printHepMCBeforeEvtGen

bool EvtInclusiveDecay::m_printHepMCBeforeEvtGen
private

Definition at line 131 of file EvtInclusiveDecay.h.

◆ m_printHepMCHighlighted

bool EvtInclusiveDecay::m_printHepMCHighlighted
private

Definition at line 133 of file EvtInclusiveDecay.h.

◆ m_printHepMCHighLightTopLevelDecays

bool EvtInclusiveDecay::m_printHepMCHighLightTopLevelDecays
private

Definition at line 134 of file EvtInclusiveDecay.h.

◆ m_prohibitFinalStateDecay

bool EvtInclusiveDecay::m_prohibitFinalStateDecay
private

Definition at line 119 of file EvtInclusiveDecay.h.

◆ m_prohibitReDecay

bool EvtInclusiveDecay::m_prohibitReDecay
private

Definition at line 120 of file EvtInclusiveDecay.h.

◆ m_prohibitRemoveSelfDecay

bool EvtInclusiveDecay::m_prohibitRemoveSelfDecay
private

Definition at line 122 of file EvtInclusiveDecay.h.

◆ m_prohibitUnDecay

bool EvtInclusiveDecay::m_prohibitUnDecay
private

Definition at line 121 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 102 of file EvtInclusiveDecay.h.

102{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 114 of file EvtInclusiveDecay.h.

◆ m_readExisting

bool EvtInclusiveDecay::m_readExisting
private

Definition at line 118 of file EvtInclusiveDecay.h.

◆ m_rndmSvc

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

Definition at line 96 of file EvtInclusiveDecay.h.

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

◆ m_setVMtransversePol

bool EvtInclusiveDecay::m_setVMtransversePol
private

Definition at line 153 of file EvtInclusiveDecay.h.

◆ m_userDecayFile

std::string EvtInclusiveDecay::m_userDecayFile
private

Definition at line 113 of file EvtInclusiveDecay.h.

◆ m_userSelMaxDimuMass

double EvtInclusiveDecay::m_userSelMaxDimuMass
private

Definition at line 151 of file EvtInclusiveDecay.h.

◆ m_userSelMinDimuMass

double EvtInclusiveDecay::m_userSelMinDimuMass
private

Definition at line 150 of file EvtInclusiveDecay.h.

◆ m_userSelMu1MaxEta

double EvtInclusiveDecay::m_userSelMu1MaxEta
private

Definition at line 148 of file EvtInclusiveDecay.h.

◆ m_userSelMu1MinPt

double EvtInclusiveDecay::m_userSelMu1MinPt
private

Definition at line 146 of file EvtInclusiveDecay.h.

◆ m_userSelMu2MaxEta

double EvtInclusiveDecay::m_userSelMu2MaxEta
private

Definition at line 149 of file EvtInclusiveDecay.h.

◆ m_userSelMu2MinPt

double EvtInclusiveDecay::m_userSelMu2MinPt
private

Definition at line 147 of file EvtInclusiveDecay.h.

◆ m_userSelRequireOppositeSignedMu

bool EvtInclusiveDecay::m_userSelRequireOppositeSignedMu
private

Definition at line 145 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 128 of file EvtInclusiveDecay.h.

◆ m_whiteListSet

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

Definition at line 129 of file EvtInclusiveDecay.h.


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