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

COMET::ITimeShiftedCherenHit::ITimeShiftedCherenHit(const IG4HitCheren& hit, 
                                                    double time_shift)
    : IG4HitCheren(hit),
      IVTimeShiftedHit() 
    {
        OffsetTime(time_shift);
    }

void COMET::ITimeShiftedCherenHit::OffsetTime(double time_shift){
    // Offset the timing by the needed time shift
    fStartPos.vT += time_shift;
    fStopPos.vT += time_shift;
}

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

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