[u/mrichter/AliRoot.git] / PWGGA / EMCALJetTasks / AliHadCorrTask.h

#ifndef ALIHADCORRTASK_H
#define ALIHADCORRTASK_H

// $Id$

class TClonesArray;
class TList;
class TH1F;
class TH2F;
class AliEmcalParticle;
class TString;

#include "AliAnalysisTaskEmcal.h"

class AliHadCorrTask : public AliAnalysisTaskEmcal {

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

  void         UserCreateOutputObjects();
  void         Terminate(Option_t *);

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

 protected:

  virtual Bool_t       Run()                                          ;
  virtual Bool_t       FillHistograms()                { return kTRUE ; }
  Int_t                GetMomBin(Double_t pt)                    const;
  Double_t             GetEtaSigma(Int_t pbin)                   const;
  Double_t             GetPhiMean(Int_t pbin, Int_t centbin)     const;
  Double_t             GetPhiSigma(Int_t pbin, Int_t centbin)    const;
  void                 DoTrackClusLoop()                              ;
  void                 DoMatchedTracksLoop(AliEmcalParticle *emccluster, Double_t &totalTrkP, Int_t &Nmatches);
  Double_t             ApplyHadCorrOneTrack(AliEmcalParticle *emccluster, Double_t hadCorr)                   ;
  Double_t             ApplyHadCorrAllTracks(AliEmcalParticle *emccluster, Double_t hadCorr)                  ;


  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

  TClonesArray          *fOutClusters;            //!output cluster collection

  TH2F                  *fHistMatchEtaPhi[8][9][2];  //!output histograms
  TH2F                  *fHistMatchEvsP[4];          //!output histograms
  TH2F                  *fHistNMatchEnergy[4];       //!output histograms
  TH2F                  *fHistNCellsEnergy[8][4];    //!output histograms
  TH2F                  *fHistMatchdRvsEP[4];        //!output histograms
  TH1F                  *fHistNclusvsCent;           //!output histograms
  TH1F                  *fHistNclusMatchvsCent;      //!output histograms
  TH1F                  *fHistEbefore;               //!output histograms
  TH1F                  *fHistEafter;                //!output histograms
  TH2F                  *fHistEoPCent;               //!output histograms
  TH2F                  *fHistNMatchCent;            //!output histograms
  TH2F                  *fHistNMatchCent_trk;        //!output histograms
  TH1F                  *fHistEsubPch[8];            //!output histograms
  TH2F                  *fHistEsubPchRat[8];         //!output histograms
  TH1F                  *fHistCentrality;            //!output histograms
  TH2F                  *fHistNoMatchEtaPhi;         //!output histograms

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

  ClassDef(AliHadCorrTask, 9) // Hadronic correction task
};
#endif
Commit	Line	Data
0b777a09	1	#ifndef ALIHADCORRTASK_H
	2	#define ALIHADCORRTASK_H
	3
	4	// $Id$
	5
	6	class TClonesArray;
0b777a09	7	class TList;
1f6c8e39	8	class TH1F;
	9	class TH2F;
	10	class AliEmcalParticle;
1f6fff78	11	class TString;
0b777a09	12
1f6c8e39	13	#include "AliAnalysisTaskEmcal.h"
	14
	15	class AliHadCorrTask : public AliAnalysisTaskEmcal {
0b777a09	16
0b777a09	17	public:
0b777a09	18	AliHadCorrTask();
d6f334b5	19	AliHadCorrTask(const char *name);
5d845887	20	AliHadCorrTask(const char *name, Bool_t histo);
0b777a09	21	virtual ~AliHadCorrTask();
	22
	23	void UserCreateOutputObjects();
1f6c8e39	24	void Terminate(Option_t *);
0b777a09	25
1f6c8e39	26	void SetOutClusName(const char *n) { fOutCaloName = n ; }
	27	void SetEtaMatch(Double_t eta) { fEtaMatch = eta ; }
	28	void SetPhiMatch(Double_t phi) { fPhiMatch = phi ; }
	29	void SetTrackClus(Int_t c) { fDoTrackClus = c ; }
	30	void SetHadCorr(Double_t c) { fHadCorr = c ; }
	31	void SetEexcl(Double_t Emin) { fEexclCell = Emin ; }
0b777a09	32
0b777a09	33	protected:
0b777a09	34
1f6c8e39	35	virtual Bool_t Run() ;
	36	virtual Bool_t FillHistograms() { return kTRUE ; }
	37	Int_t GetMomBin(Double_t pt) const;
	38	Double_t GetEtaSigma(Int_t pbin) const;
	39	Double_t GetPhiMean(Int_t pbin, Int_t centbin) const;
	40	Double_t GetPhiSigma(Int_t pbin, Int_t centbin) const;
	41	void DoTrackClusLoop() ;
	42	void DoMatchedTracksLoop(AliEmcalParticle *emccluster, Double_t &totalTrkP, Int_t &Nmatches);
	43	Double_t ApplyHadCorrOneTrack(AliEmcalParticle *emccluster, Double_t hadCorr) ;
	44	Double_t ApplyHadCorrAllTracks(AliEmcalParticle *emccluster, Double_t hadCorr) ;
	45
	46
d6f334b5	47	TString fOutCaloName; // name of output clusters
a3f43e7e	48	Double_t fPhiMatch; // phi match value (pp=0.050)
a3f43e7e	49	Double_t fEtaMatch; // eta match value (pp=0.025)
bcf52e4a	50	Int_t fDoTrackClus; // loop over tracks first
d6f334b5	51	Double_t fHadCorr; // hadronic correction (fraction)
1f6c8e39	52	Double_t fEexclCell; // Energy/cell that we cannot subtract from the clusters
1f6c8e39	53
d6f334b5	54	TClonesArray *fOutClusters; //!output cluster collection
1f6c8e39	55
	56	TH2F *fHistMatchEtaPhi[8][9][2]; //!output histograms
	57	TH2F *fHistMatchEvsP[4]; //!output histograms
	58	TH2F *fHistNMatchEnergy[4]; //!output histograms
	59	TH2F *fHistNCellsEnergy[8][4]; //!output histograms
	60	TH2F *fHistMatchdRvsEP[4]; //!output histograms
	61	TH1F *fHistNclusvsCent; //!output histograms
	62	TH1F *fHistNclusMatchvsCent; //!output histograms
	63	TH1F *fHistEbefore; //!output histograms
	64	TH1F *fHistEafter; //!output histograms
	65	TH2F *fHistEoPCent; //!output histograms
	66	TH2F *fHistNMatchCent; //!output histograms
	67	TH2F *fHistNMatchCent_trk; //!output histograms
	68	TH1F *fHistEsubPch[8]; //!output histograms
	69	TH2F *fHistEsubPchRat[8]; //!output histograms
	70	TH1F *fHistCentrality; //!output histograms
	71	TH2F *fHistNoMatchEtaPhi; //!output histograms
0b777a09	72
	73	private:
	74	AliHadCorrTask(const AliHadCorrTask&); // not implemented
	75	AliHadCorrTask &operator=(const AliHadCorrTask&); // not implemented
	76
1f6c8e39	77	ClassDef(AliHadCorrTask, 9) // Hadronic correction task
0b777a09	78	};
0b777a09	79	#endif