// $Id$
//
//    File: DReaction_factory_Thrown.cc
// Created: Tue Aug  9 14:29:24 EST 2011
// Creator: pmatt
//

#include "DReaction_factory_Thrown.h"

//------------------
// init
//------------------
jerror_t DReaction_factory_Thrown::init(void)
{
	MAX_dReactionStepPoolSize = 5;

	return NOERROR;
}

//------------------
// brun
//------------------
jerror_t DReaction_factory_Thrown::brun(jana::JEventLoop *locEventLoop, int runnumber)
{
 	vector<const DAnalysisUtilities*> locAnalysisUtilitiesVector;
	locEventLoop->Get(locAnalysisUtilitiesVector);
	dAnalysisUtilities = locAnalysisUtilitiesVector[0];
	return NOERROR;
}

//------------------
// evnt
//------------------
jerror_t DReaction_factory_Thrown::evnt(jana::JEventLoop *locEventLoop, int eventnumber)
{
	// delete pool sizes if too large, preventing memory-leakage-like behavor.
	if(dReactionStepPool_All.size() > MAX_dReactionStepPoolSize){
		for(size_t loc_i = MAX_dReactionStepPoolSize; loc_i < dReactionStepPool_All.size(); ++loc_i)
			delete dReactionStepPool_All[loc_i];
		dReactionStepPool_All.resize(MAX_dReactionStepPoolSize);
	}
	dReactionStepPool_Available = dReactionStepPool_All;

 	vector<const DMCReaction*> locMCReactions;
	locEventLoop->Get(locMCReactions);

	deque<pair<const DMCThrown*, deque<const DMCThrown*> > > locThrownSteps;
	dAnalysisUtilities->Get_ThrownParticleSteps(locEventLoop, locThrownSteps);

	DReaction* locReaction = new DReaction("Thrown");
	DReactionStep* locReactionStep = Get_ReactionStepResource();

	if(!locMCReactions.empty())
	{
		locReactionStep->Set_InitialParticleID(locMCReactions[0]->beam.PID());
		locReactionStep->Set_TargetParticleID(locMCReactions[0]->target.PID());
	}
	else //guess
	{
		locReactionStep->Set_InitialParticleID(Gamma);
		locReactionStep->Set_TargetParticleID(Proton);
	}

	for(size_t loc_i = 0; loc_i < locThrownSteps.size(); ++loc_i)
	{
		if(loc_i != 0) //else beam & target already set
		{
			locReactionStep = Get_ReactionStepResource();
			locReactionStep->Set_InitialParticleID(locThrownSteps[loc_i].first->PID());
			locReactionStep->Set_TargetParticleID(Unknown); //default (disabled)
		}
		for(size_t loc_j = 0; loc_j < locThrownSteps[loc_i].second.size(); ++loc_j)
			locReactionStep->Add_FinalParticleID(locThrownSteps[loc_i].second[loc_j]->PID());
		locReaction->Add_ReactionStep(locReactionStep);
	}

	_data.push_back(locReaction);

	return NOERROR;
}

DReactionStep* DReaction_factory_Thrown::Get_ReactionStepResource(void)
{
	DReactionStep* locReactionStep;
	if(dReactionStepPool_Available.empty())
	{
		locReactionStep = new DReactionStep();
		dReactionStepPool_All.push_back(locReactionStep);
	}
	else
	{
		locReactionStep = dReactionStepPool_Available.back();
		locReactionStep->Reset();
		dReactionStepPool_Available.pop_back();
	}
	return locReactionStep;
}

//------------------
// erun
//------------------
jerror_t DReaction_factory_Thrown::erun(void)
{
	return NOERROR;
}

//------------------
// fini
//------------------
jerror_t DReaction_factory_Thrown::fini(void)
{
	for(size_t loc_i = 0; loc_i < dReactionStepPool_All.size(); ++loc_i)
		delete dReactionStepPool_All[loc_i];
	return NOERROR;
}