#ifndef ALIAODCONVERSIONPHOTON_H
#define ALIAODCONVERSIONPHOTON_H

#include "AliConversionPhotonBase.h"
#include "AliKFConversionPhoton.h"
#include "AliAODConversionParticle.h"

class AliAODConversionPhoton : public AliAODConversionParticle, public AliConversionPhotonBase {

	public: 

		enum caloPhotonMCFlags_t {
			kIsPhoton 				= 0x001, kIsElectron 		= 0x002, kIsConversion 	= 0x004, kIsConversionFullyContained 	= 0x008,
			kIsMerged 				= 0x010, kIsMergedPartConv 	= 0x020, kIsDalitz 		= 0x040, kIsDalitzMerged 				= 0x080,
            kIsPhotonWithElecMother = 0x100, kIsShower 			= 0x200, kIsSubLeadingEM= 0x400
		};
		
		//Constructors
		AliAODConversionPhoton();    
		AliAODConversionPhoton(AliKFConversionPhoton *kfphoton);
		AliAODConversionPhoton(TLorentzVector *vec);

		//Copy Constructor
		AliAODConversionPhoton(const AliAODConversionPhoton & g);           
		//assignment operator
		AliAODConversionPhoton & operator = (const AliAODConversionPhoton & g);

		//Destructor
		virtual ~AliAODConversionPhoton();

		// Overwrite GetLabelFunctions to Make it accessible via AliAODConversionParticle
        virtual Int_t GetLabel(Int_t i) const { return AliConversionPhotonBase::GetTrackLabel(i); }
        virtual Int_t GetLabel1() const { return AliConversionPhotonBase::GetTrackLabelPositive(); }
        virtual Int_t GetLabel2() const { return AliConversionPhotonBase::GetTrackLabelNegative(); }

		virtual Double_t GetPhotonMass() const {return AliAODConversionParticle::M();}
		virtual Double_t GetPhotonPt() const {return AliAODConversionParticle::Pt();}
		virtual Double_t GetPhotonP() const {return AliAODConversionParticle::P();}
		virtual Double_t GetPhotonEta() const {return AliAODConversionParticle::Eta();}
		virtual Double_t GetPhotonTheta() const {return AliAODConversionParticle::Theta();}
		virtual Double_t GetPhotonPhi() const {return AliAODConversionParticle::Phi();}
		virtual Double_t GetPx() const { return AliAODConversionParticle::Px();}
		virtual Double_t GetPy() const { return AliAODConversionParticle::Py();}
		virtual Double_t GetPz() const { return AliAODConversionParticle::Pz();}
		void CalculateDistanceOfClossetApproachToPrimVtx(const AliVVertex* primVertex);
		void SetMassToZero() { SetE(P()); }
		
		void SetIsTrueConvertedPhoton(){
			fCaloPhoton = 0;
			fCaloPhotonMCFlags = 1;	
		}	
		
		Float_t GetDCAzToPrimVtx()const {return fDCAzPrimVtx;}
		Float_t GetDCArToPrimVtx()const {return fDCArPrimVtx;}
		
		void SetIsCaloPhoton(){fCaloPhoton =1;}
		Bool_t GetIsCaloPhoton(){return fCaloPhoton;}
		void SetCaloPhotonMCLabel(Int_t i, Int_t labelCaloPhoton){fCaloPhotonMCLabels[i] = labelCaloPhoton;}
		Int_t GetCaloPhotonMCLabel(Int_t i){return fCaloPhotonMCLabels[i];}
		void SetNCaloPhotonMCLabels(Int_t nLabels){fNCaloPhotonMCLabels = nLabels;}
		Int_t GetNCaloPhotonMCLabels(){return fNCaloPhotonMCLabels;}
        Int_t GetNCaloPhotonMotherMCLabels(){return fNCaloPhotonMotherMCLabels;}
		void SetCaloPhotonMCFlags(AliStack *MCStack);
		void SetCaloPhotonMCFlagsAOD(AliVEvent* event);
		void SetCaloClusterRef(Long_t ref){fCaloClusterRef = ref;}
		Long_t GetCaloClusterRef()const {return fCaloClusterRef;}
		
		//Calo cluster MC identifiers
		Bool_t IsLargestComponentPhoton(){return fCaloPhotonMCFlags&kIsPhoton;} 					// largest contribution to cluster is photon
		Bool_t IsLargestComponentElectron(){return fCaloPhotonMCFlags&kIsElectron;}					// largest contribution to cluster is electron
		Bool_t IsConversion(){return fCaloPhotonMCFlags&kIsConversion;}								// largest contribution to cluster is converted electron
		Bool_t IsConversionFullyContained(){return fCaloPhotonMCFlags&kIsConversionFullyContained;}	// largest contribution to cluster is converted electron & other electron has been found in cluster as well
		Bool_t IsMerged(){return fCaloPhotonMCFlags&kIsMerged;}										// largest contribution to cluster is photon, second photon or electron from dalitz decay is found in cluster as well
		Bool_t IsMergedPartConv(){return fCaloPhotonMCFlags&kIsMergedPartConv;}						// cluster contains more than 1 particle belonging to the same mother particle (i.e. pi0, eta ...) 
																									// & at least one of the decays was a conversion
		Bool_t IsDalitz(){return fCaloPhotonMCFlags&kIsDalitz;}										// cluster contains particle from Dalitz decay
		Bool_t IsDalitzMerged(){return fCaloPhotonMCFlags&kIsDalitzMerged;}							// cluster contains particle from Dalitz decay & more than one particle of this decay is contained in cluster
		Bool_t IsPhotonWithElecMother(){return fCaloPhotonMCFlags&kIsPhotonWithElecMother;}			// largest contribution to cluster is photon which stems from an electron (i.e. radiation)
		Bool_t IsShower(){return fCaloPhotonMCFlags&kIsShower;}										// largest contribution to cluster seems to stem from a shower
		Bool_t IsEMNonLeading(){return !(fCaloPhotonMCFlags&kIsPhoton) && !(fCaloPhotonMCFlags&kIsElectron);} // largest contribution is from hadron
        Bool_t IsSubLeadingEM(){return fCaloPhotonMCFlags&kIsSubLeadingEM;}                         // cluster contains at least one electron or photon from a pi0 or eta in subleading contribution
		
		Bool_t IsTrueConvertedPhoton() {
			if (!fCaloPhoton && fCaloPhotonMCFlags == 1) return kTRUE;
				else return kFALSE;
		}
		
		Float_t fDCArPrimVtx;
		Float_t fDCAzPrimVtx;
		Bool_t fCaloPhoton;
		Long_t fCaloClusterRef;
		Int_t fNCaloPhotonMCLabels;
        Int_t fNCaloPhotonMotherMCLabels;
        Int_t fCaloPhotonMCFlags;
        Int_t fCaloPhotonMCLabels[20];
        Int_t fCaloPhotonMotherMCLabels[20];
		
	
	ClassDef(AliAODConversionPhoton,4)
};


#endif