Fork me on GitHub

source: svn/trunk/Utilities/HepMC/src/GenVertex.cc@ 408

Last change on this file since 408 was 349, checked in by severine ovyn, 16 years ago

first test

File size: 32.9 KB
Line 
1//////////////////////////////////////////////////////////////////////////
2// Matt.Dobbs@Cern.CH, September 1999
3// GenVertex within an event
4//////////////////////////////////////////////////////////////////////////
5
6#include "GenParticle.h"
7#include "GenVertex.h"
8#include "GenEvent.h"
9#include "SearchVector.h"
10#include <iomanip> // needed for formatted output
11
12namespace HepMC {
13
14 GenVertex::GenVertex( const FourVector& position,
15 int id, const WeightContainer& weights )
16 : m_position(position), m_id(id), m_weights(weights), m_event(0),
17 m_barcode(0)
18 {}
19 //{
20 //s_counter++;
21 //}
22
23 GenVertex::GenVertex( const GenVertex& invertex )
24 : m_position( invertex.position() ),
25 m_particles_in(),
26 m_particles_out(),
27 m_id( invertex.id() ),
28 m_weights( invertex.weights() ),
29 m_event(0),
30 m_barcode(0)
31 {
32 /// Shallow copy: does not copy the FULL list of particle pointers.
33 /// Creates a copy of - invertex
34 /// - outgoing particles of invertex, but sets the
35 /// decay vertex of these particles to NULL
36 /// - all incoming particles which do not have a
37 /// creation vertex.
38 /// (i.e. it creates copies of all particles which it owns)
39 /// (note - impossible to copy the FULL list of particle pointers
40 /// while having the vertex
41 /// and particles in/out point-back to one another -- unless you
42 /// copy the entire tree -- which we don't want to do)
43 //
44 for ( particles_in_const_iterator
45 part1 = invertex.particles_in_const_begin();
46 part1 != invertex.particles_in_const_end(); ++part1 ) {
47 if ( !(*part1)->production_vertex() ) {
48 GenParticle* pin = new GenParticle(**part1);
49 add_particle_in(pin);
50 }
51 }
52 for ( particles_out_const_iterator
53 part2 = invertex.particles_out_const_begin();
54 part2 != invertex.particles_out_const_end(); part2++ ) {
55 GenParticle* pin = new GenParticle(**part2);
56 add_particle_out(pin);
57 }
58 suggest_barcode( invertex.barcode() );
59 //
60 //s_counter++;
61 }
62
63 GenVertex::~GenVertex() {
64 //
65 // need to delete any particles previously owned by this vertex
66 if ( parent_event() ) parent_event()->remove_barcode(this);
67 delete_adopted_particles();
68 //s_counter--;
69 }
70
71 void GenVertex::swap( GenVertex & other)
72 {
73 m_position.swap( other.m_position );
74 m_particles_in.swap( other.m_particles_in );
75 m_particles_out.swap( other.m_particles_out );
76 std::swap( m_id, other.m_id );
77 m_weights.swap( other.m_weights );
78 std::swap( m_event, other.m_event );
79 std::swap( m_barcode, other.m_barcode );
80 }
81
82 GenVertex& GenVertex::operator=( const GenVertex& invertex ) {
83 /// Shallow: does not copy the FULL list of particle pointers.
84 /// Creates a copy of - invertex
85 /// - outgoing particles of invertex, but sets the
86 /// decay vertex of these particles to NULL
87 /// - all incoming particles which do not have a
88 /// creation vertex.
89 /// - it does not alter *this's m_event (!)
90 /// (i.e. it creates copies of all particles which it owns)
91 /// (note - impossible to copy the FULL list of particle pointers
92 /// while having the vertex
93 /// and particles in/out point-back to one another -- unless you
94 /// copy the entire tree -- which we don't want to do)
95 ///
96
97 // best practices implementation
98 GenVertex tmp( invertex );
99 swap( tmp );
100 return *this;
101 }
102
103 bool GenVertex::operator==( const GenVertex& a ) const {
104 /// Returns true if the positions and the particles in the lists of a
105 /// and this are identical. Does not compare barcodes.
106 /// Note that it is impossible for two vertices to point to the same
107 /// particle's address, so we need to do more than just compare the
108 /// particle pointers
109 //
110 if ( a.position() != this->position() ) return false;
111 // if the size of the inlist differs, return false.
112 if ( a.particles_in_size() != this->particles_in_size() ) return false;
113 // if the size of the outlist differs, return false.
114 if ( a.particles_out_size() != this->particles_out_size() ) return false;
115 // loop over the inlist and ensure particles are identical
116 // (only do this if the lists aren't empty - we already know
117 // if one isn't, then other isn't either!)
118 if ( a.particles_in_const_begin() != a.particles_in_const_end() ) {
119 for ( GenVertex::particles_in_const_iterator
120 ia = a.particles_in_const_begin(),
121 ib = this->particles_in_const_begin();
122 ia != a.particles_in_const_end(); ia++, ib++ ){
123 if ( **ia != **ib ) return false;
124 }
125 }
126 // loop over the outlist and ensure particles are identical
127 // (only do this if the lists aren't empty - we already know
128 // if one isn't, then other isn't either!)
129 if ( a.particles_out_const_begin() != a.particles_out_const_end() ) {
130 for ( GenVertex::particles_out_const_iterator
131 ia = a.particles_out_const_begin(),
132 ib = this->particles_out_const_begin();
133 ia != a.particles_out_const_end(); ia++, ib++ ){
134 if ( **ia != **ib ) return false;
135 }
136 }
137 return true;
138 }
139
140 bool GenVertex::operator!=( const GenVertex& a ) const {
141 // Returns true if the positions and lists of a and this are not equal.
142 return !( a == *this );
143 }
144
145 void GenVertex::print( std::ostream& ostr ) const {
146 if ( barcode()!=0 ) {
147 if ( position() != FourVector(0,0,0,0) ) {
148 ostr << "Vertex:";
149 ostr.width(9);
150 ostr << barcode();
151 ostr << " ID:";
152 ostr.width(5);
153 ostr << id();
154 ostr << " (X,cT)=";
155 ostr.width(9);
156 ostr.precision(2);
157 ostr.setf(std::ios::scientific, std::ios::floatfield);
158 ostr.setf(std::ios_base::showpos);
159 ostr << position().x() << ",";
160 ostr.width(9);
161 ostr.precision(2);
162 ostr << position().y() << ",";
163 ostr.width(9);
164 ostr.precision(2);
165 ostr << position().z() << ",";
166 ostr.width(9);
167 ostr.precision(2);
168 ostr << position().t();
169 ostr.setf(std::ios::fmtflags(0), std::ios::floatfield);
170 ostr.unsetf(std::ios_base::showpos);
171 ostr << std::endl;
172 } else {
173 ostr << "GenVertex:";
174 ostr.width(9);
175 ostr << barcode();
176 ostr << " ID:";
177 ostr.width(5);
178 ostr << id();
179 ostr << " (X,cT):0";
180 ostr << std::endl;
181 }
182 } else {
183 // If the vertex doesn't have a unique barcode assigned, then
184 // we print its memory address instead... so that the
185 // print out gives us a unique tag for the particle.
186 if ( position() != FourVector(0,0,0,0) ) {
187 ostr << "Vertex:";
188 ostr.width(9);
189 ostr << (void*)this;
190 ostr << " ID:";
191 ostr.width(5);
192 ostr << id();
193 ostr << " (X,cT)=";
194 ostr.width(9);
195 ostr.precision(2);
196 ostr.setf(std::ios::scientific, std::ios::floatfield);
197 ostr.setf(std::ios_base::showpos);
198 ostr << position().x();
199 ostr.width(9);
200 ostr.precision(2);
201 ostr << position().y();
202 ostr.width(9);
203 ostr.precision(2);
204 ostr << position().z();
205 ostr.width(9);
206 ostr.precision(2);
207 ostr << position().t();
208 ostr.setf(std::ios::fmtflags(0), std::ios::floatfield);
209 ostr.unsetf(std::ios_base::showpos);
210 ostr << std::endl;
211 } else {
212 ostr << "GenVertex:";
213 ostr.width(9);
214 ostr << (void*)this;
215 ostr << " ID:";
216 ostr.width(5);
217 ostr << id();
218 ostr << " (X,cT):0";
219 ostr << std::endl;
220 }
221 }
222
223 // print the weights if there are any
224 if ( ! weights().empty() ) {
225 ostr << " Wgts(" << weights().size() << ")=";
226 for ( WeightContainer::const_iterator wgt = weights().begin();
227 wgt != weights().end(); wgt++ ) { ostr << *wgt << " "; }
228 ostr << std::endl;
229 }
230 // print out all the incoming, then outgoing particles
231 for ( particles_in_const_iterator part1 = particles_in_const_begin();
232 part1 != particles_in_const_end(); part1++ ) {
233 if ( part1 == particles_in_const_begin() ) {
234 ostr << " I:";
235 ostr.width(2);
236 ostr << m_particles_in.size();
237 } else { ostr << " "; }
238 //(*part1)->print( ostr ); //uncomment for long debugging printout
239 ostr << **part1 << std::endl;
240 }
241 for ( particles_out_const_iterator part2 = particles_out_const_begin();
242 part2 != particles_out_const_end(); part2++ ) {
243 if ( part2 == particles_out_const_begin() ) {
244 ostr << " O:";
245 ostr.width(2);
246 ostr << m_particles_out.size();
247 } else { ostr << " "; }
248 //(*part2)->print( ostr ); // uncomment for long debugging printout
249 ostr << **part2 << std::endl;
250 }
251 }
252
253 double GenVertex::check_momentum_conservation() const {
254 /// finds the difference between the total momentum out and the total
255 /// momentum in vectors, and returns the magnitude of this vector
256 /// i.e. returns | vec{p_in} - vec{p_out} |
257 double sumpx = 0, sumpy = 0, sumpz = 0;
258 for ( particles_in_const_iterator part1 = particles_in_const_begin();
259 part1 != particles_in_const_end(); part1++ ) {
260 sumpx += (*part1)->momentum().px();
261 sumpy += (*part1)->momentum().py();
262 sumpz += (*part1)->momentum().pz();
263 }
264 for ( particles_out_const_iterator part2 = particles_out_const_begin();
265 part2 != particles_out_const_end(); part2++ ) {
266 sumpx -= (*part2)->momentum().px();
267 sumpy -= (*part2)->momentum().py();
268 sumpz -= (*part2)->momentum().pz();
269 }
270 return sqrt( sumpx*sumpx + sumpy*sumpy + sumpz*sumpz );
271 }
272
273 void GenVertex::add_particle_in( GenParticle* inparticle ) {
274 if ( !inparticle ) return;
275 // if inparticle previously had a decay vertex, remove it from that
276 // vertex's list
277 if ( inparticle->end_vertex() ) {
278 inparticle->end_vertex()->remove_particle_in( inparticle );
279 }
280 m_particles_in.push_back( inparticle );
281 inparticle->set_end_vertex_( this );
282 }
283
284 void GenVertex::add_particle_out( GenParticle* outparticle ) {
285 if ( !outparticle ) return;
286 // if outparticle previously had a production vertex,
287 // remove it from that vertex's list
288 if ( outparticle->production_vertex() ) {
289 outparticle->production_vertex()->remove_particle_out( outparticle );
290 }
291 m_particles_out.push_back( outparticle );
292 outparticle->set_production_vertex_( this );
293 }
294
295 GenParticle* GenVertex::remove_particle( GenParticle* particle ) {
296 /// this finds *particle in the in and/or out list and removes it from
297 /// these lists ... it DOES NOT DELETE THE PARTICLE or its relations.
298 /// you could delete the particle too as follows:
299 /// delete vtx->remove_particle( particle );
300 /// or if the particle has an end vertex, you could:
301 /// delete vtx->remove_particle( particle )->end_vertex();
302 /// which would delete the particle's end vertex, and thus would
303 /// also delete the particle, since the particle would be
304 /// owned by the end vertex.
305 if ( !particle ) return 0;
306 if ( particle->end_vertex() == this ) {
307 particle->set_end_vertex_( 0 );
308 remove_particle_in(particle);
309 }
310 if ( particle->production_vertex() == this ) {
311 particle->set_production_vertex_(0);
312 remove_particle_out(particle);
313 }
314 return particle;
315 }
316
317 void GenVertex::remove_particle_in( GenParticle* particle ) {
318 /// this finds *particle in m_particles_in and removes it from that list
319 if ( !particle ) return;
320 m_particles_in.erase( already_in_vector( &m_particles_in, particle ) );
321 }
322
323 void GenVertex::remove_particle_out( GenParticle* particle ) {
324 /// this finds *particle in m_particles_out and removes it from that list
325 if ( !particle ) return;
326 m_particles_out.erase( already_in_vector( &m_particles_out, particle ) );
327 }
328
329 void GenVertex::delete_adopted_particles() {
330 /// deletes all particles which this vertex owns
331 /// to be used by the vertex destructor and operator=
332 //
333 if ( m_particles_out.empty() && m_particles_in.empty() ) return;
334 // 1. delete all outgoing particles which don't have decay vertices.
335 // those that do become the responsibility of the decay vertex
336 // and have their productionvertex pointer set to NULL
337 for ( std::vector<GenParticle*>::iterator part1 = m_particles_out.begin();
338 part1 != m_particles_out.end(); ) {
339 if ( !(*part1)->end_vertex() ) {
340 delete *(part1++);
341 } else {
342 (*part1)->set_production_vertex_(0);
343 ++part1;
344 }
345 }
346 m_particles_out.clear();
347 //
348 // 2. delete all incoming particles which don't have production
349 // vertices. those that do become the responsibility of the
350 // production vertex and have their decayvertex pointer set to NULL
351 for ( std::vector<GenParticle*>::iterator part2 = m_particles_in.begin();
352 part2 != m_particles_in.end(); ) {
353 if ( !(*part2)->production_vertex() ) {
354 delete *(part2++);
355 } else {
356 (*part2)->set_end_vertex_(0);
357 ++part2;
358 }
359 }
360 m_particles_in.clear();
361 }
362
363 bool GenVertex::suggest_barcode( int the_bar_code )
364 {
365 /// allows a barcode to be suggested for this vertex.
366 /// In general it is better to let the event pick the barcode for
367 /// you, which is automatic.
368 /// Returns TRUE if the suggested barcode has been accepted (i.e. the
369 /// suggested barcode has not already been used in the event,
370 /// and so it was used).
371 /// Returns FALSE if the suggested barcode was rejected, or if the
372 /// vertex is not yet part of an event, such that it is not yet
373 /// possible to know if the suggested barcode will be accepted).
374 if ( the_bar_code >0 ) {
375 std::cerr << "GenVertex::suggest_barcode WARNING, vertex bar codes"
376 << "\n MUST be negative integers. Positive integers "
377 << "\n are reserved for particles only. Your suggestion "
378 << "\n has been rejected." << std::endl;
379 return false;
380 }
381 bool success = false;
382 if ( parent_event() ) {
383 success = parent_event()->set_barcode( this, the_bar_code );
384 } else { set_barcode_( the_bar_code ); }
385 return success;
386 }
387
388 void GenVertex::set_parent_event_( GenEvent* new_evt )
389 {
390 GenEvent* orig_evt = m_event;
391 m_event = new_evt;
392 //
393 // every time a vertex's parent event changes, the map of barcodes
394 // in the new and old parent event needs to be modified to
395 // reflect this
396 if ( orig_evt != new_evt ) {
397 if (new_evt) new_evt->set_barcode( this, barcode() );
398 if (orig_evt) orig_evt->remove_barcode( this );
399 // we also need to loop over all the particles which are owned by
400 // this vertex, and remove their barcodes from the old event.
401 for ( particles_in_const_iterator part1=particles_in_const_begin();
402 part1 != particles_in_const_end(); part1++ ) {
403 if ( !(*part1)->production_vertex() ) {
404 if ( orig_evt ) orig_evt->remove_barcode( *part1 );
405 if ( new_evt ) new_evt->set_barcode( *part1,
406 (*part1)->barcode() );
407 }
408 }
409 for ( particles_out_const_iterator
410 part2 = particles_out_const_begin();
411 part2 != particles_out_const_end(); part2++ ) {
412 if ( orig_evt ) orig_evt->remove_barcode( *part2 );
413 if ( new_evt ) new_evt->set_barcode( *part2,
414 (*part2)->barcode() );
415 }
416 }
417 }
418
419 void GenVertex::change_parent_event_( GenEvent* new_evt )
420 {
421 //
422 // this method is for use with swap
423 // particles and vertices have already been exchanged,
424 // but the backpointer needs to be fixed
425 //GenEvent* orig_evt = m_event;
426 m_event = new_evt;
427 }
428
429 /////////////
430 // Static //
431 /////////////
432 //unsigned int GenVertex::counter() { return s_counter; }
433 //unsigned int GenVertex::s_counter = 0;
434
435 /////////////
436 // Friends //
437 /////////////
438
439 /// send vertex information to ostr for printing
440 std::ostream& operator<<( std::ostream& ostr, const GenVertex& vtx ) {
441 if ( vtx.barcode()!=0 ) ostr << "BarCode " << vtx.barcode();
442 else ostr << "Address " << &vtx;
443 ostr << " (X,cT)=";
444 if ( vtx.position() != FourVector(0,0,0,0)) {
445 ostr << vtx.position().x() << ","
446 << vtx.position().y() << ","
447 << vtx.position().z() << ","
448 << vtx.position().t();
449 } else { ostr << 0; }
450 ostr << " #in:" << vtx.particles_in_size()
451 << " #out:" << vtx.particles_out_size();
452 return ostr;
453 }
454
455 /////////////////////////////
456 // edge_iterator // (protected - for internal use only)
457 /////////////////////////////
458 // If the user wants the functionality of the edge_iterator, he should
459 // use particle_iterator with IteratorRange = family, parents, or children
460 //
461
462 GenVertex::edge_iterator::edge_iterator() : m_vertex(0), m_range(family),
463 m_is_inparticle_iter(false), m_is_past_end(true)
464 {}
465
466 GenVertex::edge_iterator::edge_iterator( const GenVertex& vtx,
467 IteratorRange range ) :
468 m_vertex(&vtx), m_range(family)
469 {
470 // Note: (26.1.2000) the original version of edge_iterator inheritted
471 // from set<GenParticle*>::const_iterator() rather than using
472 // composition as it does now.
473 // The inheritted version suffered from a strange bug, which
474 // I have not fully understood --- it only occurred after many
475 // events were processed and only when I called the delete
476 // function on past events. I believe it had something to do with
477 // the past the end values, which are now robustly coded in this
478 // version as boolean members.
479 //
480 // default range is family, only other choices are children/parents
481 // descendants/ancestors not allowed & recasted ot children/parents
482 if ( range == descendants || range == children ) m_range = children;
483 if ( range == ancestors || range == parents ) m_range = parents;
484 //
485 if ( m_vertex->m_particles_in.empty() &&
486 m_vertex->m_particles_out.empty() ) {
487 // Case: particles_in and particles_out is empty.
488 m_is_inparticle_iter = false;
489 m_is_past_end = true;
490 } else if ( m_range == parents && m_vertex->m_particles_in.empty() ){
491 // Case: particles in is empty and parents is requested.
492 m_is_inparticle_iter = true;
493 m_is_past_end = true;
494 } else if ( m_range == children && m_vertex->m_particles_out.empty() ){
495 // Case: particles out is empty and children is requested.
496 m_is_inparticle_iter = false;
497 m_is_past_end = true;
498 } else if ( m_range == children ) {
499 // Case: particles out is NOT empty, and children is requested
500 m_set_iter = m_vertex->m_particles_out.begin();
501 m_is_inparticle_iter = false;
502 m_is_past_end = false;
503 } else if ( m_range == family && m_vertex->m_particles_in.empty() ) {
504 // Case: particles in is empty, particles out is NOT empty,
505 // and family is requested. Then skip ahead to partilces out.
506 m_set_iter = m_vertex->m_particles_out.begin();
507 m_is_inparticle_iter = false;
508 m_is_past_end = false;
509 } else {
510 // Normal scenario: start with the first incoming particle
511 m_set_iter = m_vertex->m_particles_in.begin();
512 m_is_inparticle_iter = true;
513 m_is_past_end = false;
514 }
515 }
516
517 GenVertex::edge_iterator::edge_iterator( const edge_iterator& p ) {
518 *this = p;
519 }
520
521 GenVertex::edge_iterator::~edge_iterator() {}
522
523 GenVertex::edge_iterator& GenVertex::edge_iterator::operator=(
524 const edge_iterator& p ) {
525 m_vertex = p.m_vertex;
526 m_range = p.m_range;
527 m_set_iter = p.m_set_iter;
528 m_is_inparticle_iter = p.m_is_inparticle_iter;
529 m_is_past_end = p.m_is_past_end;
530 return *this;
531 }
532
533 GenParticle* GenVertex::edge_iterator::operator*(void) const {
534 if ( !m_vertex || m_is_past_end ) return 0;
535 return *m_set_iter;
536 }
537
538 GenVertex::edge_iterator& GenVertex::edge_iterator::operator++(void){
539 // Pre-fix increment
540 //
541 // increment the set iterator (unless we're past the end value)
542 if ( m_is_past_end ) return *this;
543 ++m_set_iter;
544 // handle cases where m_set_iter points past the end
545 if ( m_range == family && m_is_inparticle_iter &&
546 m_set_iter == m_vertex->m_particles_in.end() ) {
547 // at the end on in particle set, and range is family, so move to
548 // out particle set
549 m_set_iter = m_vertex->m_particles_out.begin();
550 m_is_inparticle_iter = false;
551 } else if ( m_range == parents &&
552 m_set_iter == m_vertex->m_particles_in.end() ) {
553 // at the end on in particle set, and range is parents only, so
554 // move into past the end state
555 m_is_past_end = true;
556 // might as well bail out now
557 return *this;
558 }
559 // are we iterating over input or output particles?
560 if( m_is_inparticle_iter ) {
561 // the following is not else if because we might have range=family
562 // with an empty particles_in set.
563 if ( m_set_iter == m_vertex->m_particles_in.end() ) {
564 //whenever out particles end is reached, go into past the end state
565 m_is_past_end = true;
566 }
567 } else {
568 // the following is not else if because we might have range=family
569 // with an empty particles_out set.
570 if ( m_set_iter == m_vertex->m_particles_out.end() ) {
571 //whenever out particles end is reached, go into past the end state
572 m_is_past_end = true;
573 }
574 }
575 return *this;
576 }
577
578 GenVertex::edge_iterator GenVertex::edge_iterator::operator++(int){
579 // Post-fix increment
580 edge_iterator returnvalue = *this;
581 ++*this;
582 return returnvalue;
583 }
584
585 bool GenVertex::edge_iterator::is_parent() const {
586 if ( **this && (**this)->end_vertex() == m_vertex ) return true;
587 return false;
588 }
589
590 bool GenVertex::edge_iterator::is_child() const {
591 if ( **this && (**this)->production_vertex() == m_vertex ) return true;
592 return false;
593 }
594
595 int GenVertex::edges_size( IteratorRange range ) const {
596 if ( range == children ) return m_particles_out.size();
597 if ( range == parents ) return m_particles_in.size();
598 if ( range == family ) return m_particles_out.size()
599 + m_particles_in.size();
600 return 0;
601 }
602
603 /////////////////////
604 // vertex_iterator //
605 /////////////////////
606
607 GenVertex::vertex_iterator::vertex_iterator()
608 : m_vertex(0), m_range(), m_visited_vertices(0), m_it_owns_set(0),
609 m_recursive_iterator(0)
610 {}
611
612 GenVertex::vertex_iterator::vertex_iterator( GenVertex& vtx_root,
613 IteratorRange range )
614 : m_vertex(&vtx_root), m_range(range)
615 {
616 // standard public constructor
617 //
618 m_visited_vertices = new std::set<const GenVertex*>;
619 m_it_owns_set = 1;
620 m_visited_vertices->insert( m_vertex );
621 m_recursive_iterator = 0;
622 m_edge = m_vertex->edges_begin( m_range );
623 // advance to the first good return value
624 if ( !follow_edge_() &&
625 m_edge != m_vertex->edges_end( m_range )) ++*this;
626 }
627
628 GenVertex::vertex_iterator::vertex_iterator( GenVertex& vtx_root,
629 IteratorRange range, std::set<const HepMC::GenVertex*>& visited_vertices ) :
630 m_vertex(&vtx_root), m_range(range),
631 m_visited_vertices(&visited_vertices), m_it_owns_set(0),
632 m_recursive_iterator(0)
633 {
634 // This constuctor is only to be called internally by this class
635 // for use with its recursive member pointer (m_recursive_iterator).
636 // Note: we do not need to insert m_vertex_root in the vertex - that is
637 // the responsibility of this iterator's mother, which is normally
638 // done just before calling this protected constructor.
639 m_edge = m_vertex->edges_begin( m_range );
640 // advance to the first good return value
641 if ( !follow_edge_() &&
642 m_edge != m_vertex->edges_end( m_range )) ++*this;
643 }
644
645 GenVertex::vertex_iterator::vertex_iterator( const vertex_iterator& v_iter)
646 : m_vertex(0), m_visited_vertices(0), m_it_owns_set(0),
647 m_recursive_iterator(0)
648 {
649 *this = v_iter;
650 }
651
652 GenVertex::vertex_iterator::~vertex_iterator() {
653 if ( m_recursive_iterator ) delete m_recursive_iterator;
654 if ( m_it_owns_set ) delete m_visited_vertices;
655 }
656
657 GenVertex::vertex_iterator& GenVertex::vertex_iterator::operator=(
658 const vertex_iterator& v_iter )
659 {
660 // Note: when copying a vertex_iterator that is NOT the owner
661 // of its set container, the pointer to the set is copied. Beware!
662 // (see copy_with_own_set() if you want a different set pointed to)
663 // In practise the user never needs to worry
664 // since such iterators are only intended to be used internally.
665 //
666 // destruct data member pointers
667 if ( m_recursive_iterator ) delete m_recursive_iterator;
668 m_recursive_iterator = 0;
669 if ( m_it_owns_set ) delete m_visited_vertices;
670 m_visited_vertices = 0;
671 m_it_owns_set = 0;
672 // copy the target vertex_iterator to this iterator
673 m_vertex = v_iter.m_vertex;
674 m_range = v_iter.m_range;
675 if ( v_iter.m_it_owns_set ) {
676 // i.e. this vertex will own its set if v_iter points to any
677 // vertex set regardless of whether v_iter owns the set or not!
678 m_visited_vertices =
679 new std::set<const GenVertex*>(*v_iter.m_visited_vertices);
680 m_it_owns_set = 1;
681 } else {
682 m_visited_vertices = v_iter.m_visited_vertices;
683 m_it_owns_set = 0;
684 }
685 //
686 // Note: m_vertex_root is already included in the set of
687 // tv_iter.m_visited_vertices, we do not need to insert it.
688 //
689 m_edge = v_iter.m_edge;
690 copy_recursive_iterator_( v_iter.m_recursive_iterator );
691 return *this;
692 }
693
694 GenVertex* GenVertex::vertex_iterator::operator*(void) const {
695 // de-reference operator
696 //
697 // if this iterator has an iterator_node, then we return the iterator
698 // node.
699 if ( m_recursive_iterator ) return **m_recursive_iterator;
700 //
701 // an iterator can only return its m_vertex -- any other vertex
702 // is returned by means of a recursive iterator_node
703 // (so this is the only place in the iterator code that a vertex
704 // is returned!)
705 if ( m_vertex ) return m_vertex;
706 return 0;
707 }
708
709 GenVertex::vertex_iterator& GenVertex::vertex_iterator::operator++(void) {
710 // Pre-fix incremental operator
711 //
712 // check for "past the end condition" denoted by m_vertex=0
713 if ( !m_vertex ) return *this;
714 // if at the last edge, move into the "past the end condition"
715 if ( m_edge == m_vertex->edges_end( m_range ) ) {
716 m_vertex = 0;
717 return *this;
718 }
719 // check to see if we need to create a new recursive iterator by
720 // following the current edge only if a recursive iterator doesn't
721 // already exist. If a new recursive_iterator is created, return it.
722 if ( follow_edge_() ) {
723 return *this;
724 }
725 //
726 // if a recursive iterator already exists, increment it, and return its
727 // value (unless the recursive iterator has null root_vertex [its
728 // root vertex is set to null if it has already returned its root]
729 // - in which case we delete it)
730 // and return the vertex pointed to by the edge.
731 if ( m_recursive_iterator ) {
732 ++(*m_recursive_iterator);
733 if ( **m_recursive_iterator ) {
734 return *this;
735 } else {
736 delete m_recursive_iterator;
737 m_recursive_iterator = 0;
738 }
739 }
740 //
741 // increment to the next particle edge
742 ++m_edge;
743 // if m_edge is at the end, then we have incremented through all
744 // edges, and it is time to return m_vertex, which we accomplish
745 // by returning *this
746 if ( m_edge == m_vertex->edges_end( m_range ) ) return *this;
747 // otherwise we follow the current edge by recursively ++ing.
748 return ++(*this);
749 }
750
751 GenVertex::vertex_iterator GenVertex::vertex_iterator::operator++(int) {
752 // Post-fix increment
753 vertex_iterator returnvalue(*this);
754 ++(*this);
755 return returnvalue;
756 }
757
758 void GenVertex::vertex_iterator::copy_with_own_set(
759 const vertex_iterator& v_iter,
760 std::set<const HepMC::GenVertex*>& visited_vertices ) {
761 /// intended for internal use only. (use with care!)
762 /// this is the same as the operator= method, but it allows the
763 /// user to specify which set container m_visited_vertices points to.
764 /// in all cases, this vertex will NOT own its set.
765 //
766 // destruct data member pointers
767 if ( m_recursive_iterator ) delete m_recursive_iterator;
768 m_recursive_iterator = 0;
769 if ( m_it_owns_set ) delete m_visited_vertices;
770 m_visited_vertices = 0;
771 m_it_owns_set = false;
772 // copy the target vertex_iterator to this iterator
773 m_vertex = v_iter.m_vertex;
774 m_range = v_iter.m_range;
775 m_visited_vertices = &visited_vertices;
776 m_it_owns_set = false;
777 m_edge = v_iter.m_edge;
778 copy_recursive_iterator_( v_iter.m_recursive_iterator );
779 }
780
781 GenVertex* GenVertex::vertex_iterator::follow_edge_() {
782 // follows the edge pointed to by m_edge by creating a
783 // recursive iterator for it.
784 //
785 // if a m_recursive_iterator already exists,
786 // this routine has nothing to do,
787 // if there's no m_vertex, there's no point following anything,
788 // also there's no point trying to follow a null edge.
789 if ( m_recursive_iterator || !m_vertex || !*m_edge ) return 0;
790 //
791 // if the range is parents, children, or family (i.e. <= family)
792 // then only the iterator which owns the set is allowed to create
793 // recursive iterators (i.e. recursivity is only allowed to go one
794 // layer deep)
795 if ( m_range <= family && m_it_owns_set == 0 ) return 0;
796 //
797 // M.Dobbs 2001-07-16
798 // Take care of the very special-rare case where a particle might
799 // point to the same vertex for both production and end
800 if ( (*m_edge)->production_vertex() ==
801 (*m_edge)->end_vertex() ) return 0;
802 //
803 // figure out which vertex m_edge is pointing to
804 GenVertex* vtx = ( m_edge.is_parent() ?
805 (*m_edge)->production_vertex() :
806 (*m_edge)->end_vertex() );
807 // if the pointed to vertex doesn't exist or has already been visited,
808 // then return null
809 if ( !vtx || !(m_visited_vertices->insert(vtx).second) ) return 0;
810 // follow that edge by creating a recursive iterator
811 m_recursive_iterator = new vertex_iterator( *vtx, m_range,
812 *m_visited_vertices);
813 // and return the vertex pointed to by m_recursive_iterator
814 return **m_recursive_iterator;
815 }
816
817 void GenVertex::vertex_iterator::copy_recursive_iterator_(
818 const vertex_iterator* recursive_v_iter ) {
819 // to properly copy the recursive iterator, we need to ensure
820 // the proper set container is transfered ... then do this
821 // operation .... you guessed it .... recursively!
822 //
823 if ( !recursive_v_iter ) return;
824 m_recursive_iterator = new vertex_iterator();
825 m_recursive_iterator->m_vertex = recursive_v_iter->m_vertex;
826 m_recursive_iterator->m_range = recursive_v_iter->m_range;
827 m_recursive_iterator->m_visited_vertices = m_visited_vertices;
828 m_recursive_iterator->m_it_owns_set = 0;
829 m_recursive_iterator->m_edge = recursive_v_iter->m_edge;
830 m_recursive_iterator->copy_recursive_iterator_(
831 recursive_v_iter->m_recursive_iterator );
832 }
833
834 ///////////////////////////////
835 // particle_iterator //
836 ///////////////////////////////
837
838 GenVertex::particle_iterator::particle_iterator() {}
839
840 GenVertex::particle_iterator::particle_iterator( GenVertex& vertex_root,
841 IteratorRange range ) {
842 // General Purpose Constructor
843 //
844 if ( range <= family ) {
845 m_edge = GenVertex::edge_iterator( vertex_root, range );
846 } else {
847 m_vertex_iterator = GenVertex::vertex_iterator(vertex_root, range);
848 m_edge = GenVertex::edge_iterator( **m_vertex_iterator,
849 m_vertex_iterator.range() );
850 }
851 advance_to_first_();
852 }
853
854 GenVertex::particle_iterator::particle_iterator(
855 const particle_iterator& p_iter ){
856 *this = p_iter;
857 }
858
859 GenVertex::particle_iterator::~particle_iterator() {}
860
861 GenVertex::particle_iterator&
862 GenVertex::particle_iterator::operator=( const particle_iterator& p_iter )
863 {
864 m_vertex_iterator = p_iter.m_vertex_iterator;
865 m_edge = p_iter.m_edge;
866 return *this;
867 }
868
869 GenParticle* GenVertex::particle_iterator::operator*(void) const {
870 return *m_edge;
871 }
872
873 GenVertex::particle_iterator&
874 GenVertex::particle_iterator::operator++(void) {
875 //Pre-fix increment
876 //
877 if ( *m_edge ) {
878 ++m_edge;
879 } else if ( *m_vertex_iterator ) { // !*m_edge is implicit
880 // past end of edge, but still have more vertices to visit
881 // increment the vertex, checking that the result is valid
882 if ( !*(++m_vertex_iterator) ) return *this;
883 m_edge = GenVertex::edge_iterator( **m_vertex_iterator,
884 m_vertex_iterator.range() );
885 } else { // !*m_edge and !*m_vertex_iterator are implicit
886 // past the end condition: do nothing
887 return *this;
888 }
889 advance_to_first_();
890 return *this;
891 }
892
893 GenVertex::particle_iterator GenVertex::particle_iterator::operator++(int){
894 //Post-fix increment
895 particle_iterator returnvalue(*this);
896 ++(*this);
897 return returnvalue;
898 }
899
900 GenParticle* GenVertex::particle_iterator::advance_to_first_() {
901 /// if the current edge is not a suitable return value ( because
902 /// it is a parent of the vertex root that itself belongs to a
903 /// different vertex ) it advances to the first suitable return value
904 if ( !*m_edge ) return *(++*this);
905 // if the range is relatives, we need to uniquely assign each particle
906 // to a single vertex so as to guarantee particles are returned
907 // exactly once.
908 if ( m_vertex_iterator.range() == relatives &&
909 m_edge.is_parent() &&
910 (*m_edge)->production_vertex() ) return *(++*this);
911 return *m_edge;
912 }
913
914 /// scale the position vector
915 /// this method is only for use by GenEvent
916 /// convert_position assumes that 4th component of the position vector
917 /// is ctau rather than time and has units of length-time
918 void GenVertex::convert_position( const double& f ) {
919 m_position = FourVector( f*m_position.x(),
920 f*m_position.y(),
921 f*m_position.z(),
922 f*m_position.t() );
923 }
924
925} // HepMC
Note: See TracBrowser for help on using the repository browser.