Updated list of files

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

#ifndef ALIANALYSISTASKGAMMACONVERSION_H\r
#define ALIANALYSISTASKGAMMACONVERSION_H\r
 \r
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice                               */\r
\r
////////////////////////////////////////////////\r
//--------------------------------------------- \r
// Class used to do analysis on conversion pairs\r
//---------------------------------------------\r
////////////////////////////////////////////////\r
 \r
#include "AliAnalysisTaskSE.h"\r
#include <vector>\r
#include "AliV0Reader.h"\r
\r
class TNtuple;\r
class AliGammaConversionHistograms;\r
class AliESDv0;\r
class AliKFParticle;\r
class AliESDInputHandler;\r
class AliESDEvent;\r
class AliAODEvent;\r
class TList;\r
class AliStack;\r
\r
class AliAnalysisTaskGammaConversion : public AliAnalysisTaskSE\r
{\r
 public:\r
  AliAnalysisTaskGammaConversion();\r
  AliAnalysisTaskGammaConversion(const char* name);\r
  virtual ~AliAnalysisTaskGammaConversion() ;// virtual destructor\r
 \r
  // Implementation of interface methods\r
  virtual void UserCreateOutputObjects();\r
  virtual void Init();\r
  virtual void LocalInit() {Init();}\r
  virtual void Exec(Option_t *option);\r
  virtual void Terminate(Option_t *option);\r
  virtual void ConnectInputData(Option_t *);\r
	\r
  void ProcessMCData();\r
  void ProcessV0sNoCut();\r
  void ProcessV0s();\r
  void ProcessGammasForNeutralMesonAnalysis();\r
  void SetHistograms(AliGammaConversionHistograms *const histograms){fHistograms=histograms;}\r
  void SetDoMCTruth(Bool_t flag){fDoMCTruth=flag;}\r
  void SetElectronMass(Double_t electronMass){fElectronMass = electronMass;}\r
  void SetGammaMass(Double_t gammaMass){fGammaMass = gammaMass;}\r
  void SetGammaWidth(Double_t gammaWidth){fGammaWidth = gammaWidth;}\r
  void SetPi0Mass(Double_t pi0Mass){fPi0Mass = pi0Mass;}\r
  void SetPi0Width(Double_t pi0Width){fPi0Width = pi0Width;}\r
  void SetEtaMass(Double_t etaMass){fEtaMass = etaMass;}\r
  void SetEtaWidth(Double_t etaWidth){fEtaWidth = etaWidth;}\r
  void SetMinOpeningAngleGhostCut(Double_t ghostCut){fMinOpeningAngleGhostCut = ghostCut;}\r
  void SetV0Reader(AliV0Reader* const reader){fV0Reader=reader;}\r
  void SetCalculateBackground(Bool_t bg){fCalculateBackground=bg;}\r
  void CalculateBackground();\r
  void SetWriteNtuple(Bool_t writeNtuple){fWriteNtuple = writeNtuple;}\r
  void FillNtuple();\r
  Double_t GetMCOpeningAngle(TParticle* const daughter0, TParticle* const daughter1) const;\r
  void CheckV0Efficiency();\r
\r
\r
  //////////////////Chi_c Analysis////////////////////////////\r
  void GetPID(AliESDtrack *track, Stat_t &pid, Stat_t &weight);	\r
  double GetSigmaToVertex(AliESDtrack* t);\r
  void ElectronBackground(TString hBg, vector <TLorentzVector> e);\r
  void FillAngle(TString histoName,vector <TLorentzVector> tlVeNeg, vector <TLorentzVector> tlVePos);\r
  void FillElectronInvMass(TString histoName, vector <TLorentzVector> negativeElectron, \r
	vector <TLorentzVector> positiveElectron);\r
  void FillGammaElectronInvMass(TString histoMass,TString histoDiff,vector <AliKFParticle> fKFGammas,\r
        vector <TLorentzVector> tlVeNeg,vector<TLorentzVector> tlVePos);\r
  void CleanWithAngleCuts(vector <AliESDtrack*> negativeElectrons,\r
	vector <AliESDtrack*> positiveElectrons, vector <AliKFParticle> gammas);\r
  vector <TLorentzVector> GetTLorentzVector(vector <AliESDtrack*> esdTrack);	\r
  void ProcessGammaElectronsForChicAnalysis();\r
  ///////////////////////////////////////////////////////////////\r
\r
\r
\r
 private:\r
  AliAnalysisTaskGammaConversion(const AliAnalysisTaskGammaConversion&); // Not implemented\r
  AliAnalysisTaskGammaConversion& operator=(const AliAnalysisTaskGammaConversion&); // Not implemented\r
\r
  AliV0Reader* fV0Reader; // The V0 reader object \r
\r
  AliStack * fStack; // pointer to the MC particle stack\r
  AliESDEvent* fESDEvent; //pointer to the ESDEvent\r
  TList * fOutputContainer ; // Histogram container\r
\r
  AliGammaConversionHistograms *fHistograms; // Pointer to the histogram handling class\r
\r
  Bool_t fDoMCTruth; // Flag to switch on/off MC truth \r
    \r
  vector<TParticle*> fMCAllGammas; // vector containing all MC gammas\r
  vector<TParticle*> fMCPi0s; //vector containing all MC Pi0s\r
  vector<TParticle*> fMCEtas; //vector containing all MC Etas\r
  vector<TParticle*> fMCGammaChic; //vector containing all MC Chi_c's\r
\r
  vector<AliKFParticle> fKFReconstructedGammas; // vector containing all reconstructed gammas\r
  vector<Bool_t> fIsTrueReconstructedGammas;    // vector containing information if this was a true gamma or not (follows the index of fKFReconstructedGammas)\r
  vector<Int_t> fElectronv1; // vector containing index of electron 1\r
  vector<Int_t> fElectronv2; // vector containing index of electron 2\r
\r
  ///////Chi_c Analysis///////////////////////////\r
  vector<AliESDtrack*> fCurrentEventPosElectron;       // comment here\r
  vector<AliESDtrack*> fPreviousEventPosElectron;      //comment here\r
  vector<AliESDtrack*> fCurrentEventNegElectron;       //comment here\r
  vector<AliESDtrack*> fPreviousEventNegElectron;      //comment here\r
  vector<AliKFParticle> fKFReconstructedGammasCut;     //comment here\r
  vector<TLorentzVector> fPreviousEventTLVNegElectron; //comment here\r
  vector<TLorentzVector> fPreviousEventTLVPosElectron; //comment here\r
  //////////////////////////////////////////////////	\r
\r
  //mass defines\r
  Double_t fElectronMass; //electron mass\r
  Double_t fGammaMass;    //gamma mass\r
  Double_t fPi0Mass;      //pi0mass\r
  Double_t fEtaMass;      //eta mass\r
\r
  // width defines\r
  Double_t fGammaWidth; //gamma width cut\r
  Double_t fPi0Width;   // pi0 width cut\r
  Double_t fEtaWidth;   // eta width cut\r
\r
  Double_t fMinOpeningAngleGhostCut; // minimum angle cut\r
\r
  Bool_t fCalculateBackground; //flag to set backgrount calculation on/off\r
  Bool_t fWriteNtuple;         // flag to set if writing to ntuple on/off\r
  TNtuple *fGammaNtuple;       // Ntuple for gamma values\r
  TNtuple *fNeutralMesonNtuple;// NTuple for mesons\r
\r
  Int_t fTotalNumberOfAddedNtupleEntries; // number of added ntuple entries\r
\r
  ClassDef(AliAnalysisTaskGammaConversion, 3); // Analysis task for gamma conversions\r
};\r
 \r
#endif //ALIANALYSISTASKGAMMA_H\r