PWG/EMCAL/AliEmcalMCTrackSelector.h

   1 #ifndef ALIEMCALMCTRACKSELECTOR_H
   2 #define ALIEMCALMCTRACKSELECTOR_H
   3
   4 // $Id$
   5
   6 class TClonesArray;
   7 class TString;
   8
   9 #include "AliAnalysisTaskSE.h"
  10
  11 class AliEmcalMCTrackSelector : public AliAnalysisTaskSE {
  12  public:
  13   AliEmcalMCTrackSelector();
  14   AliEmcalMCTrackSelector(const char *name);
  15   virtual ~AliEmcalMCTrackSelector();
  16
  17   void UserCreateOutputObjects();
  18   void UserExec(Option_t *option);
  19
  20   void SetTracksOutName(const char *name)            { fTracksOutName    = name ; }
  21   void SetRejectNK(Bool_t r = kTRUE)                 { fRejectNK         = r    ; }
  22   void SetChargedMC(Bool_t c = kTRUE)                { fChargedMC        = c    ; }
  23
  24  protected:
  25   TString            fTracksOutName;        // name of output track array
  26   Bool_t             fRejectNK;             // true = reject k0l and neutrons
  27   Bool_t             fChargedMC;            // true = only charged particles
  28   TClonesArray      *fTracksOut;            //!track array out
  29
  30  private:
  31   AliEmcalMCTrackSelector(const AliEmcalMCTrackSelector&);            // not implemented
  32   AliEmcalMCTrackSelector &operator=(const AliEmcalMCTrackSelector&); // not implemented
  33
  34   ClassDef(AliEmcalMCTrackSelector, 1); // Task to select tracks in MC events
  35 };
  36 #endif