[u/mrichter/AliRoot.git] / TEvtGen / HepMC / TempParticleMap.h

//--------------------------------------------------------------------------
#ifndef HEPMC_TempParticleMap_H
#define HEPMC_TempParticleMap_H

//////////////////////////////////////////////////////////////////////////
// garren@fnal.gov, October 2007
//
// Used by IO classes
//////////////////////////////////////////////////////////////////////////

#include <map>

namespace HepMC {

    class GenParticle;

    //! TempParticleMap is a temporary GenParticle* container used during input.

    ///
    /// \class  TempParticleMap
    /// Used by IO classes for recoverable particle ordering.
    /// Map GenParticle* against both outgoing vertex and particle order.
    ///
    class TempParticleMap {
    public:
        typedef std::map<HepMC::GenParticle*,int> TempMap;
        typedef std::map<int,HepMC::GenParticle*> TempOrderMap;
	typedef TempMap::iterator     TempMapIterator;
	typedef TempOrderMap::iterator  orderIterator;
	
	TempParticleMap() 
	: m_particle_to_end_vertex(), m_particle_order() {}
	
	~TempParticleMap() {}
	
	TempMapIterator begin() { return m_particle_to_end_vertex.begin(); }
	TempMapIterator end() { return m_particle_to_end_vertex.end(); }
	orderIterator order_begin() { return m_particle_order.begin(); }
	orderIterator order_end() { return m_particle_order.end(); }
	
	int end_vertex( GenParticle* );
	
	void addEndParticle( GenParticle*, int& );
	
    private:
        TempMap       m_particle_to_end_vertex;
	TempOrderMap  m_particle_order;
    };
    
    inline int TempParticleMap::end_vertex( GenParticle* p )
    { 
        //return m_particle_to_end_vertex[p]->second; 
	TempMapIterator it = m_particle_to_end_vertex.find(p);
	if( it == end() ) return 0;
	return m_particle_to_end_vertex[p];
    }

    inline void TempParticleMap::addEndParticle( GenParticle* p, int& end_vtx_code )
    {
	m_particle_order[p->barcode()] = p;
        m_particle_to_end_vertex[p] = end_vtx_code;
    }        

} // HepMC

#endif  // HEPMC_TempParticleMap_H
//--------------------------------------------------------------------------
Commit	Line	Data
0ca57c2f	1	//--------------------------------------------------------------------------
	2	#ifndef HEPMC_TempParticleMap_H
	3	#define HEPMC_TempParticleMap_H
	4
	5	//////////////////////////////////////////////////////////////////////////
	6	// garren@fnal.gov, October 2007
	7	//
	8	// Used by IO classes
	9	//////////////////////////////////////////////////////////////////////////
	10
	11	#include <map>
	12
	13	namespace HepMC {
	14
	15	class GenParticle;
	16
	17	//! TempParticleMap is a temporary GenParticle* container used during input.
	18
	19	///
	20	/// \class TempParticleMap
	21	/// Used by IO classes for recoverable particle ordering.
	22	/// Map GenParticle* against both outgoing vertex and particle order.
	23	///
	24	class TempParticleMap {
	25	public:
	26	typedef std::map<HepMC::GenParticle*,int> TempMap;
	27	typedef std::map<int,HepMC::GenParticle*> TempOrderMap;
	28	typedef TempMap::iterator TempMapIterator;
	29	typedef TempOrderMap::iterator orderIterator;
	30
	31	TempParticleMap()
	32	: m_particle_to_end_vertex(), m_particle_order() {}
	33
	34	~TempParticleMap() {}
	35
	36	TempMapIterator begin() { return m_particle_to_end_vertex.begin(); }
	37	TempMapIterator end() { return m_particle_to_end_vertex.end(); }
	38	orderIterator order_begin() { return m_particle_order.begin(); }
	39	orderIterator order_end() { return m_particle_order.end(); }
	40
	41	int end_vertex( GenParticle* );
	42
	43	void addEndParticle( GenParticle*, int& );
	44
	45	private:
	46	TempMap m_particle_to_end_vertex;
	47	TempOrderMap m_particle_order;
	48	};
	49
	50	inline int TempParticleMap::end_vertex( GenParticle* p )
	51	{
	52	//return m_particle_to_end_vertex[p]->second;
	53	TempMapIterator it = m_particle_to_end_vertex.find(p);
	54	if( it == end() ) return 0;
	55	return m_particle_to_end_vertex[p];
	56	}
	57
	58	inline void TempParticleMap::addEndParticle( GenParticle* p, int& end_vtx_code )
	59	{
	60	m_particle_order[p->barcode()] = p;
	61	m_particle_to_end_vertex[p] = end_vtx_code;
	62	}
	63
	64	} // HepMC
65
66	#endif // HEPMC_TempParticleMap_H
67	//--------------------------------------------------------------------------