[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliAnalysisTaskCounter.h

#ifndef ALIANALYSISTASKCOUNTER_H
#define ALIANALYSISTASKCOUNTER_H

//_________________________________________________________________________
//
// Count events with different selections
//
// Author: Gustavo Conesa Balbastre (LPSC)
//
//_________________________________________________________________________

class TH1F;
class TList;
class AliESDtrackCuts;
class AliTriggerAnalysis;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskCounter : public AliAnalysisTaskSE { 
  
 public:  
  AliAnalysisTaskCounter();  
  AliAnalysisTaskCounter(const char *name);  
  virtual ~AliAnalysisTaskCounter() ;
  
  virtual void   UserCreateOutputObjects();  
  virtual void   UserExec(Option_t *option);
  virtual void   FinishTaskOutput();
  
  static  Bool_t PythiaInfoFromFile(TString currFile, Float_t & xsec, Float_t & trials) ;
  virtual Bool_t Notify();

  void    SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut   = eta    ; }  
  void    SetZVertexCut(Float_t vcut)             { fZVertexCut        = vcut   ; }
  
  void    AcceptFastCluster()                     { fAcceptFastCluster = kTRUE  ; } 
  void    RejectFastCluster()                     { fAcceptFastCluster = kFALSE ; }  
  Bool_t  IsFastClusterAccepted()       const     { return fAcceptFastCluster   ; }   
  
  Bool_t  CheckForPrimaryVertex() ;

 private: 
  Bool_t               fAcceptFastCluster; // Accept events from fast cluster, exclude thiese events for LHC11a
  Float_t              fZVertexCut;        // Z vertex cut  
  Float_t              fTrackMultEtaCut;   // Track multiplicity eta cut  
  Float_t              fAvgTrials;         // avg trials
  TList*               fOutputContainer;   //! Histogram container  
  AliESDtrackCuts    * fESDtrackCuts;      // Track cut    
  AliTriggerAnalysis * fTriggerAnalysis;   // Trigger algorithm 
  TString              fCurrFileName;      // current file path name
  
  //Histograms
  TH1I *  fhNEvents;      //! Events that delivers the analysis frame after different assumptions  
  TH1F *  fhXVertex;      //! X Vertex distribution
  TH1F *  fhYVertex;      //! Y Vertex distribution
  TH1F *  fhZVertex;      //! Z Vertex distribution
  TH1F *  fhXGoodVertex;  //! X Vertex good distribution
  TH1F *  fhYGoodVertex;  //! Y Vertex good distribution
  TH1F *  fhZGoodVertex;  //! Z Vertex good distribution  
  TH1F *  fhCentrality;   //! centrality
  TH1F *  fhEventPlaneAngle; //! Histogram with Event plane angle

  TH1F *  fh1Xsec ;                       //! Xsec pythia
  TH1F *  fh1Trials ;                     //! trials pythia
  
  AliAnalysisTaskCounter(           const AliAnalysisTaskCounter&); // not implemented  
  AliAnalysisTaskCounter& operator=(const AliAnalysisTaskCounter&); // not implemented
  
  ClassDef(AliAnalysisTaskCounter, 5);

};

#endif //ALIANALYSISTASKCOUNTER_H
Commit	Line	Data
dd094cf8	1	#ifndef ALIANALYSISTASKCOUNTER_H
	2	#define ALIANALYSISTASKCOUNTER_H
	3
	4	//_________________________________________________________________________
	5	//
	6	// Count events with different selections
	7	//
	8	// Author: Gustavo Conesa Balbastre (LPSC)
	9	//
	10	//_________________________________________________________________________
	11
	12	class TH1F;
	13	class TList;
	14	class AliESDtrackCuts;
	15	class AliTriggerAnalysis;
	16
	17	#include "AliAnalysisTaskSE.h"
	18
	19	class AliAnalysisTaskCounter : public AliAnalysisTaskSE {
	20
	21	public:
	22	AliAnalysisTaskCounter();
	23	AliAnalysisTaskCounter(const char *name);
	24	virtual ~AliAnalysisTaskCounter() ;
	25
46f9f991	26	virtual void UserCreateOutputObjects();
	27	virtual void UserExec(Option_t *option);
	28	virtual void FinishTaskOutput();
dd094cf8	29
46f9f991	30	static Bool_t PythiaInfoFromFile(TString currFile, Float_t & xsec, Float_t & trials) ;
	31	virtual Bool_t Notify();
	32
c5693f62	33	void SetTrackMultiplicityEtaCut(Float_t eta) { fTrackMultEtaCut = eta ; }
997b261e	34	void SetZVertexCut(Float_t vcut) { fZVertexCut = vcut ; }
cd2e4ce6	35
c5693f62	36	void AcceptFastCluster() { fAcceptFastCluster = kTRUE ; }
	37	void RejectFastCluster() { fAcceptFastCluster = kFALSE ; }
	38	Bool_t IsFastClusterAccepted() const { return fAcceptFastCluster ; }
dd094cf8	39
c5693f62	40	Bool_t CheckForPrimaryVertex() ;
46f9f991	41
dd094cf8	42	private:
cd2e4ce6	43	Bool_t fAcceptFastCluster; // Accept events from fast cluster, exclude thiese events for LHC11a
	44	Float_t fZVertexCut; // Z vertex cut
	45	Float_t fTrackMultEtaCut; // Track multiplicity eta cut
46f9f991	46	Float_t fAvgTrials; // avg trials
cd2e4ce6	47	TList* fOutputContainer; //! Histogram container
	48	AliESDtrackCuts * fESDtrackCuts; // Track cut
	49	AliTriggerAnalysis * fTriggerAnalysis; // Trigger algorithm
46f9f991	50	TString fCurrFileName; // current file path name
dd094cf8	51
	52	//Histograms
	53	TH1I * fhNEvents; //! Events that delivers the analysis frame after different assumptions
	54	TH1F * fhXVertex; //! X Vertex distribution
	55	TH1F * fhYVertex; //! Y Vertex distribution
	56	TH1F * fhZVertex; //! Z Vertex distribution
	57	TH1F * fhXGoodVertex; //! X Vertex good distribution
	58	TH1F * fhYGoodVertex; //! Y Vertex good distribution
	59	TH1F * fhZGoodVertex; //! Z Vertex good distribution
e19ca838	60	TH1F * fhCentrality; //! centrality
0792da48	61	TH1F * fhEventPlaneAngle; //! Histogram with Event plane angle
0792da48	62
46f9f991	63	TH1F * fh1Xsec ; //! Xsec pythia
	64	TH1F * fh1Trials ; //! trials pythia
	65
d2655d46	66	AliAnalysisTaskCounter( const AliAnalysisTaskCounter&); // not implemented
c5693f62	67	AliAnalysisTaskCounter& operator=(const AliAnalysisTaskCounter&); // not implemented
c5693f62	68
997b261e	69	ClassDef(AliAnalysisTaskCounter, 5);
dd094cf8	70
	71	};
	72
	73	#endif //ALIANALYSISTASKCOUNTER_H