[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskHJetEmbed.h

#ifndef ALIANALYSISTASKHJETEMBED_H
#define ALIANALYSISTASKHJETEMBED_H

#include <vector>

class TH2F;
class TH1F;
class TF1;
class TH3F;
class THnSparse;
class TClonesArray;
class TObject;
class TString;
class AliNamedString;
class AliAODEvent;
class AliESDEvent;
class AliMCEvent;
class AliRhoParameter;
class TRandom3;
class AliEmcalJet;
class AliVTrack;

template<class T> class TParameter;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskHJetEmbed : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskHJetEmbed();
  AliAnalysisTaskHJetEmbed(const char *name);
  virtual ~AliAnalysisTaskHJetEmbed();

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

  void SetMaxVtxZ(Double_t z)                     { fMaxVtxZ=z;                     }
  void SetCollisionSystem(char *s)                { fCollisionSystem=s;             }
  void SetRunPeriod(char *p)                      { fPeriod=p;                      }
  void SetMCParticleArrName(char *s)              { fMCParticleArrName=s;           }
  void SetTrackArrName(char *s)                   { fTrackArrName=s;                }
  void SetTrkPtRange(Double_t min, Double_t max)  { fMinTrkPt=min; fMaxTrkPt=max;   }
  void SetTrkPhiRange(Double_t min, Double_t max) { fMinTrkPhi=min; fMaxTrkPhi=max; }
  void SetTrkEtaRange(Double_t min, Double_t max) { fMinTrkEta=min; fMaxTrkEta=max; }
  void SetTTtype(Int_t type)                      { fTTtype=type;                   }
  void SetRadius(Double_t rad)                    { fRadius=rad;                    }
  void SetPLJetArrName(char *s)                   { fPLJetArrName=s;                }
  void SetDLJetArrName(char *s)                   { fDLJetArrName=s;                }
  void SetJetArrName(char *s)                     { fJetArrName=s;                  }
  void SetRhoName(char *s)                        { fRhoName=s;                     }
  void SetRunQA(Bool_t run)                       { fRunQA=run;                     }
  void SetRunPL(Bool_t run)                       { fRunPL=run;                     }
  void SetRunDL(Bool_t run)                       { fRunDL=run;                     }
  void SetRunHJet(Bool_t run)                     { fRunHJet=run;                   }
  void SetRunMatch(Bool_t run)                    { fRunMatch=run;                  }
  void SetTTRanges(Double_t *min, Double_t *max)
  {
    for(Int_t i=0; i<kNTT; i++)
      {
	fMinTTPt[i] = min[i];
	fMaxTTPt[i] = max[i];
      }
  }


protected:
  void         RunQA();
  void         RunHJet(const Double_t minPt, const Double_t maxPt);
  void         RunMatch(const TClonesArray *tracks, const Int_t leadingIndex);
  void         FillHJetCor(const TClonesArray *tracks, const Int_t leadingIndex, const TClonesArray *jetArray, THnSparse *hTT, THnSparse *hn, Bool_t isBkg = kFALSE);
  Int_t        FindGeoMatchedJet(const AliEmcalJet* jet, const TClonesArray *jetArray, Double_t &dR);
  Int_t        FindEnergyMatchedJet(const AliEmcalJet* jet, const TClonesArray *jetArray, Double_t &dR, Double_t &fraction);
  Bool_t       AcceptTrack(const AliVParticle *track);
  Bool_t       IsGoodJet(const AliEmcalJet* jet);
  Double_t     GetZ(const Double_t trkPx, const Double_t trkPy, const Double_t trkPz, const Double_t jetPx, const Double_t jetPy, const Double_t jetPz);
  Double_t     GetDPhi(const Double_t phi1, const Double_t phi2);
  Double_t     CalculateDPhi(const Double_t phi1, const Double_t phi2);
  Double_t     GetJetDistance(const AliEmcalJet *jet1, const AliEmcalJet* jet2);

  enum { kNTrig = 3  };
  enum { kNTT = 4 };

 private:
  Int_t             fVerbosity;            //! Control output
  Int_t             fAnaType;              //  0-local; 1-grid
  TString           fPeriod;               //  Run period
  TString           fCollisionSystem;      //  Collision system
  AliVEvent         *fEvent;               //! Input event 
  Int_t             fTriggerType;          //! Trigger type of the event
  Double_t          fCentrality;           //! V0M for current event
  Double_t          fMaxVtxZ;              //  Maximum z of vertices
  TString           fMCParticleArrName;    //
  TClonesArray      *fMCParticleArray;     //!
  TString           fTrackArrName;         //  Name of the input track array
  TClonesArray      *fTrackArray;          //! Array of input tracks
  Int_t             fTriggerTrkIndex;      //! Index of the trigger track in the event 
  Double_t          fMinTrkPt;             //
  Double_t          fMaxTrkPt;             //
  Double_t          fMinTrkEta;            //
  Double_t          fMaxTrkEta;            //
  Double_t          fMinTrkPhi;            //
  Double_t          fMaxTrkPhi;            //
  Int_t             fTTtype;               //
  Double_t          fMinTTPt[kNTT];        //
  Double_t          fMaxTTPt[kNTT];        //
  Double_t          fRadius;               //  Jet radius
  TString           fJetArrName;           //  Name of the found jet array
  TString           fPLJetArrName;         //  Name of the embedded PYTHIA jet array on particle level
  TString           fDLJetArrName;         //  Name of the embedded PYTHIA jet array on detector level
  TClonesArray      *fJetArray;            //! Array of the found jets
  TClonesArray      *fPLJetArray;          //! Array of the embedded PYTHIA jet array on particle level
  TClonesArray      *fDLJetArray;          //! Array of the embedded PYTHIA jet array on detector level
  TString           fRhoName;              //  Name of the rho parameter
  AliRhoParameter   *fRho;                 //! Rho parameter
  Double_t          fRhoValue;             //! Value of the rho parameter
  AliNamedString    *fPtHardBinName;       //!Pt hard bin param
  Int_t             fPtHardBin;            //!        
  TRandom3          *fRandom;              //!

  Bool_t            fRunQA;                //  Flag to run QA
  Bool_t            fRunHJet;              //  Flag to run h+jet 
  Bool_t            fRunMatch;             //  Flag to run matching
  Bool_t            fRunPL;                //
  Bool_t            fRunDL;                //

  TList             *fOutputList;                //! Output list
  TH1F              *fhEventStat;                //!
  TH1F              *fhPtHardBins;               //!

  TH1F              *fhVtxZ[kNTrig];             //!
  TH1F              *fhCentrality[kNTrig];       //!
  TH2F              *fhRhoVsCent[kNTrig];        //!

  // QA histograms
  THnSparse         *fhPLJetPtVsCent[kNTrig];    //!
  THnSparse         *fhDLJetPtVsCent[kNTrig];    //!

  // HJet
  THnSparse         *fhPLTT[kNTrig];             //!
  THnSparse         *fhDLTT[kNTrig];             //!          
  THnSparse         *fhPLHJet[kNTrig];           //!
  THnSparse         *fhDLHJet[kNTrig];           //!
  THnSparse         *fhTTPtQA[kNTrig];           //!
  THnSparse         *fhTTPt[kNTrig];             //!
  THnSparse         *fhHJet[kNTrig];             //!


  // Matching
  THnSparse         *fhJetPtGeoMatch[kNTrig];    //!
  THnSparse         *fhJetPtEnMatch[kNTrig];     //!

  THnSparse         *fhJetPhiGeoMatch[kNTrig];   //!
  THnSparse         *fhJetPhiEnMatch[kNTrig];    //!

  AliAnalysisTaskHJetEmbed(const AliAnalysisTaskHJetEmbed&);            // not implemented
  AliAnalysisTaskHJetEmbed &operator=(const AliAnalysisTaskHJetEmbed&); // not implemented

  ClassDef(AliAnalysisTaskHJetEmbed, 3);
};

#endif
Commit	Line	Data
ee87a008	1	#ifndef ALIANALYSISTASKHJETEMBED_H
	2	#define ALIANALYSISTASKHJETEMBED_H
	3
	4	#include <vector>
	5
	6	class TH2F;
	7	class TH1F;
	8	class TF1;
	9	class TH3F;
	10	class THnSparse;
	11	class TClonesArray;
	12	class TObject;
	13	class TString;
aaea4d43	14	class AliNamedString;
ee87a008	15	class AliAODEvent;
	16	class AliESDEvent;
	17	class AliMCEvent;
	18	class AliRhoParameter;
	19	class TRandom3;
	20	class AliEmcalJet;
	21	class AliVTrack;
	22
	23	template<class T> class TParameter;
	24
	25	#include "AliAnalysisTaskSE.h"
	26
	27	class AliAnalysisTaskHJetEmbed : public AliAnalysisTaskSE {
	28	public:
	29	AliAnalysisTaskHJetEmbed();
	30	AliAnalysisTaskHJetEmbed(const char *name);
	31	virtual ~AliAnalysisTaskHJetEmbed();
	32
	33	void UserCreateOutputObjects();
	34	void UserExec(Option_t *option);
	35	void Terminate(Option_t *);
	36	void PrintConfig();
	37
	38	void SetMaxVtxZ(Double_t z) { fMaxVtxZ=z; }
	39	void SetCollisionSystem(char *s) { fCollisionSystem=s; }
	40	void SetRunPeriod(char *p) { fPeriod=p; }
	41	void SetMCParticleArrName(char *s) { fMCParticleArrName=s; }
	42	void SetTrackArrName(char *s) { fTrackArrName=s; }
	43	void SetTrkPtRange(Double_t min, Double_t max) { fMinTrkPt=min; fMaxTrkPt=max; }
	44	void SetTrkPhiRange(Double_t min, Double_t max) { fMinTrkPhi=min; fMaxTrkPhi=max; }
	45	void SetTrkEtaRange(Double_t min, Double_t max) { fMinTrkEta=min; fMaxTrkEta=max; }
3ec095a2	46	void SetTTtype(Int_t type) { fTTtype=type; }
ee87a008	47	void SetRadius(Double_t rad) { fRadius=rad; }
	48	void SetPLJetArrName(char *s) { fPLJetArrName=s; }
	49	void SetDLJetArrName(char *s) { fDLJetArrName=s; }
	50	void SetJetArrName(char *s) { fJetArrName=s; }
	51	void SetRhoName(char *s) { fRhoName=s; }
	52	void SetRunQA(Bool_t run) { fRunQA=run; }
aaea4d43	53	void SetRunPL(Bool_t run) { fRunPL=run; }
aaea4d43	54	void SetRunDL(Bool_t run) { fRunDL=run; }
ee87a008	55	void SetRunHJet(Bool_t run) { fRunHJet=run; }
ee87a008	56	void SetRunMatch(Bool_t run) { fRunMatch=run; }
aaea4d43	57	void SetTTRanges(Double_t min, Double_t max)
	58	{
	59	for(Int_t i=0; i<kNTT; i++)
	60	{
	61	fMinTTPt[i] = min[i];
	62	fMaxTTPt[i] = max[i];
	63	}
	64	}
ee87a008	65
	66
	67
	68	protected:
	69	void RunQA();
aaea4d43	70	void RunHJet(const Double_t minPt, const Double_t maxPt);
aaea4d43	71	void RunMatch(const TClonesArray *tracks, const Int_t leadingIndex);
3ec095a2	72	void FillHJetCor(const TClonesArray tracks, const Int_t leadingIndex, const TClonesArray jetArray, THnSparse hTT, THnSparse hn, Bool_t isBkg = kFALSE);
ee87a008	73	Int_t FindGeoMatchedJet(const AliEmcalJet* jet, const TClonesArray *jetArray, Double_t &dR);
aaea4d43	74	Int_t FindEnergyMatchedJet(const AliEmcalJet* jet, const TClonesArray *jetArray, Double_t &dR, Double_t &fraction);
ee87a008	75	Bool_t AcceptTrack(const AliVParticle *track);
	76	Bool_t IsGoodJet(const AliEmcalJet* jet);
	77	Double_t GetZ(const Double_t trkPx, const Double_t trkPy, const Double_t trkPz, const Double_t jetPx, const Double_t jetPy, const Double_t jetPz);
	78	Double_t GetDPhi(const Double_t phi1, const Double_t phi2);
	79	Double_t CalculateDPhi(const Double_t phi1, const Double_t phi2);
aaea4d43	80	Double_t GetJetDistance(const AliEmcalJet jet1, const AliEmcalJet jet2);
ee87a008	81
ee87a008	82	enum { kNTrig = 3 };
aaea4d43	83	enum { kNTT = 4 };
ee87a008	84
	85	private:
	86	Int_t fVerbosity; //! Control output
	87	Int_t fAnaType; // 0-local; 1-grid
	88	TString fPeriod; // Run period
	89	TString fCollisionSystem; // Collision system
	90	AliVEvent *fEvent; //! Input event
	91	Int_t fTriggerType; //! Trigger type of the event
	92	Double_t fCentrality; //! V0M for current event
	93	Double_t fMaxVtxZ; // Maximum z of vertices
	94	TString fMCParticleArrName; //
	95	TClonesArray *fMCParticleArray; //!
	96	TString fTrackArrName; // Name of the input track array
	97	TClonesArray *fTrackArray; //! Array of input tracks
	98	Int_t fTriggerTrkIndex; //! Index of the trigger track in the event
	99	Double_t fMinTrkPt; //
	100	Double_t fMaxTrkPt; //
	101	Double_t fMinTrkEta; //
	102	Double_t fMaxTrkEta; //
	103	Double_t fMinTrkPhi; //
	104	Double_t fMaxTrkPhi; //
3ec095a2	105	Int_t fTTtype; //
aaea4d43	106	Double_t fMinTTPt[kNTT]; //
aaea4d43	107	Double_t fMaxTTPt[kNTT]; //
ee87a008	108	Double_t fRadius; // Jet radius
	109	TString fJetArrName; // Name of the found jet array
	110	TString fPLJetArrName; // Name of the embedded PYTHIA jet array on particle level
	111	TString fDLJetArrName; // Name of the embedded PYTHIA jet array on detector level
	112	TClonesArray *fJetArray; //! Array of the found jets
	113	TClonesArray *fPLJetArray; //! Array of the embedded PYTHIA jet array on particle level
	114	TClonesArray *fDLJetArray; //! Array of the embedded PYTHIA jet array on detector level
	115	TString fRhoName; // Name of the rho parameter
	116	AliRhoParameter *fRho; //! Rho parameter
	117	Double_t fRhoValue; //! Value of the rho parameter
aaea4d43	118	AliNamedString *fPtHardBinName; //!Pt hard bin param
3ec095a2	119	Int_t fPtHardBin; //!
3ec095a2	120	TRandom3 *fRandom; //!
ee87a008	121
	122	Bool_t fRunQA; // Flag to run QA
	123	Bool_t fRunHJet; // Flag to run h+jet
	124	Bool_t fRunMatch; // Flag to run matching
aaea4d43	125	Bool_t fRunPL; //
aaea4d43	126	Bool_t fRunDL; //
ee87a008	127
	128	TList *fOutputList; //! Output list
	129	TH1F *fhEventStat; //!
	130	TH1F *fhPtHardBins; //!
	131
	132	TH1F *fhVtxZ[kNTrig]; //!
	133	TH1F *fhCentrality[kNTrig]; //!
	134	TH2F *fhRhoVsCent[kNTrig]; //!
	135
	136	// QA histograms
	137	THnSparse *fhPLJetPtVsCent[kNTrig]; //!
	138	THnSparse *fhDLJetPtVsCent[kNTrig]; //!
	139
	140	// HJet
	141	THnSparse *fhPLTT[kNTrig]; //!
	142	THnSparse *fhDLTT[kNTrig]; //!
	143	THnSparse *fhPLHJet[kNTrig]; //!
	144	THnSparse *fhDLHJet[kNTrig]; //!
	145	THnSparse *fhTTPtQA[kNTrig]; //!
	146	THnSparse *fhTTPt[kNTrig]; //!
	147	THnSparse *fhHJet[kNTrig]; //!
	148
	149
	150	// Matching
	151	THnSparse *fhJetPtGeoMatch[kNTrig]; //!
	152	THnSparse *fhJetPtEnMatch[kNTrig]; //!
	153
	154	THnSparse *fhJetPhiGeoMatch[kNTrig]; //!
	155	THnSparse *fhJetPhiEnMatch[kNTrig]; //!
	156
	157	AliAnalysisTaskHJetEmbed(const AliAnalysisTaskHJetEmbed&); // not implemented
	158	AliAnalysisTaskHJetEmbed &operator=(const AliAnalysisTaskHJetEmbed&); // not implemented
	159
aaea4d43	160	ClassDef(AliAnalysisTaskHJetEmbed, 3);
ee87a008	161	};
	162
	163	#endif
aaea4d43	164
aaea4d43	165