/* -*- 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 GRID_GRAPH_H

#define GRID_GRAPH_H

#include <lemon/core.h>

#include <lemon/bits/graph_extender.h>

#include <lemon/dim2.h>

#include <lemon/assert.h>

///\ingroup graphs

///\file

///\brief GridGraph class.

namespace lemon {

  class GridGraphBase {

  public:

    typedef GridGraphBase Graph;

    class Node;

    class Edge;

    class Arc;

  public:

    GridGraphBase() {}

  protected:

    void construct(int width, int height) {

       _width = width; _height = height;

      _node_num = width * height;

      _edge_num = 2 * _node_num - width - height;

      _edge_limit = _node_num - _width;

    }

  public:

    Node operator()(int i, int j) const {

      LEMON_DEBUG(0 <= i && i < _width &&

                  0 <= j  && j < _height, "Index out of range");

      return Node(i + j * _width);

    }

    int col(Node n) const {

      return n._id % _width;

    }

    int row(Node n) const {

      return n._id / _width;

    }

    dim2::Point<int> pos(Node n) const {

      return dim2::Point<int>(col(n), row(n));

    }

    int width() const {

      return _width;

    }

    int height() const {

      return _height;

    }

    typedef True NodeNumTag;

    typedef True EdgeNumTag;

    typedef True ArcNumTag;

    int nodeNum() const { return _node_num; }

    int edgeNum() const { return _edge_num; }

    int arcNum() const { return 2 * _edge_num; }

    Node u(Edge edge) const {

      if (edge._id < _edge_limit) {

        return edge._id;

      } else {

        return (edge._id - _edge_limit) % (_width - 1) +

          (edge._id - _edge_limit) / (_width - 1) * _width;

      }

    }

    Node v(Edge edge) const {

      if (edge._id < _edge_limit) {

        return edge._id + _width;

      } else {

        return (edge._id - _edge_limit) % (_width - 1) +

          (edge._id - _edge_limit) / (_width - 1) * _width + 1;

      }

    }

    Node source(Arc arc) const {

      return (arc._id & 1) == 1 ? u(arc) : v(arc);

    }

    Node target(Arc arc) const {

      return (arc._id & 1) == 1 ? v(arc) : u(arc);

    }

    static int id(Node node) { return node._id; }

    static int id(Edge edge) { return edge._id; }

    static int id(Arc arc) { return arc._id; }

    int maxNodeId() const { return _node_num - 1; }

    int maxEdgeId() const { return _edge_num - 1; }

    int maxArcId() const { return 2 * _edge_num - 1; }

    static Node nodeFromId(int id) { return Node(id);}

    static Edge edgeFromId(int id) { return Edge(id);}

    static Arc arcFromId(int id) { return Arc(id);}

    typedef True FindEdgeTag;

    typedef True FindArcTag;

    Edge findEdge(Node u, Node v, Edge prev = INVALID) const {

      if (prev != INVALID) return INVALID;

      if (v._id > u._id) {

        if (v._id - u._id == _width)

          return Edge(u._id);

        if (v._id - u._id == 1 && u._id % _width < _width - 1) {

          return Edge(u._id / _width * (_width - 1) +

                      u._id % _width + _edge_limit);

        }

      } else {

        if (u._id - v._id == _width)

          return Edge(v._id);

        if (u._id - v._id == 1 && v._id % _width < _width - 1) {

          return Edge(v._id / _width * (_width - 1) +

                      v._id % _width + _edge_limit);

        }

      }

      return INVALID;

    }

    Arc findArc(Node u, Node v, Arc prev = INVALID) const {

      if (prev != INVALID) return INVALID;

      if (v._id > u._id) {

        if (v._id - u._id == _width)

          return Arc((u._id << 1) | 1);

        if (v._id - u._id == 1 && u._id % _width < _width - 1) {

          return Arc(((u._id / _width * (_width - 1) +

                       u._id % _width + _edge_limit) << 1) | 1);

        }

      } else {

        if (u._id - v._id == _width)

          return Arc(v._id << 1);

        if (u._id - v._id == 1 && v._id % _width < _width - 1) {

          return Arc((v._id / _width * (_width - 1) +

                       v._id % _width + _edge_limit) << 1);

        }

      }

      return INVALID;

    }

    class Node {

      friend class GridGraphBase;

    protected:

      int _id;

      Node(int id) : _id(id) {}

    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 Edge {

      friend class GridGraphBase;

      friend class Arc;

    protected:

      int _id;

      Edge(int id) : _id(id) {}

    public:

      Edge() {}

      Edge (Invalid) : _id(-1) {}

      bool operator==(const Edge edge) const {return _id == edge._id;}

      bool operator!=(const Edge edge) const {return _id != edge._id;}

      bool operator<(const Edge edge) const {return _id < edge._id;}

    };

    class Arc {

      friend class GridGraphBase;

    protected:

      int _id;

      Arc(int id) : _id(id) {}

    public:

      Arc() {}

      Arc (Invalid) : _id(-1) {}

      operator Edge() const { return _id != -1 ? Edge(_id >> 1) : INVALID; }

      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;}

    };

    static bool direction(Arc arc) {

      return (arc._id & 1) == 1;

    }

    static Arc direct(Edge edge, bool dir) {

      return Arc((edge._id << 1) | (dir ? 1 : 0));

    }

    void first(Node& node) const {

      node._id = _node_num - 1;

    }

    static void next(Node& node) {

      --node._id;

    }

    void first(Edge& edge) const {

      edge._id = _edge_num - 1;

    }

    static void next(Edge& edge) {

      --edge._id;

    }

    void first(Arc& arc) const {

      arc._id = 2 * _edge_num - 1;

    }

    static void next(Arc& arc) {

      --arc._id;

    }

    void firstOut(Arc& arc, const Node& node) const {

      if (node._id % _width < _width - 1) {

        arc._id = (_edge_limit + node._id % _width +

                   (node._id / _width) * (_width - 1)) << 1 | 1;

        return;

      }

      if (node._id < _node_num - _width) {

        arc._id = node._id << 1 | 1;

        return;

      }

      if (node._id % _width > 0) {

        arc._id = (_edge_limit + node._id % _width +

                   (node._id / _width) * (_width - 1) - 1) << 1;

        return;

      }

      if (node._id >= _width) {

        arc._id = (node._id - _width) << 1;

        return;

      }

      arc._id = -1;

    }

    void nextOut(Arc& arc) const {

      int nid = arc._id >> 1;

      if ((arc._id & 1) == 1) {

        if (nid >= _edge_limit) {

          nid = (nid - _edge_limit) % (_width - 1) +

            (nid - _edge_limit) / (_width - 1) * _width;

          if (nid < _node_num - _width) {

            arc._id = nid << 1 | 1;

            return;

          }

        }

        if (nid % _width > 0) {

          arc._id = (_edge_limit + nid % _width +

                     (nid / _width) * (_width - 1) - 1) << 1;

          return;

        }

        if (nid >= _width) {

          arc._id = (nid - _width) << 1;

          return;

        }

      } else {

        if (nid >= _edge_limit) {

          nid = (nid - _edge_limit) % (_width - 1) +

            (nid - _edge_limit) / (_width - 1) * _width + 1;

          if (nid >= _width) {

            arc._id = (nid - _width) << 1;

            return;

          }

        }

      }

      arc._id = -1;

    }

    void firstIn(Arc& arc, const Node& node) const {

      if (node._id % _width < _width - 1) {

        arc._id = (_edge_limit + node._id % _width +

                   (node._id / _width) * (_width - 1)) << 1;

        return;

      }

      if (node._id < _node_num - _width) {

        arc._id = node._id << 1;

        return;

      }

      if (node._id % _width > 0) {

        arc._id = (_edge_limit + node._id % _width +

                   (node._id / _width) * (_width - 1) - 1) << 1 | 1;

        return;

      }

      if (node._id >= _width) {

        arc._id = (node._id - _width) << 1 | 1;

        return;

      }

      arc._id = -1;

    }

    void nextIn(Arc& arc) const {

      int nid = arc._id >> 1;

      if ((arc._id & 1) == 0) {

        if (nid >= _edge_limit) {

          nid = (nid - _edge_limit) % (_width - 1) +

            (nid - _edge_limit) / (_width - 1) * _width;

          if (nid < _node_num - _width) {

            arc._id = nid << 1;

            return;

          }

        }

        if (nid % _width > 0) {

          arc._id = (_edge_limit + nid % _width +

                     (nid / _width) * (_width - 1) - 1) << 1 | 1;

          return;

        }

        if (nid >= _width) {

          arc._id = (nid - _width) << 1 | 1;

          return;

        }

      } else {

        if (nid >= _edge_limit) {

          nid = (nid - _edge_limit) % (_width - 1) +

            (nid - _edge_limit) / (_width - 1) * _width + 1;

          if (nid >= _width) {

            arc._id = (nid - _width) << 1 | 1;

            return;

          }

        }

      }

      arc._id = -1;

    }

    void firstInc(Edge& edge, bool& dir, const Node& node) const {

      if (node._id % _width < _width - 1) {

        edge._id = _edge_limit + node._id % _width +

          (node._id / _width) * (_width - 1);

        dir = true;

        return;

      }

      if (node._id < _node_num - _width) {

        edge._id = node._id;

        dir = true;

        return;

      }

      if (node._id % _width > 0) {

        edge._id = _edge_limit + node._id % _width +

          (node._id / _width) * (_width - 1) - 1;

        dir = false;

        return;

      }

      if (node._id >= _width) {

        edge._id = node._id - _width;

        dir = false;

        return;

      }

      edge._id = -1;

      dir = true;

    }

    void nextInc(Edge& edge, bool& dir) const {

      int nid = edge._id;

      if (dir) {

        if (nid >= _edge_limit) {

          nid = (nid - _edge_limit) % (_width - 1) +

            (nid - _edge_limit) / (_width - 1) * _width;

          if (nid < _node_num - _width) {

            edge._id = nid;

            return;

          }

        }

        if (nid % _width > 0) {

          edge._id = _edge_limit + nid % _width +

            (nid / _width) * (_width - 1) - 1;

          dir = false;

          return;

        }

        if (nid >= _width) {

          edge._id = nid - _width;

          dir = false;

          return;

        }

      } else {

        if (nid >= _edge_limit) {

          nid = (nid - _edge_limit) % (_width - 1) +

            (nid - _edge_limit) / (_width - 1) * _width + 1;

          if (nid >= _width) {

            edge._id = nid - _width;

            return;

          }

        }

      }

      edge._id = -1;

      dir = true;

    }

    Arc right(Node n) const {

      if (n._id % _width < _width - 1) {

        return Arc(((_edge_limit + n._id % _width +

                    (n._id / _width) * (_width - 1)) << 1) | 1);

      } else {

        return INVALID;

      }

    }

    Arc left(Node n) const {

      if (n._id % _width > 0) {

        return Arc((_edge_limit + n._id % _width +

                     (n._id / _width) * (_width - 1) - 1) << 1);

      } else {

        return INVALID;

      }

    }

    Arc up(Node n) const {

      if (n._id < _edge_limit) {

        return Arc((n._id << 1) | 1);

      } else {

        return INVALID;

      }

    }

    Arc down(Node n) const {

      if (n._id >= _width) {

        return Arc((n._id - _width) << 1);

      } else {

        return INVALID;

      }

    }

  private:

    int _width, _height;

    int _node_num, _edge_num;

    int _edge_limit;

  };

  typedef GraphExtender<GridGraphBase> ExtendedGridGraphBase;

  /// \ingroup graphs

  ///

  /// \brief Grid graph class

  ///

  /// This class implements a special graph type. The nodes of the

  /// graph can be indexed by two integer \c (i,j) value where \c i is

  /// in the \c [0..width()-1] range and j is in the \c

  /// [0..height()-1] range.  Two nodes are connected in the graph if

  /// the indexes differ exactly on one position and exactly one is

  /// the difference. The nodes of the graph can be indexed by position

  /// with the \c operator()() function. The positions of the nodes can be

  /// get with \c pos(), \c col() and \c row() members. The outgoing

  /// arcs can be retrieved with the \c right(), \c up(), \c left()

  /// and \c down() functions, where the bottom-left corner is the

  /// origin.

  ///

  /// \image html grid_graph.png

  /// \image latex grid_graph.eps "Grid graph" width=\textwidth

  ///

  /// A short example about the basic usage:

  ///\code

  /// GridGraph graph(rows, cols);

  /// GridGraph::NodeMap<int> val(graph);

  /// for (int i = 0; i < graph.width(); ++i) {

  ///   for (int j = 0; j < graph.height(); ++j) {

  ///     val[graph(i, j)] = i + j;

  ///   }

  /// }

  ///\endcode

  ///

  /// This graph type fully conforms to the \ref concepts::Graph

  /// "Graph" concept, and it also has an important extra feature

  /// that its maps are real \ref concepts::ReferenceMap

  /// "reference map"s.

  class GridGraph : public ExtendedGridGraphBase {

  public:

    typedef ExtendedGridGraphBase Parent;

    /// \brief Map to get the indices of the nodes as dim2::Point<int>.

    ///

    /// Map to get the indices of the nodes as dim2::Point<int>.

    class IndexMap {

    public:

      /// \brief The key type of the map

      typedef GridGraph::Node Key;

      /// \brief The value type of the map

      typedef dim2::Point<int> Value;

      /// \brief Constructor

      ///

      /// Constructor

      IndexMap(const GridGraph& graph) : _graph(graph) {}

      /// \brief The subscript operator

      ///

      /// The subscript operator.

      Value operator[](Key key) const {

        return _graph.pos(key);

      }

    private:

      const GridGraph& _graph;

    };

    /// \brief Map to get the column of the nodes.

    ///

    /// Map to get the column of the nodes.

    class ColMap {

    public:

      /// \brief The key type of the map

      typedef GridGraph::Node Key;

      /// \brief The value type of the map

      typedef int Value;

      /// \brief Constructor

      ///

      /// Constructor

      ColMap(const GridGraph& graph) : _graph(graph) {}

      /// \brief The subscript operator

      ///

      /// The subscript operator.

      Value operator[](Key key) const {

        return _graph.col(key);

      }

    private:

      const GridGraph& _graph;

    };

    /// \brief Map to get the row of the nodes.

    ///

    /// Map to get the row of the nodes.

    class RowMap {

    public:

      /// \brief The key type of the map

      typedef GridGraph::Node Key;

      /// \brief The value type of the map

      typedef int Value;

      /// \brief Constructor

      ///

      /// Constructor

      RowMap(const GridGraph& graph) : _graph(graph) {}

      /// \brief The subscript operator

      ///

      /// The subscript operator.

      Value operator[](Key key) const {

        return _graph.row(key);

      }

    private:

      const GridGraph& _graph;

    };

    /// \brief Constructor

    ///

    /// Construct a grid graph with given size.

    GridGraph(int width, int height) { construct(width, height); }

    /// \brief Resize the graph

    ///

    /// Resize the graph. The function will fully destroy and rebuild

    /// the graph.  This cause that the maps of the graph will

    /// reallocated automatically and the previous values will be

    /// lost.

    void resize(int width, int height) {

      Parent::notifier(Arc()).clear();

      Parent::notifier(Edge()).clear();

      Parent::notifier(Node()).clear();

      construct(width, height);

      Parent::notifier(Node()).build();

      Parent::notifier(Edge()).build();

      Parent::notifier(Arc()).build();

    }

    /// \brief The node on the given position.

    ///

    /// Gives back the node on the given position.

    Node operator()(int i, int j) const {

      return Parent::operator()(i, j);

    }

    /// \brief Gives back the column index of the node.

    ///

    /// Gives back the column index of the node.

    int col(Node n) const {

      return Parent::col(n);

    }

    /// \brief Gives back the row index of the node.

    ///

    /// Gives back the row index of the node.

    int row(Node n) const {

      return Parent::row(n);

    }

    /// \brief Gives back the position of the node.

    ///

    /// Gives back the position of the node, ie. the <tt>(col,row)</tt> pair.

    dim2::Point<int> pos(Node n) const {

      return Parent::pos(n);

    }

    /// \brief Gives back the number of the columns.

    ///

    /// Gives back the number of the columns.

    int width() const {

      return Parent::width();

    }

    /// \brief Gives back the number of the rows.

    ///

    /// Gives back the number of the rows.

    int height() const {

      return Parent::height();

    }

    /// \brief Gives back the arc goes right from the node.

    ///

    /// Gives back the arc goes right from the node. If there is not

    /// outgoing arc then it gives back INVALID.

    Arc right(Node n) const {

      return Parent::right(n);

    }

    /// \brief Gives back the arc goes left from the node.

    ///

    /// Gives back the arc goes left from the node. If there is not

    /// outgoing arc then it gives back INVALID.

    Arc left(Node n) const {

      return Parent::left(n);

    }

    /// \brief Gives back the arc goes up from the node.

    ///

    /// Gives back the arc goes up from the node. If there is not

    /// outgoing arc then it gives back INVALID.

    Arc up(Node n) const {

      return Parent::up(n);

    }

    /// \brief Gives back the arc goes down from the node.

    ///

    /// Gives back the arc goes down from the node. If there is not

    /// outgoing arc then it gives back INVALID.

    Arc down(Node n) const {

      return Parent::down(n);

    }

    /// \brief Index map of the grid graph

    ///

    /// Just returns an IndexMap for the grid graph.

    IndexMap indexMap() const {

      return IndexMap(*this);

    }

    /// \brief Row map of the grid graph

    ///

    /// Just returns a RowMap for the grid graph.

    RowMap rowMap() const {

      return RowMap(*this);

    }

    /// \brief Column map of the grid graph

    ///

    /// Just returns a ColMap for the grid graph.

    ColMap colMap() const {

      return ColMap(*this);

    }

  };

}

#endif