fix: full chain vertexing

Examples/Algorithms/TrackFinding/include/ActsExamples/TrackFinding/TrackFindingAlgorithm.hpp

@@ -58,6 +58,8 @@ class TrackFindingAlgorithm final : public BareAlgorithm {
     std::string inputInitialTrackParameters;
     /// Output find trajectories collection.
     std::string outputTrajectories;
+    /// Output track parameters collection.
+    std::string outputTrackParameters;
     /// Type erased track finder function.
     std::shared_ptr<TrackFinderFunction> findTracks;
     /// CKF measurement selector config

Examples/Algorithms/TrackFinding/src/TrackFindingAlgorithm.cpp

@@ -51,6 +51,9 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
   TrajectoriesContainer trajectories;
   trajectories.reserve(initialParameters.size());
 
+  // Prepare the output data with TrackParameters
+  TrackParametersContainer trackParametersContainer;
+
   // Construct a perigee surface as the target surface
   auto pSurface = Acts::Surface::makeShared<Acts::PerigeeSurface>(
       Acts::Vector3{0., 0., 0.});
@@ -104,6 +107,13 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
           std::move(trackFindingOutput.fittedStates),
           std::move(trackFindingOutput.lastMeasurementIndices),
           std::move(trackFindingOutput.fittedParameters));
+
+      const auto& traj = trajectories.back();
+      for (const auto tip : traj.tips()) {
+        if (traj.hasTrackParameters(tip)) {
+          trackParametersContainer.push_back(traj.trackParameters(tip));
+        }
+      }
     } else {
       ACTS_WARNING("Track finding failed for seed " << iseed << " with error"
                                                     << result.error());
@@ -117,5 +127,7 @@ ActsExamples::ProcessCode ActsExamples::TrackFindingAlgorithm::execute(
                                              << " track candidates.");
 
   ctx.eventStore.add(m_cfg.outputTrajectories, std::move(trajectories));
+  ctx.eventStore.add(m_cfg.outputTrackParameters,
+                     std::move(trackParametersContainer));
   return ActsExamples::ProcessCode::SUCCESS;
 }

Examples/Algorithms/Vertexing/src/AdaptiveMultiVertexFinderAlgorithm.cpp

@@ -129,11 +129,11 @@ ActsExamples::AdaptiveMultiVertexFinderAlgorithm::execute(
     // find vertices and measure elapsed time
     auto result = finder.find(inputTrackPointers, finderOpts, state);
 
-    if (not result.ok()) {
+    if (result.ok()) {
+      vertices = std::move(result.value());
+    } else {
       ACTS_ERROR("Error in vertex finder: " << result.error().message());
-      return ProcessCode::ABORT;
     }
-    vertices = *result;
   }
 
   auto t2 = std::chrono::high_resolution_clock::now();

Examples/Algorithms/Vertexing/src/IterativeVertexFinderAlgorithm.cpp

@@ -106,11 +106,12 @@ ActsExamples::ProcessCode ActsExamples::IterativeVertexFinderAlgorithm::execute(
   auto result = finder.find(inputTrackPointers, finderOpts, state);
   auto t2 = std::chrono::high_resolution_clock::now();
 
-  if (not result.ok()) {
+  std::vector<Acts::Vertex<Acts::BoundTrackParameters>> vertices;
+  if (result.ok()) {
+    vertices = std::move(result.value());
+  } else {
     ACTS_ERROR("Error in vertex finder: " << result.error().message());
-    return ProcessCode::ABORT;
   }
-  auto vertices = *result;
 
   // show some debug output
   ACTS_INFO("Found " << vertices.size() << " vertices in event");

Examples/Algorithms/Vertexing/src/VertexFitterAlgorithm.cpp

@@ -88,6 +88,11 @@ ActsExamples::ProcessCode ActsExamples::VertexFitterAlgorithm::execute(
     inputTrackPtrCollection.clear();
     inputTrackPtrCollection.reserve(protoVertex.size());
     for (const auto& trackIdx : protoVertex) {
+      if (trackIdx >= trackParameters.size()) {
+        ACTS_ERROR("track parameters " << trackIdx << " does not exist");
+        continue;
+      }
+
       inputTrackPtrCollection.push_back(&trackParameters[trackIdx]);
     }
 
@@ -123,10 +128,14 @@ ActsExamples::ProcessCode ActsExamples::VertexFitterAlgorithm::execute(
       }
     }
 
-    ACTS_DEBUG("Fitted Vertex "
-               << fittedVertices.back().fullPosition().transpose());
-    ACTS_DEBUG(
-        "Tracks at fitted Vertex: " << fittedVertices.back().tracks().size());
+    if (fittedVertices.size() > 0) {
+      ACTS_DEBUG("Fitted Vertex "
+                 << fittedVertices.back().fullPosition().transpose());
+      ACTS_DEBUG(
+          "Tracks at fitted Vertex: " << fittedVertices.back().tracks().size());
+    } else {
+      ACTS_DEBUG("No fitted vertex");
+    }
   }
 
   ctx.eventStore.add(m_cfg.outputVertices, std::move(fittedVertices));

Examples/Framework/src/Framework/Sequencer.cpp

@@ -307,6 +307,7 @@ int ActsExamples::Sequencer::run() {
               StopWatch sw(localClocksAlgorithms[ialgo++]);
               ACTS_VERBOSE("Execute algorithm: " << alg->name());
               if (alg->execute(++context) != ProcessCode::SUCCESS) {
+                ACTS_ERROR("Failed to execute algorithm: " << alg->name());
                 throw std::runtime_error("Failed to process event data");
               }
             }

Examples/Python/src/TrackFinding.cpp

@@ -314,6 +314,7 @@ void addTrackFinding(Context& ctx) {
     ACTS_PYTHON_MEMBER(inputSourceLinks);
     ACTS_PYTHON_MEMBER(inputInitialTrackParameters);
     ACTS_PYTHON_MEMBER(outputTrajectories);
+    ACTS_PYTHON_MEMBER(outputTrackParameters);
     ACTS_PYTHON_MEMBER(findTracks);
     ACTS_PYTHON_MEMBER(measurementSelectorCfg);
     ACTS_PYTHON_STRUCT_END();

Examples/Python/tests/test_examples.py

@@ -1021,5 +1021,5 @@ def test_full_chain_odd_example(tmp_path):
     assert script.exists()
     env = os.environ.copy()
     env["NEVENTS"] = "1"
-    env["ACTS_LOG_FAILURE_THRESHOLD"] = "WARNING"
+    env["ACTS_LOG_FAILURE_THRESHOLD"] = "FATAL"
     subprocess.check_call([str(script)], cwd=tmp_path, env=env)

Examples/Scripts/Python/ckf_tracks.py

@@ -62,6 +62,7 @@ def addCKFTracks(
         inputSourceLinks="sourcelinks",
         inputInitialTrackParameters="estimatedparameters",
         outputTrajectories="trajectories",
+        outputTrackParameters="fittedTrackParameters",
         findTracks=acts.examples.TrackFindingAlgorithm.makeTrackFinderFunction(
             trackingGeometry, field
         ),

Examples/Scripts/Python/full_chain_odd.py

@@ -33,7 +33,7 @@
     s,
     MomentumConfig(1.0 * u.GeV, 10.0 * u.GeV, True),
     EtaConfig(-3.0, 3.0, True),
-    ParticleConfig(1, acts.PdgParticle.eMuon, True),
+    ParticleConfig(2, acts.PdgParticle.eMuon, True),
     rnd=rnd,
 )
 s = addFatras(
@@ -67,11 +67,24 @@
     CKFPerformanceConfig(ptMin=400.0 * u.MeV, nMeasurementsMin=6),
     outputDirRoot=outputDir,
 )
+s.addAlgorithm(
+    acts.examples.TrackSelector(
+        level=acts.logging.INFO,
+        inputTrackParameters="fittedTrackParameters",
+        outputTrackParameters="trackparameters",
+        outputTrackIndices="outputTrackIndices",
+        removeNeutral=True,
+        absEtaMax=2.5,
+        loc0Max=4.0 * u.mm,  # rho max
+        ptMin=500 * u.MeV,
+    )
+)
 s = addVertexFitting(
     s,
     field,
-    vertexFinder=VertexFinder.Truth,
+    vertexFinder=VertexFinder.Iterative,
     outputDirRoot=outputDir,
+    logLevel=acts.logging.VERBOSE,
 )
 
 s.run()

Examples/Scripts/Python/vertex_fitting.py

@@ -37,7 +37,7 @@ def addVertexFitting(
     field,
     outputDirRoot: Optional[Union[Path, str]] = None,
     associatedParticles: str = "particles_input",
-    trackParameters: str = "estimatedparameters",
+    trackParameters: str = "trackparameters",
     vertexFinder: VertexFinder = VertexFinder.Truth,
     logLevel: Optional[acts.logging.Level] = None,
 ):
@@ -88,7 +88,6 @@ def customLogLevel(custom: acts.logging.Level = acts.logging.INFO):
             outputVertices=outputVertices,
         )
         s.addAlgorithm(fitVertices)
-
     elif vertexFinder == VertexFinder.Iterative:
         findVertices = IterativeVertexFinderAlgorithm(
             level=customLogLevel(),
@@ -108,7 +107,6 @@ def customLogLevel(custom: acts.logging.Level = acts.logging.INFO):
             outputVertices=outputVertices,
             outputTime=outputTime,
         )
-
         s.addAlgorithm(findVertices)
     else:
         raise RuntimeError("Invalid finder argument")