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

#ifndef ALIANALYSISTASKSAJF_H
#define ALIANALYSISTASKSAJF_H

// $Id$

class TClonesArray;
class TString;
class TH1F;
class TH2F;
class AliRhoParameter;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskSAJF : public AliAnalysisTaskEmcalJet {
 public:

  AliAnalysisTaskSAJF();
  AliAnalysisTaskSAJF(const char *name);
  virtual ~AliAnalysisTaskSAJF();

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

  void                        SetJetMinRC2LJ(Float_t d)                            { fMinRC2LJ                = d          ; } 
  void                        SetEmbJetsName(const char *n)                        { fEmbJetsName             = n          ; }
  void                        SetRandTracksName(const char *n)                     { fRandTracksName          = n          ; }
  void                        SetRandClusName(const char *n)                       { fRandCaloName            = n          ; }
  void                        SetEmbTracksName(const char *n)                      { fEmbTracksName          = n          ; }
  void                        SetEmbClusName(const char *n)                        { fEmbCaloName            = n          ; }
  void                        SetRhoName(const char *n)                            { fRhoName                 = n          ; } 

 protected:

  Bool_t                      RetrieveEventObjects()                                                                        ;
  Bool_t                      FillHistograms()                                                                              ;
  void                        GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex)                                       ;
  void                        DoJetLoop()                                                                                   ;
  void                        DoEmbJetLoop(AliEmcalJet* &embJet, TObject* &embPart)                                         ;
  void                        DoTrackLoop()                                                                                 ;
  void                        DoClusterLoop()                                                                               ;
  void                        GetRigidCone(Float_t &pt, Float_t &eta, Float_t &phi, Bool_t acceptMC = kFALSE, 
					   AliEmcalJet *jet = 0, TClonesArray* tracks = 0, TClonesArray* clusters = 0) const;

  Float_t                     fMinRC2LJ;                   // Minimum distance random cone to leading jet
  TString                     fEmbJetsName;                // Name of embedded jet collection
  TString                     fEmbTracksName;              // Name of embedded track collection
  TString                     fEmbCaloName;                // Name of embedded calo cluster collection
  TString                     fRandTracksName;             // Name of randomized track collection
  TString                     fRandCaloName;               // Name of randomized calo cluster collection
  TString                     fRhoName;                    // Name of rho object

  TClonesArray               *fEmbJets;                    //!Embedded Jets
  TClonesArray               *fEmbTracks;                  //!Embedded tracks
  TClonesArray               *fEmbCaloClusters;            //!Embedded clusters  
  TClonesArray               *fRandTracks;                 //!Randomized tracks
  TClonesArray               *fRandCaloClusters;           //!Randomized clusters
  AliRhoParameter            *fRho;                        //!Event rho

  // General histograms
  TH1F                       *fHistCentrality;             //!Event centrality distribution
  TH1F                       *fHistEvents[4];              //!Events accepted/rejected
  TH1F                       *fHistTracksPt[4];            //!Inclusive track pt spectrum
  TH1F                       *fHistClustersPt[4];          //!Inclusive clusters pt spectrum
  TH2F                       *fHistJetPhiEta[4];           //!Phi-Eta distribution of jets
  TH1F                       *fHistJetsPt[4];              //!Inclusive jet pt spectrum
  TH2F                       *fHistJetsPtArea[4];          //!Jet pt vs. area
  TH1F                       *fHistLeadingJetPt[4];        //!Leading jet pt spectrum
  TH1F                       *fHist2LeadingJetPt[4];       //!Second leading jet pt spectrum
  TH2F                       *fHistJetsNEFvsPt[4];         //!Jet neutral energy fraction vs. jet pt
  TH2F                       *fHistJetsZvsPt[4];           //!Constituent Pt over Jet Pt ratio vs. jet pt
  TH2F                       *fHistMaxTrackPtvsJetPt[4];   //!Max constituent track pt vs. jet pt
  TH2F                       *fHistMaxClusPtvsJetPt[4];    //!Max constituent cluster pt vs. jet pt
  TH2F                       *fHistMaxPartPtvsJetPt[4];    //!Max constituent particle (track or cluster) pt vs. jet pt
  TH2F                       *fHistMaxTrackPtvsJetCorrPt[4];   //!Max constituent track pt vs. jet pt
  TH2F                       *fHistMaxClusPtvsJetCorrPt[4];    //!Max constituent cluster pt vs. jet pt
  TH2F                       *fHistMaxPartPtvsJetCorrPt[4];    //!Max constituent particle (track or cluster) pt vs. jet pt

  // Rho
  TH1F                       *fHistRho[4];                    //!Rho distribution
  TH2F                       *fHistRhoVSleadJetPt;            //!Area(leadjetarea) * rho vs. leading jet pt
  TH1F                       *fHistCorrJetsPt[4];             //!Corrected inclusive jet pt spectrum
  TH2F                       *fHistCorrJetsPtArea[4];         //!Jet pt vs. area
  TH1F                       *fHistCorrLeadingJetPt[4];       //!Corrected leading jet pt spectrum

  // Random cones
  TH2F                       *fHistRCPhiEta;               //!Phi-Eta distribution of random cones
  TH1F                       *fHistRCPt[4];                //!Random cone pt
  TH1F                       *fHistRCPtExLJ[4];            //!Random cone pt, imposing min distance from leading jet
  TH1F                       *fHistRCPtRand[4];            //!Random cone pt, randomized particles
  TH2F                       *fHistRCPtExLJVSDPhiLJ;       //!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet
  TH2F                       *fHistRhoVSRCPt;              //!Rho vs. Pt(RCExLJ) / Area(RCExLJ)
  TH1F                       *fHistDeltaPtRC[4];           //!deltaPt = Pt(RC) - A * rho
  TH1F                       *fHistDeltaPtRCExLJ[4];       //!deltaPt = Pt(RC) - A * rho, imposing min distance from leading jet
  TH1F                       *fHistDeltaPtRCRand[4];       //!deltaPt = Pt(RC) - A * rho, randomzied particles

  // Jet embedding
  TH1F                       *fHistEmbJetsPt[4];           //!Pt distribution of embedded jets
  TH1F                       *fHistEmbJetsCorrPt[4];       //!Pt distribution of embedded jets
  TH1F                       *fHistEmbPart[4];             //!Pt distribution of embedded particle
  TH2F                       *fHistEmbJetPhiEta;           //!Phi-Eta distribution of embedded jets
  TH2F                       *fHistEmbPartPhiEta;          //!Phi-Eta distribution of embedded particles
  TH2F                       *fHistRhoVSEmbBkg;            //!Area(embjet) * rho vs. Pt(embjet) - Pt(embtrack)
  TH1F                       *fHistDeltaPtEmb[4];          //!deltaPt = Pt(embjet) - Area(embjet) * rho - Pt(embtrack)

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

  ClassDef(AliAnalysisTaskSAJF, 7) // jet analysis task
};
#endif
Commit	Line	Data
00514d01	1	#ifndef ALIANALYSISTASKSAJF_H
00514d01	2	#define ALIANALYSISTASKSAJF_H
25283b37	3
00514d01	4	// $Id$
25283b37	5
	6	class TClonesArray;
	7	class TString;
25283b37	8	class TH1F;
25283b37	9	class TH2F;
e44e8726	10	class AliRhoParameter;
25283b37	11
6fd5039f	12	#include "AliAnalysisTaskEmcalJet.h"
25283b37	13
6fd5039f	14	class AliAnalysisTaskSAJF : public AliAnalysisTaskEmcalJet {
25283b37	15	public:
91f4b7c5	16
00514d01	17	AliAnalysisTaskSAJF();
	18	AliAnalysisTaskSAJF(const char *name);
	19	virtual ~AliAnalysisTaskSAJF();
25283b37	20
25283b37	21	void UserCreateOutputObjects();
25283b37	22	void Terminate(Option_t *option);
a55e4f1d	23	void Init();
25283b37	24
6fd5039f	25	void SetJetMinRC2LJ(Float_t d) { fMinRC2LJ = d ; }
	26	void SetEmbJetsName(const char *n) { fEmbJetsName = n ; }
	27	void SetRandTracksName(const char *n) { fRandTracksName = n ; }
	28	void SetRandClusName(const char *n) { fRandCaloName = n ; }
e44e8726	29	void SetEmbTracksName(const char *n) { fEmbTracksName = n ; }
	30	void SetEmbClusName(const char *n) { fEmbCaloName = n ; }
	31	void SetRhoName(const char *n) { fRhoName = n ; }
25283b37	32
	33	protected:
	34
6fd5039f	35	Bool_t RetrieveEventObjects() ;
	36	Bool_t FillHistograms() ;
	37	void GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex) ;
	38	void DoJetLoop() ;
e44e8726	39	void DoEmbJetLoop(AliEmcalJet* &embJet, TObject* &embPart) ;
	40	void DoTrackLoop() ;
	41	void DoClusterLoop() ;
b12a85c3	42	void GetRigidCone(Float_t &pt, Float_t &eta, Float_t &phi, Bool_t acceptMC = kFALSE,
b12a85c3	43	AliEmcalJet jet = 0, TClonesArray tracks = 0, TClonesArray* clusters = 0) const;
c554a987	44
b12a85c3	45	Float_t fMinRC2LJ; // Minimum distance random cone to leading jet
e44e8726	46	TString fEmbJetsName; // Name of embedded jet collection
	47	TString fEmbTracksName; // Name of embedded track collection
	48	TString fEmbCaloName; // Name of embedded calo cluster collection
b12a85c3	49	TString fRandTracksName; // Name of randomized track collection
b12a85c3	50	TString fRandCaloName; // Name of randomized calo cluster collection
226f511d	51	TString fRhoName; // Name of rho object
df43b607	52
2bee31e9	53	TClonesArray *fEmbJets; //!Embedded Jets
e44e8726	54	TClonesArray *fEmbTracks; //!Embedded tracks
e44e8726	55	TClonesArray *fEmbCaloClusters; //!Embedded clusters
b12a85c3	56	TClonesArray *fRandTracks; //!Randomized tracks
b12a85c3	57	TClonesArray *fRandCaloClusters; //!Randomized clusters
e44e8726	58	AliRhoParameter *fRho; //!Event rho
b12a85c3	59
b12a85c3	60	// General histograms
226f511d	61	TH1F *fHistCentrality; //!Event centrality distribution
e44e8726	62	TH1F *fHistEvents[4]; //!Events accepted/rejected
	63	TH1F *fHistTracksPt[4]; //!Inclusive track pt spectrum
	64	TH1F *fHistClustersPt[4]; //!Inclusive clusters pt spectrum
df43b607	65	TH2F *fHistJetPhiEta[4]; //!Phi-Eta distribution of jets
226f511d	66	TH1F *fHistJetsPt[4]; //!Inclusive jet pt spectrum
df43b607	67	TH2F *fHistJetsPtArea[4]; //!Jet pt vs. area
226f511d	68	TH1F *fHistLeadingJetPt[4]; //!Leading jet pt spectrum
226f511d	69	TH1F *fHist2LeadingJetPt[4]; //!Second leading jet pt spectrum
b12a85c3	70	TH2F *fHistJetsNEFvsPt[4]; //!Jet neutral energy fraction vs. jet pt
b12a85c3	71	TH2F *fHistJetsZvsPt[4]; //!Constituent Pt over Jet Pt ratio vs. jet pt
e44e8726	72	TH2F *fHistMaxTrackPtvsJetPt[4]; //!Max constituent track pt vs. jet pt
	73	TH2F *fHistMaxClusPtvsJetPt[4]; //!Max constituent cluster pt vs. jet pt
	74	TH2F *fHistMaxPartPtvsJetPt[4]; //!Max constituent particle (track or cluster) pt vs. jet pt
	75	TH2F *fHistMaxTrackPtvsJetCorrPt[4]; //!Max constituent track pt vs. jet pt
	76	TH2F *fHistMaxClusPtvsJetCorrPt[4]; //!Max constituent cluster pt vs. jet pt
	77	TH2F *fHistMaxPartPtvsJetCorrPt[4]; //!Max constituent particle (track or cluster) pt vs. jet pt
b12a85c3	78
b12a85c3	79	// Rho
2bddb6ae	80	TH1F *fHistRho[4]; //!Rho distribution
	81	TH2F fHistRhoVSleadJetPt; //!Area(leadjetarea) rho vs. leading jet pt
	82	TH1F *fHistCorrJetsPt[4]; //!Corrected inclusive jet pt spectrum
	83	TH2F *fHistCorrJetsPtArea[4]; //!Jet pt vs. area
2bddb6ae	84	TH1F *fHistCorrLeadingJetPt[4]; //!Corrected leading jet pt spectrum
b12a85c3	85
	86	// Random cones
	87	TH2F *fHistRCPhiEta; //!Phi-Eta distribution of random cones
226f511d	88	TH1F *fHistRCPt[4]; //!Random cone pt
226f511d	89	TH1F *fHistRCPtExLJ[4]; //!Random cone pt, imposing min distance from leading jet
b12a85c3	90	TH1F *fHistRCPtRand[4]; //!Random cone pt, randomized particles
	91	TH2F *fHistRCPtExLJVSDPhiLJ; //!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet
	92	TH2F *fHistRhoVSRCPt; //!Rho vs. Pt(RCExLJ) / Area(RCExLJ)
	93	TH1F fHistDeltaPtRC[4]; //!deltaPt = Pt(RC) - A rho
	94	TH1F fHistDeltaPtRCExLJ[4]; //!deltaPt = Pt(RC) - A rho, imposing min distance from leading jet
	95	TH1F fHistDeltaPtRCRand[4]; //!deltaPt = Pt(RC) - A rho, randomzied particles
	96
	97	// Jet embedding
e44e8726	98	TH1F *fHistEmbJetsPt[4]; //!Pt distribution of embedded jets
e44e8726	99	TH1F *fHistEmbJetsCorrPt[4]; //!Pt distribution of embedded jets
226f511d	100	TH1F *fHistEmbPart[4]; //!Pt distribution of embedded particle
b12a85c3	101	TH2F *fHistEmbJetPhiEta; //!Phi-Eta distribution of embedded jets
	102	TH2F *fHistEmbPartPhiEta; //!Phi-Eta distribution of embedded particles
	103	TH2F fHistRhoVSEmbBkg; //!Area(embjet) rho vs. Pt(embjet) - Pt(embtrack)
	104	TH1F fHistDeltaPtEmb[4]; //!deltaPt = Pt(embjet) - Area(embjet) rho - Pt(embtrack)
25283b37	105
25283b37	106	private:
00514d01	107	AliAnalysisTaskSAJF(const AliAnalysisTaskSAJF&); // not implemented
00514d01	108	AliAnalysisTaskSAJF &operator=(const AliAnalysisTaskSAJF&); // not implemented
25283b37	109
98750b70	110	ClassDef(AliAnalysisTaskSAJF, 7) // jet analysis task
25283b37	111	};
25283b37	112	#endif