InDetConversionFinderTools.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
/***************************************************************************
        InDetConversionFinderTools.h  -  Description
        -------------------
        begin   : 28-08-2006
        authors : Tatjana Lenz
        email   : tatjana.lenz@cern.ch
        changes :
***************************************************************************/
#ifndef INDETCONVERSIONFINDERTOOLS_INDETCONVERSIONFINDERTOOLS_H
#define INDETCONVERSIONFINDERTOOLS_INDETCONVERSIONFINDERTOOLS_H
 
#include "InDetRecToolInterfaces/IVertexFinder.h"
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
/* Tools*/
#include "TrkVertexFitterInterfaces/IVertexFitter.h"
#include "InDetConversionFinderTools/TrackPairsSelector.h"
#include "InDetConversionFinderTools/ConversionPostSelector.h"
#include "InDetConversionFinderTools/SingleTrackConversionTool.h"
#include "InDetConversionFinderTools/ConversionFinderUtils.h"
#include "InDetConversionFinderTools/VertexPointEstimator.h"
#include "TrkToolInterfaces/ITrackSelectorTool.h"
 
#include "TrkTrack/TrackCollection.h"
/*xAOD collections */
#include "xAODTracking/TrackParticle.h"
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
 
#include <vector>
 
namespace InDet {
 
class InDetConversionFinderTools
  : public AthAlgTool
  , virtual public IVertexFinder
{
public:
  InDetConversionFinderTools(const std::string& t,
                             const std::string& n,
                             const IInterface* p);
  ~InDetConversionFinderTools();
 
  virtual StatusCode initialize() override;
  virtual StatusCode finalize() override;
 
  bool m_removeTrt{};
  bool m_isConversion{};
  bool m_decorateVertices;
 
  using IVertexFinder::findVertex;
  virtual std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
  findVertex(const EventContext& ctx,
             const TrackCollection* trk_coll) const override;
 
  virtual std::pair<xAOD::VertexContainer*, xAOD::VertexAuxContainer*>
  findVertex(const EventContext& ctx,
             const xAOD::TrackParticleContainer* trk_coll) const override;
 
protected:
 
  ToolHandle<Trk::IVertexFitter> m_iVertexFitter{ this,
                                                  "VertexFitterTool",
                                                  "Trk::TrkVKalVrtFitter",
                                                  "Vertex fitter Tool" };
  ToolHandle<InDet::TrackPairsSelector> m_trackPairsSelector{
    this,
    "TrackPairsSelector",
    "InDet::TrackPairsSelector",
    "Track Pair Selector Tool"
  };
  ToolHandle<InDet::VertexPointEstimator> m_vertexEstimator{
    this,
    "VertexPointEstimator",
    "InDet::VertexPointEstimator",
    "Vertex point estimator"
  };
  ToolHandle<InDet::ConversionPostSelector> m_postSelector{
    this,
    "PostSelector",
    "InDet::ConversionPostSelector",
    "Tool for post selection of conversion candidates"
  };
  ToolHandle<InDet::SingleTrackConversionTool> m_singleTrkConvTool{
    this,
    "SingleTrackConversionTool",
    "InDet::SingleTrackConversionTool",
    "Tool for single track conversions"
  };
 
  ToolHandle<Trk::ITrackSelectorTool> m_trkSelector{
    this,
    "TrackSelectorTool",
    "InDet::TrackSelectorTool",
    "Tool for track Selection"
  };
 
  bool passPreSelection(TrackPairsSelector::Cache& cache,
                        const xAOD::TrackParticle* track_pos,
                        const xAOD::TrackParticle* track_neg,
                        std::vector<Amg::Vector3D>& trackList,
                        Amg::Vector3D& initPos,
                        int& flag,
                        std::map<std::string, float>& intersectionDecors) const;
 
  double m_mindR;
  double m_maxdR;
  double m_MinInitVtxR;
  double m_MinFlightAngle;
};
 
}
 
#endif // INDETCONVERSIONFINDERTOOLS_INDETCONVERSIONFINDERTOOL_H