AddTRTMomConstr.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
// @file AddTRTMomConstr.cxx
//
// @brief Algorithm for 
// This code is derived from TRT_AlignAlgs/AddTRTMomConstr.h/.cxx
//
// @author Oleg Brandt
// many thanks to the TRTAlignAlgs designers, Pawel Bruckman, Markus Elsing and Thijs Cornelissen!
//
// $Id: AddTRTMomConstr.h,v 1.8 2009-05-18 15:48:05 moles Exp $
 
#ifndef INDETALIGNGENALGS_ADDTRTMOMCONSTR_H
#define INDETALIGNGENALGS_ADDTRTMOMCONSTR_H
 
#include <string>
#include "AthenaBaseComps/AthAlgorithm.h"
#include "AthenaBaseComps/AthMsgStreamMacros.h"
#include "GaudiKernel/ToolHandle.h"
#include "TrkFitterUtils/FitterTypes.h"
#include "TrkParameters/TrackParameters.h"
 
class TRT_ID ;
class AtlasDetectorID ;
 
namespace Trk {
  class ITrackFitter ;
  class Track ;
  class ITrackSummaryTool ;
  class TrackStateOnSurface ;
  class PseudoMeasurementOnTrack ;
}
 
class AddTRTMomConstr: public AthAlgorithm {
 public:
  AddTRTMomConstr(const std::string& name, ISvcLocator* pSvcLocator) ;
  ~AddTRTMomConstr() ;
  StatusCode initialize() ; 
  StatusCode execute()    ; 
  StatusCode finalize()   ; 
 
  MsgStream& dump( MsgStream& outst ) const ;
 
 private:
  // helper functions
  const Trk::TrackStateOnSurface* findouterscthit( const Trk::Track* track ) ;
  const Trk::TrackStateOnSurface* findinnertrthit( const Trk::Track* track ) ;
  bool accept( const Trk::Track& track ) ;
  const Trk::MeasurementSet addPM( Trk::MeasurementSet& ms, const Trk::PseudoMeasurementOnTrack* pm ) ;
  const Trk::PseudoMeasurementOnTrack* createPMfromSi ( const Trk::Perigee* mp ) ;
  const Trk::PseudoMeasurementOnTrack* createPMfromTRT( const Trk::Perigee* mpSi
                                                      , const Trk::Perigee* mpTRT ) ;
  Trk::Track* addTRTMomentumConstraint( const Trk::Track* track ) ;
 
  // infrastructure  
  const AtlasDetectorID*  m_idHelper  ; 
  const TRT_ID*     m_trtid     ; 
  std::string m_trackListInput  ; 
  std::string m_trackListOutput ; 
  ToolHandle<Trk::ITrackSummaryTool> m_trackSummaryTool ; 
  ToolHandle<Trk::ITrackFitter> m_trackFitter ; 
 
  // algorithm-internal members
  bool   m_useThetaCorrection ; 
  double m_thetaCorr          ; 
  bool   m_applyTrkSel        ; 
  double m_selPtMin           ; 
  double m_selEtaMin          ; 
  double m_selEtaMax          ; 
  int    m_selNHitPIXMin      ; 
  int    m_selNHitSCTMin      ; 
  int    m_selNHitTRTMin      ; 
  int    m_selNHitTRTMinCrack ; 
  double m_selEtaCrackMin     ; 
  double m_selEtaCrackMax     ; 
  double m_selChiSqPerDOFMin  ; 
  size_t m_nTracksProcessed   ; 
  size_t m_nTracksPresel      ; 
  size_t m_nTracksAccepted    ; 
  size_t m_nRejectPIX         ; 
  size_t m_nRejectSCT         ; 
  size_t m_nRejectTRT         ; 
  size_t m_nRejectPM          ; 
} ;
 
MsgStream& operator<<( MsgStream& outst, const AddTRTMomConstr& alg ) ;
 
#endif