| 77 |
|
#ifndef MITANA_DATATREE_TRACK_H |
| 78 |
|
#define MITANA_DATATREE_TRACK_H |
| 79 |
|
|
| 80 |
+ |
#include "MitAna/DataCont/interface/BitMask.h" |
| 81 |
+ |
#include "MitAna/DataTree/interface/BaseVertex.h" |
| 82 |
|
#include "MitAna/DataTree/interface/DataObject.h" |
| 81 |
– |
#include "MitAna/DataTree/interface/SuperCluster.h" |
| 83 |
|
#include "MitAna/DataTree/interface/MCParticle.h" |
| 84 |
< |
#include "MitAna/DataTree/interface/BitMask.h" |
| 84 |
< |
#include "MitAna/DataTree/interface/BaseVertex.h" |
| 85 |
< |
#include "MitAna/DataTree/interface/Types.h" |
| 84 |
> |
#include "MitAna/DataTree/interface/SuperCluster.h" |
| 85 |
|
|
| 86 |
|
namespace mithep |
| 87 |
|
{ |
| 143 |
|
fChi2(0), fNdof(0), fEtaEcal(0), fPhiEcal(0) {} |
| 144 |
|
~Track() {} |
| 145 |
|
|
| 146 |
< |
Int_t Charge() const { return (fQOverP>0) ? 1 : -1; } |
| 147 |
< |
Double_t Chi2() const { return fChi2; } |
| 148 |
< |
Double_t RChi2() const { return fChi2/(Double_t)fNdof; } |
| 149 |
< |
void ClearHit(EHitLayer l) { fHits.ClearBit(l); } |
| 150 |
< |
Double_t D0() const { return -fDxy; } |
| 151 |
< |
Double_t D0Corrected(const BaseVertex *iVertex) const; |
| 152 |
< |
Double_t D0Err() const { return fDxyErr; } |
| 153 |
< |
Double_t Dsz() const { return fDsz; } |
| 154 |
< |
Double_t DszErr() const { return fDszErr; } |
| 155 |
< |
Double_t Dxy() const { return fDxy; } |
| 156 |
< |
Double_t DxyErr() const { return fDxyErr; } |
| 157 |
< |
Double_t E(Double_t m) const { return TMath::Sqrt(E2(m)); } |
| 158 |
< |
Double_t E2(Double_t m) const { return P2()+m*m; } |
| 159 |
< |
Double_t Eta() const { return Mom().Eta(); } |
| 160 |
< |
Double_t EtaEcal() const { return fEtaEcal; } |
| 161 |
< |
Bool_t Hit(EHitLayer l) const { return fHits.TestBit(l); } |
| 162 |
< |
const BitMask48 &Hits() const { return fHits; } |
| 163 |
< |
Double_t Lambda() const { return fLambda; } |
| 164 |
< |
Double_t LambdaErr() const { return fLambdaErr; } |
| 165 |
< |
const MCParticle *MCPart() const; |
| 166 |
< |
ThreeVector Mom() const { return ThreeVector(Px(),Py(),Pz()); } |
| 167 |
< |
FourVector Mom4(Double_t m) const { return FourVector(Px(),Py(),Pz(),E(m)); } |
| 168 |
< |
UInt_t Ndof() const { return fNdof; } |
| 169 |
< |
UInt_t NHits() const { return fHits.NBitsSet(); } |
| 170 |
< |
UInt_t NStereoHits() const { return StereoHits().NBitsSet(); } |
| 171 |
< |
Double_t P2() const { return P()*P(); } |
| 172 |
< |
Double_t P() const { return TMath::Abs(1./fQOverP); } |
| 173 |
< |
Double_t Phi() const { return fPhi0; } |
| 174 |
< |
Double_t Phi0() const { return fPhi0; } |
| 175 |
< |
Double_t Phi0Err() const { return fPhi0Err; } |
| 176 |
< |
Double_t PhiEcal() const { return fPhiEcal; } |
| 177 |
< |
Double_t Prob() const { return TMath::Prob(fChi2,fNdof); } |
| 178 |
< |
Double_t Pt() const { return TMath::Abs(TMath::Cos(fLambda)/fQOverP); } |
| 179 |
< |
Double_t Px() const { return Pt()*TMath::Cos(fPhi0); } |
| 180 |
< |
Double_t Py() const { return Pt()*TMath::Sin(fPhi0); } |
| 181 |
< |
Double_t Pz() const { return P()*TMath::Sin(fLambda); } |
| 182 |
< |
Double_t QOverP() const { return fQOverP; } |
| 183 |
< |
Double_t QOverPErr() const { return fQOverPErr; } |
| 184 |
< |
Double_t Theta() const { return (TMath::PiOver2() - fLambda); } |
| 185 |
< |
Double_t Z0() const { return fDsz/TMath::Cos(fLambda); } |
| 186 |
< |
void SetChi2(Double_t chi2) { fChi2 = chi2; } |
| 187 |
< |
void SetErrors(Double_t qOverPErr, Double_t lambdaErr, Double_t phi0Err, |
| 188 |
< |
Double_t dXyErr, Double_t dSzErr); |
| 189 |
< |
void SetEtaEcal(Double_t eta) { fEtaEcal = eta; } |
| 190 |
< |
void SetHelix (Double_t qOverP, Double_t lambda, Double_t phi0, |
| 191 |
< |
Double_t dXy, Double_t dSz); |
| 192 |
< |
void SetHit(EHitLayer l) { fHits.SetBit(l); } |
| 193 |
< |
void SetHits(const BitMask48 &hits) { fHits = hits; } |
| 194 |
< |
void SetNdof(UInt_t dof) { fNdof = dof; } |
| 195 |
< |
void SetMCPart(MCParticle *p) { fMCParticleRef = p; } |
| 196 |
< |
void SetPhiEcal(Double_t phi) { fPhiEcal = phi; } |
| 197 |
< |
void SetSCluster(SuperCluster* sc) { fSuperClusterRef = sc; } |
| 198 |
< |
const SuperCluster *SCluster() const; |
| 199 |
< |
const BitMask48 StereoHits() const { return (fHits & StereoLayers()); } |
| 200 |
< |
static const BitMask48 StereoLayers(); |
| 146 |
> |
Int_t Charge() const { return (fQOverP>0) ? 1 : -1; } |
| 147 |
> |
Double_t Chi2() const { return fChi2; } |
| 148 |
> |
void ClearHit(EHitLayer l) { fHits.ClearBit(l); } |
| 149 |
> |
Double_t D0() const { return -fDxy; } |
| 150 |
> |
Double_t D0Corrected(const BaseVertex &iVertex) const; |
| 151 |
> |
Double_t D0Err() const { return fDxyErr; } |
| 152 |
> |
Double_t Dsz() const { return fDsz; } |
| 153 |
> |
Double_t DszErr() const { return fDszErr; } |
| 154 |
> |
Double_t Dxy() const { return fDxy; } |
| 155 |
> |
Double_t DxyErr() const { return fDxyErr; } |
| 156 |
> |
Double_t E(Double_t m) const { return TMath::Sqrt(E2(m)); } |
| 157 |
> |
Double_t E2(Double_t m) const { return P2()+m*m; } |
| 158 |
> |
Double_t Eta() const { return Mom().Eta(); } |
| 159 |
> |
Double_t EtaEcal() const { return fEtaEcal; } |
| 160 |
> |
Bool_t Hit(EHitLayer l) const { return fHits.TestBit(l); } |
| 161 |
> |
const BitMask48 &Hits() const { return fHits; } |
| 162 |
> |
Double_t Lambda() const { return fLambda; } |
| 163 |
> |
Double_t LambdaErr() const { return fLambdaErr; } |
| 164 |
> |
const MCParticle *MCPart() const { return fMCParticleRef.Obj(); } |
| 165 |
> |
const ThreeVectorC &Mom() const; |
| 166 |
> |
FourVectorM Mom4(Double_t m) const { return FourVectorM(Pt(),Eta(),Phi(),m); } |
| 167 |
> |
UShort_t Ndof() const { return fNdof; } |
| 168 |
> |
UInt_t NHits() const { return fHits.NBitsSet(); } |
| 169 |
> |
UInt_t NStereoHits() const { return StereoHits().NBitsSet(); } |
| 170 |
> |
EObjType ObjType() const { return kTrack; } |
| 171 |
> |
Double_t P2() const { return 1./fQOverP/fQOverP; } |
| 172 |
> |
Double_t P() const { return TMath::Abs(1./fQOverP); } |
| 173 |
> |
Double_t Phi() const { return fPhi0; } |
| 174 |
> |
Double_t Phi0() const { return fPhi0; } |
| 175 |
> |
Double_t Phi0Err() const { return fPhi0Err; } |
| 176 |
> |
Double_t PhiEcal() const { return fPhiEcal; } |
| 177 |
> |
Double_t Prob() const { return TMath::Prob(fChi2,fNdof); } |
| 178 |
> |
Double_t Pt() const { return Mom().Rho(); } |
| 179 |
> |
Double_t Px() const { return Mom().X(); } |
| 180 |
> |
Double_t Py() const { return Mom().Y(); } |
| 181 |
> |
Double_t Pz() const { return Mom().Z(); } |
| 182 |
> |
Double_t QOverP() const { return fQOverP; } |
| 183 |
> |
Double_t QOverPErr() const { return fQOverPErr; } |
| 184 |
> |
Double_t RChi2() const { return fChi2/(Double_t)fNdof; } |
| 185 |
> |
Double_t Theta() const { return (TMath::PiOver2() - fLambda); } |
| 186 |
> |
const SuperCluster *SCluster() const { return fSuperClusterRef.Obj(); } |
| 187 |
> |
const BitMask48 StereoHits() const { return (fHits & StereoLayers()); } |
| 188 |
> |
void SetChi2(Double_t chi2) { fChi2 = chi2; } |
| 189 |
> |
void SetErrors(Double_t qOverPErr, Double_t lambdaErr, Double_t phi0Err, |
| 190 |
> |
Double_t dXyErr, Double_t dSzErr); |
| 191 |
> |
void SetEtaEcal(Double_t eta) { fEtaEcal = eta; } |
| 192 |
> |
void SetHelix (Double_t qOverP, Double_t lambda, Double_t phi0, |
| 193 |
> |
Double_t dXy, Double_t dSz); |
| 194 |
> |
void SetHit(EHitLayer l) { fHits.SetBit(l); } |
| 195 |
> |
void SetHits(const BitMask48 &hits) { fHits = hits; } |
| 196 |
> |
void SetNdof(UShort_t dof) { fNdof = dof; } |
| 197 |
> |
void SetMCPart(const MCParticle *p) { fMCParticleRef = p; } |
| 198 |
> |
void SetPhiEcal(Double_t phi) { fPhiEcal = phi; } |
| 199 |
> |
void SetSCluster(const SuperCluster* sc) { fSuperClusterRef = sc; } |
| 200 |
> |
Double_t Z0() const { return fDsz/TMath::Cos(fLambda); } |
| 201 |
> |
|
| 202 |
> |
static |
| 203 |
> |
const BitMask48 StereoLayers(); |
| 204 |
|
|
| 205 |
|
protected: |
| 206 |
< |
BitMask48 fHits; //storage for mostly hit information |
| 207 |
< |
Double_t fQOverP; //signed inverse of momentum [1/GeV] |
| 208 |
< |
Double_t fQOverPErr; //error of q/p |
| 209 |
< |
Double_t fLambda; //pi/2 - polar angle at the reference point |
| 210 |
< |
Double_t fLambdaErr; //error of lambda |
| 211 |
< |
Double_t fPhi0; //azimuth angle at the given point |
| 212 |
< |
Double_t fPhi0Err; //error of azimuthal angle |
| 213 |
< |
Double_t fDxy; //transverse distance to reference point [cm] |
| 214 |
< |
Double_t fDxyErr; //error of transverse distance |
| 215 |
< |
Double_t fDsz; //longitudinal distance to reference point [cm] |
| 216 |
< |
Double_t fDszErr; //error of longitudinal distance |
| 217 |
< |
Double_t fChi2; //chi squared of track fit |
| 218 |
< |
UInt_t fNdof; //degree-of-freedom of track fit |
| 219 |
< |
Double32_t fEtaEcal; //Eta of track at Ecal front face |
| 220 |
< |
Double32_t fPhiEcal; //phi of track at Ecal front face |
| 221 |
< |
TRef fSuperClusterRef; //superCluster crossed by track |
| 222 |
< |
TRef fMCParticleRef; //reference to sim particle (for monte carlo) |
| 206 |
> |
void ClearMom() const { fCacheMomFlag.ClearCache(); } |
| 207 |
> |
void GetMom() const; |
| 208 |
> |
|
| 209 |
> |
BitMask48 fHits; //storage for mostly hit information |
| 210 |
> |
Double32_t fQOverP; //[0,0,14]signed inverse of momentum [1/GeV] |
| 211 |
> |
Double32_t fQOverPErr; //[0,0,14]error of q/p |
| 212 |
> |
Double32_t fLambda; //[0,0,14]pi/2 - polar angle at the reference point |
| 213 |
> |
Double32_t fLambdaErr; //[0,0,14]error of lambda |
| 214 |
> |
Double32_t fPhi0; //[0,0,14]azimuth angle at the given point |
| 215 |
> |
Double32_t fPhi0Err; //[0,0,14]error of azimuthal angle |
| 216 |
> |
Double32_t fDxy; //[0,0,14]trans. distance to reference point [cm] |
| 217 |
> |
Double32_t fDxyErr; //[0,0,14]error of transverse distance |
| 218 |
> |
Double32_t fDsz; //[0,0,14]long. distance to reference point [cm] |
| 219 |
> |
Double32_t fDszErr; //[0,0,14]error of longitudinal distance |
| 220 |
> |
Double32_t fChi2; //[0,0,12]chi squared of track fit |
| 221 |
> |
UShort_t fNdof; //degree-of-freedom of track fit |
| 222 |
> |
Double32_t fEtaEcal; //[0,0,12]eta of track at Ecal front face |
| 223 |
> |
Double32_t fPhiEcal; //[0,0,12]phi of track at Ecal front face |
| 224 |
> |
Ref<SuperCluster> fSuperClusterRef; //superCluster crossed by track |
| 225 |
> |
Ref<MCParticle> fMCParticleRef; //reference to sim particle (for monte carlo) |
| 226 |
> |
mutable CacheFlag fCacheMomFlag; //||cache validity flag for momentum |
| 227 |
> |
mutable ThreeVectorC fCachedMom; //!cached momentum vector |
| 228 |
|
|
| 229 |
< |
ClassDef(Track, 2) // Track class |
| 229 |
> |
ClassDef(Track, 1) // Track class |
| 230 |
|
}; |
| 231 |
|
} |
| 232 |
|
|
| 233 |
|
//-------------------------------------------------------------------------------------------------- |
| 234 |
< |
inline Double_t mithep::Track::D0Corrected(const BaseVertex *iVertex) const |
| 234 |
> |
inline void mithep::Track::GetMom() const |
| 235 |
> |
{ |
| 236 |
> |
// Compute three momentum. |
| 237 |
> |
|
| 238 |
> |
Double_t pt = TMath::Abs(TMath::Cos(fLambda)/fQOverP); |
| 239 |
> |
Double_t eta = - TMath::Log(TMath::Tan(Theta()/2.)); |
| 240 |
> |
fCachedMom.SetCoordinates(pt,eta,Phi()); |
| 241 |
> |
} |
| 242 |
> |
|
| 243 |
> |
//-------------------------------------------------------------------------------------------------- |
| 244 |
> |
inline const mithep::ThreeVectorC &mithep::Track::Mom() const |
| 245 |
> |
{ |
| 246 |
> |
// Return cached momentum value. |
| 247 |
> |
|
| 248 |
> |
if (!fCacheMomFlag.IsValid()) { |
| 249 |
> |
GetMom(); |
| 250 |
> |
fCacheMomFlag.SetValid(); |
| 251 |
> |
} |
| 252 |
> |
return fCachedMom; |
| 253 |
> |
} |
| 254 |
> |
|
| 255 |
> |
//-------------------------------------------------------------------------------------------------- |
| 256 |
> |
inline Double_t mithep::Track::D0Corrected(const BaseVertex &iVertex) const |
| 257 |
|
{ |
| 258 |
< |
// Return corrected d0 with respect to primary vertex or beamspot |
| 258 |
> |
// Return corrected d0 with respect to primary vertex or beamspot. |
| 259 |
|
|
| 260 |
|
Double_t lXM = -TMath::Sin(Phi()) * D0(); |
| 261 |
|
Double_t lYM = TMath::Cos(Phi()) * D0(); |
| 262 |
< |
Double_t lDX = (lXM + iVertex->X()); |
| 263 |
< |
Double_t lDY = (lYM + iVertex->Y()); |
| 262 |
> |
Double_t lDX = (lXM + iVertex.X()); |
| 263 |
> |
Double_t lDY = (lYM + iVertex.Y()); |
| 264 |
|
Double_t d0Corr = (Px()*lDY - Py()*lDX)/Pt(); |
| 265 |
|
|
| 266 |
|
return d0Corr; |
| 238 |
– |
|
| 267 |
|
} |
| 268 |
|
|
| 269 |
|
//-------------------------------------------------------------------------------------------------- |
| 270 |
|
inline |
| 271 |
|
void mithep::Track::SetHelix(Double_t qOverP, Double_t lambda, Double_t phi0, |
| 272 |
< |
Double_t dxy, Double_t dsz) |
| 272 |
> |
Double_t dxy, Double_t dsz) |
| 273 |
|
{ |
| 274 |
|
// Set helix parameters. |
| 275 |
|
|
| 278 |
|
fPhi0 = phi0; |
| 279 |
|
fDxy = dxy; |
| 280 |
|
fDsz = dsz; |
| 281 |
+ |
ClearMom(); |
| 282 |
|
} |
| 283 |
|
|
| 284 |
|
//-------------------------------------------------------------------------------------------------- |
| 285 |
|
inline |
| 286 |
|
void mithep::Track::SetErrors(Double_t qOverPErr, Double_t lambdaErr, Double_t phi0Err, |
| 287 |
< |
Double_t dxyErr, Double_t dszErr) |
| 287 |
> |
Double_t dxyErr, Double_t dszErr) |
| 288 |
|
{ |
| 289 |
|
// Set helix errors. |
| 290 |
|
|
| 297 |
|
|
| 298 |
|
//-------------------------------------------------------------------------------------------------- |
| 299 |
|
inline |
| 271 |
– |
const mithep::MCParticle *mithep::Track::MCPart() const |
| 272 |
– |
{ |
| 273 |
– |
// Get reference to simulated particle. |
| 274 |
– |
|
| 275 |
– |
return static_cast<const MCParticle*>(fMCParticleRef.GetObject()); |
| 276 |
– |
} |
| 277 |
– |
|
| 278 |
– |
//-------------------------------------------------------------------------------------------------- |
| 279 |
– |
inline const mithep::SuperCluster *mithep::Track::SCluster() const |
| 280 |
– |
{ |
| 281 |
– |
// Return Super cluster |
| 282 |
– |
|
| 283 |
– |
return static_cast<const SuperCluster*>(fSuperClusterRef.GetObject()); |
| 284 |
– |
} |
| 285 |
– |
|
| 286 |
– |
//-------------------------------------------------------------------------------------------------- |
| 287 |
– |
inline |
| 300 |
|
const mithep::BitMask48 mithep::Track::StereoLayers() |
| 301 |
|
{ |
| 302 |
< |
// Build and return BitMask of stereo layers |
| 302 |
> |
// Build and return BitMask of stereo layers. |
| 303 |
|
|
| 304 |
|
mithep::BitMask48 stereoLayers; |
| 293 |
– |
|
| 305 |
|
stereoLayers.SetBit(mithep::Track::TIB1S); |
| 306 |
|
stereoLayers.SetBit(mithep::Track::TIB2S); |
| 307 |
|
stereoLayers.SetBit(mithep::Track::TID1S); |
| 318 |
|
stereoLayers.SetBit(mithep::Track::TEC7S); |
| 319 |
|
stereoLayers.SetBit(mithep::Track::TEC8S); |
| 320 |
|
stereoLayers.SetBit(mithep::Track::TEC9S); |
| 310 |
– |
|
| 321 |
|
return stereoLayers; |
| 322 |
|
} |
| 323 |
|
#endif |
