AliAnalysisTaskClusterQA.h

Go to the documentation of this file.

#ifndef AliAnalysisClusterQA_cxx
#define AliAnalysisClusterQA_cxx

#include "AliAnalysisTaskSE.h"
#include "AliConversionPhotonBase.h"
#include "TH1.h"
#include "TH2.h"
#include "TTreeStream.h"
#include "AliLog.h"
#include <vector>
#include "AliV0ReaderV1.h"
#include "AliCaloPhotonCuts.h"
#include "AliConvEventCuts.h"
#include "AliConversionPhotonCuts.h"
#include "AliConversionMesonCuts.h"
#include "TList.h"
#include "AliMCEvent.h"
#include "TClonesArray.h"


using namespace std;


class AliAnalysisTaskClusterQA : public AliAnalysisTaskSE{

  public:

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

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

    void SetV0Reader                        ( AliV0ReaderV1 *v0Reader )                   { fV0Reader=v0Reader                  ; }
    void SetV0ReaderName(TString name){fV0ReaderName=name; return;}
    void SetEventCuts                       ( AliConvEventCuts* conversionCuts,
                                              Bool_t IsHeavyIon )                         {
                                                                                            fEventCuts=conversionCuts           ;
                                                                                            fIsHeavyIon = IsHeavyIon            ;
                                                                                          }
    void SetClusterCutsEMC                     ( AliCaloPhotonCuts* clusterCuts,
                                              Bool_t IsHeavyIon )                         {
                                                                                            fClusterCutsEMC=clusterCuts           ;
                                                                                            fIsHeavyIon = IsHeavyIon            ;
                                                                                          }
    void SetClusterCutsDMC                     ( AliCaloPhotonCuts* clusterCuts,
                                              Bool_t IsHeavyIon )                         {
                                                                                            fClusterCutsDMC=clusterCuts           ;
                                                                                            fIsHeavyIon = IsHeavyIon            ;
                                                                                          }
    void SetMesonCuts                     ( AliConversionMesonCuts* mesonCuts,
                                              Bool_t IsHeavyIon )                         {
                                                                                            fMesonCuts=mesonCuts           ;
                                                                                            fIsHeavyIon = IsHeavyIon            ;
                                                                                          }

    void FillType                           ( Double_t fillTree,
                                              Bool_t fillHistorams)                       {
                                                                                            ffillTree = fillTree                ;
                                                                                            ffillHistograms = fillHistorams     ;
                                                                                          }
    void SetIsMC                            ( Bool_t isMC )                               { fIsMC                 = isMC        ; }
    void SetDoAdditionalHistos              ( Bool_t val )                                { fSaveAdditionalHistos = val        ; }
    void SetSaveEventProperties             ( Bool_t val  )                               { fSaveEventProperties  = val         ; }
    void SetSaveClusterCells                ( Bool_t val  )                               { fSaveCells            = val         ; }
    void SetSaveSurroundingCells            ( Bool_t val  )                               { fSaveSurroundingCells         = val         ; }
    void SetSaveSurroundingTracks            ( Bool_t val  )                               { fSaveTracks         = val         ; }
    void SetMaxConeRadius                      ( Float_t val  )                                { fConeRadius    = val         ; }
    void SetMinTrackPt                      ( Float_t val  )                                { fMinTrackPt    = val         ; }
    void SetMinClusterEnergy                      ( Float_t val  )                                { fMinClusterEnergy    = val         ; }
    void SetMinMaxNLMCut                    ( Int_t valmin, Int_t valmax  )               { fMinNLMCut    = valmin              ;
                                                                                            fMaxNLMCut    = valmax              ; }
    void SetSaveMCInformation               ( Bool_t val  )                               { fSaveMCInformation    = val         ; }
    // Function to set correction task setting
    void SetCorrectionTaskSetting(TString setting) {fCorrTaskSetting = setting;}
    Int_t       FindLargestCellInCluster(AliVCluster* cluster, AliVEvent* event);
    void GetRowAndColumnFromAbsCellID(Int_t cellIndex, Int_t& row, Int_t& column);
    Int_t MakePhotonCandidates(AliVCluster* clus, AliVCaloCells* cells, Long_t indexCluster);
    void ProcessTracksAndMatching(AliVCluster* clus, Long_t indexCluster);
    Int_t  GetMCClusterFlag(AliVCluster* clus, AliVCaloCells* cells);
   Float_t GetCentrality(AliVEvent *event);
    void ResetBuffer();
  private:
        
    AliAnalysisTaskClusterQA     ( const AliAnalysisTaskClusterQA& ); // Prevent copy-construction
    AliAnalysisTaskClusterQA &operator=( const AliAnalysisTaskClusterQA& ); // Prevent assignment

    ULong64_t GetUniqueEventID      ( AliVHeader *header);
    void ProcessQATreeCluster       ( AliVEvent *event, AliVCluster* cluster, Long_t indexCluster);
    void ProcessQA                  ( AliAODConversionPhoton *gamma );
    void RelabelAODPhotonCandidates ( Bool_t mode );
    void ProcessTrueQAESD           ( AliAODConversionPhoton *TruePhotonCandidate,
                                      AliESDtrack *elec,
                                      AliESDtrack *posi );
    void ProcessTrueQAAOD           ( AliAODConversionPhoton *TruePhotonCandidate,
                                      AliAODTrack *elec,
                                      AliAODTrack *posi );
    Int_t ProcessTrueClusterCandidates(AliAODConversionPhoton *TrueClusterCandidate, Float_t m02,
                                        AliAODConversionPhoton *TrueSubClusterCandidate1,
                                        AliAODConversionPhoton *TrueSubClusterCandidate2);
    Int_t ProcessTrueClusterCandidatesAOD(AliAODConversionPhoton *TrueClusterCandidate, Float_t m02,
                                        AliAODConversionPhoton *TrueSubClusterCandidate1,
                                        AliAODConversionPhoton *TrueSubClusterCandidate2);
    UInt_t IsTruePhotonESD          ( AliAODConversionPhoton *TruePhotonCandidate );
    UInt_t IsTruePhotonAOD          ( AliAODConversionPhoton *TruePhotonCandidate );
    void CountTracks                ();
    void SetLogBinningXTH2          ( TH2* histoRebin );
        
  protected:
    AliV0ReaderV1*              fV0Reader;                  //
    TString                     fV0ReaderName;
    TString                     fCorrTaskSetting;
    AliConversionPhotonCuts*    fConversionCuts;            // Cuts used by the V0Reader
    AliConvEventCuts*           fEventCuts;                 // Cuts used by the V0Reader
    AliCaloPhotonCuts*          fClusterCutsEMC;               // Cuts used by the V0Reader
    AliCaloPhotonCuts*          fClusterCutsDMC;               // Cuts used by the V0Reader
    AliConversionMesonCuts*     fMesonCuts;               // Cuts used by the V0Reader
    Int_t                       fMinNLMCut;                   
    Int_t                       fMaxNLMCut;                   
    AliVEvent*                  fInputEvent;                //
    AliMCEvent*                 fMCEvent;                   //
    Double_t                    fWeightJetJetMC;                                      // weight for Jet-Jet MC
    AliEMCALGeometry*           fGeomEMCAL;                     // pointer to EMCAL geometry
    TTree*                      fClusterTree;                    //
    Bool_t                      fIsHeavyIon;                //
    Double_t                    ffillTree;                  //
    Bool_t                      ffillHistograms;            //
    TList*                      fOutputList;                //
    Bool_t                      fIsMC;                      //
    Bool_t                      fCorrectForNonlinearity;                      //
    
    // Save flags
    Bool_t          fSaveEventProperties;                 
    Bool_t          fSaveCells;                           
    Bool_t          fSaveSurroundingCells;                        
    Bool_t          fSaveTracks;                        
    Float_t          fConeRadius;                        
    Float_t          fMinTrackPt;                        
    Float_t          fMinClusterEnergy;                        
    Bool_t          fSaveMCInformation;                   
    Bool_t          fSaveAdditionalHistos;                   

    // Option flags
    std::vector<Float_t> fExtractionPercentages;          
    std::vector<Float_t> fExtractionPercentagePtBins;     
    
    // Buffers that will be added to the tree
    ULong64_t       fBuffer_EventID;                     
    Float_t         fBuffer_ClusterE;                     
    Float_t         fBuffer_ClusterPhi;                   
    Float_t         fBuffer_ClusterEta;                   
    Bool_t          fBuffer_ClusterIsEMCAL;                   
    Int_t           fBuffer_MC_Cluster_Flag;                   
    Int_t           fBuffer_ClusterNumCells;              
    Int_t           fBuffer_LeadingCell_ID;              
    Float_t         fBuffer_LeadingCell_E;              
    Float_t         fBuffer_LeadingCell_Eta;              
    Float_t         fBuffer_LeadingCell_Phi;              
    Float_t         fBuffer_ClusterM02;              
    Float_t         fBuffer_ClusterM20;              

    Float_t         fBuffer_Event_Vertex_X;               
    Float_t         fBuffer_Event_Vertex_Y;               
    Float_t         fBuffer_Event_Vertex_Z;               
    Float_t         fBuffer_Event_Multiplicity;             
    Int_t           fBuffer_Event_NumActiveCells;          

    Int_t*          fBuffer_Cells_ID;                      
    Float_t*        fBuffer_Cells_E;                      
    Float_t*        fBuffer_Cells_RelativeEta;                      
    Float_t*        fBuffer_Cells_RelativePhi;                      

    Int_t           fBuffer_Surrounding_NCells;                
    Int_t*          fBuffer_Surrounding_Cells_ID;                
    Float_t*        fBuffer_Surrounding_Cells_R;                
    Float_t*        fBuffer_Surrounding_Cells_E;                
    Float_t*        fBuffer_Surrounding_Cells_RelativeEta;              
    Float_t*        fBuffer_Surrounding_Cells_RelativePhi;              
    Int_t           fBuffer_Surrounding_NTracks;                
    Float_t*        fBuffer_Surrounding_Tracks_R;              
    Float_t*        fBuffer_Surrounding_Tracks_Pt;              
    Float_t*        fBuffer_Surrounding_Tracks_RelativeEta;              
    Float_t*        fBuffer_Surrounding_Tracks_RelativePhi;              
    
    Int_t           fBuffer_Cluster_MC_Label;              
    Int_t           fBuffer_Mother_MC_Label;              

    TH2F*           hNCellsInClustersVsCentrality;
    TH2F*           hNActiveCellsVsCentrality;
    TH2F*           hNActiveCellsAbove50MeVVsCentrality;
    TH2F*           hNActiveCellsAbove80MeVVsCentrality;
    TH2F*           hNActiveCellsAbove100MeVVsCentrality;
    TH2F*           hNActiveCellsAbove150MeVVsCentrality;
    TH2F*           hECellsInClustersVsCentrality;
    TH2F*           hEActiveCellsVsCentrality;
    TH2F*           hEActiveCells50MeVVsCentrality;
    TH2F*           hEActiveCells80MeVVsCentrality;
    TH2F*           hEActiveCells100MeVVsCentrality;
    TH2F*           hEActiveCells150MeVVsCentrality;
    
    ClassDef(AliAnalysisTaskClusterQA, 7);
};

const Int_t kMaxActiveCells = 18000;
const Int_t kMaxNTracks = 4000;

#endif