dd/d40/AddTRTMomConstr_8h_source.html

/*

  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