/* -*- C++ -*-

  *

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

  *

  * Copyright (C) 2003-2006

  * 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_BPUGRAPH_ADAPTOR_H

 #define LEMON_BPUGRAPH_ADAPTOR_H

 #include <lemon/bits/invalid.h>

 #include <lemon/maps.h>

 #include <lemon/bits/graph_adaptor_extender.h>

 #include <lemon/bits/traits.h>

 #include <iostream>

 ///\ingroup graph_adaptors

 ///\file

 ///\brief Several graph adaptors.

 ///

 ///This file contains several useful bpugraph adaptor functions.

 ///

 ///\author Balazs Dezso

 namespace lemon {

   /// \ingroup graph_adaptors

   ///

   /// \brief Base type for the Bipartite Undirected Graph Adaptors

   ///

   /// This is the base type for most of LEMON bpugraph adaptors. 

   /// This class implements a trivial graph adaptor i.e. it only wraps the 

   /// functions and types of the graph. The purpose of this class is to 

   /// make easier implementing graph adaptors. E.g. if an adaptor is 

   /// considered which differs from the wrapped graph only in some of its 

   /// functions or types, then it can be derived from BpUGraphAdaptor, and 

   /// only the differences should be implemented.

   ///

   /// \author Balazs Dezso 

   template<typename _BpUGraph>

   class BpUGraphAdaptorBase {

   public:

     typedef _BpUGraph Graph;

     typedef Graph ParentGraph;

   protected:

     Graph* graph;

     BpUGraphAdaptorBase() : graph(0) {}

     void setGraph(Graph& _graph) { graph = &_graph; }

     Graph& getGraph() { return *graph; }

     const Graph& getGraph() const { return *graph; }

   public:

     BpUGraphAdaptorBase(Graph& _graph) : graph(&_graph) {}

     typedef typename Graph::Node Node;

     typedef typename Graph::ANode ANode;

     typedef typename Graph::BNode BNode;

     typedef typename Graph::Edge Edge;

     typedef typename Graph::UEdge UEdge;

     void first(Node& i) const { graph->first(i); }

     void firstANode(Node& i) const { graph->firstANode(i); }

     void firstBNode(Node& i) const { graph->firstBNode(i); }

     void first(Edge& i) const { graph->first(i); }

     void first(UEdge& i) const { graph->first(i); }

     void firstIn(Edge& i, const Node& n) const { graph->firstIn(i, n); }

     void firstOut(Edge& i, const Node& n ) const { graph->firstOut(i, n); }

     void firstInc(UEdge &i, bool &d, const Node &n) const {

       graph->firstInc(i, d, n);

     }

     void next(Node& i) const { graph->next(i); }

     void nextANode(Node& i) const { graph->nextANode(i); }

     void nextBNode(Node& i) const { graph->nextBNode(i); }

     void next(Edge& i) const { graph->next(i); }

     void next(UEdge& i) const { graph->next(i); }

     void nextIn(Edge& i) const { graph->nextIn(i); }

     void nextOut(Edge& i) const { graph->nextOut(i); }

     void nextInc(UEdge &i, bool &d) const { graph->nextInc(i, d); }

     Node source(const UEdge& e) const { return graph->source(e); }

     Node target(const UEdge& e) const { return graph->target(e); }

     Node source(const Edge& e) const { return graph->source(e); }

     Node target(const Edge& e) const { return graph->target(e); }

     Node aNode(const UEdge& e) const { return graph->aNode(e); }

     Node bNode(const UEdge& e) const { return graph->bNode(e); }

     bool aNode(const Node& n) const { return graph->aNode(n); }

     bool bNode(const Node& n) const { return graph->bNode(n); }

     typedef NodeNumTagIndicator<Graph> NodeNumTag;

     int nodeNum() const { return graph->nodeNum(); }

     int aNodeNum() const { return graph->aNodeNum(); }

     int bNodeNum() const { return graph->bNodeNum(); }

     typedef EdgeNumTagIndicator<Graph> EdgeNumTag;

     int edgeNum() const { return graph->edgeNum(); }

     int uEdgeNum() const { return graph->uEdgeNum(); }

     typedef FindEdgeTagIndicator<Graph> FindEdgeTag;

     Edge findEdge(const Node& source, const Node& target, 

 		  const Edge& prev = INVALID) {

       return graph->findEdge(source, target, prev);

     }

     UEdge findUEdge(const Node& source, const Node& target, 

                     const UEdge& prev = INVALID) {

       return graph->findUEdge(source, target, prev);

     }

     Node addANode() const { return graph->addANode(); }

     Node addBNode() const { return graph->addBNode(); }

     UEdge addEdge(const Node& source, const Node& target) const { 

       return graph->addEdge(source, target); 

     }

     void erase(const Node& i) const { graph->erase(i); }

     void erase(const UEdge& i) const { graph->erase(i); }

     void clear() const { graph->clear(); }

     bool direction(const Edge& e) const { return graph->direction(e); }

     Edge direct(const UEdge& e, bool d) const { return graph->direct(e, d); }

     int id(const Node& v) const { return graph->id(v); }

     int id(const ANode& v) const { return graph->id(v); }

     int id(const BNode& v) const { return graph->id(v); }

     int id(const Edge& e) const { return graph->id(e); }

     int id(const UEdge& e) const { return graph->id(e); }

     Node fromNodeId(int id) const { return graph->fromNodeId(id); }

     ANode fromANodeId(int id) const { return graph->fromANodeId(id); }

     BNode fromBNodeId(int id) const { return graph->fromBNodeId(id); }

     Edge fromEdgeId(int id) const { return graph->fromEdgeId(id); }

     UEdge fromUEdgeId(int id) const { return graph->fromUEdgeId(id); }

     int maxNodeId() const { return graph->maxNodeId(); }

     int maxANodeId() const { return graph->maxANodeId(); }

     int maxBNodeId() const { return graph->maxBNodeId(); }

     int maxEdgeId() const { return graph->maxEdgeId(); }

     int maxUEdgeId() const { return graph->maxEdgeId(); }

     typedef typename ItemSetTraits<Graph, Node>::ItemNotifier NodeNotifier;

     NodeNotifier& getNotifier(Node) const {

       return graph->getNotifier(Node()); 

     } 

     typedef typename ItemSetTraits<Graph, ANode>::ItemNotifier ANodeNotifier;

     ANodeNotifier& getNotifier(ANode) const {

       return graph->getNotifier(ANode());

     } 

     typedef typename ItemSetTraits<Graph, BNode>::ItemNotifier BNodeNotifier;

     BNodeNotifier& getNotifier(BNode) const {

       return graph->getNotifier(BNode());

     } 

     typedef typename ItemSetTraits<Graph, Edge>::ItemNotifier EdgeNotifier;

     EdgeNotifier& getNotifier(Edge) const {

       return graph->getNotifier(Edge());

     } 

     typedef typename ItemSetTraits<Graph, UEdge>::ItemNotifier UEdgeNotifier;

     UEdgeNotifier& getNotifier(UEdge) const {

       return graph->getNotifier(UEdge());

     } 

     template <typename _Value>

     class NodeMap : public Graph::template NodeMap<_Value> {

     public:

       typedef typename Graph::template NodeMap<_Value> Parent;

       explicit NodeMap(const BpUGraphAdaptorBase<Graph>& ga) 

 	: Parent(*ga.graph) {}

       NodeMap(const BpUGraphAdaptorBase<Graph>& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       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 ANodeMap : public Graph::template ANodeMap<_Value> {

     public:

       typedef typename Graph::template ANodeMap<_Value> Parent;

       explicit ANodeMap(const BpUGraphAdaptorBase& ga) 

 	: Parent(*ga.graph) {}

       ANodeMap(const BpUGraphAdaptorBase& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       ANodeMap& operator=(const ANodeMap& cmap) {

 	return operator=<ANodeMap>(cmap);

       }

       template <typename CMap>

       ANodeMap& operator=(const CMap& cmap) {

         Parent::operator=(cmap);

 	return *this;

       }

     };

     template <typename _Value>

     class BNodeMap : public Graph::template BNodeMap<_Value> {

     public:

       typedef typename Graph::template BNodeMap<_Value> Parent;

       explicit BNodeMap(const BpUGraphAdaptorBase<Graph>& ga) 

 	: Parent(*ga.graph) {}

       BNodeMap(const BpUGraphAdaptorBase<Graph>& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       BNodeMap& operator=(const BNodeMap& cmap) {

 	return operator=<BNodeMap>(cmap);

       }

       template <typename CMap>

       BNodeMap& operator=(const CMap& cmap) {

         Parent::operator=(cmap);

 	return *this;

       }

     };

     template <typename _Value>

     class EdgeMap : public Graph::template EdgeMap<_Value> {

     public:

       typedef typename Graph::template EdgeMap<_Value> Parent;

       explicit EdgeMap(const BpUGraphAdaptorBase<Graph>& ga) 

 	: Parent(*ga.graph) {}

       EdgeMap(const BpUGraphAdaptorBase<Graph>& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       EdgeMap& operator=(const EdgeMap& cmap) {

 	return operator=<EdgeMap>(cmap);

       }

       template <typename CMap>

       EdgeMap& operator=(const CMap& cmap) {

         Parent::operator=(cmap);

 	return *this;

       }

     };

     template <typename _Value>

     class UEdgeMap : public Graph::template UEdgeMap<_Value> {

     public:

       typedef typename Graph::template UEdgeMap<_Value> Parent;

       explicit UEdgeMap(const BpUGraphAdaptorBase<Graph>& ga) 

 	: Parent(*ga.graph) {}

       UEdgeMap(const BpUGraphAdaptorBase<Graph>& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       UEdgeMap& operator=(const UEdgeMap& cmap) {

 	return operator=<UEdgeMap>(cmap);

       }

       template <typename CMap>

       UEdgeMap& operator=(const CMap& cmap) {

         Parent::operator=(cmap);

 	return *this;

       }

     };

   };

   /// \ingroup graph_adaptors

   ///

   /// \brief Trivial Bipartite Undirected Graph Adaptor

   ///

   /// Trivial Bipartite Undirected Graph Adaptor. It does not change

   /// the functionality of the adapted graph.

   template <typename _BpUGraph>

   class BpUGraphAdaptor 

     : public BpUGraphAdaptorExtender< BpUGraphAdaptorBase<_BpUGraph> > { 

   public:

     typedef _BpUGraph Graph;

     typedef BpUGraphAdaptorExtender<BpUGraphAdaptorBase<_BpUGraph> > Parent;

   protected:

     BpUGraphAdaptor() : Parent() {}

   public:

     explicit BpUGraphAdaptor(Graph& _graph) { setGraph(_graph); }

   };

   template <typename _BpUGraph>

   class SwapBpUGraphAdaptorBase : public BpUGraphAdaptorBase<_BpUGraph> {

   public:

     typedef _BpUGraph Graph;

     typedef BpUGraphAdaptorBase<_BpUGraph> Parent;

   protected:

     SwapBpUGraphAdaptorBase() {}

   public:

     typedef typename Parent::Node Node;

     typedef typename Parent::BNode ANode;

     typedef typename Parent::ANode BNode;

     typedef typename Parent::Edge Edge;

     typedef typename Parent::UEdge UEdge;

     bool direction(const Edge& e) const { return !Parent::direction(e); }

     Edge direct(const UEdge& e, bool b) const { return Parent::direct(e, !b); }

     Node aNode(const UEdge& e) const { return Parent::bNode(e); }

     Node bNode(const UEdge& e) const { return Parent::aNode(e); }

     bool aNode(const Node& n) const { return Parent::bNode(n); }

     bool bNode(const Node& n) const { return Parent::aNode(n); }

     void firstANode(Node& i) const { Parent::firstBNode(i); }

     void firstBNode(Node& i) const { Parent::firstANode(i); }

     void nextANode(Node& i) const { Parent::nextBNode(i); }

     void nextBNode(Node& i) const { Parent::nextANode(i); }

     int id(const ANode& v) const { return Parent::id(v); }

     int id(const BNode& v) const { return Parent::id(v); }

     ANode fromANodeId(int id) const { return Parent::fromBNodeId(id); }

     BNode fromBNodeId(int id) const { return Parent::fromANodeId(id); }

     int maxANodeId() const { return Parent::maxBNodeId(); }

     int maxBNodeId() const { return Parent::maxANodeId(); }

     int aNodeNum() const { return Parent::bNodeNum(); }

     int bNodeNum() const { return Parent::aNodeNum(); }

     typedef typename Parent::BNodeNotifier ANodeNotifier;

     ANodeNotifier& getNotifier(ANode) const {

       return Parent::getNotifier(typename Parent::BNode());

     } 

     typedef typename Parent::ANodeNotifier BNodeNotifier;

     BNodeNotifier& getNotifier(BNode) const {

       return Parent::getNotifier(typename Parent::ANode());

     } 

     template <typename _Value>

     class ANodeMap : public Graph::template BNodeMap<_Value> {

     public:

       typedef typename Graph::template BNodeMap<_Value> Parent;

       explicit ANodeMap(const SwapBpUGraphAdaptorBase& ga) 

 	: Parent(*ga.graph) {}

       ANodeMap(const SwapBpUGraphAdaptorBase& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       ANodeMap& operator=(const ANodeMap& cmap) {

 	return operator=<ANodeMap>(cmap);

       }

       template <typename CMap>

       ANodeMap& operator=(const CMap& cmap) {

         Parent::operator=(cmap);

 	return *this;

       }

     };

     template <typename _Value>

     class BNodeMap : public Graph::template ANodeMap<_Value> {

     public:

       typedef typename Graph::template ANodeMap<_Value> Parent;

       explicit BNodeMap(const SwapBpUGraphAdaptorBase<Graph>& ga) 

 	: Parent(*ga.graph) {}

       BNodeMap(const SwapBpUGraphAdaptorBase<Graph>& ga, const _Value& value)

 	: Parent(*ga.graph, value) {}

       BNodeMap& operator=(const BNodeMap& cmap) {

 	return operator=<BNodeMap>(cmap);

       }

       template <typename CMap>

       BNodeMap& operator=(const CMap& cmap) {

         Parent::operator=(cmap);

 	return *this;

       }

     };

   };

   /// \ingroup graph_adaptors

   ///

   /// \brief Bipartite graph adaptor which swaps the two nodeset.

   ///

   /// Bipartite graph adaptor which swaps the two nodeset. The adaptor's

   /// a-nodeset will be the original graph's b-nodeset and the adaptor's

   /// b-nodeset will be the original graph's a-nodeset. 

   template <typename _BpUGraph>

   class SwapBpUGraphAdaptor 

     : public BpUGraphAdaptorExtender<SwapBpUGraphAdaptorBase<_BpUGraph> > { 

   public:

     typedef _BpUGraph Graph;

     typedef BpUGraphAdaptorExtender<SwapBpUGraphAdaptorBase<_BpUGraph> > 

     Parent;

   protected:

     SwapBpUGraphAdaptor() : Parent() {}

   public:

     explicit SwapBpUGraphAdaptor(Graph& _graph) { setGraph(_graph); }

   };

   template <typename _BpUGraph, 

             typename _ANMatchingMap, typename _BNMatchingMap>

   class MatchingBpUGraphAdaptorBase 

     : public BpUGraphAdaptorBase<const _BpUGraph>

   {

   public:

     typedef _BpUGraph Graph;

     typedef _ANMatchingMap ANodeMatchingMap;

     typedef _BNMatchingMap BNodeMatchingMap;

     typedef BpUGraphAdaptorBase<const _BpUGraph> Parent;

   protected:

     MatchingBpUGraphAdaptorBase() {}

     void setANodeMatchingMap(ANodeMatchingMap& _anode_matching) {

       anode_matching = &_anode_matching;

     }

     void setBNodeMatchingMap(BNodeMatchingMap& _bnode_matching) {

       bnode_matching = &_bnode_matching;

     }

   public:

     typedef typename Parent::Node Node;

     typedef typename Parent::Edge Edge;

     void firstOut(Edge& edge, const Node& node) const {

       if (Parent::aNode(node)) {

         Parent::firstOut(edge, node);

         if (edge != INVALID && (*anode_matching)[node] == edge) {

           Parent::nextOut(edge);

         }

       } else {

         edge = (*bnode_matching)[node] != INVALID ?

           Parent::direct((*bnode_matching)[node], false) : INVALID;

       }

     }

     void nextOut(Edge& edge) const {

       if (Parent::aNode(Parent::source(edge))) {

         Parent::nextOut(edge);

         if (edge != INVALID && (*anode_matching)[Parent::aNode(edge)] == edge) {

           Parent::nextOut(edge);

         }

       } else {

         edge = INVALID;

       }

     }

     void firstIn(Edge& edge, const Node& node) const {

       if (Parent::aNode(node)) {

         edge = (*bnode_matching)[node] != INVALID ?

           Parent::direct((*anode_matching)[node], false) : INVALID;

       } else {

         Parent::firstIn(edge, node);

         if (edge != INVALID && (*bnode_matching)[node] == edge) {

           Parent::nextIn(edge);

         }

       }

     }

     void nextIn(Edge& edge) const {

       if (Parent::aNode(Parent::target(edge))) {

         edge = INVALID;

       } else {

         Parent::nextIn(edge);

         if (edge != INVALID && (*bnode_matching)[Parent::bNode(edge)] == edge) {

           Parent::nextIn(edge);

         }

       }

     } 

   private:

     ANodeMatchingMap* anode_matching;

     BNodeMatchingMap* bnode_matching;

   };

   /// \ingroup graph_adaptors

   ///

   /// \brief Bipartite graph adaptor to implement matching algorithms.

   ///

   /// Bipartite graph adaptor to implement matchings. It implements

   /// the residual graph of the matching.

   template <typename _BpUGraph, 

             typename _ANMatchingMap, typename _BNMatchingMap>

   class MatchingBpUGraphAdaptor 

     : public BpUGraphAdaptorExtender<

     MatchingBpUGraphAdaptorBase<_BpUGraph, _ANMatchingMap, _BNMatchingMap> > 

   { 

   public:

     typedef _BpUGraph BpUGraph;

     typedef _BpUGraph Graph;

     typedef _ANMatchingMap ANodeMatchingMap;

     typedef _BNMatchingMap BNodeMatchingMap;

     typedef BpUGraphAdaptorExtender<

       MatchingBpUGraphAdaptorBase<BpUGraph, 

                                   ANodeMatchingMap, BNodeMatchingMap> > 

     Parent;

   protected:

     MatchingBpUGraphAdaptor() : Parent() {}

   public:

     MatchingBpUGraphAdaptor(const Graph& _graph, 

                             ANodeMatchingMap& _anode_matching,

                             BNodeMatchingMap& _bnode_matching) {

       setGraph(_graph);

       setANodeMatchingMap(_anode_matching);

       setBNodeMatchingMap(_bnode_matching);

     }

   };

 }

 #endif