PWG1/AliGenInfoTask.h

   1 #ifndef ALIGENINFOTASK_H
   2 #define ALIGENINFOTASK_H
   3
   4 // ROOT includes
   5 #include <TList.h>
   6 #include <TH1.h>
   7 #include <TH2.h>
   8
   9 // AliRoot includes
  10 #include <AliAnalysisTask.h>
  11 #include <AliESDEvent.h>
  12 #include <AliESDfriend.h>
  13 #include <AliESDtrack.h>
  14 #include <AliESDfriendTrack.h>
  15 #include <AliTPCseed.h>
  16 class AliGenInfoMaker;
  17
  18 class AliGenInfoTask : public AliAnalysisTask {
  19  public:
  20  AliGenInfoTask();
  21  AliGenInfoTask(const char *name);
  22   virtual ~AliGenInfoTask() {};
  23
  24   virtual void   ConnectInputData(Option_t *);
  25   virtual void   CreateOutputObjects();
  26   virtual void   Exec(Option_t *option);
  27   virtual void   Terminate(Option_t *);
  28   virtual void   SetDebugLevel(Int_t level) {fDebug = level;}
  29   virtual void   SetMaxTracks(Int_t max=10) {fMaxTracks = max;}
  30
  31  protected:
  32   AliGenInfoTask(const AliGenInfoTask& /*info*/);
  33   AliGenInfoTask& operator=(const AliGenInfoTask& /*info*/) { return *this;}
  34
  35   virtual Int_t FillTrackHistograms(Int_t nTracks, AliESDtrack* track,
  36                                     AliESDfriendTrack* friendTrack,
  37                                     AliTPCseed* seed);
  38   AliGenInfoMaker *fGenMaker;    // gen Maker
  39   Int_t         fDebug;          //  Debug flag
  40   AliESDEvent*  fESD;            //! ESD
  41   AliESDfriend* fESDfriend;      //! ESD friend
  42   TList*        fListOfHists;    //! Output list of histograms
  43
  44   Int_t         fMaxTracks;      // Max tracks in histogram
  45   TH1F*         hESDTracks;      //! N ESD tracks
  46   TH1F*         hGoodTracks;     //! GOOD tracks
  47
  48   ClassDef(AliGenInfoTask, 1); // Analysis task base class for TPC tracks and clusters
  49 };
  50
  51 #endif