Updated with SIScone

[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
#include "TNtuple.h"\r
\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 *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* 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* daughter0, TParticle* daughter1) const;\r
\r
 private:\r
  AliAnalysisTaskGammaConversion(const AliAnalysisTaskGammaConversion&); // Not implemented\r
  AliAnalysisTaskGammaConversion& operator=(const AliAnalysisTaskGammaConversion&); // Not implemented\r
\r
  AliV0Reader* fV0Reader;\r
\r
  AliStack * fStack;\r
\r
  TList * fOutputContainer ; // Histogram container\r
\r
  AliGammaConversionHistograms *fHistograms;\r
\r
  Bool_t fDoMCTruth;\r
    \r
  vector<TParticle*> fMCAllGammas;\r
  vector<TParticle*> fMCPi0s;\r
  vector<TParticle*> fMCEtas;\r
  vector<TParticle*> fMCGammaChic;\r
\r
  vector<AliKFParticle> fKFReconstructedGammas;\r
  vector<Bool_t> fIsTrueReconstructedGammas;\r
  vector<Int_t> electronv1;\r
  vector<Int_t> electronv2;\r
\r
  //mass defines\r
  Double_t fElectronMass;\r
  Double_t fGammaMass;\r
  Double_t fPi0Mass;\r
  Double_t fEtaMass;\r
\r
  // width defines\r
  Double_t fGammaWidth;\r
  Double_t fPi0Width;\r
  Double_t fEtaWidth;\r
\r
  Double_t fMinOpeningAngleGhostCut;\r
\r
  Bool_t fCalculateBackground;\r
  Bool_t fWriteNtuple;\r
  TNtuple *fGammaNtuple;\r
  TNtuple *fNeutralMesonNtuple;\r
\r
  Int_t fTotalNumberOfAddedNtupleEntries;\r
\r
  ClassDef(AliAnalysisTaskGammaConversion, 2); // Analysis task for gamma conversions\r
};\r
 \r
#endif //ALIANALYSISTASKGAMMA_H\r