PWGPP/TPC/AliTPCtaskPID.h

   1 #ifndef ALITPCTASKPID_H
   2 #define ALITPCTASKPID_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 #include <TString.h>
  17 class AliMCEventHadnler;
  18 class TParticle;
  19 class AliMCInfo;
  20 class AliESDRecInfo;
  21 class AliESDEvent;
  22 class AliMCEvent;
  23 class THnSparse;
  24 class TObjArray;
  25
  26 class AliTPCtaskPID : public AliAnalysisTask {
  27 public:
  28   AliTPCtaskPID();
  29   AliTPCtaskPID(const char *name);
  30   AliTPCtaskPID(const AliTPCtaskPID& info);
  31   virtual ~AliTPCtaskPID();
  32   virtual void   ConnectInputData(Option_t *);
  33   virtual void   CreateOutputObjects();
  34   virtual void   Exec(Option_t *option);
  35   virtual void   Terminate(Option_t *);
  36   virtual void   FinishTaskOutput();
  37   void           Init();
  38   //
  39   void           ProcessMCInfo();
  40   //
  41   THnSparse * GetTPCsignal(){return fTPCsignal;}
  42   THnSparse * GetTPCsignalNorm(){return fTPCsignalNorm;}
  43   THnSparse * GetTPCr(){return fTPCr;}
  44   //
  45   static void BinLogX(TAxis *axis);
  46 protected:
  47   void RegisterDebugOutput();
  48   AliTPCtaskPID& operator=(const AliTPCtaskPID& /*info*/) { return *this;}
  49   AliMCEvent  * fMCinfo;          //! MC event handler
  50   AliESDEvent * fESD;             //! current esd event
  51   //
  52   //
  53   //
  54   TObjArray  *fList; //TList output object
  55   THnSparse * fTPCsignal;         //raw tpc signal - dEdx
  56   THnSparse * fTPCsignalNorm;     //normalized TPC signal
  57   THnSparse * fTPCr;              //TPC pid info from ESD
  58   ClassDef(AliTPCtaskPID, 1); // Analysis task base class for tracks
  59 };
  60
  61 #endif