ROOT logo
#ifndef ALIANALYSISTASKCHARGEDHADRONSPECTRA_H
#define ALIANALYSISTASKCHARGEDHADRONSPECTRA_H

//////////////////////////////////////////////////////////////////////////////
//                                                                          //
// This analysis extracts pT-spectra of charged kaons, protons, and pions.  //
// It is based on particles identifation via the dE/dx signal of the TPC.   //
//                                                                          //
//////////////////////////////////////////////////////////////////////////////

class TH1;
class TH1F;
class TH2D;
class TH3D;
class TList;
class TObjArray;
class AliESDEvent;
class AliESDtrack;
class AliESDtrackCuts;
class AliHeader;
class AliESDpid;


#include "AliAnalysisTaskSE.h"
#include "THnSparse.h"

class AliAnalysisCombinedHadronSpectra : public AliAnalysisTaskSE {
 public:
  AliAnalysisCombinedHadronSpectra(const char *name);
  AliAnalysisCombinedHadronSpectra();
  virtual ~AliAnalysisCombinedHadronSpectra() {}
  //
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  //
  Bool_t         SelectOnImpPar(AliESDtrack* t);
  //
  void           SetESDtrackCuts(AliESDtrackCuts * trackCuts){fESDtrackCuts = trackCuts;};
  void           SetAlephParameters(const Double_t * parameters){for(Int_t j=0;j<5;j++) fAlephParameters[j] = parameters[j]; Initialize();};
  void           SetIsMCtrue(Bool_t isMCdata = kTRUE){fMCtrue = isMCdata;};
  void           SetUseHBTmultiplicity(Bool_t useHBTmultiplicity = kTRUE){fUseHBTmultiplicity = useHBTmultiplicity;};
  void		 SetUseTPConlyTracks(Bool_t useTPConlyTracks = kTRUE){fUseTPConlyTracks = useTPConlyTracks;};
  void           SetSaveMotherPDG(Bool_t saveMotherPDG =kTRUE){fSaveMotherPDG = saveMotherPDG;};
  void           SetSmallTHnSparse(Bool_t smallTHnSparse = kFALSE) {fSmallTHnSparse = smallTHnSparse;};
  void           SetTPCnSigmaCuts(Double_t nSigmaTPCLow = -3., Double_t nSigmaTPCHigh = 3.){fTPCnSigmaCutLow = nSigmaTPCLow; fTPCnSigmaCutHigh = nSigmaTPCHigh;};
  void           SetRapidityCuts(Double_t rapidityLow = -0.2, Double_t rapidityHigh = 0.2){fRapidityCutLow = rapidityLow; fRapidityCutHigh = rapidityHigh;};
  void           SetEvenDCAbinning(Bool_t EvenDCAbinning = kTRUE) {fEvenDCAbinning = EvenDCAbinning;};
  void           Initialize();
  //
  
 private:
  //
  void  BinLogAxis(const TH1 *h);
  Int_t GetPythiaEventProcessType(const AliHeader* aHeader, Bool_t adebug = kFALSE) const;
  //
  AliESDEvent *fESD;                   //! ESD object
  TList       *fListHist;              //! list for histograms
  //
  AliESDtrackCuts * fESDtrackCuts;     // basic cut variables
  AliESDtrackCuts * fESDTrackCutsMult; // cuts for the MULTIPLICITY DETERMINATION
  AliESDpid       * fESDpid;           // basic TPC object for n-sigma cuts
  Bool_t        fMCtrue;               // flag if real data or MC is processed
  Bool_t        fOnlyQA;               // flag if only QA histograms should be filled
  Bool_t        fUseHBTmultiplicity;   // flag if multiplicity determination should be done as in the HBT paper
  Bool_t	fUseTPConlyTracks;     // flag if TPConly-track should be used
  Bool_t        fSaveMotherPDG;        // flag if PDG of mother should be saved (weak decays)
  Bool_t        fSmallTHnSparse;       // flag if to do cuts on TPC n-sigma and rapidity in task or not
  Double_t      fTPCnSigmaCutLow;      // low border for TPC n-sigma cut
  Double_t      fTPCnSigmaCutHigh;     // high border for TPC n-sigma cut
  Double_t      fRapidityCutLow;       // low border for rapidity cut
  Double_t      fRapidityCutHigh;      // high border for rapidity cut
  Double_t      fEvenDCAbinning;       // same bin width for all bins in DCA xy
  Double_t      fAlephParameters[5];   // Aleph Parameters for Bethe-Bloch

  //
  //
  //
  THnSparseF * fHistRealTracks;        //! histogram with all necessary information for real tracks
  THnSparseF * fHistMCparticles;       //! histogram with all necessary information for MC particles
  //
  TH3D       * fHistPidQA;             //! histogram for the QA of the PID
  TH2D       * fHistMult;              //! control histogram for multiplicity
  TH1D       * fHistCentrality;        //! control histogram for centrality
  //
  AliAnalysisCombinedHadronSpectra(const AliAnalysisCombinedHadronSpectra&); 
  AliAnalysisCombinedHadronSpectra& operator=(const AliAnalysisCombinedHadronSpectra&); 

  ClassDef(AliAnalysisCombinedHadronSpectra, 1); 
};

#endif
 AliAnalysisCombinedHadronSpectra.h:1
 AliAnalysisCombinedHadronSpectra.h:2
 AliAnalysisCombinedHadronSpectra.h:3
 AliAnalysisCombinedHadronSpectra.h:4
 AliAnalysisCombinedHadronSpectra.h:5
 AliAnalysisCombinedHadronSpectra.h:6
 AliAnalysisCombinedHadronSpectra.h:7
 AliAnalysisCombinedHadronSpectra.h:8
 AliAnalysisCombinedHadronSpectra.h:9
 AliAnalysisCombinedHadronSpectra.h:10
 AliAnalysisCombinedHadronSpectra.h:11
 AliAnalysisCombinedHadronSpectra.h:12
 AliAnalysisCombinedHadronSpectra.h:13
 AliAnalysisCombinedHadronSpectra.h:14
 AliAnalysisCombinedHadronSpectra.h:15
 AliAnalysisCombinedHadronSpectra.h:16
 AliAnalysisCombinedHadronSpectra.h:17
 AliAnalysisCombinedHadronSpectra.h:18
 AliAnalysisCombinedHadronSpectra.h:19
 AliAnalysisCombinedHadronSpectra.h:20
 AliAnalysisCombinedHadronSpectra.h:21
 AliAnalysisCombinedHadronSpectra.h:22
 AliAnalysisCombinedHadronSpectra.h:23
 AliAnalysisCombinedHadronSpectra.h:24
 AliAnalysisCombinedHadronSpectra.h:25
 AliAnalysisCombinedHadronSpectra.h:26
 AliAnalysisCombinedHadronSpectra.h:27
 AliAnalysisCombinedHadronSpectra.h:28
 AliAnalysisCombinedHadronSpectra.h:29
 AliAnalysisCombinedHadronSpectra.h:30
 AliAnalysisCombinedHadronSpectra.h:31
 AliAnalysisCombinedHadronSpectra.h:32
 AliAnalysisCombinedHadronSpectra.h:33
 AliAnalysisCombinedHadronSpectra.h:34
 AliAnalysisCombinedHadronSpectra.h:35
 AliAnalysisCombinedHadronSpectra.h:36
 AliAnalysisCombinedHadronSpectra.h:37
 AliAnalysisCombinedHadronSpectra.h:38
 AliAnalysisCombinedHadronSpectra.h:39
 AliAnalysisCombinedHadronSpectra.h:40
 AliAnalysisCombinedHadronSpectra.h:41
 AliAnalysisCombinedHadronSpectra.h:42
 AliAnalysisCombinedHadronSpectra.h:43
 AliAnalysisCombinedHadronSpectra.h:44
 AliAnalysisCombinedHadronSpectra.h:45
 AliAnalysisCombinedHadronSpectra.h:46
 AliAnalysisCombinedHadronSpectra.h:47
 AliAnalysisCombinedHadronSpectra.h:48
 AliAnalysisCombinedHadronSpectra.h:49
 AliAnalysisCombinedHadronSpectra.h:50
 AliAnalysisCombinedHadronSpectra.h:51
 AliAnalysisCombinedHadronSpectra.h:52
 AliAnalysisCombinedHadronSpectra.h:53
 AliAnalysisCombinedHadronSpectra.h:54
 AliAnalysisCombinedHadronSpectra.h:55
 AliAnalysisCombinedHadronSpectra.h:56
 AliAnalysisCombinedHadronSpectra.h:57
 AliAnalysisCombinedHadronSpectra.h:58
 AliAnalysisCombinedHadronSpectra.h:59
 AliAnalysisCombinedHadronSpectra.h:60
 AliAnalysisCombinedHadronSpectra.h:61
 AliAnalysisCombinedHadronSpectra.h:62
 AliAnalysisCombinedHadronSpectra.h:63
 AliAnalysisCombinedHadronSpectra.h:64
 AliAnalysisCombinedHadronSpectra.h:65
 AliAnalysisCombinedHadronSpectra.h:66
 AliAnalysisCombinedHadronSpectra.h:67
 AliAnalysisCombinedHadronSpectra.h:68
 AliAnalysisCombinedHadronSpectra.h:69
 AliAnalysisCombinedHadronSpectra.h:70
 AliAnalysisCombinedHadronSpectra.h:71
 AliAnalysisCombinedHadronSpectra.h:72
 AliAnalysisCombinedHadronSpectra.h:73
 AliAnalysisCombinedHadronSpectra.h:74
 AliAnalysisCombinedHadronSpectra.h:75
 AliAnalysisCombinedHadronSpectra.h:76
 AliAnalysisCombinedHadronSpectra.h:77
 AliAnalysisCombinedHadronSpectra.h:78
 AliAnalysisCombinedHadronSpectra.h:79
 AliAnalysisCombinedHadronSpectra.h:80
 AliAnalysisCombinedHadronSpectra.h:81
 AliAnalysisCombinedHadronSpectra.h:82
 AliAnalysisCombinedHadronSpectra.h:83
 AliAnalysisCombinedHadronSpectra.h:84
 AliAnalysisCombinedHadronSpectra.h:85
 AliAnalysisCombinedHadronSpectra.h:86
 AliAnalysisCombinedHadronSpectra.h:87
 AliAnalysisCombinedHadronSpectra.h:88
 AliAnalysisCombinedHadronSpectra.h:89
 AliAnalysisCombinedHadronSpectra.h:90
 AliAnalysisCombinedHadronSpectra.h:91
 AliAnalysisCombinedHadronSpectra.h:92