CKFTracking.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2022 Whitney Armstrong, Wouter Deconinck
 
#ifndef JUGGLER_JUGRECO_CKFTracking_HH
#define JUGGLER_JUGRECO_CKFTracking_HH
 
#include <functional>
#include <random>
#include <stdexcept>
#include <vector>
 
#include "Gaudi/Property.h"
#include "GaudiAlg/GaudiAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "JugBase/DataHandle.h"
#include "JugBase/IGeoSvc.h"
#include "JugBase/BField/DD4hepBField.h"
#include "JugTrack/GeometryContainers.hpp"
#include "JugTrack/Index.hpp"
#include "JugTrack/IndexSourceLink.hpp"
#include "JugTrack/Measurement.hpp"
#include "JugTrack/Track.hpp"
#include "JugTrack/Trajectories.hpp"
 
#include "eicd/TrackerHitCollection.h"
 
#include "Acts/Definitions/Common.hpp"
#include "Acts/Geometry/TrackingGeometry.hpp"
#include "Acts/TrackFinding/CombinatorialKalmanFilter.hpp"
#include "Acts/TrackFinding/MeasurementSelector.hpp"
 
namespace Jug::Reco {
 
/** Fitting algorithm implmentation .
 *
 * \ingroup tracking
 */
class CKFTracking : public GaudiAlgorithm {
public:
  /// Track finder function that takes input measurements, initial trackstate
  /// and track finder options and returns some track-finder-specific result.
  using TrackFinderOptions  = Acts::CombinatorialKalmanFilterOptions<IndexSourceLinkAccessor::Iterator>;
  using TrackFinderResult   = std::vector<Acts::Result<Acts::CombinatorialKalmanFilterResult>>;
 
  /// Find function that takes the above parameters
  /// @note This is separated into a virtual interface to keep compilation units
  /// small
  class CKFTrackingFunction {
   public:
    virtual ~CKFTrackingFunction() = default;
    virtual TrackFinderResult operator()(const TrackParametersContainer&,
                                         const TrackFinderOptions&) const = 0;
  };
 
  /// Create the track finder function implementation.
  /// The magnetic field is intentionally given by-value since the variant
  /// contains shared_ptr anyways.
  static std::shared_ptr<CKFTrackingFunction> makeCKFTrackingFunction(
    std::shared_ptr<const Acts::TrackingGeometry> trackingGeometry,
    std::shared_ptr<const Acts::MagneticFieldProvider> magneticField);
 
public:
  DataHandle<IndexSourceLinkContainer> m_inputSourceLinks{"inputSourceLinks", Gaudi::DataHandle::Reader, this};
  DataHandle<MeasurementContainer> m_inputMeasurements{"inputMeasurements", Gaudi::DataHandle::Reader, this};
  DataHandle<TrackParametersContainer> m_inputInitialTrackParameters{"inputInitialTrackParameters",
                                                                     Gaudi::DataHandle::Reader, this};
  DataHandle<TrajectoriesContainer> m_outputTrajectories{"outputTrajectories", Gaudi::DataHandle::Writer, this};
 
  Gaudi::Property<std::vector<double>> m_etaBins{this, "etaBins", {}};
  Gaudi::Property<std::vector<double>> m_chi2CutOff{this, "chi2CutOff", {15.}};
  Gaudi::Property<std::vector<size_t>> m_numMeasurementsCutOff{this, "numMeasurementsCutOff", {10}};
 
  std::shared_ptr<CKFTrackingFunction> m_trackFinderFunc;
  SmartIF<IGeoSvc> m_geoSvc;
 
  std::shared_ptr<const Jug::BField::DD4hepBField> m_BField = nullptr;
  Acts::GeometryContext m_geoctx;
  Acts::CalibrationContext m_calibctx;
  Acts::MagneticFieldContext m_fieldctx;
 
  Acts::MeasurementSelector::Config m_sourcelinkSelectorCfg;
  Acts::Logging::Level m_actsLoggingLevel = Acts::Logging::INFO;
 
  CKFTracking(const std::string& name, ISvcLocator* svcLoc);
 
  StatusCode initialize() override;
 
  StatusCode execute() override;
};
 
} // namespace Jug::Reco
 
#endif