InDetTrackSplitterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETINDETTRACKSPLITTERTOOL_H
#define INDETINDETTRACKSPLITTERTOOL_H

 
#include "InDetTrackSplitterTool/IInDetTrackSplitterTool.h"
 
#include "GaudiKernel/IAlgTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "TrkTrack/Track.h"
#include "TrkTrack/TrackCollection.h"
 
 
class TRT_ID;
class SCT_ID;
 
class IInDetTrackSplitterTool;
 
namespace Trk {
  class ITrackFitter ;
  class ITrackSelectorTool;
  class RIO_OnTrack;
  class MeasurementBase;
  class MeasurementBaseComparisonFunction;
  class PseudoMeasurementOnTrack;
  class StraightLineSurface;
}
 
 
namespace InDet{
  
 
  class InDetTrackSplitterTool : public AthAlgTool, virtual public IInDetTrackSplitterTool
    {
      
    public:
      InDetTrackSplitterTool(std::string const&, std::string const&,IInterface const *);
      
      virtual ~InDetTrackSplitterTool();

      virtual StatusCode initialize();
      virtual StatusCode finalize();
      
      virtual void splitTracks(TrackCollection const* inputTracks) const;
      
      virtual std::pair<Trk::Track*, Trk::Track*> splitInUpperLowerTrack(Trk::Track const& input, bool siliconHitsOnly = false) const;

      virtual Trk::Track* stripTrack(Trk::Track const& input, bool removeSilicon = true, bool applyConstraint = true) const;
      
      virtual std::pair<Trk::Track*, Trk::Track*> splitInOddEvenHitsTrack(Trk::Track const& input) const;

      std::string const inline UpperTracksKey() const {return m_outputUpperTracksName.key();}
      std::string const inline LowerTracksKey() const {return m_outputLowerTracksName.key();}
 
    private:
       
      bool trackIsCandidate(Trk::Track const& inputTrack) const;
      
      std::vector<Trk::MeasurementBase const*> getSCTHits(Trk::Track const& input) const;

      std::vector<Trk::MeasurementBase const*>::iterator findSCTHitsFromSameSpacePoint(Trk::MeasurementBase const* sctHit, std::vector<Trk::MeasurementBase const*>& listOfSCTHits) const;

      bool isConstrained(unsigned int nPixelHits, unsigned int nSCTHits, unsigned int nTRTHits, unsigned int nPseudomeasurements) const;
      
      Trk::Track* stripSiFromTrack(Trk::Track const& input) const;
      
      Trk::PseudoMeasurementOnTrack const* makeThetaZ0Constraint(Trk::Perigee const* originialPerigee) const;

      Trk::Track* stripTRTFromTrack(Trk::Track const& input, bool applyConstraint = true) const;
      
      Trk::PseudoMeasurementOnTrack const* makePConstraint(Trk::Perigee const* originialPerigee
                 ,Trk::StraightLineSurface const* trtSurf) const;

      SG::WriteHandleKey<TrackCollection> m_outputUpperTracksName;
      SG::WriteHandleKey<TrackCollection> m_outputLowerTracksName;
      
      ToolHandle<Trk::ITrackFitter>  m_trkfitter;
      TRT_ID const* m_trtid{} ;
      SCT_ID const* m_sctid{} ;
      
      bool m_keepmaterial;
      
    };
}
 
 
#endif