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

COMET::ITimeShiftedGasRegionDR::ITimeShiftedGasRegionDR(const IPAIDeltaRay& deltaRay, 
                                                double time_shift)
    : IPAIDeltaRay(deltaRay),
      IVTimeShiftedDeltaRay() 
    {
        OffsetTime(time_shift);
    }

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

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

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