ANALYSIS/AliCentralitySelectionTask.h

   1 #ifndef ALICENTRALITYSELECTIONTASK_H
   2 #define ALICENTRALITYSELECTIONTASK_H
   3
   4 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 //*****************************************************
   8 //   Class AliCentralitySelectionTask
   9 //   author: Alberica Toia
  10 //*****************************************************
  11
  12 #include "AliAnalysisTaskSE.h"
  13
  14 class TFile;
  15 class TH1F;
  16 class TH2F;
  17 class TList;
  18 class TString;
  19
  20 class AliESDEvent;
  21 class AliESDtrackCuts;
  22
  23 class AliCentralitySelectionTask : public AliAnalysisTaskSE {
  24
  25  public:
  26
  27   AliCentralitySelectionTask();
  28   AliCentralitySelectionTask(const char *name);
  29   AliCentralitySelectionTask& operator= (const AliCentralitySelectionTask& ana);
  30   AliCentralitySelectionTask(const AliCentralitySelectionTask& c);
  31   virtual ~AliCentralitySelectionTask();
  32
  33   // Implementation of interface methods
  34   virtual void UserCreateOutputObjects();
  35   virtual void UserExec(Option_t *option);
  36   virtual void Terminate(Option_t *option);
  37
  38   virtual void  SetDebugLevel(Int_t level) {fDebug = level;}
  39   void SetInput(const char* input)         {fAnalysisInput = input;}
  40   void SetMCInput()                        {fIsMCInput = kTRUE;}
  41   void DontUseScaling()                    {fUseScaling=kFALSE;}
  42
  43   void ReadCentralityHistos(TString filename);
  44   void ReadCentralityHistos2(TString filename);
  45  private:
  46
  47   Int_t SetupRun(AliESDEvent* esd);
  48   Bool_t IsOutlierV0MSPD(Float_t spd, Float_t v0, Int_t cent);
  49   Bool_t IsOutlierV0MTPC(Int_t tracks, Float_t v0, Int_t cent);
  50   Bool_t IsOutlierV0MZDC(Float_t zdc, Float_t v0);
  51   Float_t MyGetScaleFactor(Int_t runnumber, Int_t flag);
  52   void MyInitScaleFactor();
  53   Float_t MyGetScaleFactorMC(Int_t runnumber);
  54   void MyInitScaleFactorMC();
  55
  56   Int_t    fDebug;              // Debug flag
  57   TString  fAnalysisInput;      // "ESD", "AOD"
  58   Bool_t   fIsMCInput;          // true when input is MC
  59   TFile   *fFile;               // file that holds the centrality vs multiplicity 1d
  60   TFile   *fFile2;              // file that holds the centrality vs multiplicity 2d
  61   Int_t    fCurrentRun;         // current run number
  62   Int_t    fRunNo;              // reference run number
  63   Int_t    fLowRunN;            // first run
  64   Int_t    fHighRunN;           // last run
  65   Bool_t   fUseScaling;
  66   Float_t V0MScaleFactor[2012]; // number of runs in PbPb 2010
  67   Float_t SPDScaleFactor[2012]; // number of runs in PbPb 2010
  68   Float_t TPCScaleFactor[2012]; // number of runs in PbPb 2010
  69   Float_t V0MScaleFactorMC[2012]; // number of runs in PbPb 2010
  70
  71   AliESDtrackCuts* fTrackCuts;  //! optional track cuts
  72
  73   Float_t  fZVCut;              //! z-vertex cut (in cm)
  74   Float_t  fOutliersCut;        //! outliers cut (in n-sigma)
  75   Int_t    fQuality;            //! quality for centrality determination
  76
  77   Float_t  fCentV0M;            // percentile centrality from V0
  78   Float_t  fCentFMD;            // percentile centrality from FMD
  79   Float_t  fCentTRK;            // percentile centrality from tracks
  80   Float_t  fCentTKL;            // percentile centrality from tracklets
  81   Float_t  fCentCL0;            // percentile centrality from clusters in layer 0
  82   Float_t  fCentCL1;            // percentile centrality from clusters in layer 0
  83   Float_t  fCentV0MvsFMD;       // percentile centrality from V0 vs FMD
  84   Float_t  fCentTKLvsV0M;       // percentile centrality from tracklets vs V0
  85   Float_t  fCentZEMvsZDC;       // percentile centrality from ZEM vs ZDC
  86
  87   TH1F    *fHtempV0M;           // histogram with centrality vs multiplicity using V0
  88   TH1F    *fHtempFMD;           // histogram with centrality vs multiplicity using FMD
  89   TH1F    *fHtempTRK;           // histogram with centrality vs multiplicity using tracks
  90   TH1F    *fHtempTKL;           // histogram with centrality vs multiplicity using tracklets
  91   TH1F    *fHtempCL0;           // histogram with centrality vs multiplicity using clusters in layer 0
  92   TH1F    *fHtempCL1;           // histogram with centrality vs multiplicity using clusters in layer 0
  93   TH1F    *fHtempV0MvsFMD;           // histogram with centrality vs multiplicity using V0 vs FMD
  94   TH1F    *fHtempTKLvsV0M;           // histogram with centrality vs multiplicity using tracklets vs V0
  95   TH1F    *fHtempZEMvsZDC;           // histogram with centrality vs multiplicity using ZEM vs ZDC
  96
  97   TList   *fOutputList; // output list
  98
  99   TH1F *fHOutCentV0M     ;    //control histogram for centrality
 100   TH1F *fHOutCentFMD     ;    //control histogram for centrality
 101   TH1F *fHOutCentTRK     ;    //control histogram for centrality
 102   TH1F *fHOutCentTKL     ;    //control histogram for centrality
 103   TH1F *fHOutCentCL0     ;    //control histogram for centrality
 104   TH1F *fHOutCentCL1     ;    //control histogram for centrality
 105   TH1F *fHOutCentV0MvsFMD;    //control histogram for centrality
 106   TH1F *fHOutCentTKLvsV0M;    //control histogram for centrality
 107   TH1F *fHOutCentZEMvsZDC;    //control histogram for centrality
 108
 109   TH1F *fHOutMultV0M ;        //control histogram for multiplicity
 110   TH1F *fHOutMultV0R ;        //control histogram for multiplicity
 111   TH1F *fHOutMultFMD ;        //control histogram for multiplicity
 112   TH1F *fHOutMultTRK ;        //control histogram for multiplicity
 113   TH1F *fHOutMultTKL ;        //control histogram for multiplicity
 114   TH1F *fHOutMultCL0 ;        //control histogram for multiplicity
 115   TH1F *fHOutMultCL1 ;        //control histogram for multiplicity
 116
 117   TH2F *fHOutMultV0MvsZDN;    //control histogram for multiplicity
 118   TH2F *fHOutMultZEMvsZDN;    //control histogram for multiplicity
 119   TH2F *fHOutMultV0MvsZDC;    //control histogram for multiplicity
 120   TH2F *fHOutMultZEMvsZDC;    //control histogram for multiplicity
 121   TH2F *fHOutMultV0MvsCL1;    //control histogram for multiplicity
 122   TH2F *fHOutMultV0MvsTRK;    //control histogram for multiplicity
 123   TH2F *fHOutMultTRKvsCL1;    //control histogram for multiplicity
 124
 125   TH1F *fHOutCentV0M_qual1     ;    //control histogram for centrality quality 1
 126   TH1F *fHOutCentTRK_qual1     ;    //control histogram for centrality quality 1
 127   TH1F *fHOutCentCL1_qual1     ;    //control histogram for centrality quality 1
 128
 129   TH1F *fHOutCentV0M_qual2     ;    //control histogram for centrality quality 2
 130   TH1F *fHOutCentTRK_qual2     ;    //control histogram for centrality quality 2
 131   TH1F *fHOutCentCL1_qual2     ;    //control histogram for centrality quality 2
 132
 133   TH1F *fHOutQuality ;        //control histogram for quality
 134   TH1F *fHOutVertex ;        //control histogram for vertex
 135
 136   ClassDef(AliCentralitySelectionTask, 5);
 137 };
 138
 139 #endif
 140