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

#ifndef ALIANALYSISTASKSAQA_H
#define ALIANALYSISTASKSAQA_H

// $Id$

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

#include "AliAnalysisTaskSE.h"

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

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

  void                        SetClusName(const char *n)                    { fCaloName      = 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;
  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                     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                       *fHistTracksCent;         // Number of tracks vs. centrality
  TH2F                       *fHistClusCent;           // Number of clusters vs. centrality
  TH1F                       *fHistTracksPt;           // Pt spectrum of tracks
  TH1F                       *fHistClustersEnergy;     // Energy spectrum of clusters
  TH2F                       *fHistEPcorrelation;      // Energy-momentum correlation
  TH2F                       *fHistTrPhiEta;           // Phi-Eta distribution of tracks
  TH2F                       *fHistClusPhiEta;         // Phi-Eta distribution of clusters
  TH1F                       *fHistMaxTrgCluster;      // Energy distribution of max trigger clusters
  TH1F                       *fHistClusPhiCorr;        // Clusters phi correlations
  TH1F                       *fHistTracksPhiCorr;      // Tracks phi correlations
  TH1F                       *fHistTrackPhi[5];        // Phi distribution of hybrid tracks
  TH1F                       *fHistTrackEta[5];        // Eta distribution of hybrid tracks

  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:
  AliAnalysisTaskSAQA(const AliAnalysisTaskSAQA&);            // not implemented
  AliAnalysisTaskSAQA &operator=(const AliAnalysisTaskSAQA&); // not implemented

  ClassDef(AliAnalysisTaskSAQA, 1) // Quality task for Emcal analysis
};
#endif
Commit	Line	Data
00514d01	1	#ifndef ALIANALYSISTASKSAQA_H
00514d01	2	#define ALIANALYSISTASKSAQA_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;
25283b37	13
	14	#include "AliAnalysisTaskSE.h"
	15
00514d01	16	class AliAnalysisTaskSAQA : public AliAnalysisTaskSE {
25283b37	17	public:
00514d01	18	AliAnalysisTaskSAQA();
	19	AliAnalysisTaskSAQA(const char *name);
	20	virtual ~AliAnalysisTaskSAQA();
25283b37	21
	22	void UserCreateOutputObjects();
	23	void UserExec(Option_t *option);
	24	void Terminate(Option_t *option);
	25
	26	void SetClusName(const char *n) { fCaloName = n ; }
25283b37	27	void SetTracksName(const char *n) { fTracksName = n ; }
e82e282c	28	void SetTrgClusName(const char *n) { fTrgClusName = n ; }
25283b37	29
	30	protected:
	31
	32	AliVTrack *GetTrack(const Int_t i) const;
	33	Int_t GetNumberOfTracks() const;
	34	AliVCluster *GetCaloCluster(const Int_t i) const;
	35	Int_t GetNumberOfCaloClusters() const;
e82e282c	36	AliVCluster *GetTrgCluster(const Int_t i) const;
e82e282c	37	Int_t GetNumberOfTrgClusters() const;
25283b37	38	void FillHistograms() ;
25283b37	39	void RetrieveEventObjects() ;
f0a0fd33	40	Bool_t AcceptTrack(AliVTrack* /track/) ;
25283b37	41
	42	TList *fOutput; // Output list
	43
	44	TString fTracksName; // name of track collection
	45	TString fCaloName; // name of calo cluster collection
e82e282c	46	TString fTrgClusName; // name of trg clus name
25283b37	47	TClonesArray *fTracks; //!Tracks
	48	TClonesArray *fCaloClusters; //!Clusters
	49	TClonesArray *fJets; //!Jets
e82e282c	50	TClonesArray *fTrgClusters; //!Trg Clusters
f0a0fd33	51	AliCentrality *fCent; // Event centrality
f0a0fd33	52	TH1F *fHistCentrality; // Event centrality distribution
00514d01	53	TH2F *fHistTracksCent; // Number of tracks vs. centrality
	54	TH2F *fHistClusCent; // Number of clusters vs. centrality
	55	TH1F *fHistTracksPt; // Pt spectrum of tracks
	56	TH1F *fHistClustersEnergy; // Energy spectrum of clusters
	57	TH2F *fHistEPcorrelation; // Energy-momentum correlation
00514d01	58	TH2F *fHistTrPhiEta; // Phi-Eta distribution of tracks
00514d01	59	TH2F *fHistClusPhiEta; // Phi-Eta distribution of clusters
00514d01	60	TH1F *fHistMaxTrgCluster; // Energy distribution of max trigger clusters
c554a987	61	TH1F *fHistClusPhiCorr; // Clusters phi correlations
c554a987	62	TH1F *fHistTracksPhiCorr; // Tracks phi correlations
00514d01	63	TH1F *fHistTrackPhi[5]; // Phi distribution of hybrid tracks
	64	TH1F *fHistTrackEta[5]; // Eta distribution of hybrid tracks
	65
25283b37	66	Int_t Ptbins; // No. of pt bins
	67	Float_t Ptlow; // Min pt
	68	Float_t Ptup; // Max pt
	69	Int_t Ebins; // No. of e bins
	70	Float_t Elow; // Min e
	71	Float_t Eup; // Max e
	72
	73	private:
00514d01	74	AliAnalysisTaskSAQA(const AliAnalysisTaskSAQA&); // not implemented
00514d01	75	AliAnalysisTaskSAQA &operator=(const AliAnalysisTaskSAQA&); // not implemented
25283b37	76
00514d01	77	ClassDef(AliAnalysisTaskSAQA, 1) // Quality task for Emcal analysis
25283b37	78	};
25283b37	79	#endif