#include <EventHandler.h>

Collaboration diagram for MuonCalib::EventHandler:

Classes
struct	AuthorIndices

Public Types
using	ExtendedTrackPtr = MuonCalibExtendedRawHits::ExtendedTrkPtr
using	ExtendedSegmentPtr = MuonCalibExtendedRawHits::ExtendedSegPtr
using	TruthPtr = MuonCalibTruthCollection::TruthPtr
using	TruthMdtPtr = MuonCalibTruthCollection::TruthMdtPtr
using	TruthRpcPtr = MuonCalibTruthCollection::TruthRpcPtr
using	TruthTgcPtr = MuonCalibTruthCollection::TruthTgcPtr
using	TruthCscPtr = MuonCalibTruthCollection::TruthCscPtr
using	TrackVec = std::vector<ExtendedTrackPtr>
using	AuthorTrackVecMap = std::map<int, TrackVec>
using	SegmentVec = std::vector<ExtendedSegmentPtr>
using	AuthorSegmentVecMap = std::map<int, SegmentVec>
using	AuthorSegmentVecIt = AuthorSegmentVecMap::iterator
using	AuthorSegmentVecCit = AuthorSegmentVecMap::const_iterator

Public Member Functions
	EventHandler ()
	default constructor
	~EventHandler ()
	default destructor
void	clear ()
	clear all buffers
void	setEvent (std::shared_ptr< const MuonCalibEvent_E > event)
	set a new event (will call clear on the previous)
const TrackVec &	extendedTracks (int author) const
	get extended tracks
const SegmentVec &	extendedSegments (int author) const
	get extended tracks
const MuonCalibExtendedRawHits &	extendedRawHits () const
	get extended raw hits
const MuonCalibEvent_E &	event () const
	get event
const AuthorIndices &	mooreIndices () const
	get author indices for Moore
const AuthorIndices &	muonboyIndices () const
	get author indices for Muonboy
std::string	dumpRawHits () const
	dump routines
std::string	dumpTracks () const
std::string	dumpTracks (const TrackVec &tracks) const
std::string	dumpSegments (const SegmentVec &segments) const
std::string	dumpSegments () const
std::string	dumpEvent () const
void	setdbg (const bool dbg_on)

Private Member Functions
void	createExtendedTracks ()
	create extended tracks for event
void	createExtendedTruthTracks ()
	create extended tracks for event
void	createExtendedSegments ()
	create extended segments for event
void	createExtendedRawHits ()
	create extend raw hits
void	associateSegmentsWithTracks (SegmentVec &segments, TrackVec &tracks)
	create association between tracks and segments
int	associateSegmentWithTrack (const MuonCalibExtendedSegment &segment, const MuonCalibExtendedTrack &track)
	check whether segment and track share the same hits
int	segmentOverlap (const MuonCalibExtendedSegment &segment1, const MuonCalibExtendedSegment &segment2)
	check whether segments share the same hits
void	linkTracks (int referenceIndex)
	link tracks to the reference track using the hits on the track
void	match (const MuonCalibExtendedTrack &reference, TrackVec &tracks)
	find best match of tracks with reference
template<typename T>
void	split_along_barcode (const std::vector< T > &container, std::map< int, std::set< T > > &barcode_map)
	Spltis the truth hit container into a map having the barcode of the particle as key.
std::shared_ptr< MuonCalibTrack_E >	transform_to_trk (const TruthPtr &truth_part, const std::map< int, std::set< TruthMdtPtr > > &mdt_hits, const std::map< int, std::set< TruthCscPtr > > &csc_hits, const std::map< int, std::set< TruthRpcPtr > > &rpc_hits, const std::map< int, std::set< TruthTgcPtr > > &tgc_hits)

Private Attributes
std::shared_ptr< const MuonCalibEvent_E >	m_event
AuthorTrackVecMap	m_extendedTracks
AuthorSegmentVecMap	m_extendedSegments
MuonCalibExtendedRawHits	m_extendedRawHits
AuthorIndices	m_mooreIndices
AuthorIndices	m_muonboyIndices
bool	m_debug

Detailed Description

Definition at line 23 of file EventHandler.h.

Member Typedef Documentation

◆ AuthorSegmentVecCit

using MuonCalib::EventHandler::AuthorSegmentVecCit = AuthorSegmentVecMap::const_iterator

Definition at line 47 of file EventHandler.h.

◆ AuthorSegmentVecIt

using MuonCalib::EventHandler::AuthorSegmentVecIt = AuthorSegmentVecMap::iterator

Definition at line 46 of file EventHandler.h.

◆ AuthorSegmentVecMap

using MuonCalib::EventHandler::AuthorSegmentVecMap = std::map<int, SegmentVec>

Definition at line 45 of file EventHandler.h.

◆ AuthorTrackVecMap

using MuonCalib::EventHandler::AuthorTrackVecMap = std::map<int, TrackVec>

Definition at line 42 of file EventHandler.h.

◆ ExtendedSegmentPtr

using MuonCalib::EventHandler::ExtendedSegmentPtr = MuonCalibExtendedRawHits::ExtendedSegPtr

Definition at line 33 of file EventHandler.h.

◆ ExtendedTrackPtr

using MuonCalib::EventHandler::ExtendedTrackPtr = MuonCalibExtendedRawHits::ExtendedTrkPtr

Definition at line 32 of file EventHandler.h.

◆ SegmentVec

using MuonCalib::EventHandler::SegmentVec = std::vector<ExtendedSegmentPtr>

Definition at line 44 of file EventHandler.h.

◆ TrackVec

using MuonCalib::EventHandler::TrackVec = std::vector<ExtendedTrackPtr>

Definition at line 41 of file EventHandler.h.

◆ TruthCscPtr

using MuonCalib::EventHandler::TruthCscPtr = MuonCalibTruthCollection::TruthCscPtr

Definition at line 39 of file EventHandler.h.

◆ TruthMdtPtr

using MuonCalib::EventHandler::TruthMdtPtr = MuonCalibTruthCollection::TruthMdtPtr

Definition at line 36 of file EventHandler.h.

◆ TruthPtr

using MuonCalib::EventHandler::TruthPtr = MuonCalibTruthCollection::TruthPtr

Definition at line 35 of file EventHandler.h.

◆ TruthRpcPtr

using MuonCalib::EventHandler::TruthRpcPtr = MuonCalibTruthCollection::TruthRpcPtr

Definition at line 37 of file EventHandler.h.

◆ TruthTgcPtr

using MuonCalib::EventHandler::TruthTgcPtr = MuonCalibTruthCollection::TruthTgcPtr

Definition at line 38 of file EventHandler.h.

Constructor & Destructor Documentation

◆ EventHandler()

MuonCalib::EventHandler::EventHandler ( )

default constructor

Definition at line 35 of file EventHandler.cxx.

35: m_event(nullptr), m_mooreIndices(4, 0), m_muonboyIndices(3, 100), m_debug(false) {}

MuonCalib::EventHandler::m_mooreIndices

AuthorIndices m_mooreIndices

Definition EventHandler.h:124

MuonCalib::EventHandler::m_debug

bool m_debug

Definition EventHandler.h:126

MuonCalib::EventHandler::m_muonboyIndices

AuthorIndices m_muonboyIndices

Definition EventHandler.h:125

MuonCalib::EventHandler::m_event

std::shared_ptr< const MuonCalibEvent_E > m_event

Definition EventHandler.h:118

◆ ~EventHandler()

MuonCalib::EventHandler::~EventHandler ( )

default destructor

Definition at line 37 of file EventHandler.cxx.

37{ clear(); }

MuonCalib::EventHandler::clear

void clear()

clear all buffers

Definition EventHandler.cxx:68

Member Function Documentation

◆ associateSegmentsWithTracks()

void MuonCalib::EventHandler::associateSegmentsWithTracks	(	SegmentVec &	segments,
		TrackVec &	tracks )

private

create association between tracks and segments

helper struct to evaluate the segment match quality

Definition at line 405 of file EventHandler.cxx.

                                                                                         {
        struct segmentMatchQual {
            ExtendedSegmentPtr seg;
            int assoc_quality{0};
            bool vetoed{false};
            bool operator<(const segmentMatchQual& other) const {
                if (assoc_quality != other.assoc_quality) return assoc_quality > other.assoc_quality;
                return seg->chi2() < other.seg->chi2();
            };
            segmentMatchQual(ExtendedSegmentPtr ptr, int score) : seg{std::move(ptr)}, assoc_quality{score} {}
        };
 
        for (const ExtendedTrackPtr& trk : tracks) {
            std::vector<segmentMatchQual> segmentTrackMatch;
            for (const ExtendedSegmentPtr& seg : segments) {
                int match = associateSegmentWithTrack(*seg, *trk);
                if (match > 200) { segmentTrackMatch.emplace_back(seg, match); }
            }
            std::vector<segmentMatchQual>::iterator mit = segmentTrackMatch.begin();
            std::vector<segmentMatchQual>::iterator mit_end = segmentTrackMatch.end();
            std::sort(mit, mit_end);
            for (; mit != mit_end; ++mit) {
                std::vector<segmentMatchQual>::iterator mit2 = mit + 1;
                for (; mit2 != mit_end; ++mit2) {
                    if (segmentOverlap(*mit->seg, *mit2->seg) > 1) { mit2->vetoed = true; }
                }
 
                std::cout << "We need to refine the track to segment association " << __FILE__ << " " << __LINE__ << std::endl;
                // if (!mit->vetoed) {
                //     mit->seg->addTrack(trk);
                //     trk->addSegment(mit->seg);
                // }
            }
        }
    }

◆ associateSegmentWithTrack()

int MuonCalib::EventHandler::associateSegmentWithTrack	(	const MuonCalibExtendedSegment &	segment,
		const MuonCalibExtendedTrack &	track )

private

check whether segment and track share the same hits

Definition at line 487 of file EventHandler.cxx.

                                                                                                                            {
        double minRadius = 1.;  // the sign of all hits with a drift radius smaller that minRadius is not taken into account
        unsigned int noppositeSign{0}, nsameSign{0}, nnotOnTrack{0};
        for (const MuonCalibSegment::MdtHitPtr& mdt_hit : segment.mdtHOT()) {
            const MuonFixedId& id = mdt_hit->identify();
            if (!id.isValid()) {
                MsgStream log(Athena::getMessageSvc(), "EventHandler");
                log << MSG::WARNING << "associateSegmentWithTrack(), invalid MDT id! " << endmsg;
                continue;
            }
            // look up hit on track
            IdHitMap::const_iterator trkHit = track.idHitMap().find(id);
            if (trkHit != track.idHitMap().end()) {
                // found hit
                double signSeg = mdt_hit->driftRadius() < -minRadius ? -1. : 1.;
                double signTrk = trkHit->second->driftRadius() < -minRadius ? -1. : 1.;
                if (signSeg == signTrk)
                    ++nsameSign;
                else
                    ++noppositeSign;
            } else {
                ++nnotOnTrack;
            }
        }
 
        for (const MuonCalibSegment::CscHitPtr& csc_hit : segment.cscHOT()) {
            MuonFixedId id = csc_hit->identify();
            if (!id.isValid()) {
                MsgStream log(Athena::getMessageSvc(), "EventHandler");
                log << MSG::WARNING << "associateSegmentWithTrack(), invalid CSC id! " << endmsg;
                continue;
            }
            // look up hit on track
            IdHitMap::const_iterator trkHit = track.idHitMap().find(id);
            if (trkHit != track.idHitMap().end()) {
                // found hit
                ++nsameSign;
            } else {
                ++nnotOnTrack;
            }
        }
        // Okay the weighting factors seem a bit arbitrary
        return nnotOnTrack + 10 * noppositeSign + 100 * nsameSign;
    }

◆ clear()

void MuonCalib::EventHandler::clear ( )

clear all buffers

Definition at line 68 of file EventHandler.cxx.

                             {
        m_extendedTracks.clear();
        m_extendedSegments.clear();
    }

◆ createExtendedRawHits()

void MuonCalib::EventHandler::createExtendedRawHits ( )

private

create extend raw hits

Definition at line 167 of file EventHandler.cxx.

                                             {
        if (!m_event || !m_event->rawHitCollection()) return;
 
        AuthorTrackVecMap::iterator trk_itr = m_extendedTracks.find(m_mooreIndices.track);
        AuthorSegmentVecMap::iterator segIt = m_extendedSegments.find(m_mooreIndices.segment);
 
        SegmentVec seg_to_pipe;
        TrackVec trk_to_pipe;
        if (segIt != m_extendedSegments.end()) seg_to_pipe = segIt->second;
        if (trk_itr != m_extendedTracks.end()) trk_to_pipe = trk_itr->second;
 
        m_extendedRawHits = MuonCalibExtendedRawHits(*m_event->rawHitCollection(), seg_to_pipe, trk_to_pipe);
    }

◆ createExtendedSegments()

void MuonCalib::EventHandler::createExtendedSegments ( )

private

create extended segments for event

Definition at line 388 of file EventHandler.cxx.

                                              {
        // check that event is set
        if (!m_event) return;
 
        for (const MuonCalibEvent::PatternPtr& pat_ptr : m_event->pattern()) {
            for (const MuonCalibPattern::MuonCalibSegPtr& seg_ptr : pat_ptr->muonSegs()) {
                // get author
                int author = seg_ptr->author();
                SegmentVec& segVec = m_extendedSegments[author];
 
                // create new extended segment
                MuonCalibExtendedSegment* extendedSegment = new MuonCalibExtendedSegment(*seg_ptr);
                segVec.emplace_back(extendedSegment);
            }
        }
    }

◆ createExtendedTracks()

void MuonCalib::EventHandler::createExtendedTracks ( )

private

create extended tracks for event

Definition at line 181 of file EventHandler.cxx.

                                            {
        // check that event is set
        if (!m_event) return;
 
        // hack to get the associations right...
        ExtendedTrackPtr previousMooreTrack;
 
        for (const MuonCalibEvent_E::MuonCalibTrkPtr& track_ptr : m_event->tracks()) {
            // get author Moore/Muonboy tracks author == 0 and corresponding TrackVec
            int author = track_ptr->author();
            TrackVec& trackVec = m_extendedTracks[author];
 
            // create new extended track and add it to TrackVec
            ExtendedTrackPtr extendedTrack = std::make_shared<MuonCalibExtendedTrack>(*track_ptr, 0, 0);
 
            // store previous moore track so we can add the back extrapolated ones if any
#ifdef TO_DO
            if (author == 0) {
                previousMooreTrack = extendedTrack;
            } else if ((author == 1 || author == 2) && previousMooreTrack) {
                std::cout << "Perform proper track association " << __FILE__ << " " << __LINE__ << std::endl;
                // previousMooreTrack->addAssociatedTrack(extendedTrack);
                // extendedTrack->addAssociatedTrack(previousMooreTrack);
            } else if (author == 10 && track_ptr->nrHits() == 0) {
                // Associate MuidSA/MooreExtrapolate to IP to author 0 if no hits on track
                TrackVec& trackVec0 = m_extendedTracks[0];
                double dotprodmax = -1.;
                if (trackVec0.size() > 0) {
                    int trackcount10 = trackVec.size();
                    TrackVec::iterator it0 = trackVec0.begin();
                    TrackVec::iterator it0_end = trackVec0.end();
                    int trackcount0 = 0;
                    const MuonCalibExtendedTrack* associatedMooreTrack = 0;
                    for (; it0 != it0_end; ++it0) {
                        trackcount0++;
                        if (trackcount10 > trackcount0) continue;
                        double dotprod = extendedTrack->direction().dot((*it0)->direction());
                        if (extendedTrack->isAssociated(*it0)) continue;
                        if ((*it0)->isAssociated(extendedTrack)) continue;
                        if (dotprod > dotprodmax) {
                            dotprodmax = dotprod;
                            associatedMooreTrack = *it0;
                            if (dotprodmax > 0.8) break;
                        }
                    }
                    if (dotprodmax > 0.8 && associatedMooreTrack) {
                        extendedTrack->addAssociatedTrack(associatedMooreTrack);
                        associatedMooreTrack->addAssociatedTrack(extendedTrack);
                    }
                }
            }
#endif
            trackVec.push_back(extendedTrack);
        }
    }

◆ createExtendedTruthTracks()

void MuonCalib::EventHandler::createExtendedTruthTracks ( )

private

create extended tracks for event

Definition at line 245 of file EventHandler.cxx.

                                                 {
        // check that event is set
        if (!m_event) return;
        //
        // Truth track collection
        //
 
        const MuonCalibTruthCollection* truth = m_event->calibTruthCollection();
        if (!truth) return;
 
        if (truth->TruthContainer().empty()) return;
 
        std::map<int, TruthPtr> truthTrkME, truthTrkIP, truthTrkRest;
 
        std::map<int, std::set<TruthMdtPtr>> truthTrkMdtHitMap;
        std::map<int, std::set<TruthCscPtr>> truthTrkCscHitMap;
        std::map<int, std::set<TruthRpcPtr>> truthTrkRpcHitMap;
        std::map<int, std::set<TruthTgcPtr>> truthTrkTgcHitMap;
 
        MsgStream log(Athena::getMessageSvc(), "EventHandler");
        for (const TruthPtr& truth_part : truth->TruthContainer()) {
            int barcode = truth_part->barCode();
            if (barcode <= 0) continue;
            if (log.level() <= MSG::DEBUG) log << MSG::DEBUG << "createExtendedTruthTracks() truth track barcode " << barcode << endmsg;
 
            if (truth_part->position().perp() < 4000 && std::abs(truth_part->position().z()) < 6000.) {
                if (truth_part->position().perp() < 100 && std::abs(truth_part->position().z()) < 500.) {
                    // Close to IP
                    truthTrkIP[barcode] = truth_part;
                } else {
                    truthTrkRest[barcode] = truth_part;
                }
            } else {
                // Take track at Muon Entry
                truthTrkME[barcode] = truth_part;
            }
        }
 
        //
        //  Mdt truth hit information
        //
        split_along_barcode(truth->MdtContainer(), truthTrkMdtHitMap);
 
        if (truthTrkMdtHitMap.empty()) return;
 
        //
        //  Csc truth hit information
        //
        split_along_barcode(truth->CscContainer(), truthTrkCscHitMap);
        //
        //  Rpc truth hit information
        //
 
        split_along_barcode(truth->RpcContainer(), truthTrkRpcHitMap);
 
        //
        //  Tgc truth hit information
        //
        split_along_barcode(truth->TgcContainer(), truthTrkTgcHitMap);
 
        for (const auto& it_IP : truthTrkIP) {
            std::shared_ptr<MuonCalibTrack_E> truth_trk =
                transform_to_trk(it_IP.second, truthTrkMdtHitMap, truthTrkCscHitMap, truthTrkRpcHitMap, truthTrkTgcHitMap);
            int barcode = it_IP.first;
            int pdgcode = it_IP.second->PDGCode();
            int author = -13;
            if (std::abs(pdgcode) != 13) author = -113;
            TrackVec& trackVec = m_extendedTracks[author];
            std::shared_ptr<MuonCalibExtendedTrack> extendedTrack = std::make_shared<MuonCalibExtendedTrack>(*truth_trk, pdgcode, barcode);
            trackVec.push_back(extendedTrack);
            // dump truth track
            if (m_debug) extendedTrack->dump();
        }
 
        for (const auto& it_ME : truthTrkME) {
            std::shared_ptr<MuonCalibTrack_E> truth_trk =
                transform_to_trk(it_ME.second, truthTrkMdtHitMap, truthTrkCscHitMap, truthTrkRpcHitMap, truthTrkTgcHitMap);
 
            int barcode = it_ME.first;
            int pdgcode = it_ME.second->PDGCode();
            int author = -1013;
            if (std::abs(pdgcode) != 13) author = -1113;
            TrackVec& trackVec = m_extendedTracks[author];
            std::shared_ptr<MuonCalibExtendedTrack> extendedTrack = std::make_shared<MuonCalibExtendedTrack>(*truth_trk, pdgcode, barcode);
            trackVec.push_back(extendedTrack);
            // dump truth track
            if (m_debug) extendedTrack->dump();
        }
    }

◆ dumpEvent()

std::string MuonCalib::EventHandler::dumpEvent ( ) const

Definition at line 147 of file EventHandler.cxx.

                                            {
        std::ostringstream sout;
        sout << dumpRawHits() << std::endl << dumpTracks() << std::endl << dumpSegments();
        return sout.str();
    }

◆ dumpRawHits()

std::string MuonCalib::EventHandler::dumpRawHits ( ) const

dump routines

Definition at line 103 of file EventHandler.cxx.

                                              {
        std::ostringstream sout;
        sout << m_extendedRawHits.dumpMdtHits();
        return sout.str();
    }

◆ dumpSegments() [1/2]

std::string MuonCalib::EventHandler::dumpSegments ( ) const

Definition at line 133 of file EventHandler.cxx.

                                               {
        std::ostringstream sout;
        if (m_extendedSegments.count(m_muonboyIndices.segment) && !extendedSegments(m_muonboyIndices.segment).empty()) {
            sout << " Muonboy Segments " << extendedSegments(m_muonboyIndices.segment).size() << std::endl
                 << dumpSegments(extendedSegments(m_muonboyIndices.segment));
            if (!extendedSegments(m_mooreIndices.segment).empty()) sout << std::endl;
        }
        if (m_extendedSegments.count(m_mooreIndices.segment) && !extendedSegments(m_mooreIndices.segment).empty()) {
            sout << " Moore Segments " << extendedSegments(m_mooreIndices.segment).size() << std::endl
                 << dumpSegments(extendedSegments(m_mooreIndices.segment));
        }
        return sout.str();
    }

◆ dumpSegments() [2/2]

std::string MuonCalib::EventHandler::dumpSegments ( const SegmentVec & segments ) const

Definition at line 121 of file EventHandler.cxx.

                                                                                     {
        std::ostringstream sout;
        SegmentVec::const_iterator seg = segments.begin();
        SegmentVec::const_iterator seg_end = segments.end();
        SegmentVec::const_iterator seg_last = seg_end - 1;
        for (; seg != seg_end; ++seg) {
            sout << (*seg)->dump();
            if (seg != seg_last) sout << std::endl;
        }
        return sout.str();
    }

◆ dumpTracks() [1/2]

std::string MuonCalib::EventHandler::dumpTracks ( ) const

Definition at line 153 of file EventHandler.cxx.

                                             {
        std::ostringstream sout;
        for (const auto& it : m_extendedTracks) {
            // skip standalone tracks at IP
            if (it.first == 1 || it.first == 2) continue;
 
            sout << " Track Author " << it.first << " number of tracks " << it.second.size();
            // only dump Muon tracks
            if (it.first < 1000) sout << std::endl << dumpTracks(it.second);
            sout << std::endl;
        }
        return sout.str();
    }

◆ dumpTracks() [2/2]

std::string MuonCalib::EventHandler::dumpTracks ( const TrackVec & tracks ) const

Definition at line 109 of file EventHandler.cxx.

                                                                               {
        std::ostringstream sout;
        TrackVec::const_iterator trk = tracks.begin();
        TrackVec::const_iterator trk_end = tracks.end();
        TrackVec::const_iterator trk_last = trk_end - 1;
        for (; trk != trk_end; ++trk) {
            sout << (*trk)->dump();
            if (trk != trk_last) sout << std::endl;
        }
        return sout.str();
    }

◆ event()

const MuonCalibEvent_E & MuonCalib::EventHandler::event ( ) const

get event

Definition at line 58 of file EventHandler.cxx.

58{ return *m_event; }

◆ extendedRawHits()

const MuonCalibExtendedRawHits & MuonCalib::EventHandler::extendedRawHits ( ) const

get extended raw hits

Definition at line 55 of file EventHandler.cxx.

55{ return m_extendedRawHits; }

◆ extendedSegments()

const EventHandler::SegmentVec & MuonCalib::EventHandler::extendedSegments ( int author ) const

get extended tracks

Definition at line 47 of file EventHandler.cxx.

                                                                                 {
        AuthorSegmentVecMap::const_iterator itr = m_extendedSegments.find(author);
        if (itr != m_extendedSegments.end()) return itr->second;
        static const SegmentVec dummy_vec;
        return dummy_vec;
    }

◆ extendedTracks()

const EventHandler::TrackVec & MuonCalib::EventHandler::extendedTracks ( int author ) const

get extended tracks

Definition at line 39 of file EventHandler.cxx.

                                                                             {
        AuthorTrackVecMap::const_iterator itr = m_extendedTracks.find(author);
        if (itr != m_extendedTracks.end()) return itr->second;
        static const TrackVec dummy_vec;
        return dummy_vec;
    }

◆ linkTracks()

void MuonCalib::EventHandler::linkTracks ( int referenceIndex )

private

link tracks to the reference track using the hits on the track

Definition at line 532 of file EventHandler.cxx.

                                                    {
        // check if author found in list
        AuthorTrackVecMap::iterator pos = m_extendedTracks.find(referenceIndex);
        if (pos == m_extendedTracks.end()) return;
 
        TrackVec& refTracks = pos->second;
 
        // loop over reference tracks and associate other authors
        TrackVec::iterator rit = refTracks.begin();
        TrackVec::iterator rit_end = refTracks.end();
        for (; rit != rit_end; ++rit) {
            // loop over authors
            AuthorTrackVecMap::iterator ait = m_extendedTracks.begin();
            AuthorTrackVecMap::iterator ait_end = m_extendedTracks.end();
            for (; ait != ait_end; ++ait) {
                // skip same collection
                if (ait->first == referenceIndex) continue;
 
                TrackVec& tracks = ait->second;
 
                match(**rit, tracks);
            }
        }
    }

◆ match()

void MuonCalib::EventHandler::match	(	const MuonCalibExtendedTrack &	reference,
		EventHandler::TrackVec &	tracks )

private

find best match of tracks with reference

Definition at line 557 of file EventHandler.cxx.

                                                                                                  {
        const MuonCalibExtendedTrack* bestMatch = nullptr;
        MuonCalibExtendedTrackOverlap bestOverlap;
        // loop over tracks and calcualte overlap with reference
        TrackVec::iterator it = tracks.begin();
        TrackVec::iterator it_end = tracks.end();
        for (; it != it_end; ++it) {
            // check whether the two tracks are not already associated
 
            // if (reference.isAssociated(*it)) continue;
 
            MuonCalibExtendedTrackOverlap overlap = reference.calculateHitOverlap(**it);
            if (overlap.sharedPrecisionHits() > 0) {
                if (overlap.sharedPrecisionHits() > bestOverlap.sharedPrecisionHits()) {
                    bestMatch = it->get();
                    bestOverlap = overlap;
                }
            }
        }
 
        std::cout << " Still need to be done a proper association " << __FILE__ << " " << __LINE__ << std::endl;
        if (bestMatch) {
            //     reference.addAssociatedTrack(bestMatch);
            //     bestMatch->addAssociatedTrack(&reference);
        }
    }

◆ mooreIndices()

const EventHandler::AuthorIndices & MuonCalib::EventHandler::mooreIndices ( ) const

get author indices for Moore

Definition at line 61 of file EventHandler.cxx.

61{ return m_mooreIndices; }

◆ muonboyIndices()

const EventHandler::AuthorIndices & MuonCalib::EventHandler::muonboyIndices ( ) const

get author indices for Muonboy

Definition at line 64 of file EventHandler.cxx.

64{ return m_muonboyIndices; }

◆ segmentOverlap()

int MuonCalib::EventHandler::segmentOverlap	(	const MuonCalibExtendedSegment &	segment1,
		const MuonCalibExtendedSegment &	segment2 )

private

check whether segments share the same hits

Definition at line 441 of file EventHandler.cxx.

                                                                                                                       {
        int overlap = 0;
        if (segment1.mdtHitsOnTrack() > 0 && segment2.mdtHitsOnTrack() > 0) {
            std::set<MuonFixedId> segmentIds;
            for (const MuonCalibSegment::MdtHitPtr& mdt_hit : segment1.mdtHOT()) {
                const MuonFixedId& id = mdt_hit->identify();
                if (!id.isValid()) {
                    MsgStream log(Athena::getMessageSvc(), "EventHandler");
                    log << MSG::WARNING << "segmentOverlap(), invalid MDT id! " << endmsg;
                    continue;
                }
                segmentIds.insert(id);
            }
            for (const MuonCalibSegment::MdtHitPtr& mdt_hit : segment2.mdtHOT()) {
                const MuonFixedId& id = mdt_hit->identify();
                if (!id.isValid()) {
                    MsgStream log(Athena::getMessageSvc(), "EventHandler");
                    log << MSG::WARNING << "segmentOverlap(), invalid MDT id! " << endmsg;
                    continue;
                }
                overlap += (segmentIds.count(id) > 0);
            }
            return overlap;
        } else if (segment1.cscHitsOnTrack() > 0 && segment2.cscHitsOnTrack() > 0) {
            std::set<MuonFixedId> segmentIds;
            for (const MuonCalibSegment::CscHitPtr& csc_hit : segment1.cscHOT()) {
                const MuonFixedId& id = csc_hit->identify();
                if (!id.isValid()) {
                    MsgStream log(Athena::getMessageSvc(), "EventHandler");
                    log << MSG::WARNING << "segmentOverlap(), invalid CSC id! " << endmsg;
                    continue;
                }
                segmentIds.insert(id);
            }
            for (const MuonCalibSegment::CscHitPtr& csc_hit : segment1.cscHOT()) {
                MuonFixedId id = csc_hit->identify();
                if (!id.isValid()) {
                    MsgStream log(Athena::getMessageSvc(), "EventHandler");
                    log << MSG::WARNING << "segmentOverlap(), invalid CSC id! " << endmsg;
                    continue;
                }
                overlap += (segmentIds.count(id) > 0);
            }
        }
        return overlap;
    }

◆ setdbg()

void MuonCalib::EventHandler::setdbg ( const bool dbg_on )

Definition at line 66 of file EventHandler.cxx.

66{ m_debug = dbg_on; }

◆ setEvent()

void MuonCalib::EventHandler::setEvent ( std::shared_ptr< const MuonCalibEvent_E > event )

set a new event (will call clear on the previous)

Definition at line 73 of file EventHandler.cxx.

                                                                           {
        // reset previous event and set new event
        clear();
        m_event = std::move(event);
 
        // create extended tracks
        createExtendedTracks();
 
        // create extended truth tracks
        createExtendedTruthTracks();
 
        // create extended tracks
        createExtendedSegments();
 
        // create extended raw hits
        createExtendedRawHits();
 
        AuthorTrackVecMap::iterator mit = m_extendedTracks.begin();
        AuthorTrackVecMap::iterator mit_end = m_extendedTracks.end();
        for (; mit != mit_end; ++mit) {
            if (mit->first < 500) linkTracks(mit->first);
 
            // associate segments with tracks
            if (mit->first < 99)
                associateSegmentsWithTracks(m_extendedSegments[m_mooreIndices.segment], mit->second);
            else
                associateSegmentsWithTracks(m_extendedSegments[m_muonboyIndices.segment], mit->second);
        }
    }

◆ split_along_barcode()

template<typename T>

void MuonCalib::EventHandler::split_along_barcode	(	const std::vector< T > &	container,
		std::map< int, std::set< T > > &	barcode_map )

private

Spltis the truth hit container into a map having the barcode of the particle as key.

Definition at line 237 of file EventHandler.cxx.

                                                                                                                                   {
        for (const auto& hit_ptr : container) {
            int barcode = hit_ptr->barCode();
            if (barcode <= 0) continue;
            barcode_map[barcode].insert(hit_ptr);
        }
    }

◆ transform_to_trk()

std::shared_ptr< MuonCalibTrack_E > MuonCalib::EventHandler::transform_to_trk	(	const TruthPtr &	truth_part,
		const std::map< int, std::set< TruthMdtPtr > > &	mdt_hits,
		const std::map< int, std::set< TruthCscPtr > > &	csc_hits,
		const std::map< int, std::set< TruthRpcPtr > > &	rpc_hits,
		const std::map< int, std::set< TruthTgcPtr > > &	tgc_hits )

private

Definition at line 334 of file EventHandler.cxx.

                                                                                                                       {
        const int barcode = truth_part->barCode();
        MuonCalibTrack_E::defineParams build_pars{};
        build_pars.x0 = truth_part->position().x();
        build_pars.y0 = truth_part->position().y();
        build_pars.z0 = truth_part->position().z();
        build_pars.phi = truth_part->momentum().phi();
        build_pars.theta = truth_part->momentum().theta();
        build_pars.author = std::abs(truth_part->PDGCode());
        build_pars.qOverP = (1. - 2. * (truth_part->PDGCode() < 0)) / (truth_part->momentum().mag());
        build_pars.cov.setZero();
        std::shared_ptr<MuonCalibTrack_E> track_ptr = std::make_shared<MuonCalibTrack_E>(build_pars);
 
        std::map<int, std::set<TruthMdtPtr>>::const_iterator mdt_itr = mdt_hits.find(barcode);
        if (mdt_itr != mdt_hits.end()) {
            for (const TruthMdtPtr& mdt_hit : mdt_itr->second) {
                MuonCalibHit_E::definePars hit_pars{};
                hit_pars.driftRadius = mdt_hit->driftRadius();
                hit_pars.pos = Amg::Vector3D(0., mdt_hit->positionAlongTube(), 0.);
                track_ptr->addHit(std::make_shared<MuonCalibHit_E>(mdt_hit->identify(), hit_pars));
            }
        }
        std::map<int, std::set<TruthCscPtr>>::const_iterator csc_itr = csc_hits.find(barcode);
        if (csc_itr != csc_hits.end()) {
            for (const TruthCscPtr& csc_hit : csc_itr->second) {
                MuonCalibHit_E::definePars hit_pars{};
                hit_pars.driftRadius = csc_hit->time();
                track_ptr->addHit(std::make_shared<MuonCalibHit_E>(csc_hit->identify(), hit_pars));
            }
        }
 
        std::map<int, std::set<TruthRpcPtr>>::const_iterator rpc_itr = rpc_hits.find(barcode);
        if (rpc_itr != rpc_hits.end()) {
            for (const TruthRpcPtr& rpc_hit : rpc_itr->second) {
                MuonCalibHit_E::definePars hit_pars{};
                hit_pars.driftRadius = rpc_hit->time();
                track_ptr->addHit(std::make_shared<MuonCalibHit_E>(rpc_hit->identify(), hit_pars));
            }
        }
        std::map<int, std::set<TruthTgcPtr>>::const_iterator tgc_itr = tgc_hits.find(barcode);
        if (tgc_itr != tgc_hits.end()) {
            for (const TruthTgcPtr& tgc_hit : tgc_itr->second) {
                MuonCalibHit_E::definePars hit_pars{};
                hit_pars.driftRadius = tgc_hit->time();
                track_ptr->addHit(std::make_shared<MuonCalibHit_E>(tgc_hit->identify(), hit_pars));
            }
        }
        return track_ptr;
    }

Member Data Documentation

◆ m_debug

bool MuonCalib::EventHandler::m_debug

private

Definition at line 126 of file EventHandler.h.

◆ m_event

std::shared_ptr<const MuonCalibEvent_E> MuonCalib::EventHandler::m_event

private

Definition at line 118 of file EventHandler.h.

◆ m_extendedRawHits

MuonCalibExtendedRawHits MuonCalib::EventHandler::m_extendedRawHits

private

Definition at line 121 of file EventHandler.h.

◆ m_extendedSegments

AuthorSegmentVecMap MuonCalib::EventHandler::m_extendedSegments

private

Definition at line 120 of file EventHandler.h.

◆ m_extendedTracks

AuthorTrackVecMap MuonCalib::EventHandler::m_extendedTracks

private

Definition at line 119 of file EventHandler.h.

◆ m_mooreIndices

AuthorIndices MuonCalib::EventHandler::m_mooreIndices

private

Definition at line 124 of file EventHandler.h.

◆ m_muonboyIndices

AuthorIndices MuonCalib::EventHandler::m_muonboyIndices

private

Definition at line 125 of file EventHandler.h.

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

Classes

Public Types

Public Member Functions

Private Member Functions

Private Attributes

Detailed Description

Member Typedef Documentation

◆ AuthorSegmentVecCit

◆ AuthorSegmentVecIt

◆ AuthorSegmentVecMap

◆ AuthorTrackVecMap

◆ ExtendedSegmentPtr

◆ ExtendedTrackPtr

◆ SegmentVec

◆ TrackVec

◆ TruthCscPtr

◆ TruthMdtPtr

◆ TruthPtr

◆ TruthRpcPtr

◆ TruthTgcPtr

Constructor & Destructor Documentation

◆ EventHandler()

◆ ~EventHandler()

Member Function Documentation

◆ associateSegmentsWithTracks()

◆ associateSegmentWithTrack()

◆ clear()

◆ createExtendedRawHits()

◆ createExtendedSegments()

◆ createExtendedTracks()

◆ createExtendedTruthTracks()

◆ dumpEvent()

◆ dumpRawHits()

◆ dumpSegments() [1/2]

◆ dumpSegments() [2/2]

◆ dumpTracks() [1/2]

◆ dumpTracks() [2/2]

◆ event()

◆ extendedRawHits()

◆ extendedSegments()

◆ extendedTracks()

◆ linkTracks()

◆ match()

◆ mooreIndices()

◆ muonboyIndices()

◆ segmentOverlap()

◆ setdbg()

◆ setEvent()

◆ split_along_barcode()

◆ transform_to_trk()

Member Data Documentation

◆ m_debug

◆ m_event

◆ m_extendedRawHits

◆ m_extendedSegments

◆ m_extendedTracks

◆ m_mooreIndices

◆ m_muonboyIndices