/////////////////////////////////////////////////////////////////
//
// FOR THE DEMONSTRATION
// this class just apply the linear fitting for all track candidates
// Y.Fujii
//
/////////////////////////////////////////////////////////////////


#ifndef TLinearFitter_hxx_seen
#define TLinearFitter_hxx_seen

#include <vector>

#include <TObject.h>
#include <TVector3.h>

#include <IAlgorithm.hxx>
#include <ICOMETLog.hxx>

#include <IVTrackFitter.hxx>

#include <IHandle.hxx>
#include <ICOMETEvent.hxx>
#include <IReconTrack.hxx>
#include <IReconTrackCand.hxx>

#include <TVirtualFitter.h>

class ILinearFitter : public IVTrackFitter {
public:
    ILinearFitter(const char* name, const char* title);
    virtual ~ILinearFitter();

    /// Main task called by `ICOMETReconLoopFunction`
    COMET::IHandle<COMET::IAlgorithmResult> Process(const COMET::IAlgorithmResult& input);

    /// load the event
    virtual int  Load(COMET::ICOMETEvent& anEvent);

    /// save the event
    virtual int  Save(COMET::ICOMETEvent* anEvent);

    /// called at the begin of run or else (should not be in event-by-event)
    virtual int  Init();

    /// called at the end of run or else (should not be in event-by-event)
    virtual int  Finish();

    /// called at the begin of each event
    virtual int  BeginOfEvent();

    /// called at the end of each event
    virtual int  EndOfEvent();

protected:
    TVirtualFitter *fFitter;
};
#endif