InDetSecVtxTrackSelectionTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
//Author: Lianyou Shan <lianyou.shan@cern.ch>
// -*- c++ -*-
#ifndef INDETSECVTXTRACKSELECTIONTOOL_H
#define INDETSECVTXTRACKSELECTIONTOOL_H
 
// Local include(s):
#include "InDetTrackSelectionTool/IInDetTrackSelectionTool.h"
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
// #include "CxxUtils/checker_macros.h" // ATLAS_THREAD_SAFE
#ifndef XAOD_ANALYSIS
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "TrkToolInterfaces/ITrackSummaryTool.h"
#endif
 
#include <atomic>
#include <limits>
#include <map>
// #include <mutex>
 
namespace InDet {
 
  // forward declaration of helper classes
  class SecVtxTrackAccessor;
  class SecVtxTrackCut;
 
  class InDetSecVtxTrackSelectionTool :
    public virtual IInDetTrackSelectionTool, 
    public asg::AsgTool {
    
    friend class SecVtxTrackCut;
 
    ASG_TOOL_CLASS2( InDetSecVtxTrackSelectionTool,
             IAsgSelectionTool,
             InDet::IInDetTrackSelectionTool )
    
  public:
    InDetSecVtxTrackSelectionTool( const std::string& name );
 
    // The default destructor is OK but it must be defined in the
    // implementation file in order to forward declare with unique_ptr
    ~InDetSecVtxTrackSelectionTool();
 
    
    virtual StatusCode initialize() override;
    virtual StatusCode finalize() override;
    
    
    
    virtual const asg::AcceptInfo& getAcceptInfo() const override;
    
    virtual  asg::AcceptData accept( const xAOD::IParticle* ) const override;
    
    
    
    virtual  asg::AcceptData accept( const xAOD::TrackParticle& track,
                     const xAOD::Vertex* vertex = nullptr ) const override;
 
#ifndef XAOD_ANALYSIS
    virtual  asg::AcceptData accept( const Trk::Track& track,
                       const Trk::Vertex* vertex = nullptr) const override;
#endif
    
    virtual void setCutLevel( InDet::CutLevel level, Bool_t overwrite = true ) override
       __attribute__ ((deprecated("For consistency with the athena interface, the cut level is best set through the \"CutLevel\" property.")));
 
  private:
    bool m_isInitialized = false; // flag whether or not the tool has been initialized, to check erroneous use cases.
    mutable std::atomic<bool> m_warnInit = false; // flag to keep track of whether we have warned about a lack of initialization
 
    std::unordered_map< std::string, std::shared_ptr<SecVtxTrackAccessor> > m_trackAccessors; 
 
    // first element is cut family, second is the set of cuts
    std::map< std::string, std::vector< std::unique_ptr<SecVtxTrackCut> > > m_trackCuts; 
 
    mutable std::atomic<ULong64_t> m_numTracksProcessed = 0; 
    mutable std::atomic<ULong64_t> m_numTracksPassed = 0; 
    //    mutable std::vector<ULong64_t> m_numTracksPassedCuts ATLAS_THREAD_SAFE; //!< tracks the number of tracks that passed each cut family Guarded by m_mutex
    //    mutable std::mutex m_mutex;
 
    constexpr static Double_t LOCAL_MAX_DOUBLE = 1.0e16;
    constexpr static Int_t LOCAL_MAX_INT = std::numeric_limits<Int_t>::max();
 
    asg::AcceptInfo m_acceptInfo; 
    
    DoubleProperty m_minD0{this, "minD0", -1., "Minimum |d0| of tracks"};
    IntegerProperty m_NPixel0TRT
      {this, "minNPixelHitsAtZeroTRT", -1, "Minimum number of Pixel hit upon zero TRT hit"};
    IntegerProperty m_minInDetHits
      {this, "minTotalHits", -1, "Minimum number of Pixel + SCT + TRT hits"};
 
//    ToolHandle< InDet::IInDetTrackSelectionTool > m_trkFilter ;
#ifndef XAOD_ANALYSIS
    Bool_t m_initTrkTools = false; 
    Bool_t m_trackSumToolAvailable = false; 
#endif // XAOD_ANALYSIS
 
  }; // class InDetSecVtxTrackSelectionTool
 
} // namespace InDet
 
#endif // INDETSECVTXTRACKSELECTIONTOOL_INDETSECVTXTRACKSELECTIONTOOL_H