[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliAnalysisTaskStrange.h

#ifndef AliAnalysisTaskStrange_cxx
#define AliAnalysisTaskStrange_cxx

/*  See cxx source for full Copyright notice */

//-----------------------------------------------------------------
//                 AliAnalysisTaskStrange class
//       This task is for single strange study from ESD/AOD
//          Origin: H.Ricaud, Helene.Ricaud@IReS.in2p3.fr
//-----------------------------------------------------------------

class TString;
class TList;
class TH1F;
class TH2F;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskStrange : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskStrange();
  AliAnalysisTaskStrange(const char *name,  const char *optCuts);
  virtual ~AliAnalysisTaskStrange() {}
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);

  void   SetCollidingSystems(Int_t collidingSystems = 0) {fCollidingSystems = collidingSystems;}
  void   SetAnalysisType(const char* analysisType) {fAnalysisType = analysisType;}
  
 private:
  TString      fAnalysisType;                   //  ESD or AOD
  Int_t        fCollidingSystems;               //  Colliding systems 0/1 for pp/PbPb  
  TString      fOption;                         //  Option for the selections
  TList       *fListHist;                       //! List of histograms
  TH1F        *fHistPrimaryVertexPosX;          //! Primary vertex position in X
  TH1F        *fHistPrimaryVertexPosY;          //! Primary vertex position in Y
  TH1F        *fHistPrimaryVertexPosZ;          //! Primary vertex position in Z
  TH1F        *fHistTrackMultiplicity;          //! V0 multiplicity distribution
  TH1F        *fHistV0Multiplicity;

  TH2F        *fHistDcaPosToPrimVertex;         //! Dca of V0 positive daughter to primary vertex
  TH2F        *fHistDcaNegToPrimVertex;         //! Dca of V0 negative daughter to primary vertex
  TH2F        *fHistDcaPosToPrimVertexZoom;     //! Zoom
  TH2F        *fHistDcaNegToPrimVertexZoom;     //! Zoom
  TH2F        *fHistRadiusV0;                   //! V0 radial distance distribution
  TH2F        *fHistDecayLengthV0;              //! V0 decay length distribution
  TH2F        *fHistDcaV0Daughters;             //! Dca between V0 daughters
  TH2F        *fHistChi2;                       //! V0 chi2 distribution
  TH2F        *fHistCosPointAngle;              //! Cosine of V0 pointing angle
  TH2F        *fHistCosPointAngleZoom;          //! Zoom

              // V0 offline distributions
  TH1F        *fHistV0MultiplicityOff;          //! V0 multiplicity distribution offline
  TH2F        *fHistPtVsYK0sOff;                //! Pt vs.Y with K0s assumption
  TH2F        *fHistPtVsYLambdaOff;             //! Pt vs.Y with Lambda assumption
  TH2F        *fHistPtVsYAntiLambdaOff;         //! Pt vs.Y with Anti-Lambda assumption
  TH1F        *fHistMassK0sOff;                 //! Invariant mass of K0s
  TH1F        *fHistMassLambdaOff;		//! Invariant mass of Lambda
  TH1F        *fHistMassAntiLambdaOff;		//! Invariant mass of Anti-Lambda
  TH2F        *fHistMassVsRadiusK0sOff;         //! Invariant mass vs. radius of K0s	 
  TH2F        *fHistMassVsRadiusLambdaOff;	//! Invariant mass vs. radius of Lambda	 
  TH2F        *fHistMassVsRadiusAntiLambdaOff;	//! Invariant mass vs. radius of Anti-Lambda
  TH2F        *fHistPtVsMassK0sOff;             //! Pt vs. invariant mass of K0s	    
  TH2F        *fHistPtVsMassLambdaOff;		//! Pt vs. invariant mass of Lambda	    
  TH2F        *fHistPtVsMassAntiLambdaOff;	//! Pt vs. invariant mass of Anti-Lambda
  TH2F        *fHistArmenterosPodolanskiOff;    //! Armenteros-Podolanski distribution       

              // V0 on-the-fly distributions
  TH1F        *fHistV0MultiplicityOn;          //! V0 multiplicity distribution on-the-fly
  TH2F        *fHistPtVsYK0sOn;                //! Pt vs.Y with K0s assumption		    
  TH2F        *fHistPtVsYLambdaOn;	       //! Pt vs.Y with Lambda assumption	    
  TH2F        *fHistPtVsYAntiLambdaOn;	       //! Pt vs.Y with Anti-Lambda assumption	    
  TH1F        *fHistMassK0sOn;                 //! Invariant mass of K0s		    
  TH1F        *fHistMassLambdaOn;	       //! Invariant mass of Lambda		     
  TH1F        *fHistMassAntiLambdaOn;	       //! Invariant mass of Anti-Lambda	    
  TH2F        *fHistMassVsRadiusK0sOn;	       //! Invariant mass vs. radius of K0s	    
  TH2F        *fHistMassVsRadiusLambdaOn;      //! Invariant mass vs. radius of Lambda	    
  TH2F        *fHistMassVsRadiusAntiLambdaOn;  //! Invariant mass vs. radius of Anti-Lambda
  TH2F        *fHistPtVsMassK0sOn;	       //! Pt vs. invariant mass of K0s	    
  TH2F        *fHistPtVsMassLambdaOn;	       //! Pt vs. invariant mass of Lambda	    
  TH2F        *fHistPtVsMassAntiLambdaOn;      //! Pt vs. invariant mass of Anti-Lambda    
  TH2F        *fHistArmenterosPodolanskiOn;    //! Armenteros-Podolanski distribution
   
  AliAnalysisTaskStrange(const AliAnalysisTaskStrange&);            // not implemented 
  AliAnalysisTaskStrange& operator=(const AliAnalysisTaskStrange&); // not implemented 

  ClassDef(AliAnalysisTaskStrange, 1); 
};

#endif
Commit	Line	Data
9ea746fc	1	#ifndef AliAnalysisTaskStrange_cxx
	2	#define AliAnalysisTaskStrange_cxx
	3
	4	/* See cxx source for full Copyright notice */
	5
	6	//-----------------------------------------------------------------
	7	// AliAnalysisTaskStrange class
	8	// This task is for single strange study from ESD/AOD
	9	// Origin: H.Ricaud, Helene.Ricaud@IReS.in2p3.fr
	10	//-----------------------------------------------------------------
	11
	12	class TString;
	13	class TList;
	14	class TH1F;
	15	class TH2F;
	16
	17	#include "AliAnalysisTaskSE.h"
	18
	19	class AliAnalysisTaskStrange : public AliAnalysisTaskSE {
	20	public:
	21	AliAnalysisTaskStrange();
	22	AliAnalysisTaskStrange(const char name, const char optCuts);
	23	virtual ~AliAnalysisTaskStrange() {}
	24
	25	virtual void UserCreateOutputObjects();
	26	virtual void UserExec(Option_t *option);
	27	virtual void Terminate(Option_t *);
	28
	29	void SetCollidingSystems(Int_t collidingSystems = 0) {fCollidingSystems = collidingSystems;}
	30	void SetAnalysisType(const char* analysisType) {fAnalysisType = analysisType;}
	31
	32	private:
	33	TString fAnalysisType; // ESD or AOD
	34	Int_t fCollidingSystems; // Colliding systems 0/1 for pp/PbPb
	35	TString fOption; // Option for the selections
	36	TList *fListHist; //! List of histograms
	37	TH1F *fHistPrimaryVertexPosX; //! Primary vertex position in X
	38	TH1F *fHistPrimaryVertexPosY; //! Primary vertex position in Y
	39	TH1F *fHistPrimaryVertexPosZ; //! Primary vertex position in Z
	40	TH1F *fHistTrackMultiplicity; //! V0 multiplicity distribution
	41	TH1F *fHistV0Multiplicity;
	42
	43	TH2F *fHistDcaPosToPrimVertex; //! Dca of V0 positive daughter to primary vertex
	44	TH2F *fHistDcaNegToPrimVertex; //! Dca of V0 negative daughter to primary vertex
	45	TH2F *fHistDcaPosToPrimVertexZoom; //! Zoom
	46	TH2F *fHistDcaNegToPrimVertexZoom; //! Zoom
	47	TH2F *fHistRadiusV0; //! V0 radial distance distribution
	48	TH2F *fHistDecayLengthV0; //! V0 decay length distribution
	49	TH2F *fHistDcaV0Daughters; //! Dca between V0 daughters
	50	TH2F *fHistChi2; //! V0 chi2 distribution
	51	TH2F *fHistCosPointAngle; //! Cosine of V0 pointing angle
	52	TH2F *fHistCosPointAngleZoom; //! Zoom
	53
	54	// V0 offline distributions
	55	TH1F *fHistV0MultiplicityOff; //! V0 multiplicity distribution offline
	56	TH2F *fHistPtVsYK0sOff; //! Pt vs.Y with K0s assumption
	57	TH2F *fHistPtVsYLambdaOff; //! Pt vs.Y with Lambda assumption
	58	TH2F *fHistPtVsYAntiLambdaOff; //! Pt vs.Y with Anti-Lambda assumption
	59	TH1F *fHistMassK0sOff; //! Invariant mass of K0s
	60	TH1F *fHistMassLambdaOff; //! Invariant mass of Lambda
	61	TH1F *fHistMassAntiLambdaOff; //! Invariant mass of Anti-Lambda
	62	TH2F *fHistMassVsRadiusK0sOff; //! Invariant mass vs. radius of K0s
	63	TH2F *fHistMassVsRadiusLambdaOff; //! Invariant mass vs. radius of Lambda
	64	TH2F *fHistMassVsRadiusAntiLambdaOff; //! Invariant mass vs. radius of Anti-Lambda
65	TH2F *fHistPtVsMassK0sOff; //! Pt vs. invariant mass of K0s
66	TH2F *fHistPtVsMassLambdaOff; //! Pt vs. invariant mass of Lambda
67	TH2F *fHistPtVsMassAntiLambdaOff; //! Pt vs. invariant mass of Anti-Lambda
68	TH2F *fHistArmenterosPodolanskiOff; //! Armenteros-Podolanski distribution
69
70	// V0 on-the-fly distributions
71	TH1F *fHistV0MultiplicityOn; //! V0 multiplicity distribution on-the-fly
72	TH2F *fHistPtVsYK0sOn; //! Pt vs.Y with K0s assumption
73	TH2F *fHistPtVsYLambdaOn; //! Pt vs.Y with Lambda assumption
74	TH2F *fHistPtVsYAntiLambdaOn; //! Pt vs.Y with Anti-Lambda assumption
75	TH1F *fHistMassK0sOn; //! Invariant mass of K0s
76	TH1F *fHistMassLambdaOn; //! Invariant mass of Lambda
77	TH1F *fHistMassAntiLambdaOn; //! Invariant mass of Anti-Lambda
78	TH2F *fHistMassVsRadiusK0sOn; //! Invariant mass vs. radius of K0s
79	TH2F *fHistMassVsRadiusLambdaOn; //! Invariant mass vs. radius of Lambda
80	TH2F *fHistMassVsRadiusAntiLambdaOn; //! Invariant mass vs. radius of Anti-Lambda
81	TH2F *fHistPtVsMassK0sOn; //! Pt vs. invariant mass of K0s
82	TH2F *fHistPtVsMassLambdaOn; //! Pt vs. invariant mass of Lambda
83	TH2F *fHistPtVsMassAntiLambdaOn; //! Pt vs. invariant mass of Anti-Lambda
84	TH2F *fHistArmenterosPodolanskiOn; //! Armenteros-Podolanski distribution
85
86	AliAnalysisTaskStrange(const AliAnalysisTaskStrange&); // not implemented
87	AliAnalysisTaskStrange& operator=(const AliAnalysisTaskStrange&); // not implemented
88
89	ClassDef(AliAnalysisTaskStrange, 1);
90	};
91
92	#endif