HG flag copied

[u/mrichter/AliRoot.git] / ANALYSIS / AliKineTrackCuts.h

diff --git a/ANALYSIS/AliKineTrackCuts.h b/ANALYSIS/AliKineTrackCuts.h
index 67ccacd4b3d4c55bcc48be9bd43066a0d3785d22..b3b3f6acb9fda9eab8d2647debb25437dd706b63 100644 (file)
--- a/ANALYSIS/AliKineTrackCuts.h
+++ b/ANALYSIS/AliKineTrackCuts.h
@@ -7,6 +7,7 @@
 #include "AliAnalysisCuts.h"
 
 class  TObject;
+class  TList;
 
 class AliKineTrackCuts : public AliAnalysisCuts
 {
@@ -15,28 +16,32 @@ public:
                       AliKineTrackCuts(const Char_t* name = "AliKineTrackCuts", const Char_t* title = "");
   virtual            ~AliKineTrackCuts(){;}
   
-            Bool_t    IsSelected(TObject* obj);
-           Bool_t    IsSelected(TList* /*list*/) {return kTRUE;}
+              Bool_t  IsSelected(TObject* obj);
+             Bool_t  IsSelected(TList* /*list*/) {return kTRUE;}
             
+              void    SetFinalParticles( Bool_t val=kTRUE )          { fOnlyFinalParticles = val; }
+              void    SetPrimaryParticles( Bool_t val=kTRUE )        { fOnlyPrimary = val; }
   // track kinematic cut setters
-              void    SetPRange(Float_t r1=0, Float_t r2=1e10)       {fPMin=r1;   fPMax=r2;}
-              void    SetPtRange(Float_t r1=0, Float_t r2=1e10)      {fPtMin=r1;  fPtMax=r2;}
-              void    SetPxRange(Float_t r1=-1e10, Float_t r2=1e10)  {fPxMin=r1;  fPxMax=r2;}
-              void    SetPyRange(Float_t r1=-1e10, Float_t r2=1e10)  {fPyMin=r1;  fPyMax=r2;}
-              void    SetPzRange(Float_t r1=-1e10, Float_t r2=1e10)  {fPzMin=r1;  fPzMax=r2;}
-              void    SetEtaRange(Float_t r1=-1e10, Float_t r2=1e10) {fEtaMin=r1; fEtaMax=r2;}
-              void    SetRapRange(Float_t r1=-1e10, Float_t r2=1e10) {fRapMin=r1; fRapMax=r2;}
+             void    SetPRange(Float_t r1=0, Float_t r2=1e10)       { fPMin=r1;   fPMax=r2;}
+              void    SetPtRange(Float_t r1=0, Float_t r2=1e10)      { fPtMin=r1;  fPtMax=r2;}
+              void    SetPxRange(Float_t r1=-1e10, Float_t r2=1e10)  { fPxMin=r1;  fPxMax=r2;}
+              void    SetPyRange(Float_t r1=-1e10, Float_t r2=1e10)  { fPyMin=r1;  fPyMax=r2;}
+              void    SetPzRange(Float_t r1=-1e10, Float_t r2=1e10)  { fPzMin=r1;  fPzMax=r2;}
+              void    SetEtaRange(Float_t r1=-1e10, Float_t r2=1e10) { fEtaMin=r1; fEtaMax=r2;}
+              void    SetRapRange(Float_t r1=-1e10, Float_t r2=1e10) { fRapMin=r1; fRapMax=r2;}
                   
 protected:        
                   
+           Bool_t    fOnlyFinalParticles;   // true => skip part with GetStatusCode()!=1
+           Bool_t    fOnlyPrimary;          // Only Primary Particles
   // kinematics cuts
-           Float_t   fPMin,   fPMax;             // definition of the range of the P
-           Float_t   fPtMin,  fPtMax;            // definition of the range of the Pt
-           Float_t   fPxMin,  fPxMax;            // definition of the range of the Px
-           Float_t   fPyMin,  fPyMax;            // definition of the range of the Py
-           Float_t   fPzMin,  fPzMax;            // definition of the range of the Pz
-           Float_t   fEtaMin, fEtaMax;           // definition of the range of the eta
-           Float_t   fRapMin, fRapMax;           // definition of the range of the y
+           Float_t   fPMin,   fPMax;        // definition of the range of the P
+           Float_t   fPtMin,  fPtMax;       // definition of the range of the Pt
+           Float_t   fPxMin,  fPxMax;       // definition of the range of the Px
+           Float_t   fPyMin,  fPyMax;       // definition of the range of the Py
+           Float_t   fPzMin,  fPzMax;       // definition of the range of the Pz
+           Float_t   fEtaMin, fEtaMax;      // definition of the range of the eta
+           Float_t   fRapMin, fRapMax;      // definition of the range of the y
   
   
   ClassDef(AliKineTrackCuts, 1)