PWGJE/AliPWG4HighPtSpectra.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 //-----------------------------------------------------------------------
  17 // Author : Marta Verweij - UU
  18 //-----------------------------------------------------------------------
  19
  20 #ifndef ALIPWG4HIGHPTSPECTRA_H
  21 #define ALIPWG4HIGHPTSPECTRA_H
  22
  23 #include "AliAnalysisTask.h"
  24 #include "AliCFManager.h"
  25
  26 class TH1I;
  27 class TH1F;
  28 class TH1D;
  29 class TH2F;
  30 class TProfile;
  31 class TFile;
  32 class TList;
  33
  34 //class AliCFManager;
  35 class AliESDtrackCuts;
  36 class AliESDEvent;
  37 class AliESDVertex;
  38 class AliMCEvent;
  39 class AliStack;
  40 class AliGenPythiaEventHeader;
  41
  42 class AliPWG4HighPtSpectra : public AliAnalysisTask {
  43  public:
  44
  45   enum {
  46     kStepReconstructed          = 0,
  47     kStepSecondaries            = 1,
  48     kStepReconstructedMC        = 2,
  49     kStepMCAcceptance           = 3
  50   };
  51
  52   AliPWG4HighPtSpectra();
  53   AliPWG4HighPtSpectra(const Char_t* name);
  54   // AliPWG4HighPtSpectra& operator= (const AliPWG4HighPtSpectra& c);
  55   //  AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra& c);
  56   ~AliPWG4HighPtSpectra() {;};
  57
  58   // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
  59   virtual void   LocalInit();
  60   virtual void   ConnectInputData(Option_t *);
  61   virtual void   CreateOutputObjects();
  62   virtual void   Exec(Option_t *option);
  63   virtual void   Terminate(Option_t *);
  64   virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
  65
  66   Bool_t IsPbPb() {return fIsPbPb;}  //is PbPb data?
  67   Bool_t SelectEvent();    //decides if event is used for analysis
  68   Int_t CalculateCentrality(AliESDEvent *esd);
  69
  70   //Setters
  71   void SetIsPbPb(Bool_t cs)                {fIsPbPb = cs;}
  72   void SetCentralityClass(int cent)        {fCentClass=cent;}
  73   void SetTriggerMask(UInt_t t)             {fTriggerMask=t;}
  74
  75   // CORRECTION FRAMEWORK RELATED FUNCTIONS
  76   void     SetCFManagerPos(const AliCFManager* io1) {fCFManagerPos = io1;}   // global correction manager
  77   const AliCFManager * GetCFManagerPos() const {return fCFManagerPos;}           // get corr manager
  78   void     SetCFManagerNeg(const AliCFManager* io2) {fCFManagerNeg = io2;}   // global correction manager
  79   const AliCFManager * GetCFManagerNeg() const {return fCFManagerNeg;}            // get corr manager
  80
  81   //if fTrackType=0 (Global)
  82   //if fTrackType=1 (TPConly)
  83   //if fTrackType=2 (TPConly constrained)
  84   void SetTrackType(Int_t trackType) {fTrackType = trackType;}
  85   //AliESDtrackCuts setters
  86   void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
  87   void SetCutsReject(AliESDtrackCuts* trackCuts) {fTrackCutsReject = trackCuts;}
  88
  89   void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
  90
  91   // Data types
  92   Bool_t IsReadAODData()   const {return fReadAODData;}
  93   void   SetReadAODData(Bool_t flag=kTRUE) {fReadAODData=flag;}
  94
  95   static AliGenPythiaEventHeader*  GetPythiaEventHeader(AliMCEvent *mcEvent);
  96   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
  97
  98  protected:
  99   Bool_t              fReadAODData ;       // flag for AOD/ESD input files
 100   const AliCFManager  *fCFManagerPos    ;  // pointer to the CF manager for positive charged particles
 101   const AliCFManager  *fCFManagerNeg    ;  // pointer to the CF manager for negative charged particles
 102
 103   AliESDEvent *fESD;      //! ESD object
 104   AliMCEvent  *fMC;       //! MC event object
 105   AliStack    *fStack;    //! stack object
 106
 107   const AliESDVertex   *fVtx;     //! vertex object
 108
 109   UInt_t      fTriggerMask;          // Trigger mask to select events
 110   Bool_t      fIsPbPb;               // kTRUE if PbPb
 111   Int_t       fCentClass;            // Select only events from predefined centrality class
 112
 113   Int_t   fTrackType;     // Type of track to be used in analysis
 114   //AliESDtrackCuts options. Must be setted in AddTaskPWG4HighPTSpectra.C. They correspond with different steps in container.
 115   AliESDtrackCuts *fTrackCuts;           // trackCuts applied to global tracks
 116   AliESDtrackCuts *fTrackCutsReject;     // trackCuts to reject tracks (hybrid case)
 117
 118   Double_t fSigmaConstrainedMax;  // max sigma on constrained fit
 119
 120  private:
 121   AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra&);
 122   AliPWG4HighPtSpectra& operator=(const AliPWG4HighPtSpectra&);
 123
 124   Float_t fAvgTrials;             // Average number of trials
 125
 126   // Histograms
 127   //Number of events
 128   TList *fHistList;             //! List of output histograms
 129   TH1F  *fNEventAll;            //! Event counter
 130   TH1F  *fNEventSel;            //! Event counter: Selected events for analysis
 131   TH1F  *fNEventReject;         //! Book keeping of reason of rejecting events
 132
 133   TH1F *fh1Centrality;                         //! Centrality
 134
 135   TProfile*     fh1Xsec;                       //! pythia cross section and trials
 136   TH1F*         fh1Trials;                     //! trials which are added
 137   TH1F*         fh1PtHard;                     //! pt hard of the event
 138   TH1F*         fh1PtHardTrials;               //! pt hard of the event
 139
 140   TH2F *fPtRelUncertainty1PtPrim;              //! Pt vs relUncertainty1Pt for primary particles
 141   TH2F *fPtRelUncertainty1PtSec;               //! Pt vs relUncertainty1Pt for secondary particles
 142
 143   ClassDef(AliPWG4HighPtSpectra,3);
 144 };
 145
 146 #endif