ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/Morgan/interface/TRootParticle.h
(Generate patch)

Comparing UserCode/Morgan/interface/TRootParticle.h (file contents):
Revision 1.2 by mlethuil, Wed Jul 2 16:22:18 2008 UTC vs.
Revision 1.7 by lethuill, Thu Dec 18 18:01:49 2008 UTC

# Line 7 | Line 7
7   #include "Rtypes.h"
8   #include "TLorentzVector.h"
9   #include "TVector3.h"
10 + #include "TRef.h"
11  
12   using namespace std;
13  
# Line 15 | Line 16 | class TRootParticle : public TLorentzVec
16  
17   public:
18          
19 <        TRootParticle() : TLorentzVector(), vertex_(), type_(0), charge_(0.) {;}
20 <        TRootParticle(const TRootParticle& particle) : TLorentzVector(particle), vertex_(particle.vertex_), type_(particle.type_), charge_(particle.charge_) {;}
21 <        TRootParticle(Double_t px, Double_t py, Double_t pz, Double_t e) : TLorentzVector(px,py,pz,e), vertex_(), type_(0), charge_(0.) {;}
22 <        TRootParticle(Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vtx_x, Double_t vtx_y, Double_t vtx_z) : TLorentzVector(px,py,pz,e), vertex_(vtx_x,vtx_y,vtx_z), type_(0), charge_(0.) {;}
23 <        TRootParticle(Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vtx_x, Double_t vtx_y, Double_t vtx_z, Int_t type, Float_t charge) : TLorentzVector(px,py,pz,e), vertex_(vtx_x,vtx_y,vtx_z), type_(type), charge_(charge) {;}
24 <        TRootParticle(const TLorentzVector &momentum) : TLorentzVector(momentum), vertex_(), type_(0), charge_(0.) {;}
25 <        TRootParticle(const TLorentzVector &momentum, const TVector3 &vertex, Int_t type, Float_t charge) : TLorentzVector(momentum), vertex_(vertex), type_(type), charge_(charge) {;}
19 >        TRootParticle() :
20 >                TLorentzVector()
21 >                ,vertex_()
22 >                ,type_(0)
23 >                ,charge_(0.)
24 >                ,genParticleIndex_(-1)
25 >                ,genParticle_(0)
26 >                {;}
27 >
28 >        TRootParticle(const TRootParticle& particle) :
29 >                TLorentzVector(particle)
30 >                ,vertex_(particle.vertex_)
31 >                ,type_(particle.type_)
32 >                ,charge_(particle.charge_)
33 >                ,genParticleIndex_(particle.genParticleIndex_)
34 >                ,genParticle_(particle.genParticle_)
35 >                {;}
36 >
37 >        TRootParticle(Double_t px, Double_t py, Double_t pz, Double_t e) :
38 >                TLorentzVector(px,py,pz,e)
39 >                ,vertex_()
40 >                ,type_(0)
41 >                ,charge_(0.)
42 >                ,genParticleIndex_(-1)
43 >                ,genParticle_(0)
44 >                {;}
45 >
46 >        TRootParticle(Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vtx_x, Double_t vtx_y, Double_t vtx_z) :
47 >                TLorentzVector(px,py,pz,e)
48 >                ,vertex_(vtx_x,vtx_y,vtx_z)
49 >                ,type_(0)
50 >                ,charge_(0.)
51 >                ,genParticleIndex_(-1)
52 >                ,genParticle_(0)
53 >                {;}
54 >
55 >        TRootParticle(Double_t px, Double_t py, Double_t pz, Double_t e, Double_t vtx_x, Double_t vtx_y, Double_t vtx_z,Int_t type, Float_t charge) :
56 >                TLorentzVector(px,py,pz,e)
57 >                ,vertex_(vtx_x,vtx_y,vtx_z)
58 >                ,type_(type)
59 >                ,charge_(charge)
60 >                ,genParticleIndex_(-1)
61 >                ,genParticle_(0)
62 >                {;}
63 >
64 >
65 >        TRootParticle(const TLorentzVector &momentum) :
66 >                TLorentzVector(momentum)
67 >                ,vertex_()
68 >                ,type_(0)
69 >                ,charge_(0.)
70 >                ,genParticleIndex_(-1)
71 >                ,genParticle_(0)
72 >                {;}
73 >
74 >        TRootParticle(const TLorentzVector &momentum, const TVector3 &vertex) :
75 >                TLorentzVector(momentum)
76 >                ,vertex_(vertex)
77 >                ,type_(0)
78 >                ,charge_(0.)
79 >                ,genParticleIndex_(-1)
80 >                ,genParticle_(0)
81 >                {;}
82 >
83 >        TRootParticle(const TLorentzVector &momentum, const TVector3 &vertex, Int_t type, Float_t charge) :
84 >                TLorentzVector(momentum)
85 >                ,vertex_(vertex)
86 >                ,type_(type)
87 >                ,charge_(charge)
88 >                ,genParticleIndex_(-1)
89 >                ,genParticle_(0)
90 >                {;}
91 >
92          ~TRootParticle() {;}
93  
94 +
95          Double_t vx() const  { return vertex_.x(); }
96          Double_t vy() const  { return vertex_.y(); }
97          Double_t vz() const  { return vertex_.z(); }
98          Int_t type() const  { return type_; }
99          Float_t charge() const  { return charge_; }
100 +        Int_t genParticleIndex() const { return genParticleIndex_; }
101 +        TRootParticle* genParticle() const { return (TRootParticle*) genParticle_.GetObject() ;}
102 +        virtual TString typeName() const { return "TRootParticle"; }
103 +
104  
105 <        // FIXME setVx, setVy and setVz must modify the TLorentzVector
105 >        // FIXME setVx, setVy and setVz must modify the TLorentzVector ?
106          void setVx(Double_t vx) { vertex_.SetX(vx); }
107          void setVy(Double_t vy) { vertex_.SetY(vy); }
108          void setVz(Double_t vz) { vertex_.SetZ(vz); }
109          void setType(Int_t type) { type_ = type; }
110          void setCharge(Int_t charge) { charge_ = charge; }
111 +        void setGenParticleIndex(Int_t genParticleIndex) { genParticleIndex_ = genParticleIndex; }
112 +        void setGenParticle(TObject* genParticle) { genParticle_ = genParticle; }
113  
114 <        friend std::ostream& operator<< (std::ostream& stream, const TRootParticle& part) {
114 >        friend std::ostream& operator<< (std::ostream& stream, const TRootParticle& part)
115 >        {
116                  stream << "Type=" << part.type_ << "  Charge=" << part.charge_ << " (Et,eta,phi)=("<< part.Et() <<","<< part.Eta() <<","<< part.Phi() << ")"
117 <                                << " vertex(x,y,z)=("<< part.vx() <<","<< part.vy() <<","<< part.vz() << ")";
118 <                                return stream;
117 >                        << " vertex(x,y,z)=("<< part.vx() <<","<< part.vy() <<","<< part.vz() << ")";
118 >                return stream;
119          };
120 <        
121 <                
120 >
121 >
122 >
123   protected:
124          
125          TVector3 vertex_;
126          Int_t type_;
127          Float_t charge_;
128 +        Int_t genParticleIndex_;
129 +        TRef genParticle_;
130  
131          ClassDef (TRootParticle,1);
132   };
133  
134   #endif
135 +
136 +

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines