AliHadCorrTask.h

Go to the documentation of this file.

#ifndef ALIHADCORRTASK_H
#define ALIHADCORRTASK_H

class TClonesArray;
class TList;
class TH1;
class TH2;
class AliVCluster;
class TString;

#include "AliAnalysisTaskEmcal.h"

class AliHadCorrTask : public AliAnalysisTaskEmcal {

 public:
  AliHadCorrTask();
  AliHadCorrTask(const char *name, Bool_t histo=kFALSE); 
  virtual ~AliHadCorrTask();

  void                   UserCreateOutputObjects();

  void                   SetEexcl(Double_t Emin)                 { fEexclCell      = Emin ; }
  void                   SetEtaMatch(Double_t eta)               { fEtaMatch       = eta  ; }
  void                   SetHadCorr(Double_t c)                  { fHadCorr        = c    ; }
  void                   SetOutClusName(const char *n)           { fOutCaloName    = n    ; }
  void                   SetPhiMatch(Double_t phi)               { fPhiMatch       = phi  ; }
  void                   SetTrackClus(Int_t c)                   { fDoTrackClus    = c    ; }
  void                   SetDoExact(Bool_t d)                    { fDoExact        = d    ; }

 protected:
  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;
  Bool_t                 Run()                                          ;
  void                   ExecOnce()                                     ;

  // Task configuration
  TString                fOutCaloName;               // name of output clusters
  Double_t               fPhiMatch;                  // phi match value (pp=0.050)
  Double_t               fEtaMatch;                  // eta match value (pp=0.025)
  Int_t                  fDoTrackClus;               // loop over tracks first
  Double_t               fHadCorr;                   // hadronic correction (fraction)
  Double_t               fEexclCell;                 // energy/cell that we cannot subtract from the clusters
  Bool_t                 fDoExact;                   // do exact correction (embedding only) 

  // Service fields (non-streamed)
  Bool_t                 fEsdMode;                   
  TClonesArray          *fOutClusters;               

  // QA plots
  TH2                   *fHistMatchEtaPhi[10][9][2];  
  TH2                   *fHistMatchEtaPhiAll;        
  TH2                   *fHistMatchEtaPhiAllTr;      
  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:
  AliHadCorrTask(const AliHadCorrTask&);            // not implemented
  AliHadCorrTask &operator=(const AliHadCorrTask&); // not implemented

  ClassDef(AliHadCorrTask, 15) // Hadronic correction task
};
#endif