// -*- c++ -*-

 #ifndef HUGO_SAGE_GRAPH_H

 #define HUGO_SAGE_GRAPH_H

 #include <iostream>

 #include <vector>

 #include <hugo/invalid.h>

 namespace hugo {

   template <typename It>

   int count(It it) { 

     int i=0;

     for( ; it.valid(); ++it) { ++i; } 

     return i;

   }

   class SageGraph {

     struct node_item;

     struct edge_item;

   public:

     class Node;

     class NodeIt;

     class Edge;

     class EdgeIt;

     class OutEdgeIt;

     class InEdgeIt;

     class SymEdgeIt;

     template <typename T> class NodeMap;

     template <typename T> class EdgeMap;

 //  private:

     template <typename T> friend class NodeMap;

     template <typename T> friend class EdgeMap;

     template <typename T>

     class NodeMap {

       const SageGraph& G; 

       std::vector<T> container;

     public:

       typedef T ValueType;

       typedef Node KeyType;

       NodeMap(const SageGraph& _G) : G(_G), container(G.node_id) { }

       NodeMap(const SageGraph& _G, T a) : 

 	G(_G), container(G.node_id, a) { }

       void set(Node n, T a) { container[/*G.id(n)*/n.node->id]=a; }

 //      T get(Node n) const { return container[/*G.id(n)*/n.node->id]; }

       typename std::vector<T>::reference operator[](Node n) { 

 	return container[/*G.id(n)*/n.node->id]; }

       typename std::vector<T>::const_reference operator[](Node n) const { 

 	return container[/*G.id(n)*/n.node->id]; 

       }

       void update() { container.resize(G.node_id); }

       void update(T a) { container.resize(G.node_id, a); }

     };

     template <typename T>

     class EdgeMap {

       const SageGraph& G; 

       std::vector<T> container;

     public:

       typedef T ValueType;

       typedef Edge KeyType;

       EdgeMap(const SageGraph& _G) : G(_G), container(G.edge_id) { }

       EdgeMap(const SageGraph& _G, T a) : 

 	G(_G), container(G.edge_id, a) { }

       void set(Edge e, T a) { container[/*G.id(e)*/e.edge->id]=a; }

 //      T get(Edge e) const { return container[/*G.id(e)*/e.edge->id]; }

       typename std::vector<T>::reference operator[](Edge e) { 

 	return container[/*G.id(e)*/e.edge->id]; } 

       typename std::vector<T>::const_reference operator[](Edge e) const { 

 	return container[/*G.id(e)*/e.edge->id]; 

       } 

       void update() { container.resize(G.edge_id); }

       void update(T a) { container.resize(G.edge_id, a); }

     };

   private:

     int node_id;

     int edge_id;

     int _node_num;

     int _edge_num;

     node_item* _first_node;

     node_item* _last_node;

     struct node_item {

       int id;

       edge_item* _first_out_edge;

       edge_item* _last_out_edge;

       edge_item* _first_in_edge;

       edge_item* _last_in_edge;

       node_item* _next_node;

       node_item* _prev_node;

     };

     struct edge_item {

       int id;

       node_item* _tail;

       node_item* _head;

       edge_item* _next_out;

       edge_item* _prev_out;

       edge_item* _next_in;

       edge_item* _prev_in;

     };

     node_item* _add_node() { 

       node_item* p=new node_item;

       p->id=node_id++;

       p->_first_out_edge=0;

       p->_last_out_edge=0;

       p->_first_in_edge=0;

       p->_last_in_edge=0;

       p->_prev_node=_last_node;

       p->_next_node=0;

       if (_last_node) _last_node->_next_node=p;

       _last_node=p;

       if (!_first_node) _first_node=p;

       ++_node_num;

       return p;

     }

     edge_item* _add_edge(node_item* _tail, node_item* _head) {

       edge_item* e=new edge_item;

       e->id=edge_id++;

       e->_tail=_tail;

       e->_head=_head;

       e->_prev_out=_tail->_last_out_edge;

       if (_tail->_last_out_edge) (_tail->_last_out_edge)->_next_out=e;

       _tail->_last_out_edge=e;

       if (!_tail->_first_out_edge) _tail->_first_out_edge=e; 

       e->_next_out=0;

       e->_prev_in=_head->_last_in_edge;

       if (_head->_last_in_edge) (_head->_last_in_edge)->_next_in=e;

       _head->_last_in_edge=e;

       if (!_head->_first_in_edge) { _head->_first_in_edge=e; } 

       e->_next_in=0;

       ++_edge_num;

       return e;

     }

     //deletes a node which has no out edge and no in edge

     void _delete_node(node_item* v) {

       if (v->_next_node) (v->_next_node)->_prev_node=v->_prev_node; else 

 	_last_node=v->_prev_node;

       if (v->_prev_node) (v->_prev_node)->_next_node=v->_next_node; else 

 	_first_node=v->_next_node;

       delete v;

       --_node_num;

     }

     void _delete_edge(edge_item* e) {

       if (e->_next_out) (e->_next_out)->_prev_out=e->_prev_out; else 

 	(e->_tail)->_last_out_edge=e->_prev_out;

       if (e->_prev_out) (e->_prev_out)->_next_out=e->_next_out; else 

 	(e->_tail)->_first_out_edge=e->_next_out;

       if (e->_next_in) (e->_next_in)->_prev_in=e->_prev_in; else 

 	(e->_head)->_last_in_edge=e->_prev_in;

       if (e->_prev_in) (e->_prev_in)->_next_in=e->_next_in; else 

 	(e->_head)->_first_in_edge=e->_next_in;

       delete e;

       --_edge_num;

     }

     void _set_tail(edge_item* e, node_item* _tail) {

       if (e->_next_out) (e->_next_out)->_prev_out=e->_prev_out; else 

 	(e->_tail)->_last_out_edge=e->_prev_out;

       if (e->_prev_out) (e->_prev_out)->_next_out=e->_next_out; else 

 	(e->_tail)->_first_out_edge=e->_next_out;

       e->_tail=_tail;

       e->_prev_out=_tail->_last_out_edge;

       if (_tail->_last_out_edge) (_tail->_last_out_edge)->_next_out=e;

       _tail->_last_out_edge=e;

       if (!_tail->_first_out_edge) _tail->_first_out_edge=e; 

       e->_next_out=0;

     }

     void _set_head(edge_item* e, node_item* _head) {

       if (e->_next_in) (e->_next_in)->_prev_in=e->_prev_in; else 

 	(e->_head)->_last_in_edge=e->_prev_in;

       if (e->_prev_in) (e->_prev_in)->_next_in=e->_next_in; else 

 	(e->_head)->_first_in_edge=e->_next_in;

       e->_head=_head;

       e->_prev_in=_head->_last_in_edge;

       if (_head->_last_in_edge) (_head->_last_in_edge)->_next_in=e;

       _head->_last_in_edge=e;

       if (!_head->_first_in_edge) { _head->_first_in_edge=e; } 

       e->_next_in=0;

     }

   public:

     /* default constructor */

     SageGraph() : node_id(0), edge_id(0), _node_num(0), _edge_num(0), _first_node(0), _last_node(0) { }

     ~SageGraph() { 

       NodeIt n;

       while (this->valid(first(n))) erase(n);

       //while (first<NodeIt>().valid()) erase(first<NodeIt>());

     }

     int nodeNum() const { return _node_num; }

     int edgeNum() const { return _edge_num; }

     /* functions to construct iterators from the graph, or from each other */

     //NodeIt firstNode() const { return NodeIt(*this); }

     //EdgeIt firstEdge() const { return EdgeIt(*this); }

     //OutEdgeIt firstOutEdge(const Node v) const { return OutEdgeIt(v); }

     //InEdgeIt firstInEdge(const Node v) const { return InEdgeIt(v); }

     //SymEdgeIt firstSymEdge(const Node v) const { return SymEdgeIt(v); }

     Node tail(Edge e) const { return e.tailNode(); }

     Node head(Edge e) const { return e.headNode(); }

     Node aNode(const OutEdgeIt& e) const { return e.aNode(); }

     Node aNode(const InEdgeIt& e) const { return e.aNode(); }

     Node aNode(const SymEdgeIt& e) const { return e.aNode(); }

     Node bNode(const OutEdgeIt& e) const { return e.bNode(); }

     Node bNode(const InEdgeIt& e) const { return e.bNode(); }

     Node bNode(const SymEdgeIt& e) const { return e.bNode(); }

     //Node invalid_node() { return Node(); }

     //Edge invalid_edge() { return Edge(); }

     //OutEdgeIt invalid_out_edge() { return OutEdgeIt(); }

     //InEdgeIt invalid_in_edge() { return InEdgeIt(); }

     //SymEdgeIt invalid_sym_edge() { return SymEdgeIt(); }

     /* same methods in other style */

     /* for experimental purpose */

     NodeIt& first(NodeIt& v) const { 

       v=NodeIt(*this); return v; }

     EdgeIt& first(EdgeIt& e) const { 

       e=EdgeIt(*this); return e; }

     OutEdgeIt& first(OutEdgeIt& e, Node v) const { 

       e=OutEdgeIt(*this, v); return e; }

     InEdgeIt& first(InEdgeIt& e, Node v) const { 

       e=InEdgeIt(*this, v); return e; }

     SymEdgeIt& first(SymEdgeIt& e, Node v) const { 

       e=SymEdgeIt(*this, v); return e; }

     //void getTail(Node& n, const Edge& e) const { n=tail(e); }

     //void getHead(Node& n, const Edge& e) const { n=head(e); }

     //void getANode(Node& n, const OutEdgeIt& e) const { n=e.aNode(); }

     //void getANode(Node& n, const InEdgeIt& e) const { n=e.aNode(); }

     //void getANode(Node& n, const SymEdgeIt& e) const { n=e.aNode(); }

     //void getBNode(Node& n, const OutEdgeIt& e) const { n=e.bNode(); }

     //void getBNode(Node& n, const InEdgeIt& e) const { n=e.bNode(); }

     //void getBNode(Node& n, const SymEdgeIt& e) const { n=e.bNode(); }

     //void get_invalid(Node& n) { n=Node(); }

     //void get_invalid(Edge& e) { e=Edge(); }

     //void get_invalid(OutEdgeIt& e) { e=OutEdgeIt(); }

     //void get_invalid(InEdgeIt& e) { e=InEdgeIt(); }

     //void get_invalid(SymEdgeIt& e) { e=SymEdgeIt(); }

 //     template< typename It >

 //     It first() const { 

 //       It e;

 //       first(e);

 //       return e; 

 //     }

 //     template< typename It >

 //     It first(Node v) const { 

 //       It e;

 //       first(e, v);

 //       return e; 

 //     }

     bool valid(Node n) const { return n.valid(); }

     bool valid(Edge e) const { return e.valid(); }

 //    template <typename It> It getNext(It it) const { 

 //      It tmp(it); next(tmp); return tmp; }

 //     NodeIt& next(NodeIt& it) const { return ++it; }

 //     EdgeIt& next(EdgeIt& it) const { return ++it; }

 //     OutEdgeIt& next(OutEdgeIt& it) const { return ++it; }

 //     InEdgeIt& next(InEdgeIt& it) const { return ++it; }

 //     SymEdgeIt& next(SymEdgeIt& it) const { return ++it; }

 //    template <typename It> It& next(It& it) const { return ++it; }

     template <typename It> It& next(It& it) const { ++it; return it; }

     /* for getting id's of graph objects */

     /* these are important for the implementation of property vectors */

     int id(Node v) const { return v.node->id; }

     int id(Edge e) const { return e.edge->id; }

     /* adding nodes and edges */

     Node addNode() { return Node(_add_node()); }

     Edge addEdge(Node u, Node v) {

       return Edge(_add_edge(u.node, v.node)); 

     }

     void erase(Node i) { 

       { 

 	OutEdgeIt e;

 	while (this->valid(first(e, i))) erase(e);

       }

       {

 	InEdgeIt e;

 	while (this->valid(first(e, i))) erase(e);

       }

       //while (first<OutEdgeIt>(i).valid()) erase(first<OutEdgeIt>(i));

       //while (first<InEdgeIt>(i).valid()) erase(first<InEdgeIt>(i));

       _delete_node(i.node); 

     }

     void erase(Edge e) { _delete_edge(e.edge); }

     void clear() { 

       NodeIt e;

       while (this->valid(first(e))) erase(e); 

       //while (first<NodeIt>().valid()) erase(first<NodeIt>());

     }

     void setTail(Edge e, Node tail) {

       _set_tail(e.edge, tail.node); 

     }

     void setHead(Edge e, Node head) {

       _set_head(e.edge, head.node); 

     }

     /* stream operations, for testing purpose */

 //     friend std::ostream& operator<<(std::ostream& os, const Node& i) { 

 //       if (i.valid())

 // 	os << i.node->id; 

 //       else

 // 	os << "invalid";

 //       return os; 

 //     }

 //     friend std::ostream& operator<<(std::ostream& os, const Edge& i) { 

 //       if (i.valid()) 

 // 	os << "(" << i.edge->_tail->id << "--" << i.edge->id << "->" << i.edge->_head->id << ")"; 

 //       else 

 // 	os << "invalid";

 //       return os; 

 //     }

     class Node {

       friend class SageGraph;

       template <typename T> friend class NodeMap;

       friend class Edge;

       friend class OutEdgeIt;

       friend class InEdgeIt;

       friend class SymEdgeIt;

       //public:  //FIXME: It is required by op= of NodeIt

     protected:

       node_item* node;

     protected:

       friend int SageGraph::id(Node v) const; 

     public:

       Node() /*: node(0)*/ { }

       Node(const Invalid&) : node(0) { }

     protected:

       Node(node_item* _node) : node(_node) { }

       bool valid() const { return (node); }

     public:

       //void makeInvalid() { node=0; }

       friend bool operator==(Node u, Node v) { return v.node==u.node; } 

       friend bool operator!=(Node u, Node v) { return v.node!=u.node; } 

       friend std::ostream& operator<<(std::ostream& os, const Node& i);

     };

     class NodeIt : public Node {

       friend class SageGraph;

       //protected:

     public: //for everybody but marci

       NodeIt(const SageGraph& G) : Node(G._first_node) { }

     public:

       NodeIt() : Node() { }

       NodeIt(const Invalid& i) : Node(i) { }

     protected:

       NodeIt(node_item* v) : Node(v) { }

       NodeIt& operator++() { node=node->_next_node; return *this; }

       //FIXME::

       //      NodeIt& operator=(const Node& e)

       //      { node=e.node; return *this; }

     };

     class Edge {

       friend class SageGraph;

       template <typename T> friend class EdgeMap;

       friend class Node;

       friend class NodeIt;

     protected:

       edge_item* edge;

       friend int SageGraph::id(Edge e) const;

     public:

       Edge() /*: edge(0)*/ { }

       Edge(const Invalid&) : edge(0) { }

       //Edge() { }

     protected:

       Edge(edge_item* _edge) : edge(_edge) { }

       bool valid() const { return (edge); }

     public:

       //void makeInvalid() { edge=0; }

       friend bool operator==(Edge u, Edge v) { return v.edge==u.edge; } 

       friend bool operator!=(Edge u, Edge v) { return v.edge!=u.edge; } 

     protected:

       Node tailNode() const { return Node(edge->_tail); }

       Node headNode() const { return Node(edge->_head); }

     public:

       friend std::ostream& operator<<(std::ostream& os, const Edge& i);

     };

     class EdgeIt : public Edge {

       friend class SageGraph;

       //protected: 

     public: //for alpar

       EdgeIt(const SageGraph& G) {

 	node_item* v=G._first_node;

 	if (v) edge=v->_first_out_edge; else edge=0;

 	while (v && !edge) { v=v->_next_node; if (v) edge=v->_first_out_edge; }

       }

     public:

       EdgeIt() : Edge() { }

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

     protected:

       EdgeIt(edge_item* _e) : Edge(_e) { }

       EdgeIt& operator++() { 

 	node_item* v=edge->_tail;

 	edge=edge->_next_out; 

 	while (v && !edge) { v=v->_next_node; if (v) edge=v->_first_out_edge; }

 	return *this;

       }

     };

     class OutEdgeIt : public Edge {

       friend class SageGraph;

       //node_item* v;

       //protected: 

     protected: //for alpar

       OutEdgeIt(const Node& _v) /*: v(_v.node)*/ { edge=_v.node->_first_out_edge; }

     public:

       OutEdgeIt() : Edge()/*, v(0)*/ { }

       OutEdgeIt(const Invalid& i) : Edge(i) { }

       OutEdgeIt(const SageGraph&, Node _v) /*: v(_v.node)*/ { edge=_v.node->_first_out_edge; }

     protected:

       OutEdgeIt& operator++() { edge=edge->_next_out; return *this; }

     protected:

       Node aNode() const { return Node(edge->_tail); }

       Node bNode() const { return Node(edge->_head); }

     };

     class InEdgeIt : public Edge {

       friend class SageGraph;

       //node_item* v;

       //protected:

     protected: //for alpar

       InEdgeIt(const Node& _v) /*: v(_v.node)*/ { edge=_v.node->_first_in_edge; }

     public:

       InEdgeIt() : Edge()/*, v(0)*/ { }

       InEdgeIt(const Invalid& i) : Edge(i) { }

       InEdgeIt(const SageGraph&, Node _v) /*: v(_v.node)*/ { edge=_v.node->_first_in_edge; }

     protected:

       InEdgeIt& operator++() { edge=edge->_next_in; return *this; }

     protected:

       Node aNode() const { return Node(edge->_head); }

       Node bNode() const { return Node(edge->_tail); }

     };

     class SymEdgeIt : public Edge {

       friend class SageGraph;

       bool out_or_in; //1 iff out, 0 iff in

       //node_item* v;

       //protected:

     protected: //for alpar

       SymEdgeIt(const Node& _v) /*: v(_v.node)*/ { 

 	out_or_in=1;

 	edge=_v.node->_first_out_edge; 

 	if (!edge) { edge=_v.node->_first_in_edge; out_or_in=0; }

       }

     public:

       SymEdgeIt() : Edge() /*, v(0)*/ { }

       SymEdgeIt(const Invalid& i) : Edge(i) { }

       SymEdgeIt(const SageGraph&, Node _v) /*: v(_v.node)*/ { 

 	out_or_in=1;

 	edge=_v.node->_first_out_edge; 

 	if (!edge) { edge=_v.node->_first_in_edge; out_or_in=0; }

       }

     protected:

       SymEdgeIt& operator++() { 

 	if (out_or_in) { 

 	  node_item* v=edge->_tail;

 	  edge=edge->_next_out; 

 	  if (!edge) { out_or_in=0; edge=v->_first_in_edge; }

 	} else {

 	  edge=edge->_next_in; 

 	}

 	return *this;

       }

     protected:

       Node aNode() const { 

 	return (out_or_in) ? Node(edge->_tail) : Node(edge->_head); }

       Node bNode() const { 

 	return (out_or_in) ? Node(edge->_head) : Node(edge->_tail); }

     };

   };

   inline

   std::ostream& operator<<(std::ostream& os, const SageGraph::Node& i) { 

     if (i.valid())

       os << i.node->id;

     else

       os << "invalid";

     return os; 

   }

   inline

   std::ostream& operator<<(std::ostream& os, const SageGraph::Edge& i) { 

     if (i.valid()) 

       os << "(" << i.tailNode() << "--" << i.edge->id << "->" 

 	 << i.headNode() << ")"; 

     else 

       os << "invalid";

     return os; 

   }

   class UndirSageGraph : public SageGraph {

   public:

     typedef SymEdgeIt OutEdgeIt;

     typedef SymEdgeIt InEdgeIt;

   };

 } //namespace hugo

 #endif //HUGO_SAGE_GRAPH_H