#include <SegmentAmbiSolver.h>

Inheritance diagram for MuonR4::SegmentAmbiSolver:

Collaboration diagram for MuonR4::SegmentAmbiSolver:

Classes
struct	Config

Public Types
enum	Resolution { Resolution::noOverlap, Resolution::superSet, Resolution::subSet }

using	SegmentVec = std::vector< std::unique_ptr< Segment > >

Public Member Functions
	SegmentAmbiSolver (const std::string &name, Config &&config)

SegmentVec	resolveAmbiguity (const ActsGeometryContext &gctx, SegmentVec &&toResolve) const

bool	msgLvl (const MSG::Level lvl) const
	Test the output level. More...

MsgStream &	msg () const
	The standard message stream. More...

MsgStream &	msg (const MSG::Level lvl) const
	The standard message stream. More...

void	setLevel (MSG::Level lvl)
	Change the current logging level. More...

Private Types
using	MeasurementSet = std::unordered_set< const xAOD::UncalibratedMeasurement * >

Private Member Functions
std::vector< int >	driftSigns (const ActsGeometryContext &gctx, const Segment &segment, const Segment::MeasVec &measurements) const

MeasurementSet	extractPrds (const Segment &segment) const
	Extract the Uncalibrated measurements used to build the segment. More...

unsigned int	countShared (const MeasurementSet &measSet1, const MeasurementSet &measSet2) const
	counts the number of measurements that're in both sets More...

double	redChi2 (const Segment &segment) const
	Returns the reduced chi2 of the segment. More...

void	initMessaging () const
	Initialize our message level and MessageSvc. More...

Private Attributes
const Config	m_cfg {}

std::string	m_nm
	Message source name. More...

boost::thread_specific_ptr< MsgStream >	m_msg_tls
	MsgStream instance (a std::cout like with print-out levels) More...

std::atomic< IMessageSvc * >	m_imsg { nullptr }
	MessageSvc pointer. More...

std::atomic< MSG::Level >	m_lvl { MSG::NIL }
	Current logging level. More...

std::atomic_flag m_initialized	ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT
	Messaging initialized (initMessaging) More...

Detailed Description

Definition at line 16 of file SegmentAmbiSolver.h.

Member Typedef Documentation

◆ MeasurementSet

using MuonR4::SegmentAmbiSolver::MeasurementSet = std::unordered_set<const xAOD::UncalibratedMeasurement*>

private

Definition at line 49 of file SegmentAmbiSolver.h.

◆ SegmentVec

using MuonR4::SegmentAmbiSolver::SegmentVec = std::vector<std::unique_ptr<Segment> >

Definition at line 32 of file SegmentAmbiSolver.h.

Member Enumeration Documentation

◆ Resolution

enum MuonR4::SegmentAmbiSolver::Resolution

strong

Enumerator
noOverlap
superSet
subSet

Definition at line 37 of file SegmentAmbiSolver.h.

                                  {
                 noOverlap,
                 superSet,
                 subSet
             };

Constructor & Destructor Documentation

◆ SegmentAmbiSolver()

MuonR4::SegmentAmbiSolver::SegmentAmbiSolver	(	const std::string &	name,
		Config &&	config
	)

Definition at line 12 of file SegmentAmbiSolver.cxx.

                                                                            :
         AthMessaging{name},
         m_cfg{std::move(cfg)} {}

Member Function Documentation

◆ countShared()

unsigned int MuonR4::SegmentAmbiSolver::countShared	(	const MeasurementSet &	measSet1,
		const MeasurementSet &	measSet2
	)		const

private

counts the number of measurements that're in both sets

Definition at line 120 of file SegmentAmbiSolver.cxx.

                                                                                       {
         if (measSet1.size() > measSet2.size()) {
             return std::count_if(measSet2.begin(),measSet2.end(),[&measSet1](const xAOD::UncalibratedMeasurement* meas){
                 return measSet1.count(meas);
             });
         }
         return std::count_if(measSet1.begin(),measSet1.end(),[&measSet2](const xAOD::UncalibratedMeasurement* meas){
                 return measSet2.count(meas);
             }); 
     }

◆ driftSigns()

std::vector< int > MuonR4::SegmentAmbiSolver::driftSigns	(	const ActsGeometryContext &	gctx,
		const Segment &	segment,
		const Segment::MeasVec &	measurements
	)		const

private

Definition at line 97 of file SegmentAmbiSolver.cxx.

                                                                                              {
         const auto [locPos, locDir] = makeLine(localSegmentPars(gctx, segment));
         ATH_MSG_VERBOSE("Fetch drift signs for segment "<<segment.msSector()->identString()<<" -- "<<Amg::toString(locPos)
                         <<Amg::toString(locDir));
         return SegmentFitHelpers::driftSigns(locPos, locDir, measurements,  msg());
     }

◆ extractPrds()

SegmentAmbiSolver::MeasurementSet MuonR4::SegmentAmbiSolver::extractPrds ( const Segment & segment ) const

private

Extract the Uncalibrated measurements used to build the segment.

Definition at line 106 of file SegmentAmbiSolver.cxx.

                                                                    {
         
         MeasurementSet meas{};
         for (const Segment::MeasVec::value_type& hit : segment.measurements()) {
             if (!hit->spacePoint() || hit->fitState() != CalibratedSpacePoint::State::Valid || !hit->measuresEta()) {
                 continue;
             }
             meas.insert(hit->spacePoint()->primaryMeasurement());
             if (hit->spacePoint()->secondaryMeasurement()) {
                  meas.insert(hit->spacePoint()->secondaryMeasurement());
             }
         }
         return meas;
     }

◆ initMessaging()

void AthMessaging::initMessaging ( ) const

privateinherited

Initialize our message level and MessageSvc.

This method should only be called once.

Definition at line 39 of file AthMessaging.cxx.

 {
   m_imsg = Athena::getMessageSvc();
   m_lvl = m_imsg ?
     static_cast<MSG::Level>( m_imsg.load()->outputLevel(m_nm) ) :
     MSG::INFO;
 }

◆ msg() [1/2]

MsgStream & AthMessaging::msg ( ) const

inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 164 of file AthMessaging.h.

 {
   MsgStream* ms = m_msg_tls.get();
   if (!ms) {
     if (!m_initialized.test_and_set()) initMessaging();
     ms = new MsgStream(m_imsg,m_nm);
     m_msg_tls.reset( ms );
   }
  
   ms->setLevel (m_lvl);
   return *ms;
 }

◆ msg() [2/2]

MsgStream & AthMessaging::msg ( const MSG::Level lvl ) const

inlineinherited

The standard message stream.

Returns a reference to the default message stream May not be invoked before sysInitialize() has been invoked.

Definition at line 179 of file AthMessaging.h.

180 { return msg() << lvl; }

◆ msgLvl()

bool AthMessaging::msgLvl ( const MSG::Level lvl ) const

inlineinherited

Test the output level.

Parameters

lvl	The message level to test against

Returns: boolean Indicating if messages at given level will be printed

Return values

true	Messages at level "lvl" will be printed

Definition at line 151 of file AthMessaging.h.

 {
   if (!m_initialized.test_and_set()) initMessaging();
   if (m_lvl <= lvl) {
     msg() << lvl;
     return true;
   } else {
     return false;
   }
 }

◆ redChi2()

double MuonR4::SegmentAmbiSolver::redChi2 ( const Segment & segment ) const

private

Returns the reduced chi2 of the segment.

Definition at line 16 of file SegmentAmbiSolver.cxx.

                                                                   {
         return segment.chi2() / segment.nDoF();
     }

◆ resolveAmbiguity()

SegmentVec MuonR4::SegmentAmbiSolver::resolveAmbiguity	(	const ActsGeometryContext &	gctx,
		SegmentVec &&	toResolve
	)		const

Mark the first object as resolved

Fetch first the Prds

Check whether the two segments share hits at all

Re-evaluate the drift signs of the accepted measurement w.r.t. good one

Left-right solutions differ & the two segments have the same nDOF

Segments below that threshold are not considered for outlier removal Take the one which has more degrees of freedom

No overlap detected thus far

Definition at line 19 of file SegmentAmbiSolver.cxx.

                                                                                  {
         
         std::ranges::stable_sort(toResolve,[this](const SegmentVec::value_type& a,
                                               const SegmentVec::value_type& b){
             const double redChi2A = redChi2(*a);
             const double redChi2B = redChi2(*b);
             if (redChi2A < m_cfg.selectByNDoFChi2 && redChi2B < m_cfg.selectByNDoFChi2){
                 return a->nDoF() > b->nDoF();
             }
             return redChi2A < redChi2B;
         });
         SegmentVec resolved{};
  
         resolved.push_back(std::move(toResolve[0]));
         toResolve.erase(toResolve.begin());
         std::vector<std::vector<int>> segmentSigns{driftSigns(gctx, *resolved.front(), resolved.front()->measurements())};
         std::vector<MeasurementSet> segMeasurements{extractPrds(*resolved.front())};
  
         for (std::unique_ptr<Segment>& resolveMe : toResolve) {
             ATH_MSG_VERBOSE("Try to resolve new segment "<<toString(localSegmentPars(gctx, *resolveMe))
                           <<" redChi2: "<<redChi2(*resolveMe)<<" nDoF: "<<resolveMe->nDoF());
             MeasurementSet testMeas{extractPrds(*resolveMe)};
             Resolution reso{Resolution::noOverlap};
             unsigned int resolvedIdx{0};
             for (std::unique_ptr<Segment>& goodSeg : resolved) {
                 ATH_MSG_VERBOSE("Test against segment "<<toString(localSegmentPars(gctx, *goodSeg))
                         <<" redChi2: "<<redChi2(*goodSeg)<<" nDoF: "<<goodSeg->nDoF());
                 MeasurementSet& resolvedM = segMeasurements[resolvedIdx];
                 std::vector<int>& existSigns{segmentSigns[resolvedIdx++]};
                 unsigned int shared = countShared(resolvedM, testMeas);
                 if (shared < m_cfg.sharedPrecHits) {
                     ATH_MSG_VERBOSE("Too few shared measurements "<<shared<<" (Required: "<<m_cfg.sharedPrecHits<<").");
                     continue;
                 }
                 const std::vector<int> reEvaluatedSigns{driftSigns(gctx, *resolveMe, goodSeg->measurements())};
  
                 unsigned int sameSides{0};
                 for (unsigned int s =0 ; s < existSigns.size(); ++s) {
                     sameSides += (reEvaluatedSigns[s] == existSigns[s]);
                 }
                 if (!m_cfg.remLeftRightAmbi && sameSides != existSigns.size() && resolveMe->nDoF() == goodSeg->nDoF()) {
                     ATH_MSG_VERBOSE("Reference signs: "<<existSigns<<" / re-evaluated: "<<reEvaluatedSigns);
                     continue;
                 }
                 const double resolvedChi2  = redChi2(*goodSeg);
                 const double resolveMeChi2 = redChi2(*resolveMe);
                 reso = resolveMeChi2 < resolvedChi2 ? Resolution::superSet : Resolution::subSet;
  
                 ATH_MSG_VERBOSE("Chi2 good "<<resolvedChi2<<", candidate chi2: "<<resolveMeChi2);
                 if (resolveMeChi2 < m_cfg.selectByNDoFChi2 && resolvedChi2 < m_cfg.selectByNDoFChi2) {
                     reso =  goodSeg->nDoF() > resolveMe->nDoF() ? Resolution::subSet : Resolution::superSet;
                 }
                 if (reso == Resolution::superSet) {
                     std::swap(goodSeg, resolveMe);
                     std::swap(resolvedM, testMeas);
                     existSigns = driftSigns(gctx, *resolveMe, resolveMe->measurements());
                     break;
                 } else if (reso == Resolution::subSet) {
                     break;
                 }
             }
             if (reso == Resolution::noOverlap) {
                 segMeasurements.push_back(std::move(testMeas));
                 segmentSigns.push_back(driftSigns(gctx, *resolveMe, resolveMe->measurements()));
                 resolved.push_back(std::move(resolveMe));
             }
         }
         return resolved;
     }

◆ setLevel()

void AthMessaging::setLevel ( MSG::Level lvl )

inherited

Change the current logging level.

Use this rather than msg().setLevel() for proper operation with MT.

Definition at line 28 of file AthMessaging.cxx.

 {
   m_lvl = lvl;
 }

Member Data Documentation

◆ ATLAS_THREAD_SAFE

std::atomic_flag m_initialized AthMessaging::ATLAS_THREAD_SAFE = ATOMIC_FLAG_INIT

mutableprivateinherited

Messaging initialized (initMessaging)

Definition at line 141 of file AthMessaging.h.

◆ m_cfg

const Config MuonR4::SegmentAmbiSolver::m_cfg {}

private

Definition at line 43 of file SegmentAmbiSolver.h.

◆ m_imsg

std::atomic<IMessageSvc*> AthMessaging::m_imsg { nullptr }

mutableprivateinherited

MessageSvc pointer.

Definition at line 135 of file AthMessaging.h.

◆ m_lvl

std::atomic<MSG::Level> AthMessaging::m_lvl { MSG::NIL }

mutableprivateinherited

Current logging level.

Definition at line 138 of file AthMessaging.h.

◆ m_msg_tls

boost::thread_specific_ptr<MsgStream> AthMessaging::m_msg_tls

mutableprivateinherited

MsgStream instance (a std::cout like with print-out levels)

Definition at line 132 of file AthMessaging.h.

◆ m_nm

std::string AthMessaging::m_nm

privateinherited

Message source name.

Definition at line 129 of file AthMessaging.h.

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

Classes

Public Types

Public Member Functions

Private Types

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ MeasurementSet

◆ SegmentVec

Member Enumeration Documentation

◆ Resolution

Constructor & Destructor Documentation

◆ SegmentAmbiSolver()

Member Function Documentation

◆ countShared()

◆ driftSigns()

◆ extractPrds()

◆ initMessaging()

◆ msg() [1/2]

◆ msg() [2/2]

◆ msgLvl()

◆ redChi2()

◆ resolveAmbiguity()

◆ setLevel()

Member Data Documentation

◆ ATLAS_THREAD_SAFE

◆ m_cfg

◆ m_imsg

◆ m_lvl

◆ m_msg_tls

◆ m_nm