egammaTrkRefitterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EGAMMATRACKTOOLS_EGAMMATRKREFITTERTOOL_H
#define EGAMMATRACKTOOLS_EGAMMATRKREFITTERTOOL_H
 
#include "egammaInterfaces/IegammaTrkRefitterTool.h"
 
#include "egammaInterfaces/ICaloCluster_OnTrackBuilder.h"
#include "AthenaBaseComps/AthAlgTool.h"
 
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "TrkMeasurementBase/MeasurementBase.h"
#include "TrkParameters/TrackParameters.h"
#include "TrkTrack/Track.h"
 
#include <memory>
 
class AtlasDetectorID;
 
class egammaTrkRefitterTool final
  : virtual public IegammaTrkRefitterTool
  , public AthAlgTool
{
 
public:
  egammaTrkRefitterTool(const std::string&,
                        const std::string&,
                        const IInterface*);

  virtual ~egammaTrkRefitterTool() = default; 

  virtual StatusCode initialize() override;

  virtual StatusCode finalize() override;
 
  typedef IegammaTrkRefitterTool::Cache Cache;

  virtual StatusCode refitTrack(const EventContext& ctx,
                                const Trk::Track*,
                                Cache& cache) const override final;
 
private:
  std::vector<const Trk::MeasurementBase*> getIDHits(
    const Trk::Track* track) const;
 
  struct MeasurementsAndTrash
  {
    /*
     * we need to take care of returning all the relevant measurements
     * while at the same time keeping proper ownership only for the ones
     * not handled by the EDM
     */
    std::vector<const Trk::MeasurementBase*> m_measurements;
    std::vector<std::unique_ptr<const Trk::MeasurementBase>> m_trash;
  };

  MeasurementsAndTrash addPointsToTrack(
    const EventContext& ctx,
    const Trk::Track* track,
    const xAOD::Electron* eg = nullptr) const;

  ToolHandle<Trk::ITrackFitter> m_ITrackFitter{
    this,
    "FitterTool",
    "Trk__GaussianSumFitter/GSFTrackFitter",
    "ToolHandle for track fitter implementation"
  };
 
  ToolHandle<ICaloCluster_OnTrackBuilder>
    m_CCOTBuilder{ this, "CCOTBuilder", "CaloCluster_OnTrackBuilder", "" };

  Gaudi::Property<int> m_matEffects{
    this,
    "matEffects",
    1,
    "Type of material interaction in extrapolation (Default Electron)"
  };

  Trk::ParticleHypothesis m_ParticleHypothesis;
 
  const AtlasDetectorID* m_idHelper;
 
  Gaudi::Property<bool> m_useClusterPosition{
    this,
    "useClusterPosition",
    false,
    "Switch to control use of Cluster position measurement"
  };

  Gaudi::Property<bool> m_RemoveTRT{ this,
                                     "RemoveTRTHits",
                                     false,
                                     "RemoveTRT Hits" };
};
 
#endif