TRT_SegmentsToTrack.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRT_SEGMENTSTOTRACK_H
#define TRT_SEGMENTSTOTRACK_H
 
//======================================================
//
// Algorithm for creating Tracks out of TrackSegments
// by feeding them through a TrackFitter
//
// Author: Christian Schmitt <Christian.Schmitt@cern.ch>
//
//=======================================================
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkSegment/SegmentCollection.h"
#include "TrkTruthData/PRD_MultiTruthCollection.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
#include "TrkFitterInterfaces/ITrackFitter.h" 
#include "TrkExInterfaces/IExtrapolator.h"
// for shared hits:
#include "TrkToolInterfaces/IPRDtoTrackMapTool.h"
#include "TrkEventUtils/PRDtoTrackMap.h"
 
class PRD_MultiTruthCollection;
class AtlasDetectorID;
class TRT_ID;
 
namespace Trk {
  class Track;
  class Segment;
}
 
namespace InDet{
  class TRT_DriftCircle;
}
 
 
namespace InDet 
{
 class TRT_SegmentsToTrack : public AthAlgorithm
  {
  public:
    TRT_SegmentsToTrack(const std::string& name, ISvcLocator* pSvcLocator);
    ~TRT_SegmentsToTrack();
    
    StatusCode initialize();
    StatusCode execute();
    StatusCode finalize();
 
  private:
    int getNumberReal(const InDet::TRT_DriftCircle*,const EventContext& ctx) const; 
 
 
    double getRealFractionTRT(const Trk::Track *track,const EventContext& ctx) const; 
    
    double getNoiseProbability(const Trk::Track *track) const; 
    
    int nTRTHits(const Trk::Track *track) const; 
    
    int nHTHits(const Trk::Track *track) const; 
 
    void combineSegments(const EventContext& ctx) const;
 
    
    SG::ReadHandleKey<Trk::SegmentCollection> m_inputSegmentCollectionName{this,"InputSegmentsCollection","TrackSegments","RHK to retrieve input track collection"}; 
 
    SG::WriteHandleKey<TrackCollection> m_outputTrackCollectionName{this,"OutputTrackCollection","SegmentTracks","WHK to store output tracks"};  
 
    ToolHandle<Trk::ITrackFitter> m_trackFitter
       {this,"TrackFitter","Trk::KalmanFitter/TrkKalmanFitter",""};          
 
    ToolHandle<Trk::IExtrapolator> m_extrapolator
       {this,"ExtrapolationTool","Trk::Extrapolator/InDetExtrapolator",""};  
 
    ToolHandle<Trk::IExtendedTrackSummaryTool> m_trkSummaryTool
       {this, "SummaryTool","",""};
    ToolHandle<Trk::IPRDtoTrackMapTool>        m_assoTool
       {this, "AssociationTool", "","" };
    SG::WriteHandleKey<Trk::PRDtoTrackMap>    m_assoMapName
       {this,"AssociationMapName","",""};  
    SG::ReadHandleKey<Trk::PRDtoTrackMap>     m_inputAssoMapName
       {this,"InputAssociationMapName","",""};  
 
    
    double m_noiseCut;                        
    int  m_minTRTHits;                        
    
    const AtlasDetectorID*                       m_idHelper; //<! Detector ID helper
    
    const TRT_ID* m_trtid ;
    
 
    SG::ReadHandleKey<PRD_MultiTruthCollection> m_multiTruthCollectionTRTName{this,"PRDTruthCollectionTRT","","RHK to retrieve TRT turth info"}; 
 
    int m_nTracksReal;                        
    int m_nTracksFake;                        
    
    double m_noiseratio;                      
    
    mutable std::atomic<int> m_events;        
    std::map<int,int> m_MapReal;              
    std::map<int,int> m_MapFake;              
    
    bool m_materialEffects;                   
    bool m_outlierRemoval;                    
    
 
    bool m_combineSegments;                   
 
    SG::ReadHandleKey<Trk::SegmentCollection> m_barrelSegments{this,"BarrelSegments","TRTBarrelSegments","RHK to retrieve barrel track segments"}; 
    SG::ReadHandleKey<Trk::SegmentCollection> m_endcapSegments{this,"EndcapSegments","TRTEndcapSegments","RHK to retrieve endcap track segments"}; 
 
    SG::WriteHandleKey<TrackCollection> m_BECCollectionName{this,"BarrelEndcapTracks","TRT_Barrel_EC","WHK to write tracks"};  
 
    mutable std::atomic<int> m_n_combined_fit;
 
    
 
  };
}
 
#endif