#include "ITimeShiftedMonHit.hxx"
#include <TLorentzVector.h>

COMET::ITimeShiftedMonHit::ITimeShiftedMonHit(const IG4HitMonitor& hit, 
                                                double time_shift)
    : IG4HitMonitor(hit),
      IVTimeShiftedHit() 
    {
        OffsetTime(time_shift);
    }

void COMET::ITimeShiftedMonHit::OffsetTime(double time_shift){
    // Offset the timing by the needed time shift
    fPosT += time_shift;
}

void COMET::ITimeShiftedMonHit::GetTrackIDs(std::set<int>& oldIDs){
    // Insert the needed primary track ID
    oldIDs.insert(fPrimaryId);
}

void COMET::ITimeShiftedMonHit::SetTrackIDs(const std::map<int,int>& newIDs){
    // Reset the primary ID to the new value
    fPrimaryId=GetId(fPrimaryId,newIDs);
}