#ifndef ALIANALYSISTASKGAMMACONVERSION_H
#define ALIANALYSISTASKGAMMACONVERSION_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
#include "AliAnalysisTaskSE.h"
#include "AliV0Reader.h"
//#include "TH2F.h"
//#include "TH1F.h"
#include "AliGammaConversionHistograms.h"
#include "AliESDv0.h"
#include <vector>
//#include "TLorentzVector.h"
#include "AliKFParticle.h"

//class AliAnaGamma;
class AliESDInputHandler;
class AliESDEvent;
class AliAODEvent;
class TList;
class AliStack;

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 ProcessV0s();
    void ProcessGammasForNeutralMesonAnalysis();
    void SetHistograms(AliGammaConversionHistograms *histograms){fHistograms=histograms;}
    void SetDoMCTruth(Bool_t flag){fDoMCTruth=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 SetV0Reader(AliV0Reader* reader){fV0Reader=reader;}
    void SetCalculateBackground(Bool_t bg){fCalculateBackground=bg;}
    void CalculateBackground();
    Double_t GetMCOpeningAngle(TParticle* daughter0,TParticle* daughter1);

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

    AliV0Reader* fV0Reader;

    AliStack * fStack;

    TList * fOutputContainer ; // Histogram container

    AliGammaConversionHistograms *fHistograms;

    Bool_t fDoMCTruth;
    
    vector<TParticle*> fMCAllGammas;
    vector<TParticle*> fMCPi0s;
    vector<TParticle*> fMCEtas;
    vector<TParticle*> fMCGammaChi_c;

    vector<AliKFParticle> fKFReconstructedGammas;

    //mass defines
    Double_t fElectronMass;
    Double_t fGammaMass;
    Double_t fPi0Mass;
    Double_t fEtaMass;

    // width defines
    Double_t fGammaWidth;
    Double_t fPi0Width;
    Double_t fEtaWidth;
    Bool_t fCalculateBackground;


    ClassDef(AliAnalysisTaskGammaConversion, 0); // Analysis task for gamma conversions
};
 
#endif //ALIANALYSISTASKGAMMA_H