ReFitTrack.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2019 CERN for the benefit of the ATLAS collaboration
*/
 
// ReFit.h, (c) ATLAS Detector Softwareop 
 
#ifndef TRKREFITALG_REFITTRACK_H
#define TRKREFITALG_REFITTRACK_H
 
// Base class
#include "AthenaBaseComps/AthAlgorithm.h"
#include "GaudiKernel/ToolHandle.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/MsgStream.h"
#include "TrkTrack/Track.h"
#include "TrkFitterUtils/FitterTypes.h"
#include "TrkTrack/TrackCollection.h"
#include "TrkEventPrimitives/ParticleHypothesis.h"
#include "TrkEventUtils/TrkParametersComparisonFunction.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkToolInterfaces/ITrackSelectorTool.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
#include "TrkToolInterfaces/IPRDtoTrackMapTool.h"
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "BeamSpotConditionsData/BeamSpotData.h"
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
class VxContainer;
 
namespace Trk{
 
class ReFitTrack : public AthAlgorithm  {
 
public:
 
  typedef Trk::Track Track;
 
  ReFitTrack(const std::string &name,ISvcLocator *pSvcLocator);
 
  virtual StatusCode initialize();
  virtual StatusCode execute();
  virtual StatusCode finalize();
 
private:
  ReFitTrack();
  ReFitTrack(const ReFitTrack&);
  ReFitTrack &operator=(const ReFitTrack&);
  
  const Trk::RecVertex* setupConstrainedFit();
  
 
  // --- job options
  SG::ReadHandleKey<TrackCollection>  m_trackName;
  SG::WriteHandleKey<TrackCollection> m_newTrackName;
 
  // --- fitter steering
  Trk::RunOutlierRemoval              m_runOutlier;             
  int                                 m_matEffects;             
  Trk::ParticleHypothesis             m_ParticleHypothesis;     
 
  
  // -- algorithm members
  const TrackCollection                     *m_tracks;          
  ToolHandle<Trk::ITrackFitter>              m_ITrackFitter;    
  ToolHandle<Trk::ITrackFitter>              m_ITrackFitterTRT; 
  ToolHandle<Trk::IExtendedTrackSummaryTool> m_trkSummaryTool;  
  ToolHandle<Trk::IPRDtoTrackMapTool>        m_assoTool
     {this, "AssociationTool", "" };                            
 
  ToolHandle<Trk::ITrackSelectorTool> m_trkSelectorTool;        
                                                            
  unsigned int                        m_constrainFitMode;       
  SG::ReadHandleKey<VxContainer>      m_vxContainerName;   
  SG::ReadCondHandleKey<InDet::BeamSpotData> m_beamSpotKey { this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot" };
  // For P->T conversion ID tracks
  SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_pixelDetEleCollKey{this, "PixelDetEleCollKey", "PixelDetectorElementCollection", "Key of SiDetectorElementCollection for Pixel"};
  SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_SCTDetEleCollKey{this, "SCTDetEleCollKey", "SCT_DetectorElementCollection", "Key of SiDetectorElementCollection for SCT"};
  ToolHandle<Trk::IExtrapolator>      m_extrapolator;           
  
  bool m_usetrackhypo;                                       
 
  
};
 
} 
 
#endif //TRKREFITALG_REFITTRACK_H