CKF Tracking

JeffersonLab · Nov 19, 2024 · aee1808 · aee1808
1 parent 2a501db
commit aee1808
Show file tree

Hide file tree

Showing 6 changed files with 589 additions and 57 deletions.
diff --git a/source/tdis/CMakeLists.txt b/source/tdis/CMakeLists.txt
@@ -50,7 +50,8 @@ add_executable(tdis
         # tracking/Measurement2DFactory.h
         tracking/TruthTrackParameterFactory.h
         tracking/KalmanFittingFactory.h
-
+        tracking/CKFTracking.h
+        tracking/CKFTracking.cc
 )
 
 # ---------- FIND REQUIRED PACKAGES -------------

diff --git a/source/tdis/io/PodioWriteProcessor.hpp b/source/tdis/io/PodioWriteProcessor.hpp
@@ -75,7 +75,9 @@ class PodioWriteProcessor : public JEventProcessor {
         "DigitizedMtpcMcHit",
 
         // Digitized hits
-        "TrackerHit"
+        "TrackerHit",
+        "Measurement2D",
+        "TruthTrackInitParameters",
     };
 
   PodioWriteProcessor(JApplication * app);

diff --git a/source/tdis/tdis_main.cpp b/source/tdis/tdis_main.cpp
@@ -9,9 +9,9 @@
 // #include <podio/ROOTFrameWriter.h>
 // #include <podio/ROOTFrameReader.h>
 
+#include <JANA/Components/JOmniFactoryGeneratorT.h>
 #include <JANA/JApplication.h>
 #include <JANA/Services/JParameterManager.h>
-#include <JANA/Components/JOmniFactoryGeneratorT.h>
 
 #include <utility>
 
@@ -21,6 +21,7 @@
 #include "services/LogService.hpp"
 #include "tracking/ActsGeometryService.h"
 #include "tracking/ReconstructedHitFactory.h"
+#include "tracking/TruthTrackParameterFactory.h"
 // #include "tracking/Measurement2DFactory.h"
 
 struct ProgramArguments {
@@ -123,6 +124,13 @@ int main(int argc, char* argv[]) {
         {"TrackerHit", "Measurement2D"});
     app.Add(reco_hit_generator);
 
+    auto truth_track_init_generator = new JOmniFactoryGeneratorT<tdis::tracking::TruthTrackParameterFactory>();
+    truth_track_init_generator->AddWiring(
+        "TruthTrackParameterGenerator",
+        {"DigitizedMtpcMcTrack"},
+        {"TruthTrackInitParameters"});
+    app.Add(truth_track_init_generator);
+
     // auto measurement_2d_generator = new JOmniFactoryGeneratorT<tdis::tracking::Measurement2DFactory>();
     // measurement_2d_generator->AddWiring("TrackerHitGenerator", {"TrackerHit"}, {"Measurement2D"});
     // app.Add(measurement_2d_generator);