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;


    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
//--------------------------------------------------------------------------