GenRanges.h

#ifndef HEPMC_GEN_EVENT_ITERATORS_H
#define HEPMC_GEN_EVENT_ITERATORS_H

//--------------------------------------------------------------------------
// garren@fnal.gov, May 2009
// 
//--------------------------------------------------------------------------

#include <stdexcept>

#include "HepMC/GenEvent.h"
#include "HepMC/GenVertex.h"

namespace HepMC {


class GenEventVertexRange {

public:

  GenEventVertexRange( GenEvent & e ) : m_event(e) {}
  GenEvent::vertex_iterator begin() { return m_event.vertices_begin(); }
  GenEvent::vertex_iterator end()   { return m_event.vertices_end(); }

private:
  GenEventVertexRange& operator=( GenEventVertexRange & );

private:
  GenEvent & m_event;

};


class ConstGenEventVertexRange {

public:

  ConstGenEventVertexRange( GenEvent const & e ) : m_event(e) {}
  GenEvent::vertex_const_iterator begin() const { return m_event.vertices_begin(); }
  GenEvent::vertex_const_iterator end()   const { return m_event.vertices_end(); }

private:
  ConstGenEventVertexRange& operator=( ConstGenEventVertexRange & );

private:
  GenEvent const & m_event;

};


class GenEventParticleRange {

public:

  GenEventParticleRange( GenEvent & e ) : m_event(e) {}
  GenEvent::particle_iterator begin() { return m_event.particles_begin(); }
  GenEvent::particle_iterator end()   { return m_event.particles_end(); }

private:
  GenEventParticleRange& operator=( GenEventParticleRange & );

private:
  GenEvent & m_event;

};


class ConstGenEventParticleRange {

public:

  ConstGenEventParticleRange( GenEvent const & e ) : m_event(e) {}
  GenEvent::particle_const_iterator begin() const { return m_event.particles_begin(); }
  GenEvent::particle_const_iterator end()   const { return m_event.particles_end(); }

private:
  ConstGenEventParticleRange& operator=( ConstGenEventParticleRange & );

private:
  GenEvent const & m_event;

};


class GenVertexParticleRange {

public:

  GenVertexParticleRange( GenVertex & v, IteratorRange range = relatives ) 
  : m_vertex(v),m_range(range) {}
  GenVertex::particle_iterator begin() { return m_vertex.particles_begin(m_range); }
  GenVertex::particle_iterator end()   { return m_vertex.particles_end(m_range); }

private:
  GenVertexParticleRange& operator=( GenVertexParticleRange & );

private:
  GenVertex     & m_vertex;
  IteratorRange   m_range;

};


class GenParticleProductionRange {

public:

  GenParticleProductionRange( GenParticle const & p, IteratorRange range = relatives ) 
  : m_particle(p),m_range(range) {}
  GenVertex::particle_iterator begin();
  GenVertex::particle_iterator end(); 

private:
  GenParticleProductionRange& operator=( GenParticleProductionRange & );

private:
  GenParticle const & m_particle;
  IteratorRange       m_range;

};

class ConstGenParticleProductionRange {

public:

  ConstGenParticleProductionRange( GenParticle const & p, IteratorRange range = relatives ) 
  : m_particle(p),m_range(range) {}
  GenVertex::particle_iterator begin();
  GenVertex::particle_iterator end(); 

private:
  ConstGenParticleProductionRange& operator=( ConstGenParticleProductionRange & );

private:
  GenParticle const & m_particle;
  IteratorRange       m_range;

};


class GenParticleEndRange {

public:

  GenParticleEndRange( GenParticle const & p, IteratorRange range = relatives ) 
  : m_particle(p),m_range(range) {}
  GenVertex::particle_iterator begin();
  GenVertex::particle_iterator end(); 

private:
  GenParticleEndRange& operator=( GenParticleEndRange & );

private:
  GenParticle const & m_particle;
  IteratorRange       m_range;

};

class ConstGenParticleEndRange {

public:

  ConstGenParticleEndRange( GenParticle const & p, IteratorRange range = relatives ) 
  : m_particle(p),m_range(range) {}
  GenVertex::particle_iterator begin();
  GenVertex::particle_iterator end(); 

private:
  ConstGenParticleEndRange& operator=( ConstGenParticleEndRange & );

private:
  GenParticle const & m_particle;
  IteratorRange       m_range;

};


inline GenVertex::particle_iterator GenParticleProductionRange::begin() 
{ 
    if ( ! m_particle.production_vertex() ) 
        throw(std::range_error("GenParticleProductionRange: GenParticle has no production_vertex"));
    return m_particle.production_vertex()->particles_begin(m_range); 
}

inline GenVertex::particle_iterator GenParticleProductionRange::end()
{ 
    if ( ! m_particle.production_vertex() ) 
        throw(std::range_error("GenParticleProductionRange: GenParticle has no production_vertex"));
    return m_particle.production_vertex()->particles_end(m_range); 
}


inline GenVertex::particle_iterator ConstGenParticleProductionRange::begin() 
{ 
    if ( ! m_particle.production_vertex() ) 
        throw(std::range_error("ConstGenParticleProductionRange: GenParticle has no production_vertex"));
    return m_particle.production_vertex()->particles_begin(m_range); 
}

inline GenVertex::particle_iterator ConstGenParticleProductionRange::end()
{ 
    if ( ! m_particle.production_vertex() ) 
        throw(std::range_error("ConstGenParticleProductionRange: GenParticle has no production_vertex"));
    return m_particle.production_vertex()->particles_end(m_range); 
}

inline GenVertex::particle_iterator GenParticleEndRange::begin() 
{ 
    if ( ! m_particle.end_vertex() ) 
        throw(std::range_error("GenParticleEndRange: GenParticle has no end_vertex"));
    return m_particle.end_vertex()->particles_begin(m_range); 
}
inline GenVertex::particle_iterator GenParticleEndRange::end()
{ 
    if ( ! m_particle.end_vertex() ) 
        throw(std::range_error("GenParticleEndRange: GenParticle has no end_vertex"));
    return m_particle.end_vertex()->particles_end(m_range); 
}

inline GenVertex::particle_iterator ConstGenParticleEndRange::begin() 
{ 
    if ( ! m_particle.end_vertex() ) 
        throw(std::range_error("ConstGenParticleEndRange: GenParticle has no end_vertex"));
    return m_particle.end_vertex()->particles_begin(m_range); 
}
inline GenVertex::particle_iterator ConstGenParticleEndRange::end()
{ 
    if ( ! m_particle.end_vertex() ) 
        throw(std::range_error("ConstGenParticleEndRange: GenParticle has no end_vertex"));
    return m_particle.end_vertex()->particles_end(m_range); 
}

} // HepMC

#endif  // HEPMC_GEN_EVENT_ITERATORS_H