// $Id$
//
//    File: DParticle_factory_HDParSim.cc
// Created: Tue Feb  3 11:27:46 EST 2009
// Creator: davidl (on Darwin harriet.jlab.org 9.6.0 i386)
//


#include <iostream>
#include <iomanip>
using namespace std;

#include <JANA/JApplication.h>

#include <PID/DParticle.h>
#include <TRACKING/DMCThrown.h>

#include "DFactoryGeneratorHDParSim.h"
#include "DTrackingResolutionGEANT.h"
#include "DParticle_factory_HDParSim.h"
using namespace jana;

// Routine used If we're a plugin
extern "C"{
void InitPlugin(JApplication *app){
	InitJANAPlugin(app);
	app->AddFactoryGenerator(new DFactoryGeneratorHDParSim());
}
} // "C"


//------------------
// init
//------------------
jerror_t DParticle_factory_HDParSim::init(void)
{
	res = new DTrackingResolutionGEANT();

	return NOERROR;
}

//------------------
// brun
//------------------
jerror_t DParticle_factory_HDParSim::brun(jana::JEventLoop *eventLoop, int runnumber)
{
	return NOERROR;
}

//------------------
// evnt
//------------------
jerror_t DParticle_factory_HDParSim::evnt(JEventLoop *loop, int eventnumber)
{
	// The simplest way to do this is to get the list of DParticle
	// objects made from the DParticle:THROWN factory and copy those
	// into our own DParticle objects, but with smeared values.
	vector<const DParticle*> particles_thrn;
	loop->Get(particles_thrn, "THROWN");
	
	for(unsigned int i=0; i<particles_thrn.size(); i++){
		
		// Create our own DParticle and copy thrown values into it.
		// If it turns out this track
		// is lost due to inefficiency/acceptance, then the object will be
		// deleted below.
		DParticle *part = new DParticle;
		*part = *(particles_thrn[i]);
		
		// Associated objects are not copied by default so we do them "by hand"
		vector<const JObject*> assoc_objs;
		particles_thrn[i]->GetT(assoc_objs);
		for(unsigned int j=0; j<assoc_objs.size(); j++)part->AddAssociatedObject(assoc_objs[j]);

		// Get the GEANT particle type. The DMCThrown object used to create
		// this DParticle contains this info and a pointer to it should
		// be kept as an associated object.
		vector<const DMCThrown*> throwns;
		part->Get(throwns);
		if(throwns.size()!=1){
			_DBG_<<"No associated DMCThrown object with DParticle object obtained"<<endl;
			_DBG_<<"from DParticle:THROWN factory!"<<endl;
			delete part;
			continue;
		}
		const DMCThrown *thrown = throwns[0];
		
		// Simultaneously smear the momentum of the particle and test whether
		// it passes the efficiency/acceptance cut.
		DVector3 mom = part->momentum();
		if(res->Smear(thrown->type, mom)){
			part->setMomentum(mom);
			_data.push_back(part);
		}else{
			delete part;
		}
	}

	return NOERROR;
}

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

//------------------
// fini
//------------------
jerror_t DParticle_factory_HDParSim::fini(void)
{
	return NOERROR;
}