InDet::TRT_SegmentsToTrack Class Reference

#include <TRT_SegmentsToTrack.h>

Inheritance diagram for InDet::TRT_SegmentsToTrack:

Collaboration diagram for InDet::TRT_SegmentsToTrack:

Public Member Functions
	TRT_SegmentsToTrack (const std::string &name, ISvcLocator *pSvcLocator)
	~TRT_SegmentsToTrack ()
StatusCode	initialize ()
StatusCode	execute ()
StatusCode	finalize ()
virtual StatusCode	sysInitialize () override
	Override sysInitialize. More...
virtual const DataObjIDColl &	extraOutputDeps () const override
	Return the list of extra output dependencies. More...
ServiceHandle< StoreGateSvc > &	evtStore ()
	The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc. More...
const ServiceHandle< StoreGateSvc > &	evtStore () const
	The standard StoreGateSvc (event store) Returns (kind of) a pointer to the StoreGateSvc. More...
const ServiceHandle< StoreGateSvc > &	detStore () const
	The standard StoreGateSvc/DetectorStore Returns (kind of) a pointer to the StoreGateSvc. More...
virtual StatusCode	sysStart () override
	Handle START transition. More...
virtual std::vector< Gaudi::DataHandle * >	inputHandles () const override
	Return this algorithm's input handles. More...
virtual std::vector< Gaudi::DataHandle * >	outputHandles () const override
	Return this algorithm's output handles. More...
Gaudi::Details::PropertyBase &	declareProperty (Gaudi::Property< T > &t)
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKey &hndl, const std::string &doc, const SG::VarHandleKeyType &)
	Declare a new Gaudi property. More...
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleBase &hndl, const std::string &doc, const SG::VarHandleType &)
	Declare a new Gaudi property. More...
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, SG::VarHandleKeyArray &hndArr, const std::string &doc, const SG::VarHandleKeyArrayType &)
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc, const SG::NotHandleType &)
	Declare a new Gaudi property. More...
Gaudi::Details::PropertyBase *	declareProperty (const std::string &name, T &property, const std::string &doc="none")
	Declare a new Gaudi property. More...
void	updateVHKA (Gaudi::Details::PropertyBase &)
MsgStream &	msg () const
MsgStream &	msg (const MSG::Level lvl) const
bool	msgLvl (const MSG::Level lvl) const

Protected Member Functions
void	renounceArray (SG::VarHandleKeyArray &handlesArray)
	remove all handles from I/O resolution More...
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. More...

Private Types
typedef ServiceHandle< StoreGateSvc >	StoreGateSvc_t

Private Member Functions
int	getNumberReal (const InDet::TRT_DriftCircle *, const EventContext &ctx) const
	Get the number of truth particles associated with this hit. More...
double	getRealFractionTRT (const Trk::Track *track, const EventContext &ctx) const
	Get the fraction of truth TRT hits on this Track. More...
double	getNoiseProbability (const Trk::Track *track) const
	Get the fraction of noise TRT hits on this Track. More...
int	nTRTHits (const Trk::Track *track) const
	Count number of TRT Hits on track. More...
int	nHTHits (const Trk::Track *track) const
	Count number of TRT HT Hits on track. More...
void	combineSegments (const EventContext &ctx) const
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyType &)
	specialization for handling Gaudi::Property<SG::VarHandleKey> More...
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleKeyArrayType &)
	specialization for handling Gaudi::Property<SG::VarHandleKeyArray> More...
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &hndl, const SG::VarHandleType &)
	specialization for handling Gaudi::Property<SG::VarHandleBase> More...
Gaudi::Details::PropertyBase &	declareGaudiProperty (Gaudi::Property< T > &t, const SG::NotHandleType &)
	specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray> More...

Private Attributes
SG::ReadHandleKey< Trk::SegmentCollection >	m_inputSegmentCollectionName {this,"InputSegmentsCollection","TrackSegments","RHK to retrieve input track collection"}
	Name of the TrackSegment Collection to read in. More...
SG::WriteHandleKey< TrackCollection >	m_outputTrackCollectionName {this,"OutputTrackCollection","SegmentTracks","WHK to store output tracks"}
	Name of the TrackCollection to write out. More...
ToolHandle< Trk::ITrackFitter >	m_trackFitter {this,"TrackFitter","Trk::KalmanFitter/TrkKalmanFitter",""}
	The TrackFitter. More...
ToolHandle< Trk::IExtrapolator >	m_extrapolator {this,"ExtrapolationTool","Trk::Extrapolator/InDetExtrapolator",""}
	The Extrapolator. More...
ToolHandle< Trk::IExtendedTrackSummaryTool >	m_trkSummaryTool {this, "SummaryTool","",""}
ToolHandle< Trk::IPRDtoTrackMapTool >	m_assoTool {this, "AssociationTool", "","" }
SG::WriteHandleKey< Trk::PRDtoTrackMap >	m_assoMapName {this,"AssociationMapName","",""}
	key for the PRDtoTrackMap to filled by the ambiguity score processor. More...
SG::ReadHandleKey< Trk::PRDtoTrackMap >	m_inputAssoMapName {this,"InputAssociationMapName","",""}
	key to be set to optionally store PRD to track association map More...
double	m_noiseCut
	All tracks with a TRT Noise fraction larger than this variable will be thrown away. More...
int	m_minTRTHits
	All tracks with less Hits (after the track fit) will be thrown away. More...
const AtlasDetectorID *	m_idHelper
const TRT_ID *	m_trtid
SG::ReadHandleKey< PRD_MultiTruthCollection >	m_multiTruthCollectionTRTName {this,"PRDTruthCollectionTRT","","RHK to retrieve TRT turth info"}
	Name of the TRT MultiTruthCollection. More...
int	m_nTracksReal
	Counter for real reconstructed Tracks. More...
int	m_nTracksFake
	Counter for fake reconstructed Track. More...
double	m_noiseratio
	average percentage of noise in real tracks More...
std::atomic< int >	m_events
	Event counter. More...
std::map< int, int >	m_MapReal
	Map of hits and real tracks. More...
std::map< int, int >	m_MapFake
	Map of hits and fake tracks. More...
bool	m_materialEffects
	Flag to switch on Material Effects in the Fitter. More...
bool	m_outlierRemoval
	Flag to switch on the outlier removal in the track fitter. More...
bool	m_combineSegments
	Try to combine segments from Barrel and Endcap. More...
SG::ReadHandleKey< Trk::SegmentCollection >	m_barrelSegments {this,"BarrelSegments","TRTBarrelSegments","RHK to retrieve barrel track segments"}
	Name of Barrel segment collection. More...
SG::ReadHandleKey< Trk::SegmentCollection >	m_endcapSegments {this,"EndcapSegments","TRTEndcapSegments","RHK to retrieve endcap track segments"}
	Name of Endcap segment collection. More...
SG::WriteHandleKey< TrackCollection >	m_BECCollectionName {this,"BarrelEndcapTracks","TRT_Barrel_EC","WHK to write tracks"}
	Name of the combined (TRT Barrel+EC) TrackCollection to write out. More...
std::atomic< int >	m_n_combined_fit
DataObjIDColl	m_extendedExtraObjects
StoreGateSvc_t	m_evtStore
	Pointer to StoreGate (event store by default) More...
StoreGateSvc_t	m_detStore
	Pointer to StoreGate (detector store by default) More...
std::vector< SG::VarHandleKeyArray * >	m_vhka
bool	m_varHandleArraysDeclared

Detailed Description

Definition at line 53 of file TRT_SegmentsToTrack.h.

Member Typedef Documentation

StoreGateSvc_t

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

privateinherited

Definition at line 388 of file AthCommonDataStore.h.

Constructor & Destructor Documentation

TRT_SegmentsToTrack()

InDet::TRT_SegmentsToTrack::TRT_SegmentsToTrack	(	const std::string &	name,
		ISvcLocator *	pSvcLocator
	)

Definition at line 55 of file TRT_SegmentsToTrack.cxx.

                                                                                            :
  AthAlgorithm(name, pSvcLocator),
  m_idHelper(nullptr),
  m_trtid(nullptr),
  m_nTracksReal(0),
  m_nTracksFake(0),
  m_noiseratio(0.),
  m_events(0),
  m_n_combined_fit(0)
{
  declareProperty("NoiseCut",                m_noiseCut                    = -1.);
  declareProperty("MinNHit",                 m_minTRTHits                  =  1);
  declareProperty("MaterialEffects",         m_materialEffects             = false);
  declareProperty("OutlierRemoval",          m_outlierRemoval              = false);
  declareProperty("CombineSegments",         m_combineSegments             = false);
}

~TRT_SegmentsToTrack()

InDet::TRT_SegmentsToTrack::~TRT_SegmentsToTrack

(

)

Definition at line 72 of file TRT_SegmentsToTrack.cxx.

{
}

Member Function Documentation

combineSegments()

void InDet::TRT_SegmentsToTrack::combineSegments ( const EventContext &  ctx ) const

private

Definition at line 573 of file TRT_SegmentsToTrack.cxx.

{
  //Idea:
  // - get from endcap segment z-phi dependence
  // - get from barrel segment r-phi dependence and z dependence from endcap segment fit
 
  int n_combined_fit=0;
  SG::ReadHandle<Trk::SegmentCollection> BarrelSegments(m_barrelSegments,ctx);
  SG::ReadHandle<Trk::SegmentCollection> EndcapSegments(m_endcapSegments,ctx);
 
  StatusCode sc;
 
  SG::WriteHandle<TrackCollection> outputCombiCollection(m_BECCollectionName,ctx);
  sc = outputCombiCollection.record(std::make_unique<TrackCollection>());
  if (sc.isFailure()) return;
 
  if (!BarrelSegments.isValid()){
    ATH_MSG_FATAL("Could not open barrel segments collection : " <<   m_barrelSegments.key());
    return;
  }
 
  if (!EndcapSegments.isValid()) {
    ATH_MSG_FATAL("Could not open endcap segments collection : " <<   m_endcapSegments.key());
    return;
  }
 
  ATH_MSG_VERBOSE("Got both barrel and endcap segment collections of size "<<BarrelSegments->size()<<" "<<EndcapSegments->size());
 
  Trk::SegmentCollection::const_iterator iseg = BarrelSegments->begin();
  Trk::SegmentCollection::const_iterator isegE = BarrelSegments->end();
 
  int scount=0;
  for(;iseg!=isegE;++iseg,scount++){
    ATH_MSG_VERBOSE("Barrel Segment "<<scount<<" : phi="<<(*iseg)->localParameters()[Trk::phi]);
  }
 
  iseg  = EndcapSegments->begin();
  isegE = EndcapSegments->end();
 
  scount=0;
  for(;iseg!=isegE;++iseg,scount++){
    ATH_MSG_VERBOSE("Endcap Segment "<<scount<<" : phi="<<(*iseg)->localParameters()[Trk::phi]);
  }
 
  if(BarrelSegments->size()==1 && EndcapSegments->size()==1){
    ATH_MSG_VERBOSE("Here we go: one barrel segment and one endcap segment!");
 
    Trk::SegmentCollection::const_iterator isegBarrel = BarrelSegments->begin();
    Trk::SegmentCollection::const_iterator isegBarrelE = BarrelSegments->end();
 
    Trk::SegmentCollection::const_iterator isegEndcap = EndcapSegments->begin();
    Trk::SegmentCollection::const_iterator isegEndcapE = EndcapSegments->end();
 
    //loop over all Barrel Segments
    for(;isegBarrel!=isegBarrelE;++isegBarrel){
      for(;isegEndcap!=isegEndcapE;++isegEndcap){
 
    std::vector< Trk::PseudoMeasurementOnTrack*> tobedeleted;
 
    ATH_MSG_VERBOSE("Barrel Segment : phi="<<(*isegBarrel)->localParameters()[Trk::phi]<<" with "<<(*isegBarrel)->numberOfMeasurementBases()<<" hits");
    ATH_MSG_VERBOSE("Endcap Segment : phi="<<(*isegEndcap)->localParameters()[Trk::phi]<<" with "<<(*isegEndcap)->numberOfMeasurementBases()<<" hits");
 
    int count=0;
    Trk::MeasurementSet myset,myset2,echits=(*isegEndcap)->containedMeasurements();
 
        bool barreldown=((*isegBarrel)->measurement((*isegBarrel)->numberOfMeasurementBases()/2)->globalPosition().y()<0);
        double theta=0;
        if((*isegEndcap)->localParameters().contains(Trk::theta)){
          theta = (*isegEndcap)->localParameters()[Trk::theta];
        }
 
        //correct theta to point towards the barrel
        if(( (*isegEndcap)->globalPosition().y()>0 && (*isegEndcap)->globalPosition().z()>0) ||
           ((*isegEndcap)->globalPosition().y()<0 && (*isegEndcap)->globalPosition().z()<0)
           ){
          //endcap A top pointing to barrel A bottom or Barrel C top pointing to Endcap C bottom
          //--> Theta between 90 and 180 degrees
          if(theta<M_PI/2.) {
            theta=M_PI-theta;
            std::reverse(echits.begin(),echits.end());
          }
        }else{
          //Theta between 0 and 90 degrees
          if(theta>M_PI/2.) {
            theta=M_PI-theta;
            std::reverse(echits.begin(),echits.end());
          }
        }
 
    //decide if barrel or endcap hits come first depending on global position of segments
 
    if(!barreldown && (*isegEndcap)->globalPosition().y()>0){
      ATH_MSG_VERBOSE("Both barrel and endcap segments in top sectors, won't fit");
      continue;
    }
 
    if(barreldown && (*isegEndcap)->globalPosition().y()<0){
      ATH_MSG_VERBOSE("Both barrel and endcap segments in lower sectors, won't fit");
      continue;
    }
        int firstechit=0,lastechit=0;
    if((*isegEndcap)->globalPosition().y()>0){
      //first endcap then barrel
      for(int i=0;i<(int)echits.size();++i){
            const Trk::MeasurementBase *meas=echits[i];
        const Amg::VectorX LocalParameters = meas->localParameters();
        const Amg::MatrixX LocalErrorMatrix = meas->localCovariance();
        double z=meas->globalPosition().z();
        double phi=meas->globalPosition().phi();
        double r=meas->globalPosition().perp();
 
        const InDet::TRT_DriftCircleOnTrack* trtcirc = dynamic_cast<const InDet::TRT_DriftCircleOnTrack*>(meas);
        if(trtcirc){
          myset.push_back(meas);
          ATH_MSG_VERBOSE("Endcap : (phi,r,z) = ( "<<phi<<" , "<<r<<" , "<<z<<" ) ");
          count++;
        }
      }
      firstechit=0;
          lastechit=count-1;
 
      for(unsigned int i=0;i<(*isegBarrel)->numberOfMeasurementBases();++i){
        const Amg::VectorX LocalParameters = (*isegBarrel)->measurement(i)->localParameters();
        const Amg::MatrixX LocalErrorMatrix = (*isegBarrel)->measurement(i)->localCovariance();
        double z=(*isegBarrel)->measurement(i)->globalPosition().z();
        double phi=(*isegBarrel)->measurement(i)->globalPosition().phi();
        double r=(*isegBarrel)->measurement(i)->globalPosition().perp();
 
        const Trk::MeasurementBase *mesb=(*isegBarrel)->measurement(i);
        const InDet::TRT_DriftCircleOnTrack* trtcirc = dynamic_cast<const InDet::TRT_DriftCircleOnTrack*>(mesb);
        if(trtcirc){
          myset.push_back((*isegBarrel)->measurement(i));
          ATH_MSG_VERBOSE("Barrel : (phi,r,z) = ( "<<phi<<" , "<<r<<" , "<<z<<" ) ");
          count++;
        }
 
      }
    }else{
      //first barrel then endcap
      for(unsigned int i=0;i<(*isegBarrel)->numberOfMeasurementBases();++i){
        const Amg::VectorX LocalParameters = (*isegBarrel)->measurement(i)->localParameters();
        const Amg::MatrixX LocalErrorMatrix = (*isegBarrel)->measurement(i)->localCovariance();
        double z=(*isegBarrel)->measurement(i)->globalPosition().z();
        double phi=(*isegBarrel)->measurement(i)->globalPosition().phi();
        double r=(*isegBarrel)->measurement(i)->globalPosition().perp();
 
        const Trk::MeasurementBase *mesb=(*isegBarrel)->measurement(i);
        const InDet::TRT_DriftCircleOnTrack* trtcirc = dynamic_cast<const InDet::TRT_DriftCircleOnTrack*>(mesb);
        if(trtcirc){
          myset.push_back((*isegBarrel)->measurement(i));
          ATH_MSG_VERBOSE("Barrel : (phi,r,z) = ( "<<phi<<" , "<<r<<" , "<<z<<" ) ");
          count++;
        }
 
      }
          firstechit=count;
 
      for(int i=0;i<(int)echits.size();++i){
            const Trk::MeasurementBase *meas=echits[i];
        const Amg::VectorX LocalParameters = meas->localParameters();
        const Amg::MatrixX LocalErrorMatrix = meas->localCovariance();
        double z=meas->globalPosition().z();
        double phi=meas->globalPosition().phi();
        double r=meas->globalPosition().perp();
 
        const InDet::TRT_DriftCircleOnTrack* trtcirc = dynamic_cast<const InDet::TRT_DriftCircleOnTrack*>(meas);
        if(trtcirc){
          myset.push_back(meas);
          ATH_MSG_VERBOSE("Endcap : (phi,r,z) = ( "<<phi<<" , "<<r<<" , "<<z<<" ) ");
          count++;
        }
 
      }
          lastechit=count-1;
    }
 
 
 
    Amg::Vector3D inputMatchingPos((*isegBarrel)->globalPosition());
 
    if((*isegEndcap)->globalPosition().y()>0)
      inputMatchingPos=(*isegEndcap)->globalPosition();
 
    Amg::Vector3D inputMatchingMom;
    double p = 10e7;
 
    if((*isegBarrel)->localParameters().contains(Trk::qOverP)){
      p = ((*isegBarrel)->localParameters()[Trk::qOverP]!=0.) ? fabs(1./((*isegBarrel)->localParameters()[Trk::qOverP])) : 10e7;
    }
 
    double phi=0.;
    if((*isegBarrel)->localParameters().contains(Trk::phi)){
      phi = (*isegBarrel)->localParameters()[Trk::phi];
    }
 
 
    inputMatchingMom[Trk::px] = p * cos(phi) * sin(theta);
    inputMatchingMom[Trk::py] = p * sin(phi) * sin(theta);
    inputMatchingMom[Trk::pz] = p * cos(theta);
 
 
    ATH_MSG_VERBOSE("Global position: "<<inputMatchingPos<<" Globalmomentum: "<<inputMatchingMom);
 
 
 
    //add two pseudomeasurements at the beginning and at the end of the track
    Trk::MeasurementSet::const_iterator mit;
    Trk::MeasurementSet::const_iterator mitE=myset.end();
 
    int count2=0;
    for(mit=myset.begin();mit!=mitE;++mit){
      myset2.push_back(*mit);
      if(count2==firstechit || count2==lastechit){
        const InDet::TRT_DriftCircleOnTrack* trtcirc = dynamic_cast<const InDet::TRT_DriftCircleOnTrack*>(*mit);
        if(trtcirc){
          const Trk::StraightLineSurface *line=dynamic_cast<const
        Trk::StraightLineSurface*>(&trtcirc->associatedSurface());
          if (line) {
        //add pseudomeasurement
        double locz=-200.;
        if ((count2==firstechit && firstechit!=0) || (count2==lastechit && firstechit==0)) locz=200;
        Trk::DefinedParameter dp(locz,Trk::locZ);
        Trk::LocalParameters par(dp);
 
        Amg::MatrixX cov(1,1);
        cov(0,0) = 16./12. ; // take actual length of straw instead !!!!!
 
        //create new surface
        Amg::Vector3D C = line->transform().translation();
 
        //decide on movement
 
        const Amg::Vector3D& gpos=(*mit)->globalPosition();
 
        if(fabs(gpos.z())>800){
          if(theta>M_PI/2.){
            C[2]-=1.;
          }else{
            C[2]+=1.;
          }
        }else{
          C[1]-=1.;
        }
 
 
        Amg::Transform3D T;
        T = line->transform().rotation();
        T *= Amg::Translation3D(C.x(),C.y(),C.z());
        Trk::StraightLineSurface* surface = new Trk::StraightLineSurface(T);
 
        Trk::PseudoMeasurementOnTrack *pseudo=new Trk::PseudoMeasurementOnTrack( std::move(par),
                                                                             std::move(cov),
                                                                             *surface);
 
        tobedeleted.push_back(pseudo);
 
        delete surface;
 
        myset2.push_back(pseudo);
          }
        }
      }
      count2++;
    }
 
 
 
    const Trk::StraightLineSurface* testSf;
    if((*isegEndcap)->globalPosition().y()>0){
      testSf= dynamic_cast<const Trk::StraightLineSurface*>(&((*isegEndcap)->associatedSurface()));
    }else{
      testSf= dynamic_cast<const Trk::StraightLineSurface*>(&((*isegBarrel)->associatedSurface()));
    }
 
    const Trk::AtaStraightLine* inputMatchLine = nullptr;
    const Trk::Perigee* inputMatchPerigee = nullptr;
 
    if(!testSf){
      ATH_MSG_VERBOSE("No straightLineSurface !! Trying Perigee ...");
 
      const Trk::PerigeeSurface *testPSf=dynamic_cast<const Trk::PerigeeSurface*>(&((*isegBarrel)->associatedSurface()));
 
      if(!testPSf){
        ATH_MSG_VERBOSE("Associated surface dynamic_cast into PerigeeSurface failed. "<<(*isegBarrel)->associatedSurface());
        ATH_MSG_VERBOSE("Leaving input matching perigee as nullptr, will not get a fittedTrack");
      }else{
        ATH_MSG_VERBOSE("Ok, it seems to be a PerigeeSurface");
        inputMatchPerigee = new Trk::Perigee(inputMatchingPos,inputMatchingMom, 1., *testPSf);
      }
 
    }else{
      inputMatchLine = new Trk::AtaStraightLine(inputMatchingPos,inputMatchingMom, 1., *testSf);
      ATH_MSG_VERBOSE("Created testSf : " << (*inputMatchLine));
    }
 
    Amg::Vector3D startMomentum( 0., 0., 1.);
        std::unique_ptr<Trk::Track> fittedTrack;
    if(m_materialEffects){
      if(inputMatchLine){
            fittedTrack=m_trackFitter->fit(ctx,
                                           myset2,
                       *inputMatchLine,
                       false,
                       Trk::muon
                       );
      }else if (inputMatchPerigee){
            fittedTrack=m_trackFitter->fit(ctx,
                                           myset2,
                       *inputMatchPerigee,
                       false,
                       Trk::muon
                       );
      }
    }else{
      if(inputMatchLine){
            fittedTrack=m_trackFitter->fit(ctx,
                                           myset2,
                       *inputMatchLine,
                       false,
                       Trk::nonInteracting
                       );
      }else if (inputMatchPerigee){
            fittedTrack=m_trackFitter->fit(ctx,
                                           myset2,
                       *inputMatchPerigee,
                       false,
                       Trk::muon
                       );
      }
    }
 
    if(fittedTrack){
      n_combined_fit++;
      ATH_MSG_DEBUG("Successful Barrel+Endcap fit of segment. ");
      ATH_MSG_DEBUG("Quality of Track:   "<<fittedTrack->fitQuality()->chiSquared()<<" / "<<fittedTrack->fitQuality()->numberDoF());
      ATH_MSG_VERBOSE(*fittedTrack);
      outputCombiCollection->push_back(std::move(fittedTrack));
}
 
    delete inputMatchPerigee;
    delete inputMatchLine;
 
    for(size_t i=0;i<tobedeleted.size();i++)
      delete tobedeleted[i];
    tobedeleted.clear();
      }
    }
  }
  m_n_combined_fit += n_combined_fit;
 
  if (!outputCombiCollection.isValid()) {
    ATH_MSG_ERROR("Could not write track Barrel+EC collection TRT_Barrel_EC");
  }
}

declareGaudiProperty() [1/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty ( Gaudi::Property< T > &  hndl, const SG::VarHandleKeyArrayType &    )

inlineprivateinherited

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

Definition at line 170 of file AthCommonDataStore.h.

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

declareGaudiProperty() [2/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty ( Gaudi::Property< T > &  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());
 
  }

declareGaudiProperty() [3/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty ( Gaudi::Property< T > &  hndl, const SG::VarHandleType &    )

inlineprivateinherited

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

Definition at line 184 of file AthCommonDataStore.h.

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

declareGaudiProperty() [4/4]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareGaudiProperty ( Gaudi::Property< T > &  t, const SG::NotHandleType &    )

inlineprivateinherited

specialization for handling everything that's not a Gaudi::Property<SG::VarHandleKey> or a <SG::VarHandleKeyArray>

Definition at line 199 of file AthCommonDataStore.h.

  {
    return PBASE::declareProperty(t);
  }

declareProperty() [1/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( const std::string &  name, SG::VarHandleBase &  hndl, const std::string &  doc, const SG::VarHandleType &    )

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleBase. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 245 of file AthCommonDataStore.h.

  {
    this->declare(hndl.vhKey());
    hndl.vhKey().setOwner(this);
 
    return PBASE::declareProperty(name,hndl,doc);
  }

declareProperty() [2/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( const std::string &  name, SG::VarHandleKey &  hndl, const std::string &  doc, const SG::VarHandleKeyType &    )

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
hndl	Object holding the property value.
doc	Documentation string for the property.

This is the version for types that derive from SG::VarHandleKey. The property value object is put on the input and output lists as appropriate; then we forward to the base class.

Definition at line 221 of file AthCommonDataStore.h.

  {
    this->declare(hndl);
    hndl.setOwner(this);
 
    return PBASE::declareProperty(name,hndl,doc);
  }

declareProperty() [3/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( const std::string &  name, SG::VarHandleKeyArray &  hndArr, const std::string &  doc, const SG::VarHandleKeyArrayType &    )

inlineinherited

Definition at line 259 of file AthCommonDataStore.h.

  {
 
    // std::ostringstream ost;
    // ost << Algorithm::name() << " VHKA declareProp: " << name 
    //     << " size: " << hndArr.keys().size() 
    //     << " mode: " << hndArr.mode() 
    //     << "  vhka size: " << m_vhka.size()
    //     << "\n";
    // debug() << ost.str() << endmsg;
 
    hndArr.setOwner(this);
    m_vhka.push_back(&hndArr);
 
    Gaudi::Details::PropertyBase* p =  PBASE::declareProperty(name, hndArr, doc);
    if (p != 0) {
      p->declareUpdateHandler(&AthCommonDataStore<PBASE>::updateVHKA, this);
    } else {
      ATH_MSG_ERROR("unable to call declareProperty on VarHandleKeyArray " 
                    << name);
    }
 
    return p;
 
  }

declareProperty() [4/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( const std::string &  name, T &  property, const std::string &  doc, const SG::NotHandleType &    )

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This is the generic version, for types that do not derive from SG::VarHandleKey. It just forwards to the base class version of declareProperty.

Definition at line 333 of file AthCommonDataStore.h.

  {
    return PBASE::declareProperty(name, property, doc);
  }

declareProperty() [5/6]

Gaudi::Details::PropertyBase* AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( const std::string &  name, T &  property, const std::string &  doc = "none"  )

inlineinherited

Declare a new Gaudi property.

Parameters

name	Name of the property.
property	Object holding the property value.
doc	Documentation string for the property.

This dispatches to either the generic declareProperty or the one for VarHandle/Key/KeyArray.

Definition at line 352 of file AthCommonDataStore.h.

  {
    typedef typename SG::HandleClassifier<T>::type htype;
    return declareProperty (name, property, doc, htype());
  }

declareProperty() [6/6]

Gaudi::Details::PropertyBase& AthCommonDataStore< AthCommonMsg< Algorithm > >::declareProperty ( Gaudi::Property< T > &  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< Algorithm > >::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() [1/2]

ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< Algorithm > >::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; }

evtStore() [2/2]

const ServiceHandle<StoreGateSvc>& AthCommonDataStore< AthCommonMsg< Algorithm > >::evtStore ( ) const

inlineinherited

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

Definition at line 90 of file AthCommonDataStore.h.

{ return m_evtStore; }

execute()

StatusCode InDet::TRT_SegmentsToTrack::execute

(

)

Definition at line 137 of file TRT_SegmentsToTrack.cxx.

{
  int segmentCounter=0;
  const EventContext& ctx = Gaudi::Hive::currentContext();
  m_events++;
 
  ATH_MSG_DEBUG(name() << " execute() start");
 
  StatusCode sc                          = StatusCode::SUCCESS;
 
  //combine segments
  if( m_combineSegments)
    combineSegments(ctx);
 
  // input TrackSegment Collection
 
  SG::ReadHandle<Trk::SegmentCollection> inputSegments(m_inputSegmentCollectionName,ctx);
  if (!inputSegments.isValid()) {
    ATH_MSG_ERROR("Could not open: " <<m_inputSegmentCollectionName.key());
    sc =  StatusCode::FAILURE;
    return sc;
  }
 
 
  //output collections for fitted tracks
 
  SG::WriteHandle<TrackCollection> final_outputTrackCollection(m_outputTrackCollectionName,ctx);
  ATH_CHECK(final_outputTrackCollection.record(std::make_unique<TrackCollection>()));
 
  std::vector<std::unique_ptr<Trk::Track> > output_track_collection;
  //try to get truth information
 
  if(not m_multiTruthCollectionTRTName.empty()){
    SG::ReadHandle<PRD_MultiTruthCollection> truthCollectionTRT(m_multiTruthCollectionTRTName,ctx);
 
    if (!truthCollectionTRT.isValid()){
      ATH_MSG_FATAL("Could not open PRD_MultiTruthCollection : " <<  m_multiTruthCollectionTRTName.key());
      return StatusCode::FAILURE;
    }
  }
 
  Trk::SegmentCollection::const_iterator iseg = inputSegments->begin();
  Trk::SegmentCollection::const_iterator isegEnd = inputSegments->end();
 
 
 
 for(iseg=inputSegments->begin(); iseg != isegEnd; ++ iseg) {
 
    segmentCounter++;
 
 
    if((*iseg)->numberOfMeasurementBases()<10) continue; //only convert segments that are large enough
 
    Trk::MeasurementSet myset;
    for(unsigned int i=0;i<(*iseg)->numberOfMeasurementBases();++i){
      const Amg::VectorX& LocalParameters = (*iseg)->measurement(i)->localParameters();
      const Amg::MatrixX& LocalErrorMatrix = (*iseg)->measurement(i)->localCovariance();
      double z=(*iseg)->measurement(i)->globalPosition().z();
      ATH_MSG_DEBUG("Segment "<<segmentCounter<<" rioOnTrack "<<i<<" (z="<<z<<") : "<<LocalParameters[0]
                   <<" ; "<<std::sqrt(LocalErrorMatrix(Trk::locR,Trk::locR)));
      myset.push_back((*iseg)->measurement(i));
    } //end of loop over measurements
 
    if((*iseg)->numberOfMeasurementBases()>0){
      ATH_MSG_DEBUG("numberOfContainedRots: " << (*iseg)->numberOfMeasurementBases());
 
      const Trk::StraightLineSurface* testSf
        = dynamic_cast<const Trk::StraightLineSurface*>(&((*iseg)->associatedSurface()));
 
 
      const Trk::AtaStraightLine* inputMatchLine =nullptr;
      const Trk::Perigee* inputMatchPerigee =nullptr;
      const Amg::VectorX &p = dynamic_cast<const Amg::VectorX&>((**iseg).localParameters());
 
      if(!testSf){
    ATH_MSG_DEBUG("No straightLineSurface !! Trying Perigee ...");
    const Trk::PerigeeSurface *testPSf=dynamic_cast<const Trk::PerigeeSurface*>(&((*iseg)->associatedSurface()));
 
    if(!testPSf){
      ATH_MSG_DEBUG("Associated surface dynamic_cast into PerigeeSurface failed: "<<(*iseg)->associatedSurface());
      ATH_MSG_DEBUG("Leaving input matching perigee as nullptr, will not get a fittedTrack");
    }else{
      ATH_MSG_DEBUG("Ok, it seems to be a PerigeeSurface");
      inputMatchPerigee = new Trk::Perigee(p(0),p(1),p(2),p(3),p(4), *testPSf);
    }
      }else{
    inputMatchLine = new Trk::AtaStraightLine(p(0),p(1),p(2),p(3),p(4),*testSf);
 
    ATH_MSG_DEBUG("Created testSf : " << (*inputMatchLine));
        int nmeas=(*iseg)->numberOfMeasurementBases();
        Amg::Vector3D surfpos(.5*((*iseg)->measurement(nmeas/2)->globalPosition()+(*iseg)->measurement(nmeas/2+1)->globalPosition()));
        Trk::PerigeeSurface persurf(surfpos);
        std::unique_ptr<const Trk::TrackParameters> tmp =
          m_extrapolator->extrapolateDirectly(ctx, *inputMatchLine, persurf);
        if (tmp && tmp->associatedSurface().type() == Trk::SurfaceType::Perigee) {
          inputMatchPerigee = static_cast<const Trk::Perigee*>(tmp.release());
        }
      }
      ATH_MSG_DEBUG("Created inputMatchLine");
 
      std::unique_ptr<Trk::Track> fittedTrack;
      const Trk::TrackParameters *inputpar=nullptr;
      if (inputMatchPerigee) inputpar=inputMatchPerigee;
      else if (inputMatchLine) inputpar=inputMatchLine;
 
      if (inputpar) {
         fittedTrack = m_trackFitter->fit(ctx,
                                          myset,
                                          *inputpar,
                                          m_outlierRemoval,
                                          m_materialEffects ? Trk::muon : Trk::nonInteracting);
      }
      if(fittedTrack){
        Trk::TrackStates::const_iterator itSet = fittedTrack->trackStateOnSurfaces()->begin();
        Trk::TrackStates::const_iterator itSetEnd = fittedTrack->trackStateOnSurfaces()->end();
        const Trk::TrackParameters *measpar=nullptr;
        double mindist=9999;
        for ( ; itSet!=itSetEnd; ++itSet) {
          if ((**itSet).type(Trk::TrackStateOnSurface::Measurement) && (**itSet).trackParameters()->position().perp()<mindist) {
            mindist=(**itSet).trackParameters()->position().perp();
            measpar=(**itSet).trackParameters();
          }
        }
        std::unique_ptr<Trk::TrackParameters> myper;
        if (measpar){
          myper = m_extrapolator->extrapolate(ctx,*measpar,Trk::PerigeeSurface(),Trk::anyDirection,false, m_materialEffects ? Trk::muon : Trk::nonInteracting);
        }
        if (!myper){
          fittedTrack.reset();
        }
        else {
          auto trajectory = std::make_unique<Trk::TrackStates>();
          itSet = fittedTrack->trackStateOnSurfaces()->begin();
          for ( ; itSet!=itSetEnd; ++itSet) {
            if (!(**itSet).type(Trk::TrackStateOnSurface::Perigee)) {
              auto trackpar=(**itSet).trackParameters() ? (**itSet).trackParameters()->uniqueClone() : nullptr;
              auto measurement=(**itSet).measurementOnTrack() ? (**itSet).measurementOnTrack()->uniqueClone() : nullptr;
              auto fitQual=(**itSet).fitQualityOnSurface() ;
              auto mateff=(**itSet).materialEffectsOnTrack() ? (**itSet).materialEffectsOnTrack()->uniqueClone() : nullptr;
              std::bitset<Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes> typePattern(0);
              if ((**itSet).type(Trk::TrackStateOnSurface::Measurement)) typePattern.set(Trk::TrackStateOnSurface::Measurement);
              else if ((**itSet).type(Trk::TrackStateOnSurface::Outlier)) typePattern.set(Trk::TrackStateOnSurface::Outlier);
              else if ((**itSet).type(Trk::TrackStateOnSurface::Scatterer)) typePattern.set(Trk::TrackStateOnSurface::Scatterer);
              else if ((**itSet).type(Trk::TrackStateOnSurface::BremPoint)) typePattern.set(Trk::TrackStateOnSurface::BremPoint);
              trajectory->push_back(
                new Trk::TrackStateOnSurface(fitQual,
                                             std::move(measurement),
                                             std::move(trackpar),
                                             std::move(mateff),
                                             typePattern));
            }
          }
          bool peradded=false;
          itSet = trajectory->begin()+1;
          itSetEnd = trajectory->end();
          std::bitset<Trk::TrackStateOnSurface::NumberOfTrackStateOnSurfaceTypes> typePattern(0);
          typePattern.set(Trk::TrackStateOnSurface::Perigee);
          const auto myPosition {myper->position()};
          const auto myMomentum {myper->momentum()};
          const Trk::TrackStateOnSurface* pertsos =
            new Trk::TrackStateOnSurface(nullptr,
                                         std::move(myper),
                                         nullptr,
                                         typePattern);
 
          int index=1;
          for ( ; itSet!=itSetEnd; ++itSet) {
            double inprod1=((**itSet).trackParameters()->position()-myPosition).dot(myMomentum);
            --itSet;
            double inprod2=((**itSet).trackParameters()->position()-myPosition).dot(myMomentum);
            ++itSet;
            if (inprod1>0 && inprod2<0) {
              trajectory->insert(trajectory->begin()+index,pertsos);
              peradded=true;
              break;
            }
            index++;
          }
          if (!peradded){
            itSet = trajectory->begin();
            double inprod=((**itSet).trackParameters()->position()-myPosition).dot(myMomentum);
            if (inprod>0) trajectory->insert(trajectory->begin(),pertsos);
            else trajectory->push_back(pertsos);
          }
          std::unique_ptr<Trk::Track> track =
            std::make_unique<Trk::Track>(fittedTrack->info(),
                                         std::move(trajectory),
                                         fittedTrack->fitQuality()->uniqueClone());
          fittedTrack = std::move(track);
        }
      }
      if(fittedTrack){
 
    int nHT=nHTHits(fittedTrack.get());
    ATH_MSG_DEBUG("Number of HT hits: "<<nHT);
    ATH_MSG_DEBUG(" Successful fit of track.");
    if ((*iseg)->fitQuality()) ATH_MSG_DEBUG("Quality of Segment: chi^2 /ndof "<<(*iseg)->fitQuality()->chiSquared()<<" / "<<(*iseg)->fitQuality()->numberDoF());
    ATH_MSG_DEBUG("Quality of Track:  chi^2 /ndof "<<fittedTrack->fitQuality()->chiSquared()<<" / "<<fittedTrack->fitQuality()->numberDoF());
    ATH_MSG_DEBUG("Noise probability: "<<getNoiseProbability(fittedTrack.get()));
    ATH_MSG_DEBUG((*fittedTrack));
 
    if( nTRTHits(fittedTrack.get())>=m_minTRTHits){
          if(getNoiseProbability(fittedTrack.get())>m_noiseCut){
            double truefraction=getRealFractionTRT(fittedTrack.get(),ctx);
 
        int nhits=(*iseg)->numberOfMeasurementBases();
        ATH_MSG_DEBUG("Real/Noise : "<< truefraction << " chi2="<<fittedTrack->fitQuality()->chiSquared()/double(fittedTrack->fitQuality()->numberDoF()));
        if(truefraction>0.5){
          ATH_MSG_DEBUG("==> Real Track");
 
          if(m_MapReal.find(nhits)==m_MapReal.end()){
        m_MapReal[nhits]=1;
          }else{
        m_MapReal[nhits]++;
          }
 
          m_nTracksReal++;
          m_noiseratio+=(1-truefraction);
        }else{
          ATH_MSG_DEBUG(" ==> Fake Track");
 
          if(m_MapFake.find(nhits)==m_MapFake.end()){
        m_MapFake[nhits]=1;
          }else{
        m_MapFake[nhits]++;
          }
 
          m_nTracksFake++;
        }
      }
      output_track_collection.push_back(std::move(fittedTrack));
    }else{
          ATH_MSG_DEBUG("This tracks has only "<<nTRTHits(fittedTrack.get())<<" Hits! Will reject it!");
    }
 
      }else{
    ATH_MSG_DEBUG("Fit did not converge!");
      } //end of if(fittedTrack)
 
      delete inputMatchLine;
      delete inputMatchPerigee;
 
    } //end of if: (*iseg)->numberOfMeasurementBases()>0
  } //end of loop over segments
 
  std::unique_ptr<Trk::PRDtoTrackMap> prd_to_track_map;
  if (m_assoTool.isEnabled()) {
     // create internal PRD-to-track map
     prd_to_track_map = m_assoTool->createPRDtoTrackMap();
     if (!m_inputAssoMapName.key().empty()) {
        SG::ReadHandle<Trk::PRDtoTrackMap> input_asso_map(m_inputAssoMapName,ctx);
        if (!input_asso_map.isValid()) {
           ATH_MSG_FATAL("No valid input PRT-to-track map with key " << m_inputAssoMapName.key());
        }
        *prd_to_track_map = *input_asso_map;
        ATH_MSG_INFO("PRD to track map input " << m_inputAssoMapName.key()  );
     }
     for (const std::unique_ptr<Trk::Track> &track : output_track_collection) {
       ATH_CHECK( m_assoTool->addPRDs(*prd_to_track_map, *track) );
     }
  }
  // @TODO sort output track collection ?
  final_outputTrackCollection->reserve(output_track_collection.size());
  if (m_trkSummaryTool.isEnabled()) {
     for (std::unique_ptr<Trk::Track> &track : output_track_collection) {
        m_trkSummaryTool->computeAndReplaceTrackSummary(*track,prd_to_track_map.get());
        final_outputTrackCollection->push_back(std::move(track));
     }
  }
  else {
     for (std::unique_ptr<Trk::Track> &track : output_track_collection) {
        final_outputTrackCollection->push_back(std::move(track));
     }
  }
  if (!m_assoMapName.key().empty()) {
     ATH_CHECK(SG::WriteHandle<Trk::PRDtoTrackMap>(m_assoMapName,ctx).record(std::move(prd_to_track_map)));
  }
 
  if (!final_outputTrackCollection.isValid()) {
     // @TODO never reached (?)
    ATH_MSG_ERROR("Could not write track collection " << m_outputTrackCollectionName.key());
    return StatusCode::FAILURE;
    //return sc;
  }
 
  return sc;
}

extraDeps_update_handler()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::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

extraOutputDeps()

const DataObjIDColl & AthAlgorithm::extraOutputDeps ( ) const

overridevirtualinherited

Return the list of extra output dependencies.

This list is extended to include symlinks implied by inheritance relations.

Definition at line 50 of file AthAlgorithm.cxx.

{
  // If we didn't find any symlinks to add, just return the collection
  // from the base class.  Otherwise, return the extended collection.
  if (!m_extendedExtraObjects.empty()) {
    return m_extendedExtraObjects;
  }
  return Algorithm::extraOutputDeps();
}

finalize()

StatusCode InDet::TRT_SegmentsToTrack::finalize

(

)

Definition at line 111 of file TRT_SegmentsToTrack.cxx.

{
  ATH_MSG_INFO("Summary of" << m_events << " Events");
  ATH_MSG_INFO("Found Real Tracks : " << m_nTracksReal);
  ATH_MSG_INFO("Found Fake Tracks : " << m_nTracksFake);
 
  if(m_nTracksReal>0) {
    ATH_MSG_INFO("Average noise percentage " << m_noiseratio/double(m_nTracksReal));
  }
 
  for (const auto& mitr : m_MapReal) {
    ATH_MSG_INFO("Real tracks with " << mitr.first << " hits: " << mitr.second);
  }
  for (const auto& mitr : m_MapFake) {
    ATH_MSG_INFO("Fake tracks with " << mitr.first << " hits: " << mitr.second);
  }
 
  if(m_combineSegments){
    ATH_MSG_INFO("Number of combined Barrel+Endcap tracks: "<<m_n_combined_fit);
  }
 
  ATH_MSG_DEBUG(name() << " finalize() successful ");
 
  return StatusCode::SUCCESS;
}

getNoiseProbability()

double InDet::TRT_SegmentsToTrack::getNoiseProbability ( const Trk::Track *  track ) const

private

Get the fraction of noise TRT hits on this Track.

Definition at line 534 of file TRT_SegmentsToTrack.cxx.

{
  double fraction=0.;
  int nDriftReal=0;
  int nDriftNoise=0;
 
  //loop over the hits
  for (const Trk::TrackStateOnSurface* tsos : *track->trackStateOnSurfaces()) {
    const Trk::RIO_OnTrack* hitOnTrack = dynamic_cast <const Trk::RIO_OnTrack*>(tsos->measurementOnTrack());
    if (hitOnTrack) {
      const Identifier& surfaceID = hitOnTrack->identify();
 
      //take only TRT hits
      if(m_idHelper->is_trt(surfaceID)){
 
    const InDet::TRT_DriftCircleOnTrack *dcot= dynamic_cast <const InDet::TRT_DriftCircleOnTrack*>(hitOnTrack);
    if (dcot) {
      const InDet::TRT_DriftCircle *dc=dcot->prepRawData();
 
      if(dc){
 
        if(dc->isNoise())
          nDriftNoise++;
        else
          nDriftReal++;
      }
    }
      }
    }
  }
 
  if(nDriftReal>0)
    fraction=double(nDriftReal)/double(nDriftNoise+nDriftReal);
 
  return fraction;
}

getNumberReal()

int InDet::TRT_SegmentsToTrack::getNumberReal ( const InDet::TRT_DriftCircle *  driftcircle, const EventContext &  ctx  ) const

private

Get the number of truth particles associated with this hit.

Definition at line 425 of file TRT_SegmentsToTrack.cxx.

{
  int numBarcodes = 0;
  using iter = PRD_MultiTruthCollection::const_iterator;
 
  if(m_multiTruthCollectionTRTName.empty()) return 0;
 
  else{
    SG::ReadHandle<PRD_MultiTruthCollection> truthCollectionTRT(m_multiTruthCollectionTRTName,ctx);
    if(truthCollectionTRT.isValid()){
      std::pair<iter,iter> range = truthCollectionTRT->equal_range(driftcircle->identify());
      numBarcodes+=std::distance(range.first,range.second);
    }
  }
  return numBarcodes;
}

getRealFractionTRT()

double InDet::TRT_SegmentsToTrack::getRealFractionTRT ( const Trk::Track *  track, const EventContext &  ctx  ) const

private

Get the fraction of truth TRT hits on this Track.

Definition at line 444 of file TRT_SegmentsToTrack.cxx.

{
  double fraction=0.;
  int nDriftReal=0;
  int nDriftNoise=0;
 
  //loop over the hits
  for (const Trk::TrackStateOnSurface* tsos : *track->trackStateOnSurfaces()) {
 
    const Trk::RIO_OnTrack* hitOnTrack = dynamic_cast <const Trk::RIO_OnTrack*>(tsos->measurementOnTrack());
    if (hitOnTrack) {
      const Identifier& surfaceID = hitOnTrack->identify();
 
      //take only TRT hits
      if(m_idHelper->is_trt(surfaceID)){
 
    const InDet::TRT_DriftCircleOnTrack *dcot= dynamic_cast <const InDet::TRT_DriftCircleOnTrack*>(hitOnTrack);
    if(dcot) {
      const InDet::TRT_DriftCircle *dc=dcot->prepRawData();
 
      if(dc){
            int nreal=getNumberReal(dc,ctx);
 
        if(nreal>0) nDriftReal++;
        else nDriftNoise++;
 
      }
    }
      }
    }
  }
 
  if(nDriftReal>0)
    fraction=double(nDriftReal)/double(nDriftNoise+nDriftReal);
 
  return fraction;
}

initialize()

StatusCode InDet::TRT_SegmentsToTrack::initialize

(

)

Definition at line 76 of file TRT_SegmentsToTrack.cxx.

{
 
  ATH_MSG_DEBUG("TrkSegmenttoTrack initialize()");
 
  m_n_combined_fit=0;
 
  m_nTracksReal=0;
  m_nTracksFake=0;
  m_events=0;
  m_noiseratio=0.;
 
  // Initialize handle keys
  ATH_CHECK(m_inputSegmentCollectionName.initialize());
  ATH_CHECK(m_multiTruthCollectionTRTName.initialize(!m_multiTruthCollectionTRTName.empty()));
  ATH_CHECK(m_barrelSegments.initialize(m_combineSegments));
  ATH_CHECK(m_endcapSegments.initialize(m_combineSegments));
  ATH_CHECK(m_outputTrackCollectionName.initialize());
  ATH_CHECK(m_BECCollectionName.initialize());
 
  //--------- Set up fitter -------------
  CHECK( m_trackFitter.retrieve());
 
  CHECK(detStore()->retrieve(m_trtid));
 
  CHECK(detStore()->retrieve(m_idHelper, "AtlasID"));
 
  CHECK(m_extrapolator.retrieve());
  ATH_CHECK(m_trkSummaryTool.retrieve( DisableTool{ m_trkSummaryTool.empty() } ));
  ATH_CHECK(m_assoTool.retrieve( DisableTool{ m_assoTool.empty() || (m_trkSummaryTool.empty() && m_assoMapName.empty()) } ));
  ATH_CHECK(m_inputAssoMapName.initialize( !m_inputAssoMapName.empty() && m_assoTool.isEnabled()));
  ATH_CHECK(m_assoMapName.initialize( !m_assoMapName.empty() && m_assoTool.isEnabled()));
  return StatusCode::SUCCESS;
}

inputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< Algorithm > >::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.

msg() [1/2]

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

inlineinherited

Definition at line 24 of file AthCommonMsg.h.

                                {
    return this->msgStream();
  }

msg() [2/2]

MsgStream& AthCommonMsg< Algorithm >::msg ( const MSG::Level  lvl ) const

inlineinherited

Definition at line 27 of file AthCommonMsg.h.

                                                  {
    return this->msgStream(lvl);
  }

msgLvl()

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

inlineinherited

Definition at line 30 of file AthCommonMsg.h.

                                               {
    return this->msgLevel(lvl);
  }

nHTHits()

int InDet::TRT_SegmentsToTrack::nHTHits ( const Trk::Track *  track ) const

private

Count number of TRT HT Hits on track.

Definition at line 483 of file TRT_SegmentsToTrack.cxx.

{
 
  int nHT=0;
 
  //loop over the hits
  for (const Trk::TrackStateOnSurface* tsos : *track->trackStateOnSurfaces()) {
 
    const Trk::RIO_OnTrack* hitOnTrack = dynamic_cast <const Trk::RIO_OnTrack*>(tsos->measurementOnTrack());
    if (hitOnTrack != nullptr) {
      const Identifier& surfaceID = hitOnTrack->identify();
 
      //take only TRT hits
      if(m_idHelper->is_trt(surfaceID) && !tsos->type(Trk::TrackStateOnSurface::Outlier)){
        const InDet::TRT_DriftCircleOnTrack* trtcirc = dynamic_cast<const InDet::TRT_DriftCircleOnTrack*>(hitOnTrack);
        if(trtcirc) {
          if(trtcirc->highLevel()){
            nHT++;
          }
        }
      }
    }
  }
 
  return nHT;
}

nTRTHits()

int InDet::TRT_SegmentsToTrack::nTRTHits ( const Trk::Track *  track ) const

private

Count number of TRT Hits on track.

Definition at line 512 of file TRT_SegmentsToTrack.cxx.

{
  int nhits=0;
 
  //loop over the hits
  for (const Trk::TrackStateOnSurface* tsos : *track->trackStateOnSurfaces()) {
 
    const Trk::RIO_OnTrack* hitOnTrack = dynamic_cast <const Trk::RIO_OnTrack*>(tsos->measurementOnTrack());
    if (hitOnTrack != nullptr) {
      const Identifier& surfaceID = hitOnTrack->identify();
 
      //take only TRT hits
      if(m_idHelper->is_trt(surfaceID) && !tsos->type(Trk::TrackStateOnSurface::Outlier)){
    nhits++;
      }
    }
  }
 
  return nhits;
}

outputHandles()

virtual std::vector<Gaudi::DataHandle*> AthCommonDataStore< AthCommonMsg< Algorithm > >::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.

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< Algorithm > >::renounce ( T &  h )

inlineprotectedinherited

Definition at line 380 of file AthCommonDataStore.h.

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

renounceArray()

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

inlineprotectedinherited

remove all handles from I/O resolution

Definition at line 364 of file AthCommonDataStore.h.

                                                          {
    handlesArray.renounce();
  }

sysInitialize()

StatusCode AthAlgorithm::sysInitialize ( )

overridevirtualinherited

Override sysInitialize.

Override sysInitialize from the base class.

Loop through all output handles, and if they're WriteCondHandles, automatically register them and this Algorithm with the CondSvc

Scan through all outputHandles, and if they're WriteCondHandles, register them with the CondSvc

Reimplemented from AthCommonDataStore< AthCommonMsg< Algorithm > >.

Reimplemented in AthAnalysisAlgorithm, AthFilterAlgorithm, PyAthena::Alg, and AthHistogramAlgorithm.

Definition at line 66 of file AthAlgorithm.cxx.

                                       {
  StatusCode sc=AthCommonDataStore<AthCommonMsg<Algorithm>>::sysInitialize();
 
  if (sc.isFailure()) {
    return sc;
  }
  ServiceHandle<ICondSvc> cs("CondSvc",name());
  for (auto h : outputHandles()) {
    if (h->isCondition() && h->mode() == Gaudi::DataHandle::Writer) {
      // do this inside the loop so we don't create the CondSvc until needed
      if ( cs.retrieve().isFailure() ) {
        ATH_MSG_WARNING("no CondSvc found: won't autoreg WriteCondHandles");
        return StatusCode::SUCCESS;
      }
      if (cs->regHandle(this,*h).isFailure()) {
        sc = StatusCode::FAILURE;
        ATH_MSG_ERROR("unable to register WriteCondHandle " << h->fullKey()
                      << " with CondSvc");
      }
    }
  }
  return sc;
}

sysStart()

virtual StatusCode AthCommonDataStore< AthCommonMsg< Algorithm > >::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.

updateVHKA()

void AthCommonDataStore< AthCommonMsg< Algorithm > >::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_assoMapName

SG::WriteHandleKey<Trk::PRDtoTrackMap> InDet::TRT_SegmentsToTrack::m_assoMapName {this,"AssociationMapName","",""}

private

key for the PRDtoTrackMap to filled by the ambiguity score processor.

key to be set to optionally store PRD to track association map

Definition at line 93 of file TRT_SegmentsToTrack.h.

m_assoTool

ToolHandle<Trk::IPRDtoTrackMapTool> InDet::TRT_SegmentsToTrack::m_assoTool {this, "AssociationTool", "","" }

private

Definition at line 90 of file TRT_SegmentsToTrack.h.

m_barrelSegments

SG::ReadHandleKey<Trk::SegmentCollection> InDet::TRT_SegmentsToTrack::m_barrelSegments {this,"BarrelSegments","TRTBarrelSegments","RHK to retrieve barrel track segments"}

private

Name of Barrel segment collection.

Definition at line 124 of file TRT_SegmentsToTrack.h.

m_BECCollectionName

SG::WriteHandleKey<TrackCollection> InDet::TRT_SegmentsToTrack::m_BECCollectionName {this,"BarrelEndcapTracks","TRT_Barrel_EC","WHK to write tracks"}

private

Name of the combined (TRT Barrel+EC) TrackCollection to write out.

Definition at line 127 of file TRT_SegmentsToTrack.h.

m_combineSegments

bool InDet::TRT_SegmentsToTrack::m_combineSegments

private

Try to combine segments from Barrel and Endcap.

Definition at line 122 of file TRT_SegmentsToTrack.h.

m_detStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_detStore

privateinherited

Pointer to StoreGate (detector store by default)

Definition at line 393 of file AthCommonDataStore.h.

m_endcapSegments

SG::ReadHandleKey<Trk::SegmentCollection> InDet::TRT_SegmentsToTrack::m_endcapSegments {this,"EndcapSegments","TRTEndcapSegments","RHK to retrieve endcap track segments"}

private

Name of Endcap segment collection.

Definition at line 125 of file TRT_SegmentsToTrack.h.

m_events

std::atomic<int> InDet::TRT_SegmentsToTrack::m_events

mutableprivate

Event counter.

Definition at line 114 of file TRT_SegmentsToTrack.h.

m_evtStore

StoreGateSvc_t AthCommonDataStore< AthCommonMsg< Algorithm > >::m_evtStore

privateinherited

Pointer to StoreGate (event store by default)

Definition at line 390 of file AthCommonDataStore.h.

m_extendedExtraObjects

DataObjIDColl AthAlgorithm::m_extendedExtraObjects

privateinherited

Definition at line 79 of file AthAlgorithm.h.

m_extrapolator

ToolHandle<Trk::IExtrapolator> InDet::TRT_SegmentsToTrack::m_extrapolator {this,"ExtrapolationTool","Trk::Extrapolator/InDetExtrapolator",""}

private

The Extrapolator.

Definition at line 85 of file TRT_SegmentsToTrack.h.

m_idHelper

const AtlasDetectorID* InDet::TRT_SegmentsToTrack::m_idHelper

private

Definition at line 102 of file TRT_SegmentsToTrack.h.

m_inputAssoMapName

SG::ReadHandleKey<Trk::PRDtoTrackMap> InDet::TRT_SegmentsToTrack::m_inputAssoMapName {this,"InputAssociationMapName","",""}

private

key to be set to optionally store PRD to track association map

Definition at line 95 of file TRT_SegmentsToTrack.h.

m_inputSegmentCollectionName

SG::ReadHandleKey<Trk::SegmentCollection> InDet::TRT_SegmentsToTrack::m_inputSegmentCollectionName {this,"InputSegmentsCollection","TrackSegments","RHK to retrieve input track collection"}

private

Name of the TrackSegment Collection to read in.

Definition at line 78 of file TRT_SegmentsToTrack.h.

m_MapFake

std::map<int,int> InDet::TRT_SegmentsToTrack::m_MapFake

private

Map of hits and fake tracks.

Definition at line 116 of file TRT_SegmentsToTrack.h.

m_MapReal

std::map<int,int> InDet::TRT_SegmentsToTrack::m_MapReal

private

Map of hits and real tracks.

Definition at line 115 of file TRT_SegmentsToTrack.h.

m_materialEffects

bool InDet::TRT_SegmentsToTrack::m_materialEffects

private

Flag to switch on Material Effects in the Fitter.

Definition at line 118 of file TRT_SegmentsToTrack.h.

m_minTRTHits

int InDet::TRT_SegmentsToTrack::m_minTRTHits

private

All tracks with less Hits (after the track fit) will be thrown away.

Definition at line 100 of file TRT_SegmentsToTrack.h.

m_multiTruthCollectionTRTName

SG::ReadHandleKey<PRD_MultiTruthCollection> InDet::TRT_SegmentsToTrack::m_multiTruthCollectionTRTName {this,"PRDTruthCollectionTRT","","RHK to retrieve TRT turth info"}

private

Name of the TRT MultiTruthCollection.

Definition at line 107 of file TRT_SegmentsToTrack.h.

m_n_combined_fit

std::atomic<int> InDet::TRT_SegmentsToTrack::m_n_combined_fit

mutableprivate

Definition at line 129 of file TRT_SegmentsToTrack.h.

m_noiseCut

double InDet::TRT_SegmentsToTrack::m_noiseCut

private

All tracks with a TRT Noise fraction larger than this variable will be thrown away.

Definition at line 99 of file TRT_SegmentsToTrack.h.

m_noiseratio

double InDet::TRT_SegmentsToTrack::m_noiseratio

private

average percentage of noise in real tracks

Definition at line 112 of file TRT_SegmentsToTrack.h.

m_nTracksFake

int InDet::TRT_SegmentsToTrack::m_nTracksFake

private

Counter for fake reconstructed Track.

Definition at line 110 of file TRT_SegmentsToTrack.h.

m_nTracksReal

int InDet::TRT_SegmentsToTrack::m_nTracksReal

private

Counter for real reconstructed Tracks.

Definition at line 109 of file TRT_SegmentsToTrack.h.

m_outlierRemoval

bool InDet::TRT_SegmentsToTrack::m_outlierRemoval

private

Flag to switch on the outlier removal in the track fitter.

Definition at line 119 of file TRT_SegmentsToTrack.h.

m_outputTrackCollectionName

SG::WriteHandleKey<TrackCollection> InDet::TRT_SegmentsToTrack::m_outputTrackCollectionName {this,"OutputTrackCollection","SegmentTracks","WHK to store output tracks"}

private

Name of the TrackCollection to write out.

Definition at line 80 of file TRT_SegmentsToTrack.h.

m_trackFitter

ToolHandle<Trk::ITrackFitter> InDet::TRT_SegmentsToTrack::m_trackFitter {this,"TrackFitter","Trk::KalmanFitter/TrkKalmanFitter",""}

private

The TrackFitter.

Definition at line 82 of file TRT_SegmentsToTrack.h.

m_trkSummaryTool

ToolHandle<Trk::IExtendedTrackSummaryTool> InDet::TRT_SegmentsToTrack::m_trkSummaryTool {this, "SummaryTool","",""}

private

Definition at line 88 of file TRT_SegmentsToTrack.h.

m_trtid

const TRT_ID* InDet::TRT_SegmentsToTrack::m_trtid

private

Definition at line 104 of file TRT_SegmentsToTrack.h.

m_varHandleArraysDeclared

bool AthCommonDataStore< AthCommonMsg< Algorithm > >::m_varHandleArraysDeclared

privateinherited

Definition at line 399 of file AthCommonDataStore.h.

m_vhka

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

privateinherited

Definition at line 398 of file AthCommonDataStore.h.

---

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

• TRT_SegmentsToTrack.h
• TRT_SegmentsToTrack.cxx