]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - CORRFW/AliCFTrackIsPrimaryCuts.h
cut class prepared to remove cut duplication, wait for update of AliESDtrackCuts...
[u/mrichter/AliRoot.git] / CORRFW / AliCFTrackIsPrimaryCuts.h
index e7a5b8e9d59a3e2c258036cc206072a580aac1ad..fe485cba9f2f8fda6ceb6352564a6d18f950492d 100644 (file)
 // ESD and AOD data.
 // It mainly consists of a IsSelected function that returns a boolean.
 // This function checks whether the considered track passes a set of cuts:
-// - distance to main vertex in units of sigma (resolution)
+// - min. and max. distance to main vertex in transverse plane (xy)
+// - min. and max. longitudinal distance to main vertex (z)
+// - min. and max. distance to main vertex as ellpise in xy - z plane
+// - all above cuts on absolute values or in units of sigma (resolution)
+// - min. and max. distance to main vertex in units of sigma (resolution)
+// - max. transverse (xy) and longitudinal (z) impact parameter resolution
 // - require that the dca calculation doesn't fail
 // - accept or not accept daughter tracks of kink decays
 //
+// By default, the distance to 'vertex calculated from tracks' is used.
+// Optionally the SPD (tracklet based) or TPC (TPC only tracks based) vertex
+// can be used.
+// Note: the distance to the TPC-vertex is already stored in the ESD,
+// the distance to the SPD-vertex has to be re-calculated by propagating each
+// track while executing this cut.
+//
 // 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
 // added in the Analysis Framwork's class AliAnalysisFilter and applied by
 #define ALICFTRACKISPRIMARYCUTS_H
 
 #include "AliCFCutBase.h"
-
+#include "AliAODTrack.h"
 #include <TH2.h>
-// class TH2 ;
+#include "AliESDtrackCuts.h"
 class TBits;
-class AliESDtrack ;
+class AliESDtrack;
+class AliESD;
 
 class AliCFTrackIsPrimaryCuts : public AliCFCutBase
 {
@@ -58,12 +71,23 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
 
   Bool_t IsSelected(TObject* obj);
   Bool_t IsSelected(TList* /*list*/) {return kTRUE;}
-  void GetSigmaToVertex(AliESDtrack* esdTrack);
 
   // cut value setter
-  void SetMaxNSigmaToVertex(Double_t sigma=3)  {fNSigmaToVertexMax = sigma;}
-  void SetRequireSigmaToVertex(Bool_t b=kTRUE) {fRequireSigmaToVertex=b;}
-  void SetAcceptKinkDaughters(Bool_t b=kTRUE)  {fAcceptKinkDaughters=b;}
+  void UseSPDvertex(Bool_t b=kFALSE);
+  void UseTPCvertex(Bool_t b=kFALSE);
+  void SetMinDCAToVertexXY(Float_t dist=0.)             {fMinDCAToVertexXY = dist;}
+  void SetMinDCAToVertexZ(Float_t dist=0.)              {fMinDCAToVertexZ = dist;}
+  void SetMaxDCAToVertexXY(Float_t dist=1e10)           {fMaxDCAToVertexXY = dist;}
+  void SetMaxDCAToVertexZ(Float_t dist=1e10)            {fMaxDCAToVertexZ = dist;}
+  void SetDCAToVertex2D(Bool_t b=kFALSE)                {fDCAToVertex2D = b;}
+  void SetAbsDCAToVertex(Bool_t b=kTRUE)                {fAbsDCAToVertex = b;}
+  void SetMinNSigmaToVertex(Double_t sigmaMin=0.)      {fNSigmaToVertexMin = sigmaMin;}
+  void SetMaxNSigmaToVertex(Double_t sigmaMax=1.e+10)  {fNSigmaToVertexMax = sigmaMax;}
+  void SetMaxSigmaDCAxy(Double_t sigmaMax=1.e+10)      {fSigmaDCAxy = sigmaMax;}
+  void SetMaxSigmaDCAz(Double_t sigmaMax=1.e+10)       {fSigmaDCAz = sigmaMax;}
+  void SetRequireSigmaToVertex(Bool_t b=kFALSE)                {fRequireSigmaToVertex=b;}
+  void SetAcceptKinkDaughters(Bool_t b=kTRUE)          {fAcceptKinkDaughters=b;}
+  void SetAODType(Char_t type=AliAODTrack::kUndef)      {fAODType = type;}
 
   // QA histograms
   void DrawHistograms();
@@ -73,6 +97,7 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
   // please use indices from the enumeration below
   void SetHistogramBins(Int_t index, Int_t nbins, Double_t *bins);
   void SetHistogramBins(Int_t index, Int_t nbins, Double_t xmin, Double_t xmax);
+  virtual void SetEvtInfo(TObject* esd) ;
 
   // indeces/counters for single selections
   enum { 
@@ -83,24 +108,41 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
     kDcaZ,                     // controll histogram: dca along z axis
     kDcaXYnorm,                        // controll histogram: normalised dca in xy plane
     kDcaZnorm,                 // controll histogram: normalised dca along z axis
-    kNCuts=3,                  // number of single selections
+    kSigmaDcaXY,               // controll histogram: impact parameter resolution in transverse plane
+    kSigmaDcaZ,                        // controll histogram: impact parameter resolution along z axis
+    kCutAODType,                // cut on AliAODTrack::fType
+    kNCuts=9,                  // number of single selections
     kNStepQA=2,                        // number of QA steps (before/after the cuts)
-    kNHist=7                   // number of QA histograms
+    kNHist=9                   // number of QA histograms
   };
 
  private:
   void SelectionBitMap(TObject* obj);
+  void GetDCA(AliESDtrack* esdTrack);
   void DefineHistograms();             // books histograms and TList
   void Initialise();                   // sets everything to 0
   void FillHistograms(TObject* obj, Bool_t b);
                                        // Fills histograms before and after cuts
-  Double_t fNSigmaToVertex;            // track distance to main vertex in units of sigma
+  AliESD*  fESD;                       // pointer to event, needed for SPD vertex
+  Bool_t   fUseSPDvertex;              // flag: calculate dca to SPD-vertex, off by default
+  Bool_t   fUseTPCvertex;              // flag: calculate dca to TPC-vertex, off by default
+  Double_t fMinDCAToVertexXY;          // cut value: min distance to main vertex in transverse plane
+  Double_t fMinDCAToVertexZ;           // cut value: min longitudinal distance to main vertex
+  Double_t fMaxDCAToVertexXY;          // cut value: max distance to main vertex in transverse plane
+  Double_t fMaxDCAToVertexZ;           // cut value: max longitudinal distance to main vertex
+  Bool_t   fDCAToVertex2D;             // flag: cut on ellipse in xy - z plane
+  Bool_t   fAbsDCAToVertex;            // flag: cut on absolute values or in units of sigma
+  Double_t fNSigmaToVertexMin;         // cut value: min distance to main vertex in units of sigma
   Double_t fNSigmaToVertexMax;         // cut value: max distance to main vertex in units of sigma
-  Bool_t  fRequireSigmaToVertex;       // require calculable distance to main vertex
+  Double_t fSigmaDCAxy;                        // cut value: impact parameter resolution in xy plane
+  Double_t fSigmaDCAz;                 // cut value: impact parameter resolution in z direction
+  Double_t fDCA[6];                    // impact parameters
+  Bool_t   fRequireSigmaToVertex;      // require calculable distance to main vertex
+  Char_t   fAODType;                    // type of AOD track (undef, primary, secondary, orphan)
+                                        // applicable at AOD level only !
 
   TH2F* fhDcaXYvsDcaZ[2];              // Histogram: dca xy vs. z
-  TH2F* fhDcaXYvsDcaZnorm[2];          // Histogram: (dca xy / sigma xy) vs. (dca z / simga z)
-  Bool_t  fAcceptKinkDaughters;                // accepting kink daughters
+  Bool_t fAcceptKinkDaughters;         // accepting kink daughters
 
   TH1F* fhCutStatistics;               // Histogram: statistics of what cuts the tracks did not survive
   TH2F* fhCutCorrelation;              // Histogram: 2d statistics plot
@@ -116,6 +158,8 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
   Int_t fhNBinsDcaZ;                   // number of bins+1: dca along beam axis
   Int_t fhNBinsDcaXYnorm;              // number of bins+1: normalised dca in transverse plane
   Int_t fhNBinsDcaZnorm;               // number of bins+1: normalised dca along beam axis
+  Int_t fhNBinsSigmaDcaXY;             // number of bins+1: impact parameter resolution in transverse plane
+  Int_t fhNBinsSigmaDcaZ;              // number of bins+1: impact parameter resolution along beam axis
 
   Double_t *fhBinLimNSigma; //[fhNBinsNSigma] bin limits: dca in units of sigma
   Double_t *fhBinLimRequireSigma;//[fhNBinsRequireSigma] bin limits: require successful calcuation
@@ -124,8 +168,10 @@ class AliCFTrackIsPrimaryCuts : public AliCFCutBase
   Double_t *fhBinLimDcaZ; //[fhNBinsDcaZ] bin limits: dca along beam axis
   Double_t *fhBinLimDcaXYnorm; //[fhNBinsDcaXYnorm] bin limits: normalised dca in transverse plane
   Double_t *fhBinLimDcaZnorm;//[fhNBinsDcaZnorm] bin limits: normalised dca along beam axis
+  Double_t *fhBinLimSigmaDcaXY; //[fhNBinsSigmaDcaXY] bin limits: impact parameter in transverse plane
+  Double_t *fhBinLimSigmaDcaZ; //[fhNBinsSigmaDcaZ] bin limits: impact parameter along beam axis
 
-  ClassDef(AliCFTrackIsPrimaryCuts,2);
+  ClassDef(AliCFTrackIsPrimaryCuts,3);
 };
 
 #endif