[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / LambdaK0 / AliAnalysisTaskExtractV0.h

#ifndef ALIANALYSISTASKEXTRACTV0_H
#define ALIANALYSISTASKEXTRACTV0_H

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

//-----------------------------------------------------------------
//      AliAnalysisTaskExtractV0 class
//      ------------------------------
//
//    Please see cxx file for more details.   
//             
//-----------------------------------------------------------------

// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
//
// --- This version: 23rd March 2012 
//
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+

class TList;
class TH1F;
class TH2F;
class TH3F;
class TVector3;
class THnSparse;

class AliESDpid;
class AliESDtrackCuts;
class AliESDEvent;
class AliPhysicsSelection;
class AliCFContainer;

//#include "TString.h"
//#include "AliESDtrackCuts.h"
#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskExtractV0 : public AliAnalysisTaskSE {
 public:
	AliAnalysisTaskExtractV0();
	AliAnalysisTaskExtractV0(const char *name);
	virtual ~AliAnalysisTaskExtractV0();
	
	virtual void	 UserCreateOutputObjects();
	virtual void	 UserExec(Option_t *option);
	virtual void	 Terminate(Option_t *);
  void CheckChargeV0(AliESDv0 *thisv0);

  void SetIsNuclear           (Bool_t lIsNuclear   = kTRUE ) { fkIsNuclear   = lIsNuclear;   }
  void SetINT7Trigger         (Bool_t lSwitchINT7  = kTRUE ) { fkSwitchINT7  = lSwitchINT7; }
  void SetUseOnTheFly         (Bool_t lUseOnTheFly = kTRUE ) { fkUseOnTheFly = lUseOnTheFly; }
  void SetTakeAllTracks       (Bool_t lTakeAllTracks = kTRUE ) { fkTakeAllTracks = lTakeAllTracks; }

 private:
				// Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
				// your data member object is created on the worker nodes and streaming is not needed.
				// http://root.cern.ch/download/doc/11InputOutput.pdf, page 14
	TList	*fListHistV0;	//! List of output objects
	TTree	*fTree;							//! Output Tree

  AliPIDResponse *fPIDResponse;     // PID response object
  AliESDtrackCuts *fESDtrackCuts;   // ESD track cuts used for primary track definition

  //Objects Controlling Task Behaviour 
  
  Bool_t fkIsNuclear;   // if true, replace multiplicity est. by centrality (default FALSE) 
  Bool_t fkSwitchINT7; // if true, skip FASTOnly (default FALSE)
  Bool_t fkUseOnTheFly; // if true, will use On-the-fly V0s instead of Offline V0s (default FALSE)
  Bool_t fkTakeAllTracks; // if true, no TPC crossed rows and ratio cut

  //Variables for Tree
	Float_t fTreeVariableChi2V0;         //!
	Float_t fTreeVariableDcaV0Daughters; //!
	Float_t fTreeVariableDcaV0ToPrimVertex; //!
	Float_t fTreeVariableDcaPosToPrimVertex; //!
	Float_t fTreeVariableDcaNegToPrimVertex; //!
	Float_t fTreeVariableV0CosineOfPointingAngle; //!
	Float_t fTreeVariableV0Radius; //!
	Float_t fTreeVariablePt; //!
	Float_t fTreeVariableRapK0Short; //!
	Float_t fTreeVariableRapLambda; //!
	Float_t fTreeVariableInvMassK0s; //!
	Float_t fTreeVariableInvMassLambda; //!
	Float_t fTreeVariableInvMassAntiLambda; //!
	Float_t fTreeVariableAlphaV0; //!
	Float_t fTreeVariablePtArmV0;//!
	Float_t fTreeVariableNegTotMomentum; //!	
	Float_t fTreeVariablePosTotMomentum; //!
	Float_t fTreeVariableNegdEdxSig; //!
	Float_t fTreeVariablePosdEdxSig; //!
	Float_t fTreeVariableNegEta; //!
	Float_t fTreeVariablePosEta; //!

	Float_t fTreeVariableNSigmasPosProton; //!
	Float_t fTreeVariableNSigmasPosPion; //! 
	Float_t fTreeVariableNSigmasNegProton; //!
	Float_t fTreeVariableNSigmasNegPion; //! 
	
	Float_t fTreeVariableDistOverTotMom;//!
	Int_t   fTreeVariableLeastNbrCrossedRows;//!
	Float_t fTreeVariableLeastRatioCrossedRowsOverFindable;//!
	Int_t   fTreeVariableMultiplicity ;//!

  Int_t   fTreeVariableRunNumber; //! 
  ULong64_t fTreeVariableEventNumber; //!

  Float_t fTreeVariableV0x; //!
  Float_t fTreeVariableV0y; //!
  Float_t fTreeVariableV0z; //!

  Float_t fTreeVariableV0Px; //!
  Float_t fTreeVariableV0Py; //!
  Float_t fTreeVariableV0Pz; //!

  Float_t fTreeVariablePVx; //!
  Float_t fTreeVariablePVy; //!
  Float_t fTreeVariablePVz; //!

//Note: TDistOverTotMom needs a mass hypothesis to be converted to proper decaylength.

	TH1F    *fHistV0MultiplicityBeforeTrigSel;             //! V0 multiplicity distribution
	TH1F    *fHistV0MultiplicityForTrigEvt;                //! V0 multiplicity distribution
	TH1F    *fHistV0MultiplicityForSelEvt;                 //! V0 multiplicity distribution
	TH1F    *fHistV0MultiplicityForSelEvtNoTPCOnly;        //! V0 multiplicity distribution
	TH1F    *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup;//! V0 multiplicity distribution

	TH1F    *fHistMultiplicityBeforeTrigSel; 	        //! multiplicity distribution    
	TH1F    *fHistMultiplicityForTrigEvt;  		        //! multiplicity distribution
	TH1F    *fHistMultiplicity;     					        //! multiplicity distribution
	TH1F    *fHistMultiplicityNoTPCOnly;			        //! multiplicity distribution
	TH1F    *fHistMultiplicityNoTPCOnlyNoPileup;			//! multiplicity distribution

  //Raw Data for Vertex Z position estimator change
	TH2F    *f2dHistMultiplicityVsVertexZBeforeTrigSel; 	        //! multiplicity distribution    
	TH2F    *f2dHistMultiplicityVsVertexZForTrigEvt;  		        //! multiplicity distribution
	TH2F    *f2dHistMultiplicityVsVertexZ;     					        //! multiplicity distribution
	TH2F    *f2dHistMultiplicityVsVertexZNoTPCOnly;			        //! multiplicity distribution
	TH2F    *f2dHistMultiplicityVsVertexZNoTPCOnlyNoPileup;			//! multiplicity distribution


	TH1F    *fHistPVx;     					        //! multiplicity distribution
	TH1F    *fHistPVy;     					        //! multiplicity distribution
	TH1F    *fHistPVz;     					        //! multiplicity distribution
	TH1F    *fHistPVxAnalysis;     					        //! multiplicity distribution
	TH1F    *fHistPVyAnalysis;     					        //! multiplicity distribution
	TH1F    *fHistPVzAnalysis;     					        //! multiplicity distribution
	TH1F    *fHistSwappedV0Counter;     					        //! Swapped V0 Counter

   AliAnalysisTaskExtractV0(const AliAnalysisTaskExtractV0&);            // not implemented
   AliAnalysisTaskExtractV0& operator=(const AliAnalysisTaskExtractV0&); // not implemented
  
   ClassDef(AliAnalysisTaskExtractV0, 11);
};

#endif
Commit	Line	Data
bcb6ffdb	1	#ifndef ALIANALYSISTASKEXTRACTV0_H
	2	#define ALIANALYSISTASKEXTRACTV0_H
	3
	4	/**************************************************************************
	5	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	6	* *
	7	* Author: The ALICE Off-line Project. *
	8	* Contributors are mentioned in the code where appropriate. *
	9	* *
	10	* Permission to use, copy, modify and distribute this software and its *
	11	* documentation strictly for non-commercial purposes is hereby granted *
	12	* without fee, provided that the above copyright notice appears in all *
	13	* copies and that both the copyright notice and this permission notice *
	14	* appear in the supporting documentation. The authors make no claims *
	15	* about the suitability of this software for any purpose. It is *
	16	* provided "as is" without express or implied warranty. *
	17	**************************************************************************/
	18
	19	//-----------------------------------------------------------------
	20	// AliAnalysisTaskExtractV0 class
	21	// ------------------------------
	22	//
	23	// Please see cxx file for more details.
	24	//
	25	//-----------------------------------------------------------------
	26
	27	// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
	28	//
	29	// --- This version: 23rd March 2012
	30	//
	31	// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
	32
	33	class TList;
	34	class TH1F;
	35	class TH2F;
	36	class TH3F;
	37	class TVector3;
	38	class THnSparse;
	39
	40	class AliESDpid;
	41	class AliESDtrackCuts;
	42	class AliESDEvent;
	43	class AliPhysicsSelection;
	44	class AliCFContainer;
	45
	46	//#include "TString.h"
	47	//#include "AliESDtrackCuts.h"
	48	#include "AliAnalysisTaskSE.h"
	49
	50	class AliAnalysisTaskExtractV0 : public AliAnalysisTaskSE {
	51	public:
	52	AliAnalysisTaskExtractV0();
	53	AliAnalysisTaskExtractV0(const char *name);
	54	virtual ~AliAnalysisTaskExtractV0();
	55
	56	virtual void UserCreateOutputObjects();
	57	virtual void UserExec(Option_t *option);
	58	virtual void Terminate(Option_t *);
4f8ed367	59	void CheckChargeV0(AliESDv0 *thisv0);
bcb6ffdb	60
548bd092	61	void SetIsNuclear (Bool_t lIsNuclear = kTRUE ) { fkIsNuclear = lIsNuclear; }
048f4f8f	62	void SetINT7Trigger (Bool_t lSwitchINT7 = kTRUE ) { fkSwitchINT7 = lSwitchINT7; }
548bd092	63	void SetUseOnTheFly (Bool_t lUseOnTheFly = kTRUE ) { fkUseOnTheFly = lUseOnTheFly; }
85b81f83	64	void SetTakeAllTracks (Bool_t lTakeAllTracks = kTRUE ) { fkTakeAllTracks = lTakeAllTracks; }
548bd092	65
bcb6ffdb	66	private:
	67	// Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
	68	// your data member object is created on the worker nodes and streaming is not needed.
	69	// http://root.cern.ch/download/doc/11InputOutput.pdf, page 14
	70	TList *fListHistV0; //! List of output objects
	71	TTree *fTree; //! Output Tree
	72
76029adc	73	AliPIDResponse *fPIDResponse; // PID response object
76029adc	74	AliESDtrackCuts *fESDtrackCuts; // ESD track cuts used for primary track definition
bcb6ffdb	75
548bd092	76	//Objects Controlling Task Behaviour
	77
	78	Bool_t fkIsNuclear; // if true, replace multiplicity est. by centrality (default FALSE)
048f4f8f	79	Bool_t fkSwitchINT7; // if true, skip FASTOnly (default FALSE)
548bd092	80	Bool_t fkUseOnTheFly; // if true, will use On-the-fly V0s instead of Offline V0s (default FALSE)
85b81f83	81	Bool_t fkTakeAllTracks; // if true, no TPC crossed rows and ratio cut
bcb6ffdb	82
548bd092	83	//Variables for Tree
bcb6ffdb	84	Float_t fTreeVariableChi2V0; //!
	85	Float_t fTreeVariableDcaV0Daughters; //!
	86	Float_t fTreeVariableDcaV0ToPrimVertex; //!
	87	Float_t fTreeVariableDcaPosToPrimVertex; //!
	88	Float_t fTreeVariableDcaNegToPrimVertex; //!
	89	Float_t fTreeVariableV0CosineOfPointingAngle; //!
	90	Float_t fTreeVariableV0Radius; //!
	91	Float_t fTreeVariablePt; //!
	92	Float_t fTreeVariableRapK0Short; //!
	93	Float_t fTreeVariableRapLambda; //!
	94	Float_t fTreeVariableInvMassK0s; //!
	95	Float_t fTreeVariableInvMassLambda; //!
	96	Float_t fTreeVariableInvMassAntiLambda; //!
	97	Float_t fTreeVariableAlphaV0; //!
	98	Float_t fTreeVariablePtArmV0;//!
	99	Float_t fTreeVariableNegTotMomentum; //!
	100	Float_t fTreeVariablePosTotMomentum; //!
	101	Float_t fTreeVariableNegdEdxSig; //!
	102	Float_t fTreeVariablePosdEdxSig; //!
	103	Float_t fTreeVariableNegEta; //!
	104	Float_t fTreeVariablePosEta; //!
	105
	106	Float_t fTreeVariableNSigmasPosProton; //!
	107	Float_t fTreeVariableNSigmasPosPion; //!
	108	Float_t fTreeVariableNSigmasNegProton; //!
	109	Float_t fTreeVariableNSigmasNegPion; //!
	110
	111	Float_t fTreeVariableDistOverTotMom;//!
	112	Int_t fTreeVariableLeastNbrCrossedRows;//!
	113	Float_t fTreeVariableLeastRatioCrossedRowsOverFindable;//!
	114	Int_t fTreeVariableMultiplicity ;//!
	115
0eebce93	116	Int_t fTreeVariableRunNumber; //!
	117	ULong64_t fTreeVariableEventNumber; //!
	118
57d075b9	119	Float_t fTreeVariableV0x; //!
	120	Float_t fTreeVariableV0y; //!
	121	Float_t fTreeVariableV0z; //!
	122
	123	Float_t fTreeVariableV0Px; //!
	124	Float_t fTreeVariableV0Py; //!
	125	Float_t fTreeVariableV0Pz; //!
bcb6ffdb	126
653265df	127	Float_t fTreeVariablePVx; //!
	128	Float_t fTreeVariablePVy; //!
	129	Float_t fTreeVariablePVz; //!
	130
bcb6ffdb	131	//Note: TDistOverTotMom needs a mass hypothesis to be converted to proper decaylength.
	132
	133	TH1F *fHistV0MultiplicityBeforeTrigSel; //! V0 multiplicity distribution
	134	TH1F *fHistV0MultiplicityForTrigEvt; //! V0 multiplicity distribution
	135	TH1F *fHistV0MultiplicityForSelEvt; //! V0 multiplicity distribution
	136	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnly; //! V0 multiplicity distribution
	137	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup;//! V0 multiplicity distribution
	138
	139	TH1F *fHistMultiplicityBeforeTrigSel; //! multiplicity distribution
	140	TH1F *fHistMultiplicityForTrigEvt; //! multiplicity distribution
	141	TH1F *fHistMultiplicity; //! multiplicity distribution
	142	TH1F *fHistMultiplicityNoTPCOnly; //! multiplicity distribution
	143	TH1F *fHistMultiplicityNoTPCOnlyNoPileup; //! multiplicity distribution
	144
1d94e33b	145	//Raw Data for Vertex Z position estimator change
	146	TH2F *f2dHistMultiplicityVsVertexZBeforeTrigSel; //! multiplicity distribution
	147	TH2F *f2dHistMultiplicityVsVertexZForTrigEvt; //! multiplicity distribution
	148	TH2F *f2dHistMultiplicityVsVertexZ; //! multiplicity distribution
	149	TH2F *f2dHistMultiplicityVsVertexZNoTPCOnly; //! multiplicity distribution
	150	TH2F *f2dHistMultiplicityVsVertexZNoTPCOnlyNoPileup; //! multiplicity distribution
	151
	152
bcb6ffdb	153	TH1F *fHistPVx; //! multiplicity distribution
	154	TH1F *fHistPVy; //! multiplicity distribution
	155	TH1F *fHistPVz; //! multiplicity distribution
	156	TH1F *fHistPVxAnalysis; //! multiplicity distribution
	157	TH1F *fHistPVyAnalysis; //! multiplicity distribution
	158	TH1F *fHistPVzAnalysis; //! multiplicity distribution
4f8ed367	159	TH1F *fHistSwappedV0Counter; //! Swapped V0 Counter
bcb6ffdb	160
	161	AliAnalysisTaskExtractV0(const AliAnalysisTaskExtractV0&); // not implemented
	162	AliAnalysisTaskExtractV0& operator=(const AliAnalysisTaskExtractV0&); // not implemented
	163
	164	ClassDef(AliAnalysisTaskExtractV0, 11);
	165	};
	166
	167	#endif