// $Id: DTrackWireBased_factory.h 5569 2009-10-02 22:27:08Z staylor $
//
//    File: DTrackWireBased_factory.h
// Created: Wed Sep  3 09:33:40 EDT 2008
// Creator: davidl (on Darwin harriet.jlab.org 8.11.1 i386)
//

#ifndef _DTrackWireBased_factory_
#define _DTrackWireBased_factory_

#include <JANA/JFactory.h>
#include <JANA/JObject.h>

#include <TRACKING/DTrackFitter.h>
#include <TRACKING/DTrackHitSelector.h>
#include "PID/DParticleID.h"

#include <TH2.h>
#include <TH1.h>

class DTrackCandidate;

#include "DTrackWireBased.h"

///////////////////////////////////////////////////////////////////////
/// The DTrackWireBased_factory class coordinates the fitting of wire-based
/// tracks. While the hit selection is done here, the actual heavy
/// lifting of the fit is done by the DTrackFitter class (or, more
/// specifically, a class that inherits from DTrackFitter).
///
/// This grabs a DTrackFitter object using the default Tag through JANA
/// and uses it to fit the DTrackCandidate objects which it grabs
/// also using the default Tag.
/// 
/// The DTrackWireBased objects are wire-based tracks (no drift time 
/// information is used). As such, this is hardwired to set the fit type for 
/// the DTrackFitter to kWireBased. See the DTrackTimeBased classes for the
/// time-based counterpart.
///
/// This may appear uneccessarily complex, but it provides for using the
/// exact same code for fitting both wire-based and time-based tracks
/// as well as allowing a lot of flexibility in swapping out the 
/// DTrackFitter class used by both the wire-based and time-based 
/// stages using the same DEFTAG mechanism used by the rest of JANA.
///////////////////////////////////////////////////////////////////////

class DTrackWireBased_factory:public jana::JFactory<DTrackWireBased>{
	public:
		DTrackWireBased_factory(){};
		~DTrackWireBased_factory(){};


	private:
		jerror_t init(void);						///< Called once at program start.
		jerror_t brun(jana::JEventLoop *loop, int runnumber);	///< Called everytime a new run number is detected.
		jerror_t evnt(jana::JEventLoop *loop, int eventnumber);	///< Called every event.
		jerror_t erun(void);						///< Called everytime run number changes, provided brun has been called.
		jerror_t fini(void);						///< Called after last event of last event source has been processed.

		int DEBUG_LEVEL;
		DTrackFitter *fitter;
		vector<DReferenceTrajectory*> rtv;
		vector<double> mass_hypotheses_positive;
		vector<double> mass_hypotheses_negative;
		size_t MAX_DReferenceTrajectoryPoolSize;

		void FilterDuplicates(void);
		void DoFit(unsigned int c_id,const DTrackCandidate *candidate,
			   DReferenceTrajectory *rt,jana::JEventLoop *loop, 
			   double mass); 
		const DGeometry *geom;

		bool DEBUG_HISTS;
		bool SKIP_MASS_HYPOTHESES_WIRE_BASED;
		bool USE_HITS_FROM_CANDIDATE;
		double MIN_FIT_P;

		DParticleID* dPIDAlgorithm;
};

#endif // _DTrackWireBased_factory_