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

#ifndef ALIHADCORRTASK_H
#define ALIHADCORRTASK_H

// $Id$

class TClonesArray;
class TList;
class TH1;
class TH2;

#include "AliAnalysisTaskSE.h"

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

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

  void         SetClusName(const char *n)              { fCaloName       = n;   }
  void         SetEtaMatch(Double_t eta)               { fEtaMatch       = eta; }
  void         SetHadCorr(Double_t c)                  { fHadCorr        = c;   }
  void         SetMinPt(Double_t min)                  { fMinPt          = min; }
  void         SetOutClusName(const char *n)           { fOutCaloName    = n;   }
  void         SetPhiMatch(Double_t phi)               { fPhiMatch       = phi; }
  void         SetTracksName(const char *n)            { fTracksName     = n;   }
  void         SetTrackClus(Int_t c)                   { fDoTrackClus    = c;   }

 protected:
  Int_t        GetCentBin(Double_t cent) const;
  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;
  TString      GetBeamType();

  TString                fTracksName;             // name of track collection
  TString                fCaloName;               // name of calo cluster collection
  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               fMinPt;                  // minimum pt (on tracks and clusters)
  Bool_t                 fCreateHisto;            // whether or not create histograms
  TClonesArray          *fOutClusters;            //!output cluster collection
  TList                 *fOutputList;             //!output list
  TH2                   *fHistMatchEtaPhi[8][9];  //!output histograms
  TH2                   *fHistMatchEvsP[4];       //!output histograms
  TH2                   *fHistMatchdRvsEP[4];     //!output histograms
  TH1                   *fHistNclusvsCent;        //!output histograms
  TH1                   *fHistNclusMatchvsCent;   //!output histograms
  TH1                   *fHistEbefore;            //!output histograms
  TH1                   *fHistEafter;             //!output histograms
  TH2                   *fHistEoPCent;            //!output histograms
  TH2                   *fHistNMatchCent;         //!output histograms
  TH2                   *fHistNMatchCent_trk;     //!output histograms
  TH1                   *fHistEsubPch[4][3];      //!output histograms
  TH2                   *fHistEsubPchRat[4][3];   //!output histograms
  TH1                   *fHistCentrality;         //!output histograms

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

  ClassDef(AliHadCorrTask, 6) // 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;
	8	class TH1;
	9	class TH2;
	10
	11	#include "AliAnalysisTaskSE.h"
	12
	13	class AliHadCorrTask : public AliAnalysisTaskSE {
	14	public:
0b777a09	15	AliHadCorrTask();
d6f334b5	16	AliHadCorrTask(const char *name);
5d845887	17	AliHadCorrTask(const char *name, Bool_t histo);
0b777a09	18	virtual ~AliHadCorrTask();
	19
	20	void UserCreateOutputObjects();
	21	void UserExec(Option_t *option);
	22	void Terminate(Option_t *option);
	23
bcf52e4a	24	void SetClusName(const char *n) { fCaloName = n; }
	25	void SetEtaMatch(Double_t eta) { fEtaMatch = eta; }
	26	void SetHadCorr(Double_t c) { fHadCorr = c; }
	27	void SetMinPt(Double_t min) { fMinPt = min; }
	28	void SetOutClusName(const char *n) { fOutCaloName = n; }
	29	void SetPhiMatch(Double_t phi) { fPhiMatch = phi; }
	30	void SetTracksName(const char *n) { fTracksName = n; }
	31	void SetTrackClus(Int_t c) { fDoTrackClus = c; }
0b777a09	32
0b777a09	33	protected:
d6f334b5	34	Int_t GetCentBin(Double_t cent) const;
d6f334b5	35	Int_t GetMomBin(Double_t pt) const;
f6e8a928	36	Double_t GetEtaSigma(Int_t pbin) const;
	37	Double_t GetPhiMean(Int_t pbin, Int_t centbin) const;
	38	Double_t GetPhiSigma(Int_t pbin, Int_t centbin) const;
3a385c49	39	TString GetBeamType();
0b777a09	40
	41	TString fTracksName; // name of track collection
	42	TString fCaloName; // name of calo cluster collection
d6f334b5	43	TString fOutCaloName; // name of output clusters
a3f43e7e	44	Double_t fPhiMatch; // phi match value (pp=0.050)
a3f43e7e	45	Double_t fEtaMatch; // eta match value (pp=0.025)
bcf52e4a	46	Int_t fDoTrackClus; // loop over tracks first
d6f334b5	47	Double_t fHadCorr; // hadronic correction (fraction)
d6f334b5	48	Double_t fMinPt; // minimum pt (on tracks and clusters)
bcf52e4a	49	Bool_t fCreateHisto; // whether or not create histograms
d6f334b5	50	TClonesArray *fOutClusters; //!output cluster collection
d6f334b5	51	TList *fOutputList; //!output list
65c0ba0f	52	TH2 *fHistMatchEtaPhi[8][9]; //!output histograms
d6f334b5	53	TH2 *fHistMatchEvsP[4]; //!output histograms
	54	TH2 *fHistMatchdRvsEP[4]; //!output histograms
	55	TH1 *fHistNclusvsCent; //!output histograms
	56	TH1 *fHistNclusMatchvsCent; //!output histograms
	57	TH1 *fHistEbefore; //!output histograms
	58	TH1 *fHistEafter; //!output histograms
	59	TH2 *fHistEoPCent; //!output histograms
	60	TH2 *fHistNMatchCent; //!output histograms
32bf39af	61	TH2 *fHistNMatchCent_trk; //!output histograms
65c0ba0f	62	TH1 *fHistEsubPch[4][3]; //!output histograms
65c0ba0f	63	TH2 *fHistEsubPchRat[4][3]; //!output histograms
32bf39af	64	TH1 *fHistCentrality; //!output histograms
0b777a09	65
	66	private:
	67	AliHadCorrTask(const AliHadCorrTask&); // not implemented
	68	AliHadCorrTask &operator=(const AliHadCorrTask&); // not implemented
	69
f6e8a928	70	ClassDef(AliHadCorrTask, 6) // Hadronic correction task
0b777a09	71	};
0b777a09	72	#endif