source: svn/trunk/Utilities/HepMC/interface/TempParticleMap.h@ 929

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