]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliAODTrack.h
removerd double filling of histogram
[u/mrichter/AliRoot.git] / STEER / AliAODTrack.h
index 6e740adbbfd0a723ab4e09ea882f8d4aa44f325c..6fe59c03933e698868dd249442da214bd5cb0dfe 100644 (file)
@@ -6,18 +6,19 @@
 /* $Id$ */
 
 //-------------------------------------------------------------------------
-//     AOD track implementation of AliVParticle
+//     AOD track implementation of AliVTrack
 //     Author: Markus Oldenburg, CERN
 //-------------------------------------------------------------------------
 
 #include <TRef.h>
+#include <TBits.h>
 
-#include "AliVParticle.h"
+#include "AliVTrack.h"
 #include "AliAODVertex.h"
 #include "AliAODRedCov.h"
 #include "AliAODPid.h"
 
-class AliAODTrack : public AliVParticle {
+class AliAODTrack : public AliVTrack {
 
  public:
   
@@ -61,7 +62,8 @@ class AliAODTrack : public AliVParticle {
              Bool_t usedForVtxFit,
              Bool_t usedForPrimVtxFit,
              AODTrk_t ttype=kUndef,
-             UInt_t selectInfo=0);
+             UInt_t selectInfo=0,
+             Float_t chi2perNDF = -999.);
 
   AliAODTrack(Short_t id,
              Int_t label,
@@ -77,7 +79,8 @@ class AliAODTrack : public AliVParticle {
              Bool_t usedForVtxFit,
              Bool_t usedForPrimVtxFit,
              AODTrk_t ttype=kUndef,
-             UInt_t selectInfo=0);
+             UInt_t selectInfo=0,
+             Float_t chi2perNDF = -999.);
 
   virtual ~AliAODTrack();
   AliAODTrack(const AliAODTrack& trk); 
@@ -126,15 +129,16 @@ class AliAODTrack : public AliVParticle {
  
   template <class T> void SetPID(const T *pid) {
     if(pid) for(Int_t i=0; i<10; ++i) fPID[i]=pid[i];
-    else {for(Int_t i=0; i<10; fPID[i++]=0.); fPID[AliAODTrack::kUnknown]=1.;}}
+    else {for(Int_t i=0; i<10; fPID[i++]=0.) ; fPID[AliAODTrack::kUnknown]=1.;}}
 
   Bool_t IsOn(Int_t mask) const {return (fFlags&mask)>0;}
   ULong_t GetStatus() const { return GetFlags(); }
   ULong_t GetFlags() const { return fFlags; }
 
-  Short_t GetID() const { return fID; }
+  Int_t   GetID() const { return (Int_t)fID; }
   Int_t   GetLabel() const { return fLabel; } 
   Char_t  GetType() const { return fType;}
+  Bool_t  IsPrimaryCandidate() const;
   Bool_t  GetUsedForVtxFit() const { return TestBit(kUsedForVtxFit); }
   Bool_t  GetUsedForPrimVtxFit() const { return TestBit(kUsedForPrimVtxFit); }
 
@@ -156,6 +160,12 @@ class AliAODTrack : public AliVParticle {
     if(!fCovMatrix) return kFALSE;
     fCovMatrix->GetCovMatrix(covMatrix); return kTRUE;}
 
+  Bool_t GetXYZ(Double_t *p) const {
+    return GetPosition(p); }
+
+  Bool_t GetCovarianceXYZPxPyPz(Double_t cv[21]) const {
+    return GetCovMatrix(cv);}
+
   void RemoveCovMatrix() {delete fCovMatrix; fCovMatrix=NULL;}
 
   Double_t XAtDCA() const { return fPositionAtDCA[0]; }
@@ -177,13 +187,21 @@ class AliAODTrack : public AliVParticle {
   Double_t PAtDCA() const { return TMath::Sqrt(PxAtDCA()*PxAtDCA() + PyAtDCA()*PyAtDCA() + PzAtDCA()*PzAtDCA()); }
   Bool_t   PxPyPzAtDCA(Double_t p[3]) const { p[0] = PxAtDCA(); p[1] = PyAtDCA(); p[2] = PzAtDCA(); return kTRUE; }
   
+  Double_t GetRAtAbsorberEnd() const { return fRAtAbsorberEnd; }
+  
   UChar_t  GetITSClusterMap() const       { return (UChar_t)(fITSMuonClusterMap&0xff); }
   UShort_t GetHitsPatternInTrigCh() const { return (UShort_t)((fITSMuonClusterMap&0xff00)>>8); }
   UInt_t   GetMUONClusterMap() const      { return (fITSMuonClusterMap&0x3ff0000)>>16; }
   UInt_t   GetITSMUONClusterMap() const   { return fITSMuonClusterMap; }
   
   Bool_t  TestFilterBit(UInt_t filterBit) const {return (Bool_t) ((filterBit & fFilterMap) != 0);}
+  Bool_t  TestFilterMask(UInt_t filterMask) const {return (Bool_t) ((filterMask & fFilterMap) == filterMask);}
 
+  const TBits& GetTPCClusterMap() const {return fTPCClusterMap;}
+  const TBits& GetTPCSharedMap() const {return fTPCSharedMap;}
+  void    SetTPCClusterMap(const TBits amap) {fTPCClusterMap = amap;}
+  void    SetTPCSharedMap(const TBits amap) {fTPCSharedMap = amap;}
+  
   AliAODPid    *GetDetPid() const { return fDetPid; }
   AliAODVertex *GetProdVertex() const { return (AliAODVertex*)fProdVertex.GetObject(); }
   
@@ -203,7 +221,7 @@ class AliAODTrack : public AliVParticle {
   void SetUsedForVtxFit(Bool_t used = kTRUE) { used ? SetBit(kUsedForVtxFit) : ResetBit(kUsedForVtxFit); }
   void SetUsedForPrimVtxFit(Bool_t used = kTRUE) { used ? SetBit(kUsedForPrimVtxFit) : ResetBit(kUsedForPrimVtxFit); }
 
-  void SetOneOverPt(Double_t oneOverPt) { fMomentum[0] = oneOverPt; }
+  void SetOneOverPt(Double_t oneOverPt) { fMomentum[0] = 1. / oneOverPt; }
   void SetPt(Double_t pt) { fMomentum[0] = pt; };
   void SetPhi(Double_t phi) { fMomentum[1] = phi; }
   void SetTheta(Double_t theta) { fMomentum[2] = theta; }
@@ -213,6 +231,8 @@ class AliAODTrack : public AliVParticle {
   void SetXYAtDCA(Double_t x, Double_t y) {fPositionAtDCA[0] = x; fPositionAtDCA[1] = y;}
   void SetPxPyPzAtDCA(Double_t pX, Double_t pY, Double_t pZ) {fMomentumAtDCA[0] = pX; fMomentumAtDCA[1] = pY; fMomentumAtDCA[2] = pZ;}
   
+  void SetRAtAbsorberEnd(Double_t r) { fRAtAbsorberEnd = r; }
+  
   void SetCharge(Short_t q) { fCharge = q; }
   void SetChi2perNDF(Double_t chi2perNDF) { fChi2perNDF = chi2perNDF; }
 
@@ -227,23 +247,24 @@ class AliAODTrack : public AliVParticle {
                                        //  2 Muon track match Low pt cut
                                        //  3 Muon track match High pt cut
   void     SetMatchTrigger(Int_t MatchTrigger);
-  Int_t    MatchTrigger() const { return (GetMatchTrigger()>0)?1:0; }  //  Muon track matches trigger track
-  Int_t    MatchTriggerAnyPt()   const  { return (GetMatchTrigger()>0)?1:0; }  //  Muon track matches trigger track
-  Int_t    MatchTriggerLowPt()   const  { return (GetMatchTrigger()>1)?1:0; }  //  Muon track matches trigger track and passes Low pt cut
-  Int_t    MatchTriggerHighPt()  const  { return (GetMatchTrigger()>2)?1:0; }  //  Muon track matches trigger track and passes High pt cut
+  Bool_t   MatchTrigger() const { return (GetMatchTrigger()>0); }        //  Muon track matches trigger track
+  Bool_t   MatchTriggerLowPt()   const  { return (GetMatchTrigger()>1); } //  Muon track matches trigger track and passes Low pt cut
+  Bool_t   MatchTriggerHighPt()  const  { return (GetMatchTrigger()>2); } //  Muon track matches trigger track and passes High pt cut
+  Bool_t   MatchTriggerDigits()  const;                                   //  Muon track matches trigger digits
   Double_t GetChi2MatchTrigger() const  { return fChi2MatchTrigger;}
   void     SetChi2MatchTrigger(Double_t Chi2MatchTrigger) {fChi2MatchTrigger = Chi2MatchTrigger; }
-  Int_t    HitsMT(Int_t istation, Int_t iplane, Char_t *cathode=0);  // Check if track hits Muon chambers
-  Int_t    HitsMuonChamber(Int_t MuonChamber);  // Check if track hits Muon chambers
+  Bool_t   HitsMuonChamber(Int_t MuonChamber, Int_t cathode = -1) const;  // Check if track hits Muon chambers
   Bool_t   IsMuonTrack() const { return (GetMUONClusterMap()>0) ? kTRUE : kFALSE; }
+  
+  void     Connected(Bool_t flag) {flag ? SETBIT(fITSMuonClusterMap,26) : CLRBIT(fITSMuonClusterMap,26);}
+  Bool_t   IsConnected() const {return TESTBIT(fITSMuonClusterMap,26);}
 
   void     SetProdVertex(TObject *vertex) { fProdVertex = vertex; }
   void     SetType(AODTrk_t ttype) { fType=ttype; }
 
-  // track operations
-  Bool_t   PropagateTo(Double_t xk, Double_t b);
-
-
+  // Dummy
+  Int_t    PdgCode() const {return 0;}
+  
  private :
 
   // Momentum & position
@@ -253,7 +274,9 @@ class AliAODTrack : public AliVParticle {
   Double32_t    fMomentumAtDCA[3];  // momentum (px,py,pz) at DCA
   Double32_t    fPositionAtDCA[2];  // trasverse position (x,y) at DCA
   
-  Double32_t    fChi2perNDF;        // chi2/NDF of mometum fit
+  Double32_t    fRAtAbsorberEnd;    // transverse position r at the end of the muon absorber
+  
+  Double32_t    fChi2perNDF;        // chi2/NDF of momentum fit
   Double32_t    fChi2MatchTrigger;  // chi2 of trigger/track matching
   Double32_t    fPID[10];           // [0.,1.,8] pointer to PID object
 
@@ -264,16 +287,30 @@ class AliAODTrack : public AliVParticle {
                                     // (ITS: bit 1-8, muon trigger: bit 9-16, muon tracker: bit 17-26, muon match trigger: bit 31-32) 
   UInt_t        fFilterMap;         // filter information, one bit per set of cuts
 
+  TBits         fTPCClusterMap;     // Map of clusters, one bit per padrow; 1 if has a cluster on given padrow
+  TBits         fTPCSharedMap;      // Map of clusters, one bit per padrow; 1 if has a shared cluster on given padrow
+
   Short_t       fID;                // unique track ID, points back to the ESD track
 
   Char_t        fCharge;            // particle charge
   Char_t        fType;              // Track Type
-
+  
   AliAODRedCov<6> *fCovMatrix;      // covariance matrix (x, y, z, px, py, pz)
   AliAODPid    *fDetPid;            // more detailed or detector specific pid information
   TRef          fProdVertex;        // vertex of origin
 
-  ClassDef(AliAODTrack,8);
+  ClassDef(AliAODTrack, 10);
 };
 
+inline Bool_t  AliAODTrack::IsPrimaryCandidate() const
+{
+    // True of track passes primary particle selection (independent of type) 
+    // 
+    if (fFilterMap) {
+       return kTRUE;
+    } else {
+       return kFALSE;
+    }
+}
+
 #endif