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

#ifndef ALIANALYSISTASKSAJF_H
#define ALIANALYSISTASKSAJF_H

// $Id$

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

#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                        SetRhoName(const char *n)                            { fRhoName                 = n          ; }
  void                        SetSkipEventsWithNoJets(Bool_t s)                    { fSkipEventsWithNoJets    = s          ; } 

 protected:

  Bool_t                      RetrieveEventObjects()                                                                        ;
  Bool_t                      FillHistograms()                                                                              ;
  void                        GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex)                                       ;
  void                        DoJetLoop()                                                                                   ;
  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
  Bool_t                      fSkipEventsWithNoJets;       // Whether or not skip events with no jets

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

  // General histograms
  TH1F                       *fHistCentrality;             //!Event centrality distribution
  TH1F                       *fHistRejectedEvents;         //!Rejected events
  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

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