// $Id$
//
//    File: DCustomAction_taggerCoincidence.cc
// Created: Thu Jan 22 08:06:18 EST 2015
// Creator: jrsteven (on Linux ifarm1401 2.6.32-431.el6.x86_64 x86_64)
//

#include "DCustomAction_taggerCoincidence.h"

void DCustomAction_taggerCoincidence::Initialize(JEventLoop* locEventLoop)
{

	// get PID algos
	const DParticleID* locParticleID = NULL;
        locEventLoop->GetSingle(locParticleID);
	dParticleID = locParticleID;

	japp->RootWriteLock(); //ACQUIRE ROOT LOCK!!
	{
		//Required: Create a folder in the ROOT output file that will contain all of the output ROOT objects (if any) for this action.
			//If another thread has already created the folder, it just changes to it. 
		CreateAndChangeTo_ActionDirectory();

		if(gDirectory->Get("Match_E_DeltaT") == NULL) //check to see if already created by another thread
			dMatch_E_DeltaT = new TH2D("Match_E_DeltaT", "Match SC-TAGGER: Energy vs #Delta t; #Delta t; Energy", 100, -50, 50, 160, 2., 10.);
		else //already created by another thread
			dMatch_E_DeltaT = static_cast<TH2D*>(gDirectory->Get("Match_E_DeltaT"));
	}
	japp->RootUnLock(); //RELEASE ROOT LOCK!!
}

bool DCustomAction_taggerCoincidence::Perform_Action(JEventLoop* locEventLoop, const DParticleCombo* locParticleCombo)
{
	const DDetectorMatches* locDetectorMatches = NULL;
        locEventLoop->GetSingle(locDetectorMatches);
		
	const DParticleComboStep* locParticleComboStep = locParticleCombo->Get_ParticleComboStep(1);
	if(locParticleComboStep->Get_InitialParticleID() != omega)
		return false;

	// get beam photon time
	const DKinematicData* locBeamPhoton = locParticleCombo->Get_ParticleComboStep(0)->Get_InitialParticle();
	double locBeamPhotonTime = locBeamPhoton->time();
	double locBeamPhotonEnergy = locBeamPhoton->energy();

	// get final state particles
	deque<const DKinematicData*> locParticles;
	if(!Get_UseKinFitResultsFlag()) //measured
		locParticleComboStep->Get_FinalParticles_Measured(locParticles);
	else
		locParticleComboStep->Get_FinalParticles(locParticles);

	// loop final state particles
	for(size_t loc_i = 0; loc_i < locParticles.size(); ++loc_i) {
		if(locParticleComboStep->Get_FinalParticleID(loc_i) != PiPlus && locParticleComboStep->Get_FinalParticleID(loc_i) != PiMinus) continue;

		// get time based track
		const DChargedTrack* locChargedTrack = static_cast<const DChargedTrack*>(locParticleComboStep->Get_FinalParticle_SourceObject(loc_i));
		//if(locChargedTrack == NULL) continue; // skip pi0's in omega decay?

		const DChargedTrackHypothesis* locChargedTrackHypothesis = locChargedTrack->Get_BestFOM();
		const DTrackTimeBased* locTrackTimeBased = NULL;
		locChargedTrackHypothesis->GetSingleT(locTrackTimeBased);
		
		// get match to SC
		DSCHitMatchParams locSCHitMatchParams;
		bool foundSC = dParticleID->Get_BestSCMatchParams(locTrackTimeBased, locDetectorMatches, locSCHitMatchParams);
                if(foundSC){
                        double locSCTime = locSCHitMatchParams.dHitTime - locSCHitMatchParams.dFlightTime;
			double locDeltaT = locSCTime - locBeamPhotonTime;
			
			japp->RootWriteLock(); //ACQUIRE ROOT LOCK!!
			{
				dMatch_E_DeltaT->Fill(locDeltaT, locBeamPhotonEnergy);
			}
			japp->RootUnLock(); //RELEASE ROOT LOCK!!

			// reject out of time beam photons
			if(fabs(locDeltaT) < dDeltaTMax) {
				return true;
			}
		}
	}
	
	// didn't find SC match so reject combo
	return false; //return false if you want to use this action to apply a cut (and it fails the cut!)
}