RhodeCode

/* -*- mode: C++; indent-tabs-mode: nil; -*-

 *

 * This file is a part of LEMON, a generic C++ optimization library.

 *

 * Copyright (C) 2003-2009

 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport

 * (Egervary Research Group on Combinatorial Optimization, EGRES).

 *

 * Permission to use, modify and distribute this software is granted

 * provided that this copyright notice appears in all copies. For

 * precise terms see the accompanying LICENSE file.

 *

 * This software is provided "AS IS" with no warranty of any kind,

 * express or implied, and with no claim as to its suitability for any

 * purpose.

 *

 */

#ifndef LEMON_LIST_GRAPH_H

#define LEMON_LIST_GRAPH_H

///\ingroup graphs

///\file

///\brief ListDigraph and ListGraph classes.

#include <lemon/core.h>

#include <lemon/error.h>

#include <lemon/bits/graph_extender.h>

#include <vector>

#include <list>

namespace lemon {

  class ListDigraphBase {

  protected:

    struct NodeT {

      int first_in, first_out;

      int prev, next;

    };

    struct ArcT {

      int target, source;

      int prev_in, prev_out;

      int next_in, next_out;

    };

    std::vector<NodeT> nodes;

    int first_node;

    int first_free_node;

    std::vector<ArcT> arcs;

    int first_free_arc;

  public:

    typedef ListDigraphBase Digraph;

    class Node {

      friend class ListDigraphBase;

    protected:

      int id;

      explicit Node(int pid) { id = pid;}

    public:

      Node() {}

      Node (Invalid) { id = -1; }

      bool operator==(const Node& node) const {return id == node.id;}

      bool operator!=(const Node& node) const {return id != node.id;}

      bool operator<(const Node& node) const {return id < node.id;}

    };

    class Arc {

      friend class ListDigraphBase;

    protected:

      int id;

      explicit Arc(int pid) { id = pid;}

    public:

      Arc() {}

      Arc (Invalid) { id = -1; }

      bool operator==(const Arc& arc) const {return id == arc.id;}

      bool operator!=(const Arc& arc) const {return id != arc.id;}

      bool operator<(const Arc& arc) const {return id < arc.id;}

    };

    ListDigraphBase()

      : nodes(), first_node(-1),

        first_free_node(-1), arcs(), first_free_arc(-1) {}

    int maxNodeId() const { return nodes.size()-1; }

    int maxArcId() const { return arcs.size()-1; }

    Node source(Arc e) const { return Node(arcs[e.id].source); }

    Node target(Arc e) const { return Node(arcs[e.id].target); }

    void first(Node& node) const {

      node.id = first_node;

    }

    void next(Node& node) const {

      node.id = nodes[node.id].next;

    }

    void first(Arc& arc) const {

      int n;

      for(n = first_node;

          n!=-1 && nodes[n].first_in == -1;

          n = nodes[n].next) {}

      arc.id = (n == -1) ? -1 : nodes[n].first_in;

    }

    void next(Arc& arc) const {

      if (arcs[arc.id].next_in != -1) {

        arc.id = arcs[arc.id].next_in;

      } else {

    }

    ///Contract two nodes.

    ///This function contracts the given two nodes.

    ///Node \c v is removed, but instead of deleting its

    ///incident arcs, they are joined to node \c u.

    ///If the last parameter \c r is \c true (this is the default value),

    ///then the newly created loops are removed.

    ///

    ///\note The moved arcs are joined to node \c u using changeSource()

    ///or changeTarget(), thus \c ArcIt and \c OutArcIt iterators are

    ///invalidated for the outgoing arcs of node \c v and \c InArcIt

    ///iterators are invalidated for the incomming arcs of \c v.

    ///Moreover all iterators referencing node \c v or the removed 

    ///loops are also invalidated. Other iterators remain valid.

    ///

    ///\warning This functionality cannot be used together with the Snapshot

    ///feature.

    void contract(Node u, Node v, bool r = true)

    {

      for(OutArcIt e(*this,v);e!=INVALID;) {

        OutArcIt f=e;

        ++f;

        if(r && target(e)==u) erase(e);

        else changeSource(e,u);

        e=f;

      }

      for(InArcIt e(*this,v);e!=INVALID;) {

        InArcIt f=e;

        ++f;

        if(r && source(e)==u) erase(e);

        else changeTarget(e,u);

        e=f;

      }

      erase(v);

    }

    ///Split a node.

    ///This function splits the given node. First, a new node is added

    ///to the digraph, then the source of each outgoing arc of node \c n

    ///is moved to this new node.

    ///If the second parameter \c connect is \c true (this is the default

    ///value), then a new arc from node \c n to the newly created node

    ///is also added.

    ///\return The newly created node.

    ///

    ///

    ///\warning This functionality cannot be used together with the

    ///Snapshot feature.

    Node split(Node n, bool connect = true) {

      Node b = addNode();

      }

      if (connect) addArc(n,b);

      return b;

    }

    ///Split an arc.

    ///This function splits the given arc. First, a new node \c v is

    ///added to the digraph, then the target node of the original arc

    ///is set to \c v. Finally, an arc from \c v to the original target

    ///is added.

    ///\return The newly created node.

    ///

    ///\note \c InArcIt iterators referencing the original arc are

    ///invalidated. Other iterators remain valid.

    ///

    ///\warning This functionality cannot be used together with the

    ///Snapshot feature.

    Node split(Arc a) {

      Node v = addNode();

      addArc(v,target(a));

      changeTarget(a,v);

      return v;

    }

    ///Clear the digraph.

    ///This function erases all nodes and arcs from the digraph.

    ///

    void clear() {

      Parent::clear();

    }

    /// Reserve memory for nodes.

    /// Using this function, it is possible to avoid superfluous memory

    /// allocation: if you know that the digraph you want to build will

    /// be large (e.g. it will contain millions of nodes and/or arcs),

    /// then it is worth reserving space for this amount before starting

    /// to build the digraph.

    /// \sa reserveArc()

    void reserveNode(int n) { nodes.reserve(n); };

    /// Reserve memory for arcs.

    /// Using this function, it is possible to avoid superfluous memory

    /// allocation: if you know that the digraph you want to build will

    /// be large (e.g. it will contain millions of nodes and/or arcs),