[u/mrichter/AliRoot.git] / PWGJE / UserTasks / AliAnalysisTaskJetsTriggerTRD.h

#include "TH1.h"
#include "TH2.h"
#include "TH3.h"

#include "AliLog.h"

#include "AliAnalysisTaskSE.h"

#define ID(x) x, #x

class TList;

class AliAnalysisTaskJetsTriggerTRD :
  public AliAnalysisTaskSE
{
public:
  AliAnalysisTaskJetsTriggerTRD(const char *name = "jets_trg_trd");
  ~AliAnalysisTaskJetsTriggerTRD();

  // analysis operations
  virtual void   UserCreateOutputObjects();
  virtual Bool_t Notify();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(const Option_t *option);

  // task configuration
  void  SetNumberOfJetPtBins(Int_t n) { fNoJetPtBins = n; }
  Int_t GetNumberOfJetPtBins() const { return fNoJetPtBins; }

  void    SetJetPtBinMax(Float_t ptmax) { fJetPtBinMax = ptmax; }
  Float_t GetJetPtBinMax() const { return fJetPtBinMax; }

  void SetJetBranchName(const char* branchName) { strncpy(fJetBranchName, branchName, fgkStringLength-1); }
  const char* GetJetBranchName() const { return fJetBranchName; }

  // histograms
  enum Hist_t {
      kHistStat = 0,
      kHistJetPtMC,
      kHistNoJets,
      kHistTrackGTU,
      kHistNPtMin,
      kHistLeadJetPt,
      kHistJetPt,
      kHistJetPtITS,
      kHistJetPt3x3,
      kHistJetPtNoTracks3,
      kHistLast
  };

  // statistics
  enum Stat_t {
      kStatSeen = 1,
      kStatTrg,
      kStatUsed,
      kStatEvCuts,
      kStatLast
  };

  // trigger conditions
  enum Trigger_t {
    // untriggered
    kTrgMinBias = 1, // CINT1
    kTrgInt7,
    kTrgInt8,
    kTrgEMC7,
    kTrgEMC8,
    // TRD jet trigger (HJT)
    kTrgInt7WUHJT,
    kTrgInt8WUHJT,
    kTrgEMC7WUHJT,
    kTrgEMC8WUHJT,
    kTrgEMCEJE,
    kTrgEMCEGA,
    //
    kTrgLast
  };	   

protected:
  UInt_t fTriggerMask;		// internal representation of trigger conditions

  Bool_t DetectTriggers();
  void   MarkTrigger(Trigger_t trg) { fTriggerMask |= (1 << trg); }
  Bool_t IsTrigger(Trigger_t trg) const { return (fTriggerMask & (1 << trg)); }

  // output objects
  TList *fOutputList;		// list of output objects

  // histogram management
  TH1  *fHist[kHistLast];	//! pointers to histogram
  const char *fShortTaskId;	//! short identifier for the task

  TH1*&  GetHistogram(Hist_t hist, Int_t idx = 0) { return fHist[hist + idx]; }

  TH1*   AddHistogram(Hist_t hist, const char *hid, TString title,
                      Int_t xbins, Float_t xmin, Float_t xmax, Int_t binType = 1);
  TH2*   AddHistogram(Hist_t hist, const char *hid, TString title,
                      Int_t xbins, Float_t xmin, Float_t xmax,
                      Int_t ybins, Float_t ymin, Float_t ymax, Int_t binType = 1);
  TH3*   AddHistogram(Hist_t hist, const char *hid, TString title,
                      Int_t xbins, Float_t xmin, Float_t xmax,
                      Int_t ybins, Float_t ymin, Float_t ymax,
                      Int_t zbins, Float_t zmin, Float_t zmax, Int_t binType = 1);

  void    FillH1(Hist_t hist, Float_t x, Float_t weight = 1., Int_t idx = 0)
  { GetHistogram(hist, idx)->Fill(x, weight); }
  void    FillH2(Hist_t hist, Float_t x, Float_t y, Float_t weight = 1., Int_t idx = 0)
  { ((TH2*) GetHistogram(hist, idx))->Fill(x, y, weight); }
  void    FillH3(Hist_t hist, Float_t x, Float_t y, Float_t z, Float_t weight = 1., Int_t idx = 0)
  { ((TH3*) GetHistogram(hist, idx))->Fill(x, y, z, weight); }

  // task configuration
  Int_t    fNoJetPtBins;                // number of bins for jet pt
  Float_t  fJetPtBinMax;                // max jet pt (GeV) in histograms

  Float_t  fXsection;			// x-section from PYTHIA
  Float_t  fAvgTrials;			// ratio of PYTHIA events
					// over accepted events
  Float_t  fPtHard;			// pt hard

  static const Int_t fgkStringLength = 100; // max length for the jet branch name
  char fJetBranchName[fgkStringLength];     // jet branch name

  // not implemented
  AliAnalysisTaskJetsTriggerTRD(const AliAnalysisTaskJetsTriggerTRD &rhs);
  AliAnalysisTaskJetsTriggerTRD& operator=(const AliAnalysisTaskJetsTriggerTRD &rhs);

  ClassDef(AliAnalysisTaskJetsTriggerTRD, 1);
};
Commit	Line	Data
7032c043	1	#include "TH1.h"
	2	#include "TH2.h"
	3	#include "TH3.h"
	4
	5	#include "AliLog.h"
	6
	7	#include "AliAnalysisTaskSE.h"
	8
e7808ddf	9	#define ID(x) x, #x
7032c043	10
	11	class TList;
	12
	13	class AliAnalysisTaskJetsTriggerTRD :
	14	public AliAnalysisTaskSE
	15	{
	16	public:
	17	AliAnalysisTaskJetsTriggerTRD(const char *name = "jets_trg_trd");
	18	~AliAnalysisTaskJetsTriggerTRD();
	19
	20	// analysis operations
	21	virtual void UserCreateOutputObjects();
	22	virtual Bool_t Notify();
	23	virtual void UserExec(Option_t *option);
	24	virtual void Terminate(const Option_t *option);
	25
	26	// task configuration
	27	void SetNumberOfJetPtBins(Int_t n) { fNoJetPtBins = n; }
	28	Int_t GetNumberOfJetPtBins() const { return fNoJetPtBins; }
	29
	30	void SetJetPtBinMax(Float_t ptmax) { fJetPtBinMax = ptmax; }
	31	Float_t GetJetPtBinMax() const { return fJetPtBinMax; }
	32
f61ddc78	33	void SetJetBranchName(const char* branchName) { strncpy(fJetBranchName, branchName, fgkStringLength-1); }
7032c043	34	const char* GetJetBranchName() const { return fJetBranchName; }
7032c043	35
e7808ddf	36	// histograms
	37	enum Hist_t {
	38	kHistStat = 0,
a26539a7	39	kHistJetPtMC,
e7808ddf	40	kHistNoJets,
e7808ddf	41	kHistTrackGTU,
a26539a7	42	kHistNPtMin,
	43	kHistLeadJetPt,
	44	kHistJetPt,
	45	kHistJetPtITS,
	46	kHistJetPt3x3,
e7808ddf	47	kHistJetPtNoTracks3,
	48	kHistLast
	49	};
	50
	51	// statistics
	52	enum Stat_t {
	53	kStatSeen = 1,
a26539a7	54	kStatTrg,
e7808ddf	55	kStatUsed,
a26539a7	56	kStatEvCuts,
e7808ddf	57	kStatLast
	58	};
	59
	60	// trigger conditions
a26539a7	61	enum Trigger_t {
	62	// untriggered
	63	kTrgMinBias = 1, // CINT1
	64	kTrgInt7,
	65	kTrgInt8,
	66	kTrgEMC7,
	67	kTrgEMC8,
	68	// TRD jet trigger (HJT)
	69	kTrgInt7WUHJT,
	70	kTrgInt8WUHJT,
	71	kTrgEMC7WUHJT,
	72	kTrgEMC8WUHJT,
6883db3e	73	kTrgEMCEJE,
6883db3e	74	kTrgEMCEGA,
a26539a7	75	//
a26539a7	76	kTrgLast
e7808ddf	77	};
7032c043	78
7032c043	79	protected:
e7808ddf	80	UInt_t fTriggerMask; // internal representation of trigger conditions
	81
	82	Bool_t DetectTriggers();
	83	void MarkTrigger(Trigger_t trg) { fTriggerMask \|= (1 << trg); }
	84	Bool_t IsTrigger(Trigger_t trg) const { return (fTriggerMask & (1 << trg)); }
	85
	86	// output objects
7032c043	87	TList *fOutputList; // list of output objects
	88
	89	// histogram management
	90	TH1 *fHist[kHistLast]; //! pointers to histogram
	91	const char *fShortTaskId; //! short identifier for the task
	92
f61ddc78	93	TH1*& GetHistogram(Hist_t hist, Int_t idx = 0) { return fHist[hist + idx]; }
7032c043	94
	95	TH1* AddHistogram(Hist_t hist, const char *hid, TString title,
	96	Int_t xbins, Float_t xmin, Float_t xmax, Int_t binType = 1);
	97	TH2* AddHistogram(Hist_t hist, const char *hid, TString title,
	98	Int_t xbins, Float_t xmin, Float_t xmax,
	99	Int_t ybins, Float_t ymin, Float_t ymax, Int_t binType = 1);
	100	TH3* AddHistogram(Hist_t hist, const char *hid, TString title,
	101	Int_t xbins, Float_t xmin, Float_t xmax,
	102	Int_t ybins, Float_t ymin, Float_t ymax,
	103	Int_t zbins, Float_t zmin, Float_t zmax, Int_t binType = 1);
	104
	105	void FillH1(Hist_t hist, Float_t x, Float_t weight = 1., Int_t idx = 0)
	106	{ GetHistogram(hist, idx)->Fill(x, weight); }
	107	void FillH2(Hist_t hist, Float_t x, Float_t y, Float_t weight = 1., Int_t idx = 0)
	108	{ ((TH2*) GetHistogram(hist, idx))->Fill(x, y, weight); }
	109	void FillH3(Hist_t hist, Float_t x, Float_t y, Float_t z, Float_t weight = 1., Int_t idx = 0)
	110	{ ((TH3*) GetHistogram(hist, idx))->Fill(x, y, z, weight); }
	111
	112	// task configuration
	113	Int_t fNoJetPtBins; // number of bins for jet pt
	114	Float_t fJetPtBinMax; // max jet pt (GeV) in histograms
	115
a26539a7	116	Float_t fXsection; // x-section from PYTHIA
	117	Float_t fAvgTrials; // ratio of PYTHIA events
	118	// over accepted events
	119	Float_t fPtHard; // pt hard
	120
7032c043	121	static const Int_t fgkStringLength = 100; // max length for the jet branch name
	122	char fJetBranchName[fgkStringLength]; // jet branch name
	123
	124	// not implemented
	125	AliAnalysisTaskJetsTriggerTRD(const AliAnalysisTaskJetsTriggerTRD &rhs);
	126	AliAnalysisTaskJetsTriggerTRD& operator=(const AliAnalysisTaskJetsTriggerTRD &rhs);
	127
	128	ClassDef(AliAnalysisTaskJetsTriggerTRD, 1);
	129	};