InDetExtensionProcessor.h

Go to the documentation of this file.

/*
   Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
 */
 
// InDetExtensionProcessor.h, (c) ATLAS Detector Softwareop
 
#ifndef INDETEXTENSIONPROCESSOR_H
#define INDETEXTENSIONPROCESSOR_H
 
// Base class
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "CxxUtils/checker_macros.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkTrack/TrackExtensionMap.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "TrkToolInterfaces/ITrackScoringTool.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
 
#include "GaudiKernel/ToolHandle.h"
 
#include <vector>
#include <array>
#include <mutex>
 
namespace InDet {
  class InDetExtensionProcessor: public AthReentrantAlgorithm  {
  public:
    InDetExtensionProcessor(const std::string& name, ISvcLocator* pSvcLocator);
 
    virtual StatusCode initialize();
    virtual StatusCode execute(const EventContext& ctx) const;
    virtual StatusCode finalize();
  private:
    MsgStream &dumpStat(MsgStream &out) const;
 
    TrackCollection* createExtendedTracks(const EventContext& ctx,
                                          const TrackCollection* tracks_in,
                                          const TrackExtensionMap* track_extension_map) const;
 
    Trk::Track* trackPlusExtension(const EventContext& ctx,
                                   const Trk::Track*,
                                   const std::vector<const Trk::MeasurementBase*>&) const;
 
    //
    // --- job options
    //
    SG::ReadHandleKey<TrackCollection>    m_trackName;        
    bool m_cosmics;          
    SG::ReadHandleKey<TrackExtensionMap>  m_extensionMapName; 
    SG::WriteHandleKey<TrackCollection>   m_newTrackName;     
    ToolHandle<Trk::ITrackFitter>         m_trackFitter;      
 
    ToolHandle<Trk::IExtendedTrackSummaryTool> m_trackSummaryTool
    //      {this, "TrackSummaryTool", "InDetTrackSummaryTool"};
    {
      this, "TrackSummaryTool", ""
    };
 
    ToolHandle<Trk::ITrackScoringTool>    m_scoringTool;      
    Trk::RunOutlierRemoval m_runOutlier;               
    bool m_keepFailedExtensionOnTrack;
    bool m_refitPrds;
    int m_matEffects;
    bool m_suppressHoleSearch;  
    bool m_tryBremFit;          
    bool m_caloSeededBrem;      
    float m_pTminBrem;           
 
    // -- algorithm members
    Trk::ParticleHypothesis m_particleHypothesis; 
 
    enum StatIndex {
      iAll, iBarrel, iTransi, iEndcap, Nregions
    };
 
    enum StatTrkType {
      nInput, nRecognised, nExtended, nRejected, nFailed, nRecoveryBremFits, 
      nBremFits, nFits, nNotExtended, nLowScoreBremFits, nExtendedBrem, nTypes 
    };
 
    // -- Using atomics to be multi-thread safe
    mutable std::array< std::array<std::atomic<int>, Nregions>, nTypes > m_counters ATLAS_THREAD_SAFE;
    mutable std::atomic<int> m_Nevents ATLAS_THREAD_SAFE;
    
    std::vector<float>  m_etabounds;           
 
    void incrementRegionCounter(std::array<std::atomic<int>, 4>&, const Trk::Track*, bool = true) const;
  };
}
 
#endif //INDETEXTENSIONPROCESSOR_H