]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - PWGLF/SPECTRA/PiKaPr/TestAOD/AliSpectraAODTrackCuts.h
cxx file
[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / AliSpectraAODTrackCuts.h
index dad40cf0c85e1d5cb1f94aab5df0eadefef1a7db..abc5f9f50fd936440f8850fdd1b0dbc78517a9c5 100644 (file)
 // Authors: Michele Floris, CERN, Philip Versteeg, UU, Redmer Bertens, UU
 //-------------------------------------------------------------------------
 
-class AliAODEvent;
-//class AliSpectraAODHistoManager;
 class TH1I;
+class AliAODEvent;
+class AliPIDResponse;  
 class AliAODMCParticle;
 class AliAODTrack;
+class TH2F;
+class TH1F;
 
-#include "AliSpectraAODHistoManager.h"
 #include "TNamed.h"
-using namespace AliSpectraNameSpace;
 
 class AliSpectraAODTrackCuts : public TNamed
 {
@@ -30,76 +29,78 @@ class AliSpectraAODTrackCuts : public TNamed
   enum { kTrkBit = 0, kTrkCuts, kTrkEta, kTrkDCA, kTrkP, kTrkPt,kTrkPtTOF,kTOFMatching,kTrTOFout,kTrTIME,kTrTOFpid,kAccepted,kNTrkCuts};
   
   
- AliSpectraAODTrackCuts() : TNamed(), fIsSelected(0), fTrackBits(0), fMinTPCcls(0), fEtaCutMin(0), fEtaCutMax(0), fPCut(0), fPtCut(0), fPtCutTOFMatching(0), fHistoCuts(0), fHistoNSelectedPos(0), fHistoNSelectedNeg(0), fHistoNMatchedPos(0), fHistoNMatchedNeg(0), fHistoEtaPhiHighPt(0), fTrack(0) {}
+ AliSpectraAODTrackCuts() : TNamed(), fIsSelected(0), fTrackBits(0), fMinTPCcls(0), fEtaCutMin(0), fEtaCutMax(0),fDCACut(0), fPCut(0), fPtCut(0),fYCut(0), fPtCutTOFMatching(0), fHistoCuts(0), fHistoNSelectedPos(0), fHistoNSelectedNeg(0), fHistoNMatchedPos(0), fHistoNMatchedNeg(0), fHistoEtaPhiHighPt(0), fTrack(0), fPIDResponse(0) {}
   
   AliSpectraAODTrackCuts(const char *name);
   virtual  ~AliSpectraAODTrackCuts() {} // To be implemented
   
   Bool_t IsSelected(AliAODTrack * track,Bool_t FillHistStat);
   
+  void SetEta(Float_t etamin,Float_t etamax)   { fEtaCutMin = etamin;fEtaCutMax = etamax; }
+  void SetDCA(Float_t dca)   { fDCACut = dca; }
+  void SetP(Float_t p)       { fPCut = p; }
+  void SetPt(Float_t pt)     { fPtCut = pt; }
+  void SetY(Float_t y) { fYCut = y;}
+  void SetPtTOFMatching(Float_t pt)     { fPtCutTOFMatching = pt; }
   void SetTrackType(UInt_t bit);
+  void SetTrackBits(UInt_t TrackBits) {fTrackBits=TrackBits;}
+  void SetMinTPCcls(UInt_t MinTPCcls) {fMinTPCcls=MinTPCcls;}
+  
+  UInt_t GetTrackType()  const    { return fTrackBits;}
+  TH1I * GetHistoCuts()      { return fHistoCuts; }
+  TH1F * GetHistoNSelectedPos()      { return fHistoNSelectedPos; } 
+  TH1F * GetHistoNSelectedNeg()      { return fHistoNSelectedNeg; }
+  TH1F * GetHistoNMatchedPos()      { return fHistoNMatchedPos; }
+  TH1F * GetHistoNMatchedNeg()      { return fHistoNMatchedNeg; }
+  TH2F * GetHistoEtaPhiHighPt()      { return fHistoEtaPhiHighPt; }
+  Float_t GetEtaMin()       const    { return fEtaCutMin; }
+  Float_t GetEtaMax()       const    { return fEtaCutMax; }
+  Float_t GetY()         const    { return fYCut; }
+  Float_t GetDCA()       const    { return fDCACut; }
+  Float_t GetP()         const    { return fPCut; }
+  Float_t GetPt()        const    { return fPtCut; }
+  Float_t GetPtTOFMatching()        const    { return fPtCutTOFMatching; }
+  
   Bool_t CheckTrackType();
   Bool_t CheckTrackCuts();
   Bool_t CheckEtaCut();
-  Bool_t CheckYCut(AODParticleSpecies_t specie); // not included in standard cuts
+  Bool_t CheckYCut(Double_t mass); // not included in standard cuts
   Bool_t CheckDCACut();
   Bool_t CheckPCut();
   Bool_t CheckPtCut();
   Bool_t CheckTOFMatching(Bool_t FillHistStat);
   void PrintCuts() const;
   
-   UInt_t GetTrackType()  const    { return fTrackBits;}
-   TH1I * GetHistoCuts()      { return fHistoCuts; }
-   TH1F * GetHistoNSelectedPos()      { return fHistoNSelectedPos; } 
-   TH1F * GetHistoNSelectedNeg()      { return fHistoNSelectedNeg; }
-   TH1F * GetHistoNMatchedPos()      { return fHistoNMatchedPos; }
-   TH1F * GetHistoNMatchedNeg()      { return fHistoNMatchedNeg; }
-   TH2F * GetHistoEtaPhiHighPt()      { return fHistoEtaPhiHighPt; }
-   void SetEta(Float_t etamin,Float_t etamax)   { fEtaCutMin = etamin;fEtaCutMax = etamax; }
-   void SetDCA(Float_t dca)   { fDCACut = dca; }
-   void SetP(Float_t p)       { fPCut = p; }
-   void SetPt(Float_t pt)     { fPtCut = pt; }
-   void SetY(Float_t y) { fYCut = y;}
-   void SetPtTOFMatching(Float_t pt)     { fPtCutTOFMatching = pt; }
-   void SetTrackBits(UInt_t TrackBits) {fTrackBits=TrackBits;}
-   void SetMinTPCcls(UInt_t MinTPCcls) {fMinTPCcls=MinTPCcls;}
-   Float_t GetEtaMin()       const    { return fEtaCutMin; }
-   Float_t GetEtaMax()       const    { return fEtaCutMax; }
-   Float_t GetY()         const    { return fYCut; }
-   Float_t GetDCA()       const    { return fDCACut; }
-   Float_t GetP()         const    { return fPCut; }
-   Float_t GetPt()        const    { return fPtCut; }
-   Float_t GetPtTOFMatching()        const    { return fPtCutTOFMatching; }
-    
-   Long64_t Merge(TCollection* list);
+  Long64_t Merge(TCollection* list);
    
    
  private:
-   
-   Bool_t           fIsSelected;      // True if cuts are selected
-   UInt_t           fTrackBits;       // Type of track to be used
-   UInt_t           fMinTPCcls;       // min number of clusters in the TPC
-   Float_t          fEtaCutMin;          // Allowed absolute maximum value of Eta
-   Float_t          fEtaCutMax;          // Allowed absolute maximum value of Eta
-   Float_t          fDCACut;          // Maximum value of DCA
-   Float_t          fPCut;            // Maximum value of P
-   Float_t          fPtCut;           // Maximum value of Pt
-   Float_t          fYCut;           // Maximum value of Y
-   Float_t          fPtCutTOFMatching;           // TOF Matching
-   TH1I             *fHistoCuts;       // Cuts statistics
-   TH1F             *fHistoNSelectedPos;       // Selected positive tracks
-   TH1F             *fHistoNSelectedNeg;       // Selected negative tracks
-   TH1F             *fHistoNMatchedPos;       // Matched positive tracks
-   TH1F             *fHistoNMatchedNeg;       // Matched negative tracks
-   TH2F             *fHistoEtaPhiHighPt;       // EtaPhi distr at high pt (>1.5 GeV/c)
-   AliAODTrack      *fTrack;           //! Track pointer
-   static const char * kBinLabel[]; // labels of stat histo
+  
+  Bool_t           fIsSelected;      // True if cuts are selected
+  UInt_t           fTrackBits;       // Type of track to be used
+  UInt_t           fMinTPCcls;       // min number of clusters in the TPC
+  Float_t          fEtaCutMin;          // Allowed absolute maximum value of Eta
+  Float_t          fEtaCutMax;          // Allowed absolute maximum value of Eta
+  Float_t          fDCACut;          // Maximum value of DCA
+  Float_t          fPCut;            // Maximum value of P
+  Float_t          fPtCut;           // Maximum value of Pt
+  Float_t          fYCut;           // Maximum value of Y
+  Float_t          fPtCutTOFMatching;           // TOF Matching
+  TH1I             *fHistoCuts;       // Cuts statistics
+  TH1F             *fHistoNSelectedPos;       // Selected positive tracks
+  TH1F             *fHistoNSelectedNeg;       // Selected negative tracks
+  TH1F             *fHistoNMatchedPos;       // Matched positive tracks
+  TH1F             *fHistoNMatchedNeg;       // Matched negative tracks
+  TH2F             *fHistoEtaPhiHighPt;       // EtaPhi distr at high pt (>1.5 GeV/c)
+  AliAODTrack      *fTrack;           //! Track pointer
+  AliPIDResponse   *fPIDResponse;     // ! PID response object
+  static const char * kBinLabel[]; // labels of stat histo
 
    
-   AliSpectraAODTrackCuts(const AliSpectraAODTrackCuts&);
-   AliSpectraAODTrackCuts& operator=(const AliSpectraAODTrackCuts&);
+  AliSpectraAODTrackCuts(const AliSpectraAODTrackCuts&);
+  AliSpectraAODTrackCuts& operator=(const AliSpectraAODTrackCuts&);
    
-   ClassDef(AliSpectraAODTrackCuts, 1);
+  ClassDef(AliSpectraAODTrackCuts, 2);
 };
 #endif