added AOD support for single tracks ; refit flags replaced by status flag in

[u/mrichter/AliRoot.git] / CORRFW / AliCFTrackQualityCuts.h

diff --git a/CORRFW/AliCFTrackQualityCuts.h b/CORRFW/AliCFTrackQualityCuts.h
index 36b4ff3973860fd3adf3eeae2e032f1f0281e8c4..9b72996137864561a88981cc2086b144468dc0a4 100644 (file)
--- a/CORRFW/AliCFTrackQualityCuts.h
+++ b/CORRFW/AliCFTrackQualityCuts.h
@@ -25,9 +25,8 @@
 // - number of clusters in the ITS
 // - chi2 / cluster in the TPC
 // - chi2 / cluster in the ITS
-// - successful TPC refit
-// - successful ITS refit
 // - covariance matrix diagonal elements
+// - track status (cf AliESDtrack.h)
 //
 // The cut values for these cuts are set with the corresponding set functions.
 // All cut classes provided by the correction framework are supposed to be
@@ -67,10 +66,9 @@ class AliCFTrackQualityCuts : public AliCFCutBase
   void SetMinNClusterITS(Int_t cluster=-1)             {fMinNClusterITS = cluster;}
   void SetMaxChi2PerClusterTPC(Double_t chi=1e99)      {fMaxChi2PerClusterTPC = chi;}
   void SetMaxChi2PerClusterITS(Double_t chi=1e99)      {fMaxChi2PerClusterITS = chi;}
-  void SetRequireTPCRefit(Bool_t b=kFALSE)             {fRequireTPCRefit = b;}
-  void SetRequireITSRefit(Bool_t b=kFALSE)             {fRequireITSRefit = b;}
   void SetMaxCovDiagonalElements(Float_t c1=1e99, Float_t c2=1e99, Float_t c3=1e99, Float_t c4=1e99, Float_t c5=1e99) 
 {fCovariance11Max=c1;fCovariance22Max=c2;fCovariance33Max=c3;fCovariance44Max=c4;fCovariance55Max=c5;}
+  void SetStatus(ULong_t status=0) {fStatus = status ;}
 
   // QA histograms
   void DrawHistograms(Bool_t drawLogScale=kTRUE);
@@ -87,16 +85,15 @@ class AliCFTrackQualityCuts : public AliCFCutBase
     kCutClusterITS,    // number of clusters in ITS
     kCutChi2TPC,       // chi2 per cluster in TPC
     kCutChi2ITS,       // chi2 per cluster in ITS
-    kCutRefitTPC,      // require TPC refit
-    kCutRefitITS,      // require ITS refit
     kCutCovElement11,  // diagonal element 11 of covariance matrix
     kCutCovElement22,  // diagonal element 22 of covariance matrix
     kCutCovElement33,  // diagonal element 33 of covariance matrix
     kCutCovElement44,  // diagonal element 44 of covariance matrix
     kCutCovElement55,  // diagonal element 55 of covariance matrix
-    kNCuts=11,         // number of single selections
+    kCutStatus,         // track status
+    kNCuts=10,         // number of single selections
     kNStepQA=2,                // number of QA steps (before/after the cuts)
-    kNHist=11          // number of QA histograms
+    kNHist=9           // number of QA histograms
   };
 
  private:
@@ -109,8 +106,6 @@ class AliCFTrackQualityCuts : public AliCFCutBase
   Double_t fMinNClusterITS;            // min number of clusters in ITS
   Double_t fMaxChi2PerClusterTPC;      // max chi2 per clusters in TPC
   Double_t fMaxChi2PerClusterITS;      // max chi2 per clusters in ITS
-  Bool_t fRequireTPCRefit;             // require TPC refit
-  Bool_t fRequireITSRefit;             // require ITS refit
 
   Double_t fCovariance11Max ;          // max covariance matrix element 11
   Double_t fCovariance22Max ;          // max covariance matrix element 22
@@ -118,6 +113,8 @@ class AliCFTrackQualityCuts : public AliCFCutBase
   Double_t fCovariance44Max ;          // max covariance matrix element 44
   Double_t fCovariance55Max ;          // max covariance matrix element 55
 
+  ULong_t fStatus;    // track status
+
   TH1F* fhCutStatistics;               // Histogram: statistics of what cuts the tracks did not survive
   TH2F* fhCutCorrelation;              // Histogram: 2d statistics plot
 
@@ -129,8 +126,6 @@ class AliCFTrackQualityCuts : public AliCFCutBase
   Int_t fhNBinsClusterITS;             // number of bins+1: cluster ITS
   Int_t fhNBinsChi2TPC;                        // number of bins+1: chi2 per cluster TPC
   Int_t fhNBinsChi2ITS;                        // number of bins+1: chi2 per cluster ITS
-  Int_t fhNBinsRefitTPC;               // number of bins+1: require refit TPC
-  Int_t fhNBinsRefitITS;               // number of bins+1: require refit ITS  
   Int_t fhNBinsCovariance11;           // number of bins+1: covariance matrix element 11
   Int_t fhNBinsCovariance22;           // number of bins+1: covariance matrix element 22
   Int_t fhNBinsCovariance33;           // number of bins+1: covariance matrix element 33
@@ -141,15 +136,13 @@ class AliCFTrackQualityCuts : public AliCFCutBase
   Double_t *fhBinLimClusterITS;        //[fhNBinsClusterITS] bin limits: cluster ITS
   Double_t *fhBinLimChi2TPC;   //[fhNBinsChi2TPC] bin limits: chi2 per cluster TPC
   Double_t *fhBinLimChi2ITS;   //[fhNBinsChi2ITS] bin limits: chi2 per cluster ITS
-  Double_t *fhBinLimRefitTPC;  //[fhNBinsRefitTPC] bin limits: require refit TPC
-  Double_t *fhBinLimRefitITS;  //[fhNBinsRefitITS] bin limits: require refit ITS
   Double_t *fhBinLimCovariance11;//[fhNBinsCovariance11] bin limits: covariance matrix element 11
   Double_t *fhBinLimCovariance22;//[fhNBinsCovariance22] bin limits: covariance matrix element 22
   Double_t *fhBinLimCovariance33;//[fhNBinsCovariance33] bin limits: covariance matrix element 33
   Double_t *fhBinLimCovariance44;//[fhNBinsCovariance44] bin limits: covariance matrix element 44
   Double_t *fhBinLimCovariance55;//[fhNBinsCovariance55] bin limits: covariance matrix element 55
 
-  ClassDef(AliCFTrackQualityCuts,2);
+  ClassDef(AliCFTrackQualityCuts,3);
 };
 
 #endif