PWGJE/EMCALJetTasks/AliJetModelCopyTracks.h

   1 #ifndef ALIJETMODELCOPYTRACKS_H
   2 #define ALIJETMODELCOPYTRACKS_H
   3
   4 // $Id$
   5
   6 class TClonesArray;
   7 class TRandom3;
   8 class AliVParticle;
   9 class AliPicoTrack;
  10
  11 #include "AliAnalysisTaskEmcal.h"
  12
  13 class AliJetModelCopyTracks : public AliAnalysisTaskEmcal {
  14  public:
  15   enum ParticleMass {
  16     kMassive  = 0,
  17     kMassless = 1,
  18     kPionMass = 2
  19   };
  20
  21
  22   AliJetModelCopyTracks();
  23   AliJetModelCopyTracks(const char *name);
  24   virtual ~AliJetModelCopyTracks();
  25
  26   virtual void           UserCreateOutputObjects();
  27
  28   void                   SetTracksOutName(const char *n)          { fTracksOutName   = n;    }
  29   void                   SetParticleMassType(ParticleMass m)      { fParticleMass    = m;    }
  30
  31   void                   ExecOnce();
  32   Bool_t                 Run();
  33
  34   void                   CopyTracks();
  35
  36   TString                fTracksOutName;       // name of output track collection
  37   TClonesArray          *fTracksOut;           //!output track collection
  38   ParticleMass           fParticleMass;        // particle mass to use
  39
  40   //Output objects
  41   TH1F     *fHistPtOut;                        //!pT spectrum of output particles
  42
  43  private:
  44   AliJetModelCopyTracks(const AliJetModelCopyTracks&);            // not implemented
  45   AliJetModelCopyTracks &operator=(const AliJetModelCopyTracks&); // not implemented
  46
  47   ClassDef(AliJetModelCopyTracks, 1) // copy tracks class
  48 };
  49 #endif