diff --git a/src/TMS_TreeWriter.cpp b/src/TMS_TreeWriter.cpp index d312eabb..1c2f1dab 100644 --- a/src/TMS_TreeWriter.cpp +++ b/src/TMS_TreeWriter.cpp @@ -1261,6 +1261,7 @@ void TMS_TreeWriter::Fill(TMS_Event &event) { RecoTrackHitPos[itTrack][j][1] = RecoTrack->Hits[j].GetNotZ(); } RecoTrackHitPos[itTrack][j][2] = RecoTrack->Hits[j].GetZ(); + RecoTrackHitPos[itTrack][j][3] = RecoTrack->Hits[j].GetT(); } // Can manually compute direction if it hasn't been set // if ( (RecoTrackDirection[itTrack][0] == 0) && (RecoTrackDirection[itTrack][1] == 0) && (RecoTrackDirection[itTrack][2] == 0) ) @@ -1688,6 +1689,7 @@ void TMS_TreeWriter::Clear() { RecoTrackHitPos[i][k][0] = DEFAULT_CLEARING_FLOAT; RecoTrackHitPos[i][k][1] = DEFAULT_CLEARING_FLOAT; RecoTrackHitPos[i][k][2] = DEFAULT_CLEARING_FLOAT; + RecoTrackHitPos[i][k][3] = DEFAULT_CLEARING_FLOAT; } RecoTrackEnergyRange[i] = DEFAULT_CLEARING_FLOAT; RecoTrackEnergyDeposit[i] = DEFAULT_CLEARING_FLOAT; diff --git a/src/TMS_TreeWriter.h b/src/TMS_TreeWriter.h index 758923e3..93dfc81c 100644 --- a/src/TMS_TreeWriter.h +++ b/src/TMS_TreeWriter.h @@ -46,9 +46,9 @@ class TMS_TreeWriter { int nTracks; int nHitsIn3DTrack[__TMS_MAX_TRACKS__]; int nKalmanNodes[__TMS_MAX_TRACKS__]; - float RecoTrackKalmanPos[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__][3]; - float RecoTrackKalmanTruePos[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__][3]; - float RecoTrackHitPos[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__][3]; // Due to a lack of variables, but as this is taken from line hits, it would make sense (maybe times 2?) + float RecoTrackKalmanPos[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__][4]; // x,y,z,t + float RecoTrackKalmanTruePos[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__][4]; + float RecoTrackHitPos[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__][4]; // Due to a lack of variables, but as this is taken from line hits, it would make sense (maybe times 2?) float RecoTrackHitEnergies[__TMS_MAX_TRACKS__][__TMS_MAX_LINE_HITS__]; // Due to a lack of variables, but as this is taken from line hits, it would make sense (maybe times 2?) float RecoTrackStartPos[__TMS_MAX_TRACKS__][3]; float RecoTrackStartDirection[__TMS_MAX_TRACKS__][3];