AliAnalysisTaskEMCALPi0CalibSelection.h

Go to the documentation of this file.

#ifndef ALIANALYSISTASKEMCALPI0CALIBSELECTION_H
#define ALIANALYSISTASKEMCALPI0CALIBSELECTION_H

//---------------------------------------------------------------------------
//---------------------------------------------------------------------------

// Root includes
class TH1F;
#include "TH2I.h"
#include "TObjArray.h"
#include "TLorentzVector.h"

// AliRoot includes
#include "AliAnalysisTaskSE.h"
class AliEMCALGeometry;
#include "AliEMCALGeoParams.h"
class AliEMCALRecoUtils;

class AliAnalysisTaskEMCALPi0CalibSelection : public AliAnalysisTaskSE
{
    
public:

  AliAnalysisTaskEMCALPi0CalibSelection();
  
  AliAnalysisTaskEMCALPi0CalibSelection(const char* name);
    
  virtual ~AliAnalysisTaskEMCALPi0CalibSelection();
    
  void    CorrectClusters();
  
  void    FillHistograms();
  
  void    InitEnergyCalibrationFactors();

  void    InitGeometryMatrices();
  
  void    InitTemperatureCorrections();
    
  void    UserCreateOutputObjects();
  
  void    UserExec(Option_t * opt);
    
  void    PrintInfo();

  void    Terminate(Option_t* opt);
  
  void    GetMaxEnergyCellPosAndClusterPos(AliVCaloCells* cells, AliVCluster* clu, Int_t& iSM, Int_t& ieta, Int_t& iphi);
  
  Int_t   FindPositionInNoisyQuartet(Int_t irow, Int_t icol, Int_t iSM);
    
  // Analysis parameter setting
  
  void    SwitchOnCentrality()                           { fCheckCentrality  = kTRUE ; }
  void    SwitchOffCentrality()                          { fCheckCentrality  = kFALSE; }

  void    SetCentralityWithPhysSel( Bool_t ps )          { fCentWithEventSel = ps    ; }
  
  void    SetCentralityClass(TString name)               { fCentralityClass   = name ; }
  
  void    SetCentralityRange(Float_t min, Float_t max)   { fCentMin = min; fCentMax = max; }
  
  void    SetPairDTimeCut(Float_t t)                     { fDTimeCut    = t          ; }
  
  void    SetClusterMinTime(Float_t tmin)                { fTimeMin     = tmin       ; }
  
  void    SetClusterMaxTime(Float_t tmax)                { fTimeMax     = tmax       ; }

  void    SetAsymmetryCut(Float_t asy)                   { fAsyCut      = asy        ; }
  
  void    SetClusterMinEnergy(Float_t emin)              { fEmin        = emin       ; }
  
  void    SetClusterMaxEnergy(Float_t emax)              { fEmax        = emax       ; }
  
  void    SetClusterMinEnergyForBkgShapeStudy(Float_t emin)              { fEBkgmin        = emin       ; }
    
  void    SetClusterMaxEnergyForBkgShapeStudy(Float_t emax)              { fEBkgmax        = emax       ; }
  
  void    SetClusterLambda0Cuts(Float_t min, Float_t max){ fL0max       = max        ;
                                                           fL0min       = min        ; }
  void    SetClusterLambda0CutsForBkgShapeStudy(Float_t min, Float_t max){ fL0Bkgmax       = max        ; fL0Bkgmin       = min        ; }
    
  void    SetClusterOpeningAngleCutsForBkgShapeStudy(Float_t opmin, Float_t opmax){ fOpAnglemax       = opmax        ;fOpAnglemin       = opmin        ; }
    
  void    SetClusterMinNCells(Int_t n)                   { fMinNCells   = n          ; }
  
  void    SetNCellsGroup(Int_t n)                        { fGroupNCells = n          ; }
  
  void    SetLogWeight(Float_t w)                        { fLogWeight   = w          ; }
      
  void    SetPairMinMassCut(Float_t min)                 { fInvMassCutMin = min      ; }

  void    SetPairMaxMassCut(Float_t max)                 { fInvMassCutMax = max      ; }
  
  void    SwitchOnSameSM()                               { fSameSM = kTRUE           ; }
  
  void    SwitchOffSameSM()                              { fSameSM = kFALSE          ; }
  
  void    UseFilteredEventAsInput()                      { fFilteredInput = kTRUE    ; }
  
  void    UseNormalEventAsInput()                        { fFilteredInput = kFALSE   ; }
  
  void    SetTriggerName(TString name)                   { fTriggerName = name       ; }
  
  void    SwitchOffSelectOnlyCellSignalOutOfCollision()  { fSelectOnlyCellSignalOutOfCollision = kFALSE ; }
  
  void    SwitchOnSelectOnlyCellSignalOutOfCollision()   { fSelectOnlyCellSignalOutOfCollision = kTRUE ; }
  
  void    SwitchOffCellEnergyHisto()                     { fCellEnergyHiso = kFALSE ; }
  
  void    SwitchOnCellEnergyHisto()                      { fCellEnergyHiso = kTRUE ; }
  
  void    SwitchOffTopoClusterHisto()                     { fClusterTopology = kFALSE ; }
  
  void    SwitchOnTopoClusterHisto()                      { fClusterTopology = kTRUE ; }
  
  void    SwitchOffSelectOnlyPhotonsInDifferentSM()       { fSelectOnlyPhotonsInDifferentSM = kFALSE ; }
  
  void    SwitchOnSelectOnlyPhotonsInDifferentSM()        { fSelectOnlyPhotonsInDifferentSM = kTRUE ; }
  
  void    SwitchOffChangeBkgShape()                       { fChangeBkgShape = kFALSE ; }
  
  void    SwitchOnChangeBkgShape()                        { fChangeBkgShape = kTRUE ; }

  // Geometry setters
  
  void    SetGeometryName(TString name)                  { fEMCALGeoName = name      ; }
  
  TString GeometryName() const                           { return fEMCALGeoName      ; }
  
  void    SwitchOnLoadOwnGeometryMatrices()              { fLoadMatrices = kTRUE     ; }
  
  void    SwitchOffLoadOwnGeometryMatrices()             { fLoadMatrices = kFALSE    ; }
  
  void    SetGeometryMatrixInSM(TGeoHMatrix* m, Int_t i) { fMatrix[i]    = m         ; }

  void    SetOADBFilePath(TString path)                  { fOADBFilePath = path      ; }
  
  void    SetCalibrationFilePath( TString path )         { fCalibFilePath = path     ; }
  
  // Cluster recalculation
  
  void    SwitchOnClusterCorrection()                    { fCorrectClusters = kTRUE  ; }
 
  void    SwitchOffClusterCorrection()                   { fCorrectClusters = kFALSE ; }

  void    SwitchOnRecalculatePosition()                  { fRecalPosition   = kTRUE  ; }
  
  void    SwitchOffRecalculatePosition()                 { fRecalPosition   = kFALSE ; }

  void    SetEMCALRecoUtils(AliEMCALRecoUtils * ru)      { fRecoUtils = ru           ; }
  
  AliEMCALRecoUtils* GetEMCALRecoUtils()    const        { return fRecoUtils         ; }
  
  void    SetInvariantMassHistoBinRange(Int_t nBins, Float_t minbin, Float_t maxbin){
                                        fNbins     = nBins ; fMinBin     = minbin ; fMaxBin     = maxbin ; }
  
  void    SetEnergyHistoBinRange(Int_t nBins, Float_t minbin, Float_t maxbin){
    fNEnergybins     = nBins ; fMinEnergyBin     = minbin ; fMaxEnergyBin     = maxbin ; }

  void    SetTimeHistoBinRange         (Int_t nBins, Float_t minbin, Float_t maxbin){
                                        fNTimeBins = nBins ; fMinTimeBin = minbin ; fMaxTimeBin = maxbin ; }

  
  void    SetImportGeometryFromFile(Bool_t import, TString path = ""){
                                                           fImportGeometryFromFile = import ;
                                                           fImportGeometryFilePath = path   ; } 

  // Mask clusters
  
  void    SetNMaskCellColumns(Int_t n) ;
  
  void    SetMaskCellColumn(Int_t ipos, Int_t icol) ;
  
  Bool_t  MaskFrameCluster(Int_t iSM, Int_t ieta) const;
  
  
  // Define zones for clusters for pT dependance in Pi0 calibration study
  
//  Int_t IsInWhichZone(Int_t iSupMod, Int_t ieta, Int_t iphi);
  Bool_t IsInZone1(Int_t iSupMod, Int_t ieta, Int_t iphi);
  
  Bool_t IsInZone2(Int_t iSupMod, Int_t ieta, Int_t iphi);
  
  Bool_t IsInZone3(Int_t iSupMod, Int_t ieta, Int_t iphi);
  
  Bool_t IsInZone4(Int_t iSupMod, Int_t ieta, Int_t iphi);
  
  Bool_t IsInZone5(Int_t iSupMod, Int_t ieta, Int_t iphi);

  Bool_t IsInZone6(Int_t iSupMod, Int_t ieta, Int_t iphi);
  
  Bool_t IsInZone7(Int_t iSupMod, Int_t ieta, Int_t iphi);
  
private:

  AliEMCALGeometry  * fEMCALGeo;         
    
  TGeoHMatrix       * fMatrix[AliEMCALGeoParams::fgkEMCALModules];
    
  Bool_t              fLoadMatrices;     
    
  
  TString             fEMCALGeoName;     
    
  TString             fTriggerName;      
    
  AliEMCALRecoUtils * fRecoUtils;        
    
  TString             fOADBFilePath ;    
  
  TString             fCalibFilePath;    
    
  Bool_t              fCorrectClusters;  

  Bool_t              fRecalPosition;    
  
  TRefArray         * fCaloClustersArr;  
    
  AliVCaloCells     * fEMCALCells;       
  
//TList             * fCuts ;            //!<! List with analysis cuts.
    
  TList             * fOutputContainer;  
    
  Bool_t              fCheckCentrality;  
  
  TString             fCentralityClass;  
  
  Bool_t              fCentWithEventSel; 
  
  Float_t             fCentMin;          
  Float_t             fCentMax;          
  
  Double_t            fVertex[3];        
    
  Bool_t              fFilteredInput;    

  Bool_t              fImportGeometryFromFile; 
  
  TString             fImportGeometryFilePath; 
  
  // Analysis cuts
  
  Float_t             fEmin;             
  Float_t             fEmax;             
  
  Float_t             fEBkgmin;          
  Float_t             fEBkgmax;          
    
  Float_t             fL0min;            
  Float_t             fL0max;            
  
  Float_t             fL0Bkgmin;         
  Float_t             fL0Bkgmax;         
    
  Float_t             fOpAnglemin;       
  Float_t             fOpAnglemax;       

  Float_t             fDTimeCut;         
  Float_t             fTimeMax;          
  Float_t             fTimeMin;          

  Float_t             fAsyCut;           
    
  Int_t               fMinNCells;        
  Int_t               fGroupNCells;      
  
  Float_t             fLogWeight;        
    
  Bool_t              fSameSM;           

  Int_t               fNMaskCellColumns; 

  Int_t*              fMaskCellColumns;  //[fNMaskCellColumns]
  
  Bool_t              fSelectOnlyCellSignalOutOfCollision; 
  
  Bool_t              fCellEnergyHiso;   
  
  Bool_t              fClusterTopology;  
  
  Bool_t              fSelectOnlyPhotonsInDifferentSM; 
  
  Bool_t              fChangeBkgShape;   
  
    
  Float_t             fInvMassCutMin;    
  Float_t             fInvMassCutMax;    
  
  // Output histograms and settings

  Int_t               fNbins;            
  Float_t             fMinBin;           
  Float_t             fMaxBin;           
  
  Int_t               fNTimeBins;        
  Float_t             fMinTimeBin;       
  Float_t             fMaxTimeBin;       
  
  Int_t               fNEnergybins;      
  Float_t             fMinEnergyBin;     
  Float_t             fMaxEnergyBin;     
  
  // Temporal TLorentzVectors, avoir recreation per event 
    
  TLorentzVector      fMomentum1 ;       
  TLorentzVector      fMomentum2 ;       
  TLorentzVector      fMomentum12;       
    
  // Histograms
  
  TH1F*     fHmpi0[AliEMCALGeoParams::fgkEMCALModules][AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows]; 
  TH1F*     fhEnergy[AliEMCALGeoParams::fgkEMCALModules][AliEMCALGeoParams::fgkEMCALCols][AliEMCALGeoParams::fgkEMCALRows];     

  TH2F*     fHmgg;                                                                 
  TH2F*     fHmggDifferentSM;                                                      
  TH2F*     fHmggSM[AliEMCALGeoParams::fgkEMCALModules];                           
  TH2F*     fHmggSM_Zone1[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHmggSM_Zone2[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHmggSM_Zone3[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHmggSM_Zone4[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHmggSM_Zone5[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHmggSM_Zone6[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHmggSM_Zone7[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fhTopoClusterCase0[AliEMCALGeoParams::fgkEMCALModules];                
  TH2F*     fhTopoClusterCase1[AliEMCALGeoParams::fgkEMCALModules];                
  TH2F*     fhTopoClusterCase2[AliEMCALGeoParams::fgkEMCALModules];                
  TH2F*     fhTopoClusterCase3[AliEMCALGeoParams::fgkEMCALModules];                
  TH2F*     fhTopoClusterAmpCase0[AliEMCALGeoParams::fgkEMCALModules];             
  TH2F*     fhTopoClusterAmpCase1[AliEMCALGeoParams::fgkEMCALModules];             
  TH2F*     fhTopoClusterAmpCase2[AliEMCALGeoParams::fgkEMCALModules];             
  TH2F*     fhTopoClusterAmpCase3[AliEMCALGeoParams::fgkEMCALModules];             
  
  TH2F*     fhTopoClusterAmpFractionCase0[AliEMCALGeoParams::fgkEMCALModules];     
  TH2F*     fhTopoClusterAmpFractionCase1[AliEMCALGeoParams::fgkEMCALModules];     
  TH2F*     fhTopoClusterAmpFractionCase2[AliEMCALGeoParams::fgkEMCALModules];     
  TH2F*     fhTopoClusterAmpFractionCase3[AliEMCALGeoParams::fgkEMCALModules];     
  TH2F*     fHmggPairSameSectorSM[AliEMCALGeoParams::fgkEMCALModules/2];           
  TH2F*     fHmggPairSameSideSM  [AliEMCALGeoParams::fgkEMCALModules-2];           
  
  TH2F*     fHmggMaskFrame;                                                        
  TH2F*     fHmggDifferentSMMaskFrame;                                             
  TH2F*     fHmggSMMaskFrame[AliEMCALGeoParams::fgkEMCALModules];                  
  TH2F*     fHmggPairSameSectorSMMaskFrame[AliEMCALGeoParams::fgkEMCALModules/2];  
  TH2F*     fHmggPairSameSideSMMaskFrame  [AliEMCALGeoParams::fgkEMCALModules-2];  

  TH2F*     fHOpeningAngle;                                                        
  TH2F*     fHOpeningAngleDifferentSM;                                             
  TH2F*     fHOpeningAngleSM[AliEMCALGeoParams::fgkEMCALModules];                  
  TH2F*     fHOpeningAnglePairSM[AliEMCALGeoParams::fgkEMCALModules];              

  TH2F*     fHAsymmetry;                                                           
  TH2F*     fHAsymmetryDifferentSM;                                                
  TH2F*     fHAsymmetrySM[AliEMCALGeoParams::fgkEMCALModules];                     
  TH2F*     fHAsymmetryPairSM[AliEMCALGeoParams::fgkEMCALModules];                 
  
  TH2F*     fhTowerDecayPhotonHit[AliEMCALGeoParams::fgkEMCALModules] ;            
  TH2F*     fhTowerDecayPhotonEnergy[AliEMCALGeoParams::fgkEMCALModules] ;         
  TH2F*     fhTowerDecayPhotonAsymmetry[AliEMCALGeoParams::fgkEMCALModules] ;      
  TH2F*     fhTowerDecayPhotonHitMaskFrame[AliEMCALGeoParams::fgkEMCALModules] ;   

  TH1I*     fhNEvents;                                                             
 
  TH1F *    fhCentrality;                                                          
  
  TH1F *    fhCentralitySelected;                                                  
  
  // Cluster time histograms
  TH2F*     fHTpi0[4];                                                             
  TH2F*     fhClusterTime ;                                                        
  TH2F*     fhClusterTimeSM[AliEMCALGeoParams::fgkEMCALModules] ;                  
  TH2F*     fhClusterPairDiffTime;                                                 
  TH2F*     fhClusterPairDiffTimeSameSM[AliEMCALGeoParams::fgkEMCALModules];       
  TH2F*     fhClusterPairDiffTimeSameSector[AliEMCALGeoParams::fgkEMCALModules/2]; 
  TH2F*     fhClusterPairDiffTimeSameSide[AliEMCALGeoParams::fgkEMCALModules-2];   

  AliAnalysisTaskEMCALPi0CalibSelection(           const AliAnalysisTaskEMCALPi0CalibSelection&) ;
    
  AliAnalysisTaskEMCALPi0CalibSelection& operator=(const AliAnalysisTaskEMCALPi0CalibSelection&) ;
  
  ClassDef(AliAnalysisTaskEMCALPi0CalibSelection,26) ;

};

#endif //ALIANALYSISTASKEMCALPI0CALIBSELECTION_H