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

#ifndef ALIANALYSISTASKSAJF_H
#define ALIANALYSISTASKSAJF_H

// $Id$

class TClonesArray;
class TString;
class AliVTrack;
class AliVCluster;
class TList;
class TH1F;
class TH2F;
class AliEmcalJet;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskSAJF : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskSAJF();
  AliAnalysisTaskSAJF(const char *name);
  virtual ~AliAnalysisTaskSAJF();

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

  void                        SetClusName(const char *n)                    { fCaloName      = n          ; }
  void                        SetJetsName(const char *n)                    { fJetsName      = n          ; }
  void                        SetTracksName(const char *n)                  { fTracksName    = n          ; }
  void                        SetTrgClusName(const char *n)                 { fTrgClusName   = n          ; }

 protected:

  AliVTrack                  *GetTrack(const Int_t i)          const;
  Int_t                       GetNumberOfTracks()              const;
  AliVCluster                *GetCaloCluster(const Int_t i)    const;
  Int_t                       GetNumberOfCaloClusters()        const;
  AliEmcalJet                *GetJet(const Int_t i)            const;
  Int_t                       GetNumberOfJets()                const;
  AliVCluster                *GetTrgCluster(const Int_t i)     const;
  Int_t                       GetNumberOfTrgClusters()         const;
  void                        FillHistograms()                      ;
  void                        RetrieveEventObjects()                ;
  Bool_t                      AcceptTrack(AliVTrack* /*track*/)     ;

  TList                      *fOutput;                 // Output list

  TString                     fTracksName;             // name of track collection
  TString                     fCaloName;               // name of calo cluster collection
  TString                     fJetsName;               // name of jet collection
  TString                     fTrgClusName;            // name of trg clus name
  TClonesArray               *fTracks;                 //!Tracks
  TClonesArray               *fCaloClusters;           //!Clusters
  TClonesArray               *fJets;                   //!Jets
  TClonesArray               *fTrgClusters;            //!Trg Clusters
  AliCentrality              *fCent;                   // Event centrality
  TH1F                       *fHistCentrality;         // Event centrality distribution
  TH1F                       *fHistJetsE[4];           // Jet energy spectrum
  TH1F                       *fHistJetsNE[4];          // Jet neutral energy spectrum
  TH1F                       *fHistJetsNEF[4];         // Jet neutral energy fraction
  TH1F                       *fHistJetsZ[4];           // Constituent Pt over Jet E ratio
  TH1F                       *fHistLeadingJetE[4];     // Leading jet energy spectrum
  TH1F                       *fHistTracksPtLJ[4];      // Pt spectrum of tracks
  TH1F                       *fHistClusELJ[4];         // Energy spectrum of clusters
  TH1F                       *fHistTracksPtBkg[4];     // Pt spectrum of tracks
  TH1F                       *fHistClusEBkg[4];        // Energy spectrum of clusters
  Int_t                       Ptbins;                  // No. of pt bins
  Float_t                     Ptlow;                   // Min pt
  Float_t                     Ptup;                    // Max pt
  Int_t                       Ebins;                   // No. of e bins
  Float_t                     Elow;                    // Min e
  Float_t                     Eup;                     // Max e

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

  ClassDef(AliAnalysisTaskSAJF, 1) // Isolated photons 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 AliVTrack;
	9	class AliVCluster;
	10	class TList;
	11	class TH1F;
	12	class TH2F;
	13	class AliEmcalJet;
	14
	15	#include "AliAnalysisTaskSE.h"
	16
00514d01	17	class AliAnalysisTaskSAJF : public AliAnalysisTaskSE {
25283b37	18	public:
00514d01	19	AliAnalysisTaskSAJF();
	20	AliAnalysisTaskSAJF(const char *name);
	21	virtual ~AliAnalysisTaskSAJF();
25283b37	22
	23	void UserCreateOutputObjects();
	24	void UserExec(Option_t *option);
	25	void Terminate(Option_t *option);
	26
	27	void SetClusName(const char *n) { fCaloName = n ; }
	28	void SetJetsName(const char *n) { fJetsName = n ; }
	29	void SetTracksName(const char *n) { fTracksName = n ; }
e82e282c	30	void SetTrgClusName(const char *n) { fTrgClusName = n ; }
25283b37	31
	32	protected:
	33
	34	AliVTrack *GetTrack(const Int_t i) const;
	35	Int_t GetNumberOfTracks() const;
	36	AliVCluster *GetCaloCluster(const Int_t i) const;
	37	Int_t GetNumberOfCaloClusters() const;
	38	AliEmcalJet *GetJet(const Int_t i) const;
	39	Int_t GetNumberOfJets() const;
e82e282c	40	AliVCluster *GetTrgCluster(const Int_t i) const;
e82e282c	41	Int_t GetNumberOfTrgClusters() const;
25283b37	42	void FillHistograms() ;
25283b37	43	void RetrieveEventObjects() ;
f0a0fd33	44	Bool_t AcceptTrack(AliVTrack* /track/) ;
25283b37	45
	46	TList *fOutput; // Output list
	47
	48	TString fTracksName; // name of track collection
	49	TString fCaloName; // name of calo cluster collection
	50	TString fJetsName; // name of jet collection
e82e282c	51	TString fTrgClusName; // name of trg clus name
25283b37	52	TClonesArray *fTracks; //!Tracks
	53	TClonesArray *fCaloClusters; //!Clusters
	54	TClonesArray *fJets; //!Jets
e82e282c	55	TClonesArray *fTrgClusters; //!Trg Clusters
f0a0fd33	56	AliCentrality *fCent; // Event centrality
	57	TH1F *fHistCentrality; // Event centrality distribution
	58	TH1F *fHistJetsE[4]; // Jet energy spectrum
	59	TH1F *fHistJetsNE[4]; // Jet neutral energy spectrum
	60	TH1F *fHistJetsNEF[4]; // Jet neutral energy fraction
	61	TH1F *fHistJetsZ[4]; // Constituent Pt over Jet E ratio
	62	TH1F *fHistLeadingJetE[4]; // Leading jet energy spectrum
	63	TH1F *fHistTracksPtLJ[4]; // Pt spectrum of tracks
	64	TH1F *fHistClusELJ[4]; // Energy spectrum of clusters
	65	TH1F *fHistTracksPtBkg[4]; // Pt spectrum of tracks
	66	TH1F *fHistClusEBkg[4]; // Energy spectrum of clusters
25283b37	67	Int_t Ptbins; // No. of pt bins
	68	Float_t Ptlow; // Min pt
	69	Float_t Ptup; // Max pt
	70	Int_t Ebins; // No. of e bins
	71	Float_t Elow; // Min e
	72	Float_t Eup; // Max e
	73
	74	private:
00514d01	75	AliAnalysisTaskSAJF(const AliAnalysisTaskSAJF&); // not implemented
00514d01	76	AliAnalysisTaskSAJF &operator=(const AliAnalysisTaskSAJF&); // not implemented
25283b37	77
00514d01	78	ClassDef(AliAnalysisTaskSAJF, 1) // Isolated photons task
25283b37	79	};
25283b37	80	#endif