ATLAS Offline Software
Public Types | Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
VertexMatcher Class Reference

#include <VertexMatcher.h>

Inheritance diagram for VertexMatcher:
Collaboration diagram for VertexMatcher:

Public Types

typedef std::map< TIDA::Vertex *, TIDA::Vertex * > map_type
 
typedef std::map< TIDA::Vertex *, TIDA::Vertex * > rmap_type
 

Public Member Functions

 VertexMatcher (const std::string &s, double d=3)
 
 ~VertexMatcher ()
 
virtual BestMatcher< TIDA::Vertex > * clone () override
 
virtual double distance (const TIDA::Vertex *v0, const TIDA::Vertex *v1) const override
 
virtual void match (const std::vector< TIDA::Vertex * > &ref, const std::vector< TIDA::Vertex * > &test)
 
virtual const TIDA::Vertexmatched (TIDA::Vertex *t)
 
const map_typematched () const
 
virtual const TIDA::Vertexrevmatched (TIDA::Vertex *t)
 
const rmap_typerevmatched () const
 
void clear ()
 
unsigned size () const
 

Protected Member Functions

std::map< int, int > matcher (const std::vector< Tp * > &ref, const std::vector< Tq * > &test)
 

Protected Attributes

double m_d
 
std::string mname
 
map_type mmatched
 
rmap_type mrevmatched
 

Detailed Description

Definition at line 22 of file VertexMatcher.h.

Member Typedef Documentation

◆ map_type

Definition at line 28 of file TIDAAssociator.h.

◆ rmap_type

Definition at line 29 of file TIDAAssociator.h.

Constructor & Destructor Documentation

◆ VertexMatcher()

VertexMatcher::VertexMatcher ( const std::string &  s,
double  d = 3 
)
inline

Definition at line 26 of file VertexMatcher.h.

26 : BestMatcher<TIDA::Vertex>( s, d*d ) { }

◆ ~VertexMatcher()

VertexMatcher::~VertexMatcher ( )
inline

Definition at line 28 of file VertexMatcher.h.

28 { }

Member Function Documentation

◆ clear()

void TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::clear
inlineinherited

Definition at line 63 of file TIDAAssociator.h.

63 { mmatched.clear(); mrevmatched.clear(); }

◆ clone()

virtual BestMatcher<TIDA::Vertex>* VertexMatcher::clone ( )
inlineoverridevirtual

Implements TIDA::Associator< TIDA::Vertex, TIDA::Vertex >.

Definition at line 30 of file VertexMatcher.h.

30 { return new VertexMatcher(*this); }

◆ distance()

virtual double VertexMatcher::distance ( const TIDA::Vertex v0,
const TIDA::Vertex v1 
) const
inlineoverridevirtual

ha ha ha, divide by the "expected uncertainty" you must be MAD

Implements BestMatcher< TIDA::Vertex >.

Definition at line 32 of file VertexMatcher.h.

32  {
33  double d = (v0->z()-v1->z());
34  return d*d;
35 
37  // if ( v0->dz() && v1->dz() ) return d*d/( v0->dz()*v0->dz() + v1->dz()*v1->dz() );
38  // else return d*d;
39  };

◆ match()

virtual void BestMatcher< TIDA::Vertex , TIDA::Vertex >::match ( const std::vector< TIDA::Vertex * > &  ref,
const std::vector< TIDA::Vertex * > &  test 
)
inlinevirtualinherited

Implements TIDA::Associator< TIDA::Vertex, TIDA::Vertex >.

Definition at line 68 of file BestMatcher.h.

69  {
70  this->clear();
71 
72  std::map<int,int> matched = matcher( ref, test);
73 
74  std::map<int,int>::iterator mitr = matched.begin();
75  while ( mitr!=matched.end() ) {
76  this->mmatched.insert( typename TIDA::Associator<T,S>::map_type::value_type( ref[mitr->first], test[mitr->second]) );
77  this->mrevmatched.insert( typename TIDA::Associator<S,T>::map_type::value_type( test[mitr->second], ref[mitr->first] ) );
78  ++mitr;
79  }
80 
81  }

◆ matched() [1/2]

const map_type& TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::matched
inlineinherited

Definition at line 59 of file TIDAAssociator.h.

59 { return mmatched; }

◆ matched() [2/2]

virtual const TIDA::Vertex * TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::matched ( TIDA::Vertex t)
inlinevirtualinherited

Definition at line 45 of file TIDAAssociator.h.

45  {
46  typename map_type::const_iterator titr = mmatched.find(t);
47  if ( titr != mmatched.end() ) return titr->second;
48  else return 0;
49  }

◆ matcher()

std::map<int, int> BestMatcher< TIDA::Vertex , TIDA::Vertex >::matcher ( const std::vector< Tp * > &  ref,
const std::vector< Tq * > &  test 
)
inlineprotectedinherited

inserting into a multimatch sorts them into order as they are added so you get the list of best matches imediately NB: probably quicker to add them to a vector and then sort them afterward

now go through from best to worst, adding to another sorted set only the matches that do not use tracks already used by a better matching pair already found

set of unique track pairings NB: for some reason, using the multiset is faster than using a vector, even though we are only iterating through it later

hooray!! now print out the answer, make a map etc this isn't technically needed, could just use the set of "matched" objects, but hey ho

Definition at line 91 of file BestMatcher.h.

91  {
92 
98 
99  std::multiset<matched_> m;
100 
101  for (unsigned int i=0 ; i<ref.size() ; i++ ) {
102 
103  for (unsigned int j=0 ; j<test.size() ; j++ ) {
104  double d = distance(ref[i], test[j]);
105  if ( d<m_d ){
106  m.insert( matched_(d, i, j) );
107  }
108  }
109  }
110 
114 
115  std::vector<bool> refused( ref.size(), false );
116  std::vector<bool> testused( test.size(), false );
117 
122  std::multiset<matched_> unique;
123 
124  typename std::multiset<matched_>::iterator mitr = m.begin();
125 
126  double chi2 = 0;
127 
128  for ( ; mitr!=m.end() ; ++mitr ) {
129 
130  int rind = mitr->first();
131  int tind = mitr->second();
132 
133  if ( refused[rind] ) continue;
134  if ( testused[tind] ) continue;
135 
136  refused[rind] = true;
137  testused[tind] = true;
138 
139  unique.insert( *mitr );
140 
141  chi2 += (mitr->d()*mitr->d());
142 
143  }
144 
145 
146  // std::cout << "chi2 of matches " << chi2 << std::endl;
147 
151 
152  // std::cout << "\nmatched" << std::endl;
153 
154  std::map<int, int> matches;
155 
156 
157  mitr = unique.begin();
158  while ( mitr!=unique.end() ) {
159  matches.insert( std::map<int, int>::value_type( mitr->first(), mitr->second() ) );
160  // std::cout << "\tbest match " << *mitr << "\t" << ref[mitr->first()] << "\t" << test[mitr->second()] << std::endl;
161  ++mitr;
162  }
163 
164  return matches;
165 
166  }

◆ revmatched() [1/2]

const rmap_type& TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::revmatched
inlineinherited

Definition at line 60 of file TIDAAssociator.h.

60 { return mrevmatched; }

◆ revmatched() [2/2]

virtual const TIDA::Vertex * TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::revmatched ( TIDA::Vertex t)
inlinevirtualinherited

Definition at line 52 of file TIDAAssociator.h.

52  {
53  typename rmap_type::const_iterator titr = mrevmatched.find(t);
54  if ( titr != mrevmatched.end() ) return titr->second;
55  else return 0;
56  }

◆ size()

unsigned TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::size
inlineinherited

Definition at line 66 of file TIDAAssociator.h.

66 { return mmatched.size(); }

Member Data Documentation

◆ m_d

double BestMatcher< TIDA::Vertex , TIDA::Vertex >::m_d
protectedinherited

Definition at line 171 of file BestMatcher.h.

◆ mmatched

map_type TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::mmatched
protectedinherited

Definition at line 73 of file TIDAAssociator.h.

◆ mname

std::string TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::mname
protectedinherited

Definition at line 71 of file TIDAAssociator.h.

◆ mrevmatched

rmap_type TIDA::Associator< TIDA::Vertex , TIDA::Vertex >::mrevmatched
protectedinherited

Definition at line 74 of file TIDAAssociator.h.


The documentation for this class was generated from the following file:
xAOD::iterator
JetConstituentVector::iterator iterator
Definition: JetConstituentVector.cxx:68
TIDA::Associator
Definition: TIDAAssociator.h:24
python.SystemOfUnits.s
int s
Definition: SystemOfUnits.py:131
python.SystemOfUnits.m
int m
Definition: SystemOfUnits.py:91
VertexMatcher::VertexMatcher
VertexMatcher(const std::string &s, double d=3)
Definition: VertexMatcher.h:26
hist_file_dump.d
d
Definition: hist_file_dump.py:137
TIDA::Associator< TIDA::Vertex, TIDA::Vertex >::matched
const map_type & matched() const
Definition: TIDAAssociator.h:59
TIDA::Associator< TIDA::Vertex, TIDA::Vertex >::mmatched
map_type mmatched
Definition: TIDAAssociator.h:73
TrigInDetValidation_Base.test
test
Definition: TrigInDetValidation_Base.py:147
read_hist_ntuple.t
t
Definition: read_hist_ntuple.py:5
BestMatcher< TIDA::Vertex >::matcher
std::map< int, int > matcher(const std::vector< Tp * > &ref, const std::vector< Tq * > &test)
Definition: BestMatcher.h:91
BestMatcher< TIDA::Vertex >::distance
virtual double distance(const TIDA::Vertex *t0, const TIDA::Vertex *t1) const=0
TIDA::Associator< TIDA::Vertex, TIDA::Vertex >::clear
void clear()
Definition: TIDAAssociator.h:63
parseMapping.v0
def v0
Definition: parseMapping.py:149
lumiFormat.i
int i
Definition: lumiFormat.py:85
chi2
double chi2(TH1 *h0, TH1 *h1)
Definition: comparitor.cxx:523
TIDA::Vertex::z
double z() const
Definition: TIDAVertex.h:53
BestMatcher< TIDA::Vertex >
std::unique
DataModel_detail::iterator< DVL > unique(typename DataModel_detail::iterator< DVL > beg, typename DataModel_detail::iterator< DVL > end)
Specialization of unique for DataVector/List.
Definition: DVL_algorithms.h:135
ref
const boost::regex ref(r_ef)
TIDA::Associator< TIDA::Vertex, TIDA::Vertex >::mrevmatched
rmap_type mrevmatched
Definition: TIDAAssociator.h:74
BestMatcher< TIDA::Vertex >::m_d
double m_d
Definition: BestMatcher.h:171