[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliAnalysisTaskDeltaPt.h

#ifndef ALIANALYSISTASKDELTAPT_H
#define ALIANALYSISTASKDELTAPT_H

// $Id$

class TClonesArray;
class TString;
class TH1;
class TH2;
class TH3;
class AliJetContainer;
class AliParticleContainer;
class AliClusterContainer;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskDeltaPt : public AliAnalysisTaskEmcalJet {
 public:

  AliAnalysisTaskDeltaPt();
  AliAnalysisTaskDeltaPt(const char *name);
  virtual ~AliAnalysisTaskDeltaPt() {;}

  void                        UserCreateOutputObjects();

  void                        SetJetMinRC2LJ(Float_t d)                            { fMinRC2LJ                = d          ; }
  void                        SetRCperEvent(Int_t n)                               { fRCperEvent              = n          ; }
  void                        SetMCJetPtThreshold(Double_t t)                      { fMCJetPtThreshold        = t          ; }
  void                        SetConeRadius(Double_t r)                            { fConeRadius              = r          ; }
  void                        SetConeEtaPhiEMCAL() ;
  void                        SetConeEtaPhiTPC()   ;
  void                        SetConeEtaLimits(Float_t min, Float_t max)           { fConeMinEta = min, fConeMaxEta = max  ; }
  void                        SetConePhiLimits(Float_t min, Float_t max)           { fConeMinPhi = min, fConeMaxPhi = max  ; }

 protected:
  void                        AllocateHistogramArrays()                                                                     ;
  void                        ExecOnce()                                                                                    ;
  Bool_t                      FillHistograms()                                                                              ;
  void                        GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex)                                       ;
  AliEmcalJet*                NextEmbeddedJet(Bool_t reset=kFALSE)                                                          ;
  void                        DoEmbTrackLoop()                                                                              ;
  void                        DoEmbClusterLoop()                                                                            ;
  void                        GetRandomCone(Float_t &pt, Float_t &eta, Float_t &phi, AliParticleContainer* tracks, AliClusterContainer* clusters,
					    AliEmcalJet *jet = 0, Bool_t bPartialExclusion = 0) const;
  Double_t                    GetNColl() const;


  Double_t                    fMCJetPtThreshold;           // threshold for MC jets
  Float_t                     fMinRC2LJ;                   // Minimum distance random cone to leading jet
  Int_t                       fRCperEvent;                 // No. of random cones per event
  Double_t                    fConeRadius;                 // Radius of the random cones
  Float_t                     fConeMinEta;                 // Minimum eta of the random cones
  Float_t                     fConeMaxEta;                 // Maximum eta of the random cones
  Float_t                     fConeMinPhi;                 // Minimum phi of the random cones
  Float_t                     fConeMaxPhi;                 // Maximum phi of the random cones

  AliJetContainer            *fJetsCont;                   //!Jets
  AliParticleContainer       *fTracksCont;                 //!Tracks
  AliClusterContainer        *fCaloClustersCont;           //!Clusters  
  AliJetContainer            *fEmbJetsCont;                //!Embedded jets
  AliParticleContainer       *fEmbTracksCont;              //!Embedded tracks
  AliClusterContainer        *fEmbCaloClustersCont;        //!Embedded clusters  
  AliParticleContainer       *fRandTracksCont;             //!Randomized tracks
  AliClusterContainer        *fRandCaloClustersCont;       //!Randomized clusters

  // Random cones
  TH2                        *fHistRCPhiEta;               //!Phi-Eta distribution of random cones
  TH1                       **fHistRCPt;                   //!Random cone pt
  TH1                       **fHistRCPtExLJ;               //!Random cone pt, imposing min distance from leading jet
  TH1                       **fHistRCPtExPartialLJ;        //!Random cone pt, imposing min distance from leading jet with 1/ncoll probability
  TH1                       **fHistRCPtRand;               //!Random cone pt, randomized particles
  TH2                       **fHistRhoVSRCPt;              //!Area(RC) * rho vs. Pt(RC)
  TH2                       **fHistDeltaPtRCvsEP;          //!deltaPt = Pt(RC) - A * rho vs. event plane
  TH1                       **fHistDeltaPtRCExLJ;          //!deltaPt = Pt(RC) - A * rho, imposing min distance from leading jet
  TH1                       **fHistDeltaPtRCExPartialLJ;   //!deltaPt = Pt(RC) - A * rho, imposing min distance from leading jet with 1/ncoll probability
  TH1                       **fHistDeltaPtRCRand;          //!deltaPt = Pt(RC) - A * rho, randomzied particles

  // Jet embedding
  TH3                       **fHistEmbJetsPtArea;          //!Pt vs. area of embedded jets
  TH3                       **fHistEmbJetsCorrPtArea;      //!Pt-rho*A vs. area of embedded jets
  TH2                       **fHistEmbPartPtvsJetPt;       //!MC jet pt total jet pt
  TH2                       **fHistEmbPartPtvsJetCorrPt;   //!MC jet pt total jet pt - rho*A
  TH2                       **fHistJetPtvsJetCorrPt;       //!Pt vs jet pt - rho*A
  TH1                       **fHistDistLeadPart2JetAxis;   //!Distance between leading particle and jet axis
  TH2                       **fHistEmbBkgArea;             //!Pt(embjet) - Pt(embtrack) vs. area of embedded jets
  TH2                       **fHistRhoVSEmbBkg;            //!Area(embjet) * rho vs. Pt(embjet) - Pt(embtrack)
  TH2                       **fHistDeltaPtEmbArea;         //!deltaPt = Pt(embjet) - Area(embjet) * rho - Pt(embtrack) vs. Area(embjet)
  TH2                       **fHistDeltaPtEmbvsEP;         //!deltaPt = Pt(embjet) - Area(embjet) * rho - Pt(embtrack) vs. event plane
  TH2                        *fHistRCPtExLJVSDPhiLJ;       //!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet
  TH2                        *fHistRCPtExPartialLJVSDPhiLJ;//!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet with 1/ncoll probability
  TH2                        *fHistEmbJetsPhiEta;          //!Phi-Eta distribution of embedded jets
  TH2                        *fHistLeadPartPhiEta;         //!Phi-Eta distribution of the leading particle of embedded jets

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

  ClassDef(AliAnalysisTaskDeltaPt, 5) // deltaPt analysis task
};
#endif
Commit	Line	Data
a487deae	1	#ifndef ALIANALYSISTASKDELTAPT_H
	2	#define ALIANALYSISTASKDELTAPT_H
	3
	4	// $Id$
	5
	6	class TClonesArray;
	7	class TString;
6a20534a	8	class TH1;
6a20534a	9	class TH2;
ca5c29fa	10	class TH3;
6421eeb0	11	class AliJetContainer;
	12	class AliParticleContainer;
	13	class AliClusterContainer;
a487deae	14
9239b066	15	#include "AliAnalysisTaskEmcalJet.h"
a487deae	16
9239b066	17	class AliAnalysisTaskDeltaPt : public AliAnalysisTaskEmcalJet {
a487deae	18	public:
	19
	20	AliAnalysisTaskDeltaPt();
	21	AliAnalysisTaskDeltaPt(const char *name);
6421eeb0	22	virtual ~AliAnalysisTaskDeltaPt() {;}
a487deae	23
a487deae	24	void UserCreateOutputObjects();
a487deae	25
a487deae	26	void SetJetMinRC2LJ(Float_t d) { fMinRC2LJ = d ; }
a487deae	27	void SetRCperEvent(Int_t n) { fRCperEvent = n ; }
f660c2d6	28	void SetMCJetPtThreshold(Double_t t) { fMCJetPtThreshold = t ; }
6421eeb0	29	void SetConeRadius(Double_t r) { fConeRadius = r ; }
	30	void SetConeEtaPhiEMCAL() ;
	31	void SetConeEtaPhiTPC() ;
	32	void SetConeEtaLimits(Float_t min, Float_t max) { fConeMinEta = min, fConeMaxEta = max ; }
	33	void SetConePhiLimits(Float_t min, Float_t max) { fConeMinPhi = min, fConeMaxPhi = max ; }
a487deae	34
a487deae	35	protected:
ed062c95	36	void AllocateHistogramArrays() ;
a487deae	37	void ExecOnce() ;
	38	Bool_t FillHistograms() ;
	39	void GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex) ;
6421eeb0	40	AliEmcalJet* NextEmbeddedJet(Bool_t reset=kFALSE) ;
a487deae	41	void DoEmbTrackLoop() ;
a487deae	42	void DoEmbClusterLoop() ;
6421eeb0	43	void GetRandomCone(Float_t &pt, Float_t &eta, Float_t &phi, AliParticleContainer* tracks, AliClusterContainer* clusters,
6421eeb0	44	AliEmcalJet *jet = 0, Bool_t bPartialExclusion = 0) const;
56f370b0	45	Double_t GetNColl() const;
56f370b0	46
a487deae	47
f660c2d6	48	Double_t fMCJetPtThreshold; // threshold for MC jets
a487deae	49	Float_t fMinRC2LJ; // Minimum distance random cone to leading jet
a487deae	50	Int_t fRCperEvent; // No. of random cones per event
6421eeb0	51	Double_t fConeRadius; // Radius of the random cones
	52	Float_t fConeMinEta; // Minimum eta of the random cones
	53	Float_t fConeMaxEta; // Maximum eta of the random cones
	54	Float_t fConeMinPhi; // Minimum phi of the random cones
	55	Float_t fConeMaxPhi; // Maximum phi of the random cones
a487deae	56
6421eeb0	57	AliJetContainer *fJetsCont; //!Jets
	58	AliParticleContainer *fTracksCont; //!Tracks
	59	AliClusterContainer *fCaloClustersCont; //!Clusters
	60	AliJetContainer *fEmbJetsCont; //!Embedded jets
	61	AliParticleContainer *fEmbTracksCont; //!Embedded tracks
	62	AliClusterContainer *fEmbCaloClustersCont; //!Embedded clusters
	63	AliParticleContainer *fRandTracksCont; //!Randomized tracks
	64	AliClusterContainer *fRandCaloClustersCont; //!Randomized clusters
a487deae	65
a487deae	66	// Random cones
6a20534a	67	TH2 *fHistRCPhiEta; //!Phi-Eta distribution of random cones
56f370b0	68	TH1 **fHistRCPt; //!Random cone pt
	69	TH1 **fHistRCPtExLJ; //!Random cone pt, imposing min distance from leading jet
	70	TH1 **fHistRCPtExPartialLJ; //!Random cone pt, imposing min distance from leading jet with 1/ncoll probability
	71	TH1 **fHistRCPtRand; //!Random cone pt, randomized particles
	72	TH2 *fHistRhoVSRCPt; //!Area(RC) rho vs. Pt(RC)
faf21244	73	TH2 *fHistDeltaPtRCvsEP; //!deltaPt = Pt(RC) - A rho vs. event plane
56f370b0	74	TH1 *fHistDeltaPtRCExLJ; //!deltaPt = Pt(RC) - A rho, imposing min distance from leading jet
	75	TH1 *fHistDeltaPtRCExPartialLJ; //!deltaPt = Pt(RC) - A rho, imposing min distance from leading jet with 1/ncoll probability
	76	TH1 *fHistDeltaPtRCRand; //!deltaPt = Pt(RC) - A rho, randomzied particles
a487deae	77
a487deae	78	// Jet embedding
56f370b0	79	TH3 **fHistEmbJetsPtArea; //!Pt vs. area of embedded jets
	80	TH3 *fHistEmbJetsCorrPtArea; //!Pt-rhoA vs. area of embedded jets
	81	TH2 **fHistEmbPartPtvsJetPt; //!MC jet pt total jet pt
	82	TH2 *fHistEmbPartPtvsJetCorrPt; //!MC jet pt total jet pt - rhoA
	83	TH2 *fHistJetPtvsJetCorrPt; //!Pt vs jet pt - rhoA
	84	TH1 **fHistDistLeadPart2JetAxis; //!Distance between leading particle and jet axis
	85	TH2 **fHistEmbBkgArea; //!Pt(embjet) - Pt(embtrack) vs. area of embedded jets
	86	TH2 *fHistRhoVSEmbBkg; //!Area(embjet) rho vs. Pt(embjet) - Pt(embtrack)
faf21244	87	TH2 *fHistDeltaPtEmbArea; //!deltaPt = Pt(embjet) - Area(embjet) rho - Pt(embtrack) vs. Area(embjet)
faf21244	88	TH2 *fHistDeltaPtEmbvsEP; //!deltaPt = Pt(embjet) - Area(embjet) rho - Pt(embtrack) vs. event plane
56f370b0	89	TH2 *fHistRCPtExLJVSDPhiLJ; //!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet
56f370b0	90	TH2 *fHistRCPtExPartialLJVSDPhiLJ;//!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet with 1/ncoll probability
6421eeb0	91	TH2 *fHistEmbJetsPhiEta; //!Phi-Eta distribution of embedded jets
6a20534a	92	TH2 *fHistLeadPartPhiEta; //!Phi-Eta distribution of the leading particle of embedded jets
a487deae	93
	94	private:
	95	AliAnalysisTaskDeltaPt(const AliAnalysisTaskDeltaPt&); // not implemented
	96	AliAnalysisTaskDeltaPt &operator=(const AliAnalysisTaskDeltaPt&); // not implemented
	97
6421eeb0	98	ClassDef(AliAnalysisTaskDeltaPt, 5) // deltaPt analysis task
a487deae	99	};
a487deae	100	#endif