hepmc - Rev 168

Subversion Repositories:

(root)/branches/HEPMC_02_01_branch/HepMC/TempParticleMap.h

Rev 220 - Go to most recent revision - Blame - Compare with Previous - Last modification - View Log - Download -

//--------------------------------------------------------------------------
#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(), m_counter(0) {}
       
        ~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;
        int           m_counter;
    };
   
    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_counter;
        m_particle_order[m_counter] = p;
        m_particle_to_end_vertex[p] = end_vtx_code;
    }

} // HepMC

#endif  // HEPMC_TempParticleMap_H
//--------------------------------------------------------------------------