[u/mrichter/AliRoot.git] / PWG4 / GammaConv / AliAnalysisTaskGammaConversion.h

#ifndef ALIANALYSISTASKGAMMACONVERSION_H
#define ALIANALYSISTASKGAMMACONVERSION_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

////////////////////////////////////////////////
//--------------------------------------------- 
// Class used to do analysis on conversion pairs
//---------------------------------------------
////////////////////////////////////////////////
 
#include "AliAnalysisTaskSE.h"
#include <vector>
#include "AliV0Reader.h"
#include "AliGammaConversionAODObject.h"

class TNtuple;
class AliGammaConversionHistograms;
class AliESDv0;
class AliKFParticle;
class AliESDInputHandler;
class AliESDEvent;
class AliAODEvent;
class TList;
class AliStack;
class AliESDtrackCuts;


class AliAnalysisTaskGammaConversion : public AliAnalysisTaskSE
{
 public:
  AliAnalysisTaskGammaConversion();
  AliAnalysisTaskGammaConversion(const char* name);
  virtual ~AliAnalysisTaskGammaConversion() ;// virtual destructor
 
  // Implementation of interface methods
  virtual void UserCreateOutputObjects();
  virtual void Init();
  virtual void LocalInit() {Init();}
  virtual void Exec(Option_t *option);
  virtual void Terminate(Option_t *option);
  virtual void ConnectInputData(Option_t *);
	
  void ProcessMCData();
  void ProcessV0sNoCut();
  void ProcessV0s();
  void ProcessGammasForNeutralMesonAnalysis();

  // AOD
  TString GetAODBranchName() const {return  fAODBranchName;}
  void SetAODBranchName(TString name)  {fAODBranchName = name ;}	
  void FillAODWithConversionGammas();
  // end AOD


 // for GammaJetAnalysis
  void ProcessGammasForGammaJetAnalysis();
  void CreateListOfChargedParticles();
  Double_t GetMinimumDistanceToCharge(Int_t);
  void CalculateJetCone(Int_t,Int_t);
  Int_t GetIndexHighestPtGamma();
  void SetESDtrackCuts();
  // end of Gamma Jet

  void SetMinPtForGammaJet(Double_t minPtForGammaJet){fMinPtForGammaJet=minPtForGammaJet;}
  void SetMinIsoConeSize(Double_t minIsoConeSize){fMinIsoConeSize=minIsoConeSize;}
  void SetMinPtIsoCone(Double_t minPtIsoCone){fMinPtIsoCone=minPtIsoCone;}
  void SetMinPtGamChargedCorr(Double_t minPtGamChargedCorr){fMinPtGamChargedCorr=minPtGamChargedCorr;}
  void SetMinPtJetCone(Double_t minPtJetCone){fMinPtJetCone=minPtJetCone;}

  void SetHistograms(AliGammaConversionHistograms *const histograms){fHistograms=histograms;}
  void SetDoMCTruth(Bool_t flag){fDoMCTruth=flag;}
  void SetDoNeutralMeson(Bool_t flag){fDoNeutralMeson=flag;}
  void SetDoJet(Bool_t flag){fDoJet=flag;}
  void SetDoChic(Bool_t flag){fDoChic=flag;}

  void SetElectronMass(Double_t electronMass){fElectronMass = electronMass;}
  void SetGammaMass(Double_t gammaMass){fGammaMass = gammaMass;}
  void SetGammaWidth(Double_t gammaWidth){fGammaWidth = gammaWidth;}
  void SetPi0Mass(Double_t pi0Mass){fPi0Mass = pi0Mass;}
  void SetPi0Width(Double_t pi0Width){fPi0Width = pi0Width;}
  void SetEtaMass(Double_t etaMass){fEtaMass = etaMass;}
  void SetEtaWidth(Double_t etaWidth){fEtaWidth = etaWidth;}
  void SetMinOpeningAngleGhostCut(Double_t ghostCut){fMinOpeningAngleGhostCut = ghostCut;}
  void SetV0Reader(AliV0Reader* const reader){fV0Reader=reader;}
  void SetCalculateBackground(Bool_t bg){fCalculateBackground=bg;}
  void CalculateBackground();
  void SetWriteNtuple(Bool_t writeNtuple){fWriteNtuple = writeNtuple;}
  void FillNtuple();
  Double_t GetMCOpeningAngle(TParticle* const daughter0, TParticle* const daughter1) const;
  void CheckV0Efficiency();


  //////////////////Chi_c Analysis////////////////////////////
  void GetPID(AliESDtrack *track, Stat_t &pid, Stat_t &weight);	
  double GetSigmaToVertex(AliESDtrack* t);
  void ElectronBackground(TString hBg, vector <TLorentzVector> e);
  void FillAngle(TString histoName,vector <TLorentzVector> tlVeNeg, vector <TLorentzVector> tlVePos);
  void FillElectronInvMass(TString histoName, vector <TLorentzVector> negativeElectron, 
	vector <TLorentzVector> positiveElectron);
  void FillGammaElectronInvMass(TString histoMass,TString histoDiff,vector <AliKFParticle> fKFGammas,
        vector <TLorentzVector> tlVeNeg,vector<TLorentzVector> tlVePos);
  void CleanWithAngleCuts(vector <AliESDtrack*> negativeElectrons,
	vector <AliESDtrack*> positiveElectrons, vector <AliKFParticle> gammas);
  vector <TLorentzVector> GetTLorentzVector(vector <AliESDtrack*> esdTrack);	
  void ProcessGammaElectronsForChicAnalysis();
  ///////////////////////////////////////////////////////////////


 private:
  AliAnalysisTaskGammaConversion(const AliAnalysisTaskGammaConversion&); // Not implemented
  AliAnalysisTaskGammaConversion& operator=(const AliAnalysisTaskGammaConversion&); // Not implemented

  AliV0Reader* fV0Reader; // The V0 reader object 

  AliStack * fStack; // pointer to the MC particle stack
  AliESDEvent* fESDEvent; //pointer to the ESDEvent
  TList * fOutputContainer ; // Histogram container

  AliGammaConversionHistograms *fHistograms; // Pointer to the histogram handling class

  Bool_t fDoMCTruth; // Flag to switch on/off MC truth 
  Bool_t fDoNeutralMeson;
  Bool_t fDoJet;
  Bool_t fDoChic;
    
  vector<TParticle*> fMCAllGammas; // vector containing all MC gammas
  vector<TParticle*> fMCPi0s; //vector containing all MC Pi0s
  vector<TParticle*> fMCEtas; //vector containing all MC Etas
  vector<TParticle*> fMCGammaChic; //vector containing all MC Chi_c's

  vector<AliKFParticle> fKFReconstructedGammas; // vector containing all reconstructed gammas
  vector<Bool_t> fIsTrueReconstructedGammas;    // vector containing information if this was a true gamma or not (follows the index of fKFReconstructedGammas)
  vector<Int_t> fElectronv1; // vector containing index of electron 1
  vector<Int_t> fElectronv2; // vector containing index of electron 2

  ///////Chi_c Analysis///////////////////////////
  vector<AliESDtrack*> fCurrentEventPosElectron;       // comment here
  vector<AliESDtrack*> fPreviousEventPosElectron;      //comment here
  vector<AliESDtrack*> fCurrentEventNegElectron;       //comment here
  vector<AliESDtrack*> fPreviousEventNegElectron;      //comment here
  vector<AliKFParticle> fKFReconstructedGammasCut;     //comment here
  vector<TLorentzVector> fPreviousEventTLVNegElectron; //comment here
  vector<TLorentzVector> fPreviousEventTLVPosElectron; //comment here
  //////////////////////////////////////////////////	

  //mass defines
  Double_t fElectronMass; //electron mass
  Double_t fGammaMass;    //gamma mass
  Double_t fPi0Mass;      //pi0mass
  Double_t fEtaMass;      //eta mass

  // width defines
  Double_t fGammaWidth; //gamma width cut
  Double_t fPi0Width;   // pi0 width cut
  Double_t fEtaWidth;   // eta width cut

  Double_t fMinOpeningAngleGhostCut; // minimum angle cut

  AliESDtrackCuts* fEsdTrackCuts;           // Object containing the parameters of the esd track cuts

  Bool_t fCalculateBackground; //flag to set backgrount calculation on/off
  Bool_t fWriteNtuple;         // flag to set if writing to ntuple on/off
  TNtuple *fGammaNtuple;       // Ntuple for gamma values
  TNtuple *fNeutralMesonNtuple;// NTuple for mesons

  Int_t fTotalNumberOfAddedNtupleEntries; // number of added ntuple entries

  vector<AliESDtrack*> fChargedParticles;
  vector<Int_t> fChargedParticlesId;

  Double_t fGammaPtHighest;
  Double_t fMinPtForGammaJet;
  Double_t fMinIsoConeSize;
  Double_t fMinPtIsoCone;
  Double_t fMinPtGamChargedCorr;
  Double_t fMinPtJetCone;
  Int_t    fLeadingChargedIndex;

  TClonesArray* fAODBranch ;        //! selected particles branch
  TString fAODBranchName; // New AOD branch name
  
  vector<AliGammaConversionAODObject> fAODObjects;

  ClassDef(AliAnalysisTaskGammaConversion, 4); // Analysis task for gamma conversions
};
 
#endif //ALIANALYSISTASKGAMMA_H
Commit	Line	Data
d7d7e825	1	#ifndef ALIANALYSISTASKGAMMACONVERSION_H
	2	#define ALIANALYSISTASKGAMMACONVERSION_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////////////////////////////////////////////////
	8	//---------------------------------------------
	9	// Class used to do analysis on conversion pairs
	10	//---------------------------------------------
	11	////////////////////////////////////////////////
	12
	13	#include "AliAnalysisTaskSE.h"
	14	#include <vector>
	15	#include "AliV0Reader.h"
	16	#include "AliGammaConversionAODObject.h"
	17
	18	class TNtuple;
	19	class AliGammaConversionHistograms;
	20	class AliESDv0;
	21	class AliKFParticle;
	22	class AliESDInputHandler;
	23	class AliESDEvent;
	24	class AliAODEvent;
	25	class TList;
	26	class AliStack;
	27	class AliESDtrackCuts;
	28
	29
	30	class AliAnalysisTaskGammaConversion : public AliAnalysisTaskSE
	31	{
	32	public:
	33	AliAnalysisTaskGammaConversion();
	34	AliAnalysisTaskGammaConversion(const char* name);
	35	virtual ~AliAnalysisTaskGammaConversion() ;// virtual destructor
	36
	37	// Implementation of interface methods
	38	virtual void UserCreateOutputObjects();
	39	virtual void Init();
	40	virtual void LocalInit() {Init();}
	41	virtual void Exec(Option_t *option);
	42	virtual void Terminate(Option_t *option);
	43	virtual void ConnectInputData(Option_t *);
	44
	45	void ProcessMCData();
	46	void ProcessV0sNoCut();
	47	void ProcessV0s();
	48	void ProcessGammasForNeutralMesonAnalysis();
	49
	50	// AOD
	51	TString GetAODBranchName() const {return fAODBranchName;}
	52	void SetAODBranchName(TString name) {fAODBranchName = name ;}
	53	void FillAODWithConversionGammas();
	54	// end AOD
	55
	56
	57	// for GammaJetAnalysis
	58	void ProcessGammasForGammaJetAnalysis();
	59	void CreateListOfChargedParticles();
	60	Double_t GetMinimumDistanceToCharge(Int_t);
	61	void CalculateJetCone(Int_t,Int_t);
	62	Int_t GetIndexHighestPtGamma();
	63	void SetESDtrackCuts();
	64	// end of Gamma Jet
65
66	void SetMinPtForGammaJet(Double_t minPtForGammaJet){fMinPtForGammaJet=minPtForGammaJet;}
67	void SetMinIsoConeSize(Double_t minIsoConeSize){fMinIsoConeSize=minIsoConeSize;}
68	void SetMinPtIsoCone(Double_t minPtIsoCone){fMinPtIsoCone=minPtIsoCone;}
69	void SetMinPtGamChargedCorr(Double_t minPtGamChargedCorr){fMinPtGamChargedCorr=minPtGamChargedCorr;}
70	void SetMinPtJetCone(Double_t minPtJetCone){fMinPtJetCone=minPtJetCone;}
71
72	void SetHistograms(AliGammaConversionHistograms *const histograms){fHistograms=histograms;}
73	void SetDoMCTruth(Bool_t flag){fDoMCTruth=flag;}
74	void SetDoNeutralMeson(Bool_t flag){fDoNeutralMeson=flag;}
75	void SetDoJet(Bool_t flag){fDoJet=flag;}
76	void SetDoChic(Bool_t flag){fDoChic=flag;}
77
78	void SetElectronMass(Double_t electronMass){fElectronMass = electronMass;}
79	void SetGammaMass(Double_t gammaMass){fGammaMass = gammaMass;}
80	void SetGammaWidth(Double_t gammaWidth){fGammaWidth = gammaWidth;}
81	void SetPi0Mass(Double_t pi0Mass){fPi0Mass = pi0Mass;}
82	void SetPi0Width(Double_t pi0Width){fPi0Width = pi0Width;}
83	void SetEtaMass(Double_t etaMass){fEtaMass = etaMass;}
84	void SetEtaWidth(Double_t etaWidth){fEtaWidth = etaWidth;}
85	void SetMinOpeningAngleGhostCut(Double_t ghostCut){fMinOpeningAngleGhostCut = ghostCut;}
86	void SetV0Reader(AliV0Reader* const reader){fV0Reader=reader;}
87	void SetCalculateBackground(Bool_t bg){fCalculateBackground=bg;}
88	void CalculateBackground();
89	void SetWriteNtuple(Bool_t writeNtuple){fWriteNtuple = writeNtuple;}
90	void FillNtuple();
91	Double_t GetMCOpeningAngle(TParticle* const daughter0, TParticle* const daughter1) const;
92	void CheckV0Efficiency();
93
94
95	//////////////////Chi_c Analysis////////////////////////////
96	void GetPID(AliESDtrack *track, Stat_t &pid, Stat_t &weight);
97	double GetSigmaToVertex(AliESDtrack* t);
98	void ElectronBackground(TString hBg, vector <TLorentzVector> e);
99	void FillAngle(TString histoName,vector <TLorentzVector> tlVeNeg, vector <TLorentzVector> tlVePos);
100	void FillElectronInvMass(TString histoName, vector <TLorentzVector> negativeElectron,
101	vector <TLorentzVector> positiveElectron);
102	void FillGammaElectronInvMass(TString histoMass,TString histoDiff,vector <AliKFParticle> fKFGammas,
103	vector <TLorentzVector> tlVeNeg,vector<TLorentzVector> tlVePos);
104	void CleanWithAngleCuts(vector <AliESDtrack*> negativeElectrons,
105	vector <AliESDtrack*> positiveElectrons, vector <AliKFParticle> gammas);
106	vector <TLorentzVector> GetTLorentzVector(vector <AliESDtrack*> esdTrack);
107	void ProcessGammaElectronsForChicAnalysis();
108	///////////////////////////////////////////////////////////////
109
110
111
112	private:
113	AliAnalysisTaskGammaConversion(const AliAnalysisTaskGammaConversion&); // Not implemented
114	AliAnalysisTaskGammaConversion& operator=(const AliAnalysisTaskGammaConversion&); // Not implemented
115
116	AliV0Reader* fV0Reader; // The V0 reader object
117
118	AliStack * fStack; // pointer to the MC particle stack
119	AliESDEvent* fESDEvent; //pointer to the ESDEvent
120	TList * fOutputContainer ; // Histogram container
121
122	AliGammaConversionHistograms *fHistograms; // Pointer to the histogram handling class
123
124	Bool_t fDoMCTruth; // Flag to switch on/off MC truth
125	Bool_t fDoNeutralMeson;
126	Bool_t fDoJet;
127	Bool_t fDoChic;
128
129	vector<TParticle*> fMCAllGammas; // vector containing all MC gammas
130	vector<TParticle*> fMCPi0s; //vector containing all MC Pi0s
131	vector<TParticle*> fMCEtas; //vector containing all MC Etas
132	vector<TParticle*> fMCGammaChic; //vector containing all MC Chi_c's
133
134	vector<AliKFParticle> fKFReconstructedGammas; // vector containing all reconstructed gammas
135	vector<Bool_t> fIsTrueReconstructedGammas; // vector containing information if this was a true gamma or not (follows the index of fKFReconstructedGammas)
136	vector<Int_t> fElectronv1; // vector containing index of electron 1
137	vector<Int_t> fElectronv2; // vector containing index of electron 2
138
139	///////Chi_c Analysis///////////////////////////
140	vector<AliESDtrack*> fCurrentEventPosElectron; // comment here
141	vector<AliESDtrack*> fPreviousEventPosElectron; //comment here
142	vector<AliESDtrack*> fCurrentEventNegElectron; //comment here
143	vector<AliESDtrack*> fPreviousEventNegElectron; //comment here
144	vector<AliKFParticle> fKFReconstructedGammasCut; //comment here
145	vector<TLorentzVector> fPreviousEventTLVNegElectron; //comment here
146	vector<TLorentzVector> fPreviousEventTLVPosElectron; //comment here
147	//////////////////////////////////////////////////
148
149	//mass defines
150	Double_t fElectronMass; //electron mass
151	Double_t fGammaMass; //gamma mass
152	Double_t fPi0Mass; //pi0mass
153	Double_t fEtaMass; //eta mass
154
155	// width defines
156	Double_t fGammaWidth; //gamma width cut
157	Double_t fPi0Width; // pi0 width cut
158	Double_t fEtaWidth; // eta width cut
159
160	Double_t fMinOpeningAngleGhostCut; // minimum angle cut
161
162	AliESDtrackCuts* fEsdTrackCuts; // Object containing the parameters of the esd track cuts
163
164	Bool_t fCalculateBackground; //flag to set backgrount calculation on/off
165	Bool_t fWriteNtuple; // flag to set if writing to ntuple on/off
166	TNtuple *fGammaNtuple; // Ntuple for gamma values
167	TNtuple *fNeutralMesonNtuple;// NTuple for mesons
168
169	Int_t fTotalNumberOfAddedNtupleEntries; // number of added ntuple entries
170
171	vector<AliESDtrack*> fChargedParticles;
172	vector<Int_t> fChargedParticlesId;
173
174	Double_t fGammaPtHighest;
175	Double_t fMinPtForGammaJet;
176	Double_t fMinIsoConeSize;
177	Double_t fMinPtIsoCone;
178	Double_t fMinPtGamChargedCorr;
179	Double_t fMinPtJetCone;
180	Int_t fLeadingChargedIndex;
181
182	TClonesArray* fAODBranch ; //! selected particles branch
183	TString fAODBranchName; // New AOD branch name
184
185	vector<AliGammaConversionAODObject> fAODObjects;
186
187	ClassDef(AliAnalysisTaskGammaConversion, 4); // Analysis task for gamma conversions
188	};
189
190	#endif //ALIANALYSISTASKGAMMA_H