DynamicContainer.h

Go to the documentation of this file.

#ifndef AMPGEN_DYNAMICCONTAINER_H
#define AMPGEN_DYNAMICCONTAINER_H
#include "AmpGen/MsgService.h"
#include <functional>
#include <vector>
 
namespace AmpGen
{
  template <class TYPE, class CONTAINER_TYPE = std::vector<TYPE>>
  class DynamicContainer
  {
  public:
    class Iterator
    {
    private:
      DynamicContainer* m_container;
      size_t m_pos;
    public:
      Iterator( const size_t& pos, DynamicContainer* parent ) : m_container( parent ), m_pos( pos ) {}
      TYPE* operator->() const { return &( ( *m_container )[m_pos] ); }
      TYPE operator*() const { return ( *m_container )[m_pos]; }
      Iterator operator++() const
      {
        if ( m_pos + 1 % m_container->m_buffer.size() == 0 && m_pos + 1 != m_container->m_totalSize ) {
          m_container->refresh();
        }
        return Iterator( m_pos + 1, m_container );
      }
      Iterator& operator++()
      {
        m_pos++;
        if ( m_pos % m_container->m_buffer.size() == 0 && m_pos % m_container->m_totalSize ) {
          m_container->refresh();
        }
        return *this;
      }
      bool operator==( const Iterator& rhs ) const { return m_pos == rhs.m_pos; }
      bool operator!=( const Iterator& rhs ) const { return m_pos != rhs.m_pos; }
    };
  private:
    CONTAINER_TYPE m_buffer;
    size_t m_totalSize;
    std::function<void( CONTAINER_TYPE& )> m_generator;
  public:
    template <class GENERATOR> DynamicContainer( const size_t& totalSize, const GENERATOR& generator ) : 
      m_totalSize( totalSize ), m_generator( generator )
      {
        refresh();
      }
    Iterator begin() { return Iterator( 0, this ); }
    Iterator end() { return Iterator( m_totalSize, this ); }
    const size_t& size() const { return m_totalSize; }
    TYPE& operator[]( const size_t& pos ) { return m_buffer[pos % m_buffer.size()]; }
    const TYPE& operator[]( const size_t& pos ) const { return m_buffer[pos % m_buffer.size()]; }
    void refresh()
    {
      m_buffer.clear();
      m_generator( m_buffer );
    }
    CONTAINER_TYPE& buffer() { return m_buffer; }
  };
} // namespace AmpGen
#endif