1 |
loizides |
1.6 |
// $Id: MCParticle.cc,v 1.5 2009/02/18 15:38:55 loizides Exp $
|
2 |
bendavid |
1.1 |
|
3 |
|
|
#include "MitAna/DataTree/interface/MCParticle.h"
|
4 |
|
|
|
5 |
|
|
ClassImp(mithep::MCParticle)
|
6 |
|
|
|
7 |
|
|
using namespace mithep;
|
8 |
|
|
|
9 |
|
|
//--------------------------------------------------------------------------------------------------
|
10 |
loizides |
1.3 |
const MCParticle *MCParticle::FindDaughter(Int_t pid,
|
11 |
|
|
Bool_t checkCharge, const MCParticle *start) const
|
12 |
|
|
{
|
13 |
|
|
// Return daughter with given pid. If checkCharge is false then just the type of particle is
|
14 |
|
|
// checked (ie particle and anti-particle). If start is not null, start searching from
|
15 |
|
|
// this daughter.
|
16 |
|
|
|
17 |
|
|
UInt_t i = 0;
|
18 |
|
|
if (start) {
|
19 |
|
|
for (; i<NDaughters(); ++i) {
|
20 |
|
|
if (Daughter(i)==start) {
|
21 |
|
|
++i;
|
22 |
|
|
break;
|
23 |
|
|
}
|
24 |
|
|
}
|
25 |
|
|
return 0;
|
26 |
|
|
}
|
27 |
|
|
|
28 |
|
|
for (UInt_t j=i; j<NDaughters(); ++j) {
|
29 |
|
|
if (Daughter(j)->Is(pid,checkCharge))
|
30 |
|
|
return Daughter(j);
|
31 |
|
|
}
|
32 |
|
|
|
33 |
|
|
return 0;
|
34 |
|
|
}
|
35 |
|
|
|
36 |
|
|
//--------------------------------------------------------------------------------------------------
|
37 |
|
|
const MCParticle *MCParticle::FindMother(Int_t pid, Bool_t checkCharge) const
|
38 |
|
|
{
|
39 |
|
|
// Return mother with given pid. If checkCharge is false then just the type of particle is
|
40 |
|
|
// checked (ie particle and anti-particle).
|
41 |
|
|
|
42 |
|
|
const MCParticle *mother = Mother();
|
43 |
|
|
if (!mother)
|
44 |
|
|
return 0;
|
45 |
|
|
|
46 |
|
|
if (checkCharge) {
|
47 |
loizides |
1.4 |
while (mother && mother->PdgId()!=pid)
|
48 |
loizides |
1.3 |
mother = mother->Mother();
|
49 |
|
|
return mother;
|
50 |
|
|
}
|
51 |
|
|
|
52 |
|
|
Int_t apid = pid>0?pid:-pid;
|
53 |
loizides |
1.4 |
while (mother && mother->AbsPdgId()!=apid)
|
54 |
loizides |
1.3 |
mother = mother->Mother();
|
55 |
|
|
|
56 |
|
|
return mother;
|
57 |
|
|
}
|
58 |
|
|
|
59 |
|
|
//--------------------------------------------------------------------------------------------------
|
60 |
loizides |
1.2 |
void MCParticle::Print(Option_t *opt) const
|
61 |
bendavid |
1.1 |
{
|
62 |
loizides |
1.2 |
// Print particle kinematics. In case option "l" is given then also print info about daughters.
|
63 |
|
|
|
64 |
loizides |
1.6 |
printf("id=%5d st=%2d nd=%3d gen=%d sim=%d px=%.3f py=%.3f pz=%.3f e=%.3f\n",
|
65 |
|
|
PdgId(), Status(), NDaughters(), IsGenerated(), IsSimulated(), Px(), Py(), Pz(), E());
|
66 |
loizides |
1.2 |
|
67 |
|
|
if (opt && opt[0]=='l') {
|
68 |
|
|
for (UInt_t i=0; i<NDaughters(); ++i) {
|
69 |
loizides |
1.6 |
printf(" %2d -> id=%5d st=%2d gen=%d sim=%d px=%.3f py=%.3f pz=%.3f e=%.3f\n",
|
70 |
|
|
i, Daughter(i)->PdgId(), Daughter(i)->Status(),
|
71 |
|
|
Daughter(i)->IsGenerated(), Daughter(i)->IsSimulated(),
|
72 |
loizides |
1.2 |
Daughter(i)->Px(), Daughter(i)->Py(), Daughter(i)->Pz(), Daughter(i)->E());
|
73 |
|
|
}
|
74 |
|
|
}
|
75 |
bendavid |
1.1 |
}
|