TrackAnalysisInfoWriteTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_TRACKANALYSISINFOWRITETOOL_H
#define INDETTRACKPERFMON_TRACKANALYSISINFOWRITETOOL_H


#include "AsgTools/AsgTool.h"
#include "StoreGate/WriteHandle.h"

#include "xAODCore/BaseContainer.h"

#include <string>
#include <vector>

#include "InDetTrackPerfMon/ITrackAnalysisDefinitionSvc.h"
 
 
namespace IDTPM {

  template< typename coll_t >
  using EL_t = ElementLink< coll_t >;
 
  template< typename coll_t >
  using VecEL_t = std::vector< EL_t< coll_t > >;
 
  template< typename coll_t >
  using Vec_t = std::vector< typename coll_t::const_value_type >;
 
  using VecF_t = std::vector< float >;
 
  template< typename Tcoll_t, typename Rcoll_t >
  using MatchInfo_t = std::tuple< VecEL_t<Tcoll_t>, VecEL_t<Rcoll_t>, VecF_t >;

  class TrackAnalysisCollections;
  class ITrackMatchingLookup;
 
 
  class TrackAnalysisInfoWriteTool :
      public virtual asg::IAsgTool, 
      public asg::AsgTool {
 
    ASG_TOOL_CLASS( TrackAnalysisInfoWriteTool, IAsgTool );
 
  public:

    TrackAnalysisInfoWriteTool( const std::string& name );

    virtual ~TrackAnalysisInfoWriteTool() = default;

    virtual StatusCode initialize() override;

    StatusCode write(
        SG::WriteHandle< xAOD::BaseContainer >& wh,
        TrackAnalysisCollections& trkAnaColls,
        const std::string& chain = "",
        unsigned int roiIdx = 0,
        const std::string& roiStr = "" ) const;

    std::string printInfo(
        SG::WriteHandle< xAOD::BaseContainer >& wh ) const;
 
  private:

    template< typename Tcoll_t, typename Rcoll_t >
    MatchInfo_t< Tcoll_t, Rcoll_t > getMatchInfo(
        const Vec_t< Tcoll_t >& Tvec,
        const Vec_t< Rcoll_t >& Rvec,
        const Tcoll_t* Tcoll,
        const Rcoll_t* Rcoll,
        const ITrackMatchingLookup* matches ) const;

    template< typename Tcoll_t, typename Rcoll_t >
    std::string printMatchInfo(
        const VecEL_t< Tcoll_t >& testVec,
        const VecEL_t< Rcoll_t >& refVec,
        const VecF_t& distVec ) const;
 
    StringProperty m_anaTag{ this, "AnaTag", "", "Track analysis tag" };
 
    SmartIF<ITrackAnalysisDefinitionSvc> m_trkAnaDefSvc;
 
  }; // class TrackAnalysisInfoWriteTool
 
} // namespace IDTPM
 
#include "TrackAnalysisInfoWriteTool.icc"
 
#endif // > !INDETTRACKPERFMON_TRACKANALYSISINFOWRITETOOL_H