[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / Cascades / AliAnalysisTaskExtractPerformanceCascade.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 ALIANALYSISTASKEXTRACTPERFORMANCECASCADE_H
#define ALIANALYSISTASKEXTRACTPERFORMANCECASCADE_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 AliAnalysisTaskExtractPerformanceCascade : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskExtractPerformanceCascade();
  AliAnalysisTaskExtractPerformanceCascade(const char *name);
  virtual ~AliAnalysisTaskExtractPerformanceCascade();
  
  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 SetINT7Trigger         (Bool_t lSwitchINT7  = kTRUE ) { fkSwitchINT7   = lSwitchINT7; }
  void SetpARapidityShift     (Double_t lRapShift = 0.465 ) { fpArapidityShift = lRapShift; }
  void SetCentralityEstimator (TString lCentralityEstimator = "V0M" ) { fCentralityEstimator = lCentralityEstimator; }
  
 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  *fListHist;  //! List of Cascade histograms
  TTree  *fTreeCascade;              //! Output Tree, Cascades

  //Objects that have to be streamed:
  AliPIDResponse *fPIDResponse;     // PID response object
  AliESDtrackCuts *fESDtrackCuts;   // ESD track cuts used for primary track definition

  //Objects Controlling Task Behaviour 
  // (have to be streamed too or configuration is lost)
  
  Bool_t fkIsNuclear;   //if true, replace multiplicity est. by centrality (default FALSE) 
  Bool_t fkSwitchINT7 ; //if true, skip FASTOnly (default FALSE)
  Double_t fpArapidityShift; //pA rapidity shift (should be 0.465, usually)
  TString fCentralityEstimator; //Centrality Estimator String value (default V0M)

	Double_t        fV0Sels[7];                     // Array to store the 7 values for the different selections V0 related
	Double_t        fCascSels[8];                   // Array to store the 8 values for the different selections Casc. related


//===========================================================================================
//   Variables for tree, cascades
//===========================================================================================

  Int_t fTreeCascVarCharge;         //! 
  Float_t fTreeCascVarMassAsXi;     //! 
  Float_t fTreeCascVarMassAsOmega;  //! 
  Float_t fTreeCascVarPt;           //!
  Float_t fTreeCascVarPtMC;         //!
  Float_t fTreeCascVarRapMC;        //!
  Float_t fTreeCascVarRapXi;        //!
  Float_t fTreeCascVarRapOmega;     //!
  Float_t fTreeCascVarNegEta;       //!
  Float_t fTreeCascVarPosEta;       //!
  Float_t fTreeCascVarBachEta;      //!
  Float_t fTreeCascVarDCACascDaughters; //!
  Float_t fTreeCascVarDCABachToPrimVtx; //!
  Float_t fTreeCascVarDCAV0Daughters;   //!
  Float_t fTreeCascVarDCAV0ToPrimVtx;   //!
  Float_t fTreeCascVarDCAPosToPrimVtx;  //!
  Float_t fTreeCascVarDCANegToPrimVtx;  //!
  Float_t fTreeCascVarCascCosPointingAngle; //!
  Float_t fTreeCascVarCascRadius;           //!
  Float_t fTreeCascVarV0Mass;               //!
  Float_t fTreeCascVarV0CosPointingAngle;   //!
  Float_t fTreeCascVarV0Radius;             //!
  Int_t   fTreeCascVarLeastNbrClusters;     //!
  Int_t   fTreeCascVarMultiplicity;         //!
  Float_t fTreeCascVarDistOverTotMom;       //!
  Int_t   fTreeCascVarPID;         //!
  Int_t   fTreeCascVarPIDBachelor; //!  
  Int_t   fTreeCascVarPIDNegative; //!
  Int_t   fTreeCascVarPIDPositive; //!
  Float_t fTreeCascVarPosTransMom;   //!
  Float_t fTreeCascVarNegTransMom;   //!
  Float_t fTreeCascVarPosTransMomMC; //!
  Float_t fTreeCascVarNegTransMomMC; //!
  Float_t fTreeCascVarNegNSigmaPion;   //!
  Float_t fTreeCascVarNegNSigmaProton; //!
  Float_t fTreeCascVarPosNSigmaPion;   //! 
  Float_t fTreeCascVarPosNSigmaProton; //! 
  Float_t fTreeCascVarBachNSigmaPion;  //! 
  Float_t fTreeCascVarBachNSigmaKaon;  //! 

//===========================================================================================
//   Histograms
//===========================================================================================

   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

//---> Generated Histograms

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

   TH3F      *f3dHistGenSelectedPtVsYVsMultXiMinus;      //! Generated Xi- Distrib, at event selection level
   TH3F      *f3dHistGenSelectedPtVsYVsMultXiPlus;       //! Generated Xi+ Distrib, at event selection level
   TH3F      *f3dHistGenSelectedPtVsYVsMultOmegaMinus;      //! Generated Omega- Distrib, at event selection level
   TH3F      *f3dHistGenSelectedPtVsYVsMultOmegaPlus;       //! Generated Omega+ Distrib, at event selection level

   TH3F      *f3dHistGenPtVsYCMSVsMultXiMinus;      //! Generated Xi- Distrib
   TH3F      *f3dHistGenPtVsYCMSVsMultXiPlus;       //! Generated Xi+ Distrib
   TH3F      *f3dHistGenPtVsYCMSVsMultOmegaMinus;      //! Generated Omega- Distrib
   TH3F      *f3dHistGenPtVsYCMSVsMultOmegaPlus;       //! Generated Omega+ Distrib

   TH3F      *f3dHistGenSelectedPtVsYCMSVsMultXiMinus;      //! Generated Xi- Distrib, at event selection level
   TH3F      *f3dHistGenSelectedPtVsYCMSVsMultXiPlus;       //! Generated Xi+ Distrib, at event selection level
   TH3F      *f3dHistGenSelectedPtVsYCMSVsMultOmegaMinus;      //! Generated Omega- Distrib, at event selection level
   TH3F      *f3dHistGenSelectedPtVsYCMSVsMultOmegaPlus;       //! Generated Omega+ Distrib, at event selection level

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

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

#endif
Commit	Line	Data
76029adc	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 ALIANALYSISTASKEXTRACTPERFORMANCECASCADE_H
	26	#define ALIANALYSISTASKEXTRACTPERFORMANCECASCADE_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 AliAnalysisTaskExtractPerformanceCascade : public AliAnalysisTaskSE {
	46	public:
	47	AliAnalysisTaskExtractPerformanceCascade();
	48	AliAnalysisTaskExtractPerformanceCascade(const char *name);
	49	virtual ~AliAnalysisTaskExtractPerformanceCascade();
	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;
	55
	56	void SetIsNuclear (Bool_t lIsNuclear = kTRUE ) { fkIsNuclear = lIsNuclear; }
2213f6fc	57	void SetINT7Trigger (Bool_t lSwitchINT7 = kTRUE ) { fkSwitchINT7 = lSwitchINT7; }
da3539d8	58	void SetpARapidityShift (Double_t lRapShift = 0.465 ) { fpArapidityShift = lRapShift; }
da3539d8	59	void SetCentralityEstimator (TString lCentralityEstimator = "V0M" ) { fCentralityEstimator = lCentralityEstimator; }
76029adc	60
	61	private:
	62	// Note : In ROOT, "//!" means "do not stream the data from Master node to Worker node" ...
	63	// your data member object is created on the worker nodes and streaming is not needed.
	64	// http://root.cern.ch/download/doc/11InputOutput.pdf, page 14
	65	TList *fListHist; //! List of Cascade histograms
	66	TTree *fTreeCascade; //! Output Tree, Cascades
	67
	68	//Objects that have to be streamed:
	69	AliPIDResponse *fPIDResponse; // PID response object
	70	AliESDtrackCuts *fESDtrackCuts; // ESD track cuts used for primary track definition
	71
	72	//Objects Controlling Task Behaviour
	73	// (have to be streamed too or configuration is lost)
	74
	75	Bool_t fkIsNuclear; //if true, replace multiplicity est. by centrality (default FALSE)
2213f6fc	76	Bool_t fkSwitchINT7 ; //if true, skip FASTOnly (default FALSE)
da3539d8	77	Double_t fpArapidityShift; //pA rapidity shift (should be 0.465, usually)
da3539d8	78	TString fCentralityEstimator; //Centrality Estimator String value (default V0M)
76029adc	79
d8841e95	80	Double_t fV0Sels[7]; // Array to store the 7 values for the different selections V0 related
	81	Double_t fCascSels[8]; // Array to store the 8 values for the different selections Casc. related
	82
	83
76029adc	84	//===========================================================================================
	85	// Variables for tree, cascades
	86	//===========================================================================================
	87
	88	Int_t fTreeCascVarCharge; //!
	89	Float_t fTreeCascVarMassAsXi; //!
	90	Float_t fTreeCascVarMassAsOmega; //!
	91	Float_t fTreeCascVarPt; //!
	92	Float_t fTreeCascVarPtMC; //!
	93	Float_t fTreeCascVarRapMC; //!
	94	Float_t fTreeCascVarRapXi; //!
	95	Float_t fTreeCascVarRapOmega; //!
	96	Float_t fTreeCascVarNegEta; //!
	97	Float_t fTreeCascVarPosEta; //!
	98	Float_t fTreeCascVarBachEta; //!
	99	Float_t fTreeCascVarDCACascDaughters; //!
	100	Float_t fTreeCascVarDCABachToPrimVtx; //!
	101	Float_t fTreeCascVarDCAV0Daughters; //!
	102	Float_t fTreeCascVarDCAV0ToPrimVtx; //!
	103	Float_t fTreeCascVarDCAPosToPrimVtx; //!
	104	Float_t fTreeCascVarDCANegToPrimVtx; //!
	105	Float_t fTreeCascVarCascCosPointingAngle; //!
	106	Float_t fTreeCascVarCascRadius; //!
	107	Float_t fTreeCascVarV0Mass; //!
	108	Float_t fTreeCascVarV0CosPointingAngle; //!
	109	Float_t fTreeCascVarV0Radius; //!
	110	Int_t fTreeCascVarLeastNbrClusters; //!
	111	Int_t fTreeCascVarMultiplicity; //!
	112	Float_t fTreeCascVarDistOverTotMom; //!
	113	Int_t fTreeCascVarPID; //!
	114	Int_t fTreeCascVarPIDBachelor; //!
	115	Int_t fTreeCascVarPIDNegative; //!
	116	Int_t fTreeCascVarPIDPositive; //!
	117	Float_t fTreeCascVarPosTransMom; //!
	118	Float_t fTreeCascVarNegTransMom; //!
	119	Float_t fTreeCascVarPosTransMomMC; //!
	120	Float_t fTreeCascVarNegTransMomMC; //!
76029adc	121	Float_t fTreeCascVarNegNSigmaPion; //!
	122	Float_t fTreeCascVarNegNSigmaProton; //!
	123	Float_t fTreeCascVarPosNSigmaPion; //!
	124	Float_t fTreeCascVarPosNSigmaProton; //!
	125	Float_t fTreeCascVarBachNSigmaPion; //!
	126	Float_t fTreeCascVarBachNSigmaKaon; //!
	127
	128	//===========================================================================================
	129	// Histograms
	130	//===========================================================================================
	131
	132	TH1F *fHistV0MultiplicityBeforeTrigSel; //! V0 multiplicity distribution
	133	TH1F *fHistV0MultiplicityForTrigEvt; //! V0 multiplicity distribution
	134	TH1F *fHistV0MultiplicityForSelEvt; //! V0 multiplicity distribution
	135	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnly; //! V0 multiplicity distribution
	136	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup; //! V0 multiplicity distribution
	137
	138	TH1F *fHistMultiplicityBeforeTrigSel; //! multiplicity distribution
	139	TH1F *fHistMultiplicityForTrigEvt; //! multiplicity distribution
	140	TH1F *fHistMultiplicity; //! multiplicity distribution
	141	TH1F *fHistMultiplicityNoTPCOnly; //! multiplicity distribution
	142	TH1F *fHistMultiplicityNoTPCOnlyNoPileup; //! multiplicity distribution
	143
	144	//---> Generated Histograms
	145
	146	TH3F *f3dHistGenPtVsYVsMultXiMinus; //! Generated Xi- Distrib
	147	TH3F *f3dHistGenPtVsYVsMultXiPlus; //! Generated Xi+ Distrib
	148	TH3F *f3dHistGenPtVsYVsMultOmegaMinus; //! Generated Omega- Distrib
	149	TH3F *f3dHistGenPtVsYVsMultOmegaPlus; //! Generated Omega+ Distrib
	150
	151	TH3F *f3dHistGenSelectedPtVsYVsMultXiMinus; //! Generated Xi- Distrib, at event selection level
	152	TH3F *f3dHistGenSelectedPtVsYVsMultXiPlus; //! Generated Xi+ Distrib, at event selection level
	153	TH3F *f3dHistGenSelectedPtVsYVsMultOmegaMinus; //! Generated Omega- Distrib, at event selection level
	154	TH3F *f3dHistGenSelectedPtVsYVsMultOmegaPlus; //! Generated Omega+ Distrib, at event selection level
	155
da3539d8	156	TH3F *f3dHistGenPtVsYCMSVsMultXiMinus; //! Generated Xi- Distrib
	157	TH3F *f3dHistGenPtVsYCMSVsMultXiPlus; //! Generated Xi+ Distrib
	158	TH3F *f3dHistGenPtVsYCMSVsMultOmegaMinus; //! Generated Omega- Distrib
	159	TH3F *f3dHistGenPtVsYCMSVsMultOmegaPlus; //! Generated Omega+ Distrib
	160
	161	TH3F *f3dHistGenSelectedPtVsYCMSVsMultXiMinus; //! Generated Xi- Distrib, at event selection level
	162	TH3F *f3dHistGenSelectedPtVsYCMSVsMultXiPlus; //! Generated Xi+ Distrib, at event selection level
	163	TH3F *f3dHistGenSelectedPtVsYCMSVsMultOmegaMinus; //! Generated Omega- Distrib, at event selection level
	164	TH3F *f3dHistGenSelectedPtVsYCMSVsMultOmegaPlus; //! Generated Omega+ Distrib, at event selection level
	165
76029adc	166	TH1F *fHistPVx; //! PVx distrib
	167	TH1F *fHistPVy; //! PVy distrib
	168	TH1F *fHistPVz; //! PVz distrib
	169	TH1F *fHistPVxAnalysis; //! PVx distrib
	170	TH1F *fHistPVyAnalysis; //! PVy distrib
	171	TH1F *fHistPVzAnalysis; //! PVz distrib
	172
	173	AliAnalysisTaskExtractPerformanceCascade(const AliAnalysisTaskExtractPerformanceCascade&); // not implemented
	174	AliAnalysisTaskExtractPerformanceCascade& operator=(const AliAnalysisTaskExtractPerformanceCascade&); // not implemented
	175
	176	ClassDef(AliAnalysisTaskExtractPerformanceCascade, 11);
	177	};
	178
	179	#endif