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

COMET::ITimeShiftedCaloHit::ITimeShiftedCaloHit(const IG4HitCalo& hit, 
                                                double time_shift)
    : IG4HitCalo(hit),
      IVTimeShiftedHit() 
    {
        OffsetTime(time_shift);
    }

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

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

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