ConstrainedTrackProvider.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TRKALIGNGENTOOLS_CONSTRAINEDTRACKPROVIDER_H
#define TRKALIGNGENTOOLS_CONSTRAINEDTRACKPROVIDER_H
 
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
#include "muonEvent/Muon.h"
#include "TrkAlignInterfaces/ITrackCollectionProvider.h"
#include "TrkFitterUtils/FitterTypes.h"  
 
 
 
class TFile;
class TH2F;
namespace Trk {
  class IGlobalTrackFitter;
 
  class ConstrainedTrackProvider : virtual public ITrackCollectionProvider, public AthAlgTool {
 
  public:
    ConstrainedTrackProvider(const std::string & type, const std::string & name, const IInterface * parent);
    virtual ~ConstrainedTrackProvider();
    
    virtual StatusCode initialize();
    virtual StatusCode finalize();
    
    virtual StatusCode trackCollection(const TrackCollection*& tracks);
 
    virtual void printSummary();
 
  private :
 
    void  getCorrectedValues_P(const Trk::Perigee* mp, double& correctedQoverP,double& correctedQoverPError);
    void  getCorrectedValues_d0(const Trk::Perigee* mp, double& corrected_d0,double& corrected_d0Error);
    void  getCorrectedValues_z0(const Trk::Perigee* mp, double& corrected_z0,double& corrected_z0Error);
    bool  passTrackSelection(const Trk::Track *track);
 
    ToolHandle<IGlobalTrackFitter>        m_trackFitter;     
 
    RunOutlierRemoval     m_runOutlierRemoval;         
    bool          m_scalepmaptogev;
    bool          m_doTrackSelection;
    int           m_maxRetrievalErrors;        
    bool          m_useConstrainedTrkOnly;
    int           m_constrainedTracks;
    int           m_passconstrainedRequirements;
    int           m_unconstrainedTracks;
    bool          m_useConstraintError;
    double        m_reduceConstraintUncertainty;
    double        m_reduceConstraintUncert_z0;
    double        m_deltaScaling;
 
    int           m_minPIXHits;
    int           m_minSCTHits;
    int           m_minTRTHits;
    double        m_maxd0;
    double        m_maxz0;
    double        m_minPt;
    double        m_maxPt;
    
 
    std::string     m_inputTrackCollection;
 
 
    bool            m_CorrectMomentum;
    std::string     m_constraintFileName_P; 
    TFile*          m_constraintInputFile_P;
    std::string     m_constraintHistName_P; 
    TH2F*           m_etaphiMap_P;
      // Corrections expected to be in GeV-1
 
    std::string     m_constraintFileName_d0; 
    TFile*          m_constraintInputFile_d0;
    std::string     m_constraintHistName_d0; 
    TH2F*           m_etaphiMap_d0;
      // Corrections expected to be in mm
 
    bool            m_CorrectZ0;
    std::string     m_constraintFileName_z0; 
    TFile*          m_constraintInputFile_z0;
    std::string     m_constraintHistName_z0; 
      // Corrections expected to be in mm
 
    TH2F*           m_etaphiMap_z0;
 
    bool            m_CorrectD0;
    bool            m_CorrectMeanD0;
 
    bool          m_SelectByCharge;    
    bool          m_SelectPositive;
 
    
  }; // end class
 
} // end namespace
 
#endif // TRKALIGNGENTOOLS_CONSTRAINEDTRACKPROVIDER_H