New analysis devoted to shower shape studies

[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliPWG4HighPtQAMC.h

diff --git a/PWG4/JetTasks/AliPWG4HighPtQAMC.h b/PWG4/JetTasks/AliPWG4HighPtQAMC.h
index 0a5519da081109f2893bba0c1a5d2a4ffbd47f18..7defa802ca25954cbdb52d18396a0491e08dbe66 100644 (file)
--- a/PWG4/JetTasks/AliPWG4HighPtQAMC.h
+++ b/PWG4/JetTasks/AliPWG4HighPtQAMC.h
@@ -29,6 +29,7 @@ class TH3F;
 class TList;
 class AliESDEvent;
 class AliESDtrackCuts;
+class AliMCEvent;
 
 class AliPWG4HighPtQAMC: public AliAnalysisTask {
 
@@ -45,18 +46,27 @@ class AliPWG4HighPtQAMC: public AliAnalysisTask {
   void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
   void SetCutsITS(AliESDtrackCuts* trackCutsITS) {fTrackCutsITS = trackCutsITS;}
 
+  void SetTrackType(Int_t trackType) {fTrackType = trackType;}
+  void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
+  Float_t GetPtMax()           {return fPtMax;}
+
  protected:
 
  private:
 
-  void InitHistPointers();
   AliPWG4HighPtQAMC(const AliPWG4HighPtQAMC&);
   AliPWG4HighPtQAMC& operator=(const AliPWG4HighPtQAMC&);
 
   AliESDEvent *fESD;              //! ESD object
+  AliMCEvent  *fMC;               //! MC event object
+ 
   AliESDtrackCuts *fTrackCuts;    // TrackCuts for global reconstructed vs MC comparison
   AliESDtrackCuts *fTrackCutsITS; // TrackCuts including ITSrefit
 
+  Int_t fTrackType;               // 0: global track; 1:TPConly track
+
+  Float_t fPtMax;                 // Maximum pT for histograms
+
   
   TH1F *fNEventAll;                            //! Event counter
   TH1F *fNEventSel;                            //! Event counter