// -*- c++ -*-
#ifndef HUGO_DIMACS_H
#define HUGO_DIMACS_H

#include <iostream>
#include <string>
#include <vector>
#include <hugo/maps.h>

/// \ingroup misc
/// \file
/// \brief Dimacs file format reader.

namespace hugo {


  /// \addtogroup misc
  /// @{

  /// Dimacs min cost flow reader function.

  /// This function reads a min cost flow instance from dimacs format,
  /// i.e. from dimacs files having a line starting with \c p \c "min".
  /// At the beginning \c g is cleared by \c g.clear(). The edge
  /// capacities are written to \c capacity, \c s and \c t are set to
  /// the source and the target nodes resp. and the cost of the edges
  /// are written to \c cost.
  ///
  /// \author Marton Makai
  template<typename Graph, typename CapacityMap, typename CostMap>
  void readDimacs(std::istream& is, Graph &g, CapacityMap& capacity, 
		  typename Graph::Node &s, typename Graph::Node &t, 
		  CostMap& cost) {
    g.clear();
    typename CapacityMap::ValueType _cap;
    typename CostMap::ValueType _cost;
    char d;
    std::string problem;
    char c;
    int i, j;
    std::string str;
    int n, m; 
    typename Graph::Edge e;
    std::vector<typename Graph::Node> nodes;
    while (is>>c) {
      switch (c) {
      case 'c': //comment
	getline(is, str);
	break;
      case 'p': //problem definition
	is >> problem >> n >> m;
	getline(is, str);
	nodes.resize(n+1);
	for (int k=1; k<=n; ++k) nodes[k]=g.addNode();
	break;
      case 'n': //node definition
	if (problem=="sp") { //shortest path problem
	  is >> i;
	  getline(is, str);
	  s=nodes[i];
	}
	if (problem=="max" || problem=="min") { //((max) or (min cost)) flow problem
	  is >> i >> d;
	  getline(is, str);
	  if (d=='s') s=nodes[i];
	  if (d=='t') t=nodes[i];
	}
	break;
      case 'a':
	if ( problem == "max" || problem == "sp") {
	  is >> i >> j >> _cap;
	  getline(is, str);
	  e=g.addEdge(nodes[i], nodes[j]);
	  //capacity.update();
	  capacity.set(e, _cap);
	} else {
	  if ( problem == "min" ) {
	    is >> i >> j >> _cap >> _cost;
	    getline(is, str);
	    e=g.addEdge(nodes[i], nodes[j]);
	    //capacity.update();
	    capacity.set(e, _cap);
	    //cost.update();
	    cost.set(e, _cost);
	  } else {
	    is >> i >> j;
	    getline(is, str);
	    g.addEdge(nodes[i], nodes[j]);
	  }
	}
	break;
      }
    }
  }


  /// Dimacs max flow reader function.

  /// This function reads a max flow instance from dimacs format,
  /// i.e. from dimacs files having a line starting with \c p \c
  /// "max".  At the beginning \c g is cleared by \c g.clear(). The
  /// edge capacities are written to \c capacity and \c s and \c t are
  /// set to the source and the target nodes.
  ///
  /// \author Marton Makai
  template<typename Graph, typename CapacityMap>
  void readDimacs(std::istream& is, Graph &g, CapacityMap& capacity, 
		  typename Graph::Node &s, typename Graph::Node &t) {
    NullMap<typename Graph::Edge, int> n;
    readDimacs(is, g, capacity, s, t, n);
  }


  /// Dimacs shortest path reader function.

  /// This function reads a shortest path instance from dimacs format,
  /// i.e. from dimacs files having a line starting with \c p \c "sp".
  /// At the beginning \c g is cleared by \c g.clear(). The edge
  /// capacities are written to \c capacity and \c s is set to the
  /// source node.
  ///
  /// \author Marton Makai
  template<typename Graph, typename CapacityMap>
  void readDimacs(std::istream& is, Graph &g, CapacityMap& capacity, 
		  typename Graph::Node &s) {
    NullMap<typename Graph::Edge, int> n;
    readDimacs(is, g, capacity, s, s, n);
  }


  /// Dimacs capacitated graph reader function.

  /// This function reads an edge capacitated graph instance from
  /// dimacs format.  At the beginning \c g is cleared by \c g.clear()
  /// and the edge capacities are written to \c capacity.
  ///
  /// \author Marton Makai
  template<typename Graph, typename CapacityMap>
  void readDimacs(std::istream& is, Graph &g, CapacityMap& capacity) {
    typename Graph::Node u;
    NullMap<typename Graph::Edge, int> n;
    readDimacs(is, g, capacity, u, u, n);
  }


  /// Dimacs plain graph reader function.

  /// This function reads a graph without any designated nodes and
  /// maps from dimacs format, i.e. from dimacs files having a line
  /// starting with \c p \c "mat".  At the beginning \c g is cleared
  /// by \c g.clear().
  ///
  /// \author Marton Makai
  template<typename Graph>
  void readDimacs(std::istream& is, Graph &g) {
    typename Graph::Node u;
    NullMap<typename Graph::Edge, int> n;
    readDimacs(is, g, n, u, u, n);
  }
  


  
  /// write matching problem
  template<typename Graph>
  void writeDimacs(std::ostream& os, const Graph &g) {
    typedef typename Graph::NodeIt NodeIt;
    typedef typename Graph::EdgeIt EdgeIt;  
    
    typename Graph::template NodeMap<int> nodes(g);

    os << "c matching problem" << std::endl;

    int i=1;
    for(NodeIt v(g); v!=INVALID; ++v) {
      nodes.set(v, i);
      ++i;
    }    
    
    os << "p mat " << g.nodeNum() << " " << g.edgeNum() << std::endl;

    for(EdgeIt e(g); e!=INVALID; ++e) {
      os << "a " << nodes[g.tail(e)] << " " << nodes[g.head(e)] << std::endl; 
    }

  }


  /// @}

} //namespace hugo

#endif //HUGO_DIMACS_H

//  template<typename Graph, typename CapacityMap>
//   void readDimacsMaxFlow(std::istream& is, Graph &g, 
// 			 typename Graph::Node &s, typename Graph::Node &t, CapacityMap& capacity) {
//     g.clear();
//     int cap;
//     char d;
//     std::string problem;
//     char c;
//     int i, j;
//     std::string str;
//     int n, m; 
//     typename Graph::Edge e;
//     std::vector<typename Graph::Node> nodes;
//     while (is>>c) {
//       switch (c) {
//       case 'c': //comment
// 	getline(is, str);
// 	break;
//       case 'p': //problem definition
// 	is >> problem >> n >> m;
// 	getline(is, str);
// 	nodes.resize(n+1);
// 	for (int k=1; k<=n; ++k) nodes[k]=g.addNode();
// 	break;
//       case 'n': //node definition
// 	if (problem=="sp") { //shortest path problem
// 	  is >> i;
// 	  getline(is, str);
// 	  s=nodes[i];
// 	}
// 	if (problem=="max") { //max flow problem
// 	  is >> i >> d;
// 	  getline(is, str);
// 	  if (d=='s') s=nodes[i];
// 	  if (d=='t') t=nodes[i];
// 	}
// 	break;
//       case 'a':
// 	is >> i >> j >> cap;
// 	getline(is, str);
// 	e=g.addEdge(nodes[i], nodes[j]);
// 	capacity.update();
// 	capacity.set(e, cap);
// 	break;
//       }
//     }
//   }