lemon/edge_set.h
author Alpar Juttner <alpar@cs.elte.hu>
Thu, 05 Nov 2009 10:01:02 +0100
changeset 779 c160bf9f18ef
parent 670 926c47568a56
child 787 c2230649a493
permissions -rw-r--r--
Merge
     1 /* -*- mode: C++; indent-tabs-mode: nil; -*-
     2  *
     3  * This file is a part of LEMON, a generic C++ optimization library.
     4  *
     5  * Copyright (C) 2003-2008
     6  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
     7  * (Egervary Research Group on Combinatorial Optimization, EGRES).
     8  *
     9  * Permission to use, modify and distribute this software is granted
    10  * provided that this copyright notice appears in all copies. For
    11  * precise terms see the accompanying LICENSE file.
    12  *
    13  * This software is provided "AS IS" with no warranty of any kind,
    14  * express or implied, and with no claim as to its suitability for any
    15  * purpose.
    16  *
    17  */
    18 
    19 #ifndef LEMON_EDGE_SET_H
    20 #define LEMON_EDGE_SET_H
    21 
    22 #include <lemon/core.h>
    23 #include <lemon/bits/edge_set_extender.h>
    24 
    25 /// \ingroup graphs
    26 /// \file
    27 /// \brief ArcSet and EdgeSet classes.
    28 ///
    29 /// Graphs which use another graph's node-set as own.
    30 namespace lemon {
    31 
    32   template <typename GR>
    33   class ListArcSetBase {
    34   public:
    35 
    36     typedef typename GR::Node Node;
    37     typedef typename GR::NodeIt NodeIt;
    38 
    39   protected:
    40 
    41     struct NodeT {
    42       int first_out, first_in;
    43       NodeT() : first_out(-1), first_in(-1) {}
    44     };
    45 
    46     typedef typename ItemSetTraits<GR, Node>::
    47     template Map<NodeT>::Type NodesImplBase;
    48 
    49     NodesImplBase* _nodes;
    50 
    51     struct ArcT {
    52       Node source, target;
    53       int next_out, next_in;
    54       int prev_out, prev_in;
    55       ArcT() : prev_out(-1), prev_in(-1) {}
    56     };
    57 
    58     std::vector<ArcT> arcs;
    59 
    60     int first_arc;
    61     int first_free_arc;
    62 
    63     const GR* _graph;
    64 
    65     void initalize(const GR& graph, NodesImplBase& nodes) {
    66       _graph = &graph;
    67       _nodes = &nodes;
    68     }
    69 
    70   public:
    71 
    72     class Arc {
    73       friend class ListArcSetBase<GR>;
    74     protected:
    75       Arc(int _id) : id(_id) {}
    76       int id;
    77     public:
    78       Arc() {}
    79       Arc(Invalid) : id(-1) {}
    80       bool operator==(const Arc& arc) const { return id == arc.id; }
    81       bool operator!=(const Arc& arc) const { return id != arc.id; }
    82       bool operator<(const Arc& arc) const { return id < arc.id; }
    83     };
    84 
    85     ListArcSetBase() : first_arc(-1), first_free_arc(-1) {}
    86 
    87     Node addNode() {
    88       LEMON_ASSERT(false,
    89         "This graph structure does not support node insertion");
    90       return INVALID; // avoid warning
    91     }
    92 
    93     Arc addArc(const Node& u, const Node& v) {
    94       int n;
    95       if (first_free_arc == -1) {
    96         n = arcs.size();
    97         arcs.push_back(ArcT());
    98       } else {
    99         n = first_free_arc;
   100         first_free_arc = arcs[first_free_arc].next_in;
   101       }
   102       arcs[n].next_in = (*_nodes)[v].first_in;
   103       if ((*_nodes)[v].first_in != -1) {
   104         arcs[(*_nodes)[v].first_in].prev_in = n;
   105       }
   106       (*_nodes)[v].first_in = n;
   107       arcs[n].next_out = (*_nodes)[u].first_out;
   108       if ((*_nodes)[u].first_out != -1) {
   109         arcs[(*_nodes)[u].first_out].prev_out = n;
   110       }
   111       (*_nodes)[u].first_out = n;
   112       arcs[n].source = u;
   113       arcs[n].target = v;
   114       return Arc(n);
   115     }
   116 
   117     void erase(const Arc& arc) {
   118       int n = arc.id;
   119       if (arcs[n].prev_in != -1) {
   120         arcs[arcs[n].prev_in].next_in = arcs[n].next_in;
   121       } else {
   122         (*_nodes)[arcs[n].target].first_in = arcs[n].next_in;
   123       }
   124       if (arcs[n].next_in != -1) {
   125         arcs[arcs[n].next_in].prev_in = arcs[n].prev_in;
   126       }
   127 
   128       if (arcs[n].prev_out != -1) {
   129         arcs[arcs[n].prev_out].next_out = arcs[n].next_out;
   130       } else {
   131         (*_nodes)[arcs[n].source].first_out = arcs[n].next_out;
   132       }
   133       if (arcs[n].next_out != -1) {
   134         arcs[arcs[n].next_out].prev_out = arcs[n].prev_out;
   135       }
   136 
   137     }
   138 
   139     void clear() {
   140       Node node;
   141       for (first(node); node != INVALID; next(node)) {
   142         (*_nodes)[node].first_in = -1;
   143         (*_nodes)[node].first_out = -1;
   144       }
   145       arcs.clear();
   146       first_arc = -1;
   147       first_free_arc = -1;
   148     }
   149 
   150     void first(Node& node) const {
   151       _graph->first(node);
   152     }
   153 
   154     void next(Node& node) const {
   155       _graph->next(node);
   156     }
   157 
   158     void first(Arc& arc) const {
   159       Node node;
   160       first(node);
   161       while (node != INVALID && (*_nodes)[node].first_in == -1) {
   162         next(node);
   163       }
   164       arc.id = (node == INVALID) ? -1 : (*_nodes)[node].first_in;
   165     }
   166 
   167     void next(Arc& arc) const {
   168       if (arcs[arc.id].next_in != -1) {
   169         arc.id = arcs[arc.id].next_in;
   170       } else {
   171         Node node = arcs[arc.id].target;
   172         next(node);
   173         while (node != INVALID && (*_nodes)[node].first_in == -1) {
   174           next(node);
   175         }
   176         arc.id = (node == INVALID) ? -1 : (*_nodes)[node].first_in;
   177       }
   178     }
   179 
   180     void firstOut(Arc& arc, const Node& node) const {
   181       arc.id = (*_nodes)[node].first_out;
   182     }
   183 
   184     void nextOut(Arc& arc) const {
   185       arc.id = arcs[arc.id].next_out;
   186     }
   187 
   188     void firstIn(Arc& arc, const Node& node) const {
   189       arc.id = (*_nodes)[node].first_in;
   190     }
   191 
   192     void nextIn(Arc& arc) const {
   193       arc.id = arcs[arc.id].next_in;
   194     }
   195 
   196     int id(const Node& node) const { return _graph->id(node); }
   197     int id(const Arc& arc) const { return arc.id; }
   198 
   199     Node nodeFromId(int ix) const { return _graph->nodeFromId(ix); }
   200     Arc arcFromId(int ix) const { return Arc(ix); }
   201 
   202     int maxNodeId() const { return _graph->maxNodeId(); };
   203     int maxArcId() const { return arcs.size() - 1; }
   204 
   205     Node source(const Arc& arc) const { return arcs[arc.id].source;}
   206     Node target(const Arc& arc) const { return arcs[arc.id].target;}
   207 
   208     typedef typename ItemSetTraits<GR, Node>::ItemNotifier NodeNotifier;
   209 
   210     NodeNotifier& notifier(Node) const {
   211       return _graph->notifier(Node());
   212     }
   213 
   214     template <typename V>
   215     class NodeMap : public GR::template NodeMap<V> {
   216       typedef typename GR::template NodeMap<V> Parent;
   217 
   218     public:
   219 
   220       explicit NodeMap(const ListArcSetBase<GR>& arcset)
   221         : Parent(*arcset._graph) {}
   222 
   223       NodeMap(const ListArcSetBase<GR>& arcset, const V& value)
   224         : Parent(*arcset._graph, value) {}
   225 
   226       NodeMap& operator=(const NodeMap& cmap) {
   227         return operator=<NodeMap>(cmap);
   228       }
   229 
   230       template <typename CMap>
   231       NodeMap& operator=(const CMap& cmap) {
   232         Parent::operator=(cmap);
   233         return *this;
   234       }
   235     };
   236 
   237   };
   238 
   239   /// \ingroup graphs
   240   ///
   241   /// \brief Digraph using a node set of another digraph or graph and
   242   /// an own arc set.
   243   ///
   244   /// This structure can be used to establish another directed graph
   245   /// over a node set of an existing one. This class uses the same
   246   /// Node type as the underlying graph, and each valid node of the
   247   /// original graph is valid in this arc set, therefore the node
   248   /// objects of the original graph can be used directly with this
   249   /// class. The node handling functions (id handling, observing, and
   250   /// iterators) works equivalently as in the original graph.
   251   ///
   252   /// This implementation is based on doubly-linked lists, from each
   253   /// node the outgoing and the incoming arcs make up lists, therefore
   254   /// one arc can be erased in constant time. It also makes possible,
   255   /// that node can be removed from the underlying graph, in this case
   256   /// all arcs incident to the given node is erased from the arc set.
   257   ///
   258   /// \param GR The type of the graph which shares its node set with
   259   /// this class. Its interface must conform to the
   260   /// \ref concepts::Digraph "Digraph" or \ref concepts::Graph "Graph"
   261   /// concept.
   262   ///
   263   /// This class fully conforms to the \ref concepts::Digraph
   264   /// "Digraph" concept.
   265   template <typename GR>
   266   class ListArcSet : public ArcSetExtender<ListArcSetBase<GR> > {
   267     typedef ArcSetExtender<ListArcSetBase<GR> > Parent;
   268 
   269   public:
   270 
   271     typedef typename Parent::Node Node;
   272     typedef typename Parent::Arc Arc;
   273 
   274     typedef typename Parent::NodesImplBase NodesImplBase;
   275 
   276     void eraseNode(const Node& node) {
   277       Arc arc;
   278       Parent::firstOut(arc, node);
   279       while (arc != INVALID ) {
   280         erase(arc);
   281         Parent::firstOut(arc, node);
   282       }
   283 
   284       Parent::firstIn(arc, node);
   285       while (arc != INVALID ) {
   286         erase(arc);
   287         Parent::firstIn(arc, node);
   288       }
   289     }
   290 
   291     void clearNodes() {
   292       Parent::clear();
   293     }
   294 
   295     class NodesImpl : public NodesImplBase {
   296       typedef NodesImplBase Parent;
   297 
   298     public:
   299       NodesImpl(const GR& graph, ListArcSet& arcset)
   300         : Parent(graph), _arcset(arcset) {}
   301 
   302       virtual ~NodesImpl() {}
   303 
   304     protected:
   305 
   306       virtual void erase(const Node& node) {
   307         _arcset.eraseNode(node);
   308         Parent::erase(node);
   309       }
   310       virtual void erase(const std::vector<Node>& nodes) {
   311         for (int i = 0; i < int(nodes.size()); ++i) {
   312           _arcset.eraseNode(nodes[i]);
   313         }
   314         Parent::erase(nodes);
   315       }
   316       virtual void clear() {
   317         _arcset.clearNodes();
   318         Parent::clear();
   319       }
   320 
   321     private:
   322       ListArcSet& _arcset;
   323     };
   324 
   325     NodesImpl _nodes;
   326 
   327   public:
   328 
   329     /// \brief Constructor of the ArcSet.
   330     ///
   331     /// Constructor of the ArcSet.
   332     ListArcSet(const GR& graph) : _nodes(graph, *this) {
   333       Parent::initalize(graph, _nodes);
   334     }
   335 
   336     /// \brief Add a new arc to the digraph.
   337     ///
   338     /// Add a new arc to the digraph with source node \c s
   339     /// and target node \c t.
   340     /// \return The new arc.
   341     Arc addArc(const Node& s, const Node& t) {
   342       return Parent::addArc(s, t);
   343     }
   344 
   345     /// \brief Erase an arc from the digraph.
   346     ///
   347     /// Erase an arc \c a from the digraph.
   348     void erase(const Arc& a) {
   349       return Parent::erase(a);
   350     }
   351 
   352   };
   353 
   354   template <typename GR>
   355   class ListEdgeSetBase {
   356   public:
   357 
   358     typedef typename GR::Node Node;
   359     typedef typename GR::NodeIt NodeIt;
   360 
   361   protected:
   362 
   363     struct NodeT {
   364       int first_out;
   365       NodeT() : first_out(-1) {}
   366     };
   367 
   368     typedef typename ItemSetTraits<GR, Node>::
   369     template Map<NodeT>::Type NodesImplBase;
   370 
   371     NodesImplBase* _nodes;
   372 
   373     struct ArcT {
   374       Node target;
   375       int prev_out, next_out;
   376       ArcT() : prev_out(-1), next_out(-1) {}
   377     };
   378 
   379     std::vector<ArcT> arcs;
   380 
   381     int first_arc;
   382     int first_free_arc;
   383 
   384     const GR* _graph;
   385 
   386     void initalize(const GR& graph, NodesImplBase& nodes) {
   387       _graph = &graph;
   388       _nodes = &nodes;
   389     }
   390 
   391   public:
   392 
   393     class Edge {
   394       friend class ListEdgeSetBase;
   395     protected:
   396 
   397       int id;
   398       explicit Edge(int _id) { id = _id;}
   399 
   400     public:
   401       Edge() {}
   402       Edge (Invalid) { id = -1; }
   403       bool operator==(const Edge& arc) const {return id == arc.id;}
   404       bool operator!=(const Edge& arc) const {return id != arc.id;}
   405       bool operator<(const Edge& arc) const {return id < arc.id;}
   406     };
   407 
   408     class Arc {
   409       friend class ListEdgeSetBase;
   410     protected:
   411       Arc(int _id) : id(_id) {}
   412       int id;
   413     public:
   414       operator Edge() const { return edgeFromId(id / 2); }
   415 
   416       Arc() {}
   417       Arc(Invalid) : id(-1) {}
   418       bool operator==(const Arc& arc) const { return id == arc.id; }
   419       bool operator!=(const Arc& arc) const { return id != arc.id; }
   420       bool operator<(const Arc& arc) const { return id < arc.id; }
   421     };
   422 
   423     ListEdgeSetBase() : first_arc(-1), first_free_arc(-1) {}
   424 
   425     Node addNode() {
   426       LEMON_ASSERT(false,
   427         "This graph structure does not support node insertion");
   428       return INVALID; // avoid warning
   429     }
   430 
   431     Edge addEdge(const Node& u, const Node& v) {
   432       int n;
   433 
   434       if (first_free_arc == -1) {
   435         n = arcs.size();
   436         arcs.push_back(ArcT());
   437         arcs.push_back(ArcT());
   438       } else {
   439         n = first_free_arc;
   440         first_free_arc = arcs[n].next_out;
   441       }
   442 
   443       arcs[n].target = u;
   444       arcs[n | 1].target = v;
   445 
   446       arcs[n].next_out = (*_nodes)[v].first_out;
   447       if ((*_nodes)[v].first_out != -1) {
   448         arcs[(*_nodes)[v].first_out].prev_out = n;
   449       }
   450       (*_nodes)[v].first_out = n;
   451       arcs[n].prev_out = -1;
   452 
   453       if ((*_nodes)[u].first_out != -1) {
   454         arcs[(*_nodes)[u].first_out].prev_out = (n | 1);
   455       }
   456       arcs[n | 1].next_out = (*_nodes)[u].first_out;
   457       (*_nodes)[u].first_out = (n | 1);
   458       arcs[n | 1].prev_out = -1;
   459 
   460       return Edge(n / 2);
   461     }
   462 
   463     void erase(const Edge& arc) {
   464       int n = arc.id * 2;
   465 
   466       if (arcs[n].next_out != -1) {
   467         arcs[arcs[n].next_out].prev_out = arcs[n].prev_out;
   468       }
   469 
   470       if (arcs[n].prev_out != -1) {
   471         arcs[arcs[n].prev_out].next_out = arcs[n].next_out;
   472       } else {
   473         (*_nodes)[arcs[n | 1].target].first_out = arcs[n].next_out;
   474       }
   475 
   476       if (arcs[n | 1].next_out != -1) {
   477         arcs[arcs[n | 1].next_out].prev_out = arcs[n | 1].prev_out;
   478       }
   479 
   480       if (arcs[n | 1].prev_out != -1) {
   481         arcs[arcs[n | 1].prev_out].next_out = arcs[n | 1].next_out;
   482       } else {
   483         (*_nodes)[arcs[n].target].first_out = arcs[n | 1].next_out;
   484       }
   485 
   486       arcs[n].next_out = first_free_arc;
   487       first_free_arc = n;
   488 
   489     }
   490 
   491     void clear() {
   492       Node node;
   493       for (first(node); node != INVALID; next(node)) {
   494         (*_nodes)[node].first_out = -1;
   495       }
   496       arcs.clear();
   497       first_arc = -1;
   498       first_free_arc = -1;
   499     }
   500 
   501     void first(Node& node) const {
   502       _graph->first(node);
   503     }
   504 
   505     void next(Node& node) const {
   506       _graph->next(node);
   507     }
   508 
   509     void first(Arc& arc) const {
   510       Node node;
   511       first(node);
   512       while (node != INVALID && (*_nodes)[node].first_out == -1) {
   513         next(node);
   514       }
   515       arc.id = (node == INVALID) ? -1 : (*_nodes)[node].first_out;
   516     }
   517 
   518     void next(Arc& arc) const {
   519       if (arcs[arc.id].next_out != -1) {
   520         arc.id = arcs[arc.id].next_out;
   521       } else {
   522         Node node = arcs[arc.id ^ 1].target;
   523         next(node);
   524         while(node != INVALID && (*_nodes)[node].first_out == -1) {
   525           next(node);
   526         }
   527         arc.id = (node == INVALID) ? -1 : (*_nodes)[node].first_out;
   528       }
   529     }
   530 
   531     void first(Edge& edge) const {
   532       Node node;
   533       first(node);
   534       while (node != INVALID) {
   535         edge.id = (*_nodes)[node].first_out;
   536         while ((edge.id & 1) != 1) {
   537           edge.id = arcs[edge.id].next_out;
   538         }
   539         if (edge.id != -1) {
   540           edge.id /= 2;
   541           return;
   542         }
   543         next(node);
   544       }
   545       edge.id = -1;
   546     }
   547 
   548     void next(Edge& edge) const {
   549       Node node = arcs[edge.id * 2].target;
   550       edge.id = arcs[(edge.id * 2) | 1].next_out;
   551       while ((edge.id & 1) != 1) {
   552         edge.id = arcs[edge.id].next_out;
   553       }
   554       if (edge.id != -1) {
   555         edge.id /= 2;
   556         return;
   557       }
   558       next(node);
   559       while (node != INVALID) {
   560         edge.id = (*_nodes)[node].first_out;
   561         while ((edge.id & 1) != 1) {
   562           edge.id = arcs[edge.id].next_out;
   563         }
   564         if (edge.id != -1) {
   565           edge.id /= 2;
   566           return;
   567         }
   568         next(node);
   569       }
   570       edge.id = -1;
   571     }
   572 
   573     void firstOut(Arc& arc, const Node& node) const {
   574       arc.id = (*_nodes)[node].first_out;
   575     }
   576 
   577     void nextOut(Arc& arc) const {
   578       arc.id = arcs[arc.id].next_out;
   579     }
   580 
   581     void firstIn(Arc& arc, const Node& node) const {
   582       arc.id = (((*_nodes)[node].first_out) ^ 1);
   583       if (arc.id == -2) arc.id = -1;
   584     }
   585 
   586     void nextIn(Arc& arc) const {
   587       arc.id = ((arcs[arc.id ^ 1].next_out) ^ 1);
   588       if (arc.id == -2) arc.id = -1;
   589     }
   590 
   591     void firstInc(Edge &arc, bool& dir, const Node& node) const {
   592       int de = (*_nodes)[node].first_out;
   593       if (de != -1 ) {
   594         arc.id = de / 2;
   595         dir = ((de & 1) == 1);
   596       } else {
   597         arc.id = -1;
   598         dir = true;
   599       }
   600     }
   601     void nextInc(Edge &arc, bool& dir) const {
   602       int de = (arcs[(arc.id * 2) | (dir ? 1 : 0)].next_out);
   603       if (de != -1 ) {
   604         arc.id = de / 2;
   605         dir = ((de & 1) == 1);
   606       } else {
   607         arc.id = -1;
   608         dir = true;
   609       }
   610     }
   611 
   612     static bool direction(Arc arc) {
   613       return (arc.id & 1) == 1;
   614     }
   615 
   616     static Arc direct(Edge edge, bool dir) {
   617       return Arc(edge.id * 2 + (dir ? 1 : 0));
   618     }
   619 
   620     int id(const Node& node) const { return _graph->id(node); }
   621     static int id(Arc e) { return e.id; }
   622     static int id(Edge e) { return e.id; }
   623 
   624     Node nodeFromId(int id) const { return _graph->nodeFromId(id); }
   625     static Arc arcFromId(int id) { return Arc(id);}
   626     static Edge edgeFromId(int id) { return Edge(id);}
   627 
   628     int maxNodeId() const { return _graph->maxNodeId(); };
   629     int maxEdgeId() const { return arcs.size() / 2 - 1; }
   630     int maxArcId() const { return arcs.size()-1; }
   631 
   632     Node source(Arc e) const { return arcs[e.id ^ 1].target; }
   633     Node target(Arc e) const { return arcs[e.id].target; }
   634 
   635     Node u(Edge e) const { return arcs[2 * e.id].target; }
   636     Node v(Edge e) const { return arcs[2 * e.id + 1].target; }
   637 
   638     typedef typename ItemSetTraits<GR, Node>::ItemNotifier NodeNotifier;
   639 
   640     NodeNotifier& notifier(Node) const {
   641       return _graph->notifier(Node());
   642     }
   643 
   644     template <typename V>
   645     class NodeMap : public GR::template NodeMap<V> {
   646       typedef typename GR::template NodeMap<V> Parent;
   647 
   648     public:
   649 
   650       explicit NodeMap(const ListEdgeSetBase<GR>& arcset)
   651         : Parent(*arcset._graph) {}
   652 
   653       NodeMap(const ListEdgeSetBase<GR>& arcset, const V& value)
   654         : Parent(*arcset._graph, value) {}
   655 
   656       NodeMap& operator=(const NodeMap& cmap) {
   657         return operator=<NodeMap>(cmap);
   658       }
   659 
   660       template <typename CMap>
   661       NodeMap& operator=(const CMap& cmap) {
   662         Parent::operator=(cmap);
   663         return *this;
   664       }
   665     };
   666 
   667   };
   668 
   669   /// \ingroup graphs
   670   ///
   671   /// \brief Graph using a node set of another digraph or graph and an
   672   /// own edge set.
   673   ///
   674   /// This structure can be used to establish another graph over a
   675   /// node set of an existing one. This class uses the same Node type
   676   /// as the underlying graph, and each valid node of the original
   677   /// graph is valid in this arc set, therefore the node objects of
   678   /// the original graph can be used directly with this class. The
   679   /// node handling functions (id handling, observing, and iterators)
   680   /// works equivalently as in the original graph.
   681   ///
   682   /// This implementation is based on doubly-linked lists, from each
   683   /// node the incident edges make up lists, therefore one edge can be
   684   /// erased in constant time. It also makes possible, that node can
   685   /// be removed from the underlying graph, in this case all edges
   686   /// incident to the given node is erased from the arc set.
   687   ///
   688   /// \param GR The type of the graph which shares its node set
   689   /// with this class. Its interface must conform to the
   690   /// \ref concepts::Digraph "Digraph" or \ref concepts::Graph "Graph"
   691   /// concept.
   692   ///
   693   /// This class fully conforms to the \ref concepts::Graph "Graph"
   694   /// concept.
   695   template <typename GR>
   696   class ListEdgeSet : public EdgeSetExtender<ListEdgeSetBase<GR> > {
   697     typedef EdgeSetExtender<ListEdgeSetBase<GR> > Parent;
   698 
   699   public:
   700 
   701     typedef typename Parent::Node Node;
   702     typedef typename Parent::Arc Arc;
   703     typedef typename Parent::Edge Edge;
   704 
   705     typedef typename Parent::NodesImplBase NodesImplBase;
   706 
   707     void eraseNode(const Node& node) {
   708       Arc arc;
   709       Parent::firstOut(arc, node);
   710       while (arc != INVALID ) {
   711         erase(arc);
   712         Parent::firstOut(arc, node);
   713       }
   714 
   715     }
   716 
   717     void clearNodes() {
   718       Parent::clear();
   719     }
   720 
   721     class NodesImpl : public NodesImplBase {
   722       typedef NodesImplBase Parent;
   723 
   724     public:
   725       NodesImpl(const GR& graph, ListEdgeSet& arcset)
   726         : Parent(graph), _arcset(arcset) {}
   727 
   728       virtual ~NodesImpl() {}
   729 
   730     protected:
   731 
   732       virtual void erase(const Node& node) {
   733         _arcset.eraseNode(node);
   734         Parent::erase(node);
   735       }
   736       virtual void erase(const std::vector<Node>& nodes) {
   737         for (int i = 0; i < int(nodes.size()); ++i) {
   738           _arcset.eraseNode(nodes[i]);
   739         }
   740         Parent::erase(nodes);
   741       }
   742       virtual void clear() {
   743         _arcset.clearNodes();
   744         Parent::clear();
   745       }
   746 
   747     private:
   748       ListEdgeSet& _arcset;
   749     };
   750 
   751     NodesImpl _nodes;
   752 
   753   public:
   754 
   755     /// \brief Constructor of the EdgeSet.
   756     ///
   757     /// Constructor of the EdgeSet.
   758     ListEdgeSet(const GR& graph) : _nodes(graph, *this) {
   759       Parent::initalize(graph, _nodes);
   760     }
   761 
   762     /// \brief Add a new edge to the graph.
   763     ///
   764     /// Add a new edge to the graph with node \c u
   765     /// and node \c v endpoints.
   766     /// \return The new edge.
   767     Edge addEdge(const Node& u, const Node& v) {
   768       return Parent::addEdge(u, v);
   769     }
   770 
   771     /// \brief Erase an edge from the graph.
   772     ///
   773     /// Erase the edge \c e from the graph.
   774     void erase(const Edge& e) {
   775       return Parent::erase(e);
   776     }
   777 
   778   };
   779 
   780   template <typename GR>
   781   class SmartArcSetBase {
   782   public:
   783 
   784     typedef typename GR::Node Node;
   785     typedef typename GR::NodeIt NodeIt;
   786 
   787   protected:
   788 
   789     struct NodeT {
   790       int first_out, first_in;
   791       NodeT() : first_out(-1), first_in(-1) {}
   792     };
   793 
   794     typedef typename ItemSetTraits<GR, Node>::
   795     template Map<NodeT>::Type NodesImplBase;
   796 
   797     NodesImplBase* _nodes;
   798 
   799     struct ArcT {
   800       Node source, target;
   801       int next_out, next_in;
   802       ArcT() {}
   803     };
   804 
   805     std::vector<ArcT> arcs;
   806 
   807     const GR* _graph;
   808 
   809     void initalize(const GR& graph, NodesImplBase& nodes) {
   810       _graph = &graph;
   811       _nodes = &nodes;
   812     }
   813 
   814   public:
   815 
   816     class Arc {
   817       friend class SmartArcSetBase<GR>;
   818     protected:
   819       Arc(int _id) : id(_id) {}
   820       int id;
   821     public:
   822       Arc() {}
   823       Arc(Invalid) : id(-1) {}
   824       bool operator==(const Arc& arc) const { return id == arc.id; }
   825       bool operator!=(const Arc& arc) const { return id != arc.id; }
   826       bool operator<(const Arc& arc) const { return id < arc.id; }
   827     };
   828 
   829     SmartArcSetBase() {}
   830 
   831     Node addNode() {
   832       LEMON_ASSERT(false,
   833         "This graph structure does not support node insertion");
   834       return INVALID; // avoid warning
   835     }
   836 
   837     Arc addArc(const Node& u, const Node& v) {
   838       int n = arcs.size();
   839       arcs.push_back(ArcT());
   840       arcs[n].next_in = (*_nodes)[v].first_in;
   841       (*_nodes)[v].first_in = n;
   842       arcs[n].next_out = (*_nodes)[u].first_out;
   843       (*_nodes)[u].first_out = n;
   844       arcs[n].source = u;
   845       arcs[n].target = v;
   846       return Arc(n);
   847     }
   848 
   849     void clear() {
   850       Node node;
   851       for (first(node); node != INVALID; next(node)) {
   852         (*_nodes)[node].first_in = -1;
   853         (*_nodes)[node].first_out = -1;
   854       }
   855       arcs.clear();
   856     }
   857 
   858     void first(Node& node) const {
   859       _graph->first(node);
   860     }
   861 
   862     void next(Node& node) const {
   863       _graph->next(node);
   864     }
   865 
   866     void first(Arc& arc) const {
   867       arc.id = arcs.size() - 1;
   868     }
   869 
   870     static void next(Arc& arc) {
   871       --arc.id;
   872     }
   873 
   874     void firstOut(Arc& arc, const Node& node) const {
   875       arc.id = (*_nodes)[node].first_out;
   876     }
   877 
   878     void nextOut(Arc& arc) const {
   879       arc.id = arcs[arc.id].next_out;
   880     }
   881 
   882     void firstIn(Arc& arc, const Node& node) const {
   883       arc.id = (*_nodes)[node].first_in;
   884     }
   885 
   886     void nextIn(Arc& arc) const {
   887       arc.id = arcs[arc.id].next_in;
   888     }
   889 
   890     int id(const Node& node) const { return _graph->id(node); }
   891     int id(const Arc& arc) const { return arc.id; }
   892 
   893     Node nodeFromId(int ix) const { return _graph->nodeFromId(ix); }
   894     Arc arcFromId(int ix) const { return Arc(ix); }
   895 
   896     int maxNodeId() const { return _graph->maxNodeId(); };
   897     int maxArcId() const { return arcs.size() - 1; }
   898 
   899     Node source(const Arc& arc) const { return arcs[arc.id].source;}
   900     Node target(const Arc& arc) const { return arcs[arc.id].target;}
   901 
   902     typedef typename ItemSetTraits<GR, Node>::ItemNotifier NodeNotifier;
   903 
   904     NodeNotifier& notifier(Node) const {
   905       return _graph->notifier(Node());
   906     }
   907 
   908     template <typename V>
   909     class NodeMap : public GR::template NodeMap<V> {
   910       typedef typename GR::template NodeMap<V> Parent;
   911 
   912     public:
   913 
   914       explicit NodeMap(const SmartArcSetBase<GR>& arcset)
   915         : Parent(*arcset._graph) { }
   916 
   917       NodeMap(const SmartArcSetBase<GR>& arcset, const V& value)
   918         : Parent(*arcset._graph, value) { }
   919 
   920       NodeMap& operator=(const NodeMap& cmap) {
   921         return operator=<NodeMap>(cmap);
   922       }
   923 
   924       template <typename CMap>
   925       NodeMap& operator=(const CMap& cmap) {
   926         Parent::operator=(cmap);
   927         return *this;
   928       }
   929     };
   930 
   931   };
   932 
   933 
   934   /// \ingroup graphs
   935   ///
   936   /// \brief Digraph using a node set of another digraph or graph and
   937   /// an own arc set.
   938   ///
   939   /// This structure can be used to establish another directed graph
   940   /// over a node set of an existing one. This class uses the same
   941   /// Node type as the underlying graph, and each valid node of the
   942   /// original graph is valid in this arc set, therefore the node
   943   /// objects of the original graph can be used directly with this
   944   /// class. The node handling functions (id handling, observing, and
   945   /// iterators) works equivalently as in the original graph.
   946   ///
   947   /// \param GR The type of the graph which shares its node set with
   948   /// this class. Its interface must conform to the
   949   /// \ref concepts::Digraph "Digraph" or \ref concepts::Graph "Graph"
   950   /// concept.
   951   ///
   952   /// This implementation is slightly faster than the \c ListArcSet,
   953   /// because it uses continuous storage for arcs and it uses just
   954   /// single-linked lists for enumerate outgoing and incoming
   955   /// arcs. Therefore the arcs cannot be erased from the arc sets.
   956   ///
   957   /// \warning If a node is erased from the underlying graph and this
   958   /// node is the source or target of one arc in the arc set, then
   959   /// the arc set is invalidated, and it cannot be used anymore. The
   960   /// validity can be checked with the \c valid() member function.
   961   ///
   962   /// This class fully conforms to the \ref concepts::Digraph
   963   /// "Digraph" concept.
   964   template <typename GR>
   965   class SmartArcSet : public ArcSetExtender<SmartArcSetBase<GR> > {
   966     typedef ArcSetExtender<SmartArcSetBase<GR> > Parent;
   967 
   968   public:
   969 
   970     typedef typename Parent::Node Node;
   971     typedef typename Parent::Arc Arc;
   972 
   973   protected:
   974 
   975     typedef typename Parent::NodesImplBase NodesImplBase;
   976 
   977     void eraseNode(const Node& node) {
   978       if (typename Parent::InArcIt(*this, node) == INVALID &&
   979           typename Parent::OutArcIt(*this, node) == INVALID) {
   980         return;
   981       }
   982       throw typename NodesImplBase::Notifier::ImmediateDetach();
   983     }
   984 
   985     void clearNodes() {
   986       Parent::clear();
   987     }
   988 
   989     class NodesImpl : public NodesImplBase {
   990       typedef NodesImplBase Parent;
   991 
   992     public:
   993       NodesImpl(const GR& graph, SmartArcSet& arcset)
   994         : Parent(graph), _arcset(arcset) {}
   995 
   996       virtual ~NodesImpl() {}
   997 
   998       bool attached() const {
   999         return Parent::attached();
  1000       }
  1001 
  1002     protected:
  1003 
  1004       virtual void erase(const Node& node) {
  1005         try {
  1006           _arcset.eraseNode(node);
  1007           Parent::erase(node);
  1008         } catch (const typename NodesImplBase::Notifier::ImmediateDetach&) {
  1009           Parent::clear();
  1010           throw;
  1011         }
  1012       }
  1013       virtual void erase(const std::vector<Node>& nodes) {
  1014         try {
  1015           for (int i = 0; i < int(nodes.size()); ++i) {
  1016             _arcset.eraseNode(nodes[i]);
  1017           }
  1018           Parent::erase(nodes);
  1019         } catch (const typename NodesImplBase::Notifier::ImmediateDetach&) {
  1020           Parent::clear();
  1021           throw;
  1022         }
  1023       }
  1024       virtual void clear() {
  1025         _arcset.clearNodes();
  1026         Parent::clear();
  1027       }
  1028 
  1029     private:
  1030       SmartArcSet& _arcset;
  1031     };
  1032 
  1033     NodesImpl _nodes;
  1034 
  1035   public:
  1036 
  1037     /// \brief Constructor of the ArcSet.
  1038     ///
  1039     /// Constructor of the ArcSet.
  1040     SmartArcSet(const GR& graph) : _nodes(graph, *this) {
  1041       Parent::initalize(graph, _nodes);
  1042     }
  1043 
  1044     /// \brief Add a new arc to the digraph.
  1045     ///
  1046     /// Add a new arc to the digraph with source node \c s
  1047     /// and target node \c t.
  1048     /// \return The new arc.
  1049     Arc addArc(const Node& s, const Node& t) {
  1050       return Parent::addArc(s, t);
  1051     }
  1052 
  1053     /// \brief Validity check
  1054     ///
  1055     /// This functions gives back false if the ArcSet is
  1056     /// invalidated. It occurs when a node in the underlying graph is
  1057     /// erased and it is not isolated in the ArcSet.
  1058     bool valid() const {
  1059       return _nodes.attached();
  1060     }
  1061 
  1062   };
  1063 
  1064 
  1065   template <typename GR>
  1066   class SmartEdgeSetBase {
  1067   public:
  1068 
  1069     typedef typename GR::Node Node;
  1070     typedef typename GR::NodeIt NodeIt;
  1071 
  1072   protected:
  1073 
  1074     struct NodeT {
  1075       int first_out;
  1076       NodeT() : first_out(-1) {}
  1077     };
  1078 
  1079     typedef typename ItemSetTraits<GR, Node>::
  1080     template Map<NodeT>::Type NodesImplBase;
  1081 
  1082     NodesImplBase* _nodes;
  1083 
  1084     struct ArcT {
  1085       Node target;
  1086       int next_out;
  1087       ArcT() {}
  1088     };
  1089 
  1090     std::vector<ArcT> arcs;
  1091 
  1092     const GR* _graph;
  1093 
  1094     void initalize(const GR& graph, NodesImplBase& nodes) {
  1095       _graph = &graph;
  1096       _nodes = &nodes;
  1097     }
  1098 
  1099   public:
  1100 
  1101     class Edge {
  1102       friend class SmartEdgeSetBase;
  1103     protected:
  1104 
  1105       int id;
  1106       explicit Edge(int _id) { id = _id;}
  1107 
  1108     public:
  1109       Edge() {}
  1110       Edge (Invalid) { id = -1; }
  1111       bool operator==(const Edge& arc) const {return id == arc.id;}
  1112       bool operator!=(const Edge& arc) const {return id != arc.id;}
  1113       bool operator<(const Edge& arc) const {return id < arc.id;}
  1114     };
  1115 
  1116     class Arc {
  1117       friend class SmartEdgeSetBase;
  1118     protected:
  1119       Arc(int _id) : id(_id) {}
  1120       int id;
  1121     public:
  1122       operator Edge() const { return edgeFromId(id / 2); }
  1123 
  1124       Arc() {}
  1125       Arc(Invalid) : id(-1) {}
  1126       bool operator==(const Arc& arc) const { return id == arc.id; }
  1127       bool operator!=(const Arc& arc) const { return id != arc.id; }
  1128       bool operator<(const Arc& arc) const { return id < arc.id; }
  1129     };
  1130 
  1131     SmartEdgeSetBase() {}
  1132 
  1133     Node addNode() {
  1134       LEMON_ASSERT(false,
  1135         "This graph structure does not support node insertion");
  1136       return INVALID; // avoid warning
  1137     }
  1138 
  1139     Edge addEdge(const Node& u, const Node& v) {
  1140       int n = arcs.size();
  1141       arcs.push_back(ArcT());
  1142       arcs.push_back(ArcT());
  1143 
  1144       arcs[n].target = u;
  1145       arcs[n | 1].target = v;
  1146 
  1147       arcs[n].next_out = (*_nodes)[v].first_out;
  1148       (*_nodes)[v].first_out = n;
  1149 
  1150       arcs[n | 1].next_out = (*_nodes)[u].first_out;
  1151       (*_nodes)[u].first_out = (n | 1);
  1152 
  1153       return Edge(n / 2);
  1154     }
  1155 
  1156     void clear() {
  1157       Node node;
  1158       for (first(node); node != INVALID; next(node)) {
  1159         (*_nodes)[node].first_out = -1;
  1160       }
  1161       arcs.clear();
  1162     }
  1163 
  1164     void first(Node& node) const {
  1165       _graph->first(node);
  1166     }
  1167 
  1168     void next(Node& node) const {
  1169       _graph->next(node);
  1170     }
  1171 
  1172     void first(Arc& arc) const {
  1173       arc.id = arcs.size() - 1;
  1174     }
  1175 
  1176     static void next(Arc& arc) {
  1177       --arc.id;
  1178     }
  1179 
  1180     void first(Edge& arc) const {
  1181       arc.id = arcs.size() / 2 - 1;
  1182     }
  1183 
  1184     static void next(Edge& arc) {
  1185       --arc.id;
  1186     }
  1187 
  1188     void firstOut(Arc& arc, const Node& node) const {
  1189       arc.id = (*_nodes)[node].first_out;
  1190     }
  1191 
  1192     void nextOut(Arc& arc) const {
  1193       arc.id = arcs[arc.id].next_out;
  1194     }
  1195 
  1196     void firstIn(Arc& arc, const Node& node) const {
  1197       arc.id = (((*_nodes)[node].first_out) ^ 1);
  1198       if (arc.id == -2) arc.id = -1;
  1199     }
  1200 
  1201     void nextIn(Arc& arc) const {
  1202       arc.id = ((arcs[arc.id ^ 1].next_out) ^ 1);
  1203       if (arc.id == -2) arc.id = -1;
  1204     }
  1205 
  1206     void firstInc(Edge &arc, bool& dir, const Node& node) const {
  1207       int de = (*_nodes)[node].first_out;
  1208       if (de != -1 ) {
  1209         arc.id = de / 2;
  1210         dir = ((de & 1) == 1);
  1211       } else {
  1212         arc.id = -1;
  1213         dir = true;
  1214       }
  1215     }
  1216     void nextInc(Edge &arc, bool& dir) const {
  1217       int de = (arcs[(arc.id * 2) | (dir ? 1 : 0)].next_out);
  1218       if (de != -1 ) {
  1219         arc.id = de / 2;
  1220         dir = ((de & 1) == 1);
  1221       } else {
  1222         arc.id = -1;
  1223         dir = true;
  1224       }
  1225     }
  1226 
  1227     static bool direction(Arc arc) {
  1228       return (arc.id & 1) == 1;
  1229     }
  1230 
  1231     static Arc direct(Edge edge, bool dir) {
  1232       return Arc(edge.id * 2 + (dir ? 1 : 0));
  1233     }
  1234 
  1235     int id(Node node) const { return _graph->id(node); }
  1236     static int id(Arc arc) { return arc.id; }
  1237     static int id(Edge arc) { return arc.id; }
  1238 
  1239     Node nodeFromId(int id) const { return _graph->nodeFromId(id); }
  1240     static Arc arcFromId(int id) { return Arc(id); }
  1241     static Edge edgeFromId(int id) { return Edge(id);}
  1242 
  1243     int maxNodeId() const { return _graph->maxNodeId(); };
  1244     int maxArcId() const { return arcs.size() - 1; }
  1245     int maxEdgeId() const { return arcs.size() / 2 - 1; }
  1246 
  1247     Node source(Arc e) const { return arcs[e.id ^ 1].target; }
  1248     Node target(Arc e) const { return arcs[e.id].target; }
  1249 
  1250     Node u(Edge e) const { return arcs[2 * e.id].target; }
  1251     Node v(Edge e) const { return arcs[2 * e.id + 1].target; }
  1252 
  1253     typedef typename ItemSetTraits<GR, Node>::ItemNotifier NodeNotifier;
  1254 
  1255     NodeNotifier& notifier(Node) const {
  1256       return _graph->notifier(Node());
  1257     }
  1258 
  1259     template <typename V>
  1260     class NodeMap : public GR::template NodeMap<V> {
  1261       typedef typename GR::template NodeMap<V> Parent;
  1262 
  1263     public:
  1264 
  1265       explicit NodeMap(const SmartEdgeSetBase<GR>& arcset)
  1266         : Parent(*arcset._graph) { }
  1267 
  1268       NodeMap(const SmartEdgeSetBase<GR>& arcset, const V& value)
  1269         : Parent(*arcset._graph, value) { }
  1270 
  1271       NodeMap& operator=(const NodeMap& cmap) {
  1272         return operator=<NodeMap>(cmap);
  1273       }
  1274 
  1275       template <typename CMap>
  1276       NodeMap& operator=(const CMap& cmap) {
  1277         Parent::operator=(cmap);
  1278         return *this;
  1279       }
  1280     };
  1281 
  1282   };
  1283 
  1284   /// \ingroup graphs
  1285   ///
  1286   /// \brief Graph using a node set of another digraph or graph and an
  1287   /// own edge set.
  1288   ///
  1289   /// This structure can be used to establish another graph over a
  1290   /// node set of an existing one. This class uses the same Node type
  1291   /// as the underlying graph, and each valid node of the original
  1292   /// graph is valid in this arc set, therefore the node objects of
  1293   /// the original graph can be used directly with this class. The
  1294   /// node handling functions (id handling, observing, and iterators)
  1295   /// works equivalently as in the original graph.
  1296   ///
  1297   /// \param GR The type of the graph which shares its node set
  1298   /// with this class. Its interface must conform to the
  1299   /// \ref concepts::Digraph "Digraph" or \ref concepts::Graph "Graph"
  1300   ///  concept.
  1301   ///
  1302   /// This implementation is slightly faster than the \c ListEdgeSet,
  1303   /// because it uses continuous storage for edges and it uses just
  1304   /// single-linked lists for enumerate incident edges. Therefore the
  1305   /// edges cannot be erased from the edge sets.
  1306   ///
  1307   /// \warning If a node is erased from the underlying graph and this
  1308   /// node is incident to one edge in the edge set, then the edge set
  1309   /// is invalidated, and it cannot be used anymore. The validity can
  1310   /// be checked with the \c valid() member function.
  1311   ///
  1312   /// This class fully conforms to the \ref concepts::Graph
  1313   /// "Graph" concept.
  1314   template <typename GR>
  1315   class SmartEdgeSet : public EdgeSetExtender<SmartEdgeSetBase<GR> > {
  1316     typedef EdgeSetExtender<SmartEdgeSetBase<GR> > Parent;
  1317 
  1318   public:
  1319 
  1320     typedef typename Parent::Node Node;
  1321     typedef typename Parent::Arc Arc;
  1322     typedef typename Parent::Edge Edge;
  1323 
  1324   protected:
  1325 
  1326     typedef typename Parent::NodesImplBase NodesImplBase;
  1327 
  1328     void eraseNode(const Node& node) {
  1329       if (typename Parent::IncEdgeIt(*this, node) == INVALID) {
  1330         return;
  1331       }
  1332       throw typename NodesImplBase::Notifier::ImmediateDetach();
  1333     }
  1334 
  1335     void clearNodes() {
  1336       Parent::clear();
  1337     }
  1338 
  1339     class NodesImpl : public NodesImplBase {
  1340       typedef NodesImplBase Parent;
  1341 
  1342     public:
  1343       NodesImpl(const GR& graph, SmartEdgeSet& arcset)
  1344         : Parent(graph), _arcset(arcset) {}
  1345 
  1346       virtual ~NodesImpl() {}
  1347 
  1348       bool attached() const {
  1349         return Parent::attached();
  1350       }
  1351 
  1352     protected:
  1353 
  1354       virtual void erase(const Node& node) {
  1355         try {
  1356           _arcset.eraseNode(node);
  1357           Parent::erase(node);
  1358         } catch (const typename NodesImplBase::Notifier::ImmediateDetach&) {
  1359           Parent::clear();
  1360           throw;
  1361         }
  1362       }
  1363       virtual void erase(const std::vector<Node>& nodes) {
  1364         try {
  1365           for (int i = 0; i < int(nodes.size()); ++i) {
  1366             _arcset.eraseNode(nodes[i]);
  1367           }
  1368           Parent::erase(nodes);
  1369         } catch (const typename NodesImplBase::Notifier::ImmediateDetach&) {
  1370           Parent::clear();
  1371           throw;
  1372         }
  1373       }
  1374       virtual void clear() {
  1375         _arcset.clearNodes();
  1376         Parent::clear();
  1377       }
  1378 
  1379     private:
  1380       SmartEdgeSet& _arcset;
  1381     };
  1382 
  1383     NodesImpl _nodes;
  1384 
  1385   public:
  1386 
  1387     /// \brief Constructor of the EdgeSet.
  1388     ///
  1389     /// Constructor of the EdgeSet.
  1390     SmartEdgeSet(const GR& graph) : _nodes(graph, *this) {
  1391       Parent::initalize(graph, _nodes);
  1392     }
  1393 
  1394     /// \brief Add a new edge to the graph.
  1395     ///
  1396     /// Add a new edge to the graph with node \c u
  1397     /// and node \c v endpoints.
  1398     /// \return The new edge.
  1399     Edge addEdge(const Node& u, const Node& v) {
  1400       return Parent::addEdge(u, v);
  1401     }
  1402 
  1403     /// \brief Validity check
  1404     ///
  1405     /// This functions gives back false if the EdgeSet is
  1406     /// invalidated. It occurs when a node in the underlying graph is
  1407     /// erased and it is not isolated in the EdgeSet.
  1408     bool valid() const {
  1409       return _nodes.attached();
  1410     }
  1411 
  1412   };
  1413 
  1414 }
  1415 
  1416 #endif