Coding rule violations corrected
[u/mrichter/AliRoot.git] / STEER / AliAODMCParticle.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16
17 //-------------------------------------------------------------------------
18 //     Realisation of AliVParticle for MC Particles
19 //     Basically a stripped down AliMCParicle / TParticle
20 //     with minimum information on MC tracks
21 //     Author: Christian Klein-Bösing, CERN
22 //-------------------------------------------------------------------------
23
24
25 #include "AliAODMCParticle.h"
26 #include "AliAODEvent.h"
27
28 #include "TDatabasePDG.h"
29 #include "TParticle.h"
30 #include "TClonesArray.h"
31
32
33 ClassImp(AliAODMCParticle)
34
35 TString AliAODMCParticle::fgkStdBranchName("mcparticles");
36
37 AliAODMCParticle::AliAODMCParticle():
38 AliVParticle(),
39   fPdgCode(0),
40   fFlag(0),
41   fLabel(0),
42   fMother(0),
43   fPx(0),
44   fPy(0),
45   fPz(0),
46   fE(0),
47   fVx(0),
48   fVy(0),
49   fVz(0),
50   fVt(0)
51 {
52   // Default Constructor
53   fDaughter[0] =   fDaughter[1] = 0;
54 }
55
56     
57 AliAODMCParticle::AliAODMCParticle(const AliMCParticle* mcpart, Int_t label,Int_t flag):
58     AliVParticle(),
59     fPdgCode(mcpart->Particle()->GetPdgCode()),
60     fFlag(flag),
61     fLabel(label),
62     fMother(mcpart->GetMother()),
63     fPx(mcpart->Particle()->Px()),
64     fPy(mcpart->Particle()->Py()),
65     fPz(mcpart->Particle()->Pz()),
66     fE(mcpart->Particle()->Energy()),
67     fVx(mcpart->Particle()->Vx()),
68     fVy(mcpart->Particle()->Vy()),
69     fVz(mcpart->Particle()->Vz()),
70     fVt(mcpart->Particle()->T())
71 {
72     // Constructor
73   fDaughter[0] =  mcpart->GetFirstDaughter(); 
74   fDaughter[1] =  mcpart->GetLastDaughter();
75 }
76     
77     
78 AliAODMCParticle::AliAODMCParticle(const AliAODMCParticle& mcPart) :
79     AliVParticle(mcPart),
80     fPdgCode(mcPart.fPdgCode),
81     fFlag(mcPart.fFlag),
82     fLabel(mcPart.fLabel),
83     fMother(mcPart.fMother),
84     fPx(mcPart.fPx),
85     fPy(mcPart.fPy),
86     fPz(mcPart.fPz),
87     fE(mcPart.fE),
88     fVx(mcPart.fVx),
89     fVy(mcPart.fVy),
90     fVz(mcPart.fVz),
91     fVt(mcPart.fVt)
92 {
93   // Copy constructor
94   fDaughter[0] = mcPart.fDaughter[0]; 
95   fDaughter[1] = mcPart.fDaughter[1]; 
96 }
97
98 AliAODMCParticle& AliAODMCParticle::operator=(const AliAODMCParticle& mcPart)
99
100
101   if (this!=&mcPart) { 
102     AliVParticle::operator=(mcPart);
103     fPdgCode    = mcPart.fPdgCode;
104     fFlag       = mcPart.fFlag;
105     fLabel      = mcPart.fLabel;
106     fMother     = mcPart.fMother;
107     fPx         = mcPart.fPx;
108     fPy         = mcPart.fPy;
109     fPz         = mcPart.fPz;
110     fE          = mcPart.fE;
111     fVx         = mcPart.fVx;
112     fVy         = mcPart.fVy;
113     fVz         = mcPart.fVz;
114     fVt         = mcPart.fVt;
115     fDaughter[0] = mcPart.fDaughter[0]; 
116     fDaughter[1] = mcPart.fDaughter[1]; 
117   }  
118   
119   return *this;
120
121 }
122
123 Double_t AliAODMCParticle::M()         const
124 {
125     TParticlePDG* pdg =  TDatabasePDG::Instance()->GetParticle(fPdgCode);
126     if (pdg) {
127         return (pdg->Mass());
128     } else {
129         return GetCalcMass();
130     }
131 }
132
133
134 Short_t AliAODMCParticle::Charge()     const
135 {
136     TParticlePDG* pdg =  TDatabasePDG::Instance()->GetParticle(fPdgCode);
137     if (pdg) {
138         return (Short_t (pdg->Charge()));
139     } else {
140         return -99;
141     }
142 }
143
144 void AliAODMCParticle::Print(const Option_t */*opt*/) const {
145 // Print particle information
146   if(TDatabasePDG::Instance()->GetParticle(fPdgCode)){
147     Printf(">>> PDG (%d) : %s",fPdgCode,TDatabasePDG::Instance()->GetParticle(fPdgCode)->GetName());
148   }
149   else{
150     Printf(">>> PDG (%d) : %s",fPdgCode,"Unknown");
151   }
152   Printf(">>  P(%3.3f,%3.3f,%3.3f) V((%3.3f,%3.3f,%3.3f,%3.3f)",fPx,fPy,fPz,fVx,fVy,fVz,fVt);  
153   Printf(">   Mother %d, First Daughter %d Last Daughter %d , Status %d, PhysicalPrimary %d",
154          fMother,fDaughter[0],fDaughter[1],GetStatus(),
155          IsPhysicalPrimary());
156 }