Trk::ConstrainedTrackProvider Class Reference

#include <ConstrainedTrackProvider.h>

Inheritance diagram for Trk::ConstrainedTrackProvider:

Collaboration diagram for Trk::ConstrainedTrackProvider:

Public Member Functions
	ConstrainedTrackProvider (const std::string &type, const std::string &name, const IInterface *parent)
virtual StatusCode	initialize ()
virtual StatusCode	finalize ()
virtual StatusCode	trackCollection (const TrackCollection *&tracks)
virtual void	printSummary ()
	Print statistical summary to logfile.
virtual void	setLogStream (std::ostream *os)
	sets the output stream for the logfile
virtual void	setNtuple (TFile *)
	sets ntuple
virtual StatusCode	fillNtuple ()
	writes tree to ntuple
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.
virtual StatusCode	sysInitialize () override
	Perform system initialization for an algorithm.
virtual StatusCode	sysStart () override
	Handle START transition.
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles.
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles.
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T, V, H > &t)
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
bool	msgLvl (const MSG::Level lvl) const

Static Public Member Functions
static const InterfaceID &	interfaceID ()

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution
std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void >	renounce (T &h)
void	extraDeps_update_handler (Gaudi::Details::PropertyBase &ExtraDeps)
	Add StoreName to extra input/output deps as needed.

Protected Attributes
std::ostream *	m_logStream = nullptr
	logfile output stream

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
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)
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T, V, H > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey>

Private Attributes
ToolHandle< IGlobalTrackFitter >	m_trackFitter
Gaudi::Property< Trk::RunOutlierRemoval >	m_runOutlierRemoval
BooleanProperty	m_scalepmaptogev {this, "ScalePMapToGeV", false}
BooleanProperty	m_doTrackSelection {this, "doTrackSelection", true}
BooleanProperty	m_useConstrainedTrkOnly {this, "UseConstrainedTrkOnly", false}
BooleanProperty	m_useConstraintError {this, "UseConstraintError", true}
int	m_constrainedTracks = 0
int	m_unconstrainedTracks = 0
DoubleProperty	m_reduceConstraintUncertainty
DoubleProperty	m_reduceConstraintUncert_z0
DoubleProperty	m_deltaScaling {this, "DeltaScaling", 1.}
IntegerProperty	m_minPIXHits {this, "MinPIXHits", 0}
IntegerProperty	m_minSCTHits {this, "MinSCTHits", 0}
IntegerProperty	m_minTRTHits {this, "MinTRTHits", 0}
DoubleProperty	m_maxd0 {this, "Maxd0", 500.}
DoubleProperty	m_maxz0 {this, "Maxz0", 500.}
DoubleProperty	m_minPt {this, "MinPt", 15.}
DoubleProperty	m_maxPt {this, "MaxPt", 100.}
SG::ReadHandleKey< TrackCollection >	m_inputKey
SG::WriteHandleKey< TrackCollection >	m_outputKey
BooleanProperty	m_CorrectMomentum {this, "CorrectMomentum", true}
StringProperty	m_constraintFileName_P
TFile *	m_constraintInputFile_P = nullptr
StringProperty	m_constraintHistName_P
TH2F *	m_etaphiMap_P = nullptr
StringProperty	m_constraintFileName_d0
TFile *	m_constraintInputFile_d0 = nullptr
StringProperty	m_constraintHistName_d0 {this, "d0ConstraintHistName", "EtaPhiMap"}
TH2F *	m_etaphiMap_d0 = nullptr
BooleanProperty	m_CorrectZ0 {this, "CorrectZ0", false}
StringProperty	m_constraintFileName_z0
TFile *	m_constraintInputFile_z0 = nullptr
StringProperty	m_constraintHistName_z0 {this, "z0ConstraintHistName", "EtaPhiMap"}
TH2F *	m_etaphiMap_z0 = nullptr
BooleanProperty	m_CorrectD0 {this, "CorrectD0", false}
BooleanProperty	m_CorrectMeanD0 {this, "CorrectMeanD0", false}
BooleanProperty	m_SelectByCharge {this, "SelectByCharge", false}
BooleanProperty	m_SelectPositive {this, "SelectPositive", true}
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default)
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default)
std::vector< SG::VarHandleKeyArray * >	m_vhka
bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 32 of file ConstrainedTrackProvider.h.

Member Typedef Documentation

StoreGateSvc_t

typedef ServiceHandle<StoreGateSvc> AthCommonDataStore< AthCommonMsg< AlgTool > >::StoreGateSvc_t

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

ConstrainedTrackProvider()

Trk::ConstrainedTrackProvider::ConstrainedTrackProvider	(	const std::string &	type,
		const std::string &	name,
		const IInterface *	parent )

Definition at line 21 of file ConstrainedTrackProvider.cxx.

    : AthAlgTool(type,name,parent)
   {
    declareInterface<ITrackCollectionProvider>(this);
  }

Member Function Documentation

declareGaudiProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareGaudiProperty ( Gaudi::Property< T, V, H > & hndl, const SG::VarHandleKeyType &  )

inlineprivateinherited

specialization for handling Gaudi::Property<SG::VarHandleKey>

Definition at line 156 of file AthCommonDataStore.h.

  {
    return *AthCommonDataStore<PBASE>::declareProperty(hndl.name(),
                                                       hndl.value(), 
                                                       hndl.documentation());
 
  }

declareProperty()

Gaudi::Details::PropertyBase & AthCommonDataStore< AthCommonMsg< AlgTool > >::declareProperty ( Gaudi::Property< T, V, H > & t )

inlineinherited

Definition at line 145 of file AthCommonDataStore.h.

                                                                     {
    typedef typename SG::HandleClassifier<T>::type htype;
    return AthCommonDataStore<PBASE>::declareGaudiProperty(t, htype());
  }

detStore()

const ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::detStore ( ) const

inlineinherited

The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 95 of file AthCommonDataStore.h.

{ return m_detStore; }

evtStore()

ServiceHandle< StoreGateSvc > & AthCommonDataStore< AthCommonMsg< AlgTool > >::evtStore ( )

inlineinherited

The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc.

Definition at line 85 of file AthCommonDataStore.h.

{ return m_evtStore; }

extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::extraDeps_update_handler ( Gaudi::Details::PropertyBase & ExtraDeps )

protectedinherited

Add StoreName to extra input/output deps as needed.

use the logic of the VarHandleKey to parse the DataObjID keys supplied via the ExtraInputs and ExtraOuputs Properties to add the StoreName if it's not explicitly given

fillNtuple()

virtual StatusCode Trk::ITrackCollectionProvider::fillNtuple ( )

inlinevirtualinherited

writes tree to ntuple

Reimplemented in Trk::MSConstraintTracksProvider.

Definition at line 46 of file ITrackCollectionProvider.h.

{ return StatusCode::SUCCESS; }

finalize()

StatusCode Trk::ConstrainedTrackProvider::finalize ( )

virtual

Definition at line 114 of file ConstrainedTrackProvider.cxx.

  {
    if(m_constraintInputFile_P){
      m_constraintInputFile_P->Close();
      delete m_constraintInputFile_P;
    }
 
    if(m_constraintInputFile_d0){
      m_constraintInputFile_d0->Close();
      delete m_constraintInputFile_d0;
    }
 
    if(m_constraintInputFile_z0){
      m_constraintInputFile_z0->Close();
      delete m_constraintInputFile_z0;
    }
 
    return StatusCode::SUCCESS;
  }

getCorrectedValues_d0()

void Trk::ConstrainedTrackProvider::getCorrectedValues_d0 ( const Trk::Perigee * mp, double & corrected_d0, double & corrected_d0Error )

private

Definition at line 425 of file ConstrainedTrackProvider.cxx.

  {
    ATH_MSG_DEBUG(" == getCorrectedValues_d0 == STARTED ==");
    // scale d0
 
    double charge(0);
    if( measuredPerigee->parameters()[Trk::qOverP] > 0)
      charge = 1.;
    else
      charge = -1.;
 
    double d0  = measuredPerigee->parameters()[Trk::d0];
    double eta = -log(tan(measuredPerigee->parameters()[Trk::theta]/2.));
    double phi = measuredPerigee->parameters()[Trk::phi];
    double d0err =  (*measuredPerigee->covariance())( Trk::d0, Trk::d0 );
 
 
    int binNumber = m_etaphiMap_d0->FindBin(eta, phi);
    double constraintErr = m_etaphiMap_d0->GetBinError(binNumber);
    double delta = m_etaphiMap_d0->GetBinContent(binNumber) * m_deltaScaling;
 
    correctedD0 = d0 - charge * delta * 0.5 ; // delta d0 = d0_pos - d0_neg . To make the difference null, substract half to the pos and add half to the neg track.
    if (m_CorrectMeanD0) {correctedD0 = d0 + delta;}
    correctedD0Error = d0err;
    ATH_MSG_DEBUG("Correcting d0 by delta =  " << delta << "\t, corrected d0 = "<< correctedD0 << "\t, original d0 = "<< d0 ) ;
 
    if(m_useConstraintError){
      correctedD0Error = d0err + pow( constraintErr, 2 );
    }
    ATH_MSG_DEBUG("Scaling by 1/m_reduceConstraintUncertainty  " << m_reduceConstraintUncertainty << '\t'<< pow( m_reduceConstraintUncertainty,-2)) ;
    correctedD0Error = correctedD0Error * pow( m_reduceConstraintUncertainty,-2);
 
    ATH_MSG_DEBUG(" == input d0: " << d0 << "  deltad0: " << delta << "  final d0: " << correctedD0 << " +- " << correctedD0Error);
    ATH_MSG_DEBUG(" == getCorrectedValues_d0 == Completed ==");
 }

getCorrectedValues_P()

void Trk::ConstrainedTrackProvider::getCorrectedValues_P ( const Trk::Perigee * mp, double & correctedQoverP, double & correctedQoverPError )

private

Definition at line 386 of file ConstrainedTrackProvider.cxx.

  {
 
    // scale q/p not p
 
    double charge(0);
    if( measuredPerigee->parameters()[Trk::qOverP] > 0)
      charge = 1.;
    else
      charge = -1.;
 
    double pt   = std::abs(1./(measuredPerigee->parameters()[Trk::qOverP]))* sin(measuredPerigee->parameters()[Trk::theta]) *1e-3;
    double eta = -log(tan(measuredPerigee->parameters()[Trk::theta]/2.));
    double phi = measuredPerigee->parameters()[Trk::phi];
    double perr =  (*measuredPerigee->covariance())( Trk::qOverP, Trk::qOverP );
 
 
    int binNumber = m_etaphiMap_P->FindBin(eta, phi);
    double constraintErr = m_etaphiMap_P->GetBinError(binNumber);
 
    double delta = m_etaphiMap_P->GetBinContent(binNumber) * m_deltaScaling;
 
    if (m_scalepmaptogev)
      delta = delta * 0.001;
 
    correctedQoverP = measuredPerigee->parameters()[Trk::qOverP] * (1.+ charge *pt *delta );
    correctedQoverPError = perr;
 
    if(m_useConstraintError){
      double qoverpFracError =  perr/pow(measuredPerigee->parameters()[Trk::qOverP], 2);
      qoverpFracError += pow(constraintErr,2);
      correctedQoverPError = qoverpFracError*correctedQoverP*correctedQoverP;
    }
    ATH_MSG_DEBUG("Scaling by 1/m_reduceConstraintUncertainty  " << m_reduceConstraintUncertainty << '\t'<< pow( m_reduceConstraintUncertainty,-2)) ;
    correctedQoverPError = correctedQoverPError * pow( m_reduceConstraintUncertainty,-2);
    ATH_MSG_DEBUG(" == input pt: " << pt << "  deltaSagitta: " << delta << "  final pt: " << sin(measuredPerigee->parameters()[Trk::theta]) * 1e-3/correctedQoverP);
 }

getCorrectedValues_z0()

void Trk::ConstrainedTrackProvider::getCorrectedValues_z0 ( const Trk::Perigee * mp, double & corrected_z0, double & corrected_z0Error )

private

Definition at line 461 of file ConstrainedTrackProvider.cxx.

  {
 
    // scale z0
 
    double charge(0);
    if( measuredPerigee->parameters()[Trk::qOverP] > 0)
      charge = 1.;
    else
      charge = -1.;
 
    double z0  = measuredPerigee->parameters()[Trk::z0];
    double eta = -log(tan(measuredPerigee->parameters()[Trk::theta]/2.));
    double phi = measuredPerigee->parameters()[Trk::phi];
    double z0err =  (*measuredPerigee->covariance())( Trk::z0, Trk::z0 );
 
    int binNumber = m_etaphiMap_z0->FindBin(eta, phi);
    double constraintErr = m_etaphiMap_z0->GetBinError(binNumber);
    double delta = m_etaphiMap_z0->GetBinContent(binNumber) * m_deltaScaling;
 
    correctedZ0 = z0 - charge * delta * 0.5 ; // delta z0 = z0_pos - z0_neg . To make the difference null, substract half to the pos and add half to the neg track.
    correctedZ0Error = z0err;
 
    if(m_useConstraintError){
      correctedZ0Error = z0err + pow( constraintErr, 2 );
    }
    ATH_MSG_DEBUG("Scaling by 1/m_reduceConstraintUncertainty  " << m_reduceConstraintUncertainty << '\t'<< pow( m_reduceConstraintUncertainty,-2)) ;
    correctedZ0Error = correctedZ0Error * pow( m_reduceConstraintUncertainty,-2);
    ATH_MSG_DEBUG("Scaling by 1/m_reduceConstraintUncert_z0  " << m_reduceConstraintUncert_z0 << '\t'<< pow( m_reduceConstraintUncert_z0,-2)) ;
    correctedZ0Error = correctedZ0Error * pow( m_reduceConstraintUncert_z0,-2);
    ATH_MSG_DEBUG(" == input z0: " << z0 << "  deltaz0: " << delta << "  final z0: " << correctedZ0 << " +- " << correctedZ0Error);
 }

initialize()

StatusCode Trk::ConstrainedTrackProvider::initialize ( )

virtual

Definition at line 30 of file ConstrainedTrackProvider.cxx.

  {
    // configure main track fitter
    if(m_trackFitter.retrieve().isFailure()) {
       msg(MSG::FATAL) << "Could not get " << m_trackFitter << endmsg;
       return StatusCode::FAILURE;
    }
    ATH_MSG_INFO("Retrieved " << m_trackFitter);
 
 
    ATH_MSG_INFO( "init m_useConstraintError " <<m_useConstraintError );
 
    ATH_MSG_INFO( "Init m_reduceConstraintUncertainty " << m_reduceConstraintUncertainty );
 
    ATH_MSG_INFO( "Init m_reduceConstraintUncert_z0 " << m_reduceConstraintUncert_z0 );
 
    if(m_CorrectMomentum){
 
      m_constraintInputFile_P = new TFile(m_constraintFileName_P.value().c_str(), "read");
      if ( m_constraintInputFile_P->IsZombie() || !(m_constraintInputFile_P->IsOpen()) ) {
        ATH_MSG_FATAL( " Problem reading TFile " << m_constraintFileName_P );
        return StatusCode::FAILURE;
      }
      ATH_MSG_INFO("Opened  file containing the deltaSagitta constraints" << m_constraintFileName_P);
      m_etaphiMap_P = static_cast<TH2F*>(m_constraintInputFile_P->Get(m_constraintHistName_P.value().c_str()));
      if(!m_etaphiMap_P){
        ATH_MSG_FATAL( " Problem getting constraints Hist.  Name " << m_constraintHistName_P );
        m_constraintInputFile_P->Close();
        delete m_constraintInputFile_P;
        return StatusCode::FAILURE;
      }
      ATH_MSG_INFO("Opened constraints histogram " << m_constraintHistName_P);
 
    }
 
    if(m_CorrectD0){
 
      m_constraintInputFile_d0 = new TFile(m_constraintFileName_d0.value().c_str(), "read");
      if ( m_constraintInputFile_d0->IsZombie() || !(m_constraintInputFile_d0->IsOpen()) ) {
        ATH_MSG_FATAL( " Problem reading TFile " << m_constraintFileName_d0 );
        return StatusCode::FAILURE;
      }
      ATH_MSG_INFO("Opened file containing the d0 constraints" << m_constraintFileName_d0);
      m_etaphiMap_d0 = static_cast<TH2F*>(m_constraintInputFile_d0->Get(m_constraintHistName_d0.value().c_str()));
      if(!m_etaphiMap_d0){
        ATH_MSG_FATAL( " Problem getting constraints Hist.  Name " << m_constraintHistName_d0 );
        m_constraintInputFile_d0->Close();
        delete m_constraintInputFile_d0;
        return StatusCode::FAILURE;
      }
      ATH_MSG_INFO("Opened constraints histogram " << m_constraintHistName_d0);
 
    }
 
    if(m_CorrectZ0){
 
      m_constraintInputFile_z0 = new TFile(m_constraintFileName_z0.value().c_str(),"read");
      if ( m_constraintInputFile_z0->IsZombie() || !(m_constraintInputFile_z0->IsOpen()) ) {
        ATH_MSG_FATAL( " Problem reading TFile " << m_constraintFileName_z0 );
        return StatusCode::FAILURE;
      }
      ATH_MSG_INFO("Opened  file containing the z0 constraints" << m_constraintFileName_z0);
      m_etaphiMap_z0 = static_cast<TH2F*>(m_constraintInputFile_z0->Get(m_constraintHistName_z0.value().c_str()));
      if(!m_etaphiMap_z0){
        ATH_MSG_FATAL( " Problem getting constraints Hist.  Name " << m_constraintHistName_z0 );
        m_constraintInputFile_z0->Close();
        delete m_constraintInputFile_z0;
        return StatusCode::FAILURE;
      }
      ATH_MSG_INFO("Opened constraints histogram " << m_constraintHistName_z0);
 
    }
 
    ATH_CHECK( m_inputKey.initialize() );
    ATH_CHECK( m_outputKey.initialize() );
 
    return StatusCode::SUCCESS;
  }

inputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::inputHandles ( ) const

overridevirtualinherited

Return this algorithm's input handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

interfaceID()

const InterfaceID & ITrackCollectionProvider::interfaceID ( )

inlinestaticinherited

Definition at line 57 of file ITrackCollectionProvider.h.

                                                                  {
    return   IID_TRK_ITrackCollectionProvider;
  }

msg()

MsgStream & AthCommonMsg< AlgTool >::msg ( ) const

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msgLvl()

bool AthCommonMsg< AlgTool >::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

outputHandles()

virtual std::vector< Gaudi::DataHandle * > AthCommonDataStore< AthCommonMsg< AlgTool > >::outputHandles ( ) const

overridevirtualinherited

Return this algorithm's output handles.

We override this to include handle instances from key arrays if they have not yet been declared. See comments on updateVHKA.

passTrackSelection()

bool Trk::ConstrainedTrackProvider::passTrackSelection ( const Trk::Track * track )

private

Definition at line 312 of file ConstrainedTrackProvider.cxx.

  {
    ATH_MSG_DEBUG("== TrackSelection == STARTED ==");
 
    const Trk::Perigee* perigee = track->perigeeParameters();
    if(!perigee){
      ATH_MSG_DEBUG("== TrackSelection == NO perigee on this track");
      return false;
    }
 
    const Trk::TrackSummary* summary = track->trackSummary();
 
    if(!summary){
      ATH_MSG_DEBUG("== TrackSelection == No summary on this track");
      return false;
    }
 
    const int nPixHits        = summary->get(Trk::numberOfPixelHits);
    const int nSCTHits        = summary->get(Trk::numberOfSCTHits);
    const int nTRTHits        = summary->get(Trk::numberOfTRTHits);
 
    const double qoverP = perigee->parameters()[Trk::qOverP] * 1000.;
    const double z0     = perigee->parameters()[Trk::z0];
    const double d0     = perigee->parameters()[Trk::d0];
    double pt          = 0.;
    if ( qoverP != 0 ) pt = std::abs(1.0/qoverP)*sin(perigee->parameters()[Trk::theta]);
 
    ATH_MSG_DEBUG( "== TrackSelection == track pt  : "<< pt );
    double charge(0);
    if( qoverP > 0)
      charge = 1.;
    else
      charge = -1.;
    ATH_MSG_DEBUG( " pt  : "<< pt << " ; Charge : "<< charge);
 
    if(pt < m_minPt ||  pt > m_maxPt   ){
      ATH_MSG_DEBUG("== TrackSelection == Track fails pt cut [ "<<m_minPt<<", "<< m_maxPt << "] " << pt );
      return false;
    }
 
    if (m_SelectByCharge) {
      if (m_SelectPositive){
    if (charge<0){
      ATH_MSG_DEBUG("Track fails charge cut: negative charge " << charge );
      return false;
    }
      }
      if (not m_SelectPositive){
    if (charge>0) {
      ATH_MSG_DEBUG("Track fails charge cut: positive charge " << charge);
      return false;
    }
      }
    }
 
    ATH_MSG_VERBOSE("== TrackSelection == Track N PIX hits: " << nPixHits << "  requested: "<< m_minPIXHits );
    ATH_MSG_VERBOSE("== TrackSelection == Track N SCT hits: " << nSCTHits << "  requested: "<< m_minSCTHits );
    ATH_MSG_VERBOSE("== TrackSelection == Track N TRT hits: " << nTRTHits << "  requested: "<< m_minTRTHits );
    ATH_MSG_VERBOSE("== TrackSelection == Track d0: " << d0 << "  requested: "<< m_maxd0 );
    ATH_MSG_VERBOSE("== TrackSelection == Track z0: " << z0 << "  requested: "<< m_maxz0 );
 
    if( nPixHits < m_minPIXHits ||
        nSCTHits < m_minSCTHits ||
        nTRTHits < m_minTRTHits ||
        std::abs(d0) > m_maxd0    ||
        std::abs(z0) > m_maxz0    ) {
      ATH_MSG_DEBUG("This track did not pass cuts --- nPixHits: " << nPixHits << " nSCTHits: " << nSCTHits <<
            " nTRTHits: " << nTRTHits << " idd0atIP: " << d0 << " idz0atIP: " << z0  );
     return false;
    }
 
    return true;
  }

printSummary()

void Trk::ConstrainedTrackProvider::printSummary ( )

virtual

Print statistical summary to logfile.

Reimplemented from Trk::ITrackCollectionProvider.

Definition at line 498 of file ConstrainedTrackProvider.cxx.

                                             {
 
    if(m_logStream) {
 
      *m_logStream<<"*************************************************************"<<std::endl;
      *m_logStream<<"******        ConstrainedTrackProvider Summary       ******"<<std::endl;
      *m_logStream<<"*"<<std::endl;
      *m_logStream<<"* number of combined muons failed in refit:                           " << m_constrainedTracks   << std::endl;
      *m_logStream<<"* number of combined muons succeeded in refit:                        " << m_unconstrainedTracks      << std::endl;
 
      *m_logStream<<"*"<<std::endl;
    }
  }

renounce()

std::enable_if_t< std::is_void_v< std::result_of_t< decltype(&T::renounce)(T)> > &&!std::is_base_of_v< SG::VarHandleKeyArray, T > &&std::is_base_of_v< Gaudi::DataHandle, T >, void > AthCommonDataStore< AthCommonMsg< AlgTool > >::renounce ( T & h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

  {
    h.renounce();
    PBASE::renounce (h);
  }

renounceArray()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::renounceArray ( SG::VarHandleKeyArray & handlesArray )

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

setLogStream()

virtual void Trk::ITrackCollectionProvider::setLogStream ( std::ostream * os )

inlinevirtualinherited

sets the output stream for the logfile

Definition at line 40 of file ITrackCollectionProvider.h.

{ m_logStream = os; }

setNtuple()

virtual void Trk::ITrackCollectionProvider::setNtuple ( TFile * )

inlinevirtualinherited

sets ntuple

Reimplemented in Trk::MSConstraintTracksProvider.

Definition at line 43 of file ITrackCollectionProvider.h.

{}

sysInitialize()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysInitialize ( )

overridevirtualinherited

Perform system initialization for an algorithm.

We override this to declare all the elements of handle key arrays at the end of initialization. See comments on updateVHKA.

Reimplemented in asg::AsgMetadataTool, AthCheckedComponent< AthAlgTool >, AthCheckedComponent<::AthAlgTool >, and DerivationFramework::CfAthAlgTool.

sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< AlgTool > >::sysStart ( )

overridevirtualinherited

Handle START transition.

We override this in order to make sure that conditions handle keys can cache a pointer to the conditions container.

trackCollection()

StatusCode Trk::ConstrainedTrackProvider::trackCollection ( const TrackCollection *& tracks )

virtual

Implements Trk::ITrackCollectionProvider.

Definition at line 135 of file ConstrainedTrackProvider.cxx.

  {
    SG::ReadHandle<TrackCollection> originalTracks(m_inputKey);
    if (!originalTracks.isValid()) {
      ATH_MSG_WARNING(" Can't retrieve " << m_inputKey);
      finalTracks = nullptr; // return empty collection
      return StatusCode::FAILURE;
    }
 
    SG::WriteHandle<TrackCollection> trackCollection(m_outputKey);
    ATH_CHECK(trackCollection.record(std::make_unique<TrackCollection>()));
    if (!trackCollection.isValid()) {
      ATH_MSG_ERROR("Could not record output TrackCollection " << trackCollection.name() << " to store " << trackCollection.store());
      return StatusCode::FAILURE;
    }
 
    ATH_MSG_DEBUG("Input track collection size:  " << originalTracks->size());
 
    int trackCount = 0;
    for (const Trk::Track* track : *originalTracks) {
      trackCount++;
      ATH_MSG_DEBUG("Dealing with track:  " << trackCount << " / " << originalTracks->size());
 
      bool acceptedTrack = true;
      if (m_doTrackSelection) acceptedTrack = passTrackSelection( track );
 
      if( acceptedTrack ){
    ATH_MSG_DEBUG("Track selection OK for track: " << trackCount);
        const Trk::Perigee* measuredPerigee =  track->perigeeParameters();
        if(!measuredPerigee){
          ATH_MSG_DEBUG(" no measuredPerigee");
          continue;
        }
 
        if(!(measuredPerigee->covariance()) ){
          ATH_MSG_DEBUG(" no measuredPerigee");
          continue;
        }
 
        const Trk::Surface* surf = dynamic_cast<const Trk::Surface*>( &(measuredPerigee->associatedSurface()) ) ;
        if( !surf ) {
          ATH_MSG_DEBUG("NO surface of the measuredPerigee");
          continue;
        }
 
        // Get corrections from histograms if they are requested
 
        double correctedQoverP(0), correctedQoverPError(0);
        if(m_CorrectMomentum){
          getCorrectedValues_P( measuredPerigee, correctedQoverP, correctedQoverPError);
        } else {
          correctedQoverP = measuredPerigee->parameters()[Trk::qOverP];
          correctedQoverPError = 1e6;
        }
 
        double correctedD0(0), correctedD0Error(0);
        if(m_CorrectD0){
          getCorrectedValues_d0( measuredPerigee, correctedD0, correctedD0Error);
        } else {
          correctedD0 = measuredPerigee->parameters()[Trk::d0];
          correctedD0Error = 1e6;
        }
 
        double correctedZ0(0), correctedZ0Error(0);
        if(m_CorrectZ0){
          getCorrectedValues_z0( measuredPerigee, correctedZ0, correctedZ0Error);
        } else {
          correctedZ0 = measuredPerigee->parameters()[Trk::z0];
          correctedZ0Error = 1e6;
        }
 
        // Construct pseudomeasurement
        Trk::DefinedParameter constrainedD0( correctedD0     , Trk::d0) ;
        Trk::DefinedParameter constrainedZ0( correctedZ0     , Trk::z0) ;
        Trk::DefinedParameter constrainedqOverP( correctedQoverP , Trk::qOverP) ;
 
        std::vector<Trk::DefinedParameter> constrainedPars;
        constrainedPars.push_back( constrainedD0 ) ;
        constrainedPars.push_back( constrainedZ0 ) ;
        constrainedPars.push_back( constrainedqOverP ) ;
 
        Amg::MatrixX constrainedCov( 3,3 ) ;
          constrainedCov.setZero();
        constrainedCov( 0, 0 )   =  correctedD0Error;
        constrainedCov( 1, 1 )   =  correctedZ0Error;
        constrainedCov( 2, 2 )   =  correctedQoverPError;
        // constrainedCov( 1, 0 )   =  0;  //initially we will assume the constraints are independant
        // constrainedCov( 2, 0 )   =  0;
        // constrainedCov( 2, 1 )   =  0;
 
        Trk::PseudoMeasurementOnTrack *pmot = new Trk::PseudoMeasurementOnTrack( Trk::LocalParameters( constrainedPars ),
                                                                                 std::move(constrainedCov), *surf) ;
 
 
        // Add pseusdo measurement to a list of hits to be fit;
        std::vector<const Trk::MeasurementBase*> vecOfMB;
        if (pmot) vecOfMB.push_back(pmot);
 
        // Add the remaining hits;
        DataVector<const Trk::MeasurementBase>::const_iterator it    = track->measurementsOnTrack()->begin();
        DataVector<const Trk::MeasurementBase>::const_iterator itend = track->measurementsOnTrack()->end();
        for (;it!=itend;++it)
          vecOfMB.push_back(*it);
 
        if(m_runOutlierRemoval){
          it    = track->outliersOnTrack()->begin();
          itend = track->outliersOnTrack()->end();
          // Add the remaining hits;
          for (;it!=itend;++it)
            vecOfMB.push_back(*it);
        }
 
        Trk::Track* constrainedFittedTrack = m_trackFitter->fit( Gaudi::Hive::currentContext(),
                                                                 vecOfMB, *measuredPerigee,
                                                                 m_runOutlierRemoval, Trk::pion).release();
        delete pmot;
 
        if (constrainedFittedTrack){
          ATH_MSG_DEBUG("Fit was successful");
          const Trk::Perigee* constrainedPerigee =  constrainedFittedTrack->perigeeParameters();
          if(!constrainedPerigee){
            ATH_MSG_DEBUG(" no constrainedmeasuredPerigee");
          } else {
            ATH_MSG_DEBUG("Initial track parameters --> momentum: " << 1/measuredPerigee->parameters()[Trk::qOverP] * 1e-3
              << "  pt: " << 1/measuredPerigee->parameters()[Trk::qOverP] * 1e-3 * sin(measuredPerigee->parameters()[Trk::theta])
              << "  d0: " << measuredPerigee->parameters()[Trk::d0]
              << "  z0: " << measuredPerigee->parameters()[Trk::z0] );
            ATH_MSG_DEBUG("constraint values  -->       momentum: " << 1/correctedQoverP * 1e-3
              << "  pt: " << 1/correctedQoverP * 1e-3 * sin(measuredPerigee->parameters()[Trk::theta])
              << "  d0: " << correctedD0
              << "  z0: " << correctedZ0 );
            ATH_MSG_DEBUG("Constrained fit result -->   momentum: " << 1/constrainedPerigee->parameters()[Trk::qOverP] * 1e-3
              << "  pt: " << 1/constrainedPerigee->parameters()[Trk::qOverP] * 1e-3 * sin(constrainedPerigee->parameters()[Trk::theta])
              << "  d0: " << constrainedPerigee->parameters()[Trk::d0]
              << "  z0: " << constrainedPerigee->parameters()[Trk::z0] );
          }
 
      constrainedFittedTrack->setTrackSummary( std::make_unique<Trk::TrackSummary> (*track->trackSummary()) );
 
          trackCollection->push_back(constrainedFittedTrack);
          ++m_constrainedTracks;
        } else{
          ATH_MSG_DEBUG("Constrained fit was unsuccessful");
        }
      } //--> if ( acceptedTrack)
      else {
 
        if(m_useConstrainedTrkOnly) continue;
 
        //Prob could just copy the track?
        Trk::Track* unconstrainedFittedTrack =
          m_trackFitter
            ->fit(Gaudi::Hive::currentContext(),
                  *track,
                  m_runOutlierRemoval,
                  Trk::pion)
            .release();
 
        if(unconstrainedFittedTrack) {
          ATH_MSG_DEBUG("Unconstrained fit was successful");
          trackCollection->push_back(unconstrainedFittedTrack);
          ++m_unconstrainedTracks;
        } else {
          ATH_MSG_DEBUG("Unconstrained fit was unsuccessful");
        }
      }
    }
 
    ATH_MSG_INFO ("Size of the trackCollection in this event : " << trackCollection->size() );
    finalTracks = trackCollection.cptr();
    return StatusCode::SUCCESS;
 
  }

updateVHKA()

void AthCommonDataStore< AthCommonMsg< AlgTool > >::updateVHKA ( Gaudi::Details::PropertyBase & )

inlineinherited

Definition at line 308 of file AthCommonDataStore.h.

                                               {
    // debug() << "updateVHKA for property " << p.name() << " " << p.toString() 
    //         << "  size: " << m_vhka.size() << endmsg;
    for (auto &a : m_vhka) {
      std::vector<SG::VarHandleKey*> keys = a->keys();
      for (auto k : keys) {
        k->setOwner(this);
      }
    }
  }

Member Data Documentation

m_constrainedTracks

int Trk::ConstrainedTrackProvider::m_constrainedTracks = 0

private

Definition at line 61 of file ConstrainedTrackProvider.h.

m_constraintFileName_d0

StringProperty Trk::ConstrainedTrackProvider::m_constraintFileName_d0

private

Initial value:

{
      this, "d0ConstraintFileName", "Constraint.root"}

Definition at line 93 of file ConstrainedTrackProvider.h.

                                          {
      this, "d0ConstraintFileName", "Constraint.root"};

m_constraintFileName_P

StringProperty Trk::ConstrainedTrackProvider::m_constraintFileName_P

private

Initial value:

{
      this, "MomentumConstraintFileName", "Constraint.root"}

Definition at line 85 of file ConstrainedTrackProvider.h.

                                         {
      this, "MomentumConstraintFileName", "Constraint.root"};

m_constraintFileName_z0

StringProperty Trk::ConstrainedTrackProvider::m_constraintFileName_z0

private

Initial value:

{
      this, "z0ConstraintFileName", "Constraint.root"}

Definition at line 101 of file ConstrainedTrackProvider.h.

                                          {
      this, "z0ConstraintFileName", "Constraint.root"};

m_constraintHistName_d0

StringProperty Trk::ConstrainedTrackProvider::m_constraintHistName_d0 {this, "d0ConstraintHistName", "EtaPhiMap"}

private

Definition at line 96 of file ConstrainedTrackProvider.h.

{this, "d0ConstraintHistName", "EtaPhiMap"};

m_constraintHistName_P

StringProperty Trk::ConstrainedTrackProvider::m_constraintHistName_P

private

Initial value:

{
      this, "MomentumConstraintHistName", "EtaPhiMap"}

Definition at line 88 of file ConstrainedTrackProvider.h.

                                         {
      this, "MomentumConstraintHistName", "EtaPhiMap"};

m_constraintHistName_z0

StringProperty Trk::ConstrainedTrackProvider::m_constraintHistName_z0 {this, "z0ConstraintHistName", "EtaPhiMap"}

private

Definition at line 104 of file ConstrainedTrackProvider.h.

{this, "z0ConstraintHistName", "EtaPhiMap"};

m_constraintInputFile_d0

TFile* Trk::ConstrainedTrackProvider::m_constraintInputFile_d0 = nullptr

private

Definition at line 95 of file ConstrainedTrackProvider.h.

m_constraintInputFile_P

TFile* Trk::ConstrainedTrackProvider::m_constraintInputFile_P = nullptr

private

Definition at line 87 of file ConstrainedTrackProvider.h.

m_constraintInputFile_z0

TFile* Trk::ConstrainedTrackProvider::m_constraintInputFile_z0 = nullptr

private

Definition at line 103 of file ConstrainedTrackProvider.h.

m_CorrectD0

BooleanProperty Trk::ConstrainedTrackProvider::m_CorrectD0 {this, "CorrectD0", false}

private

Definition at line 108 of file ConstrainedTrackProvider.h.

{this, "CorrectD0", false};

m_CorrectMeanD0

BooleanProperty Trk::ConstrainedTrackProvider::m_CorrectMeanD0 {this, "CorrectMeanD0", false}

private

Definition at line 109 of file ConstrainedTrackProvider.h.

{this, "CorrectMeanD0", false};

m_CorrectMomentum

BooleanProperty Trk::ConstrainedTrackProvider::m_CorrectMomentum {this, "CorrectMomentum", true}

private

Definition at line 84 of file ConstrainedTrackProvider.h.

{this, "CorrectMomentum", true};

m_CorrectZ0

BooleanProperty Trk::ConstrainedTrackProvider::m_CorrectZ0 {this, "CorrectZ0", false}

private

Definition at line 100 of file ConstrainedTrackProvider.h.

{this, "CorrectZ0", false};

m_deltaScaling

DoubleProperty Trk::ConstrainedTrackProvider::m_deltaScaling {this, "DeltaScaling", 1.}

private

Definition at line 69 of file ConstrainedTrackProvider.h.

{this, "DeltaScaling", 1.};

m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

m_doTrackSelection

BooleanProperty Trk::ConstrainedTrackProvider::m_doTrackSelection {this, "doTrackSelection", true}

private

Definition at line 57 of file ConstrainedTrackProvider.h.

{this, "doTrackSelection", true};

m_etaphiMap_d0

TH2F* Trk::ConstrainedTrackProvider::m_etaphiMap_d0 = nullptr

private

Definition at line 97 of file ConstrainedTrackProvider.h.

m_etaphiMap_P

TH2F* Trk::ConstrainedTrackProvider::m_etaphiMap_P = nullptr

private

Definition at line 90 of file ConstrainedTrackProvider.h.

m_etaphiMap_z0

TH2F* Trk::ConstrainedTrackProvider::m_etaphiMap_z0 = nullptr

private

Definition at line 105 of file ConstrainedTrackProvider.h.

m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< AlgTool > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

m_inputKey

SG::ReadHandleKey<TrackCollection> Trk::ConstrainedTrackProvider::m_inputKey

private

Initial value:

{
      this, "InputTracksCollection", "Tracks"}

Definition at line 79 of file ConstrainedTrackProvider.h.

                                               {
      this, "InputTracksCollection", "Tracks"};

m_logStream

std::ostream* Trk::ITrackCollectionProvider::m_logStream = nullptr

protectedinherited

logfile output stream

Definition at line 53 of file ITrackCollectionProvider.h.

m_maxd0

DoubleProperty Trk::ConstrainedTrackProvider::m_maxd0 {this, "Maxd0", 500.}

private

Definition at line 74 of file ConstrainedTrackProvider.h.

{this, "Maxd0", 500.};

m_maxPt

DoubleProperty Trk::ConstrainedTrackProvider::m_maxPt {this, "MaxPt", 100.}

private

Definition at line 77 of file ConstrainedTrackProvider.h.

{this, "MaxPt", 100.};

m_maxz0

DoubleProperty Trk::ConstrainedTrackProvider::m_maxz0 {this, "Maxz0", 500.}

private

Definition at line 75 of file ConstrainedTrackProvider.h.

{this, "Maxz0", 500.};

m_minPIXHits

IntegerProperty Trk::ConstrainedTrackProvider::m_minPIXHits {this, "MinPIXHits", 0}

private

Definition at line 71 of file ConstrainedTrackProvider.h.

{this, "MinPIXHits", 0};

m_minPt

DoubleProperty Trk::ConstrainedTrackProvider::m_minPt {this, "MinPt", 15.}

private

Definition at line 76 of file ConstrainedTrackProvider.h.

{this, "MinPt", 15.};

m_minSCTHits

IntegerProperty Trk::ConstrainedTrackProvider::m_minSCTHits {this, "MinSCTHits", 0}

private

Definition at line 72 of file ConstrainedTrackProvider.h.

{this, "MinSCTHits", 0};

m_minTRTHits

IntegerProperty Trk::ConstrainedTrackProvider::m_minTRTHits {this, "MinTRTHits", 0}

private

Definition at line 73 of file ConstrainedTrackProvider.h.

{this, "MinTRTHits", 0};

m_outputKey

SG::WriteHandleKey<TrackCollection> Trk::ConstrainedTrackProvider::m_outputKey

private

Initial value:

{
      this, "OutputTracksCollection", "AlignmentConstrainedTracks"}

Definition at line 81 of file ConstrainedTrackProvider.h.

                                                 {
      this, "OutputTracksCollection", "AlignmentConstrainedTracks"};

m_reduceConstraintUncert_z0

DoubleProperty Trk::ConstrainedTrackProvider::m_reduceConstraintUncert_z0

private

Initial value:

{this, "ReduceConstraintUncert_z0", 1.,
      "Reduce the uncertainty on z0 track parameter constraint by this amount"}

Definition at line 67 of file ConstrainedTrackProvider.h.

                                              {this, "ReduceConstraintUncert_z0", 1.,
      "Reduce the uncertainty on z0 track parameter constraint by this amount"};

m_reduceConstraintUncertainty

DoubleProperty Trk::ConstrainedTrackProvider::m_reduceConstraintUncertainty

private

Initial value:

{
      this, "ReduceConstraintUncertainty", 1.,
      "Reduce the uncertainty on the track parmater constraint by this amount"}

Definition at line 64 of file ConstrainedTrackProvider.h.

                                                {
      this, "ReduceConstraintUncertainty", 1.,
      "Reduce the uncertainty on the track parmater constraint by this amount"};

m_runOutlierRemoval

Gaudi::Property<Trk::RunOutlierRemoval> Trk::ConstrainedTrackProvider::m_runOutlierRemoval

private

Initial value:

{
      this, "RunOutlierRemoval", true, "run outlier removal in the GX2 fitter"}

Definition at line 54 of file ConstrainedTrackProvider.h.

                                                           {
      this, "RunOutlierRemoval", true, "run outlier removal in the GX2 fitter"};

m_scalepmaptogev

BooleanProperty Trk::ConstrainedTrackProvider::m_scalepmaptogev {this, "ScalePMapToGeV", false}

private

Definition at line 56 of file ConstrainedTrackProvider.h.

{this, "ScalePMapToGeV", false};

m_SelectByCharge

BooleanProperty Trk::ConstrainedTrackProvider::m_SelectByCharge {this, "SelectByCharge", false}

private

Definition at line 111 of file ConstrainedTrackProvider.h.

{this, "SelectByCharge", false};

m_SelectPositive

BooleanProperty Trk::ConstrainedTrackProvider::m_SelectPositive {this, "SelectPositive", true}

private

Definition at line 112 of file ConstrainedTrackProvider.h.

{this, "SelectPositive", true};

m_trackFitter

ToolHandle<IGlobalTrackFitter> Trk::ConstrainedTrackProvider::m_trackFitter

private

Initial value:

{this, "TrackFitter",
      "Trk::GlobalChi2Fitter/InDetTrackFitter", "normal track fitter"}

Definition at line 51 of file ConstrainedTrackProvider.h.

                                                {this, "TrackFitter",
      "Trk::GlobalChi2Fitter/InDetTrackFitter", "normal track fitter"};

m_unconstrainedTracks

int Trk::ConstrainedTrackProvider::m_unconstrainedTracks = 0

private

Definition at line 62 of file ConstrainedTrackProvider.h.

m_useConstrainedTrkOnly

BooleanProperty Trk::ConstrainedTrackProvider::m_useConstrainedTrkOnly {this, "UseConstrainedTrkOnly", false}

private

Definition at line 58 of file ConstrainedTrackProvider.h.

{this, "UseConstrainedTrkOnly", false};

m_useConstraintError

BooleanProperty Trk::ConstrainedTrackProvider::m_useConstraintError {this, "UseConstraintError", true}

private

Definition at line 59 of file ConstrainedTrackProvider.h.

{this, "UseConstraintError", true};

m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< AlgTool > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

m_vhka

std::vector<SG::VarHandleKeyArray*> AthCommonDataStore< AthCommonMsg< AlgTool > >::m_vhka

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

---

The documentation for this class was generated from the following files:

• ConstrainedTrackProvider.h
• ConstrainedTrackProvider.cxx