// -*- c++ -*-

#ifndef LEMON_EXTENDABLE_GRAPH_EXTENDER_H

#define LEMON_EXTENDABLE_GRAPH_EXTENDER_H

namespace lemon {

  template <typename _Base> 

  class ExtendableGraphExtender : public _Base {

  public:

    typedef ExtendableGraphExtender Graph;

    typedef _Base Parent;

    typedef typename Parent::Node Node;

    typedef typename Parent::Edge Edge;

    Node addNode() {

      Node node = Parent::addNode();

      Parent::getNotifier(Node()).add(node);

      return node;

    }

    Edge addEdge(const Node& from, const Node& to) {

      Edge edge = Parent::addEdge(from, to);

      Parent::getNotifier(Edge()).add(edge);

      return edge;

    }

  };

  template <typename _Base> 

  class ExtendableUndirGraphExtender : public _Base {

  public:

    typedef ExtendableUndirGraphExtender Graph;

    typedef _Base Parent;

    typedef typename Parent::Node Node;

    typedef typename Parent::Edge Edge;

    typedef typename Parent::UndirEdge UndirEdge;

    Node addNode() {

      Node node = Parent::addNode();

      Parent::getNotifier(Node()).add(node);

      return node;

    }

    UndirEdge addEdge(const Node& from, const Node& to) {

      UndirEdge uedge = Parent::addEdge(from, to);

      Parent::getNotifier(UndirEdge()).add(uedge);

      std::vector<Edge> edges;

      edges.push_back(Parent::direct(uedge, true));

      edges.push_back(Parent::direct(uedge, false));

      Parent::getNotifier(Edge()).add(edges);

      return uedge;

    }

  };

  template <typename _Base>

  class ExtendableUndirBipartiteGraphExtender : public _Base {

  public:

    typedef _Base Parent;

    typedef ExtendableUndirBipartiteGraphExtender Graph;

    typedef typename Parent::Node Node;

    typedef typename Parent::LowerNode LowerNode;

    typedef typename Parent::UpperNode UpperNode;

    typedef typename Parent::Edge Edge;

    typedef typename Parent::UndirEdge UndirEdge;

    Node addUpperNode() {

      Node node = Parent::addUpperNode();

      Parent::getNotifier(UpperNode()).add(node);

      Parent::getNotifier(Node()).add(node);

      return node;

    }

    Node addLowerNode() {

      Node node = Parent::addLowerNode();

      Parent::getNotifier(LowerNode()).add(node);

      Parent::getNotifier(Node()).add(node);

      return node;

    }

    UndirEdge addEdge(const Node& source, const Node& target) {

      UndirEdge undiredge = Parent::addEdge(source, target);

      Parent::getNotifier(UndirEdge()).add(undiredge);

      std::vector<Edge> edges;

      edges.push_back(Parent::direct(undiredge, true));

      edges.push_back(Parent::direct(undiredge, false));

      Parent::getNotifier(Edge()).add(edges);

      return undiredge;

    }

  };

}

#endif