InDetTrtTrackScoringTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRTTRACKSCORINGTOOL_H
#define INDETTRTTRACKSCORINGTOOL_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkEventPrimitives/TrackScore.h"
#include "TrkToolInterfaces/ITrackScoringTool.h"
// MagField cache
#include "MagFieldConditions/AtlasFieldCacheCondObj.h"
#include "MagFieldElements/AtlasFieldCache.h"
// Access to mu
#include "LumiBlockComps/ILumiBlockMuTool.h"
#include "TrkParameters/TrackParameters.h"
#include <string>
#include <vector>
 
namespace Trk {
class Track;
class TrackSummary;
}
 
class TRT_ID;
 
namespace InDet {
 
class ITrtDriftCircleCutTool;
 
class InDetTrtTrackScoringTool
  : virtual public Trk::ITrackScoringTool
  , public AthAlgTool
{
 
public:
  InDetTrtTrackScoringTool(const std::string&,
                           const std::string&,
                           const IInterface*);
  virtual ~InDetTrtTrackScoringTool() = default;
  virtual StatusCode initialize() override;
  virtual Trk::TrackScore score(const Trk::Track& track) const override;
 
  virtual Trk::TrackScore simpleScore(
    const Trk::Track& track,
    const Trk::TrackSummary& trackSum) const override;
 
  Trk::TrackScore TRT_ambigScore(const Trk::Track& track,
                                 const Trk::TrackSummary& trackSum) const;
 
private:
  void setupTRT_ScoreModifiers();
 
  bool isGoodTRT(const Trk::Track&) const;
 
  // Get Eta bin for eta-dependent TRT-track cuts
  unsigned int getEtaBin(const Trk::Perigee& perigee) const;
 
  // Get eta- and mu-dependent nTRT cut for TRT track
  double getMuDependentNtrtMinCut(unsigned int eta_bin) const;
 
  const TRT_ID* m_trtId;
 
  // these are used for ScoreModifiers
  int m_maxSigmaChi2, m_maxTrtRatio, m_maxTrtFittedRatio;
 
  std::vector<double> m_factorSigmaChi2, m_factorTrtRatio,
    m_factorTrtFittedRatio;
 
  std::vector<double> m_boundsSigmaChi2, m_boundsTrtRatio,
    m_boundsTrtFittedRatio;
 
  bool m_useAmbigFcn;
  bool m_useSigmaChi2;
 
  std::vector<Trk::TrackScore> m_summaryTypeScore;
 
  ToolHandle<ITrtDriftCircleCutTool> m_selectortool;
 
  // Read handle for conditions object to get the field cache
  SG::ReadCondHandleKey<AtlasFieldCacheCondObj> m_fieldCacheCondObjInputKey{
    this,
    "AtlasFieldCacheCondObj",
    "fieldCondObj",
    "Name of the Magnetic Field conditions object key"
  };
 
  ToolHandle<ILumiBlockMuTool> m_lumiBlockMuTool{
    this,
    "LuminosityTool",
    "LumiBlockMuTool/LumiBlockMuTool",
    "Luminosity Tool"
  };
 
  int m_minTRTonTrk; 
  double m_maxEta;   
  double m_ptmin;    
  bool
    m_parameterization; 
  bool m_oldLogic;      
  double m_minTRTprecision; 
 
  std::vector<double>
    m_TRTTrksEtaBins; 
  std::vector<double>
    m_TRTTrksMinTRTHitsThresholds; 
  std::vector<double>
    m_TRTTrksMinTRTHitsMuDependencies; 
};
 
}
#endif