ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitAna/DataTree/interface/TrackingParticle.h
Revision: 1.3
Committed: Mon Jan 18 14:33:02 2010 UTC (15 years, 3 months ago) by bendavid
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_025c_branch2, Mit_025c_branch1, Mit_025c_branch0, Mit_025d, Mit_025c, Mit_025b, Mit_025a, Mit_025, Mit_025pre2, Mit_024b, Mit_025pre1, Mit_024a, Mit_024, Mit_023, Mit_022a, Mit_022, Mit_020d, TMit_020d, Mit_020c, Mit_021, Mit_021pre2, Mit_021pre1, Mit_020b, Mit_020a, Mit_020, Mit_020pre1, Mit_018, Mit_017, Mit_017pre3, Mit_017pre2, Mit_017pre1, Mit_016, Mit_015b, Mit_015a, Mit_015, Mit_014e, Mit_014d, Mit_014c, Mit_014b, Mit_014a, Mit_014, Mit_014pre3, Mit_014pre2, Mit_014pre1, Mit_013d, Mit_013c, Mit_013b, Mit_013a, Mit_013, Mit_013pre1, Mit_012i
Branch point for: Mit_025c_branch
Changes since 1.2: +2 -2 lines
Log Message: 
Add accessor for HasObject

File Contents

# User Rev Content
1 bendavid 1.1 //--------------------------------------------------------------------------------------------------
2 bendavid 1.3 // $Id: TrackingParticle.h,v 1.2 2010/01/07 17:08:40 loizides Exp $
3 bendavid 1.1 //
4     // TrackingParticle
5     //
6     // Stores additional MC information for tracking truth.
7     //
8     // Authors: J.Bendavid
9     //--------------------------------------------------------------------------------------------------
10    
11     #ifndef MITANA_DATATREE_TRACKINGPARTICLE_H
12     #define MITANA_DATATREE_TRACKINGPARTICLE_H
13    
14     #include "MitAna/DataCont/interface/Ref.h"
15     #include "MitAna/DataTree/interface/Particle.h"
16     #include "MitAna/DataTree/interface/MCParticle.h"
17     #include "MitAna/DataTree/interface/Track.h"
18     #include "MitAna/DataCont/interface/RefArray.h"
19    
20    
21     namespace mithep
22     {
23     class TrackingParticle : public Particle
24     {
25     public:
26    
27     TrackingParticle() {}
28    
29 loizides 1.2 void AddMCPart(const MCParticle *p)
30     { fMCParts.Add(p); ClearCharge(); ClearMom(); }
31     const MCParticle *DistinctMother() const;
32     const MCParticle *FinalMCPart() const;
33 bendavid 1.3 Bool_t HasMCPart(const MCParticle *p) const { return fMCParts.HasObject(p); }
34 loizides 1.2 Bool_t Hit(Track::EHitLayer l) const { return fHits.TestBit(l); }
35     const BitMask48 &Hits() const { return fHits; }
36     const MCParticle *InitialMCPart() const;
37     const MCParticle *MCPart(UInt_t i) const { return fMCParts.At(i); }
38     UInt_t NMCParts() const { return fMCParts.Entries(); }
39     UInt_t NHits() const { return fHits.NBitsSet(); }
40     EObjType ObjType() const { return kTrackingParticle; }
41     void SetHit(Track::EHitLayer l) { fHits.SetBit(l); }
42     void SetHits(const BitMask48 &hits) { fHits = hits; }
43 bendavid 1.1
44     protected:
45 loizides 1.2 Double_t GetCharge() const;
46     void GetMom() const;
47 bendavid 1.1 BitMask48 fHits; //storage for sim hit information
48     RefArray<MCParticle> fMCParts; //reference to corresponding MCParticles
49    
50     ClassDef(TrackingParticle,1) // Generated particle class
51     };
52     }
53    
54     //--------------------------------------------------------------------------------------------------
55     inline const mithep::MCParticle *mithep::TrackingParticle::DistinctMother() const
56     {
57     // Return mother, walking up the tree until a particle with a different pdg from this one
58     // is found.
59    
60     const mithep::MCParticle *mcPart = FinalMCPart();
61    
62     if (mcPart)
63     return mcPart->DistinctMother();
64    
65     return 0;
66    
67     }
68    
69     //--------------------------------------------------------------------------------------------------
70     inline const mithep::MCParticle *mithep::TrackingParticle::FinalMCPart() const
71     {
72     // Return last MCParticle in the chain
73    
74     if (NMCParts()) {
75     return MCPart(NMCParts() - 1);
76     }
77     else {
78     return 0;
79     }
80    
81     }
82    
83     //--------------------------------------------------------------------------------------------------
84     inline const mithep::MCParticle *mithep::TrackingParticle::InitialMCPart() const
85     {
86     // Return first MCParticle in the chain
87    
88     if (NMCParts()) {
89     return MCPart(0);
90     }
91     else {
92     return 0;
93     }
94    
95     }
96    
97     //--------------------------------------------------------------------------------------------------
98     inline Double_t mithep::TrackingParticle::GetCharge() const
99     {
100     // Get charge from first MC particle.
101    
102     const MCParticle *firstMCPart = InitialMCPart();
103    
104     if (firstMCPart) {
105     return firstMCPart->Charge();
106     }
107     else {
108     return 0.0;
109     }
110    
111     }
112    
113     //--------------------------------------------------------------------------------------------------
114     inline void mithep::TrackingParticle::GetMom() const
115     {
116     // Get momentum from first MC particle.
117    
118     const MCParticle *firstMCPart = InitialMCPart();
119    
120     if (firstMCPart) {
121     fCachedMom = firstMCPart->Mom();
122     }
123     else {
124     fCachedMom = FourVectorM();
125     }
126    
127     }
128    
129     #endif