#ifndef DSelector_kpi0sigma0_h
#define DSelector_kpi0sigma0_h

#include <iostream>

#include "DSelector/DSelector.h"
#include "DSelector/DHistogramActions.h"
#include "DSelector/DCutActions.h"

#include "TH1I.h"
#include "TH2I.h"

class DSelector_kpi0sigma0 : public DSelector
{
	public:

		DSelector_kpi0sigma0(TTree* locTree = NULL) : DSelector(locTree){}
		virtual ~DSelector_kpi0sigma0(){}

		void Init(TTree *tree);
		Bool_t Process(Long64_t entry);

	private:

		void Get_ComboWrappers(void);
		void Finalize(void);

		// BEAM POLARIZATION INFORMATION
		UInt_t dPreviousRunNumber;
		bool dIsPolarizedFlag; //else is AMO
		bool dIsPARAFlag; //else is PERP or AMO

		//CREATE REACTION-SPECIFIC PARTICLE ARRAYS

		//Step 0
		DParticleComboStep* dStep0Wrapper;
		DBeamParticle* dComboBeamWrapper;
		DChargedTrackHypothesis* dKPlusWrapper;

		//Step 1
		DParticleComboStep* dStep1Wrapper;
		DKinematicData* dDecayingPi0Wrapper;
		DNeutralParticleHypothesis* dPhoton1Wrapper;
		DNeutralParticleHypothesis* dPhoton2Wrapper;

		//Step 2
		DParticleComboStep* dStep2Wrapper;
		DKinematicData* dDecayingSigma0Wrapper;
		DNeutralParticleHypothesis* dPhoton3Wrapper;

		//Step 3
		DParticleComboStep* dStep3Wrapper;
		DKinematicData* dDecayingLambdaWrapper;
		DChargedTrackHypothesis* dProtonWrapper;
		DChargedTrackHypothesis* dPiMinusWrapper;

		// DEFINE YOUR HISTOGRAMS HERE
		TH1I* dHist_NumSurvivingBeamParticles;

		// BEST COMBO
		TH1I* dHist_MandelstamT;
		TH1I* dHist_KStarMass_BestCombo;
		TH2I* dHist_KStarMassVsT;

		//BEST COMBO: POLARIZATION
		TH1I* dHist_ProdPlanePhi_PARA;
		TH2I* dHist_ProdPlanePhiVsT_Sideband_PARA;
		TH2I* dHist_ProdPlanePhiVsT_Signal_PARA;
		TH1I* dHist_ProdPlanePhi_PERP;
		TH2I* dHist_ProdPlanePhiVsT_Sideband_PERP;
		TH2I* dHist_ProdPlanePhiVsT_Signal_PERP;
		TH1I* dHist_ProdPlanePhi_AMO;
		TH2I* dHist_ProdPlanePhiVsT_Sideband_AMO;
		TH2I* dHist_ProdPlanePhiVsT_Signal_AMO;

		ClassDef(DSelector_kpi0sigma0, 0);
};

void DSelector_kpi0sigma0::Get_ComboWrappers(void)
{
	//Step 0
	dStep0Wrapper = dComboWrapper->Get_ParticleComboStep(0);
	dComboBeamWrapper = static_cast<DBeamParticle*>(dStep0Wrapper->Get_InitialParticle());
	dKPlusWrapper = static_cast<DChargedTrackHypothesis*>(dStep0Wrapper->Get_FinalParticle(0));

	//Step 1
	dStep1Wrapper = dComboWrapper->Get_ParticleComboStep(1);
	dDecayingPi0Wrapper = dStep1Wrapper->Get_InitialParticle();
	dPhoton1Wrapper = static_cast<DNeutralParticleHypothesis*>(dStep1Wrapper->Get_FinalParticle(0));
	dPhoton2Wrapper = static_cast<DNeutralParticleHypothesis*>(dStep1Wrapper->Get_FinalParticle(1));

	//Step 2
	dStep2Wrapper = dComboWrapper->Get_ParticleComboStep(2);
	dDecayingSigma0Wrapper = dStep2Wrapper->Get_InitialParticle();
	dPhoton3Wrapper = static_cast<DNeutralParticleHypothesis*>(dStep2Wrapper->Get_FinalParticle(1));

	//Step 3
	dStep3Wrapper = dComboWrapper->Get_ParticleComboStep(3);
	dDecayingLambdaWrapper = dStep3Wrapper->Get_InitialParticle();
	dProtonWrapper = static_cast<DChargedTrackHypothesis*>(dStep3Wrapper->Get_FinalParticle(0));
	dPiMinusWrapper = static_cast<DChargedTrackHypothesis*>(dStep3Wrapper->Get_FinalParticle(1));
}

#endif // DSelector_kpi0sigma0_h