DiTauTrackFinder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DITAUREC_DITAUTRACKFINDER_H
#define DITAUREC_DITAUTRACKFINDER_H
 
#include "DiTauToolBase.h"
 
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
 
#include "xAODTracking/Vertex.h"
#include "xAODTracking/TrackParticleContainer.h"
 
#include "TrkToolInterfaces/ITrackSelectorTool.h"
 
 
class DiTauTrackFinder : public DiTauToolBase {
 public:
 
  //-------------------------------------------------------------
  //-------------------------------------------------------------
  DiTauTrackFinder(const std::string& type,
           const std::string& name,
           const IInterface * parent);
 
  //-------------------------------------------------------------
  //-------------------------------------------------------------
  virtual ~DiTauTrackFinder();
 
  virtual StatusCode initialize() override;
 
  virtual StatusCode execute(DiTauCandidateData * data,
                 const EventContext& ctx) const override;
 
 
  // ------------------------------------------------------------
  // Definition of track types
  // ------------------------------------------------------------
  enum DiTauTrackType { 
    DiTauSubjetTrack = 0,
    DiTauIsoTrack = 1,
    DiTauOtherTrack = 2,
    OutsideTrack = 3
  };
 
  void getTracksFromPV( const DiTauCandidateData*,
            const xAOD::TrackParticleContainer*,
            const xAOD::Vertex*,
            std::vector<const xAOD::TrackParticle*>&,
            std::vector<const xAOD::TrackParticle*>&,
            std::vector<const xAOD::TrackParticle*>& ) const;
 
  DiTauTrackType diTauTrackType( const DiTauCandidateData*,
                 const xAOD::TrackParticle*,
                 const xAOD::Vertex* ) const;
 
 
 
 
 private:
  float m_MaxDrJet;
  float m_MaxDrSubjet;
  int m_MaxNTracksSubjet;
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_TrackParticleContainerName
    { this, "TrackParticleContainer", "InDetTrackParticles", "" };
  ToolHandle<Trk::ITrackSelectorTool> m_TrackSelectorTool;
 
};
 
#endif // DITAUREC_DITAUTRACKFINDER_H