PWGGA/EMCALTasks/AliEmcalPicoTrackMaker.h

   1 #ifndef ALIEMCALPICOTRACKMAKER_H
   2 #define ALIEMCALPICOTRACKMAKER_H
   3
   4 // $Id: AliEmcalPicoTrackMaker.h 54003 2012-01-19 16:40:42Z loizides $
   5
   6 class TClonesArray;
   7 class AliVEvent;
   8 class AliVTrack;
   9 class AliESDtrackCuts;
  10
  11 #include "AliAnalysisTaskSE.h"
  12
  13 class AliEmcalPicoTrackMaker : public AliAnalysisTaskSE {
  14  public:
  15   AliEmcalPicoTrackMaker();
  16   AliEmcalPicoTrackMaker(const char *name);
  17   virtual ~AliEmcalPicoTrackMaker();
  18
  19   void UserCreateOutputObjects();
  20   void UserExec(Option_t *option);
  21
  22   void SetAODfilterBits(Int_t b0 = 0, Int_t b1 = 0, Int_t b2 = 0)      { fAODfilterBits[0] = b0; fAODfilterBits[1] = b1; fAODfilterBits[2] = b2; }
  23   void SetTracksOutName(const char *name)                              { fTracksOutName    = name; }
  24   void SetTracksInName(const char *name)                               { fTracksInName     = name; }
  25   void SetESDtrackCuts(AliESDtrackCuts *cuts)                          { fESDtrackCuts     = cuts; }
  26
  27  protected:
  28   Bool_t             AcceptTrack(AliVTrack *track)        ;
  29   void               RetrieveEventObjects()               ;
  30   AliVTrack*         GetTrack(const Int_t i)         const;
  31   Int_t              GetNumberOfTracks()             const;
  32
  33   Int_t              fAODfilterBits[3];     // AOD track filter bit map
  34   AliESDtrackCuts   *fESDtrackCuts;         // ESD track cuts
  35   TString            fTracksOutName;        // Name of output track array
  36   TString            fTracksInName;         // Name of input track array
  37   TClonesArray      *fTracksIn;             //!Track array in
  38   TClonesArray      *fTracksOut;            //!Track array out
  39
  40  private:
  41   AliEmcalPicoTrackMaker(const AliEmcalPicoTrackMaker&);            // not implemented
  42   AliEmcalPicoTrackMaker &operator=(const AliEmcalPicoTrackMaker&); // not implemented
  43
  44   ClassDef(AliEmcalPicoTrackMaker, 1); // Class to make PicoTracks in AOD/ESD events
  45 };
  46
  47 #endif