PWGPP/TPC/AliTPCComparisonPID.h

   1 #ifndef ALITPCOMPARISONPID_H
   2 #define ALITPCOMPARISONPID_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 AliGenInfoMaker;
  18 class TTreeSRedirector;
  19 class AliMCEventHadnler;
  20 class TParticle;
  21 class AliMCInfo;
  22 class AliESDRecInfo;
  23 class AliESDEvent;
  24 class AliMCEvent;
  25 class THnSparse;
  26
  27 class AliTPCComparisonPID : public AliAnalysisTask {
  28 public:
  29   AliTPCComparisonPID();
  30   AliTPCComparisonPID(const char *name);
  31   AliTPCComparisonPID(const AliTPCComparisonPID& info);
  32   virtual ~AliTPCComparisonPID();
  33   virtual void   ConnectInputData(Option_t *);
  34   virtual void   CreateOutputObjects();
  35   virtual void   Exec(Option_t *option);
  36   virtual void   Terminate(Option_t *);
  37   virtual void   FinishTaskOutput();
  38   void           SetDebugOuputhPath(const char * name){fDebugOutputPath=name;}
  39   void           Init();
  40   //
  41   void           ProcessMCInfo();
  42   //
  43   THnSparse * GetTPCsignal(){return fTPCsignal;}
  44   THnSparse * GetTPCsignalNorm(){return fTPCsignalNorm;}
  45   //
  46   // debug streamer part
  47   //
  48   TTreeSRedirector *GetDebugStreamer();
  49   void       SetStreamLevel(Int_t streamLevel){fStreamLevel=streamLevel;}
  50   void       SetDebugLevel(Int_t level) {fDebugLevel = level;}
  51   Int_t      GetStreamLevel() const {return fStreamLevel;}
  52   Int_t      GetDebugLevel() const {return fDebugLevel;}
  53   //
  54 protected:
  55   void RegisterDebugOutput();
  56   AliTPCComparisonPID& operator=(const AliTPCComparisonPID& /*info*/) { return *this;}
  57   AliMCEvent  * fMCinfo;          //! MC event handler
  58   AliESDEvent * fESD;             //! current esd event
  59   //
  60   //
  61   //
  62   THnSparse * fTPCsignal;         //raw tpc signal - dEdx
  63   THnSparse * fTPCsignalNorm;     //normalized TPC signal
  64   //
  65   TTreeSRedirector *fDebugStreamer;     //! debug streamer
  66   Int_t  fStreamLevel;                  //  debug stream level
  67   Int_t  fDebugLevel;                   //  debug level
  68   TString      fDebugOutputPath; // debug output path
  69   ClassDef(AliTPCComparisonPID, 1); // Analysis task base class for tracks
  70 };
  71
  72 #endif