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

#ifndef ALIANALYSISTASKSAJF_H
#define ALIANALYSISTASKSAJF_H

// $Id$

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

#include "AliAnalysisTaskEmcal.h"

class AliAnalysisTaskSAJF : public AliAnalysisTaskEmcal {
 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                        SetRhoName(const char *n)                            { fRhoName        = n          ; }

 protected:

  void                        RetrieveEventObjects()                                                                        ;
  void                        FillHistograms()                                                                              ;
  void                        DoJetLoop(Int_t &maxJetIndex, Int_t &max2JetIndex)                                            ;
  void                        DoEmbJetLoop(AliEmcalJet* &embJet, TObject* &maxPart)                                         ;
  void                        DoTrackLoop(Int_t maxJetIndex)                                                                ;
  void                        DoClusterLoop(Int_t maxJetIndex)                                                              ;
  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 jets 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               *fRandTracks;                 //!Randomized tracks
  TClonesArray               *fRandCaloClusters;           //!Randomized clusters
  Float_t                     fRho;                        //!Event rho

  // General histograms
  TH1F                       *fHistCentrality;             //!Event centrality distribution
  TH2F                       *fHistJetPhiEta[4];           //!Phi-Eta distribution of jets
  TH1F                       *fHistJetsPt[4];              //!Inclusive jet pt spectrum
  TH2F                       *fHistJetsPtArea[4];          //!Jet pt vs. area
  TH1F                       *fHistJetsPtClus[4];          //!Inclusive jet pt spectrum cluster biased
  TH1F                       *fHistJetsPtTrack[4];         //!Inclusive jet pt spectrum track biased
  TH1F                       *fHistJetsPtNonBias[4];       //!Non biased inclusive jet pt spectrum
  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
  TH1F                       *fHistTracksPtLJ[4];          //!Pt spectrum of tracks belonging to the leading jet
  TH1F                       *fHistClusEtLJ[4];            //!Et spectrum of clusters belonging to the leading jet
  TH1F                       *fHistTracksPtBkg[4];         //!Pt spectrum of tracks not belonging to the leading jet
  TH1F                       *fHistClusEtBkg[4];           //!Et spectrum of clusters not belonging to the leading jet

  // Rho
  TH1F                       *fHistRho[4];                 //!Rho distribution
  TH2F                       *fHistRhoVSleadJetPt;         //!Area(leadjetarea) * rho vs. leading jet pt
  TH1F                       *fHistCorrJetsPt[4];          //!Corrected inclusive jet pt spectrum
  TH1F                       *fHistCorrJetsPtClus[4];      //!Corrected inclusive jet pt spectrum cluster biased
  TH1F                       *fHistCorrJetsPtTrack[4];     //!Corrected inclusive jet pt spectrum track biased
  TH1F                       *fHistCorrJetsPtNonBias[4];   //!Non biased corrected inclusive jet pt spectrum
  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                       *fHistEmbJets[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, 6) // 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;
25283b37	10
df43b607	11	#include "AliAnalysisTaskEmcal.h"
25283b37	12
df43b607	13	class AliAnalysisTaskSAJF : public AliAnalysisTaskEmcal {
25283b37	14	public:
91f4b7c5	15
00514d01	16	AliAnalysisTaskSAJF();
	17	AliAnalysisTaskSAJF(const char *name);
	18	virtual ~AliAnalysisTaskSAJF();
25283b37	19
25283b37	20	void UserCreateOutputObjects();
25283b37	21	void Terminate(Option_t *option);
a55e4f1d	22	void Init();
25283b37	23
df43b607	24	void SetJetMinRC2LJ(Float_t d) { fMinRC2LJ = d ; }
11d4d636	25	void SetEmbJetsName(const char *n) { fEmbJetsName = n ; }
b12a85c3	26	void SetRandTracksName(const char *n) { fRandTracksName = n ; }
b12a85c3	27	void SetRandClusName(const char *n) { fRandCaloName = n ; }
11d4d636	28	void SetRhoName(const char *n) { fRhoName = n ; }
25283b37	29
	30	protected:
	31
b12a85c3	32	void RetrieveEventObjects() ;
	33	void FillHistograms() ;
	34	void DoJetLoop(Int_t &maxJetIndex, Int_t &max2JetIndex) ;
	35	void DoEmbJetLoop(AliEmcalJet* &embJet, TObject* &maxPart) ;
	36	void DoTrackLoop(Int_t maxJetIndex) ;
	37	void DoClusterLoop(Int_t maxJetIndex) ;
	38	void GetRigidCone(Float_t &pt, Float_t &eta, Float_t &phi, Bool_t acceptMC = kFALSE,
	39	AliEmcalJet jet = 0, TClonesArray tracks = 0, TClonesArray* clusters = 0) const;
c554a987	40
b12a85c3	41	Float_t fMinRC2LJ; // Minimum distance random cone to leading jet
2bee31e9	42	TString fEmbJetsName; // Name of embedded jets collection
b12a85c3	43	TString fRandTracksName; // Name of randomized track collection
b12a85c3	44	TString fRandCaloName; // Name of randomized calo cluster collection
226f511d	45	TString fRhoName; // Name of rho object
df43b607	46
2bee31e9	47	TClonesArray *fEmbJets; //!Embedded Jets
b12a85c3	48	TClonesArray *fRandTracks; //!Randomized tracks
b12a85c3	49	TClonesArray *fRandCaloClusters; //!Randomized clusters
226f511d	50	Float_t fRho; //!Event rho
b12a85c3	51
b12a85c3	52	// General histograms
226f511d	53	TH1F *fHistCentrality; //!Event centrality distribution
df43b607	54	TH2F *fHistJetPhiEta[4]; //!Phi-Eta distribution of jets
226f511d	55	TH1F *fHistJetsPt[4]; //!Inclusive jet pt spectrum
df43b607	56	TH2F *fHistJetsPtArea[4]; //!Jet pt vs. area
b12a85c3	57	TH1F *fHistJetsPtClus[4]; //!Inclusive jet pt spectrum cluster biased
	58	TH1F *fHistJetsPtTrack[4]; //!Inclusive jet pt spectrum track biased
	59	TH1F *fHistJetsPtNonBias[4]; //!Non biased inclusive jet pt spectrum
226f511d	60	TH1F *fHistLeadingJetPt[4]; //!Leading jet pt spectrum
226f511d	61	TH1F *fHist2LeadingJetPt[4]; //!Second leading jet pt spectrum
b12a85c3	62	TH2F *fHistJetsNEFvsPt[4]; //!Jet neutral energy fraction vs. jet pt
	63	TH2F *fHistJetsZvsPt[4]; //!Constituent Pt over Jet Pt ratio vs. jet pt
	64	TH1F *fHistTracksPtLJ[4]; //!Pt spectrum of tracks belonging to the leading jet
	65	TH1F *fHistClusEtLJ[4]; //!Et spectrum of clusters belonging to the leading jet
	66	TH1F *fHistTracksPtBkg[4]; //!Pt spectrum of tracks not belonging to the leading jet
	67	TH1F *fHistClusEtBkg[4]; //!Et spectrum of clusters not belonging to the leading jet
	68
	69	// Rho
	70	TH1F *fHistRho[4]; //!Rho distribution
	71	TH2F fHistRhoVSleadJetPt; //!Area(leadjetarea) rho vs. leading jet pt
	72	TH1F *fHistCorrJetsPt[4]; //!Corrected inclusive jet pt spectrum
	73	TH1F *fHistCorrJetsPtClus[4]; //!Corrected inclusive jet pt spectrum cluster biased
	74	TH1F *fHistCorrJetsPtTrack[4]; //!Corrected inclusive jet pt spectrum track biased
	75	TH1F *fHistCorrJetsPtNonBias[4]; //!Non biased corrected inclusive jet pt spectrum
	76	TH1F *fHistCorrLeadingJetPt[4]; //!Corrected leading jet pt spectrum
	77
	78	// Random cones
	79	TH2F *fHistRCPhiEta; //!Phi-Eta distribution of random cones
226f511d	80	TH1F *fHistRCPt[4]; //!Random cone pt
226f511d	81	TH1F *fHistRCPtExLJ[4]; //!Random cone pt, imposing min distance from leading jet
b12a85c3	82	TH1F *fHistRCPtRand[4]; //!Random cone pt, randomized particles
	83	TH2F *fHistRCPtExLJVSDPhiLJ; //!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet
	84	TH2F *fHistRhoVSRCPt; //!Rho vs. Pt(RCExLJ) / Area(RCExLJ)
	85	TH1F fHistDeltaPtRC[4]; //!deltaPt = Pt(RC) - A rho
	86	TH1F fHistDeltaPtRCExLJ[4]; //!deltaPt = Pt(RC) - A rho, imposing min distance from leading jet
	87	TH1F fHistDeltaPtRCRand[4]; //!deltaPt = Pt(RC) - A rho, randomzied particles
	88
	89	// Jet embedding
226f511d	90	TH1F *fHistEmbJets[4]; //!Pt distribution of embedded jets
226f511d	91	TH1F *fHistEmbPart[4]; //!Pt distribution of embedded particle
b12a85c3	92	TH2F *fHistEmbJetPhiEta; //!Phi-Eta distribution of embedded jets
	93	TH2F *fHistEmbPartPhiEta; //!Phi-Eta distribution of embedded particles
	94	TH2F fHistRhoVSEmbBkg; //!Area(embjet) rho vs. Pt(embjet) - Pt(embtrack)
	95	TH1F fHistDeltaPtEmb[4]; //!deltaPt = Pt(embjet) - Area(embjet) rho - Pt(embtrack)
25283b37	96
25283b37	97	private:
00514d01	98	AliAnalysisTaskSAJF(const AliAnalysisTaskSAJF&); // not implemented
00514d01	99	AliAnalysisTaskSAJF &operator=(const AliAnalysisTaskSAJF&); // not implemented
25283b37	100
6e8d91c9	101	ClassDef(AliAnalysisTaskSAJF, 6) // jet analysis task
25283b37	102	};
25283b37	103	#endif