Filled tracks now passed correctly to TreeWriter

DUNE · Jan 23, 2024 · ab29c11 · ab29c11
1 parent 6bb9e83
commit ab29c11
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/TMS_TreeWriter.cpp b/src/TMS_TreeWriter.cpp
@@ -120,7 +120,8 @@ void TMS_TreeWriter::MakeBranches() {
   Reco_Tree->Branch("SliceNo", &SliceNo, "SliceNo/I");
   Reco_Tree->Branch("SpillNo", &SpillNo, "SpillNo/I");
 
-  Reco_Tree->Branch("nHits",        &nHitsIn3DTrack,    "nHits[nTracks]/I");
+  Reco_Tree->Branch("nTracks",      &nTracks,               "nTracks/I");
+  Reco_Tree->Branch("nHits",        &nHitsIn3DTrack,        "nHits[nTracks]/I");
   Reco_Tree->Branch("StartPos",     RecoTrackStartPos,      "StartPos[nTracks][3]/F");
   Reco_Tree->Branch("Direction",    RecoTrackDirection,     "Direction[nTracks][3]/F");
   Reco_Tree->Branch("EndPos",       RecoTrackEndPos,        "EndPos[nTracks][3]/F");
@@ -451,7 +452,9 @@ void TMS_TreeWriter::Fill(TMS_Event &event) {
   //TODO: Function here that uses the info from ^^^^^ to fill the 3DTrack objects
 
   int itTrack= 0;
-  std::vector<TMS_Track> Reco_Tracks;
+  std::vector<TMS_Track> Reco_Tracks = GetHoughTrack3D();
+  int nTracks = Reco_Tracks.size();
+
   for (auto RecoTrack = Reco_Tracks.begin(); RecoTrack != Reco_Tracks.end(); ++RecoTrack, ++itTrack) {
     nHitsIn3DTrack[itTrack]         = (int) RecoTrack->Hits.size(); // Do we need to cast it? idk
     RecoTrackEnergy[itTrack]        =       RecoTrack->GetEnergyRange();