// PMTTimeJitterProc.cc
// Contact person: James Sinclair <j.sinclair@sussex.ac.uk> - Author
// See PMTTimeJitterProc.hh for more details
//———————————————————————//

#include <RAT/PMTTimeJitterProc.hh>
#include <RAT/DS/EV.hh>
#include <G4UImessenger.hh>
#include <TRandom3.h>
#include <iostream>
#include <sstream>

#include <RAT/DU/PMTInfo.hh>
#include <RAT/DS/Entry.hh>
#include <RAT/DU/Utility.hh>

using namespace std;

using namespace RAT::DS;

namespace RAT {

    PMTTimeJitterProc::PMTTimeJitterProc() : Processor("PMTjitter")
    {
        fnHitPMT = 0;
        fnPMT = 0;
        fPMTid = 0;
        fSigma = 0;
        fJitter = 0;
        fPMTtime = 0;
        fFirstEvent = true;
    }

    PMTTimeJitterProc::~PMTTimeJitterProc()
    {
        // Do nothing
    }

    void PMTTimeJitterProc::BeginOfRun( DS::Run& ){
        fnPMT = DU::Utility::Get()->GetPMTInfo().GetCount();

        TRandom3 rand;
        rand.SetSeed(7);

        for (int i=0;i<fnPMT;++i){
            fPMTid = i+1;
            fJitter = rand.Gaus(0,fSigma);
            fPMTMap[fPMTid] = fJitter;
        }
    }
    void PMTTimeJitterProc::SetI(const std::string& param, const int value){
        if (param == "sigma"){
            if (value > 0){
                fSigma = value;
                cerr << "Sigma: "<< fSigma << endl;
            }
            else{
                throw ParamInvalid(param, "sigma must be > 0");
            }
        }
        else{
            throw ParamUnknown(param);
        }
    }
    void PMTTimeJitterProc::SetD(const std::string& param, const double value){
        if (param == "sigma"){
            if (value > 0){
                fSigma = value;
                cerr << "Sigma: "<< fSigma << endl;
            }
            else{
                throw ParamInvalid(param, "sigma must be > 0");
            }
        }
        else{
            throw ParamUnknown(param);
        }
    }
    Processor::Result PMTTimeJitterProc::DSEvent( DS::Run&, DS::Entry& ds ){
        for(size_t iev = 0; iev < ds.GetEVCount(); iev++){
            RAT::DS::EV& ev = ds.GetEV(iev);
            fnHitPMT = ev.GetCalPMTs().GetCount();

            for (int i=0;i<fnHitPMT;++i){
                fPMTid = ev.GetCalPMTs().GetPMT(i).GetID();
                fPMTtime = ev.GetCalPMTs().GetPMT(i).GetTime() + fPMTMap[fPMTid];
                ev.GetCalPMTs().GetPMT(i).SetTime(fPMTtime);
            }
            return OK;
        }
    }
} // namespace RAT