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.15 by sixie, Fri Aug 22 09:55:40 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 <      FourVector           Mom()                   const;
37 <      const Track         *Trk()                   const { return BestTrk();                }
38 <      Double_t             E()                     const {return SCluster()->Energy();      }
39 <
40 <
41 <      Double_t    Mass()                           const { return 0.51099892e-3;            }
42 <      Double_t    ESuperClusterOverP()             const { return fESuperClusterOverP;      }
43 <      Double_t    ESeedClusterOverPout()           const { return fESeedClusterOverPout;    }
44 <      Double_t    ESeedClusterOverPIn()            const;
45 <      Double_t    PIn()                            const { return fPIn;                     }
46 <      Double_t    POut()                           const { return fPOut;                    }
47 <      Double_t    DeltaEtaSuperClusterTrackAtVtx() const { return fDeltaEtaSuperClTrkAtVtx; }
48 <      Double_t    DeltaEtaSeedClusterTrackAtCalo() const { return fDeltaEtaSeedClTrkAtCalo; }
49 <      Double_t    DeltaPhiSuperClusterTrackAtVtx() const { return fDeltaPhiSuperClTrkAtVtx; }
50 <      Double_t    DeltaPhiSeedClusterTrackAtCalo() const { return fDeltaPhiSeedClTrkAtCalo; }
51 <      Double_t    HadronicOverEm()                 const { return fHadronicOverEm;          }
52 <      Double_t    IsEnergyScaleCorrected()         const { return fIsEnergyScaleCorrected;  }
53 <      Double_t    IsMomentumCorrected()            const { return fIsMomentumCorrected;     }
54 <      Double_t    NumberOfClusters()               const { return fNumberOfClusters;        }
55 <      Double_t    Classification()                 const { return fClassification;          }
56 <      Double_t    E33()                            const { return fE33;                     }
57 <      Double_t    E55()                            const { return fE55;                     }
58 <      Double_t    CovEtaEta()                      const { return fCovEtaEta;               }
59 <      Double_t    CovEtaPhi()                      const { return fCovEtaPhi;               }
60 <      Double_t    CovPhiPhi()                      const { return fCovPhiPhi;               }
61 <      Double_t    CaloIsolation()                  const { return fCaloIsolation;           }
62 <      Double_t    TrackIsolation()                 const { return fTrackIsolation;          }
63 <      Double_t    PassLooseID()                    const { return fPassLooseID;             }
64 <      Double_t    PassTightID()                    const { return fPassTightID;             }
65 <      Double_t    IDLikelihood()                   const { return fIDLikelihood;            }
66 <      Double_t    ComputeTrackIsolation   (  Double_t extRadius, Double_t intRadius,
67 <                                             Double_t ptLow, Double_t maxVtxZDist,
68 <                                             mithep::Collection<Track> *tracks              );
69 <      Double_t    ComputeEcalIsolation    (  Double_t coneSize, Double_t etLow,
70 <                                             mithep::Collection<BasicCluster> *basicClusters);
71 <
72 <      void        SetGsfTrk(Track* t)                           { fGsfTrackRef = t;                }
73 <      void        SetTrackerTrk(Track* t)                       { fTrackerTrackRef = t;            }
74 <      void        SetSuperCluster(SuperCluster* sc)             { fSuperClusterRef = sc;           }
75 <      void        SetESuperClusterOverP(Double_t x)             { fESuperClusterOverP = x;         }
76 <      void        SetESeedClusterOverPout(Double_t x)           { fESeedClusterOverPout = x;       }
77 <      void        SetPIn(Double_t PIn)                          { fPIn = PIn;                      }
78 <      void        SetPOut(Double_t POut)                        { fPOut = POut;                    }
79 <      void        SetDeltaEtaSuperClusterTrackAtVtx(Double_t x) { fDeltaEtaSuperClTrkAtVtx = x;    }
80 <      void        SetDeltaEtaSeedClusterTrackAtCalo(Double_t x) { fDeltaEtaSeedClTrkAtCalo = x;    }
81 <      void        SetDeltaPhiSuperClusterTrackAtVtx(Double_t x) { fDeltaPhiSuperClTrkAtVtx = x;    }
82 <      void        SetDeltaPhiSeedClusterTrackAtCalo(Double_t x) { fDeltaPhiSeedClTrkAtCalo = x;    }
83 <      void        SetHadronicOverEm(Double_t x)                 { fHadronicOverEm = x;             }
84 <      void        SetIsEnergyScaleCorrected(Double_t x)         { fIsEnergyScaleCorrected = x;     }
85 <      void        SetIsMomentumCorrected(Double_t x)            { fIsMomentumCorrected = x;        }
86 <      void        SetNumberOfClusters(Double_t x)               { fNumberOfClusters = x;           }
87 <      void        SetClassification(Double_t x)                 { fClassification = x;             }
88 <      void        SetE33(Double_t E33)                          { fE33 = E33;                      }
89 <      void        SetE55(Double_t E55)                          { fE55 = E55;                      }
90 <      void        SetCovEtaEta(Double_t CovEtaEta)              { fCovEtaEta = CovEtaEta;          }
91 <      void        SetCovEtaPhi(Double_t CovEtaPhi)              { fCovEtaPhi = CovEtaPhi;          }
92 <      void        SetCovPhiPhi(Double_t CovPhiPhi)              { fCovPhiPhi = CovPhiPhi;          }
93 <      void        SetCaloIsolation(Double_t CaloIsolation)      { fCaloIsolation = CaloIsolation;  }
94 <      void        SetTrackIsolation(Double_t TrackIsolation)    { fTrackIsolation = TrackIsolation;}
95 <      void        SetPassLooseID(Double_t passLooseID)          { fPassLooseID = passLooseID;      }
96 <      void        SetPassTightID(Double_t passTightID)          { fPassTightID = passTightID;      }
97 <      void        SetIDLikelihood(Double_t likelihood)          { fIDLikelihood = likelihood;      }
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;
138 <      Double_t             fPassLooseID;
139 <      Double_t             fPassTightID;
140 <      Double_t             fIDLikelihood;
141 <      Double_t             fPIn;
142 <      Double_t             fPOut;
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 131 | 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());
148 < }
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 <  return static_cast<const Track*>(fTrackerTrackRef.GetObject());
177 < }
157 < //--------------------------------------------------------------------------------------------------
158 < inline const mithep::SuperCluster *mithep::Electron::SCluster() const
159 < {
160 <  // Return Super cluster
176 >  Double_t p = 0;
177 >  Double_t mass = GetMass();
178  
179 <  return static_cast<const SuperCluster*>(fSuperClusterRef.GetObject());
180 < }
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 < //-------------------------------------------------------------------------------------------------
186 < inline mithep::FourVector mithep::Electron::Mom() const
167 < {
168 <  // Return Momentum of the electron. We use the direction of the
169 <  // Track and the Energy of the Super Cluster
170 <
171 <  double P = TMath::Sqrt( E()*E() - Mass()*Mass());
172 <  return FourVector(P*sin(Trk()->Theta())*cos(Trk()->Phi()),
173 <                    P*sin(Trk()->Theta())*sin(Trk()->Phi()), P*cos(Trk()->Theta()), E());
185 >  Double_t pt = TMath::Abs(p*TMath::Cos(trk->Lambda()));
186 >  fCachedMom.SetCoordinates(pt,trk->Eta(),trk->Phi(),mass);
187   }
188  
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
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   }
183
184
197   #endif
186

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines