AliEmcalCorrectionClusterHadronicCorrection.h

Go to the documentation of this file.

#ifndef ALIEMCALCORRECTIONCLUSTERHADRONICCORRECTION_H
#define ALIEMCALCORRECTIONCLUSTERHADRONICCORRECTION_H

#include "AliEmcalCorrectionComponent.h"

#if !(defined(__CINT__) || defined(__MAKECINT__))
#include "AliEmcalContainerIndexMap.h"
#endif

class TH1;
class TH2;

class AliEmcalCorrectionClusterHadronicCorrection : public AliEmcalCorrectionComponent {
 public:
  AliEmcalCorrectionClusterHadronicCorrection();
  virtual ~AliEmcalCorrectionClusterHadronicCorrection();

  // Sets up and runs the task
  Bool_t Initialize();
  void UserCreateOutputObjects();
  void ExecOnce();
  Bool_t Run();
  
protected:
  void                   GenerateTrackToContainerMap();
  Double_t               ApplyHadCorrOneTrack(Int_t icluster, Double_t hadCorr);
  Double_t               ApplyHadCorrAllTracks(Int_t icluster, Double_t hadCorr);
  void                   DoMatchedTracksLoop(Int_t icluster, Double_t &totalTrkP, Int_t &Nmatches, Double_t &trkPMCfrac, Int_t &NMCmatches);
  void                   DoTrackLoop();
  Double_t               GetEtaSigma(Int_t pbin)                   const;
  UInt_t                 GetMomBin(Double_t pt)                    const;
  Double_t               GetPhiMean(Int_t pbin, Int_t centbin)     const;
  Double_t               GetPhiSigma(Int_t pbin, Int_t centbin)    const;
  Double_t               ComputeM02Subtraction(const AliVCluster* cluster, Double_t energyclus, Int_t Nmatches, Double_t totalTrkP, Double_t hadCorr);
  
  // Task configuration
  Double_t               fPhiMatch;                  
  Double_t               fEtaMatch;                  
  Bool_t                 fDoTrackClus;               
  Double_t               fHadCorr;                   
  Double_t               fEexclCell;                 
  Bool_t                 fPlotOversubtractionHistograms; 
  Bool_t                 fDoNotOversubtract;         
  Bool_t                 fUseM02SubtractionScheme;   
  Bool_t                 fUseConstantSubtraction;    
  Double_t               fConstantSubtractionValue;  

#if !(defined(__CINT__) || defined(__MAKECINT__))
  // Handle mapping between index and containers
  AliEmcalContainerIndexMap <AliClusterContainer, AliVCluster> fClusterContainerIndexMap;    
  AliEmcalContainerIndexMap <AliParticleContainer, AliVParticle> fParticleContainerIndexMap; 
#endif
  std::map <AliVTrack *, AliParticleContainer *> fTrackToContainerMap;                       
  
  // QA plots
  TH2                   *fHistMatchEtaPhi[10][9][2];  
  TH2                   *fHistMatchEtaPhiAll;        
  TH2                   *fHistMatchEtaPhiAllCl;      
  TH2                   *fHistMatchEvsP[5];          
  TH2                   *fHistNMatchEnergy[5];       
  TH2                   *fHistNCellsEnergy[5][4];    
  TH2                   *fHistMatchdRvsEP[5];        
  TH1                   *fHistNclusvsCent;           
  TH1                   *fHistNclusMatchvsCent;      
  TH1                   *fHistEbefore;               
  TH1                   *fHistEafter;                
  TH2                   *fHistEoPCent;               
  TH2                   *fHistNMatchCent;            
  TH2                   *fHistNClusMatchCent;        
  TH1                   *fHistEsubPch[10];           
  TH2                   *fHistEsubPchRat[10];        
  TH2                   *fHistEsubPchRatAll[5];      
  TH2                   *fHistEmbTrackMatchesOversub[5];    
  TH2                   *fHistNonEmbTrackMatchesOversub[5]; 
  TH2                   *fHistOversubMCClusters[5];         
  TH2                   *fHistOversubNonMCClusters[5];      
  TH2                   *fHistOversub[5];                   

 private:
  AliEmcalCorrectionClusterHadronicCorrection(const AliEmcalCorrectionClusterHadronicCorrection &);               // Not implemented
  AliEmcalCorrectionClusterHadronicCorrection &operator=(const AliEmcalCorrectionClusterHadronicCorrection &);    // Not implemented

  // Allows the registration of the class so that it is available to be used by the correction task.
  static RegisterCorrectionComponent<AliEmcalCorrectionClusterHadronicCorrection> reg;
  
  ClassDef(AliEmcalCorrectionClusterHadronicCorrection, 5); // EMCal cluster hadronic correction component
};

#endif /* ALIEMCALCORRECTIONCLUSTERHADRONICCORRECTION_H */