[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / extra / AliAnalysisTaskEventCount.h

#ifndef ALIANALYSISTASKEVENTCOUNT_H
#define ALIANALYSISTASKEVENTCOUNT_H

// analysis task creating basic QA plots for resonance particles
// Author: Ayben Karasu Uysal

class TH1I;
class TH1F;
class TH2F;

class AliESDpid;
class AliESDEvent;
class AliESDtrackCuts;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskEventCount : public AliAnalysisTaskSE {
public:

   enum EType {
      kAll = 0,
      kPassPhysicsSel,
      kGoodPrimaryVertex,
      kHas1TrackAny,
      kHas1TrackQuality
      
      kTypes
   };

   AliAnalysisTaskEventCount(const char *name = "RsnQA");
   virtual ~AliAnalysisTaskEventCount() {}

   virtual void   UserCreateOutputObjects();
   virtual void   UserExec(Option_t *option);
   virtual void   Terminate(Option_t *);
   
   void               SetVzRange(Double_t vz)                      {fVz = vz;}
   void               SetAcceptTPCPrimaryVertex(Bool_t yn = kTRUE) {fAcceptTPC = yn;}
   static const char* Title(EType type);

private:

   Bool_t            fAcceptTPC;     // accept TPC primary vertex?
   Double_t          fVz;            // symmetric range around 0
   TList            *fOutputList;    // output container
   Bool_t            fCheck[kTypes]; // check if accepted according to criterion in enum
   TH1I             *fHEvent;        // histogram with event counts
   AliESDtrackCuts  *fTrackCuts;     // cut checker
   
   AliAnalysisTaskEventCount(const AliAnalysisTaskEventCount&);
   AliAnalysisTaskEventCount& operator=(const AliAnalysisTaskEventCount&);

   ClassDef(AliAnalysisTaskEventCount, 1);
};

inline const char* Title(AliAnalysisTaskEventCount::EType type)
{
//
// Title string depending on type
//
   
   switch (type) {
      case kAll              : return "All";
      case kPassPhysicsSel   : return "Phys Sel OK";
      case kGoodPrimaryVertex: return "Good primary vertex";
      case kHas1TrackAny     : return "1 track (any)";
      case kHas1TrackQuality : return "1 track (quality OK)";
      default                : return "Unknown";
   }
}

#endif
Commit	Line	Data
0f87171a	1	#ifndef ALIANALYSISTASKEVENTCOUNT_H
	2	#define ALIANALYSISTASKEVENTCOUNT_H
	3
	4	// analysis task creating basic QA plots for resonance particles
	5	// Author: Ayben Karasu Uysal
	6
	7	class TH1I;
	8	class TH1F;
	9	class TH2F;
	10
	11	class AliESDpid;
	12	class AliESDEvent;
	13	class AliESDtrackCuts;
	14
	15	#include "AliAnalysisTaskSE.h"
	16
	17	class AliAnalysisTaskEventCount : public AliAnalysisTaskSE {
	18	public:
	19
	20	enum EType {
	21	kAll = 0,
	22	kPassPhysicsSel,
	23	kGoodPrimaryVertex,
	24	kHas1TrackAny,
	25	kHas1TrackQuality
	26
	27	kTypes
	28	};
	29
	30	AliAnalysisTaskEventCount(const char *name = "RsnQA");
	31	virtual ~AliAnalysisTaskEventCount() {}
	32
	33	virtual void UserCreateOutputObjects();
	34	virtual void UserExec(Option_t *option);
	35	virtual void Terminate(Option_t *);
	36
	37	void SetVzRange(Double_t vz) {fVz = vz;}
	38	void SetAcceptTPCPrimaryVertex(Bool_t yn = kTRUE) {fAcceptTPC = yn;}
	39	static const char* Title(EType type);
	40
	41	private:
	42
	43	Bool_t fAcceptTPC; // accept TPC primary vertex?
	44	Double_t fVz; // symmetric range around 0
	45	TList *fOutputList; // output container
	46	Bool_t fCheck[kTypes]; // check if accepted according to criterion in enum
	47	TH1I *fHEvent; // histogram with event counts
	48	AliESDtrackCuts *fTrackCuts; // cut checker
	49
	50	AliAnalysisTaskEventCount(const AliAnalysisTaskEventCount&);
	51	AliAnalysisTaskEventCount& operator=(const AliAnalysisTaskEventCount&);
	52
	53	ClassDef(AliAnalysisTaskEventCount, 1);
	54	};
	55
	56	inline const char* Title(AliAnalysisTaskEventCount::EType type)
	57	{
	58	//
	59	// Title string depending on type
	60	//
	61
	62	switch (type) {
	63	case kAll : return "All";
	64	case kPassPhysicsSel : return "Phys Sel OK";
65	case kGoodPrimaryVertex: return "Good primary vertex";
66	case kHas1TrackAny : return "1 track (any)";
67	case kHas1TrackQuality : return "1 track (quality OK)";
68	default : return "Unknown";
69	}
70	}
71
72	#endif
73
74