Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGCF / EBYE / LRC / AliAnalysisTaskLRC.h

#ifndef AliAnalysisTaskLRC_H
#define AliAnalysisTaskLRC_H

// Analysis task for Long Range Correlation (LRC) analysis using TPC data
// This includes a TList of AliLRCProcess objects that are processing LRC analysis
// for a given eta-phi windows

// Author : Andrey Ivanov, Igor Altsybeev, St.Peterburg State University
// Email: Igor.Altsybeev.ch


#include <AliPIDResponse.h>
#include <AliPIDCombined.h>
#include <AliAnalysisTaskSE.h>

//#define kNumberOfParentParticleClassesInMC 8

//class AliLRCProcess;
class AliLRCBase;
class AliESDtrackCuts;
class TH1D;
class TH2D;
class TH1I;
class TRandom3;
class TParticle;
//class AliSimpleEvent;
class TTree;

class TStopwatch;
//enum en_AnalysisType
//{
//    en_AnalysisType_ESD = 0,
//    en_AnalysisType_AOD

//};

class AliAnalysisTaskLRC : public AliAnalysisTaskSE {

public:
    //Constructors
    AliAnalysisTaskLRC( const char *name = "AliAnalysisTaskLRC", Bool_t runKine = kFALSE );
    virtual ~AliAnalysisTaskLRC() {}

    //AliAnalysisTaskSE overloading

    virtual void   UserCreateOutputObjects();
    virtual void   UserExec(Option_t *option);
    //    virtual void   UserExecLoop( Double_t phiAdditional = 0 );//Option_t *option);
    virtual void   Terminate(Option_t *);
    //----------------------------------

    void AddLRCProcess(AliLRCBase *newProc); //Adds new AliLRCProcess to analysis task

    // Setters
    void SetMinNumberOfSPDtracklets( Int_t MinSPDtracklets );   //Sets  min number of SPD tracklets
    void SetMaxPtLimit(Double_t MaxPtLimit);   //Sets  Max Pt filter
    void SetMinPtLimit(Double_t MinPtLimit);   //Sets  Min Pt filter
    void SetKineLowPtCut(Double_t MinKineParticlePtLimit);   //Sets  Min Pt filter for kine particles
    void SetCheckForkVtx(Bool_t CheckForkVtx){fCheckForkVtx=CheckForkVtx;} // Accept only events with veretex
    void SetCheckForVtxPosition(Bool_t CheckForVtxPosition ){fCheckForVtxPosition=CheckForVtxPosition;} //Accept only events with veretex in slected range
    void SetTrackCuts(AliESDtrackCuts* const cuts)  { fEsdTrackCuts = cuts; }
    void SetAODtrackCutBit(UInt_t bit){ fAODtrackCutBit = bit;  } //AOD track cut bit
    void SetShowEventStats(Bool_t ShowEventStats)  {fShowEventStats= ShowEventStats;}
    void SetShowPerTrackStats(Bool_t ShowPerTrackStats) {fShowPerTrackStats=ShowPerTrackStats;}
    void SetVtxDiamond(Double_t Vx, Double_t Vy, Double_t Vz) {fVxMax = Vx;fVyMax =Vy;fVzMax = Vz;}
    void SetNchCuts(Int_t minNch, Int_t maxNch){fMinAcceptedTracksCut=minNch; fMaxAcceptedTracksCut=maxNch;}

    void SetNumberOfPhiSectors(Int_t nSectors){ fNumberOfPhiSectors = nSectors; }//fNeedToRotateSector = kTRUE; }
    void SetCentralityClass(Float_t minCentralityClass, Float_t maxCentralityClass ){ fMinCentralityClass = minCentralityClass; fMaxCentralityClass = maxCentralityClass; }
    
    void SetIonsAnalysis(Bool_t isIonsFlag ){ fIsIonsAnalysis = isIonsFlag; }
    void SetEtAnalysis(Bool_t isEtAnalysisFlag ){ fEtInsteadOfPt = isEtAnalysisFlag; }
    void SetUsePhiShufflingByHand(Bool_t usePhiShufflingByHand ){ fUsePhiShufflingByHand = usePhiShufflingByHand; }
    void SetUseToyEvents(Bool_t useToyEvents ){ fUseToyEvents = useToyEvents; }
    void SetNumberOfToyEvents(Int_t nEvents ){ fNumberOfToyEvents = nEvents; }
    void SetArtificialInefficiencyCoeff( Double_t artificialInefficiencyCoeff ) { fArtificialInefficiency = artificialInefficiencyCoeff; }   //Sets coeff for artificial inefficiency

    //void SetNumberOfPhiSectorsByHand( Int_t numberOfPhiSectorsByHand ) { fNumberOfPhiSectorsByHand = numberOfPhiSectorsByHand; }

    // Getters
    TList* GetListOfProcessors() { return &fLRCproc;} // Returns list of included
    AliESDtrackCuts* GetTrackCuts() const                         { return fEsdTrackCuts; }
    AliLRCBase * Proc(Int_t index);// Get Processor i

    void SetParticleTypeForTask( TString strF, TString strB );  //( char* strF, char* strB );
    //    void SetMCparticleClassForFillingLRC( TString strParticleType ) { fStrMCparticleClassForFillingLRC = strParticleType; }
    //    void SetEtaCutsForSpecMCanalysis( double etaMin, double etaMax  ) { fEtaMCanalysisCutMin = etaMin; fEtaMCanalysisCutMax = etaMax; }

    void SetV0ACMultThreshold( int minMult ) { fThresholdOnV0mult = minMult; }

    //    void SetIncludeEventTreeInOutput( Bool_t flag ) { fSetIncludeEventTreeInOutput = flag; }
    //    Bool_t GetIncludeEventTreeInOutput() { return fSetIncludeEventTreeInOutput; }

    //    void SetAnalysisType( en_AnalysisType analysisType ) { fAnalysisType = analysisType; }
    //    en_AnalysisType GetAnalysisType() { return fAnalysisType; }

    Double_t GetEventPlane(AliVEvent *event);


    //track cuts array stuff
    void AddTrackCutForBits(AliESDtrackCuts* const cuts, TString cutsName );
    void SetUseListCuts( Bool_t const useCutsList )  { fSwitchToListingCuts = useCutsList; }
    Int_t GetNumberOfTrackCutForBits() const                         { return fArrTrackCuts.GetEntries();/*fNumberOfCutsToRemember*/; }

    void SetAnalysisLevel(const char* analysisLevel) {  fAnalysisLevel = analysisLevel;}
    const char* GetAnalysisLevel() { return fAnalysisLevel.Data(); }

    void SetMCESD( Bool_t flagMCESD ) { fAnalyseMCESD = flagMCESD; }
    Bool_t GetMCESD() const { return fAnalyseMCESD; }

    void SetFlagWatchZDC( Bool_t flagWatchZDC) {  fFlagWatchZDC = flagWatchZDC;}
    void SetFlagWatchV0 ( Bool_t flagWatchV0 ) {  fFlagWatchV0  = flagWatchV0 ;}
    void SetFlagWatchFMD( Bool_t flagWatchFMD) {  fFlagWatchFMD = flagWatchFMD;}
    Bool_t GetFlagWatchZDC() {  return fFlagWatchZDC; }
    Bool_t GetFlagWatchV0 () {  return fFlagWatchV0 ; }
    Bool_t GetFlagWatchFMD() {  return fFlagWatchFMD; }

    void FillLRCProcessors( int nTracks, Double_t eventCentrality );
    void ProfilePhiByHand( int numberOfAcceptedTracksForLRC );
    void UseToyEvents();

    void SetEtaRegionForTests( Double_t etaRegionForTests ) {  fEtaRegionForTests = etaRegionForTests; }
    Double_t GetEtaRegionForTests() {  return fEtaRegionForTests; }

    void SetMultCutInEtaRegion( Double_t multCutInEtaRegion ) {  fMultCutInEtaRegion = multCutInEtaRegion; }
    Double_t GetMultCutInEtaRegion() {  return fMultCutInEtaRegion; }

    inline void FixAngleInTwoPi( Double_t &lPhi )
    {
        if ( lPhi > 2 * TMath::Pi() )
            lPhi -= 2 * TMath::Pi();
        else if ( lPhi < 0 )
            lPhi += 2 * TMath::Pi();
    }

    void SetFlagSuppressAddingSomeHistos( Bool_t flagSuppressAddingSomeHistos ) {  fFlagSuppressAddingSomeHistos = flagSuppressAddingSomeHistos; }

    enum enTaskObjectParameters { kMaxParticlesNumber = 15000, kMaxLRCprocArrayPointers = 10000 }; // default TPC & TOF pid (via GetTPCpid & GetTOFpid)

protected:
    void SetParticleTypeToProcessors( int windowId, TString strPid );
    
    Int_t fNumberOfPhiSectors; // n of phi rotations
    Bool_t fFlagSuppressAddingSomeHistos; //flag to include in output list some histos or not include
    //    AliLRCBase *fLRCprocArrayPointers[kMaxLRCprocArrayPointers];

    // Track cuts
    TString fAnalysisLevel; //ESD, AOD or MC
    AliESDtrackCuts *fEsdTrackCuts;               // esd track cuts
    UInt_t fAODtrackCutBit;//track cut bit from track selection (only used for AODs)

    // Array with different track cuts to remember in simple event Tree
    TObjArray fArrTrackCuts;    //AliESDtrackCuts*  [100];     // Arr with different track cuts
    TString  fArrCutsNames[100];     // Arr with names of different track cuts
    TH1I    *fHistCutsNamesBins;        //!  tracks passed different cut sets in histogram bins
    Bool_t fSwitchToListingCuts;      // switch to remember cuts desicions and not to drop track by fTrackCuts


    //    en_AnalysisType fAnalysisType; // type of analysis

    //SPD tracklets cut
    Int_t fMinNumberOfSPDtracklets;   //Minimum number of SPD tracklets in ESD event

    // Acceptance cuts
    Double_t fMaxPtLimit;  //Max Pt filter
    Double_t fMinPtLimit;  // Min Pt filter
    Double_t fKineLowPtCut;  // Min Pt for Kine tracks

    // Nch cuts
    Int_t fMinAcceptedTracksCut;   //Minimum number of accepted tracks in event
    Int_t fMaxAcceptedTracksCut;   //Maximum number of accepted tracks in event

    // Vtx cuts
    Bool_t fCheckForkVtx;		// Check for vertex
    Bool_t fCheckForVtxPosition;  // Check if vertex position in range
    Double_t fVxMax;	// X vrtx max
    Double_t fVyMax;	// Y vrtx max
    Double_t fVzMax;	// Z vrtx max


    TList fLRCproc;       //  AliLRCProcess objects list
    TList* fOutList;      //! Task Output data container

    Bool_t fRunKine;      // ESD/AOD  - KINE switch
    Bool_t fAnalyseMCESD;     // MCESD switch
    Bool_t fShowEventStats; //  Allows per event debug output (trigger Nch, cuts etc)
    Bool_t fShowPerTrackStats; // Allows per track debug output


    Double_t fEtaRegionForTests; //eta region to tests
    Double_t fMultCutInEtaRegion;	// cut on mult in eta region

    // QA histos

    TH1I *fHistEventCutStats;  //! Event cut statistics
    TH1I *fHistTrackCutStats;  //! Track cut statistics
    TH1I *fHistAODTrackStats;  //! AOD track bits statistics


    TH1D *fHistVx;  //! Vx hist
    TH1D *fHistVy;  //! Vy hist
    TH1D *fHistVz;  //! Vz hist

    TH1D *fHistVxMCrecoDiff;  //! Vx hist MC-reco diff
    TH1D *fHistVyMCrecoDiff;  //! Vy hist MC-reco diff
    TH1D *fHistVzMCrecoDiff;  //! Vz hist MC-reco diff

    TH1I *fHistVertexNconributors;  //! vertex contributors number
    TH1I *fHistNumberOfPileupVerticesTracks;  //! number of pileup verteces in event (ESD or AOD) by tracks
    TH1I *fHistNumberOfPileupVerticesSPD;  //! number of pileup verteces in event (ESD or AOD) by SPD

    TH2F *fHistEventPlane; //event plane distribution


    TH1F *fHistPt; //! Overal Pt spectrum
    TH1F *fHistEta; //! Overal Eta spectrum
    TH1F *fHistEtaAODpure; //! Overal Eta spectrum aod pure before cuts
    TH1F *fHistPhi; //! Overal Phi spectrum
    TH2D *fHistEtaPhi;       //! 2D plot for checking acceptance
    TH1F *fHistPhiLRCrotationsCheck; //! Overal Phi spectrum for LRC rotations
    TH1F *fHistPhiArtificialProfilingCheck; //! hist for the check of profiled phi
    TH1F *fHistPhiArtificialProfilingCheckWrtEvPlane; //! hist for the check of profiled phi wrt event plane
    TH1F *fHistPhiArtificialEvPlane; //! hist artificial event plane

    TH2D *fHistEtaVsZvCoverage; //! Statistics on tracks Zv and Eta for all tracks
    TH2D *fHistEtaVsZvCoverageAccepted; //!  Statistics on tracks Zv and Eta for accepted tracks

    TH1D *fHistMultBeforeCuts;   //! Histo: Number of tracks before applying cuts
    TH1D *fHistAcceptedMult;   //! Number of accepted tracks histo
    TH1D *fHistAcceptedTracks;   //! Number of tracks accepted for filling LRC processors, histo
    TH1D *fHistMultiplicityInEtaRegion; //! Number of tracks in |eta| region
    TH1D *fHistMultiplicityInEtaRegionAfterPtCuts; //! Number of tracks in |eta| region after Pt Cuts
    TH2D *fHist2DMultiplicityMCESDInEtaRegion; //! Number of tracks in |eta| region (MC vs ESD)
    TH1D *fHistAcceptedTracksAfterPtCuts;   //! Number of tracks accepted for filling LRC processors, histo
    TH1D *fHistAcceptedTPCtracks;   //! Number of accepted tracks with TPC inner param
    TH1D *fHistClustersTPC;   //! Number of TPC clusters distribution
    TH1D *fHistClustersTPCafterCuts;   //! Number of TPC clusters distribution after cuts
    TH1D *fHistCrossedRowsTPC;   //! Number of TPC crossed rows
    TH1D *fHistCrossedRowsTPCafterCuts;   //! Number of TPC crossed rows after cuts
    

    TH1D *fHistClustersITS;   //! Number of ITS clusters distribution
    TH1D *fHistTrackletsITS;   //! Number of ITS tracklets distribution
    TH2D *fHist2DClustersTPCvsPt;   //! Number of TPC clusters vs Pt distribution (to see the dependence!)
    TH2D *fHist2DClustersTPCvsEta;   //! Number of TPC clusters vs Eta distribution (to see the dependence!)

    TH2D *fHist2DAcceptedTracksPtvsEta;   //! rejected tracks pt vs eta

    TH1D *fHistMClabels;   //! MC labels
    TH1D *fHistRejectedTracksCharge;   //! Charge of rejected tracks
    TH1D *fHistTracksCharge;   //! Charge of accepted tracks (zero is filled only for MC truth)

    //netcharge study
    TH1D *fHistPtPlus;   //! pt distr for +
    TH1D *fHistPtMinus;   //! pt distr for -
    TH1D *fHistNetChargeVsPt;   //! Net charge vs pt of accepted tracks
    TH1D *fHistChargePlusVsPtTmp;    //! Net charge vs pt of accepted tracks +
    TH1D *fHistChargeMinusVsPtTmp;   //! Net charge vs pt of accepted tracks -
    TH2D *fHist2DNetChargeVsPt;      //! Net charge vs pt of accepted tracks 2D
    TH2D *fHist2DNetChargeVsPtCorrectedOnEventMean;              //! Net charge vs pt of accepted tracks 2D shifted by mean
    TH2D *fHist2DNetChargeVsPtCorrectedOnEventMeanNormOnNch;     //! Net charge vs pt of accepted tracks 2D norm on nCh

    AliAnalysisTaskLRC(const AliAnalysisTaskLRC&); // not implemented
    AliAnalysisTaskLRC& operator=(const AliAnalysisTaskLRC&); // not implemented

    TH1D *fHistProbabilitiesPID;  //!hist of esd pid prob-s
    //Double_t *fProbabilitiesPID;	//! array of esd pid prob-s
    TH1D *fHistESDtrackMass;  //!hist of esd particle masses
    TH1D *fHistProbabilityPion;  //!hist of pion probability
    TH1D *fHistProbabilityKaon;  //!hist of kaon probability
    TH1D *fHistProbabilityProton;  //!hist of proton probability
    TH1D *fHistParticlesDistr;  //!hist of particles distr
    TH1D *fHistParticlesDistrBeforeCuts;  //!hist of particles distr



    TH1D *fHistCentralityPercentile;        //! centrality class
    TH1D *fHistCentralityClass10;           //! centrality class by 10
    TH1D *fHistCentralityClass5;            //! centrality class by 5


    //ZDC stuff
    TH1D *fHistZDCenergy[5];     //! ZDC energy for diff mult conditions
    TH1D *fHistZDCparticipants;             //! ZDC participants

    //V0 stuff
    TH1D *fHistV0multiplicity;     //! V0 mult
    TH1D *fHistV0Amultiplicity;     //! V0 A mult
    TH1D *fHistV0Cmultiplicity;     //! V0 C mult
    TH2D *fHist2DV0ACmultiplicity;     //! V0 A-C mult
    //TH1D *fHistV0spectra;     //! V0 particle masses
    TH2D *fHist2DTracksAcceptedVsV0multiplicity;     //! V0 mult Vs N tracks Accepted

    TH1D *fHistV0AmultiplicityRing[4];     //! V0 A mult in rings
    TH1D *fHistV0CmultiplicityRing[4];     //! V0 C mult in rings
    TH2D *fHist2DV0ACmultiplicityRing[4];     //! V0 A-C mult in rings
    TH2D *fHist2DTracksAcceptedVsV0AmultiplicityRing[4];     //! V0A mult Rings Vs N tracks Accepted
    TH2D *fHist2DTracksAcceptedVsV0CmultiplicityRing[4];     //! V0C mult Rings Vs N tracks Accepted

    TH1D *fHistV0cells         ;     //! V0 cells
    TH1D *fHistV0Acells        ;     //! V0 A cells
    TH1D *fHistV0Ccells        ;     //! V0 C cells
    TH2D *fHist2DV0ACcells     ;     //! V0 A-C cells

    Int_t fThresholdOnV0mult; //min V0AC mult to analyse this event (default is 0)


    //centrality class
    Float_t fMinCentralityClass;    // min bound on centrality percentile
    Float_t fMaxCentralityClass;    // max bound on centrality percentile


    Bool_t fIsIonsAnalysis; //Ions analysis flag
    Bool_t fEtInsteadOfPt; //pass the Et instead of Pt to LRC processors

    Bool_t fUsePhiShufflingByHand; //flag for manual suffling of tracks phi
    Bool_t fUseToyEvents; //flag for manual suffling of tracks phi
    Int_t fNumberOfToyEvents; //number of toy events

    Int_t fTmpCounter; //! TMP

    const AliPIDResponse *fPIDResponse;     //! PID response object
    AliPIDCombined       *fPIDCombined;     //! combined PID object
    TH1F *fPriors[AliPID::kSPECIES];           //! priors
    TH2D *fPriorsUsed[AliPID::kSPECIES];       //! priors used
    TH2D *fProbTPCTOF[AliPID::kSPECIES];       //! combined probabilities vs mom TPC-TOF
    TH2D *fProbAllDets[AliPID::kSPECIES];       //! combined probabilities ALL dets vs mom

    TH1D *fHistPidMaxProbability;  //!hist of max probabilities for arrays PID species
    TH1D *fHistPidPureMaxProbability;  //!hist of max probabilities for arrays PID species (when detId is TPC+TOF)

    TString fStrPIDforFwd;		//PID name for FWD win
    TString fStrPIDforBwd;		//PID name for BWD win
    Bool_t fPIDsensingFlag; 		//flag that we sense PID in processors


    int fMultForZDCstudy[5]; //! threshold multiplicities for ZDC study

    //artificial inefficiency (27.09.12)
    Double_t fArtificialInefficiency;	// inefficiency by hand in [0,1], default is 0
    TH2D *fHistNumberOfDroppedByHandTracks;   //! Number of tracks which were dropped by hand vs N of accepted tracks
    TRandom3 *fRand; //random generator for some uses

    //phi artificial gaps
    Double_t fPhiArtificialGapBegin;    // inefficiency in phi - gap position edge begins
    Double_t fPhiArtificialGapEnd;      // inefficiency in phi - gap position edge ends

    //flags for inclusion of detectors info:
    Bool_t fFlagWatchZDC;   //study ZDC issues
    Bool_t fFlagWatchV0;    //study V0 issues
    Bool_t fFlagWatchFMD;   //study FMD issues


    TStopwatch *fAnalysisTimer;


    //arrays with data for LRC processors
    Double_t fArrayTracksPt[kMaxParticlesNumber];
    Double_t fArrayTracksEta[kMaxParticlesNumber];
    Double_t fArrayTracksPhi[kMaxParticlesNumber];
    Short_t fArrayTracksCharge[kMaxParticlesNumber];
    Int_t fArrayTracksPID[kMaxParticlesNumber];


    //test MC particles
    //    TH1D *fHistMCvertexRdeltaFromParent;  //!MC R hist
    //    TH1F *fHistMCparentsStat;  //! MC parent ratios for different partile classes
    //    TH1F *fHistMCparentsEta[kNumberOfParentParticleClassesInMC];  //! MC parents eta distributions for different particle classes
    //    TH1F *fHistMCchildsEta[kNumberOfParentParticleClassesInMC];  //! MC childs eta distributions for different partile classes
    //    TH1F *fHistMCdeltaEtaChildParent[kNumberOfParentParticleClassesInMC];  //! MC delta eta b/n parent and child
    //    TH1F *fHistMCdeltaPhiChildParent[kNumberOfParentParticleClassesInMC];  //! MC delta phi b/n parent and child
    //    TH2D *fHist2DMCchildrenPhiChildParent[kNumberOfParentParticleClassesInMC];  //! MC delta b/n parent and child in eta-phi

    //    TH1F *fHistMCNumberOfChildren[kNumberOfParentParticleClassesInMC]; //! Number of children for fathers in MC (for different father classes)
    //    TH1D *fHistMCchildrenEtaDeviationsFromAverage[kNumberOfParentParticleClassesInMC];  //! MC delta b/n av. eta and each child's eta for each father (for different father classes)
    //    TH1D *fHistMCchildrenPhiDeviationsFromAverage[kNumberOfParentParticleClassesInMC];  //! MC delta b/n av. phi and each child's phi for each father (for different father classes)
    //    TH2D *fHist2DMCchildrenPhiDeviationsFromAverage[kNumberOfParentParticleClassesInMC];  //! MC delta b/n av. eta-phi and each child's eta and phi for each father (for different father classes)

    //    TString fStrMCparticleClassForFillingLRC; // name of particle class for LRC filling (default is All)
    //    Double_t fEtaMCanalysisCutMin; // spec MC analysis: cut on eta
    //    Double_t fEtaMCanalysisCutMax; // spec MC analysis: cut on eta

    //    TH1F *fHistMCparentDeepness;  //! MC deepness of parent tree from "physical primary" children
    //    TH1F *fHistMCparentsInitialStat;  //! MC initial papa particle class distr

    //    TH1F *fHistMCEtaInitialQuark;  //! MC eta distr of "initial" quarks
    //    TH1F *fHistMCEtaInitialGluon;  //! MC eta distr of "initial" gluons
    //    TH1F *fHistMCEtaInitialProton;  //! MC eta distr of "initial" protons

    //    TH1F *fHistMCnumberInitialQuarksInEvent;  //! MC initial quarks number distr
    //    TH1F *fHistMCnumberInitialGluonsInEvent;  //! MC initial gluons number distr
    //    TH1F *fHistMCnumberInitialProtonInEvent;  //! MC initial proton number distr (check that there are 2)

    //    TH1F *fHistMCnumberChildrenFromInitialQuarksInEvent;  //! MC children number from initial quarks distr
    //    TH1F *fHistMCnumberChildrenFromInitialGluonsInEvent;  //! MC children number from initial gluons distr
    //    TH1F *fHistMCnumberChildrenFromInitialProtonInEvent;  //! MC children number from initial proton distr (check that there are 2)


    // 4.01.2012: MyTree stuff
    //    AliSimpleEvent *fSimpleEvent;   // instance of simple event to be filled in analysis loop
    //    Int_t fNsimpleEvents;
    //    TTree *fEventTree;              //! event tree to write into output file
    //    Bool_t fSetIncludeEventTreeInOutput;    // flag to use event tree or not

    ClassDef(AliAnalysisTaskLRC, 11 );
};

#endif