Merge branch 'master' of https://git.cern.ch/reps/AliRoot

[u/mrichter/AliRoot.git] / PWGJE / AliPWG4HighPtSpectra.h

diff --git a/PWGJE/AliPWG4HighPtSpectra.h b/PWGJE/AliPWG4HighPtSpectra.h
index 13be72ed60365388ccad314d1a01a487a98af5e8..45a44bae5f5416ed5288e8191f99b21193e9a89b 100644 (file)
--- a/PWGJE/AliPWG4HighPtSpectra.h
+++ b/PWGJE/AliPWG4HighPtSpectra.h
@@ -38,6 +38,7 @@ class AliESDVertex;
 class AliMCEvent;
 class AliStack;
 class AliGenPythiaEventHeader;
+class AliGenHijingEventHeader;
 
 class AliPWG4HighPtSpectra : public AliAnalysisTask {
  public:
@@ -85,6 +86,9 @@ class AliPWG4HighPtSpectra : public AliAnalysisTask {
   //AliESDtrackCuts setters
   void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
   void SetCutsReject(AliESDtrackCuts* trackCuts) {fTrackCutsReject = trackCuts;}
+  void SelectHIJINGOnly(Bool_t b)    {fbSelectHIJING = b;}
+
+  Bool_t IsHIJINGParticle(Int_t label);
 
   void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
 
@@ -93,7 +97,11 @@ class AliPWG4HighPtSpectra : public AliAnalysisTask {
   void   SetReadAODData(Bool_t flag=kTRUE) {fReadAODData=flag;}
 
   static AliGenPythiaEventHeader*  GetPythiaEventHeader(AliMCEvent *mcEvent);
+  static AliGenHijingEventHeader*  GetHijingEventHeader(AliMCEvent *mcEvent);
+
   static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root
+
+
   
  protected:
   Bool_t              fReadAODData ;       // flag for AOD/ESD input files
@@ -115,6 +123,8 @@ class AliPWG4HighPtSpectra : public AliAnalysisTask {
   AliESDtrackCuts *fTrackCuts;           // trackCuts applied to global tracks
   AliESDtrackCuts *fTrackCutsReject;     // trackCuts to reject tracks (hybrid case)
 
+  Bool_t fbSelectHIJING; //Select only particles from HIJING event
+
   Double_t fSigmaConstrainedMax;  // max sigma on constrained fit
 
  private: