#include <cstdarg>
#include <memory>
#include <string>
#include <iostream>
#include <sstream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <map>

#include <TFile.h>
#include <TString.h>

#include <cometEventLoop.hxx>
#include <IReconEventFunction.hxx>
#include <ICyDetReconEventFunction.hxx>
#include <IHandle.hxx>
#include <IOADatabase.hxx>
#include <IDigit.hxx>
#include <IDigitContainer.hxx>
#include <ICOMETEvent.hxx>
#include <IHitSelection.hxx>
#include <IHit.hxx>
#include <IAlgorithmResult.hxx>
#include <IReconTrack.hxx>
#include <IReconTrackCand.hxx>
#include <HEPUnits.hxx>

/// User code for the event loop.  This can be copied and used as a skeleton
/// for generic user loop programs.
class IUserLoop: public COMET::ICOMETEventLoopFunction, public ICyDetReconEventFunction {
private:

public:
    IUserLoop() {}
    virtual ~IUserLoop() {};

    void Initialize(void) {
//        ICyDetReconEventFunction::Initialize();
    }

    void Finalize(const COMET::ICOMETOutput*) {
        std::cout << "Finalize end" << std::endl;
    }

    bool operator () (COMET::ICOMETEvent& event) {
        BeginOfEvent();
        PrepareHits(event);
        COMET::IHandle<COMET::IAlgorithmResult> fits = event.GetFit("ReconCyDet_1stHitSel");
        if(!fits) return false;
        ProcessAlgorithms((*fits), event);
        if(!fits) return false;
        COMET::IHandle<COMET::IAlgorithmResult> fitResults = event.GetFit("IGenFitResult_IGenFitter_CyDet");
        if(!fitResults){
            std::cout<<"ReconCyDet can not find fit result "<<fitResults<<std::endl;
            return false;
        }
        COMET::IHandle<COMET::IReconObjectContainer> tracks = fitResults->GetResultsContainer("GenFit_TrackContainer_IGenFitter_CyDet");

        if(!tracks) return false;

        for(COMET::IReconObjectContainer::iterator reconIter = tracks->begin(); reconIter != tracks->end(); reconIter++){
            COMET::IHandle<COMET::IReconTrack> track = *reconIter;

            std::cout << "\t\t" << track->GetName() << " Track Momentum = "  << track->GetMomentum() << " MeV." << std::endl;
        }
        EndOfEvent();
        return true;
    }
};

int main(int argc, char **argv) {
    IUserLoop userLoop;
    cometEventLoop(argc, argv, userLoop);
    return 0;
}