PWGJE/EMCALJetTasks/UserTasks/AliAnalysisTaskRhoVnModulation.h

   1 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. */
   2 /* See cxx source for full Copyright notice */
   3 /* $Id$ */
   4
   5 #ifndef ALIANALYSISTASKRHOVNMODULATION_H
   6 #define ALIANALYSISTASKRHOVNMODULATION_H
   7
   8 #include <AliAnalysisTaskEmcalJet.h>
   9 #include <AliEmcalJet.h>
  10 #include <AliVEvent.h>
  11 #include <AliVTrack.h>
  12 #include <AliVCluster.h>
  13 #include <TClonesArray.h>
  14 #include <TMath.h>
  15 #include <TRandom3.h>
  16
  17 class TF1;
  18 class THF1;
  19 class THF2;
  20 class TProfile;
  21 class AliLocalRhoParameter;
  22
  23 class AliAnalysisTaskRhoVnModulation : public AliAnalysisTaskEmcalJet
  24 {
  25     public:
  26          // enumerators
  27         enum fitModulationType  { kNoFit, kV2, kV3, kCombined, kFourierSeries, kIntegratedFlow, kQC2, kQC4 }; // fit type
  28         enum qcRecovery         { kFixedRho, kNegativeVn, kTryFit };    // how to deal with negative cn value for qcn value
  29         enum runModeType        { kLocal, kGrid };                      // run mode type
  30         enum dataType           { kESD, kAOD, kESDMC, kAODMC };         // data type
  31         enum detectorType       { kTPC, kVZEROA, kVZEROC, kVZEROComb};  // detector that was used
  32         // constructors, destructor
  33                                 AliAnalysisTaskRhoVnModulation();
  34                                 AliAnalysisTaskRhoVnModulation(const char *name, runModeType type);
  35         virtual                 ~AliAnalysisTaskRhoVnModulation();
  36         // setting up the task and technical aspects
  37         void                    ExecOnce();
  38         Bool_t                  InitializeAnalysis();
  39         virtual void            UserCreateOutputObjects();
  40         virtual Bool_t          Run();
  41         TH1F*                   BookTH1F(const char* name, const char* x, Int_t bins, Double_t min, Double_t max, Int_t c = -1, Bool_t append = kTRUE);
  42         TH2F*                   BookTH2F(const char* name, const char* x, const char* y, Int_t binsx, Double_t minx, Double_t maxx, Int_t binsy, Double_t miny, Double_t maxy, Int_t c = -1, Bool_t append = kTRUE);
  43         /* inline */    Double_t PhaseShift(Double_t x) const {
  44             while (x>=TMath::TwoPi())x-=TMath::TwoPi();
  45             while (x<0.)x+=TMath::TwoPi();
  46             return x; }
  47         /* inline */    Double_t PhaseShift(Double_t x, Double_t n) const {
  48             x = PhaseShift(x);
  49             if(TMath::Nint(n)==2) while (x>TMath::Pi()) x-=TMath::Pi();
  50             if(TMath::Nint(n)==3) {
  51                 if(x>2.*TMath::TwoPi()/n) x = TMath::TwoPi() - x;
  52                 if(x>TMath::TwoPi()/n) x = TMath::TwoPi()-(x+TMath::TwoPi()/n);
  53             }
  54             return x; }
  55         /* inline */    Double_t ChiSquarePDF(Int_t ndf, Double_t x) const {
  56             Double_t n(ndf/2.), denom(TMath::Power(2, n)*TMath::Gamma(n));
  57             if (denom!=0)  return ((1./denom)*TMath::Power(x, n-1)*TMath::Exp(-x/2.));
  58             return -999; }
  59         // note that the cdf of the chisquare distribution is the normalized lower incomplete gamma function
  60         /* inline */    Double_t ChiSquareCDF(Int_t ndf, Double_t x) const { return TMath::Gamma(ndf/2., x/2.); }
  61         // setters - analysis setup
  62         void                    SetDebugMode(Int_t d)                           {fDebug = d;}
  63         void                    SetAttachToEvent(Bool_t b)                      {fAttachToEvent = b;}
  64         void                    SetFillHistograms(Bool_t b)                     {fFillHistograms = b;}
  65         void                    SetFillQAHistograms(Bool_t qa)                  {fFillQAHistograms = qa;}
  66         void                    SetReduceBinsXYByFactor(Float_t x, Float_t y)   {fReduceBinsXByFactor = x;
  67                                                                                  fReduceBinsYByFactor = y;}
  68         void                    SetNoEventWeightsForQC(Bool_t e)                {fNoEventWeightsForQC = e;}
  69         void                    SetCentralityClasses(TArrayI* c)                {fCentralityClasses = c;}
  70         void                    SetPtBinsHybrids(TArrayD* p)                    {fPtBinsHybrids = p;}
  71         void                    SetPtBinsJets(TArrayD* p)                       {fPtBinsJets = p;}
  72         void                    SetIntegratedFlow(TH1F* i, TH1F* j)             {fUserSuppliedV2 = i;
  73                                                                                  fUserSuppliedV3 = j; }
  74         void                    SetOnTheFlyResCorrection(TH1F* r2, TH1F* r3)    {fUserSuppliedR2 = r2;
  75                                                                                  fUserSuppliedR3 = r3; }
  76         void                    SetNameJetClones(const char* name)              {fNameJetClones = name; }
  77         void                    SetNamePicoTrackClones(const char* name)        {fNamePicoTrackClones = name; }
  78         void                    SetNameRho(const char* name)                    {fNameRho = name; }
  79         void                    SetUseScaledRho(Bool_t s)                       {fUseScaledRho = s; }
  80         void                    SetRandomSeed(TRandom3* r)                      {if (fRandom) delete fRandom; fRandom = r; }
  81         void                    SetModulationFit(TF1* fit);
  82         void                    SetModulationFitMinMaxP(Float_t m, Float_t n)   {fMinPvalue = m; fMaxPvalue = n; }
  83         void                    SetModulationFitType(fitModulationType type)    {fFitModulationType = type; }
  84         void                    SetQCnRecoveryType(qcRecovery type)             {fQCRecovery = type; }
  85         void                    SetModulationFitOptions(TString opt)            {fFitModulationOptions = opt; }
  86         void                    SetReferenceDetector(detectorType type)         {fDetectorType = type; }
  87         void                    SetUsePtWeight(Bool_t w)                        {fUsePtWeight = w; }
  88         void                    SetRunModeType(runModeType type)                {fRunModeType = type; }
  89         void                    SetAbsVertexZ(Float_t v)                        {fAbsVertexZ = v; }
  90         void                    SetMinDistanceRctoLJ(Float_t m)                 {fMinDisanceRCtoLJ = m; }
  91         void                    SetRandomConeRadius(Float_t r)                  {fRandomConeRadius = r; }
  92         void                    SetMaxNoRandomCones(Int_t m)                    {fMaxCones = m; }
  93         void                    SetMinLeadingHadronPt(Double_t m)               {fMinLeadingHadronPt = m; }
  94         void                    SetSetPtSub(Bool_t s)                           {fSubtractJetPt = s;}
  95         void                    SetForceAbsVnHarmonics(Bool_t f)                {fAbsVnHarmonics = f; }
  96         void                    SetExcludeLeadingJetsFromFit(Float_t n)         {fExcludeLeadingJetsFromFit = n; }
  97         void                    SetRebinSwapHistoOnTheFly(Bool_t r)             {fRebinSwapHistoOnTheFly = r; }
  98         void                    SetSaveThisPercentageOfFits(Float_t p)          {fPercentageOfFits = p; }
  99         void                    SetUseV0EventPlaneFromHeader(Bool_t h)          {fUseV0EventPlaneFromHeader = h;}
 100         void                    SetExplicitOutlierCutForYear(Int_t y)           {fExplicitOutlierCut = y;}
 101         // getters - these are used as well by AliAnalyisTaskJetFlow, so be careful when changing them
 102         TString                 GetJetsName() const                             {return fJetsName; }
 103         TString                 GetTracksName() const                           {return fTracksName; }
 104         TString                 GetLocalRhoName() const                         {return fLocalRhoName; }
 105         TArrayI*                GetCentralityClasses() const                    {return fCentralityClasses;}
 106         TArrayD*                GetPtBinsHybrids() const                        {return fPtBinsHybrids; }
 107         TArrayD*                GetPtBinsJets() const                           {return fPtBinsJets; }
 108         TProfile*               GetResolutionParameters(Int_t h, Int_t c) const {return (h==2) ? fProfV2Resolution[c] : fProfV3Resolution[c];}
 109         TList*                  GetOutputList() const                           {return fOutputList;}
 110         AliLocalRhoParameter*   GetLocalRhoParameter() const                    {return fLocalRho;}
 111         Double_t                GetJetRadius() const                            {return fJetRadius;}
 112         void                    ExecMe()                                        {ExecOnce();}
 113         AliAnalysisTaskRhoVnModulation* ReturnMe()                              {return this;}
 114         // local cuts
 115         void                    SetLocalJetMinMaxEta(Float_t min, Float_t max)  {fLocalJetMinEta = min; fLocalJetMaxEta = max;}
 116         void                    SetLocalJetMinMaxEta(Float_t R)                 {fLocalJetMinEta = - 0.9 + R; fLocalJetMaxEta = 0.9 - R; }
 117         void                    SetLocalJetMinMaxPhi(Float_t min, Float_t max)  {fLocalJetMinPhi = min; fLocalJetMaxEta = max;}
 118         void                    SetSoftTrackMinMaxPt(Float_t min, Float_t max)  {fSoftTrackMinPt = min; fSoftTrackMaxPt = max;}
 119         // numerical evaluations
 120         void                    CalculateEventPlaneVZERO(Double_t vzero[2][2]) const;
 121         void                    CalculateEventPlaneTPC(Double_t* tpc);
 122         void                    CalculateEventPlaneCombinedVZERO(Double_t* comb) const;
 123         void                    CalculateEventPlaneResolution(Double_t vzero[2][2], Double_t* vzeroComb, Double_t* tpc);
 124         Double_t                CalculateEventPlaneChi(Double_t resEP) const;
 125         void                    CalculateRandomCone(Float_t &pt, Float_t &eta, Float_t &phi, AliEmcalJet* jet = 0x0, Bool_t randomize = 0) const;
 126         Double_t                CalculateQC2(Int_t harm);
 127         Double_t                CalculateQC4(Int_t harm);
 128         // helper calculations for the q-cumulant analysis, also used by AliAnalyisTaskJetFlow
 129         void                    QCnQnk(Int_t n, Int_t k, Double_t &reQ, Double_t &imQ);
 130         void                    QCnDiffentialFlowVectors(
 131             TClonesArray* pois, TArrayD* ptBins, Bool_t vpart, Double_t* repn, Double_t* impn,
 132             Double_t *mp, Double_t *reqn, Double_t *imqn, Double_t* mq, Int_t n);
 133         Double_t                QCnS(Int_t i, Int_t j);
 134         Double_t                QCnM();
 135         Double_t                QCnM11();
 136         Double_t                QCnM1111();
 137         Bool_t                  QCnRecovery(Double_t psi2, Double_t psi3);
 138         // analysis details
 139         Bool_t                  CorrectRho(Double_t psi2, Double_t psi3);
 140         // event and track selection, also used by AliAnalyisTaskJetFlow
 141         /* inline */    Bool_t PassesCuts(const AliVTrack* track) const {
 142             if(!track) return kFALSE;
 143             return (track->Pt() < fTrackPtCut || track->Eta() < fTrackMinEta || track->Eta() > fTrackMaxEta || track->Phi() < fTrackMinPhi || track->Phi() > fTrackMaxPhi) ? kFALSE : kTRUE; }
 144         /* inline */    Bool_t PassesCuts(AliEmcalJet* jet) const {
 145             if(!jet || fJetRadius <= 0) return kFALSE;
 146             return (GetLeadingHadronPt(jet) < fMinLeadingHadronPt || jet->Pt() < fJetPtCut || jet->Area()/(fJetRadius*fJetRadius*TMath::Pi()) < fPercAreaCut || jet->Eta() < fJetMinEta || jet->Eta() > fJetMaxEta || jet->Phi() < fJetMinPhi || jet->Phi() > fJetMaxPhi) ? kFALSE : kTRUE; }
 147         Bool_t                  PassesCuts(AliVEvent* event);
 148         Bool_t                  PassesCuts(Int_t year);
 149         Bool_t                  PassesCuts(const AliVCluster* track) const;
 150         // filling histograms
 151         void                    FillHistogramsAfterSubtraction(Double_t psi2, Double_t psi3, Double_t vzero[2][2], Double_t* vzeroComb, Double_t* tpc) const;
 152         void                    FillTrackHistograms() const;
 153         void                    FillClusterHistograms() const;
 154         void                    FillCorrectedClusterHistograms() const;
 155         void                    FillEventPlaneHistograms(Double_t vzero[2][2], Double_t* vzeroComb, Double_t* tpc) const;
 156         void                    FillRhoHistograms() const;
 157         void                    FillDeltaPtHistograms(Double_t psi2, Double_t psi3) const;
 158         void                    FillJetHistograms(Double_t psi2, Double_t psi3) const;
 159         void                    FillQAHistograms(AliVTrack* vtrack) const;
 160         void                    FillQAHistograms(AliVEvent* vevent);
 161         void                    FillAnalysisSummaryHistogram() const;
 162         virtual void            Terminate(Option_t* option);
 163         // interface methods for the output file
 164         void                    SetOutputList(TList* l) {fOutputList = l;}
 165         TH1F*                   GetResolutionFromOuptutFile(detectorType detector, Int_t h = 2, TArrayD* c = 0x0);
 166         TH1F*                   CorrectForResolutionDiff(TH1F* v, detectorType detector, TArrayD* cen, Int_t c, Int_t h = 2);
 167         TH1F*                   CorrectForResolutionInt(TH1F* v, detectorType detector, TArrayD* cen, Int_t h = 2);
 168         TH1F*                   GetDifferentialQC(TProfile* refCumulants, TProfile* diffCumlants, TArrayD* ptBins, Int_t h);
 169     private:
 170         // analysis flags and settings
 171         Int_t                   fDebug;                 // debug level (0 none, 1 fcn calls, 2 verbose)
 172         Bool_t                  fLocalInit;             //! is the analysis initialized?
 173         Bool_t                  fAttachToEvent;         // attach local rho to the event
 174         Bool_t                  fFillHistograms;        // fill histograms
 175         Bool_t                  fFillQAHistograms;      // fill qa histograms
 176         Float_t                 fReduceBinsXByFactor;   // reduce the bins on x-axis of histo's by this much
 177         Float_t                 fReduceBinsYByFactor;   // reduce the bins on y-axis of histo's by this much
 178         Bool_t                  fNoEventWeightsForQC;   // don't store event weights for qc analysis
 179         TArrayI*                fCentralityClasses;     //-> centrality classes (maximum 10)
 180         TArrayD*                fPtBinsHybrids;         //-> pt bins for hybrid track vn anaysis
 181         TArrayD*                fPtBinsJets;            //-> pt bins for jet vn analysis
 182         TH1F*                   fUserSuppliedV2;        // histo with integrated v2
 183         TH1F*                   fUserSuppliedV3;        // histo with integrated v3
 184         TH1F*                   fUserSuppliedR2;        // correct the extracted v2 with this r
 185         TH1F*                   fUserSuppliedR3;        // correct the extracted v3 with this r
 186         // members
 187         Bool_t                  fUseScaledRho;          // use scaled rho
 188         Int_t                   fNAcceptedTracks;       //! number of accepted tracks
 189         Int_t                   fNAcceptedTracksQCn;    //! accepted tracks for QCn
 190         fitModulationType       fFitModulationType;     // fit modulation type
 191         qcRecovery              fQCRecovery;            // recovery type for e-by-e qc method
 192         Bool_t                  fUsePtWeight;           // use dptdphi instead of dndphi
 193         detectorType            fDetectorType;          // type of detector used for modulation fit
 194         TString                 fFitModulationOptions;  // fit options for modulation fit
 195         runModeType             fRunModeType;           // run mode type
 196         dataType                fDataType;              // datatype
 197         TRandom3*               fRandom;                //-> dont use gRandom to not interfere with other tasks
 198         Int_t                   fMappedRunNumber;       //! mapped runnumer (for QA)
 199         Int_t                   fInCentralitySelection; //! centrality bin
 200         TF1*                    fFitModulation;         //-> modulation fit for rho
 201         Float_t                 fMinPvalue;             // minimum value of p
 202         Float_t                 fMaxPvalue;             // maximum value of p
 203         const char*             fNameJetClones;         //! collection of tclones array with jets
 204         const char*             fNamePicoTrackClones;   //! collection of tclones with pico tracks
 205         const char*             fNameRho;               //! name of rho
 206         // additional jet cuts (most are inherited)
 207         Float_t                 fLocalJetMinEta;        // local eta cut for jets
 208         Float_t                 fLocalJetMaxEta;        // local eta cut for jets
 209         Float_t                 fLocalJetMinPhi;        // local phi cut for jets
 210         Float_t                 fLocalJetMaxPhi;        // local phi cut for jets
 211         Float_t                 fSoftTrackMinPt;        // min pt for soft tracks
 212         Float_t                 fSoftTrackMaxPt;        // max pt for soft tracks
 213         // event cuts
 214         Float_t                 fAbsVertexZ;            // cut on zvertex
 215         // general qa histograms
 216         TH1F*                   fHistCentrality;        //! accepted centrality
 217         TH1F*                   fHistVertexz;           //! accepted verte
 218         TH2F*                   fHistRunnumbersPhi;     //! run numbers averaged phi
 219         TH2F*                   fHistRunnumbersEta;     //! run numbers averaged eta
 220         TH1F*                   fHistPvaluePDF;         //! pdf value of chisquare p
 221         TH1F*                   fHistPvalueCDF;         //! cdf value of chisquare p
 222         // general settings
 223         Float_t                 fMinDisanceRCtoLJ;      // min distance between rc and leading jet
 224         Float_t                 fRandomConeRadius;      // radius of random cone
 225         Int_t                   fMaxCones;              // max number of random cones
 226         Bool_t                  fAbsVnHarmonics;        // force postive local rho
 227         Float_t                 fExcludeLeadingJetsFromFit;    // exclude n leading jets from fit
 228         Bool_t                  fRebinSwapHistoOnTheFly;       // rebin swap histo on the fly
 229         Float_t                 fPercentageOfFits;      // save this percentage of fits
 230         Bool_t                  fUseV0EventPlaneFromHeader;             // use the vzero event plane from the header
 231         Int_t                   fExplicitOutlierCut;    // cut on correlation of tpc and global multiplicity
 232         Double_t                fMinLeadingHadronPt;    // minimum pt for leading hadron
 233         Bool_t                  fSubtractJetPt;         // save subtracted jet pt by calling SetPtSub
 234         // transient object pointers
 235         TList*                  fOutputList;            //! output list
 236         TList*                  fOutputListGood;        //! output list for local analysis
 237         TList*                  fOutputListBad;         //! output list for local analysis
 238         TH1F*                   fHistAnalysisSummary;   //! analysis summary
 239         TH1F*                   fHistSwap;              //! swap histogram
 240         TProfile*               fProfV2;                //! extracted v2
 241         TProfile*               fProfV2Cumulant;        //! v2 cumulant
 242         TProfile*               fProfV2Resolution[10];  //! resolution parameters for v2
 243         TProfile*               fProfV3;                //! extracted v3
 244         TProfile*               fProfV3Cumulant;        //! v3 cumulant
 245         TProfile*               fProfV3Resolution[10];  //! resolution parameters for v3
 246         // qa histograms for accepted pico tracks
 247         TH1F*                   fHistPicoTrackPt[10];    //! pt of all charged tracks
 248         TH1F*                   fHistPicoTrackMult[10];  //! multiplicity of accepted pico tracks
 249         TH2F*                   fHistPicoCat1[10];       //! pico tracks spd hit and refit
 250         TH2F*                   fHistPicoCat2[10];       //! pico tracks wo spd hit w refit, constrained
 251         TH2F*                   fHistPicoCat3[10];       //! pico tracks wo spd hit wo refit, constrained
 252         // qa histograms for accepted emcal clusters
 253         /* TH1F*                   fHistClusterPt[10];      //! pt uncorrected emcal clusters */
 254         /* TH1F*                   fHistClusterPhi[10];     //! phi uncorrected emcal clusters */
 255         /* TH1F*                   fHistClusterEta[10];     //! eta uncorrected emcal clusters */
 256         // qa histograms for accepted emcal clusters aftehadronic correction
 257         /* TH1F*                   fHistClusterCorrPt[10];  //! pt corrected emcal clusters */
 258         /* TH1F*                   fHistClusterCorrPhi[10]; //! phi corrected emcal clusters */
 259         /* TH1F*                   fHistClusterCorrEta[10]; //! eta corrected emcal clusters */
 260         // qa event planes
 261         TProfile*               fHistPsiControl;         //! event plane control histogram
 262         TProfile*               fHistPsiSpread;          //! event plane spread histogram
 263         TH1F*                   fHistPsiVZEROA;          //! psi 2 from vzero a
 264         TH1F*                   fHistPsiVZEROC;          //! psi 2 from vzero c
 265         TH1F*                   fHistPsiVZERO;           //! psi 2 from combined vzero
 266         TH1F*                   fHistPsiTPC;             //! psi 2 from tpc
 267         // background
 268         TH1F*                   fHistRhoPackage[10];     //! rho as estimated by emcal jet package
 269         TH1F*                   fHistRho[10];            //! background
 270         TH2F*                   fHistRhoVsMult;          //! rho versus multiplicity
 271         TH2F*                   fHistRhoVsCent;          //! rho veruss centrality
 272         TH2F*                   fHistRhoAVsMult;         //! rho * A vs multiplicity for all jets
 273         TH2F*                   fHistRhoAVsCent;         //! rho * A vs centrality for all jets
 274         // delta pt distributions
 275         TH2F*                   fHistRCPhiEta[10];              //! random cone eta and phi
 276         TH2F*                   fHistRhoVsRCPt[10];             //! rho * A vs rcpt
 277         TH1F*                   fHistRCPt[10];                  //! rcpt
 278         TH2F*                   fHistDeltaPtDeltaPhi2[10];      //! dpt vs dphi (psi2 - phi)
 279         TH2F*                   fHistDeltaPtDeltaPhi3[10];      //! dpt vs dphi (psi3 - phi)
 280         TH2F*                   fHistRCPhiEtaExLJ[10];          //! random cone eta and phi, excl leading jet
 281         TH2F*                   fHistRhoVsRCPtExLJ[10];         //! rho * A vs rcpt, excl leading jet
 282         TH1F*                   fHistRCPtExLJ[10];              //! rcpt, excl leading jet
 283         TH2F*                   fHistDeltaPtDeltaPhi2ExLJ[10];  //! dpt vs dphi, excl leading jet
 284         TH2F*                   fHistDeltaPtDeltaPhi3ExLJ[10];  //! dpt vs dphi, excl leading jet
 285         /* TH2F*                   fHistRCPhiEtaRand[10];          //! random cone eta and phi, randomized */
 286         /* TH2F*                   fHistRhoVsRCPtRand[10];         //! rho * A vs rcpt, randomized */
 287         /* TH1F*                   fHistRCPtRand[10];              //! rcpt, randomized */
 288         /* TH2F*                   fHistDeltaPtDeltaPhi2Rand[10];  //! dpt vs dphi, randomized */
 289         /* TH2F*                   fHistDeltaPtDeltaPhi3Rand[10];  //! dpt vs dphi, randomized */
 290         // jet histograms (after kinematic cuts)
 291         TH1F*                   fHistJetPtRaw[10];              //! jet pt - no background subtraction
 292         TH1F*                   fHistJetPt[10];                 //! pt of found jets (background subtracted)
 293         TH2F*                   fHistJetEtaPhi[10];             //! eta and phi correlation
 294         TH2F*                   fHistJetPtArea[10];             //! jet pt versus area
 295         TH2F*                   fHistJetPtConstituents[10];     //! jet pt versus number of constituents
 296         TH2F*                   fHistJetEtaRho[10];             //! jet eta versus jet rho
 297         // in plane, out of plane jet spectra
 298         TH2F*                   fHistJetPsi2Pt[10];             //! psi tpc versus pt
 299         TH2F*                   fHistJetPsi3Pt[10];             //! psi vzeroc versus pt
 300
 301         AliAnalysisTaskRhoVnModulation(const AliAnalysisTaskRhoVnModulation&);                  // not implemented
 302         AliAnalysisTaskRhoVnModulation& operator=(const AliAnalysisTaskRhoVnModulation&);       // not implemented
 303
 304         ClassDef(AliAnalysisTaskRhoVnModulation, 15);
 305 };
 306
 307 #endif