RhodeCode

    class EdgeIt : public Parent::Edge { 

      const Graph* graph;

    public:

      EdgeIt() { }

      EdgeIt(Invalid i) : Edge(i) { }

      explicit EdgeIt(const Graph& _graph) : graph(&_graph) {

	_graph.first(static_cast<Edge&>(*this));

      }

      EdgeIt(const Graph& _graph, const Edge& e) : 

	Edge(e), graph(&_graph) { }

      EdgeIt& operator++() { 

	graph->next(*this);

	return *this; 

      }

    };

    class IncEdgeIt : public Parent::Edge {

      friend class EdgeSetExtender;

      const Graph* graph;

      bool direction;

    public:

      IncEdgeIt() { }

      IncEdgeIt(Invalid i) : Edge(i), direction(false) { }

      IncEdgeIt(const Graph& _graph, const Node &n) : graph(&_graph) {

	_graph.firstInc(*this, direction, n);

      }

      IncEdgeIt(const Graph& _graph, const Edge &ue, const Node &n)

	: graph(&_graph), Edge(ue) {

	direction = (_graph.source(ue) == n);

      }

      IncEdgeIt& operator++() {

	graph->nextInc(*this, direction);

	return *this; 

      }

    };

    // \brief Base node of the iterator

    //

    // Returns the base node (ie. the source in this case) of the iterator

    Node baseNode(const OutArcIt &e) const {

      return Parent::source(static_cast<const Arc&>(e));

    }

    // \brief Running node of the iterator

    //

    // Returns the running node (ie. the target in this case) of the

    // iterator

    Node runningNode(const OutArcIt &e) const {

      return Parent::target(static_cast<const Arc&>(e));

    }

    // \brief Base node of the iterator

    //

    // Returns the base node (ie. the target in this case) of the iterator

    Node baseNode(const InArcIt &e) const {

      return Parent::target(static_cast<const Arc&>(e));

    }

    // \brief Running node of the iterator

    //

    // Returns the running node (ie. the source in this case) of the

    // iterator

    Node runningNode(const InArcIt &e) const {

      return Parent::source(static_cast<const Arc&>(e));

    }

    // Base node of the iterator

    //

    // Returns the base node of the iterator

    Node baseNode(const IncEdgeIt &e) const {

      return e.direction ? u(e) : v(e);

    }

    // Running node of the iterator

    //

    // Returns the running node of the iterator

    Node runningNode(const IncEdgeIt &e) const {

      return e.direction ? v(e) : u(e);

    }

    template <typename _Value>

    class ArcMap 

      : public MapExtender<DefaultMap<Graph, Arc, _Value> > {

      typedef MapExtender<DefaultMap<Graph, Arc, _Value> > Parent;

    public:

	: Parent(_g) {}

      ArcMap(const Graph& _g, const _Value& _v) 

	: Parent(_g, _v) {}

      ArcMap& operator=(const ArcMap& cmap) {

	return operator=<ArcMap>(cmap);

      }

      template <typename CMap>

      ArcMap& operator=(const CMap& cmap) {

        Parent::operator=(cmap);

	return *this;

      }

    };

    template <typename _Value>

    class EdgeMap 

      : public MapExtender<DefaultMap<Graph, Edge, _Value> > {

      typedef MapExtender<DefaultMap<Graph, Edge, _Value> > Parent;

    public:

	: Parent(_g) {}

      EdgeMap(const Graph& _g, const _Value& _v) 

	: Parent(_g, _v) {}

      EdgeMap& operator=(const EdgeMap& cmap) {

	return operator=<EdgeMap>(cmap);

      }

      template <typename CMap>

      EdgeMap& operator=(const CMap& cmap) {

        Parent::operator=(cmap);

	return *this;

      }

    };

    // Alteration extension

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

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

      notifier(Edge()).add(edge);

      std::vector<Arc> arcs;

      arcs.push_back(Parent::direct(edge, true));

      arcs.push_back(Parent::direct(edge, false));

      notifier(Arc()).add(arcs);

      return edge;

    }

    void clear() {

      notifier(Arc()).clear();

      notifier(Edge()).clear();

      Parent::clear();

    }

    void erase(const Edge& edge) {

      std::vector<Arc> arcs;

      arcs.push_back(Parent::direct(edge, true));

      arcs.push_back(Parent::direct(edge, false));

      notifier(Arc()).erase(arcs);

      notifier(Edge()).erase(edge);

      Parent::erase(edge);

    }

    EdgeSetExtender() {

      arc_notifier.setContainer(*this);

      edge_notifier.setContainer(*this);

    }

    ~EdgeSetExtender() {

      edge_notifier.clear();

      arc_notifier.clear();

    }

  };

}

#endif

    class EdgeIt : public Parent::Edge {

      const Graph* _graph;

    public:

      EdgeIt() { }

      EdgeIt(Invalid i) : Edge(i) { }

      explicit EdgeIt(const Graph& graph) : _graph(&graph) {

        _graph->first(static_cast<Edge&>(*this));

      }

      EdgeIt(const Graph& graph, const Edge& edge) :

        Edge(edge), _graph(&graph) { }

      EdgeIt& operator++() {

        _graph->next(*this);

        return *this;

      }

    };

    class IncEdgeIt : public Parent::Edge {

      friend class GraphExtender;

      const Graph* _graph;

      bool _direction;

    public:

      IncEdgeIt() { }

      IncEdgeIt(Invalid i) : Edge(i), _direction(false) { }

      IncEdgeIt(const Graph& graph, const Node &node) : _graph(&graph) {

        _graph->firstInc(*this, _direction, node);

      }

      IncEdgeIt(const Graph& graph, const Edge &edge, const Node &node)

        : _graph(&graph), Edge(edge) {

        _direction = (_graph->source(edge) == node);

      }

      IncEdgeIt& operator++() {

        _graph->nextInc(*this, _direction);

        return *this;

      }

    };

    // \brief Base node of the iterator

    //

    // Returns the base node (ie. the source in this case) of the iterator

    Node baseNode(const OutArcIt &arc) const {

      return Parent::source(static_cast<const Arc&>(arc));

    }

    // \brief Running node of the iterator

    //

    // Returns the running node (ie. the target in this case) of the

    // iterator

    Node runningNode(const OutArcIt &arc) const {

      return Parent::target(static_cast<const Arc&>(arc));

    }

    // \brief Base node of the iterator

    //

    // Returns the base node (ie. the target in this case) of the iterator

    Node baseNode(const InArcIt &arc) const {

      return Parent::target(static_cast<const Arc&>(arc));

    }

    // \brief Running node of the iterator

    //

    // Returns the running node (ie. the source in this case) of the

    // iterator

    Node runningNode(const InArcIt &arc) const {

      return Parent::source(static_cast<const Arc&>(arc));

    }

    // Base node of the iterator

    //

    // Returns the base node of the iterator

    Node baseNode(const IncEdgeIt &edge) const {

      return edge._direction ? u(edge) : v(edge);

    }

    // Running node of the iterator

    //

    // Returns the running node of the iterator

    Node runningNode(const IncEdgeIt &edge) const {

      return edge._direction ? v(edge) : u(edge);

    }

    // Mappable extension

    template <typename _Value>

    class NodeMap

      : public MapExtender<DefaultMap<Graph, Node, _Value> > {

      typedef MapExtender<DefaultMap<Graph, Node, _Value> > Parent;

    public:

        : Parent(graph) {}

      NodeMap(const Graph& graph, const _Value& value)

        : Parent(graph, value) {}

    private:

      NodeMap& operator=(const NodeMap& cmap) {

        return operator=<NodeMap>(cmap);

      }

      template <typename CMap>

      NodeMap& operator=(const CMap& cmap) {

        Parent::operator=(cmap);

        return *this;

      }

    };

    template <typename _Value>

    class ArcMap

      : public MapExtender<DefaultMap<Graph, Arc, _Value> > {

      typedef MapExtender<DefaultMap<Graph, Arc, _Value> > Parent;

    public:

        : Parent(graph) {}

      ArcMap(const Graph& graph, const _Value& value)

        : Parent(graph, value) {}

    private:

      ArcMap& operator=(const ArcMap& cmap) {

        return operator=<ArcMap>(cmap);

      }

      template <typename CMap>

      ArcMap& operator=(const CMap& cmap) {

        Parent::operator=(cmap);

        return *this;

      }

    };

    template <typename _Value>

    class EdgeMap

      : public MapExtender<DefaultMap<Graph, Edge, _Value> > {

      typedef MapExtender<DefaultMap<Graph, Edge, _Value> > Parent;

    public:

        : Parent(graph) {}

      EdgeMap(const Graph& graph, const _Value& value)

        : Parent(graph, value) {}

    private:

      EdgeMap& operator=(const EdgeMap& cmap) {

        return operator=<EdgeMap>(cmap);

      }

      template <typename CMap>

      EdgeMap& operator=(const CMap& cmap) {

        Parent::operator=(cmap);

        return *this;

      }

    };

    // Alteration extension

    Node addNode() {

      Node node = Parent::addNode();

      notifier(Node()).add(node);

      return node;

    }

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

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

      notifier(Edge()).add(edge);

      std::vector<Arc> ev;

      ev.push_back(Parent::direct(edge, true));

      ev.push_back(Parent::direct(edge, false));

      notifier(Arc()).add(ev);

      return edge;

    }

    void clear() {

      notifier(Arc()).clear();

      notifier(Edge()).clear();

      notifier(Node()).clear();

      Parent::clear();

    }

    template <typename Graph, typename NodeRefMap, typename EdgeRefMap>

    void build(const Graph& graph, NodeRefMap& nodeRef,

               EdgeRefMap& edgeRef) {

      Parent::build(graph, nodeRef, edgeRef);

      notifier(Node()).build();

      notifier(Edge()).build();

      notifier(Arc()).build();

    }

    void erase(const Node& node) {

      Arc arc;

      Parent::firstOut(arc, node);

      while (arc != INVALID ) {

        erase(arc);

        Parent::firstOut(arc, node);

      }

      Parent::firstIn(arc, node);

      while (arc != INVALID ) {

        erase(arc);

        Parent::firstIn(arc, node);

      }

      notifier(Node()).erase(node);

      Parent::erase(node);

    }

    void erase(const Edge& edge) {

      std::vector<Arc> av;

      av.push_back(Parent::direct(edge, true));

      av.push_back(Parent::direct(edge, false));

      notifier(Arc()).erase(av);

      notifier(Edge()).erase(edge);

      Parent::erase(edge);

    }

    GraphExtender() {

      node_notifier.setContainer(*this);

      arc_notifier.setContainer(*this);

      edge_notifier.setContainer(*this);

    }

    ~GraphExtender() {

      edge_notifier.clear();

      arc_notifier.clear();

      node_notifier.clear();

    }

  };

}

#endif