/* -*- C++ -*-

  * src/lemon/graph_writer.h - Part of LEMON, a generic C++ optimization library

  *

  * Copyright (C) 2005 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.

  *

  */

 ///\ingroup io_group

 ///\file

 ///\brief Lemon Graph Format writer.

 #ifndef LEMON_GRAPH_WRITER_H

 #define LEMON_GRAPH_WRITER_H

 #include <iostream>

 #include <lemon/error.h>

 #include <lemon/lemon_writer.h>

 namespace lemon {

   /// \addtogroup io_group

   /// @{

   /// \brief The graph writer class.

   ///

   /// The \c GraphWriter class provides the graph output. To write a graph

   /// you should first give writing commands for the writer. You can declare

   /// write command as \c NodeMap or \c EdgeMap writing and labeled Node and

   /// Edge writing.

   ///

   /// \code

   /// GraphWriter<ListGraph> writer(std::cout, graph);

   /// \endcode

   ///

   /// The \c writeNodeMap() function declares a \c NodeMap writing 

   /// command in the \c GraphWriter. You should give as parameter 

   /// the name of the map and the map object. The NodeMap writing 

   /// command with name "id" should write a unique map because it 

   /// is regarded as ID map.

   ///

   /// \code

   /// IdMap<ListGraph, Node> nodeIdMap;

   /// writer.writeNodeMap("id", nodeIdMap);

   ///

   /// writer.writeNodeMap("coords", coords);

   /// writer.writeNodeMap("color", colorMap);

   /// \endcode

   ///

   /// With the \c writeEdgeMap() member function you can give an edge map

   /// writing command similar to the NodeMaps.

   ///

   /// \code

   /// DescriptorMap<ListGraph, Edge, ListGraph::EdgeMap<int> > 

   ///   edgeDescMap(graph);

   /// writer.writeEdgeMap("descriptor", edgeDescMap);

   ///

   /// writer.writeEdgeMap("weight", weightMap);

   /// writer.writeEdgeMap("label", labelMap);

   /// \endcode

   ///

   /// With \c writeNode() and \c writeEdge() functions you can 

   /// point out Nodes and Edges in the graph. By example, you can 

   /// write out the source and target of the graph.

   ///

   /// \code

   /// writer.writeNode("source", sourceNode);

   /// writer.writeNode("target", targetNode);

   ///

   /// writer.writeEdge("observed", edge);

   /// \endcode

   ///

   /// After you give all write commands you must call the \c run() member

   /// function, which execute all the writer commands.

   ///

   /// \code

   /// writer.run();

   /// \endcode

   ///

   /// \see DefaultWriterTraits

   /// \see QuotedStringWriter

   /// \see IdMap

   /// \see DescriptorMap

   /// \see \ref GraphReader

   /// \see \ref graph-io-page

   /// \author Balazs Dezso

   template <typename _Graph, typename _WriterTraits = DefaultWriterTraits> 

   class GraphWriter {

   public:

     typedef _Graph Graph;

     typedef typename Graph::Node Node;

     typedef typename Graph::Edge Edge;

     typedef _WriterTraits WriterTraits;

     /// \brief Construct a new GraphWriter.

     ///

     /// Construct a new GraphWriter. It writes the given graph

     /// to the given stream.

     GraphWriter(std::ostream& _os, const Graph& _graph) 

       : writer(new LemonWriter(_os)), own_writer(true), 

 	nodeset_writer(*writer, _graph, std::string()),

 	edgeset_writer(*writer, _graph, nodeset_writer, std::string()),

 	node_writer(*writer, nodeset_writer, std::string()),

 	edge_writer(*writer, edgeset_writer, std::string()),

 	attribute_writer(*writer, std::string()) {}

     /// \brief Construct a new GraphWriter.

     ///

     /// Construct a new GraphWriter. It writes into the given graph

     /// to the given file.

     GraphWriter(const std::string& _filename, const Graph& _graph) 

       : writer(new LemonWriter(_filename)), own_writer(true), 

 	nodeset_writer(*writer, _graph, std::string()),

 	edgeset_writer(*writer, _graph, nodeset_writer, std::string()),

 	node_writer(*writer, nodeset_writer, std::string()),

 	edge_writer(*writer, edgeset_writer, std::string()),

 	attribute_writer(*writer, std::string()) {}

     /// \brief Construct a new GraphWriter.

     ///

     /// Construct a new GraphWriter. It writes into the given graph

     /// to given LemonReader.

     GraphWriter(LemonWriter& _writer, const Graph& _graph)

       : writer(_writer), own_writer(false), 

 	nodeset_writer(*writer, _graph, std::string()),

 	edgeset_writer(*writer, _graph, nodeset_writer, std::string()),

 	node_writer(*writer, nodeset_writer, std::string()),

 	edge_writer(*writer, edgeset_writer, std::string()),

 	attribute_writer(*writer, std::string()) {}

     /// \brief Destruct the graph writer.

     ///

     /// Destruct the graph writer.

     ~GraphWriter() {

       if (own_writer) 

 	delete writer;

     }

     /// \brief Add a new node map writer command for the writer.

     ///

     /// Add a new node map writer command for the writer.

     template <typename Map>

     GraphWriter& writeNodeMap(std::string name, const Map& map) {

       nodeset_writer.writeNodeMap(name, map);

       return *this;

     }

     /// \brief Add a new node map writer command for the writer.

     ///

     /// Add a new node map writer command for the writer.

     template <typename Writer, typename Map>

     GraphWriter& writeNodeMap(std::string name, const Map& map, 

 			      const Writer& writer = Writer()) {

       nodeset_writer.writeNodeMap(name, map, writer);

       return *this;

     }

     /// \brief Add a new edge map writer command for the writer.

     ///

     /// Add a new edge map writer command for the writer.

     template <typename Map>

     GraphWriter& writeEdgeMap(std::string name, const Map& map) { 

       edgeset_writer.writeEdgeMap(name, map);

       return *this;

     }

     /// \brief Add a new edge map writer command for the writer.

     ///

     /// Add a new edge map writer command for the writer.

     template <typename Writer, typename Map>

     GraphWriter& writeEdgeMap(std::string name, const Map& map,

 			      const Writer& writer = Writer()) {

       edgeset_writer.writeEdgeMap(name, map, writer);

       return *this;

     }

     /// \brief Add a new labeled node writer for the writer.

     ///

     /// Add a new labeled node writer for the writer.

     GraphWriter& writeNode(std::string name, const Node& node) {

       node_writer.writeNode(name, node);

       return *this;

     }

     /// \brief Add a new labeled edge writer for the writer.

     ///

     /// Add a new labeled edge writer for the writer.

     GraphWriter& writeEdge(std::string name, const Edge& edge) {

       edge_writer.writeEdge(name, edge);

     }

     /// \brief Add a new attribute writer command.

     ///

     ///  Add a new attribute writer command.

     template <typename Value>

     GraphWriter& writeAttribute(std::string name, const Value& value) {

       attribute_writer.writeAttribute(name, value);

       return *this;

     }

     /// \brief Add a new attribute writer command.

     ///

     ///  Add a new attribute writer command.

     template <typename Writer, typename Value>

     GraphWriter& writeAttribute(std::string name, const Value& value, 

 			       const Writer& writer) {

       attribute_writer.writeAttribute<Writer>(name, value, writer);

       return *this;

     }

     /// \brief Conversion operator to LemonWriter.

     ///

     /// Conversion operator to LemonWriter. It make possible

     /// to access the encapsulated \e LemonWriter, this way

     /// you can attach to this writer new instances of 

     /// \e LemonWriter::SectionWriter.

     operator LemonWriter&() {

       return *writer;

     }

     /// \brief Executes the writer commands.

     ///

     /// Executes the writer commands.

     void run() {

       writer->run();

     }

     /// \brief Write the id of the given node.

     ///

     /// It writes the id of the given node. If there was written an "id"

     /// named node map then it will write the map value belongs to the node.

     void writeId(std::ostream& os, const Node& item) const {

       nodeset_writer.writeId(os, item);

     } 

     /// \brief Write the id of the given edge.

     ///

     /// It writes the id of the given edge. If there was written an "id"

     /// named edge map then it will write the map value belongs to the edge.

     void writeId(std::ostream& os, const Edge& item) const {

       edgeset_writer.writeId(os, item);

     } 

   private:

     LemonWriter* writer;

     bool own_writer;

     NodeSetWriter<Graph, WriterTraits> nodeset_writer;

     EdgeSetWriter<Graph, WriterTraits> edgeset_writer;

     NodeWriter<Graph> node_writer;

     EdgeWriter<Graph> edge_writer;

     AttributeWriter<WriterTraits> attribute_writer;

   };

   /// \brief Write a graph to the output.

   ///

   /// Write a graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   /// \param capacity The capacity map.

   /// \param s The source node.

   /// \param t The target node.

   /// \param cost The cost map.

   template<typename Graph, typename CapacityMap, typename CostMap>

   void writeGraph(std::ostream& os, const Graph &g, 

 		  const CapacityMap& capacity, const typename Graph::Node &s,

 		  const typename Graph::Node &t, const CostMap& cost) {

     GraphWriter<Graph> writer(os, g);

     IdMap<Graph, typename Graph::Node> nodeIdMap(g);

     writer.writeNodeMap("id", nodeIdMap);

     IdMap<Graph, typename Graph::Edge> edgeIdMap(g);

     writer.writeEdgeMap("id", edgeIdMap);

     writer.writeEdgeMap("capacity", capacity);

     writer.writeEdgeMap("cost", cost);

     writer.writeNode("source", s);

     writer.writeNode("target", t);

     writer.run();

   }

   /// \brief Write a graph to the output.

   ///

   /// Write a graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   /// \param capacity The capacity map.

   /// \param s The source node.

   /// \param t The target node.

   template<typename Graph, typename CapacityMap>

   void writeGraph(std::ostream& os, const Graph &g, 

 		  const CapacityMap& capacity, const typename Graph::Node &s,

 		  const typename Graph::Node &t) {

     GraphWriter<Graph> writer(os, g);

     IdMap<Graph, typename Graph::Node> nodeIdMap(g);

     writer.writeNodeMap("id", nodeIdMap);

     IdMap<Graph, typename Graph::Edge> edgeIdMap(g);

     writer.writeEdgeMap("id", edgeIdMap);

     writer.writeEdgeMap("capacity", capacity);

     writer.writeNode("source", s);

     writer.writeNode("target", t);

     writer.run();

   }

   /// \brief Write a graph to the output.

   ///

   /// Write a graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   /// \param capacity The capacity map.

   /// \param s The source node.

   template<typename Graph, typename CapacityMap>

   void writeGraph(std::ostream& os, const Graph &g, 

 		  const CapacityMap& capacity, const typename Graph::Node &s) {

     GraphWriter<Graph> writer(os, g);

     IdMap<Graph, typename Graph::Node> nodeIdMap(g);

     writer.writeNodeMap("id", nodeIdMap);

     IdMap<Graph, typename Graph::Edge> edgeIdMap(g);

     writer.writeEdgeMap("id", edgeIdMap);

     writer.writeEdgeMap("capacity", capacity);

     writer.writeNode("source", s);

     writer.run();

   }

   /// \brief Write a graph to the output.

   ///

   /// Write a graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   /// \param capacity The capacity map.

   template<typename Graph, typename CapacityMap>

   void writeGraph(std::ostream& os, const Graph &g, 

 		  const CapacityMap& capacity) {

     GraphWriter<Graph> writer(os, g);

     IdMap<Graph, typename Graph::Node> nodeIdMap(g);

     writer.writeNodeMap("id", nodeIdMap);

     IdMap<Graph, typename Graph::Edge> edgeIdMap(g);

     writer.writeEdgeMap("id", edgeIdMap);

     writer.writeEdgeMap("capacity", capacity);

     writer.run();

   }

   /// \brief Write a graph to the output.

   ///

   /// Write a graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   template<typename Graph>

   void writeGraph(std::ostream& os, const Graph &g) {

     GraphWriter<Graph> writer(os, g);

     IdMap<Graph, typename Graph::Node> nodeIdMap(g);

     writer.writeNodeMap("id", nodeIdMap);

     IdMap<Graph, typename Graph::Edge> edgeIdMap(g);

     writer.writeEdgeMap("id", edgeIdMap);

     writer.run();

   }

   /// \brief The undirected graph writer class.

   ///

   /// The \c UndirGraphWriter class provides the undir graph output. To write 

   /// a graph you should first give writing commands for the writer. You can 

   /// declare write command as \c NodeMap, \c EdgeMap or \c UndirEdgeMap 

   /// writing and labeled Node, Edge or UndirEdge writing.

   ///

   /// \code

   /// UndirGraphWriter<UndirListGraph> writer(std::cout, graph);

   /// \endcode

   ///

   /// The \c writeNodeMap() function declares a \c NodeMap writing 

   /// command in the \c UndirGraphWriter. You should give as parameter 

   /// the name of the map and the map object. The NodeMap writing 

   /// command with name "id" should write a unique map because it 

   /// is regarded as ID map.

   ///

   /// \code

   /// IdMap<UndirListGraph, Node> nodeIdMap;

   /// writer.writeNodeMap("id", nodeIdMap);

   ///

   /// writer.writeNodeMap("coords", coords);

   /// writer.writeNodeMap("color", colorMap);

   /// \endcode

   ///

   /// With the \c writeUndirEdgeMap() member function you can give an 

   /// undirected edge map writing command similar to the NodeMaps.

   ///

   /// \code

   /// DescriptorMap<ListGraph, Edge, ListGraph::EdgeMap<int> > 

   ///   edgeDescMap(graph);

   /// writer.writeUndirEdgeMap("descriptor", edgeDescMap);

   ///

   /// writer.writeUndirEdgeMap("weight", weightMap);

   /// writer.writeUndirEdgeMap("label", labelMap);

   /// \endcode

   /// 

   /// The EdgeMap handling is just a syntactical sugar. It writes

   /// two undirected edge map with '+' and '-' prefix in the name.

   ///

   /// \code

   /// writer.writeEdgeMap("capacity", capacityMap);

   /// \endcode

   ///

   ///

   /// With \c writeNode() and \c writeUndirEdge() functions you can 

   /// point out nodes and undirected edges in the graph. By example, you can 

   /// write out the source and target of the graph.

   ///

   /// \code

   /// writer.writeNode("source", sourceNode);

   /// writer.writeNode("target", targetNode);

   ///

   /// writer.writeUndirEdge("observed", undirEdge);

   /// \endcode

   ///

   /// After you give all write commands you must call the \c run() member

   /// function, which execute all the writer commands.

   ///

   /// \code

   /// writer.run();

   /// \endcode

   ///

   /// \see DefaultWriterTraits

   /// \see QuotedStringWriter

   /// \see IdMap

   /// \see DescriptorMap

   /// \see \ref GraphWriter

   /// \see \ref graph-io-page

   /// \author Balazs Dezso

   template <typename _Graph, typename _WriterTraits = DefaultWriterTraits> 

   class UndirGraphWriter {

   public:

     typedef _Graph Graph;

     typedef typename Graph::Node Node;

     typedef typename Graph::Edge Edge;

     typedef typename Graph::UndirEdge UndirEdge;

     typedef _WriterTraits WriterTraits;

     /// \brief Construct a new UndirGraphWriter.

     ///

     /// Construct a new UndirGraphWriter. It writes the given graph

     /// to the given stream.

     UndirGraphWriter(std::ostream& _os, const Graph& _graph) 

       : writer(new LemonWriter(_os)), own_writer(true), 

 	nodeset_writer(*writer, _graph, std::string()),

 	undir_edgeset_writer(*writer, _graph, nodeset_writer, std::string()),

 	node_writer(*writer, nodeset_writer, std::string()),

 	undir_edge_writer(*writer, undir_edgeset_writer, std::string()),

 	attribute_writer(*writer, std::string()) {}

     /// \brief Construct a new UndirGraphWriter.

     ///

     /// Construct a new UndirGraphWriter. It writes into the given graph

     /// to the given file.

     UndirGraphWriter(const std::string& _filename, const Graph& _graph) 

       : writer(new LemonWriter(_filename)), own_writer(true), 

 	nodeset_writer(*writer, _graph, std::string()),

 	undir_edgeset_writer(*writer, _graph, nodeset_writer, std::string()),

 	node_writer(*writer, nodeset_writer, std::string()),

 	undir_edge_writer(*writer, undir_edgeset_writer, std::string()),

 	attribute_writer(*writer, std::string()) {}

     /// \brief Construct a new UndirGraphWriter.

     ///

     /// Construct a new UndirGraphWriter. It writes into the given graph

     /// to given LemonReader.

     UndirGraphWriter(LemonWriter& _writer, const Graph& _graph)

       : writer(_writer), own_writer(false), 

 	nodeset_writer(*writer, _graph, std::string()),

 	undir_edgeset_writer(*writer, _graph, nodeset_writer, std::string()),

 	node_writer(*writer, nodeset_writer, std::string()),

 	undir_edge_writer(*writer, undir_edgeset_writer, std::string()),

 	attribute_writer(*writer, std::string()) {}

     /// \brief Destruct the graph writer.

     ///

     /// Destruct the graph writer.

     ~UndirGraphWriter() {

       if (own_writer) 

 	delete writer;

     }

     /// \brief Add a new node map writer command for the writer.

     ///

     /// Add a new node map writer command for the writer.

     template <typename Map>

     UndirGraphWriter& writeNodeMap(std::string name, const Map& map) {

       nodeset_writer.writeNodeMap(name, map);

       return *this;

     }

     /// \brief Add a new node map writer command for the writer.

     ///

     /// Add a new node map writer command for the writer.

     template <typename Writer, typename Map>

     UndirGraphWriter& writeNodeMap(std::string name, const Map& map, 

 			      const Writer& writer = Writer()) {

       nodeset_writer.writeNodeMap(name, map, writer);

       return *this;

     }

     /// \brief Add a new edge map writer command for the writer.

     ///

     /// Add a new edge map writer command for the writer.

     template <typename Map>

     UndirGraphWriter& writeEdgeMap(std::string name, const Map& map) { 

       undir_edgeset_writer.writeEdgeMap(name, map);

       return *this;

     }

     /// \brief Add a new edge map writer command for the writer.

     ///

     /// Add a new edge map writer command for the writer.

     template <typename Writer, typename Map>

     UndirGraphWriter& writeEdgeMap(std::string name, const Map& map,

 				   const Writer& writer = Writer()) {

       undir_edgeset_writer.writeEdgeMap(name, map, writer);

       return *this;

     }

     /// \brief Add a new undirected edge map writer command for the writer.

     ///

     /// Add a new undirected edge map writer command for the writer.

     template <typename Map>

     UndirGraphWriter& writeUndirEdgeMap(std::string name, const Map& map) { 

       undir_edgeset_writer.writeUndirEdgeMap(name, map);

       return *this;

     }

     /// \brief Add a new undirected edge map writer command for the writer.

     ///

     /// Add a new edge undirected map writer command for the writer.

     template <typename Writer, typename Map>

     UndirGraphWriter& writeUndirEdgeMap(std::string name, const Map& map,

 					const Writer& writer = Writer()) {

       undir_edgeset_writer.writeUndirEdgeMap(name, map, writer);

       return *this;

     }

     /// \brief Add a new labeled node writer for the writer.

     ///

     /// Add a new labeled node writer for the writer.

     UndirGraphWriter& writeNode(std::string name, const Node& node) {

       node_writer.writeNode(name, node);

       return *this;

     }

     /// \brief Add a new labeled edge writer for the writer.

     ///

     /// Add a new labeled edge writer for the writer.

     UndirGraphWriter& writeEdge(std::string name, const Edge& edge) {

       undir_edge_writer.writeEdge(name, edge);

     }

     /// \brief Add a new labeled undirected edge writer for the writer.

     ///

     /// Add a new labeled undirected edge writer for the writer.

     UndirGraphWriter& writeUndirEdge(std::string name, const UndirEdge& edge) {

       undir_edge_writer.writeUndirEdge(name, edge);

     }

     /// \brief Add a new attribute writer command.

     ///

     ///  Add a new attribute writer command.

     template <typename Value>

     UndirGraphWriter& writeAttribute(std::string name, const Value& value) {

       attribute_writer.writeAttribute(name, value);

       return *this;

     }

     /// \brief Add a new attribute writer command.

     ///

     ///  Add a new attribute writer command.

     template <typename Writer, typename Value>

     UndirGraphWriter& writeAttribute(std::string name, const Value& value, 

 			       const Writer& writer) {

       attribute_writer.writeAttribute<Writer>(name, value, writer);

       return *this;

     }

     /// \brief Conversion operator to LemonWriter.

     ///

     /// Conversion operator to LemonWriter. It make possible

     /// to access the encapsulated \e LemonWriter, this way

     /// you can attach to this writer new instances of 

     /// \e LemonWriter::SectionWriter.

     operator LemonWriter&() {

       return *writer;

     }

     /// \brief Executes the writer commands.

     ///

     /// Executes the writer commands.

     void run() {

       writer->run();

     }

     /// \brief Write the id of the given node.

     ///

     /// It writes the id of the given node. If there was written an "id"

     /// named node map then it will write the map value belongs to the node.

     void writeId(std::ostream& os, const Node& item) const {

       nodeset_writer.writeId(os, item);

     } 

     /// \brief Write the id of the given edge.

     ///

     /// It writes the id of the given edge. If there was written an "id"

     /// named edge map then it will write the map value belongs to the edge.

     void writeId(std::ostream& os, const Edge& item) const {

       undir_edgeset_writer.writeId(os, item);

     } 

     /// \brief Write the id of the given undirected edge.

     ///

     /// It writes the id of the given undirected edge. If there was written 

     /// an "id" named edge map then it will write the map value belongs to 

     /// the edge.

     void writeId(std::ostream& os, const UndirEdge& item) const {

       undir_edgeset_writer.writeId(os, item);

     } 

   private:

     LemonWriter* writer;

     bool own_writer;

     NodeSetWriter<Graph, WriterTraits> nodeset_writer;

     UndirEdgeSetWriter<Graph, WriterTraits> undir_edgeset_writer;

     NodeWriter<Graph> node_writer;

     UndirEdgeWriter<Graph> undir_edge_writer;

     AttributeWriter<WriterTraits> attribute_writer;

   };

   /// \brief Write an undirected graph to the output.

   ///

   /// Write an undirected graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   /// \param capacity The capacity undirected map.

   template<typename Graph, typename CapacityMap>

   void writeUndirGraph(std::ostream& os, const Graph &g, 

 		       const CapacityMap& capacity) {

     UndirGraphWriter<Graph> writer(os, g);

     writer.writeUndirEdgeMap("capacity", capacity);

     writer.run();

   }

   /// \brief Write an undirected graph to the output.

   ///

   /// Write an undirected graph to the output.

   /// \param os The output stream.

   /// \param g The graph.

   template<typename Graph>

   void writeUndirGraph(std::ostream& os, const Graph &g) {

     UndirGraphWriter<Graph> writer(os, g);

     writer.run();

   }

   /// @}

 }

 #endif