[u/mrichter/AliRoot.git] / PWG / EMCAL / AliHadCorrTask.h

#ifndef ALIHADCORRTASK_H
#define ALIHADCORRTASK_H

// $Id$

class TClonesArray;
class TList;
class TH1;
class TH2;
class AliEmcalParticle;
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(AliEmcalParticle *emccluster, Double_t hadCorr);
  Double_t               ApplyHadCorrAllTracks(AliEmcalParticle *emccluster, Double_t hadCorr);
  void                   DoMatchedTracksLoop(AliEmcalParticle *emccluster, 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;                   //!ESD/AOD mode
  TClonesArray          *fOutClusters;               //!output cluster collection

  // QA plots
  TH2                   *fHistMatchEtaPhi[8][9][2];  //!deta vs. dphi of matched cluster-track pairs
  TH2                   *fHistMatchEtaPhiAll;        //!deta vs. dphi of matched cluster-track pairs
  TH2                   *fHistMatchEtaPhiAllTr;      //!deta vs. dphi of all cluster-track pairs (tr loop)
  TH2                   *fHistMatchEtaPhiAllCl;      //!deta vs. dphi of all cluster-track pairs (cl loop)
  TH2                   *fHistMatchEvsP[4];          //!cluster energy vs. track momentum of matched pairs
  TH2                   *fHistNMatchEnergy[4];       //!n matches vs. cluster energy
  TH2                   *fHistNCellsEnergy[4][4];    //!n cells vs. cluster energy
  TH2                   *fHistMatchdRvsEP[4];        //!matching distance vs. E/P
  TH1                   *fHistNclusvsCent;           //!n clusters vs. centrality
  TH1                   *fHistNclusMatchvsCent;      //!n clusters matched to some track vs. centrality
  TH1                   *fHistEbefore;               //!average energy of clusters before correction vs. centrality
  TH1                   *fHistEafter;                //!average energy of clusters after correction vs. centrality
  TH2                   *fHistEoPCent;               //!E/P vs. centrality
  TH2                   *fHistNMatchCent;            //!n matches vs. centraity
  TH2                   *fHistNClusMatchCent;        //!n clusters macthed to some track (tracks allowed to match more than one cluster)
  TH1                   *fHistEsubPch[8];            //!Esub vs. total momentum of matched tracks (only 1 match)
  TH2                   *fHistEsubPchRat[8];         //!Esub/momentum of matched tracks vs. total momentum of matched tracks (only 1 match)
  TH2                   *fHistEsubPchRatAll[8];      //!Esub/momentum of matched tracks vs. total momentum of matched tracks (all number of matches)
  TH2                   *fHistEmbTrackMatchesOversub[4];    //!Over-subtracted energy / cluster energy with embedded track matches (non-embedded matches < 5%)
  TH2                   *fHistNonEmbTrackMatchesOversub[4]; //!Over-subtracted energy / cluster energy with non-embedded track matches (embedded matches < 5%)
  TH2                   *fHistOversubMCClusters[4];         //!Over-subtracted energy / cluster energy (cluster MC energy fraction > 95%)
  TH2                   *fHistOversubNonMCClusters[4];      //!Over-subtracted energy / cluster energy (cluster MC energy fraction < 5%)
  TH2                   *fHistOversub[4];                   //!Over-subtracted energy / cluster energy

 private:
  AliHadCorrTask(const AliHadCorrTask&);            // not implemented
  AliHadCorrTask &operator=(const AliHadCorrTask&); // not implemented

  ClassDef(AliHadCorrTask, 14) // Hadronic correction task
};
#endif
Commit	Line	Data
8417a2aa	1	#ifndef ALIHADCORRTASK_H
	2	#define ALIHADCORRTASK_H
	3
	4	// $Id$
	5
	6	class TClonesArray;
	7	class TList;
	8	class TH1;
	9	class TH2;
	10	class AliEmcalParticle;
	11	class TString;
	12
	13	#include "AliAnalysisTaskEmcal.h"
	14
	15	class AliHadCorrTask : public AliAnalysisTaskEmcal {
	16
	17	public:
	18	AliHadCorrTask();
	19	AliHadCorrTask(const char *name, Bool_t histo=kFALSE);
	20	virtual ~AliHadCorrTask();
	21
	22	void UserCreateOutputObjects();
	23
	24	void SetEexcl(Double_t Emin) { fEexclCell = Emin ; }
	25	void SetEtaMatch(Double_t eta) { fEtaMatch = eta ; }
	26	void SetHadCorr(Double_t c) { fHadCorr = c ; }
	27	void SetOutClusName(const char *n) { fOutCaloName = n ; }
	28	void SetPhiMatch(Double_t phi) { fPhiMatch = phi ; }
	29	void SetTrackClus(Int_t c) { fDoTrackClus = c ; }
	30	void SetDoExact(Bool_t d) { fDoExact = d ; }
	31
	32	protected:
	33	Double_t ApplyHadCorrOneTrack(AliEmcalParticle *emccluster, Double_t hadCorr);
	34	Double_t ApplyHadCorrAllTracks(AliEmcalParticle *emccluster, Double_t hadCorr);
	35	void DoMatchedTracksLoop(AliEmcalParticle *emccluster, Double_t &totalTrkP, Int_t &Nmatches, Double_t &trkPMCfrac, Int_t &NMCmatches);
	36	void DoTrackLoop();
	37	Double_t GetEtaSigma(Int_t pbin) const;
	38	UInt_t GetMomBin(Double_t pt) const;
	39	Double_t GetPhiMean(Int_t pbin, Int_t centbin) const;
	40	Double_t GetPhiSigma(Int_t pbin, Int_t centbin) const;
	41	Bool_t Run() ;
	42	void ExecOnce() ;
	43
	44	// Task configuration
	45	TString fOutCaloName; // name of output clusters
	46	Double_t fPhiMatch; // phi match value (pp=0.050)
	47	Double_t fEtaMatch; // eta match value (pp=0.025)
	48	Int_t fDoTrackClus; // loop over tracks first
	49	Double_t fHadCorr; // hadronic correction (fraction)
	50	Double_t fEexclCell; // energy/cell that we cannot subtract from the clusters
	51	Bool_t fDoExact; // do exact correction (embedding only)
	52
	53	// Service fields (non-streamed)
	54	Bool_t fEsdMode; //!ESD/AOD mode
	55	TClonesArray *fOutClusters; //!output cluster collection
	56
	57	// QA plots
	58	TH2 *fHistMatchEtaPhi[8][9][2]; //!deta vs. dphi of matched cluster-track pairs
	59	TH2 *fHistMatchEtaPhiAll; //!deta vs. dphi of matched cluster-track pairs
	60	TH2 *fHistMatchEtaPhiAllTr; //!deta vs. dphi of all cluster-track pairs (tr loop)
	61	TH2 *fHistMatchEtaPhiAllCl; //!deta vs. dphi of all cluster-track pairs (cl loop)
	62	TH2 *fHistMatchEvsP[4]; //!cluster energy vs. track momentum of matched pairs
	63	TH2 *fHistNMatchEnergy[4]; //!n matches vs. cluster energy
	64	TH2 *fHistNCellsEnergy[4][4]; //!n cells vs. cluster energy
65	TH2 *fHistMatchdRvsEP[4]; //!matching distance vs. E/P
66	TH1 *fHistNclusvsCent; //!n clusters vs. centrality
67	TH1 *fHistNclusMatchvsCent; //!n clusters matched to some track vs. centrality
68	TH1 *fHistEbefore; //!average energy of clusters before correction vs. centrality
69	TH1 *fHistEafter; //!average energy of clusters after correction vs. centrality
70	TH2 *fHistEoPCent; //!E/P vs. centrality
71	TH2 *fHistNMatchCent; //!n matches vs. centraity
72	TH2 *fHistNClusMatchCent; //!n clusters macthed to some track (tracks allowed to match more than one cluster)
73	TH1 *fHistEsubPch[8]; //!Esub vs. total momentum of matched tracks (only 1 match)
74	TH2 *fHistEsubPchRat[8]; //!Esub/momentum of matched tracks vs. total momentum of matched tracks (only 1 match)
75	TH2 *fHistEsubPchRatAll[8]; //!Esub/momentum of matched tracks vs. total momentum of matched tracks (all number of matches)
76	TH2 *fHistEmbTrackMatchesOversub[4]; //!Over-subtracted energy / cluster energy with embedded track matches (non-embedded matches < 5%)
77	TH2 *fHistNonEmbTrackMatchesOversub[4]; //!Over-subtracted energy / cluster energy with non-embedded track matches (embedded matches < 5%)
78	TH2 *fHistOversubMCClusters[4]; //!Over-subtracted energy / cluster energy (cluster MC energy fraction > 95%)
79	TH2 *fHistOversubNonMCClusters[4]; //!Over-subtracted energy / cluster energy (cluster MC energy fraction < 5%)
80	TH2 *fHistOversub[4]; //!Over-subtracted energy / cluster energy
81
82	private:
83	AliHadCorrTask(const AliHadCorrTask&); // not implemented
84	AliHadCorrTask &operator=(const AliHadCorrTask&); // not implemented
85
86	ClassDef(AliHadCorrTask, 14) // Hadronic correction task
87	};
88	#endif