[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / Hypernuclei / AliAnalysisTaskLambdaNAOD.h

#ifndef ALIANALYSISTASKLAMBDANAOD_H
#define ALIANALYSISTASKLAMBDANAOD_H


/**************************************************************************
 * Author : Nicole Alice Martin (nicole.alice.martin@cern.ch)                  *
 *                                                                        *
 * 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.                  *
 **************************************************************************/

//-----------------------------------------------------------------
//               AliAnalysisTaskLambdaNAOD  class
//  task for the investigation of (anti-)lambda-n bound state
//          uses the V0 finders, based on AODs or ESDS
//-----------------------------------------------------------------

class TF1;
class TH1F;
class TH2F;
class TH3F;
class THnSparse;

class TTree;

class AliAODInputHandler;
class AliAODEvent;
class AliAODVertex;
class AliAODv0;

class AliESDtrackCuts;

class AliESDInputHandler;
class AliESDEvent;
class AliESDVertex;
class AliESDv0;

#include <fstream>
#include "AliAnalysisTaskSE.h"
#include "THnSparse.h"
#include "AliStack.h"
#include "AliVTrack.h"

#define maxNofTracks 100000

class AliAnalysisTaskLambdaNAOD : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskLambdaNAOD();
  AliAnalysisTaskLambdaNAOD(const char *name);
  virtual ~AliAnalysisTaskLambdaNAOD();// {}
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(const Option_t*);
  Int_t  Initialize();
  Int_t  SetupEvent();
  void   ResetEvent();
  void SetAnalysisType               (const char* analysisType          = "ESD") { fAnalysisType                = analysisType;               }


  /** Types of pdgCode */
  enum PdgCodeType_t { 
    kPDGPionPlus,
    kPDGPionMinus,
    kPDGProton,
    kPDGAntiProton,
    kPDGDeuteron,
    kPDGAntiDeuteron,
    kPDGHelium3,
    kPDGAntiHelium3,
    kPDGLambda,
    kPDGAntiLambda,
    kPDGLambdaNeutron,
    kPDGAntiLambdaNeutron,
    kPDGHypertriton,
    kPDGAntiHypertriton,
    kPdgCodeMax  // Number of enum entries
  };

  /** Types of Mass */
  enum MassType_t { 
    kMassPion,
    kMassProton,
    kMassDeuteron,
    kMassTriton,
    kMassHelium3,
    kMassMax  // Number of enum entries
  };

  /** Array of types of pdgCodes */
  static const Int_t fgkPdgCode[];           //! transient

  /** Array of types of pdgCodes */
  static const Double_t fgkMass[];        //! transient

 private:


  TString         fAnalysisType;                  // "ESD" or "AOD" analysis type	

  AliInputEventHandler *fEventHandler;            //for ESDs or AODs

  AliESDtrackCuts    *fESDtrackCutsV0;                            // basic cut variables for v0's
  AliESDpid          *fESDpid;                                  // basic TPC object for n-sigma cuts
  AliPIDResponse     *fPIDResponse;                   //! PID response object

  //
   
  
  TTree             *fTreeV0;                          //!
 
  TH1F              *fHistNumberOfEvents;                      //! histo to count the number of events
  TH1F              *fHistTriggerStat;                         //! Trigger statistics
  TH1F              *fHistTriggerStatAfterEventSelection;      //! Trigger statistics
  TH1F              *fHistLambdaNeutronPtGen;                  //! for MC
  TH1F              *fHistAntiLambdaNeutronPtGen;              //! for MC
  TH1F              *fHistLambdaNeutronInvaMassGen;            //! for MC
  TH1F              *fHistAntiLambdaNeutronInvaMassGen;        //! for MC
  TH1F              *fHistLambdaNeutronDecayLengthGen;         //! for MC
  TH1F              *fHistAntiLambdaNeutronDecayLengthGen;     //! for MC
  TH1F              *fHistLambdaNeutronPtAso;                  //! for MC
  TH1F              *fHistLambdaNeutronPtAsoCuts;              //! for MC
  TH1F              *fHistAntiLambdaNeutronPtAso;              //! for MC
  TH1F              *fHistAntiLambdaNeutronPtAsoCuts;          //! for MC
  TH1F              *fHistLambdaNeutronInvaMassAso;            //! for MC
  TH1F              *fHistAntiLambdaNeutronInvaMassAso;        //! for MC
  TH1F              *fHistLambdaNeutronDecayLengthAso;         //! for MC
  TH1F              *fHistAntiLambdaNeutronDecayLengthAso;     //! for MC

  TH2F              *fof;                                      //! debug histo for OnTheFlyStatus
  TH2F              *fHistArmenterosPodolanskiDeuteronPion;    //!
  TH2F              *fHistArmenterosPodolanskiAntiDeuteronPion;//!

  TH3F              *fHistDeDxQA;                              //! histo for a QA dE/dx

  Int_t              fNTriggers;                               //! N Triggers used

  Bool_t             fMCtrue;                                  //! flag if real data or MC is processed
  Bool_t             fOnlyQA;                                  //! flag if only QA histograms should be filled
  Bool_t             fTriggerFired[5];                         //! TriggerFired 0: MB | 1: CE | 2: SC | 3: EJE | 4: EGA


  //Tree variables
  //AliAODv0 *fV0object;                                         //! Tree variable
  Int_t  fItrk;                                                //! Tree variable

  Int_t fV0finder[maxNofTracks];                               //! Tree variable
  Int_t fkMB[maxNofTracks];                                    //! Tree variable
  Int_t fkCentral[maxNofTracks];                               //! Tree variable
  Int_t fkSemiCentral[maxNofTracks];                           //! Tree variable
  Int_t fkEMCEJE[maxNofTracks];                                //! Tree variable
  Int_t fkEMCEGA[maxNofTracks];                                //! Tree variable 

  Int_t fCentrality[maxNofTracks];                             //! Tree variable
  Int_t fMultiplicity[maxNofTracks];                           //! Tree variable

  Double_t fPtotN[maxNofTracks];                               //! Tree variable
  Double_t fPtotP[maxNofTracks];                               //! Tree variable
  Double_t fMotherPt[maxNofTracks];                            //! Tree variable

  Double_t fdEdxN[maxNofTracks];                               //! Tree variable
  Double_t fdEdxP[maxNofTracks];                               //! Tree variable
  Double_t fSignN[maxNofTracks];                               //! Tree variable
  Double_t fSignP[maxNofTracks];                               //! Tree variable

  Float_t fDCAv0[maxNofTracks];                                //! Tree variable
  Float_t fCosinePAv0[maxNofTracks];                           //! Tree variable
  Float_t fDecayRadiusTree[maxNofTracks];                      //! Tree variable
  Double_t fInvaMassDeuteronPionTree[maxNofTracks];            //! Tree variable
  Int_t fChargeComboDeuteronPionTree[maxNofTracks];            //! Tree variable
  Bool_t fIsCorrectlyAssociated[maxNofTracks];                 //! Tree variable

  Double_t fAmenterosAlphaTree[maxNofTracks];                  //! Tree variable
  Double_t fAmenterosQtTree[maxNofTracks];                     //! Tree variable
  Int_t fRotationTree[maxNofTracks];                        //! Tree variable


  TObjArray         *fOutputContainer;         //! output data container for the histogramms
  //
  void  BinLogAxis(const THnSparse *h, Int_t axisNumber); //define function for log axis for search for Anti-Alpha candidates 
  //
  
  /** Check if event is triggred */
  Bool_t   IsTriggered();
  Bool_t   DeuteronPID(AliVTrack *trackP, AliVTrack *trackN, Double_t ptotP, Double_t ptotN, Int_t runNumber, Bool_t isDeuteron[3]);
  Bool_t   PionPID(AliVTrack *trackP, AliVTrack *trackN, Double_t ptotP, Double_t ptotN, Int_t runNumber, Bool_t isPion[2]);
  Bool_t   TrackCuts(AliVTrack *track, Bool_t testTrackCuts);
  //Bool_t   FilterBit(AliVTrack *track, Bool_t testFilterBit);
  Double_t MomentumInnerParam(AliVTrack *track, Double_t ptot);

 
  void               MCGenerated(AliStack* stack);             //! function to loop over the genrated particles
  
  void               MCTwoBodyDecay (AliStack* stack, const TParticle *tparticleMother, Long_t PDGMother, Long_t PDGFirstDaughter, Long_t PDGSecondDaughter, Double_t massFirstDaughter, Double_t massSecondDaughter);                         //! function to calculate the invariant mass of two daughters and the pt of the mother
  //void RotateKFParticle(AliKFParticle * kfParticle,Double_t angle, const AliVEvent * const ev);

  AliAnalysisTaskLambdaNAOD(const AliAnalysisTaskLambdaNAOD&); // not implemented
  AliAnalysisTaskLambdaNAOD& operator=(const AliAnalysisTaskLambdaNAOD&); // not implemented
  //
  ClassDef(AliAnalysisTaskLambdaNAOD, 1); // example of analysis
};

#endif
Commit	Line	Data
f291255e	1	#ifndef ALIANALYSISTASKLAMBDANAOD_H
	2	#define ALIANALYSISTASKLAMBDANAOD_H
	3
	4
e2d2636c	5	/**************************************************************************
	6	* Author : Nicole Alice Martin (nicole.alice.martin@cern.ch) *
	7	* *
	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	// AliAnalysisTaskLambdaNAOD class
	21	// task for the investigation of (anti-)lambda-n bound state
	22	// uses the V0 finders, based on AODs or ESDS
	23	//-----------------------------------------------------------------
	24
e2d2636c	25	class TF1;
	26	class TH1F;
	27	class TH2F;
	28	class TH3F;
	29	class THnSparse;
	30
	31	class TTree;
	32
	33	class AliAODInputHandler;
	34	class AliAODEvent;
	35	class AliAODVertex;
	36	class AliAODv0;
	37
	38	class AliESDtrackCuts;
	39
	40	class AliESDInputHandler;
	41	class AliESDEvent;
	42	class AliESDVertex;
	43	class AliESDv0;
	44
	45	#include <fstream>
	46	#include "AliAnalysisTaskSE.h"
	47	#include "THnSparse.h"
f291255e	48	#include "AliStack.h"
f291255e	49	#include "AliVTrack.h"
e2d2636c	50
	51	#define maxNofTracks 100000
	52
	53	class AliAnalysisTaskLambdaNAOD : public AliAnalysisTaskSE {
	54	public:
	55	AliAnalysisTaskLambdaNAOD();
	56	AliAnalysisTaskLambdaNAOD(const char *name);
	57	virtual ~AliAnalysisTaskLambdaNAOD();// {}
	58
	59	virtual void UserCreateOutputObjects();
	60	virtual void UserExec(Option_t *option);
	61	virtual void Terminate(const Option_t*);
	62	Int_t Initialize();
	63	Int_t SetupEvent();
	64	void ResetEvent();
	65	void SetAnalysisType (const char* analysisType = "ESD") { fAnalysisType = analysisType; }
	66
	67
	68	/** Types of pdgCode */
	69	enum PdgCodeType_t {
	70	kPDGPionPlus,
	71	kPDGPionMinus,
	72	kPDGProton,
	73	kPDGAntiProton,
	74	kPDGDeuteron,
	75	kPDGAntiDeuteron,
	76	kPDGHelium3,
	77	kPDGAntiHelium3,
	78	kPDGLambda,
	79	kPDGAntiLambda,
	80	kPDGLambdaNeutron,
	81	kPDGAntiLambdaNeutron,
	82	kPDGHypertriton,
	83	kPDGAntiHypertriton,
	84	kPdgCodeMax // Number of enum entries
	85	};
	86
	87	/** Types of Mass */
	88	enum MassType_t {
	89	kMassPion,
	90	kMassProton,
	91	kMassDeuteron,
	92	kMassTriton,
	93	kMassHelium3,
	94	kMassMax // Number of enum entries
	95	};
	96
	97	/** Array of types of pdgCodes */
	98	static const Int_t fgkPdgCode[]; //! transient
	99
	100	/** Array of types of pdgCodes */
	101	static const Double_t fgkMass[]; //! transient
	102
	103	private:
	104
	105
	106	TString fAnalysisType; // "ESD" or "AOD" analysis type
	107
	108	AliInputEventHandler *fEventHandler; //for ESDs or AODs
	109
	110	AliESDtrackCuts *fESDtrackCutsV0; // basic cut variables for v0's
	111	AliESDpid *fESDpid; // basic TPC object for n-sigma cuts
	112	AliPIDResponse *fPIDResponse; //! PID response object
	113
114	//
115
116
117	TTree *fTreeV0; //!
118
119	TH1F *fHistNumberOfEvents; //! histo to count the number of events
120	TH1F *fHistTriggerStat; //! Trigger statistics
8bb06d05	121	TH1F *fHistTriggerStatAfterEventSelection; //! Trigger statistics
e2d2636c	122	TH1F *fHistLambdaNeutronPtGen; //! for MC
	123	TH1F *fHistAntiLambdaNeutronPtGen; //! for MC
	124	TH1F *fHistLambdaNeutronInvaMassGen; //! for MC
	125	TH1F *fHistAntiLambdaNeutronInvaMassGen; //! for MC
	126	TH1F *fHistLambdaNeutronDecayLengthGen; //! for MC
	127	TH1F *fHistAntiLambdaNeutronDecayLengthGen; //! for MC
	128	TH1F *fHistLambdaNeutronPtAso; //! for MC
	129	TH1F *fHistLambdaNeutronPtAsoCuts; //! for MC
	130	TH1F *fHistAntiLambdaNeutronPtAso; //! for MC
	131	TH1F *fHistAntiLambdaNeutronPtAsoCuts; //! for MC
	132	TH1F *fHistLambdaNeutronInvaMassAso; //! for MC
	133	TH1F *fHistAntiLambdaNeutronInvaMassAso; //! for MC
	134	TH1F *fHistLambdaNeutronDecayLengthAso; //! for MC
	135	TH1F *fHistAntiLambdaNeutronDecayLengthAso; //! for MC
	136
	137	TH2F *fof; //! debug histo for OnTheFlyStatus
	138	TH2F *fHistArmenterosPodolanskiDeuteronPion; //!
	139	TH2F *fHistArmenterosPodolanskiAntiDeuteronPion;//!
	140
	141	TH3F *fHistDeDxQA; //! histo for a QA dE/dx
	142
	143	Int_t fNTriggers; //! N Triggers used
	144
	145	Bool_t fMCtrue; //! flag if real data or MC is processed
	146	Bool_t fOnlyQA; //! flag if only QA histograms should be filled
	147	Bool_t fTriggerFired[5]; //! TriggerFired 0: MB \| 1: CE \| 2: SC \| 3: EJE \| 4: EGA
	148
	149
	150
	151	//Tree variables
1d281cbe	152	//AliAODv0 *fV0object; //! Tree variable
e2d2636c	153	Int_t fItrk; //! Tree variable
	154
	155	Int_t fV0finder[maxNofTracks]; //! Tree variable
	156	Int_t fkMB[maxNofTracks]; //! Tree variable
	157	Int_t fkCentral[maxNofTracks]; //! Tree variable
	158	Int_t fkSemiCentral[maxNofTracks]; //! Tree variable
233f521e	159	Int_t fkEMCEJE[maxNofTracks]; //! Tree variable
233f521e	160	Int_t fkEMCEGA[maxNofTracks]; //! Tree variable
e2d2636c	161
85158d91	162	Int_t fCentrality[maxNofTracks]; //! Tree variable
	163	Int_t fMultiplicity[maxNofTracks]; //! Tree variable
	164
e2d2636c	165	Double_t fPtotN[maxNofTracks]; //! Tree variable
	166	Double_t fPtotP[maxNofTracks]; //! Tree variable
	167	Double_t fMotherPt[maxNofTracks]; //! Tree variable
	168
	169	Double_t fdEdxN[maxNofTracks]; //! Tree variable
	170	Double_t fdEdxP[maxNofTracks]; //! Tree variable
	171	Double_t fSignN[maxNofTracks]; //! Tree variable
	172	Double_t fSignP[maxNofTracks]; //! Tree variable
	173
	174	Float_t fDCAv0[maxNofTracks]; //! Tree variable
	175	Float_t fCosinePAv0[maxNofTracks]; //! Tree variable
	176	Float_t fDecayRadiusTree[maxNofTracks]; //! Tree variable
	177	Double_t fInvaMassDeuteronPionTree[maxNofTracks]; //! Tree variable
	178	Int_t fChargeComboDeuteronPionTree[maxNofTracks]; //! Tree variable
	179	Bool_t fIsCorrectlyAssociated[maxNofTracks]; //! Tree variable
	180
233f521e	181	Double_t fAmenterosAlphaTree[maxNofTracks]; //! Tree variable
233f521e	182	Double_t fAmenterosQtTree[maxNofTracks]; //! Tree variable
e2d2636c	183	Int_t fRotationTree[maxNofTracks]; //! Tree variable
	184
	185
	186
	187
	188	TObjArray *fOutputContainer; //! output data container for the histogramms
	189	//
	190	void BinLogAxis(const THnSparse *h, Int_t axisNumber); //define function for log axis for search for Anti-Alpha candidates
	191	//
	192
	193	/** Check if event is triggred */
	194	Bool_t IsTriggered();
	195	Bool_t DeuteronPID(AliVTrack trackP, AliVTrack trackN, Double_t ptotP, Double_t ptotN, Int_t runNumber, Bool_t isDeuteron[3]);
	196	Bool_t PionPID(AliVTrack trackP, AliVTrack trackN, Double_t ptotP, Double_t ptotN, Int_t runNumber, Bool_t isPion[2]);
	197	Bool_t TrackCuts(AliVTrack *track, Bool_t testTrackCuts);
0cea21e4	198	//Bool_t FilterBit(AliVTrack *track, Bool_t testFilterBit);
e2d2636c	199	Double_t MomentumInnerParam(AliVTrack *track, Double_t ptot);
	200
	201
	202	void MCGenerated(AliStack* stack); //! function to loop over the genrated particles
	203
	204	void MCTwoBodyDecay (AliStack* stack, const TParticle *tparticleMother, Long_t PDGMother, Long_t PDGFirstDaughter, Long_t PDGSecondDaughter, Double_t massFirstDaughter, Double_t massSecondDaughter); //! function to calculate the invariant mass of two daughters and the pt of the mother
233f521e	205	//void RotateKFParticle(AliKFParticle * kfParticle,Double_t angle, const AliVEvent * const ev);
233f521e	206
e2d2636c	207	AliAnalysisTaskLambdaNAOD(const AliAnalysisTaskLambdaNAOD&); // not implemented
	208	AliAnalysisTaskLambdaNAOD& operator=(const AliAnalysisTaskLambdaNAOD&); // not implemented
	209	//
	210	ClassDef(AliAnalysisTaskLambdaNAOD, 1); // example of analysis
	211	};
	212
	213	#endif