HGTD_IterativeExtensionTool.h

Go to the documentation of this file.

 
#ifndef HGTD_ITERATIVEEXTENTSIONTOOL_H
#define HGTD_ITERATIVEEXTENTSIONTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "HGTD_RecToolInterfaces/IHGTD_TrackTimeExtensionTool.h"
 
#include "GaudiKernel/ToolHandle.h"
#include "HGTD_PrepRawData/HGTD_ClusterContainer.h"
#include "HGTD_RecToolInterfaces/IHGTD_ClusterTruthTool.h"
#include "HGTD_RecToolInterfaces/IHGTD_TOFcorrectionTool.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkToolInterfaces/IUpdator.h"
#include "xAODTruth/TruthParticle.h"
 
#include <memory>
 
class HGTD_ID;
class HGTD_DetectorManager;
 
class HGTD_IterativeExtensionTool
    : public extends<AthAlgTool, IHGTD_TrackTimeExtensionTool> {
 
public:
  HGTD_IterativeExtensionTool(const std::string&, const std::string&,
                              const IInterface*);
 
  virtual StatusCode initialize() override final;

  virtual HGTD::ExtensionObject extendTrackToHGTD(
      const EventContext& ctx, const xAOD::TrackParticle& track_ptkl,
      const HGTD_ClusterContainer* container,
      const HepMC::GenEvent* hs_event = nullptr,
      const InDetSimDataCollection* sim_data = nullptr) const override final;
 
private:
  const Trk::TrackStateOnSurface*
  getLastHitOnTrack(const Trk::Track& track) const;

  static std::vector<const Trk::Surface*>
  getCompatibleSurfaces(const Trk::TrackParameters& param,
                        const Trk::Layer* layer);

  std::vector<std::unique_ptr<const Trk::TrackParameters>>
  extrapolateToSurfaces(const EventContext& ctx,
                        const Trk::TrackParameters& param,
                        const std::vector<const Trk::Surface*>& surfaces) const;

  std::unique_ptr<const Trk::TrackStateOnSurface>
  updateStateWithBestFittingCluster(
      const Trk::Track* track,
      const std::vector<std::unique_ptr<const Trk::TrackParameters>>& params,
      const HGTD_ClusterContainer* container) const;

  std::unique_ptr<const Trk::TrackStateOnSurface>
  findBestCompatibleCluster(const Trk::Track* track,
                            const Trk::TrackParameters* param,
                            const HGTD_ClusterContainer* container) const;

  std::unique_ptr<const Trk::TrackStateOnSurface>
  updateState(const Trk::Track* track, const Trk::TrackParameters* param,
              const HGTD_Cluster* cluster) const;
 
  std::pair<const HGTD_Cluster*, HGTD::ClusterTruthInfo>
  getTruthMatchedCluster(const std::vector<const Trk::Surface*>& surfaces,
                         const HGTD_ClusterContainer* container,
                         const xAOD::TruthParticle* truth_ptkl,
                         const HepMC::GenEvent* hs_event,
                         const InDetSimDataCollection* sim_data) const;
 
  // extrapolation tool
  ToolHandle<Trk::IExtrapolator> m_extrapolator{
      this, "ExtrapolatorTool", "Trk::Extrapolator/AtlasExtrapolator",
      "Tool for extrapolating the track to the HGTD surfaces"};
  // kalman updator tool
  ToolHandle<Trk::IUpdator> m_updator{
      this, "UpdatorTool", "Trk::KalmanUpdator/KalmanUpdator",
      "Tool for updating the track parameters accounting for new measurements"};
 
  ToolHandle<IHGTD_TOFcorrectionTool> m_tof_corr_tool{
      this, "TOFCorrTool", "StraightLineTOFcorrectionTool",
      "Tool for correcting for time of flight"};
 
  ToolHandle<IHGTD_ClusterTruthTool> m_truth_tool{
      this, "ClusterTruthTool", "HGTD::ClusterTruthTool/ClusterTruthTool",
      "Tool for classifying HGTD clusters with truth information"};
 
  const HGTD_DetectorManager* m_hgtd_det_mgr{nullptr};
  const HGTD_ID* m_hgtd_id_helper{nullptr};

  FloatProperty m_chi2_cut{this, "Chi2Cut", 5.0,
                           "Quality cut for decision to keep track extension"};

  IntegerProperty m_particle_hypot{
      this, "ParticleHypothesis", Trk::ParticleHypothesis::pion,
      "The hypothesis of the track's particle type"};
};
 
#endif // HGTD_ITERATIVEEXTENTSIONTOOL_H