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

Comparing UserCode/MitAna/DataTree/interface/Electron.h (file contents):
Revision 1.13 by sixie, Fri Aug 8 11:17:13 2008 UTC vs.
Revision 1.28 by bendavid, Tue Mar 3 18:03:55 2009 UTC

# Line 3 | Line 3
3   //
4   // Electron
5   //
6 < // Details to be worked out...
6 > // Details to be worked out... TODO: Needs description ala Muon class
7   //
8   // Authors: C.Loizides, J.Bendavid, S.Xie
9   //--------------------------------------------------------------------------------------------------
10  
11 < #ifndef DATATREE_ELECTRON_H
12 < #define DATATREE_ELECTRON_H
11 > #ifndef MITANA_DATATREE_ELECTRON_H
12 > #define MITANA_DATATREE_ELECTRON_H
13  
14   #include "MitAna/DataTree/interface/SuperCluster.h"
15   #include "MitAna/DataTree/interface/ChargedParticle.h"
16 + #include "MitAna/DataCont/interface/Ref.h"
17  
18   namespace mithep
19   {
20    class Electron : public ChargedParticle
21    {
22      public:
23 <      Electron() {}
24 <      ~Electron() {}
23 >      Electron() : fESuperClusterOverP(0), fESeedClusterOverPout(0), fDeltaEtaSuperClTrkAtVtx(0),
24 >                   fDeltaEtaSeedClTrkAtCalo(0), fDeltaPhiSuperClTrkAtVtx(0),
25 >                   fDeltaPhiSeedClTrkAtCalo(0), fHadronicOverEm(0), fIsEnergyScaleCorrected(0),
26 >                   fIsMomentumCorrected(0), fNumberOfClusters(0), fClassification(0), fE33(0),
27 >                   fE55(0), fCovEtaEta(0), fCoviEtaiEta(0), fCovEtaPhi(0), fCovPhiPhi(0),
28 >                   fCaloIsolation(0), fCaloTowerIsolation(0), fTrackIsolation(0),
29 >                   fEcalJurassicIsolation(0), fHcalJurassicIsolation(0), fPassLooseID(0),
30 >                   fPassTightID(0), fIDLikelihood(0), fPIn(0), fPOut(0) {}
31        
32 <      const Track         *BestTrk()         const;
33 <      const Track         *GsfTrk()          const;
34 <      const Track         *TrackerTrk()      const;
35 <      const SuperCluster  *SCluster()        const;
36 <      const Track         *Trk()             const { return BestTrk(); }
37 <
38 <      Double_t    Mass()                           const { return 0.51099892e-3;            }
39 <      Double_t    ESuperClusterOverP()             const { return fESuperClusterOverP;      }
40 <      Double_t    ESeedClusterOverPout()           const { return fESeedClusterOverPout;    }
41 <      Double_t    DeltaEtaSuperClusterTrackAtVtx() const { return fDeltaEtaSuperClTrkAtVtx; }
42 <      Double_t    DeltaEtaSeedClusterTrackAtCalo() const { return fDeltaEtaSeedClTrkAtCalo; }
43 <      Double_t    DeltaPhiSuperClusterTrackAtVtx() const { return fDeltaPhiSuperClTrkAtVtx; }
44 <      Double_t    DeltaPhiSeedClusterTrackAtCalo() const { return fDeltaPhiSeedClTrkAtCalo; }
45 <      Double_t    HadronicOverEm()                 const { return fHadronicOverEm;          }
46 <      Double_t    IsEnergyScaleCorrected()         const { return fIsEnergyScaleCorrected;  }
47 <      Double_t    IsMomentumCorrected()            const { return fIsMomentumCorrected;     }
48 <      Double_t    NumberOfClusters()               const { return fNumberOfClusters;        }
49 <      Double_t    Classification()                 const { return fClassification;          }
50 <      Double_t    SuperClusterPx()                 const { return fSuperClusterPx;          }
51 <      Double_t    SuperClusterPy()                 const { return fSuperClusterPy;          }
52 <      Double_t    SuperClusterPz()                 const { return fSuperClusterPz;          }
53 <      Double_t    SuperClusterE()                  const { return fSuperClusterE;           }
54 <      Double_t    E33()                            const { return fE33;                     }
55 <      Double_t    E55()                            const { return fE55;                     }
56 <      Double_t    CovEtaEta()                      const { return fCovEtaEta;               }
57 <      Double_t    CovEtaPhi()                      const { return fCovEtaPhi;               }
58 <      Double_t    CovPhiPhi()                      const { return fCovPhiPhi;               }
59 <      Double_t    CaloIsolation()                  const { return fCaloIsolation;           }
60 <      Double_t    TrackIsolation()                 const { return fTrackIsolation;          }
61 <      
62 <      Double_t    ComputeTrackIsolation   (  Double_t extRadius, Double_t intRadius,
63 <                                             Double_t ptLow, Double_t maxVtxZDist,
64 <                                             mithep::Collection<Track> *tracks              );
65 <      Double_t    ComputeEcalIsolation    (  Double_t coneSize, Double_t etLow,
66 <                                             mithep::Collection<BasicCluster> *basicClusters);
67 <
68 <      void        SetGsfTrk(Track* t)                           { fGsfTrackRef = t;                }
69 <      void        SetTrackerTrk(Track* t)                       { fTrackerTrackRef = t;            }
70 <      void        SetSuperCluster(SuperCluster* sc)             { fSuperClusterRef = sc;           }
71 <      void        SetESuperClusterOverP(Double_t x)             { fESuperClusterOverP = x;         }
72 <      void        SetESeedClusterOverPout(Double_t x)           { fESeedClusterOverPout = x;       }
73 <      void        SetDeltaEtaSuperClusterTrackAtVtx(Double_t x) { fDeltaEtaSuperClTrkAtVtx = x;    }
74 <      void        SetDeltaEtaSeedClusterTrackAtCalo(Double_t x) { fDeltaEtaSeedClTrkAtCalo = x;    }
75 <      void        SetDeltaPhiSuperClusterTrackAtVtx(Double_t x) { fDeltaPhiSuperClTrkAtVtx = x;    }
76 <      void        SetDeltaPhiSeedClusterTrackAtCalo(Double_t x) { fDeltaPhiSeedClTrkAtCalo = x;    }
77 <      void        SetHadronicOverEm(Double_t x)                 { fHadronicOverEm = x;             }
78 <      void        SetIsEnergyScaleCorrected(Double_t x)         { fIsEnergyScaleCorrected = x;     }
79 <      void        SetIsMomentumCorrected(Double_t x)            { fIsMomentumCorrected = x;        }
80 <      void        SetNumberOfClusters(Double_t x)               { fNumberOfClusters = x;           }
81 <      void        SetClassification(Double_t x)                 { fClassification = x;             }
82 <      void        SetSuperClusterPx(Double_t SuperClusterPx)    { fSuperClusterPx = SuperClusterPx;}
83 <      void        SetSuperClusterPy(Double_t SuperClusterPy)    { fSuperClusterPy = SuperClusterPy;}
84 <      void        SetSuperClusterPz(Double_t SuperClusterPz)    { fSuperClusterPz = SuperClusterPz;}
85 <      void        SetSuperClusterE(Double_t SuperClusterE)      { fSuperClusterE = SuperClusterE;  }
86 <      void        SetE33(Double_t E33)                          { fE33 = E33;                      }
87 <      void        SetE55(Double_t E55)                          { fE55 = E55;                      }
88 <      void        SetCovEtaEta(Double_t CovEtaEta)              { fCovEtaEta = CovEtaEta;          }
89 <      void        SetCovEtaPhi(Double_t CovEtaPhi)              { fCovEtaPhi = CovEtaPhi;          }
90 <      void        SetCovPhiPhi(Double_t CovPhiPhi)              { fCovPhiPhi = CovPhiPhi;          }
91 <      void        SetCaloIsolation(Double_t CaloIsolation)      { fCaloIsolation = CaloIsolation;  }
92 <      void        SetTrackIsolation(Double_t TrackIsolation)    { fTrackIsolation = TrackIsolation;}
93 <
94 <
32 >      const Track         *BestTrk()               const;
33 >      const Track         *GsfTrk()                const { return fGsfTrackRef.Obj();          }
34 >      const Track         *TrackerTrk()            const { return fTrackerTrackRef.Obj();      }
35 >      const SuperCluster  *SCluster()              const { return fSuperClusterRef.Obj();      }
36 >      const Track         *Trk()                   const { return BestTrk();                   }
37 >      Double_t             CaloIsolation()         const { return fCaloIsolation;              }
38 >      Double_t             CaloTowerIsolation()    const { return fCaloTowerIsolation;         }
39 >      Int_t                Classification()        const { return fClassification;             }
40 >      Double_t             CovEtaEta()             const { return fCovEtaEta;                  }
41 >      Double_t             CovEtaPhi()             const { return fCovEtaPhi;                  }
42 >      Double_t             CovPhiPhi()             const { return fCovPhiPhi;                  }
43 >      Double_t             CoviEtaiEta()           const { return fCoviEtaiEta;                }
44 >      Double_t             DeltaEtaSuperClusterTrackAtVtx() const
45 >                             { return fDeltaEtaSuperClTrkAtVtx; }
46 >      Double_t             DeltaEtaSeedClusterTrackAtCalo() const
47 >                             { return fDeltaEtaSeedClTrkAtCalo; }
48 >      Double_t             DeltaPhiSuperClusterTrackAtVtx() const
49 >                             { return fDeltaPhiSuperClTrkAtVtx; }
50 >      Double_t             DeltaPhiSeedClusterTrackAtCalo() const
51 >                             { return fDeltaPhiSeedClTrkAtCalo; }
52 >      Double_t             E33()                   const { return fE33;                        }
53 >      Double_t             E55()                   const { return fE55;                        }
54 >      Double_t             EcalJurassicIsolation() const { return fEcalJurassicIsolation;      }
55 >      Double_t             ESuperClusterOverP()    const { return fESuperClusterOverP;         }
56 >      Double_t             ESeedClusterOverPout()  const { return fESeedClusterOverPout;       }
57 >      Double_t             ESeedClusterOverPIn()   const;
58 >      Double_t             IDLikelihood()          const { return fIDLikelihood;               }
59 >      Double_t             IsEnergyScaleCorrected()const { return fIsEnergyScaleCorrected;     }
60 >      Double_t             IsMomentumCorrected()   const { return fIsMomentumCorrected;        }
61 >      Double_t             HadronicOverEm()        const { return fHadronicOverEm;             }  
62 >      Bool_t               HasGsfTrk()             const { return fGsfTrackRef.IsValid();      }
63 >      Bool_t               HasTrackerTrk()         const { return fTrackerTrackRef.IsValid();  }
64 >      Bool_t               HasSuperCluster()       const { return fSuperClusterRef.IsValid();  }
65 >      Double_t             HcalIsolation()         const { return fHcalJurassicIsolation;      }
66 >      Double_t             NumberOfClusters()      const { return fNumberOfClusters;           }
67 >      EObjType             ObjType()               const { return kElectron;                   }
68 >      Double_t             PassLooseID()           const { return fPassLooseID;                }
69 >      Double_t             PassTightID()           const { return fPassTightID;                }
70 >      Double_t             PIn()                   const { return fPIn;                        }
71 >      Double_t             POut()                  const { return fPOut;                       }
72 >      Double_t             TrackIsolation()        const { return fTrackIsolation;             }
73 >      void                 SetGsfTrk(const Track* t)                    
74 >                             { fGsfTrackRef = t;     ClearMom(); ClearCharge(); }
75 >      void                 SetTrackerTrk(const Track* t)                
76 >                             { fTrackerTrackRef = t; ClearMom(); ClearCharge(); }
77 >      void                 SetSuperCluster(const SuperCluster* sc)      
78 >                             { fSuperClusterRef = sc; ClearMom(); }
79 >      void                 SetCaloIsolation(Double_t caloiso)    { fCaloIsolation = caloiso;   }
80 >      void                 SetCaloTowerIsolation(Double_t tiso)  { fCaloTowerIsolation = tiso; }
81 >      void                 SetClassification(Int_t x)            { fClassification = x;        }
82 >      void                 SetCovEtaEta(Double_t CovEtaEta)      { fCovEtaEta = CovEtaEta;     }
83 >      void                 SetCovEtaPhi(Double_t CovEtaPhi)      { fCovEtaPhi = CovEtaPhi;     }
84 >      void                 SetCovPhiPhi(Double_t CovPhiPhi)      { fCovPhiPhi = CovPhiPhi;     }
85 >      void                 SetCoviEtaiEta(Double_t CoviEtaiEta)  { fCoviEtaiEta = CoviEtaiEta; }
86 >      void                 SetDeltaEtaSuperClusterTrackAtVtx(Double_t x)  
87 >                             { fDeltaEtaSuperClTrkAtVtx = x;   }
88 >      void                 SetDeltaEtaSeedClusterTrackAtCalo(Double_t x)  
89 >                             { fDeltaEtaSeedClTrkAtCalo = x;   }
90 >      void                 SetDeltaPhiSuperClusterTrackAtVtx(Double_t x)  
91 >                             { fDeltaPhiSuperClTrkAtVtx = x;   }
92 >      void                 SetDeltaPhiSeedClusterTrackAtCalo(Double_t x)  
93 >                             { fDeltaPhiSeedClTrkAtCalo = x;   }
94 >      void                 SetE33(Double_t E33)                  { fE33 = E33;                     }
95 >      void                 SetE55(Double_t E55)                  { fE55 = E55;                     }
96 >      void                 SetEcalJurassicIso(Double_t iso )     { fEcalJurassicIsolation = iso;   }
97 >      void                 SetESuperClusterOverP(Double_t x)     { fESuperClusterOverP = x;        }
98 >      void                 SetESeedClusterOverPout(Double_t x)   { fESeedClusterOverPout = x;      }
99 >      void                 SetHadronicOverEm(Double_t x)         { fHadronicOverEm = x;            }
100 >      void                 SetIDLikelihood(Double_t likelihood)  { fIDLikelihood = likelihood;     }
101 >      void                 SetIsEnergyScaleCorrected(Double_t x) { fIsEnergyScaleCorrected = x;    }
102 >      void                 SetIsMomentumCorrected(Double_t x)    { fIsMomentumCorrected = x;       }
103 >      void                 SetHcalIsolation(Double_t iso )       { fHcalJurassicIsolation = iso;   }
104 >      void                 SetNumberOfClusters(Double_t x)       { fNumberOfClusters = x;          }
105 >      void                 SetPassLooseID(Double_t passLooseID)  { fPassLooseID = passLooseID;     }
106 >      void                 SetPassTightID(Double_t passTightID)  { fPassTightID = passTightID;     }
107 >      void                 SetPIn(Double_t PIn)                  { fPIn = PIn;                     }
108 >      void                 SetPOut(Double_t POut)                { fPOut = POut;                   }
109 >      void                 SetTrackIsolation(Double_t trkiso)    { fTrackIsolation = trkiso;       }
110  
111      protected:
112 <      TRef                 fGsfTrackRef;     //global combined track reference
113 <      TRef                 fTrackerTrackRef; //tracker track reference
114 <      TRef                 fSuperClusterRef; //superCluster
115 <
116 <      Double_t             fESuperClusterOverP;
117 <      Double_t             fESeedClusterOverPout;
118 <      Double_t             fDeltaEtaSuperClTrkAtVtx;
119 <      Double_t             fDeltaEtaSeedClTrkAtCalo;
120 <      Double_t             fDeltaPhiSuperClTrkAtVtx;
121 <      Double_t             fDeltaPhiSeedClTrkAtCalo;
122 <      Double_t             fHadronicOverEm;
123 <      Double_t             fIsEnergyScaleCorrected;
124 <      Double_t             fIsMomentumCorrected;      
125 <      Double_t             fNumberOfClusters;
126 <      Double_t             fClassification;          
127 <      Double_t             fSuperClusterPx;
128 <      Double_t             fSuperClusterPy;
129 <      Double_t             fSuperClusterPz;
130 <      Double_t             fSuperClusterE;
131 <      Double_t             fE33;
132 <      Double_t             fE55;
133 <      Double_t             fCovEtaEta;
134 <      Double_t             fCovEtaPhi;
135 <      Double_t             fCovPhiPhi;
136 <      Double_t             fCaloIsolation;
137 <      Double_t             fTrackIsolation;
112 >      Double_t             GetMass()               const         { return 0.51099892e-3;           }
113 >      void                 GetMom()                const;
114 >
115 >      Ref<Track>           fGsfTrackRef;               //gsf track reference
116 >      Ref<Track>           fTrackerTrackRef;           //tracker track reference
117 >      Ref<SuperCluster>    fSuperClusterRef;           //reference to SuperCluster
118 >      Double32_t           fESuperClusterOverP;        //
119 >      Double32_t           fESeedClusterOverPout;      //
120 >      Double32_t           fDeltaEtaSuperClTrkAtVtx;   //
121 >      Double32_t           fDeltaEtaSeedClTrkAtCalo;   //
122 >      Double32_t           fDeltaPhiSuperClTrkAtVtx;   //
123 >      Double32_t           fDeltaPhiSeedClTrkAtCalo;   //
124 >      Double32_t           fHadronicOverEm;            //
125 >      Double32_t           fIsEnergyScaleCorrected;    //
126 >      Double32_t           fIsMomentumCorrected;       //
127 >      Double32_t           fNumberOfClusters;          //
128 >      Int_t                fClassification;            //
129 >      Double32_t           fE33;                       //
130 >      Double32_t           fE55;                       //
131 >      Double32_t           fCovEtaEta;                 //
132 >      Double32_t           fCoviEtaiEta;               //
133 >      Double32_t           fCovEtaPhi;                 //
134 >      Double32_t           fCovPhiPhi;                 //
135 >      Double32_t           fCaloIsolation;             //
136 >      Double32_t           fCaloTowerIsolation;        //
137 >      Double32_t           fTrackIsolation;            //
138 >      Double32_t           fEcalJurassicIsolation;     //
139 >      Double32_t           fHcalJurassicIsolation;     //
140 >      Double32_t           fPassLooseID;               //
141 >      Double32_t           fPassTightID;               //
142 >      Double32_t           fIDLikelihood;              //
143 >      Double32_t           fPIn;                       //
144 >      Double32_t           fPOut;                      //
145  
146      ClassDef(Electron, 1) // Electron class
147    };
# Line 123 | Line 152 | inline const mithep::Track *mithep::Elec
152   {
153    // Return "best" track.
154  
155 <  if (GsfTrk())
155 >  if (HasGsfTrk())
156      return GsfTrk();
157 <  else if (TrackerTrk())
157 >  else if (HasTrackerTrk())
158      return TrackerTrk();
159  
160    return 0;
161   }
162  
163   //--------------------------------------------------------------------------------------------------
164 < inline const mithep::Track *mithep::Electron::GsfTrk() const
164 > inline void mithep::Electron::GetMom() const
165   {
166 <  // Return global combined track.
166 >  // Get momentum of the electron. We use the direction of the
167 >  // track and the energy of the SuperCluster.
168  
169 <  return static_cast<const Track*>(fGsfTrackRef.GetObject());
140 < }
169 >  const mithep::Track *trk = Trk();
170  
171 < //--------------------------------------------------------------------------------------------------
172 < inline const mithep::Track *mithep::Electron::TrackerTrk() const
173 < {
174 <  // Return tracker track.
171 >  if (!trk) {
172 >    fCachedMom.SetCoordinates(0,0,0,0);
173 >    return;
174 >  }
175 >
176 >  Double_t p = 0;
177 >  Double_t mass = GetMass();
178 >
179 >  const mithep::SuperCluster *sc = SCluster();
180 >  if (sc)
181 >    p = TMath::Sqrt(sc->Energy()*sc->Energy() - mass*mass);
182 >  else
183 >    p = trk->P();
184  
185 <  return static_cast<const Track*>(fTrackerTrackRef.GetObject());
185 >  Double_t pt = TMath::Abs(p*TMath::Cos(trk->Lambda()));
186 >  fCachedMom.SetCoordinates(pt,trk->Eta(),trk->Phi(),mass);
187   }
149 //--------------------------------------------------------------------------------------------------
150 inline const mithep::SuperCluster *mithep::Electron::SCluster() const
151 {
152  // Return Super cluster
188  
189 <  return static_cast<const SuperCluster*>(fSuperClusterRef.GetObject());
189 > //-------------------------------------------------------------------------------------------------
190 > inline Double_t mithep::Electron::ESeedClusterOverPIn() const
191 > {
192 >  // Return energy of the SuperCluster seed divided by the magnitude
193 >  // of the track momentum at the vertex.
194 >  
195 >  return SCluster()->Seed()->Energy() / PIn();
196   }
197   #endif

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines