[u/mrichter/AliRoot.git] / PWG3 / muon / AliAnalysisTaskMuonCollisionMultiplicity.h

#ifndef ALIANALYSISTASKMUONCOLLISIONMULTIPLICITY_H
#define ALIANALYSISTASKMUONCOLLISIONMULTIPLICITY_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/// \ingroup base
/// \class AliAnalysisTaskMultiplicity
/// \compute the number of Muons tracks as a function of the SPD tracklets multiplicity
/// Author Matthieu LENHARDT - SUBATECH, Nantes

#include "AliAnalysisTaskSE.h"

class AliAODEvent;
class AliESDEvent;
class AliAODTrack;
class AliESDMuonTrack;
class AliAODDimuon;
class TList;
class TArrayD;

class AliAnalysisTaskMuonCollisionMultiplicity : public AliAnalysisTaskSE
{
 public:
  AliAnalysisTaskMuonCollisionMultiplicity();
  AliAnalysisTaskMuonCollisionMultiplicity(const AliAnalysisTaskMuonCollisionMultiplicity& rhs);
  AliAnalysisTaskMuonCollisionMultiplicity& operator=(const AliAnalysisTaskMuonCollisionMultiplicity&);
  AliAnalysisTaskMuonCollisionMultiplicity(const Char_t* name);
  virtual ~AliAnalysisTaskMuonCollisionMultiplicity();
  
  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
  virtual void UserExec(Option_t *option);
  virtual void Terminate(Option_t *option);
  virtual void NotifyRun();
  virtual void FinishTaskOutput();
  
  
  Double_t GetZCut()              {return fZCut;};
  Double_t GetEtaCut()            {return fEtaCut;};
  void SetZCut(Double_t zCut)     {fZCut = zCut;};
  void SetEtaCut(Double_t etaCut) {fEtaCut = etaCut;};
  
 private:
  
  Bool_t fIsInit;              //< Is the class initialized?
  
  AliAODEvent *fAOD;                  //!< AOD Event
  AliESDEvent *fESD;                  //!< ESD Event

  Double_t fZCut;                    //< Cut on the |z| of the primary vertex
  Double_t fEtaCut;                  //< Cut on the eta cut of the SPD tracklets

  Int_t fTrackletMultiplicity;       //< SPD tracklets multiplicity in the current event

  TList *fTriggerList;               //< list of all trigger histos
  TList *fSingleMuonList;            //< List of all single muons histos 
  TList *fDimuonList;                //< List of all dimuons histos
  TList *fMonteCarloList;            //< List of all histos containing MC info


  void Init();
  Bool_t CheckEventAOD();
  Bool_t CheckEventESD();
  void ComputeMultiplicity();
  Bool_t IsUsableMuon(AliAODTrack *track);
  Bool_t IsUsableMuon(AliESDMuonTrack *track);
  void FillHistosAOD(Int_t triggerClass);
  void FillHistosESD(Int_t triggerClass);
  void FillHistosMC();

  ClassDef(AliAnalysisTaskMuonCollisionMultiplicity, 1);
};
    
#endif
Commit	Line	Data
0a65e325	1	#ifndef ALIANALYSISTASKMUONCOLLISIONMULTIPLICITY_H
	2	#define ALIANALYSISTASKMUONCOLLISIONMULTIPLICITY_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/// \ingroup base
	7	/// \class AliAnalysisTaskMultiplicity
	8	/// \compute the number of Muons tracks as a function of the SPD tracklets multiplicity
	9	/// Author Matthieu LENHARDT - SUBATECH, Nantes
	10
	11	#include "AliAnalysisTaskSE.h"
	12
	13	class AliAODEvent;
	14	class AliESDEvent;
	15	class AliAODTrack;
	16	class AliESDMuonTrack;
	17	class AliAODDimuon;
	18	class TList;
	19	class TArrayD;
	20
	21	class AliAnalysisTaskMuonCollisionMultiplicity : public AliAnalysisTaskSE
	22	{
	23	public:
	24	AliAnalysisTaskMuonCollisionMultiplicity();
	25	AliAnalysisTaskMuonCollisionMultiplicity(const AliAnalysisTaskMuonCollisionMultiplicity& rhs);
	26	AliAnalysisTaskMuonCollisionMultiplicity& operator=(const AliAnalysisTaskMuonCollisionMultiplicity&);
	27	AliAnalysisTaskMuonCollisionMultiplicity(const Char_t* name);
	28	virtual ~AliAnalysisTaskMuonCollisionMultiplicity();
	29
	30	// Implementation of interface methods
	31	virtual void UserCreateOutputObjects();
	32	virtual void UserExec(Option_t *option);
	33	virtual void Terminate(Option_t *option);
	34	virtual void NotifyRun();
	35	virtual void FinishTaskOutput();
	36
	37
	38	Double_t GetZCut() {return fZCut;};
	39	Double_t GetEtaCut() {return fEtaCut;};
	40	void SetZCut(Double_t zCut) {fZCut = zCut;};
	41	void SetEtaCut(Double_t etaCut) {fEtaCut = etaCut;};
	42
	43	private:
	44
	45	Bool_t fIsInit; //< Is the class initialized?
	46
	47	AliAODEvent *fAOD; //!< AOD Event
	48	AliESDEvent *fESD; //!< ESD Event
	49
	50	Double_t fZCut; //< Cut on the \|z\| of the primary vertex
	51	Double_t fEtaCut; //< Cut on the eta cut of the SPD tracklets
	52
	53	Int_t fTrackletMultiplicity; //< SPD tracklets multiplicity in the current event
	54
	55	TList *fTriggerList; //< list of all trigger histos
	56	TList *fSingleMuonList; //< List of all single muons histos
	57	TList *fDimuonList; //< List of all dimuons histos
	58	TList *fMonteCarloList; //< List of all histos containing MC info
	59
	60
	61	void Init();
	62	Bool_t CheckEventAOD();
	63	Bool_t CheckEventESD();
	64	void ComputeMultiplicity();
65	Bool_t IsUsableMuon(AliAODTrack *track);
66	Bool_t IsUsableMuon(AliESDMuonTrack *track);
67	void FillHistosAOD(Int_t triggerClass);
68	void FillHistosESD(Int_t triggerClass);
69	void FillHistosMC();
70
71	ClassDef(AliAnalysisTaskMuonCollisionMultiplicity, 1);
72	};
73
74	#endif
75