[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:
  
  enum SAJFAnaType {
    kFullAcceptance  = 0,     // Full acceptance
    kEMCAL           = 1,     // EMCal acceptance only
    kEMCALFiducial   = 2      // EMCal fiduacial region only
  };

  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          ; }
  void                        SetAnaType(SAJFAnaType type)                  { fAnaType       = type       ; }

 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*/)     ;
  Bool_t                      AcceptJet(AliEmcalJet* jet)           ;

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