changes from saiola and soh

[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliEmcalPicoTrackMaker.h

diff --git a/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.h b/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.h
index 3a6120603e81a0eb4998395c29169268bfc3e20d..5c525998f1830ad6385190dee2e421345bd49227 100644 (file)
--- a/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.h
+++ b/PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.h
@@ -19,19 +19,26 @@ class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
   void UserCreateOutputObjects();
   void UserExec(Option_t *option);
 
-  void SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)  { fAODfilterBits[0] = b0; fAODfilterBits[1] = b1; }
-  void SetESDtrackCuts(AliESDtrackCuts *cuts)        { fESDtrackCuts     = cuts; }
-  void SetTracksInName(const char *name)             { fTracksInName     = name; }
-  void SetTracksOutName(const char *name)            { fTracksOutName    = name; }
-  void SetMaxTrackPt(Float_t pt)                     { fMaxTrackPt       = pt  ; }
-  void SetTrackEfficiency(Double_t eff = 0.95)       { fTrackEfficiency = eff  ; }
+  void SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0)      { fAODfilterBits[0] = b0  ; fAODfilterBits[1] = b1  ; }
+  void SetESDtrackCuts(AliESDtrackCuts *cuts)            { fESDtrackCuts     = cuts; }
+  void SetTracksInName(const char *name)                 { fTracksInName     = name; }
+  void SetTracksOutName(const char *name)                { fTracksOutName    = name; }
+  void SetTrackPtLimits(Double_t min, Double_t max)      { fMaxTrackPt       = max ; fMinTrackPt       = min ; }
+  void SetTrackEtaLimits(Double_t min, Double_t max)     { fMaxTrackEta      = max ; fMinTrackEta      = min ; }
+  void SetTrackPhiLimits(Double_t min, Double_t max)     { fMaxTrackPhi      = max ; fMinTrackPhi      = min ; }
+  void SetTrackEfficiency(Double_t eff = 0.95)           { fTrackEfficiency = eff  ; }
 
  protected:
   Int_t              fAODfilterBits[2];     // AOD track filter bit map
   AliESDtrackCuts   *fESDtrackCuts;         // ESD track cuts
   TString            fTracksOutName;        // name of output track array
   TString            fTracksInName;         // name of input track array
-  Float_t            fMaxTrackPt;           // max pt of tracks
+  Double_t           fMinTrackPt;           // mix pt of tracks
+  Double_t           fMaxTrackPt;           // max pt of tracks
+  Double_t           fMinTrackEta;          // cut on track eta
+  Double_t           fMaxTrackEta;          // cut on track eta
+  Double_t           fMinTrackPhi;          // cut on track phi
+  Double_t           fMaxTrackPhi;          // cut on track phi
   Double_t           fTrackEfficiency;      // track efficiency
   TClonesArray      *fTracksIn;             //!track array in
   TClonesArray      *fTracksOut;            //!track array out
@@ -40,6 +47,6 @@ class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
   AliEmcalPicoTrackMaker(const AliEmcalPicoTrackMaker&);            // not implemented
   AliEmcalPicoTrackMaker &operator=(const AliEmcalPicoTrackMaker&); // not implemented
 
-  ClassDef(AliEmcalPicoTrackMaker, 2); // Task to make PicoTracks in AOD/ESD events
+  ClassDef(AliEmcalPicoTrackMaker, 3); // Task to make PicoTracks in AOD/ESD events
 };
 #endif