AliAnalysisTaskGammaCaloDalitzV1.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKGAMMACALODALITZV1_cxx 
#define ALIANALYSISTASKGAMMACALODALITZV1_cxx

#include "AliAnalysisTaskSE.h"
#include "AliESDtrack.h"
#include "AliV0ReaderV1.h"
#include "AliKFConversionPhoton.h"
#include "AliDalitzElectronSelector.h"
#include "AliGammaConversionAODBGHandler.h"
#include "AliConversionAODBGHandlerRP.h"
#include "AliCaloPhotonCuts.h"
#include "AliConvEventCuts.h"
#include "AliConversionPhotonCuts.h"
#include "AliConversionMesonCuts.h"
#include "AliDalitzElectronCuts.h"
#include "AliAnalysisManager.h"
#include "TProfile2D.h"
#include "TH3.h"
#include "TH3F.h"
#include <vector>

class AliAnalysisTaskGammaCaloDalitzV1 : public AliAnalysisTaskSE {
  public:

    AliAnalysisTaskGammaCaloDalitzV1();
    AliAnalysisTaskGammaCaloDalitzV1(const char *name);
    virtual ~AliAnalysisTaskGammaCaloDalitzV1();

    virtual void   UserCreateOutputObjects();
    virtual Bool_t Notify();
    virtual void   UserExec(Option_t *);
    virtual void   Terminate(const Option_t*);
    void InitBack();

        void SetV0ReaderName(TString name){fV0ReaderName=name; return;}
    void SetIsHeavyIon(Int_t flag){
      fIsHeavyIon = flag;    
    }

    // base functions for selecting photon and meson candidates in reconstructed data
    void ProcessClusters();
    void ProcessPhotonCandidates();
    void ProcessElectronCandidates();
    void CalculatePi0DalitzCandidates();
    
    // MC functions
    void SetIsMC(Bool_t isMC){fIsMC=isMC;}
    void ProcessMCParticles();
    void ProcessAODMCParticles();
    void RelabelAODPhotonCandidates(Bool_t mode);
    void ProcessTruePhotonCandidates( AliAODConversionPhoton* TruePhotonCandidate);
    void ProcessTrueClusterCandidates( AliAODConversionPhoton* TruePhotonCandidate);
    void ProcessTrueClusterCandidatesAOD( AliAODConversionPhoton* TruePhotonCandidate);
    void ProcessTruePhotonCandidatesAOD( AliAODConversionPhoton* TruePhotonCandidate);
    void ProcessTrueMesonCandidates( AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate0, AliAODConversionPhoton *TrueGammaCandidate1, Bool_t matched);
    void ProcessTrueMesonCandidatesAOD(AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate0, AliAODConversionPhoton *TrueGammaCandidate1, Bool_t matched);
    void ProcessConversionPhotonsForMissingTags();
    void ProcessConversionPhotonsForMissingTagsAOD();
    
    // switches for additional analysis streams or outputs
    void SetDoMesonAnalysis(Bool_t flag){fDoMesonAnalysis = flag;}
    void SetDoMesonQA(Int_t flag){fDoMesonQA = flag;}
    void SetDoPhotonQA(Int_t flag){fDoPhotonQA = flag;}
    void SetDoClusterQA(Int_t flag){fDoClusterQA = flag;}
    void SetUseTHnSparse(Bool_t flag){fDoTHnSparse = flag;}
    
      // Setting the cut lists for the conversion photons
    void SetEventCutList(Int_t nCuts, TList *CutArray){
      fnCuts = nCuts;
      fEventCutArray = CutArray;
    }

    // Setting the cut lists for the conversion photons
    void SetConversionCutList(Int_t nCuts, TList *CutArray){
      fnCuts = nCuts;
      fGammaCutArray = CutArray;
    }

      // Setting the cut lists for the calo photons
    void SetCaloCutList(Int_t nCuts, TList *CutArray){
      fnCuts = nCuts;
      fClusterCutArray = CutArray;
    }
    
    // Setting the cut lists for the meson
    void SetMesonCutList(Int_t nCuts, TList *CutArray){
      fnCuts = nCuts;
      fMesonCutArray = CutArray;
    }
        
    void SetElectronCutList(Int_t nCuts, TList *CutArray){
      
      fnCuts = nCuts;
      fElectronCutArray = CutArray;
    }
    
    // BG HandlerSettings
    void CalculateBackground();
    //void CalculateBackgroundRP();
    void RotateParticle(AliAODConversionPhoton *gamma);
    void RotateParticleAccordingToEP(AliAODConversionPhoton *gamma, Double_t previousEventEP, Double_t thisEventEP);
    void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
    void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
    void UpdateEventByEventData();
    
    // Additional functions for convenience
    void SetLogBinningXTH2(TH2* histoRebin);
    Int_t GetSourceClassification(Int_t daughter, Int_t pdgCode);
    Bool_t CheckIfContainedInString(TString input, Int_t tobechecked);
    Bool_t CheckIfContainedInStringAndAppend(TString &input, Int_t tobechecked);
    Bool_t IsPi0DalitzDaughter( Int_t label ) const;
    Double_t GetPsiPair( const AliESDtrack *trackPos, const AliESDtrack *trackNeg ) const;
    Bool_t IsDalitz(TParticle *fMCMother) const;
    Int_t FindMotherOfPhoton(Int_t particleLabel );

    Bool_t CheckVectorForDoubleCount(vector<Int_t> &vec, Int_t tobechecked);
  
        void SetTrackMatcherRunningMode(Int_t mode){fTrackMatcherRunningMode = mode;}

  protected:
    AliV0ReaderV1             *fV0Reader;             // basic photon Selection Task
        TString                             fV0ReaderName;
    AliDalitzElectronSelector       *fElecSelector;             // basic electron Selection     
    AliGammaConversionAODBGHandler    **fBGClusHandler;         // BG handler for Cluster
    AliVEvent               *fInputEvent;           // current event
    AliMCEvent              *fMCEvent;              // corresponding MC event
    TList                 **fCutFolder;           // Array of lists for containers belonging to cut
    TList                 **fESDList;             // Array of lists with histograms with reconstructed properties   
    TList                 **fBackList;            // Array of lists with BG THnSparseF
    TList                 **fMotherList;            // Array of lists with Signal THnSparseF
    TList                 **fPhotonDCAList;         // Array of lists with photon dca trees
    TList                 **fTrueList;            // Array of lists with histograms with MC validated reconstructed properties
    TList                 **fMCList;              // Array of lists with histograms with pure MC information
    TList                 **fHeaderNameList;          // Array of lists with header names for MC header selection
    TList               **fClusterOutputList;           
    TList                 *fOutputContainer;          // Output container
    TClonesArray            *fReaderGammas;           // Array with conversion photons selected by V0Reader Cut
    vector<Int_t>               fSelectorElectronIndex;                         // Array of electron candidates selected by Electron Selector
    vector<Int_t>               fSelectorPositronIndex;                         // Array of positron candidates selected by Positron Selector 
    TList                 *fGammaCandidates;          // current list of photon candidates
    TList                 *fClusterCandidates;        
    TList                                                           *fVirtualGammaCandidates;                          //current list of virtual photon candidates
    TList                 *fEventCutArray;          // List with Event Cuts
    AliConvEventCuts          *fEventCuts;            // EventCutObject
    TList                 *fGammaCutArray;              // List with Conversion Cuts
    TList                                                   *fElectronCutArray;                                     // List with electron cuts
    AliConversionPhotonCuts       *fConversionCuts;         // ConversionCutObject
    TList                 *fClusterCutArray;          // List with Cluster Cuts
    AliCaloPhotonCuts           *fCaloPhotonCuts;         // CaloPhotonCutObject
    TList                 *fMesonCutArray;          // List with Meson Cuts
    AliConversionMesonCuts        *fMesonCuts;            // MesonCutObject
    
    //histograms for Conversions reconstructed quantities
    TH1F                **fHistoConvGammaPt;        
    TH1F                **fHistoConvGammaR;       
    TH1F                **fHistoConvGammaEta;       
    TH1F                **fHistoDalitzElectronPt;     
    TH1F                **fHistoDalitzPositronPt;     
    TH1F                **fHistoDalitzElectronPhi;      
    TH1F                **fHistoDalitzPositronPhi;      
    Float_t               fPtGamma;             
    Float_t               fDCAzPhoton;            
    Float_t               fRConvPhoton;           
    Float_t               fEtaPhoton;             
    UChar_t               fCharCatPhoton;           
    UChar_t               fCharPhotonMCInfo;          
                      // 0: garbage,
                      // 1: background
                      // 2: secondary photon not from eta or k0s,
                      // 3: secondary photon from eta, 
                      // 4: secondary photon from k0s, 
                      // 5: dalitz
                      // 6: primary gamma
    //histograms for mesons reconstructed quantities
    TH2F                **fHistoMotherInvMassPt;      
    TH1F                **fHistoMotherInvMassOpeningAngleGammaElectron; 
    TH2F                **fHistoMotherMatchedInvMassPt;   
    THnSparseF              **fSparseMotherInvMassPtZM;     
    TH2F                **fHistoMotherBackInvMassPt;    
    THnSparseF              **fSparseMotherBackInvMassPtZM;   
    TH2F                **fHistoMotherInvMassEalpha;    
    TH2F                **fHistoMotherPi0PtY;       
    TH2F                **fHistoMotherEtaPtY;       
    TH2F                **fHistoMotherPi0PtAlpha;     
    TH2F                **fHistoMotherEtaPtAlpha;     
    TH2F                **fHistoMotherPi0PtOpenAngle;   
    TH2F                **fHistoMotherEtaPtOpenAngle;   
    TH2F                                        **fHistoMotherPi0ConvPhotonEtaPhi;  
    TH2F                                        **fHistoMotherEtaConvPhotonEtaPhi;  
    TH2F                **fHistoMotherInvMassECalib;    
    TH2F                **fHistoMotherInvMassECalibalpha; 

    // histograms for rec photons tagged by Calo
    // histograms for rec photon clusters
    TH1F                ** fHistoClusGammaPt;       
    TH1F                ** fHistoClusOverlapHeadersGammaPt; 
                    
    //histograms for pure MC quantities
    TH1I                **fHistoMCHeaders;        
    TH1F                **fHistoMCAllGammaPt;       
    TH1F                **fHistoMCAllGammaPi0Pt;                        
    TH1F                **fHistoMCAllGammaEMCALAccPt;     
    TH1F                **fHistoMCDecayGammaPi0Pt;      
    TH1F                **fHistoMCDecayGammaRhoPt;      
    TH1F                **fHistoMCDecayGammaEtaPt;      
    TH1F                **fHistoMCDecayGammaOmegaPt;      
    TH1F                **fHistoMCDecayGammaEtapPt;     
    TH1F                **fHistoMCDecayGammaPhiPt;      
    TH1F                **fHistoMCDecayGammaSigmaPt;      
    TH1F                **fHistoMCConvGammaPt;        
    TH1F                **fHistoMCConvGammaR;       
    TH1F                **fHistoMCConvGammaEta;       
    TH1F                **fHistoMCAllPositronsPt;                       
    TH1F                **fHistoMCDecayPositronPi0Pt;                   
    TH1F                **fHistoMCAllElectronsPt;     
    TH1F                **fHistoMCDecayElectronPi0Pt;                   
    TH1F                **fHistoMCDecayNoPrimElectronPi0DalitzR;  
    TH1F                **fHistoMCDecayNoPrimPositronPi0DalitzR;  
    TH1F                **fHistoMCDecayNoPrimElectronPi0DalitzID; 
    TH1F                **fHistoMCDecayNoPrimPositronPi0DalitzID; 
    TH1F                **fHistoMCPi0GGPt;        
    TH1F                **fHistoMCPi0GGWOWeightPt;      
    TH1F                **fHistoMCPi0Pt;        
    TH1F                **fHistoMCPi0WOWeightPt;      
    TH1F                **fHistoMCEtaGGPt;                              
    TH1F                                                            **fHistoMCEtaGGWOWeightPt;                      
    TH1F                **fHistoMCEtaPt;        
    TH1F                **fHistoMCEtaWOWeightPt;      
    TH1F                **fHistoMCPi0InAccPt;       
    TH1F                **fHistoMCPi0InAccOpeningAngleGammaElectron;    
    TH1F                **fHistoMCEtaInAccPt;       
    TH2F                **fHistoMCPi0PtY;       
    TH2F                **fHistoMCEtaPtY;       
    TH2F                **fHistoMCPi0PtAlpha;       
    TH2F                **fHistoMCEtaPtAlpha;       
    TH1F                **fHistoMCK0sPt;        
    TH1F                **fHistoMCK0sWOWeightPt;      
    TH2F                **fHistoMCK0sPtY;       
    // MC validated reconstructed quantities mesons
    TH2F                **fHistoTruePi0InvMassPt;           
    TH2F                **fHistoTrueEtaInvMassPt;           
    TH2F                **fHistoTruePi0ShowerInvMassPt;           
    TH2F                **fHistoTrueEtaShowerInvMassPt;           
    TH2F                **fHistoTruePi0NoShowerInvMassPt;           
    TH2F                **fHistoTrueEtaNoShowerInvMassPt;           
    TH1F                **fHistoTruePi0OpeningAngleGammaElectron;                                       
    TH2F                **fHistoTruePi0GGInvMassPt;         
    TH2F                **fHistoTrueEtaGGInvMassPt;           
    TH2F                **fHistoTruePi0CaloPhotonInvMassPt;       
    TH2F                **fHistoTrueEtaCaloPhotonInvMassPt;       
    TH2F                **fHistoTruePi0CaloConvertedPhotonInvMassPt;  
    TH2F                **fHistoTruePi0CaloConvertedPhotonMatchedInvMassPt; 
    TH2F                **fHistoTrueEtaCaloConvertedPhotonInvMassPt;  
    TH2F                **fHistoTrueEtaCaloConvertedPhotonMatchedInvMassPt; 
    TH2F                **fHistoTruePi0CaloElectronInvMassPt;     
    TH2F                **fHistoTrueEtaCaloElectronInvMassPt;     
    TH2F                **fHistoTruePi0CaloMergedClusterInvMassPt;    
    TH2F                **fHistoTrueEtaCaloMergedClusterInvMassPt;    
    TH2F                **fHistoTrueMotherCaloEMNonLeadingInvMassPt;  
    TH2F                **fHistoTruePi0CaloMergedClusterPartConvInvMassPt; 
    TH2F                **fHistoTrueEtaCaloMergedClusterPartConvInvMassPt; 
    TH2F                **fHistoTruePrimaryPi0InvMassPt;        
    TH2F                **fHistoTruePrimaryPi0GGInvMassPt;        
    TH2F                **fHistoTruePrimaryEtaInvMassPt;        
    TH2F                **fHistoTruePrimaryEtaGGInvMassPt;        
    TH2F                **fHistoTruePrimaryPi0W0WeightingInvMassPt;   
    TH2F                **fHistoTruePrimaryEtaW0WeightingInvMassPt;   
    TProfile2D              **fProfileTruePrimaryPi0WeightsInvMassPt;   
    TProfile2D              **fProfileTruePrimaryEtaWeightsInvMassPt;   
    TH2F                **fHistoTruePrimaryPi0MCPtResolPt;        
    TH2F                **fHistoTruePrimaryEtaMCPtResolPt;        
        TH2F                                **fHistoTrueMotherPi0ConvPhotonEtaPhi;      
        TH2F                                **fHistoTrueMotherEtaConvPhotonEtaPhi;      
    TH2F                **fHistoTrueSecondaryPi0InvMassPt;        
    TH2F                **fHistoTrueSecondaryPi0GGInvMassPt;        
    TH2F                **fHistoTrueSecondaryPi0FromK0sInvMassPt;   
    TH1F                **fHistoTrueK0sWithPi0DaughterMCPt;       
    TH2F                **fHistoTrueSecondaryPi0FromEtaInvMassPt;   
    TH1F                **fHistoTrueEtaWithPi0DaughterMCPt;       
    TH2F                **fHistoTrueSecondaryPi0FromLambdaInvMassPt;  
    TH1F                **fHistoTrueLambdaWithPi0DaughterMCPt;      
    TH2F                **fHistoTrueBckGGInvMassPt;           
    TH2F                **fHistoTrueBckContInvMassPt;         
    TH2F                **fHistoTruePi0PtY;               
    TH2F                **fHistoTrueEtaPtY;               
    TH2F                **fHistoTruePi0PtAlpha;             
    TH2F                **fHistoTrueEtaPtAlpha;             
    TH2F                **fHistoTruePi0PtOpenAngle;           
    TH2F                **fHistoTrueEtaPtOpenAngle;           
    // MC validated reconstructed quantities photons
    TH1F                **fHistoTrueConvGammaPt;            
    TH1F                **fHistoTrueConvPi0GammaPt;           
    TH1F                **fHistoTrueConvGammaEta;           
    TH1F                **fHistoTruePositronPt;             
    TH1F                **fHistoTrueElectronPt;                 
    TH1F                **fHistoTrueSecPositronPt;            
    TH1F                **fHistoTrueSecElectronPt;                                              
    TH1F                **fHistoTruePi0DalitzPositronPt;          
    TH1F                **fHistoTruePi0DalitzElectronPt;          
    TH1F                **fHistoTruePi0DalitzSecPositronPt;         
    TH1F                **fHistoTruePi0DalitzSecElectronPt;         
    TH1F                **fHistoTruePrimaryConvGammaPt;         
    TH2F                **fHistoTruePrimaryConvGammaESDPtMCPt;      
    TH1F                **fHistoTrueSecondaryConvGammaPt;       
    TH1F                **fHistoTrueSecondaryConvGammaFromXFromK0sPt; 
    TH1F                **fHistoTrueSecondaryConvGammaFromXFromLambdaPt;
    TH1F                **fHistoTrueClusGammaPt;            
    TH1F                **fHistoTrueClusUnConvGammaPt;          
    TH1F                **fHistoTrueClusUnConvGammaMCPt;          
    TH1F                **fHistoTrueClusElectronPt;           
    TH1F                **fHistoTrueClusConvGammaPt;          
    TH1F                **fHistoTrueClusConvGammaMCPt;          
    TH1F                **fHistoTrueClusConvGammaFullyPt;       
    TH1F                **fHistoTrueClusMergedGammaPt;          
    TH1F                **fHistoTrueClusMergedPartConvGammaPt;      
    TH1F                **fHistoTrueClusDalitzPt;           
    TH1F                **fHistoTrueClusDalitzMergedPt;         
    TH1F                **fHistoTrueClusPhotonFromElecMotherPt;     
    TH1F                **fHistoTrueClusShowerPt;           
        TH1F                                **fHistoTrueClusSubLeadingPt;                   
        TH1I                                **fHistoTrueClusNParticles;                     
    TH1F                **fHistoTrueClusEMNonLeadingPt;         
    TH1F                **fHistoTrueNLabelsInClus;            
    TH1F                **fHistoTruePrimaryClusGammaPt;         
    TH2F                **fHistoTruePrimaryClusGammaESDPtMCPt;      
    TH1F                **fHistoTruePi0DalitzClusGammaPt;     
    TH1F                **fHistoTruePi0DalitzAllClusGammaPt;                    
    TH1F                **fHistoTruePi0DalitzClusGammaMCPt;     
    TH2F                **fHistoTruePrimaryPi0PhotonPairPtconv;     
    TH1F                **fHistoTruePrimaryPi0DCPtconv;         
    TH1F                **fHistoTruePrimaryPi0MissingPtconv;      
    TH2F                **fHistoTruePrimaryEtaPhotonPairPtconv;     
    TH1F                **fHistoTruePrimaryEtaDCPtconv;         
    TH1F                **fHistoTruePrimaryEtaMissingPtconv;      
    TH2F                **fHistoTrueSecondaryPi0PhotonPairPtconv;   
    TH1F                **fHistoTrueSecondaryPi0DCPtconv;       
    TH1F                **fHistoTrueSecondaryPi0MissingPtconv;      
    TString               *fStringRecTruePi0s;              
    TString               *fStringRecTrueEtas;              
    TH2F                **fHistoDoubleCountTruePi0InvMassPt;      
    TH2F                **fHistoDoubleCountTrueEtaInvMassPt;      
    TH2F                **fHistoDoubleCountTrueConvGammaRPt;        
    vector<Int_t>           fVectorDoubleCountTruePi0s;           
    vector<Int_t>           fVectorDoubleCountTrueEtas;           
    vector<Int_t>           fVectorDoubleCountTrueConvGammas;         
    // event histograms
    TH1I                **fHistoNEvents;                
    TH1I                **fHistoNGoodESDTracks;             
    TH1I                **fHistoNGammaCandidates;           
    TH2F                **fHistoNGoodESDTracksVsNGammaCanditates;   
    TH2F                **fHistoSPDClusterTrackletBackground;     
    TH1I                **fHistoNV0Tracks;                
    TProfile              **fProfileEtaShift;               
    
    // additional variables
    Double_t              fEventPlaneAngle;           // EventPlaneAngle
    TRandom3              fRandom;              // random 
    Int_t                 fNGammaCandidates;          // number of gamma candidates in event
    Double_t              *fUnsmearedPx;            //[fNGammaCandidates]
    Double_t              *fUnsmearedPy;            //[fNGammaCandidates]
    Double_t              *fUnsmearedPz;            //[fNGammaCandidates]
    Double_t              *fUnsmearedE;             //[fNGammaCandidates]
        Int_t                 *fMCEventPos;               //[fNGammaCandidates]
        Int_t                 *fMCEventNeg;               //[fNGammaCandidates]
    Int_t                 *fESDArrayPos;              //[fNGammaCandidates]
    Int_t                 *fESDArrayNeg;            //[fNGammaCandidates]
    Int_t                 fnCuts;               // number of cuts to be analysed in parallel
    Int_t                 fiCut;                // current cut  
    Bool_t                fMoveParticleAccordingToVertex;   // boolean for BG calculation
    Int_t               fIsHeavyIon;            // switch for pp = 0, PbPb = 1, pPb = 2
    Bool_t                fDoMesonAnalysis;         // flag for meson analysis
    Int_t                 fDoMesonQA;             // flag for meson QA
    Int_t                 fDoPhotonQA;            // flag for photon QA
    Int_t                 fDoClusterQA;           // flag for cluster QA
    Bool_t                fIsFromMBHeader;          // flag for MC headers
    Bool_t                fIsOverlappingWithOtherHeader;    // flag for particles in MC overlapping between headers
    Bool_t                fIsMC;                // flag for MC information
        Bool_t                fDoTHnSparse;                       // flag for THnSparse
        Int_t                               fTrackMatcherRunningMode;           // CaloTrackMatcher running mode
    
  private:
    AliAnalysisTaskGammaCaloDalitzV1(const AliAnalysisTaskGammaCaloDalitzV1&); // Prevent copy-construction
    AliAnalysisTaskGammaCaloDalitzV1 &operator=(const AliAnalysisTaskGammaCaloDalitzV1&); // Prevent assignment

        ClassDef(AliAnalysisTaskGammaCaloDalitzV1, 6);
};

#endif