#pragma once

#ifndef TCG_CONTAINERS_READER_H
#define TCG_CONTAINERS_READER_H

// STD includes
#include <stack>

namespace tcg {

//******************************************************************************
//    Generic Reader declaration
//******************************************************************************

template <typename Val>
class generic_containers_reader {
public:
  typedef Val value_type;

public:
  virtual void openContainer()                     = 0;
  virtual void addElement(const value_type &value) = 0;
  virtual void closeContainer()                    = 0;

  // NOTE: Algorithms may require the openContainer() function to accept the
  // container's header data. Check the algorithms description if necessary.
};

//******************************************************************************
//    Some Actual Container Readers
//******************************************************************************

template <typename Container, typename Val = typename Container::value_type>
class sequential_reader {
  Container *m_values;

public:
  typedef Container container_type;
  typedef Val value_type;

public:
  sequential_reader(Container *container) : m_values(container) {}

  void openContainer() { m_values->clear(); }
  void addElement(const value_type &val) { m_values->push_back(val); }
  void closeContainer() {}

  const container_type *values() const { return m_values; }
  container_type *values() { return m_values; }
};

//------------------------------------------------------------------

template <typename Container, typename Val = typename Container::value_type>
class ordered_reader {
  Container *m_values;

public:
  typedef Container container_type;
  typedef Val value_type;

public:
  ordered_reader(Container &container) : m_values(container) {}

  void openContainer() { m_values->clear(); }
  void addElement(const value_type &val) { m_values->insert(val); }
  void closeContainer() {}

  const container_type *values() const { return m_values; }
  container_type *values() { return m_values; }
};

//------------------------------------------------------------------

template <typename Container, typename ReaderType,
          typename Val = typename ReaderType::value_type>
class multiple_reader {
public:
  typedef Container container_type;
  typedef ReaderType reader_type;
  typedef Val value_type;

protected:
  Container *m_storage;
  std::stack<reader_type> m_stack;

public:
  multiple_reader(Container *storage) : m_storage(storage) {}

  reader_type &reader() { return m_stack.top(); }

  void openContainer() {
    m_stack.push(reader_type(new typename reader_type::container_type));
  }
  void addElement(const value_type &val) { reader().addElement(val); }
  void closeContainer() {
    if (reader().values()->empty())
      delete reader().values();
    else
      m_storage->push_back(reader().values());

    m_stack.pop();
  }

  const container_type &storage() const { return m_storage; }
  container_type &storage() { return m_storage; }
};

}  // namespace tcg

#endif  // TCG_CONTAINERS_READER_H