PWG/EMCAL/AliEmcalEsdTpcTrackTask.h

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