AliAnalysisTaskGammaCocktailMC.h

Go to the documentation of this file.

#ifndef ALIANLYSISTASKGAMMACOCKTAILMC_cxx
#define ALIANLYSISTASKGAMMACOCKTAILMC_cxx

#include "AliAnalysisTaskSE.h"
#include "AliAnalysisManager.h"
#include "AliGenEMCocktailV2.h"
#include "AliMCGenHandler.h"
#include "AliGenerator.h"

class AliAnalysisTaskGammaCocktailMC : public AliAnalysisTaskSE {
  public:

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

    virtual void   UserCreateOutputObjects();
    virtual void   UserExec(Option_t *);
    virtual void   Terminate(const Option_t*);

    // MC functions
    void SetIsMC(Int_t isMC){fIsMC=isMC;}
    void ProcessMCParticles();
    
    // additional functions
    void SetLogBinningXTH1(TH1* histoRebin);
    void SetLogBinningXTH2(TH2* histoRebin);
    void SetMaxY(Double_t maxy){fMaxY = maxy;}
    void SetLightOutput(Bool_t flag) {fDoLightOutput = flag;}
    void InitializeDecayChannelHist(TH1F* hist, Int_t np);
    void FillPythiaBranchingRatio(TH1F* histo, Int_t np);
    void GetAndSetPtParametrizations(AliGenEMCocktailV2* mcCocktailGen);
    void GetAndSetPtYDistributions(AliGenEMCocktailV2* mcCocktailGen);
    void SetHasMother(UInt_t selectedMothers);
    Int_t GetParticlePosLocal(Int_t pdg);
    TH1* SetHist1D(TH1* hist, TString histType, TString histName, TString xTitle, TString yTitle, Int_t nBinsX, Double_t xMin, Double_t xMax, Bool_t optSumw2);
    TH2* SetHist2D(TH2* hist, TString histType, TString histName, TString xTitle, TString yTitle, Int_t nBinsX, Double_t xMin, Double_t xMax, Int_t nBinsY, Double_t yMin, Double_t yMax, Bool_t optSumw2);
    TH2* SetHist2D(TH2* hist, TString histType, TString histName, TString xTitle, TString yTitle, Int_t nBinsX, Double_t xMin, Double_t xMax, Int_t nBinsY, Double_t* binsY, Bool_t optSumw2);
    Float_t GetDecayChannel(AliMCEvent *mcEvent, TParticle* part);
    
  protected:
    TList*                      fOutputContainer;               // Output container

    AliVEvent*                  fInputEvent;                    // current event
    AliMCEvent*                 fMCEvent;                       // corresponding MC event
    AliMCGenHandler*            fMCGenHandler;                  // MC gen handler
    const AliGenerator*         fMCGenerator;                   //
    AliGenEMCocktailV2*         fMCCocktailGen;                 // cocktail generator
  

  
    Bool_t                      fDoLightOutput;                 // switch for running light
    Bool_t                      fHasMother[17];                 // mother i produced
  
    // histograms events
    TH1F*                       fHistNEvents;                   // number of events histo

    // histograms mesons
    TH2F*                       fHistPtYGamma;                  
    TH2F*                       fHistPtPhiGamma;                
    TH2F**                      fHistPtPhiGammaSource;          
    TH2F**                      fHistPtPhiInput;                
    TH2F**                      fHistPtYInput;                  
    TH2F**                      fHistPtYGammaSource;            
    TH2F**                      fHistPtAlphaInput;              
    TH2F**                      fHistPtDeltaPhiInput;           
    TH1F**                      fHistDecayChannelsInput;        
    TH1F**                      fHistPythiaBR;                  
    
    TH2F**                      fHistPtGammaSourcePtInput;      
    TH2F**                      fHistPhiGammaSourcePhiInput;    
    
    TH1I*                       fHistPdgInputRest;              
    TH1I*                       fHistPdgGammaSourceRest;        
  
    Int_t*                      fParticleList;                  // array with particle Pdg values
    TString*                    fParticleListNames;             // array with particle names

    TF1*                        fPtParametrization[17];         
    TF1*                        fPtParametrizationProton;       
    TObjString*                 fCocktailSettings[12];          
    TH1D*                       fMtScalingFactors;              
    TH2F*                       fPtYDistributions[17];          

    //histos for PCM-EMCal related Pi0-tagging, splitting Gammas from Pi0s into Dalitz and NonDalitz contributions
    TH2F*                       fHistPtYGammaSourceFromDalitzPi0; 
    TH2F*                       fHistPtPhiGammaSourceFromDalitzPi0; 
    TH2F*                       fHistPtYGammaSourceFromNonDalitzPi0; 
    TH2F*                       fHistPtPhiGammaSourceFromNonDalitzPi0; 

    TList*                      fUserInfo;
    TTree*                      fOutputTree;
    Int_t                       fIsMC;                          // MC flag
    Double_t                    fMaxY;                          // Max y

  private:
    AliAnalysisTaskGammaCocktailMC(const AliAnalysisTaskGammaCocktailMC&);            // Prevent copy-construction
    AliAnalysisTaskGammaCocktailMC &operator=(const AliAnalysisTaskGammaCocktailMC&); // Prevent assignment

    ClassDef(AliAnalysisTaskGammaCocktailMC, 6);
};

#endif