[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / Cascades / AliAnalysisTaskExtractCascade.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 ALIANALYSISTASKEXTRACTCASCADE_H
#define ALIANALYSISTASKEXTRACTCASCADE_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 AliAnalysisTaskExtractCascade : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskExtractCascade();
  AliAnalysisTaskExtractCascade(const char *name);
  virtual ~AliAnalysisTaskExtractCascade();
  
  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 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)
  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

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

   AliAnalysisTaskExtractCascade(const AliAnalysisTaskExtractCascade&);            // not implemented
   AliAnalysisTaskExtractCascade& operator=(const AliAnalysisTaskExtractCascade&); // not implemented
   
   ClassDef(AliAnalysisTaskExtractCascade, 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 ALIANALYSISTASKEXTRACTCASCADE_H
	26	#define ALIANALYSISTASKEXTRACTCASCADE_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 AliAnalysisTaskExtractCascade : public AliAnalysisTaskSE {
	46	public:
	47	AliAnalysisTaskExtractCascade();
	48	AliAnalysisTaskExtractCascade(const char *name);
	49	virtual ~AliAnalysisTaskExtractCascade();
	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 SetCentralityEstimator (TString lCentralityEstimator = "V0M" ) { fCentralityEstimator = lCentralityEstimator; }
76029adc	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 *fListHist; //! List of Cascade histograms
	65	TTree *fTreeCascade; //! Output Tree, Cascades
	66
	67	//Objects that have to be streamed:
	68	AliPIDResponse *fPIDResponse; // PID response object
	69	AliESDtrackCuts *fESDtrackCuts; // ESD track cuts used for primary track definition
	70
	71	//Objects Controlling Task Behaviour
	72	// (have to be streamed too or configuration is lost)
	73
	74	Bool_t fkIsNuclear; //if true, replace multiplicity est. by centrality (default FALSE)
2213f6fc	75	Bool_t fkSwitchINT7; //if true, skip FASTOnly (default FALSE)
da3539d8	76	TString fCentralityEstimator; //Centrality Estimator String value (default V0M)
76029adc	77
d8841e95	78	Double_t fV0Sels[7]; // Array to store the 7 values for the different selections V0 related
	79	Double_t fCascSels[8]; // Array to store the 8 values for the different selections Casc. related
	80
76029adc	81	//===========================================================================================
	82	// Variables for tree, cascades
	83	//===========================================================================================
	84
	85	Int_t fTreeCascVarCharge; //!
	86	Float_t fTreeCascVarMassAsXi; //!
	87	Float_t fTreeCascVarMassAsOmega; //!
	88	Float_t fTreeCascVarPt; //!
	89	Float_t fTreeCascVarPtMC; //!
	90	Float_t fTreeCascVarRapMC; //!
	91	Float_t fTreeCascVarRapXi; //!
	92	Float_t fTreeCascVarRapOmega; //!
	93	Float_t fTreeCascVarNegEta; //!
	94	Float_t fTreeCascVarPosEta; //!
	95	Float_t fTreeCascVarBachEta; //!
	96	Float_t fTreeCascVarDCACascDaughters; //!
	97	Float_t fTreeCascVarDCABachToPrimVtx; //!
	98	Float_t fTreeCascVarDCAV0Daughters; //!
	99	Float_t fTreeCascVarDCAV0ToPrimVtx; //!
	100	Float_t fTreeCascVarDCAPosToPrimVtx; //!
	101	Float_t fTreeCascVarDCANegToPrimVtx; //!
	102	Float_t fTreeCascVarCascCosPointingAngle; //!
	103	Float_t fTreeCascVarCascRadius; //!
	104	Float_t fTreeCascVarV0Mass; //!
	105	Float_t fTreeCascVarV0CosPointingAngle; //!
	106	Float_t fTreeCascVarV0Radius; //!
	107	Int_t fTreeCascVarLeastNbrClusters; //!
	108	Int_t fTreeCascVarMultiplicity; //!
	109	Float_t fTreeCascVarDistOverTotMom; //!
	110	Int_t fTreeCascVarPID; //!
	111	Int_t fTreeCascVarPIDBachelor; //!
	112	Int_t fTreeCascVarPIDNegative; //!
	113	Int_t fTreeCascVarPIDPositive; //!
	114	Float_t fTreeCascVarPosTransMom; //!
	115	Float_t fTreeCascVarNegTransMom; //!
	116	Float_t fTreeCascVarPosTransMomMC; //!
	117	Float_t fTreeCascVarNegTransMomMC; //!
	118
	119	Float_t fTreeCascVarNegNSigmaPion; //!
	120	Float_t fTreeCascVarNegNSigmaProton; //!
	121	Float_t fTreeCascVarPosNSigmaPion; //!
	122	Float_t fTreeCascVarPosNSigmaProton; //!
	123	Float_t fTreeCascVarBachNSigmaPion; //!
	124	Float_t fTreeCascVarBachNSigmaKaon; //!
	125
	126	//===========================================================================================
	127	// Histograms
	128	//===========================================================================================
	129
	130	TH1F *fHistV0MultiplicityBeforeTrigSel; //! V0 multiplicity distribution
	131	TH1F *fHistV0MultiplicityForTrigEvt; //! V0 multiplicity distribution
	132	TH1F *fHistV0MultiplicityForSelEvt; //! V0 multiplicity distribution
	133	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnly; //! V0 multiplicity distribution
	134	TH1F *fHistV0MultiplicityForSelEvtNoTPCOnlyNoPileup; //! V0 multiplicity distribution
	135
	136	TH1F *fHistMultiplicityBeforeTrigSel; //! multiplicity distribution
	137	TH1F *fHistMultiplicityForTrigEvt; //! multiplicity distribution
	138	TH1F *fHistMultiplicity; //! multiplicity distribution
	139	TH1F *fHistMultiplicityNoTPCOnly; //! multiplicity distribution
	140	TH1F *fHistMultiplicityNoTPCOnlyNoPileup; //! multiplicity distribution
	141
	142	//---> Generated Histograms
	143
	144	TH1F *fHistPVx; //! PVx distrib
145	TH1F *fHistPVy; //! PVy distrib
146	TH1F *fHistPVz; //! PVz distrib
147	TH1F *fHistPVxAnalysis; //! PVx distrib
148	TH1F *fHistPVyAnalysis; //! PVy distrib
149	TH1F *fHistPVzAnalysis; //! PVz distrib
150
151	AliAnalysisTaskExtractCascade(const AliAnalysisTaskExtractCascade&); // not implemented
152	AliAnalysisTaskExtractCascade& operator=(const AliAnalysisTaskExtractCascade&); // not implemented
153
154	ClassDef(AliAnalysisTaskExtractCascade, 11);
155	};
156
157	#endif