PWG4/JetTasks/AliAnalysisTaskPhiCorrelations.h

   1 #ifndef AliAnalysisTaskPhiCorrelations_H
   2 #define AliAnalysisTaskPhiCorrelations_H
   3
   4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 ////////////////////////////////////////////////////////////////////////
   8 //
   9 // Analysis class for Underlying Event studies w.r.t. leading track
  10 //
  11 // Look for correlations on the tranverse regions w.r.t
  12 // the leading track in the event
  13 //
  14 // This class needs input AODs.
  15 // The output is a list of analysis-specific containers.
  16 //
  17 // The AOD can be either connected to the InputEventHandler
  18 // for a chain of AOD files
  19 // or
  20 // to the OutputEventHandler
  21 // for a chain of ESD files,
  22 // in this case the class should be in the train after the jet-finder
  23 //
  24 //    Authors:
  25 //    Arian Abrahantes Quintana
  26 //    Jan Fiete Grosse-Oetringhaus
  27 //    Ernesto Lopez Torres
  28 //    Sara Vallero
  29 //
  30 ////////////////////////////////////////////////////////////////////////
  31
  32 #include "AliAnalysisTask.h"
  33 #include "AliUEHist.h"
  34 #include "TString.h"
  35
  36 class AliAODEvent;
  37 class AliAnalyseLeadingTrackUE;
  38 class AliInputEventHandler;
  39 class AliMCEvent;
  40 class AliMCEventHandler;
  41 class AliUEHistograms;
  42 class AliVParticle;
  43 class TH1D;
  44 class TObjArray;
  45 class AliEventPoolManager;
  46 class AliESDEvent;
  47
  48 class  AliAnalysisTaskPhiCorrelations : public AliAnalysisTask
  49   {
  50   public:
  51     AliAnalysisTaskPhiCorrelations(const char* name="AliAnalysisTaskPhiCorrelations");
  52     virtual           ~AliAnalysisTaskPhiCorrelations();
  53
  54
  55     // Implementation of interace methods
  56     virtual     void   ConnectInputData(Option_t *);
  57     virtual     void   CreateOutputObjects();
  58     virtual     void   Exec(Option_t *option);
  59
  60     // Setters/Getters
  61     // general configuration
  62     virtual     void    SetDebugLevel( Int_t level )  { fDebug = level; }
  63     virtual     void    SetMode(Int_t mode)           { fMode  = mode;  }
  64     virtual     void    SetReduceMemoryFootprint(Bool_t flag) { fReduceMemoryFootprint = flag; }
  65     virtual     void    SetEventMixing(Bool_t flag) { fFillMixed = flag; }
  66     virtual     void    SetCompareCentralities(Bool_t flag) { fCompareCentralities = flag; }
  67     virtual     void    SetTwoTrackEfficiencyStudy(Bool_t flag) { fTwoTrackEfficiencyStudy = flag; }
  68
  69     // histogram settings
  70     void SetTrackingEfficiency( const TH1D* hist) { fkTrackingEfficiency = hist; }
  71
  72     // for event QA
  73     void   SetTracksInVertex( Int_t val ){ fnTracksVertex = val; }
  74     void   SetZVertex( Double_t val )    { fZVertex = val; }
  75
  76     // track cuts
  77     void   SetTrackEtaCut( Double_t val )    { fTrackEtaCut = val; }
  78     void   SetPtMin(Double_t val)            { fPtMin = val; }
  79     void   SetFilterBit( UInt_t val )        { fFilterBit = val;  }
  80
  81     void   SetEventSelectionBit( UInt_t val )        { fSelectBit = val;  }
  82     void   SetUseChargeHadrons( Bool_t val ) { fUseChargeHadrons = val; }
  83     void   SetSelectCharge(Int_t selectCharge) { fSelectCharge = selectCharge; }
  84     void   SetCentralityMethod(const char* method) { fCentralityMethod = method; }
  85     void   SetFillpT(Bool_t flag) { fFillpT = flag; }
  86
  87
  88   private:
  89     AliAnalysisTaskPhiCorrelations(const  AliAnalysisTaskPhiCorrelations &det);
  90     AliAnalysisTaskPhiCorrelations&   operator=(const  AliAnalysisTaskPhiCorrelations &det);
  91     void            AddSettingsTree();                                  // add list of settings to output list
  92     // Analysis methods
  93     void            AnalyseCorrectionMode();                            // main algorithm to get correction maps
  94     void            AnalyseDataMode();                                  // main algorithm to get raw distributions
  95     void            Initialize();                                       // initialize some common pointer
  96
  97
  98
  99     // General configuration
 100     Int_t               fDebug;           //  Debug flag
 101     Int_t               fMode;            //  fMode = 0: data-like analysis
 102                                           //  fMode = 1: corrections analysis
 103     Bool_t              fReduceMemoryFootprint; // reduce memory consumption by writing less debug histograms
 104     Bool_t              fFillMixed;             // enable event mixing (default: ON)
 105     Bool_t              fCompareCentralities;   // use the z vtx axis for a centrality comparison
 106     Bool_t              fTwoTrackEfficiencyStudy; // two-track efficiency study on
 107
 108     // Pointers to external UE classes
 109     AliAnalyseLeadingTrackUE*     fAnalyseUE;      //! points to class containing common analysis algorithms
 110     AliUEHistograms*  fHistos;       //! points to class to handle histograms/containers
 111     AliUEHistograms*  fHistosMixed;       //! points to class to handle mixed histograms/containers
 112
 113     const TH1D* fkTrackingEfficiency;       // used for study of bias by tracking
 114
 115     // Handlers and events
 116     AliAODEvent*             fAOD;             //! AOD Event
 117     AliESDEvent*             fESD;             //! ESD Event
 118     TClonesArray*            fArrayMC;         //! Array of MC particles
 119     AliInputEventHandler*    fInputHandler;    //! Generic InputEventHandler
 120     AliMCEvent*              fMcEvent;         //! MC event
 121     AliMCEventHandler*       fMcHandler;       //! MCEventHandler
 122     AliEventPoolManager*     fPoolMgr;         //! event pool manager
 123
 124     // Histogram settings
 125     TList*              fListOfHistos;    //  Output list of containers
 126
 127     // Event QA cuts
 128     Int_t               fnTracksVertex;        // QA tracks pointing to principal vertex
 129     Double_t            fZVertex;              // Position of Vertex in Z direction
 130     TString             fCentralityMethod;     // Method to determine centrality
 131
 132     // Track cuts
 133     Double_t            fTrackEtaCut;          // Eta cut on particles
 134     Double_t            fPtMin;                // Min pT to start correlations
 135     UInt_t              fFilterBit;            // Select tracks from an specific track cut (default 0xFF all track selected)
 136     UInt_t              fSelectBit;            // Select events according to AliAnalysisTaskJetServices bit maps
 137     Bool_t              fUseChargeHadrons;     // Only use charge hadrons
 138
 139     Int_t fSelectCharge;           // (un)like sign selection when building correlations: 0: no selection; 1: unlike sign; 2: like sign
 140     Bool_t fFillpT;                // fill sum pT instead of number density
 141
 142     ClassDef( AliAnalysisTaskPhiCorrelations, 1); // Analysis task for Underlying Event analysis w.r.t. leading track
 143   };
 144
 145 #endif
 146
 147