[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / LambdaK0 / AliAnalysisTaskExtractPerformanceV0.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.                  *
 **************************************************************************/

// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
//
// Modified version of AliAnalysisTaskCheckCascade.h
// Used bits of code from AliAnalysisTaskCheckPerformanceStrange
//
// --- David Dobrigkeit Chinellato
//
// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+

#ifndef ALIANALYSISTASKEXTRACTPERFORMANCEV0_H
#define ALIANALYSISTASKEXTRACTPERFORMANCEV0_H

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 AliAnalysisTaskExtractPerformanceV0 : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskExtractPerformanceV0();
  AliAnalysisTaskExtractPerformanceV0(const char *name);
  virtual ~AliAnalysisTaskExtractPerformanceV0();
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  Double_t MyRapidity(Double_t rE, Double_t rPz) const;

  void SetIsNuclear           (Bool_t lIsNuclear   = kTRUE ) { fkIsNuclear   = lIsNuclear;   }
  void SetIsLowEnergyPP       (Bool_t lLowEnergyPP = kTRUE ) { fkLowEnergyPP = lLowEnergyPP; }
  void SetUseOnTheFly         (Bool_t lUseOnTheFly = kTRUE ) { fkUseOnTheFly = lUseOnTheFly; }
  
 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 Cascade histograms
  TTree  *fTree;              //! Output Tree

  AliPIDResponse *fPIDResponse;     // PID response object

  //Objects Controlling Task Behaviour 
  
  Bool_t fkIsNuclear;   //if true, replace multiplicity est. by centrality (default FALSE) 
  Bool_t fkLowEnergyPP; //if true, skip FASTOnly (default FALSE)
  Bool_t fkUseOnTheFly; //if true, will use On-the-fly V0s instead of Offline V0s (default FALSE)

  //Variables for Tree
   Int_t    fTreeVariablePrimaryStatus;      //!
   Int_t    fTreeVariablePrimaryStatusMother;      //!
   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 fTreeVariablePtMC; //!
   Float_t fTreeVariableRapK0Short; //!
   Float_t fTreeVariableRapLambda; //!
   Float_t fTreeVariableRapMC; //!
   Float_t fTreeVariableInvMassK0s; //!
   Float_t fTreeVariableInvMassLambda; //!
   Float_t fTreeVariableInvMassAntiLambda; //!
   Float_t fTreeVariableAlphaV0; //!
   Float_t fTreeVariablePtArmV0;//!
   Float_t fTreeVariableNegTotMomentum; //!               
   Float_t fTreeVariablePosTotMomentum; //!
   Float_t fTreeVariableNegTransvMomentum; //!   
   Float_t fTreeVariablePosTransvMomentum; //!
   Float_t fTreeVariableNegTransvMomentumMC; //!   
   Float_t fTreeVariablePosTransvMomentumMC; //!
   
   Float_t fTreeVariableNSigmasPosProton; //!
   Float_t fTreeVariableNSigmasPosPion; //! 
   Float_t fTreeVariableNSigmasNegProton; //!
   Float_t fTreeVariableNSigmasNegPion; //! 

   Float_t fTreeVariablePtMother; //!
   Float_t fTreeVariableV0CreationRadius; //!
   Int_t fTreeVariablePID; //!
   Int_t fTreeVariablePIDPositive; //!
   Int_t fTreeVariablePIDNegative; //!
   Int_t fTreeVariablePIDMother; //!
   Int_t fTreeVariableIndexStatus; //!
   Int_t fTreeVariableIndexStatusMother; //!

   //Note: TDistOverTotMom needs a mass hypothesis to be converted to proper decaylength.
   Float_t fTreeVariableDistOverTotMom;//!

   Float_t fTreeVariablePosEta; //!
   Float_t fTreeVariableNegEta; //!

   Int_t fTreeVariableLeastNbrCrossedRows;//!
   Float_t fTreeVariableLeastRatioCrossedRowsOverFindable;//!
   Int_t fTreeVariableMultiplicity;//!

   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

//---> Filled At Analysis Scope

   TH3F      *f3dHistPrimAnalysisPtVsYVsMultLambda;     //! Lambda
   TH3F      *f3dHistPrimAnalysisPtVsYVsMultAntiLambda; //! AntiLambda
   TH3F      *f3dHistPrimAnalysisPtVsYVsMultK0Short;    //! K0Short

//---> Containers for monte carlo information for calculating efficiency! 

   TH3F      *f3dHistPrimRawPtVsYVsMultLambda;     //! Lambda
   TH3F      *f3dHistPrimRawPtVsYVsMultAntiLambda; //! AntiLambda
   TH3F      *f3dHistPrimRawPtVsYVsMultK0Short;    //! K0Short

//---> Filled vs Decay Length

   TH3F      *f3dHistPrimRawPtVsYVsDecayLengthLambda;     //! Lambda
   TH3F      *f3dHistPrimRawPtVsYVsDecayLengthAntiLambda; //! AntiLambda
   TH3F      *f3dHistPrimRawPtVsYVsDecayLengthK0Short;    //! K0Short

//---> Needed for FeedDown Corrections

   TH3F      *f3dHistGenPtVsYVsMultXiMinus;      //! Generated Xi- Distrib
   TH3F      *f3dHistGenPtVsYVsMultXiPlus;       //! Generated Xi+ Distrib

   TH1F      *fHistPVx;                      //! PVx distrib
   TH1F      *fHistPVy;                      //! PVy distrib
   TH1F      *fHistPVz;                      //! PVz distrib
   TH1F      *fHistPVxAnalysis;                      //! PVx distrib
   TH1F      *fHistPVyAnalysis;                      //! PVy distrib
   TH1F      *fHistPVzAnalysis;                      //! PVz distrib
   TH1F      *fHistPVxAnalysisHasHighPtLambda;                      //! PVx distrib
   TH1F      *fHistPVyAnalysisHasHighPtLambda;                      //! PVy distrib
   TH1F      *fHistPVzAnalysisHasHighPtLambda;                      //! PVz distrib

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

#endif
Commit	Line	Data
bcb6ffdb	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
	17	//
	18	// Modified version of AliAnalysisTaskCheckCascade.h
	19	// Used bits of code from AliAnalysisTaskCheckPerformanceStrange
	20	//
	21	// --- David Dobrigkeit Chinellato
	22	//
	23	// +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
	24
	25	#ifndef ALIANALYSISTASKEXTRACTPERFORMANCEV0_H
	26	#define ALIANALYSISTASKEXTRACTPERFORMANCEV0_H
	27
	28	class TList;
	29	class TH1F;
	30	class TH2F;
	31	class TH3F;
	32	class TVector3;
	33	class THnSparse;
	34
	35	class AliESDpid;
	36	class AliESDtrackCuts;
	37	class AliESDEvent;
	38	class AliPhysicsSelection;
	39	class AliCFContainer;
	40
	41	//#include "TString.h"
	42	//#include "AliESDtrackCuts.h"
	43	#include "AliAnalysisTaskSE.h"
	44
	45	class AliAnalysisTaskExtractPerformanceV0 : public AliAnalysisTaskSE {
	46	public:
	47	AliAnalysisTaskExtractPerformanceV0();
	48	AliAnalysisTaskExtractPerformanceV0(const char *name);
	49	virtual ~AliAnalysisTaskExtractPerformanceV0();
	50
	51	virtual void UserCreateOutputObjects();
	52	virtual void UserExec(Option_t *option);
	53	virtual void Terminate(Option_t *);
	54	Double_t MyRapidity(Double_t rE, Double_t rPz) const;
548bd092	55
	56	void SetIsNuclear (Bool_t lIsNuclear = kTRUE ) { fkIsNuclear = lIsNuclear; }
	57	void SetIsLowEnergyPP (Bool_t lLowEnergyPP = kTRUE ) { fkLowEnergyPP = lLowEnergyPP; }
	58	void SetUseOnTheFly (Bool_t lUseOnTheFly = kTRUE ) { fkUseOnTheFly = lUseOnTheFly; }
bcb6ffdb	59
	60	private:
	61	// Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
	62	// your data member object is created on the worker nodes and streaming is not needed.
	63	// http://root.cern.ch/download/doc/11InputOutput.pdf, page 14
	64	TList *fListHistV0; //! List of Cascade histograms
	65	TTree *fTree; //! Output Tree
	66
548bd092	67	AliPIDResponse *fPIDResponse; // PID response object
	68
	69	//Objects Controlling Task Behaviour
	70
	71	Bool_t fkIsNuclear; //if true, replace multiplicity est. by centrality (default FALSE)
	72	Bool_t fkLowEnergyPP; //if true, skip FASTOnly (default FALSE)
	73	Bool_t fkUseOnTheFly; //if true, will use On-the-fly V0s instead of Offline V0s (default FALSE)
bcb6ffdb	74
	75	//Variables for Tree
	76	Int_t fTreeVariablePrimaryStatus; //!
	77	Int_t fTreeVariablePrimaryStatusMother; //!
	78	Float_t fTreeVariableChi2V0; //!
	79	Float_t fTreeVariableDcaV0Daughters; //!
	80	Float_t fTreeVariableDcaV0ToPrimVertex; //!
	81	Float_t fTreeVariableDcaPosToPrimVertex; //!
	82	Float_t fTreeVariableDcaNegToPrimVertex; //!
	83	Float_t fTreeVariableV0CosineOfPointingAngle; //!
	84	Float_t fTreeVariableV0Radius; //!
	85	Float_t fTreeVariablePt; //!
	86	Float_t fTreeVariablePtMC; //!
	87	Float_t fTreeVariableRapK0Short; //!
	88	Float_t fTreeVariableRapLambda; //!
	89	Float_t fTreeVariableRapMC; //!
	90	Float_t fTreeVariableInvMassK0s; //!
	91	Float_t fTreeVariableInvMassLambda; //!
	92	Float_t fTreeVariableInvMassAntiLambda; //!
	93	Float_t fTreeVariableAlphaV0; //!
	94	Float_t fTreeVariablePtArmV0;//!
	95	Float_t fTreeVariableNegTotMomentum; //!
	96	Float_t fTreeVariablePosTotMomentum; //!
	97	Float_t fTreeVariableNegTransvMomentum; //!
	98	Float_t fTreeVariablePosTransvMomentum; //!
	99	Float_t fTreeVariableNegTransvMomentumMC; //!
	100	Float_t fTreeVariablePosTransvMomentumMC; //!
	101
	102	Float_t fTreeVariableNSigmasPosProton; //!
	103	Float_t fTreeVariableNSigmasPosPion; //!
	104	Float_t fTreeVariableNSigmasNegProton; //!
	105	Float_t fTreeVariableNSigmasNegPion; //!
	106
	107	Float_t fTreeVariablePtMother; //!
	108	Float_t fTreeVariableV0CreationRadius; //!
	109	Int_t fTreeVariablePID; //!
	110	Int_t fTreeVariablePIDPositive; //!
	111	Int_t fTreeVariablePIDNegative; //!
	112	Int_t fTreeVariablePIDMother; //!
	113	Int_t fTreeVariableIndexStatus; //!
	114	Int_t fTreeVariableIndexStatusMother; //!
	115
	116	//Note: TDistOverTotMom needs a mass hypothesis to be converted to proper decaylength.
	117	Float_t fTreeVariableDistOverTotMom;//!
	118
	119	Float_t fTreeVariablePosEta; //!
	120	Float_t fTreeVariableNegEta; //!
	121
	122	Int_t fTreeVariableLeastNbrCrossedRows;//!
	123	Float_t fTreeVariableLeastRatioCrossedRowsOverFindable;//!
	124	Int_t fTreeVariableMultiplicity;//!
	125
	126	TH1F *fHistV0MultiplicityBeforeTrigSel; //! V0 multiplicity distribution
	127	TH1F *fHistV0MultiplicityForTrigEvt; //! V0 multiplicity distribution
	128	TH1F *fHistV0MultiplicityForSelEvt; //! V0 multiplicity distribution
	129	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnly; //! V0 multiplicity distribution
	130	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup; //! V0 multiplicity distribution
	131
	132	TH1F *fHistMultiplicityBeforeTrigSel; //! multiplicity distribution
	133	TH1F *fHistMultiplicityForTrigEvt; //! multiplicity distribution
	134	TH1F *fHistMultiplicity; //! multiplicity distribution
	135	TH1F *fHistMultiplicityNoTPCOnly; //! multiplicity distribution
	136	TH1F *fHistMultiplicityNoTPCOnlyNoPileup; //! multiplicity distribution
	137
138	//---> Filled At Analysis Scope
139
140	TH3F *f3dHistPrimAnalysisPtVsYVsMultLambda; //! Lambda
141	TH3F *f3dHistPrimAnalysisPtVsYVsMultAntiLambda; //! AntiLambda
142	TH3F *f3dHistPrimAnalysisPtVsYVsMultK0Short; //! K0Short
143
144	//---> Containers for monte carlo information for calculating efficiency!
145
146	TH3F *f3dHistPrimRawPtVsYVsMultLambda; //! Lambda
147	TH3F *f3dHistPrimRawPtVsYVsMultAntiLambda; //! AntiLambda
148	TH3F *f3dHistPrimRawPtVsYVsMultK0Short; //! K0Short
149
150	//---> Filled vs Decay Length
151
152	TH3F *f3dHistPrimRawPtVsYVsDecayLengthLambda; //! Lambda
153	TH3F *f3dHistPrimRawPtVsYVsDecayLengthAntiLambda; //! AntiLambda
154	TH3F *f3dHistPrimRawPtVsYVsDecayLengthK0Short; //! K0Short
155
156	//---> Needed for FeedDown Corrections
157
158	TH3F *f3dHistGenPtVsYVsMultXiMinus; //! Generated Xi- Distrib
159	TH3F *f3dHistGenPtVsYVsMultXiPlus; //! Generated Xi+ Distrib
160
161	TH1F *fHistPVx; //! PVx distrib
162	TH1F *fHistPVy; //! PVy distrib
163	TH1F *fHistPVz; //! PVz distrib
164	TH1F *fHistPVxAnalysis; //! PVx distrib
165	TH1F *fHistPVyAnalysis; //! PVy distrib
166	TH1F *fHistPVzAnalysis; //! PVz distrib
167	TH1F *fHistPVxAnalysisHasHighPtLambda; //! PVx distrib
168	TH1F *fHistPVyAnalysisHasHighPtLambda; //! PVy distrib
169	TH1F *fHistPVzAnalysisHasHighPtLambda; //! PVz distrib
170
171	AliAnalysisTaskExtractPerformanceV0(const AliAnalysisTaskExtractPerformanceV0&); // not implemented
172	AliAnalysisTaskExtractPerformanceV0& operator=(const AliAnalysisTaskExtractPerformanceV0&); // not implemented
173
174	ClassDef(AliAnalysisTaskExtractPerformanceV0, 11);
175	};
176
177	#endif