[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;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskDeltaPt : public AliAnalysisTaskEmcalJet {
 public:

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

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

  void                        SetJetMinRC2LJ(Float_t d)                            { fMinRC2LJ                = d          ; }
  void                        SetEmbJetsName(const char *n)                        { fEmbJetsName             = n          ; } 
  void                        SetEmbTracksName(const char *n)                      { fEmbTracksName           = n          ; }
  void                        SetEmbClusName(const char *n)                        { fEmbCaloName             = n          ; }
  void                        SetRandTracksName(const char *n)                     { fRandTracksName          = n          ; }
  void                        SetRandClusName(const char *n)                       { fRandCaloName            = n          ; }
  void                        SetRCperEvent(Int_t n)                               { fRCperEvent              = n          ; }
  void                        SetMCJetPtThreshold(Double_t t)                      { fMCJetPtThreshold        = t          ; }

 protected:
  void                        ExecOnce()                                                                                    ;
  Bool_t                      FillHistograms()                                                                              ;
  void                        GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex)                                       ;
  AliEmcalJet*                NextEmbeddedJet(Int_t i=-1)                                                                   ;
  void                        DoEmbTrackLoop()                                                                              ;
  void                        DoEmbClusterLoop()                                                                            ;
  void                        GetRandomCone(Float_t &pt, Float_t &eta, Float_t &phi, 
					    AliEmcalJet *jet = 0, TClonesArray* tracks = 0, TClonesArray* clusters = 0) const;

  Double_t                    fMCJetPtThreshold;           // threshold for MC jets
  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
  Int_t                       fRCperEvent;                 // No. of random cones per event

  TClonesArray               *fEmbJets;                    //!Embedded jets
  TClonesArray               *fEmbTracks;                  //!Embedded tracks
  TClonesArray               *fEmbCaloClusters;            //!Embedded clusters  
  TClonesArray               *fRandTracks;                 //!Randomized tracks
  TClonesArray               *fRandCaloClusters;           //!Randomized clusters
  Int_t                       fEmbeddedClusterNIds;        //!Embedded cluster id count
  Int_t                       fEmbeddedClusterIds[999];    //!Embedded cluster ids
  Int_t                       fEmbeddedTrackNIds;          //!Embedded track id count
  Int_t                       fEmbeddedTrackIds[999];      //!Embedded track ids

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

  // Jet embedding
  TH2                        *fHistEmbNotFoundPhiEta[4];   //!Phi-Eta of "not found" embedded particles
  TH1                        *fHistEmbNotFoundPt[4];       //!Pt of "not found" embedded particles
  TH3                        *fHistEmbJetsPtArea[4];       //!Pt vs. area of embedded jets
  TH3                        *fHistEmbJetsCorrPtArea[4];   //!Pt-rho*A vs. area of embedded jets
  TH2                        *fHistEmbPartPtvsJetPt[4];    //!MC jet pt total jet pt
  TH2                        *fHistEmbPartPtvsJetCorrPt[4];//!MC jet pt total jet pt - rho*A
  TH2                        *fHistEmbJetsPhiEta;          //!Phi-Eta distribution of embedded jets<
  TH2                        *fHistLeadPartPhiEta;         //!Phi-Eta distribution of the leading particle of embedded jets
  TH1                        *fHistDistLeadPart2JetAxis[4];//!Distance between leading particle and jet axis
  TH2                        *fHistEmbBkgArea[4];          //!Pt(embjet) - Pt(embtrack) vs. area of embedded jets
  TH2                        *fHistRhoVSEmbBkg[4];         //!Area(embjet) * rho vs. Pt(embjet) - Pt(embtrack)
  TH2                        *fHistDeltaPtEmbArea[4];      //!deltaPt = Pt(embjet) - Area(embjet) * rho - Pt(embtrack)

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

  ClassDef(AliAnalysisTaskDeltaPt, 3) // 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;
a487deae	11
	12	#include "AliAnalysisTaskEmcalJet.h"
	13
	14	class AliAnalysisTaskDeltaPt : public AliAnalysisTaskEmcalJet {
	15	public:
	16
	17	AliAnalysisTaskDeltaPt();
	18	AliAnalysisTaskDeltaPt(const char *name);
	19	virtual ~AliAnalysisTaskDeltaPt();
	20
	21	void UserCreateOutputObjects();
	22	void Terminate(Option_t *option);
	23
	24	void SetJetMinRC2LJ(Float_t d) { fMinRC2LJ = d ; }
	25	void SetEmbJetsName(const char *n) { fEmbJetsName = n ; }
	26	void SetEmbTracksName(const char *n) { fEmbTracksName = n ; }
	27	void SetEmbClusName(const char *n) { fEmbCaloName = n ; }
	28	void SetRandTracksName(const char *n) { fRandTracksName = n ; }
	29	void SetRandClusName(const char *n) { fRandCaloName = n ; }
a487deae	30	void SetRCperEvent(Int_t n) { fRCperEvent = n ; }
f660c2d6	31	void SetMCJetPtThreshold(Double_t t) { fMCJetPtThreshold = t ; }
a487deae	32
	33	protected:
	34	void ExecOnce() ;
	35	Bool_t FillHistograms() ;
	36	void GetLeadingJets(Int_t &maxJetIndex, Int_t &max2JetIndex) ;
f660c2d6	37	AliEmcalJet* NextEmbeddedJet(Int_t i=-1) ;
a487deae	38	void DoEmbTrackLoop() ;
	39	void DoEmbClusterLoop() ;
	40	void GetRandomCone(Float_t &pt, Float_t &eta, Float_t &phi,
	41	AliEmcalJet jet = 0, TClonesArray tracks = 0, TClonesArray* clusters = 0) const;
	42
f660c2d6	43	Double_t fMCJetPtThreshold; // threshold for MC jets
a487deae	44	Float_t fMinRC2LJ; // Minimum distance random cone to leading jet
	45	TString fEmbJetsName; // Name of embedded jet collection
	46	TString fEmbTracksName; // Name of embedded track collection
	47	TString fEmbCaloName; // Name of embedded calo cluster collection
	48	TString fRandTracksName; // Name of randomized track collection
	49	TString fRandCaloName; // Name of randomized calo cluster collection
	50	Int_t fRCperEvent; // No. of random cones per event
	51
	52	TClonesArray *fEmbJets; //!Embedded jets
	53	TClonesArray *fEmbTracks; //!Embedded tracks
	54	TClonesArray *fEmbCaloClusters; //!Embedded clusters
	55	TClonesArray *fRandTracks; //!Randomized tracks
	56	TClonesArray *fRandCaloClusters; //!Randomized clusters
f660c2d6	57	Int_t fEmbeddedClusterNIds; //!Embedded cluster id count
	58	Int_t fEmbeddedClusterIds[999]; //!Embedded cluster ids
	59	Int_t fEmbeddedTrackNIds; //!Embedded track id count
	60	Int_t fEmbeddedTrackIds[999]; //!Embedded track ids
a487deae	61
a487deae	62	// Random cones
6a20534a	63	TH2 *fHistRCPhiEta; //!Phi-Eta distribution of random cones
	64	TH1 *fHistRCPt[4]; //!Random cone pt
	65	TH1 *fHistRCPtExLJ[4]; //!Random cone pt, imposing min distance from leading jet
	66	TH1 *fHistRCPtRand[4]; //!Random cone pt, randomized particles
	67	TH2 *fHistRCPtExLJVSDPhiLJ; //!Random cone pt, imposing min distance from leading jet, vs. deltaPhi leading jet
	68	TH2 fHistRhoVSRCPt[4]; //!Area(RC) rho vs. Pt(RC)
	69	TH1 fHistDeltaPtRC[4]; //!deltaPt = Pt(RC) - A rho
	70	TH1 fHistDeltaPtRCExLJ[4]; //!deltaPt = Pt(RC) - A rho, imposing min distance from leading jet
	71	TH1 fHistDeltaPtRCRand[4]; //!deltaPt = Pt(RC) - A rho, randomzied particles
a487deae	72
a487deae	73	// Jet embedding
6a20534a	74	TH2 *fHistEmbNotFoundPhiEta[4]; //!Phi-Eta of "not found" embedded particles
6a20534a	75	TH1 *fHistEmbNotFoundPt[4]; //!Pt of "not found" embedded particles
ca5c29fa	76	TH3 *fHistEmbJetsPtArea[4]; //!Pt vs. area of embedded jets
	77	TH3 fHistEmbJetsCorrPtArea[4]; //!Pt-rhoA vs. area of embedded jets
	78	TH2 *fHistEmbPartPtvsJetPt[4]; //!MC jet pt total jet pt
	79	TH2 fHistEmbPartPtvsJetCorrPt[4];//!MC jet pt total jet pt - rhoA
	80	TH2 *fHistEmbJetsPhiEta; //!Phi-Eta distribution of embedded jets<
6a20534a	81	TH2 *fHistLeadPartPhiEta; //!Phi-Eta distribution of the leading particle of embedded jets
6a20534a	82	TH1 *fHistDistLeadPart2JetAxis[4];//!Distance between leading particle and jet axis
	83	TH2 *fHistEmbBkgArea[4]; //!Pt(embjet) - Pt(embtrack) vs. area of embedded jets
	84	TH2 fHistRhoVSEmbBkg[4]; //!Area(embjet) rho vs. Pt(embjet) - Pt(embtrack)
	85	TH2 fHistDeltaPtEmbArea[4]; //!deltaPt = Pt(embjet) - Area(embjet) rho - Pt(embtrack)
a487deae	86
	87	private:
	88	AliAnalysisTaskDeltaPt(const AliAnalysisTaskDeltaPt&); // not implemented
	89	AliAnalysisTaskDeltaPt &operator=(const AliAnalysisTaskDeltaPt&); // not implemented
	90
f660c2d6	91	ClassDef(AliAnalysisTaskDeltaPt, 3) // deltaPt analysis task
a487deae	92	};
a487deae	93	#endif