PWG/EMCAL/AliEmcalEsdTrackFilterTask.h

   1 #ifndef ALIEMCALESDTRACKFILTERTASK_H
   2 #define ALIEMCALESDTRACKFILTERTASK_H
   3
   4 // $Id$
   5
   6 class TClonesArray;
   7 class AliESDEvent;
   8 class AliESDtrack;
   9 class AliESDtrackCuts;
  10 class AliEMCALRecoUtils;
  11
  12 #include "AliAnalysisTaskSE.h"
  13 #include "AliESDtrackCuts.h"
  14
  15 class AliEmcalEsdTrackFilterTask : public AliAnalysisTaskSE {
  16  public:
  17   AliEmcalEsdTrackFilterTask();
  18   AliEmcalEsdTrackFilterTask(const char *name);
  19   virtual ~AliEmcalEsdTrackFilterTask();
  20
  21   void UserCreateOutputObjects();
  22   void UserExec(Option_t *option);
  23
  24   void SetDist(Double_t d)                       { fDist             = d;    }
  25   void SetDoPropagation(Bool_t b)                { fDoPropagation    = b;    }
  26   void SetDoSpdVtxConstrain(Bool_t b)            { fDoSpdVtxCon      = b;    }
  27   void SetHybridTrackCuts(AliESDtrackCuts *cuts) { fHybridTrackCuts  = cuts; }
  28   void SetIncludeNoITS(Bool_t f)                 { fIncludeNoITS     = f;    }
  29   void SetTrackCuts(AliESDtrackCuts *cuts)       { fEsdTrackCuts     = cuts; }
  30   void SetTracksName(const char *name)           { fTracksName       = name; }
  31
  32  protected:
  33   AliESDtrackCuts   *fEsdTrackCuts;      // esd track cuts
  34   Bool_t             fDoSpdVtxCon;       // if true then do vertex constraint
  35   AliESDtrackCuts   *fHybridTrackCuts;   // hybrid track cuts
  36   TString            fTracksName;        // name of tracks
  37   Bool_t             fIncludeNoITS;      // includes tracks with failed ITS refit
  38   Bool_t             fDoPropagation;     // propagate all hybrid tracks to EMCal surface
  39   Double_t           fDist;              // distance to surface (430cm default)
  40   AliESDEvent       *fEsdEv;             //!esd event
  41   TClonesArray      *fTracks;            //!track array
  42
  43  private:
  44   AliEmcalEsdTrackFilterTask(const AliEmcalEsdTrackFilterTask&);            // not implemented
  45   AliEmcalEsdTrackFilterTask &operator=(const AliEmcalEsdTrackFilterTask&); // not implemented
  46
  47   ClassDef(AliEmcalEsdTrackFilterTask, 1); // Class to constrain TPC tracks to SPD vertex
  48 };
  49
  50 #endif