lemon/lgf_writer.h
author Balazs Dezso <deba@inf.elte.hu>
Tue, 30 Sep 2008 20:53:18 +0200
changeset 290 f6899946c1ac
parent 248 8fada33fc60a
child 291 d901321d6555
permissions -rw-r--r--
Simplifying exceptions

- Using asserts instead of exceptions for unitialized parameters
- Only the IO exceptions are used in the lemon
- DataFormatError is renamed to FormatError
- The IoError is used for file access errors
     1 /* -*- mode: C++; indent-tabs-mode: nil; -*-
     2  *
     3  * This file is a part of LEMON, a generic C++ optimization library.
     4  *
     5  * Copyright (C) 2003-2008
     6  * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
     7  * (Egervary Research Group on Combinatorial Optimization, EGRES).
     8  *
     9  * Permission to use, modify and distribute this software is granted
    10  * provided that this copyright notice appears in all copies. For
    11  * precise terms see the accompanying LICENSE file.
    12  *
    13  * This software is provided "AS IS" with no warranty of any kind,
    14  * express or implied, and with no claim as to its suitability for any
    15  * purpose.
    16  *
    17  */
    18 
    19 ///\ingroup lemon_io
    20 ///\file
    21 ///\brief \ref lgf-format "LEMON Graph Format" writer.
    22 
    23 
    24 #ifndef LEMON_LGF_WRITER_H
    25 #define LEMON_LGF_WRITER_H
    26 
    27 #include <iostream>
    28 #include <fstream>
    29 #include <sstream>
    30 
    31 #include <algorithm>
    32 
    33 #include <vector>
    34 #include <functional>
    35 
    36 #include <lemon/assert.h>
    37 #include <lemon/core.h>
    38 #include <lemon/maps.h>
    39 
    40 #include <lemon/concept_check.h>
    41 #include <lemon/concepts/maps.h>
    42 
    43 namespace lemon {
    44 
    45   namespace _writer_bits {
    46 
    47     template <typename Value>
    48     struct DefaultConverter {
    49       std::string operator()(const Value& value) {
    50         std::ostringstream os;
    51         os << value;
    52         return os.str();
    53       }
    54     };
    55 
    56     template <typename T>
    57     bool operator<(const T&, const T&) {
    58       throw FormatError("Label map is not comparable");
    59     }
    60 
    61     template <typename _Map>
    62     class MapLess {
    63     public:
    64       typedef _Map Map;
    65       typedef typename Map::Key Item;
    66 
    67     private:
    68       const Map& _map;
    69 
    70     public:
    71       MapLess(const Map& map) : _map(map) {}
    72 
    73       bool operator()(const Item& left, const Item& right) {
    74         return _map[left] < _map[right];
    75       }
    76     };
    77 
    78     template <typename _Graph, bool _dir, typename _Map>
    79     class GraphArcMapLess {
    80     public:
    81       typedef _Map Map;
    82       typedef _Graph Graph;
    83       typedef typename Graph::Edge Item;
    84 
    85     private:
    86       const Graph& _graph;
    87       const Map& _map;
    88 
    89     public:
    90       GraphArcMapLess(const Graph& graph, const Map& map)
    91         : _graph(graph), _map(map) {}
    92 
    93       bool operator()(const Item& left, const Item& right) {
    94         return _map[_graph.direct(left, _dir)] <
    95           _map[_graph.direct(right, _dir)];
    96       }
    97     };
    98 
    99     template <typename _Item>
   100     class MapStorageBase {
   101     public:
   102       typedef _Item Item;
   103 
   104     public:
   105       MapStorageBase() {}
   106       virtual ~MapStorageBase() {}
   107 
   108       virtual std::string get(const Item& item) = 0;
   109       virtual void sort(std::vector<Item>&) = 0;
   110     };
   111 
   112     template <typename _Item, typename _Map,
   113               typename _Converter = DefaultConverter<typename _Map::Value> >
   114     class MapStorage : public MapStorageBase<_Item> {
   115     public:
   116       typedef _Map Map;
   117       typedef _Converter Converter;
   118       typedef _Item Item;
   119 
   120     private:
   121       const Map& _map;
   122       Converter _converter;
   123 
   124     public:
   125       MapStorage(const Map& map, const Converter& converter = Converter())
   126         : _map(map), _converter(converter) {}
   127       virtual ~MapStorage() {}
   128 
   129       virtual std::string get(const Item& item) {
   130         return _converter(_map[item]);
   131       }
   132       virtual void sort(std::vector<Item>& items) {
   133         MapLess<Map> less(_map);
   134         std::sort(items.begin(), items.end(), less);
   135       }
   136     };
   137 
   138     template <typename _Graph, bool _dir, typename _Map,
   139               typename _Converter = DefaultConverter<typename _Map::Value> >
   140     class GraphArcMapStorage : public MapStorageBase<typename _Graph::Edge> {
   141     public:
   142       typedef _Map Map;
   143       typedef _Converter Converter;
   144       typedef _Graph Graph;
   145       typedef typename Graph::Edge Item;
   146       static const bool dir = _dir;
   147 
   148     private:
   149       const Graph& _graph;
   150       const Map& _map;
   151       Converter _converter;
   152 
   153     public:
   154       GraphArcMapStorage(const Graph& graph, const Map& map,
   155                          const Converter& converter = Converter())
   156         : _graph(graph), _map(map), _converter(converter) {}
   157       virtual ~GraphArcMapStorage() {}
   158 
   159       virtual std::string get(const Item& item) {
   160         return _converter(_map[_graph.direct(item, dir)]);
   161       }
   162       virtual void sort(std::vector<Item>& items) {
   163         GraphArcMapLess<Graph, dir, Map> less(_graph, _map);
   164         std::sort(items.begin(), items.end(), less);
   165       }
   166     };
   167 
   168     class ValueStorageBase {
   169     public:
   170       ValueStorageBase() {}
   171       virtual ~ValueStorageBase() {}
   172 
   173       virtual std::string get() = 0;
   174     };
   175 
   176     template <typename _Value, typename _Converter = DefaultConverter<_Value> >
   177     class ValueStorage : public ValueStorageBase {
   178     public:
   179       typedef _Value Value;
   180       typedef _Converter Converter;
   181 
   182     private:
   183       const Value& _value;
   184       Converter _converter;
   185 
   186     public:
   187       ValueStorage(const Value& value, const Converter& converter = Converter())
   188         : _value(value), _converter(converter) {}
   189 
   190       virtual std::string get() {
   191         return _converter(_value);
   192       }
   193     };
   194 
   195     template <typename Value>
   196     struct MapLookUpConverter {
   197       const std::map<Value, std::string>& _map;
   198 
   199       MapLookUpConverter(const std::map<Value, std::string>& map)
   200         : _map(map) {}
   201 
   202       std::string operator()(const Value& str) {
   203         typename std::map<Value, std::string>::const_iterator it =
   204           _map.find(str);
   205         if (it == _map.end()) {
   206           throw FormatError("Item not found");
   207         }
   208         return it->second;
   209       }
   210     };
   211 
   212     template <typename Graph>
   213     struct GraphArcLookUpConverter {
   214       const Graph& _graph;
   215       const std::map<typename Graph::Edge, std::string>& _map;
   216 
   217       GraphArcLookUpConverter(const Graph& graph,
   218                               const std::map<typename Graph::Edge,
   219                                              std::string>& map)
   220         : _graph(graph), _map(map) {}
   221 
   222       std::string operator()(const typename Graph::Arc& val) {
   223         typename std::map<typename Graph::Edge, std::string>
   224           ::const_iterator it = _map.find(val);
   225         if (it == _map.end()) {
   226           throw FormatError("Item not found");
   227         }
   228         return (_graph.direction(val) ? '+' : '-') + it->second;
   229       }
   230     };
   231 
   232     inline bool isWhiteSpace(char c) {
   233       return c == ' ' || c == '\t' || c == '\v' ||
   234         c == '\n' || c == '\r' || c == '\f';
   235     }
   236 
   237     inline bool isEscaped(char c) {
   238       return c == '\\' || c == '\"' || c == '\'' ||
   239         c == '\a' || c == '\b';
   240     }
   241 
   242     inline static void writeEscape(std::ostream& os, char c) {
   243       switch (c) {
   244       case '\\':
   245         os << "\\\\";
   246         return;
   247       case '\"':
   248         os << "\\\"";
   249         return;
   250       case '\a':
   251         os << "\\a";
   252         return;
   253       case '\b':
   254         os << "\\b";
   255         return;
   256       case '\f':
   257         os << "\\f";
   258         return;
   259       case '\r':
   260         os << "\\r";
   261         return;
   262       case '\n':
   263         os << "\\n";
   264         return;
   265       case '\t':
   266         os << "\\t";
   267         return;
   268       case '\v':
   269         os << "\\v";
   270         return;
   271       default:
   272         if (c < 0x20) {
   273           std::ios::fmtflags flags = os.flags();
   274           os << '\\' << std::oct << static_cast<int>(c);
   275           os.flags(flags);
   276         } else {
   277           os << c;
   278         }
   279         return;
   280       }
   281     }
   282 
   283     inline bool requireEscape(const std::string& str) {
   284       if (str.empty() || str[0] == '@') return true;
   285       std::istringstream is(str);
   286       char c;
   287       while (is.get(c)) {
   288         if (isWhiteSpace(c) || isEscaped(c)) {
   289           return true;
   290         }
   291       }
   292       return false;
   293     }
   294 
   295     inline std::ostream& writeToken(std::ostream& os, const std::string& str) {
   296 
   297       if (requireEscape(str)) {
   298         os << '\"';
   299         for (std::string::const_iterator it = str.begin();
   300              it != str.end(); ++it) {
   301           writeEscape(os, *it);
   302         }
   303         os << '\"';
   304       } else {
   305         os << str;
   306       }
   307       return os;
   308     }
   309 
   310     class Section {
   311     public:
   312       virtual ~Section() {}
   313       virtual void process(std::ostream& os) = 0;
   314     };
   315 
   316     template <typename Functor>
   317     class LineSection : public Section {
   318     private:
   319 
   320       Functor _functor;
   321 
   322     public:
   323 
   324       LineSection(const Functor& functor) : _functor(functor) {}
   325       virtual ~LineSection() {}
   326 
   327       virtual void process(std::ostream& os) {
   328         std::string line;
   329         while (!(line = _functor()).empty()) os << line << std::endl;
   330       }
   331     };
   332 
   333     template <typename Functor>
   334     class StreamSection : public Section {
   335     private:
   336 
   337       Functor _functor;
   338 
   339     public:
   340 
   341       StreamSection(const Functor& functor) : _functor(functor) {}
   342       virtual ~StreamSection() {}
   343 
   344       virtual void process(std::ostream& os) {
   345         _functor(os);
   346       }
   347     };
   348 
   349   }
   350 
   351   template <typename Digraph>
   352   class DigraphWriter;
   353 
   354   template <typename Digraph>
   355   DigraphWriter<Digraph> digraphWriter(std::ostream& os,
   356                                        const Digraph& digraph);
   357 
   358   template <typename Digraph>
   359   DigraphWriter<Digraph> digraphWriter(const std::string& fn,
   360                                        const Digraph& digraph);
   361 
   362   template <typename Digraph>
   363   DigraphWriter<Digraph> digraphWriter(const char *fn,
   364                                        const Digraph& digraph);
   365 
   366   /// \ingroup lemon_io
   367   ///
   368   /// \brief \ref lgf-format "LGF" writer for directed graphs
   369   ///
   370   /// This utility writes an \ref lgf-format "LGF" file.
   371   ///
   372   /// The writing method does a batch processing. The user creates a
   373   /// writer object, then various writing rules can be added to the
   374   /// writer, and eventually the writing is executed with the \c run()
   375   /// member function. A map writing rule can be added to the writer
   376   /// with the \c nodeMap() or \c arcMap() members. An optional
   377   /// converter parameter can also be added as a standard functor
   378   /// converting from the value type of the map to \c std::string. If it
   379   /// is set, it will determine how the value type of the map is written to
   380   /// the output stream. If the functor is not set, then a default
   381   /// conversion will be used. The \c attribute(), \c node() and \c
   382   /// arc() functions are used to add attribute writing rules.
   383   ///
   384   ///\code
   385   /// DigraphWriter<Digraph>(std::cout, digraph).
   386   ///   nodeMap("coordinates", coord_map).
   387   ///   nodeMap("size", size).
   388   ///   nodeMap("title", title).
   389   ///   arcMap("capacity", cap_map).
   390   ///   node("source", src).
   391   ///   node("target", trg).
   392   ///   attribute("caption", caption).
   393   ///   run();
   394   ///\endcode
   395   ///
   396   ///
   397   /// By default, the writer does not write additional captions to the
   398   /// sections, but they can be give as an optional parameter of
   399   /// the \c nodes(), \c arcs() or \c
   400   /// attributes() functions.
   401   ///
   402   /// The \c skipNodes() and \c skipArcs() functions forbid the
   403   /// writing of the sections. If two arc sections should be written
   404   /// to the output, it can be done in two passes, the first pass
   405   /// writes the node section and the first arc section, then the
   406   /// second pass skips the node section and writes just the arc
   407   /// section to the stream. The output stream can be retrieved with
   408   /// the \c ostream() function, hence the second pass can append its
   409   /// output to the output of the first pass.
   410   template <typename _Digraph>
   411   class DigraphWriter {
   412   public:
   413 
   414     typedef _Digraph Digraph;
   415     TEMPLATE_DIGRAPH_TYPEDEFS(Digraph);
   416 
   417   private:
   418 
   419 
   420     std::ostream* _os;
   421     bool local_os;
   422 
   423     const Digraph& _digraph;
   424 
   425     std::string _nodes_caption;
   426     std::string _arcs_caption;
   427     std::string _attributes_caption;
   428 
   429     typedef std::map<Node, std::string> NodeIndex;
   430     NodeIndex _node_index;
   431     typedef std::map<Arc, std::string> ArcIndex;
   432     ArcIndex _arc_index;
   433 
   434     typedef std::vector<std::pair<std::string,
   435       _writer_bits::MapStorageBase<Node>* > > NodeMaps;
   436     NodeMaps _node_maps;
   437 
   438     typedef std::vector<std::pair<std::string,
   439       _writer_bits::MapStorageBase<Arc>* > >ArcMaps;
   440     ArcMaps _arc_maps;
   441 
   442     typedef std::vector<std::pair<std::string,
   443       _writer_bits::ValueStorageBase*> > Attributes;
   444     Attributes _attributes;
   445 
   446     bool _skip_nodes;
   447     bool _skip_arcs;
   448 
   449   public:
   450 
   451     /// \brief Constructor
   452     ///
   453     /// Construct a directed graph writer, which writes to the given
   454     /// output stream.
   455     DigraphWriter(std::ostream& is, const Digraph& digraph)
   456       : _os(&is), local_os(false), _digraph(digraph),
   457         _skip_nodes(false), _skip_arcs(false) {}
   458 
   459     /// \brief Constructor
   460     ///
   461     /// Construct a directed graph writer, which writes to the given
   462     /// output file.
   463     DigraphWriter(const std::string& fn, const Digraph& digraph)
   464       : _os(new std::ofstream(fn.c_str())), local_os(true), _digraph(digraph),
   465         _skip_nodes(false), _skip_arcs(false) {
   466       if (!(*_os)) throw IoError(fn, "Cannot write file");
   467     }
   468 
   469     /// \brief Constructor
   470     ///
   471     /// Construct a directed graph writer, which writes to the given
   472     /// output file.
   473     DigraphWriter(const char* fn, const Digraph& digraph)
   474       : _os(new std::ofstream(fn)), local_os(true), _digraph(digraph),
   475         _skip_nodes(false), _skip_arcs(false) {
   476       if (!(*_os)) throw IoError(fn, "Cannot write file");
   477     }
   478 
   479     /// \brief Destructor
   480     ~DigraphWriter() {
   481       for (typename NodeMaps::iterator it = _node_maps.begin();
   482            it != _node_maps.end(); ++it) {
   483         delete it->second;
   484       }
   485 
   486       for (typename ArcMaps::iterator it = _arc_maps.begin();
   487            it != _arc_maps.end(); ++it) {
   488         delete it->second;
   489       }
   490 
   491       for (typename Attributes::iterator it = _attributes.begin();
   492            it != _attributes.end(); ++it) {
   493         delete it->second;
   494       }
   495 
   496       if (local_os) {
   497         delete _os;
   498       }
   499     }
   500 
   501   private:
   502 
   503     friend DigraphWriter<Digraph> digraphWriter<>(std::ostream& os,
   504                                                   const Digraph& digraph);
   505     friend DigraphWriter<Digraph> digraphWriter<>(const std::string& fn,
   506                                                   const Digraph& digraph);
   507     friend DigraphWriter<Digraph> digraphWriter<>(const char *fn,
   508                                                   const Digraph& digraph);
   509 
   510     DigraphWriter(DigraphWriter& other)
   511       : _os(other._os), local_os(other.local_os), _digraph(other._digraph),
   512         _skip_nodes(other._skip_nodes), _skip_arcs(other._skip_arcs) {
   513 
   514       other._os = 0;
   515       other.local_os = false;
   516 
   517       _node_index.swap(other._node_index);
   518       _arc_index.swap(other._arc_index);
   519 
   520       _node_maps.swap(other._node_maps);
   521       _arc_maps.swap(other._arc_maps);
   522       _attributes.swap(other._attributes);
   523 
   524       _nodes_caption = other._nodes_caption;
   525       _arcs_caption = other._arcs_caption;
   526       _attributes_caption = other._attributes_caption;
   527     }
   528 
   529     DigraphWriter& operator=(const DigraphWriter&);
   530 
   531   public:
   532 
   533     /// \name Writing rules
   534     /// @{
   535 
   536     /// \brief Node map writing rule
   537     ///
   538     /// Add a node map writing rule to the writer.
   539     template <typename Map>
   540     DigraphWriter& nodeMap(const std::string& caption, const Map& map) {
   541       checkConcept<concepts::ReadMap<Node, typename Map::Value>, Map>();
   542       _writer_bits::MapStorageBase<Node>* storage =
   543         new _writer_bits::MapStorage<Node, Map>(map);
   544       _node_maps.push_back(std::make_pair(caption, storage));
   545       return *this;
   546     }
   547 
   548     /// \brief Node map writing rule
   549     ///
   550     /// Add a node map writing rule with specialized converter to the
   551     /// writer.
   552     template <typename Map, typename Converter>
   553     DigraphWriter& nodeMap(const std::string& caption, const Map& map,
   554                            const Converter& converter = Converter()) {
   555       checkConcept<concepts::ReadMap<Node, typename Map::Value>, Map>();
   556       _writer_bits::MapStorageBase<Node>* storage =
   557         new _writer_bits::MapStorage<Node, Map, Converter>(map, converter);
   558       _node_maps.push_back(std::make_pair(caption, storage));
   559       return *this;
   560     }
   561 
   562     /// \brief Arc map writing rule
   563     ///
   564     /// Add an arc map writing rule to the writer.
   565     template <typename Map>
   566     DigraphWriter& arcMap(const std::string& caption, const Map& map) {
   567       checkConcept<concepts::ReadMap<Arc, typename Map::Value>, Map>();
   568       _writer_bits::MapStorageBase<Arc>* storage =
   569         new _writer_bits::MapStorage<Arc, Map>(map);
   570       _arc_maps.push_back(std::make_pair(caption, storage));
   571       return *this;
   572     }
   573 
   574     /// \brief Arc map writing rule
   575     ///
   576     /// Add an arc map writing rule with specialized converter to the
   577     /// writer.
   578     template <typename Map, typename Converter>
   579     DigraphWriter& arcMap(const std::string& caption, const Map& map,
   580                           const Converter& converter = Converter()) {
   581       checkConcept<concepts::ReadMap<Arc, typename Map::Value>, Map>();
   582       _writer_bits::MapStorageBase<Arc>* storage =
   583         new _writer_bits::MapStorage<Arc, Map, Converter>(map, converter);
   584       _arc_maps.push_back(std::make_pair(caption, storage));
   585       return *this;
   586     }
   587 
   588     /// \brief Attribute writing rule
   589     ///
   590     /// Add an attribute writing rule to the writer.
   591     template <typename Value>
   592     DigraphWriter& attribute(const std::string& caption, const Value& value) {
   593       _writer_bits::ValueStorageBase* storage =
   594         new _writer_bits::ValueStorage<Value>(value);
   595       _attributes.push_back(std::make_pair(caption, storage));
   596       return *this;
   597     }
   598 
   599     /// \brief Attribute writing rule
   600     ///
   601     /// Add an attribute writing rule with specialized converter to the
   602     /// writer.
   603     template <typename Value, typename Converter>
   604     DigraphWriter& attribute(const std::string& caption, const Value& value,
   605                              const Converter& converter = Converter()) {
   606       _writer_bits::ValueStorageBase* storage =
   607         new _writer_bits::ValueStorage<Value, Converter>(value, converter);
   608       _attributes.push_back(std::make_pair(caption, storage));
   609       return *this;
   610     }
   611 
   612     /// \brief Node writing rule
   613     ///
   614     /// Add a node writing rule to the writer.
   615     DigraphWriter& node(const std::string& caption, const Node& node) {
   616       typedef _writer_bits::MapLookUpConverter<Node> Converter;
   617       Converter converter(_node_index);
   618       _writer_bits::ValueStorageBase* storage =
   619         new _writer_bits::ValueStorage<Node, Converter>(node, converter);
   620       _attributes.push_back(std::make_pair(caption, storage));
   621       return *this;
   622     }
   623 
   624     /// \brief Arc writing rule
   625     ///
   626     /// Add an arc writing rule to writer.
   627     DigraphWriter& arc(const std::string& caption, const Arc& arc) {
   628       typedef _writer_bits::MapLookUpConverter<Arc> Converter;
   629       Converter converter(_arc_index);
   630       _writer_bits::ValueStorageBase* storage =
   631         new _writer_bits::ValueStorage<Arc, Converter>(arc, converter);
   632       _attributes.push_back(std::make_pair(caption, storage));
   633       return *this;
   634     }
   635 
   636     /// \name Section captions
   637     /// @{
   638 
   639     /// \brief Add an additional caption to the \c \@nodes section
   640     ///
   641     /// Add an additional caption to the \c \@nodes section.
   642     DigraphWriter& nodes(const std::string& caption) {
   643       _nodes_caption = caption;
   644       return *this;
   645     }
   646 
   647     /// \brief Add an additional caption to the \c \@arcs section
   648     ///
   649     /// Add an additional caption to the \c \@arcs section.
   650     DigraphWriter& arcs(const std::string& caption) {
   651       _arcs_caption = caption;
   652       return *this;
   653     }
   654 
   655     /// \brief Add an additional caption to the \c \@attributes section
   656     ///
   657     /// Add an additional caption to the \c \@attributes section.
   658     DigraphWriter& attributes(const std::string& caption) {
   659       _attributes_caption = caption;
   660       return *this;
   661     }
   662 
   663     /// \name Skipping section
   664     /// @{
   665 
   666     /// \brief Skip writing the node set
   667     ///
   668     /// The \c \@nodes section will not be written to the stream.
   669     DigraphWriter& skipNodes() {
   670       LEMON_ASSERT(!_skip_nodes, "Multiple usage of skipNodes() member");
   671       _skip_nodes = true;
   672       return *this;
   673     }
   674 
   675     /// \brief Skip writing arc set
   676     ///
   677     /// The \c \@arcs section will not be written to the stream.
   678     DigraphWriter& skipArcs() {
   679       LEMON_ASSERT(!_skip_arcs, "Multiple usage of skipArcs() member");
   680       _skip_arcs = true;
   681       return *this;
   682     }
   683 
   684     /// @}
   685 
   686   private:
   687 
   688     void writeNodes() {
   689       _writer_bits::MapStorageBase<Node>* label = 0;
   690       for (typename NodeMaps::iterator it = _node_maps.begin();
   691            it != _node_maps.end(); ++it) {
   692         if (it->first == "label") {
   693           label = it->second;
   694           break;
   695         }
   696       }
   697 
   698       *_os << "@nodes";
   699       if (!_nodes_caption.empty()) {
   700         _writer_bits::writeToken(*_os << ' ', _nodes_caption);
   701       }
   702       *_os << std::endl;
   703 
   704       if (label == 0) {
   705         *_os << "label" << '\t';
   706       }
   707       for (typename NodeMaps::iterator it = _node_maps.begin();
   708            it != _node_maps.end(); ++it) {
   709         _writer_bits::writeToken(*_os, it->first) << '\t';
   710       }
   711       *_os << std::endl;
   712 
   713       std::vector<Node> nodes;
   714       for (NodeIt n(_digraph); n != INVALID; ++n) {
   715         nodes.push_back(n);
   716       }
   717 
   718       if (label == 0) {
   719         IdMap<Digraph, Node> id_map(_digraph);
   720         _writer_bits::MapLess<IdMap<Digraph, Node> > id_less(id_map);
   721         std::sort(nodes.begin(), nodes.end(), id_less);
   722       } else {
   723         label->sort(nodes);
   724       }
   725 
   726       for (int i = 0; i < static_cast<int>(nodes.size()); ++i) {
   727         Node n = nodes[i];
   728         if (label == 0) {
   729           std::ostringstream os;
   730           os << _digraph.id(n);
   731           _writer_bits::writeToken(*_os, os.str());
   732           *_os << '\t';
   733           _node_index.insert(std::make_pair(n, os.str()));
   734         }
   735         for (typename NodeMaps::iterator it = _node_maps.begin();
   736              it != _node_maps.end(); ++it) {
   737           std::string value = it->second->get(n);
   738           _writer_bits::writeToken(*_os, value);
   739           if (it->first == "label") {
   740             _node_index.insert(std::make_pair(n, value));
   741           }
   742           *_os << '\t';
   743         }
   744         *_os << std::endl;
   745       }
   746     }
   747 
   748     void createNodeIndex() {
   749       _writer_bits::MapStorageBase<Node>* label = 0;
   750       for (typename NodeMaps::iterator it = _node_maps.begin();
   751            it != _node_maps.end(); ++it) {
   752         if (it->first == "label") {
   753           label = it->second;
   754           break;
   755         }
   756       }
   757 
   758       if (label == 0) {
   759         for (NodeIt n(_digraph); n != INVALID; ++n) {
   760           std::ostringstream os;
   761           os << _digraph.id(n);
   762           _node_index.insert(std::make_pair(n, os.str()));
   763         }
   764       } else {
   765         for (NodeIt n(_digraph); n != INVALID; ++n) {
   766           std::string value = label->get(n);
   767           _node_index.insert(std::make_pair(n, value));
   768         }
   769       }
   770     }
   771 
   772     void writeArcs() {
   773       _writer_bits::MapStorageBase<Arc>* label = 0;
   774       for (typename ArcMaps::iterator it = _arc_maps.begin();
   775            it != _arc_maps.end(); ++it) {
   776         if (it->first == "label") {
   777           label = it->second;
   778           break;
   779         }
   780       }
   781 
   782       *_os << "@arcs";
   783       if (!_arcs_caption.empty()) {
   784         _writer_bits::writeToken(*_os << ' ', _arcs_caption);
   785       }
   786       *_os << std::endl;
   787 
   788       *_os << '\t' << '\t';
   789       if (label == 0) {
   790         *_os << "label" << '\t';
   791       }
   792       for (typename ArcMaps::iterator it = _arc_maps.begin();
   793            it != _arc_maps.end(); ++it) {
   794         _writer_bits::writeToken(*_os, it->first) << '\t';
   795       }
   796       *_os << std::endl;
   797 
   798       std::vector<Arc> arcs;
   799       for (ArcIt n(_digraph); n != INVALID; ++n) {
   800         arcs.push_back(n);
   801       }
   802 
   803       if (label == 0) {
   804         IdMap<Digraph, Arc> id_map(_digraph);
   805         _writer_bits::MapLess<IdMap<Digraph, Arc> > id_less(id_map);
   806         std::sort(arcs.begin(), arcs.end(), id_less);
   807       } else {
   808         label->sort(arcs);
   809       }
   810 
   811       for (int i = 0; i < static_cast<int>(arcs.size()); ++i) {
   812         Arc a = arcs[i];
   813         _writer_bits::writeToken(*_os, _node_index.
   814                                  find(_digraph.source(a))->second);
   815         *_os << '\t';
   816         _writer_bits::writeToken(*_os, _node_index.
   817                                  find(_digraph.target(a))->second);
   818         *_os << '\t';
   819         if (label == 0) {
   820           std::ostringstream os;
   821           os << _digraph.id(a);
   822           _writer_bits::writeToken(*_os, os.str());
   823           *_os << '\t';
   824           _arc_index.insert(std::make_pair(a, os.str()));
   825         }
   826         for (typename ArcMaps::iterator it = _arc_maps.begin();
   827              it != _arc_maps.end(); ++it) {
   828           std::string value = it->second->get(a);
   829           _writer_bits::writeToken(*_os, value);
   830           if (it->first == "label") {
   831             _arc_index.insert(std::make_pair(a, value));
   832           }
   833           *_os << '\t';
   834         }
   835         *_os << std::endl;
   836       }
   837     }
   838 
   839     void createArcIndex() {
   840       _writer_bits::MapStorageBase<Arc>* label = 0;
   841       for (typename ArcMaps::iterator it = _arc_maps.begin();
   842            it != _arc_maps.end(); ++it) {
   843         if (it->first == "label") {
   844           label = it->second;
   845           break;
   846         }
   847       }
   848 
   849       if (label == 0) {
   850         for (ArcIt a(_digraph); a != INVALID; ++a) {
   851           std::ostringstream os;
   852           os << _digraph.id(a);
   853           _arc_index.insert(std::make_pair(a, os.str()));
   854         }
   855       } else {
   856         for (ArcIt a(_digraph); a != INVALID; ++a) {
   857           std::string value = label->get(a);
   858           _arc_index.insert(std::make_pair(a, value));
   859         }
   860       }
   861     }
   862 
   863     void writeAttributes() {
   864       if (_attributes.empty()) return;
   865       *_os << "@attributes";
   866       if (!_attributes_caption.empty()) {
   867         _writer_bits::writeToken(*_os << ' ', _attributes_caption);
   868       }
   869       *_os << std::endl;
   870       for (typename Attributes::iterator it = _attributes.begin();
   871            it != _attributes.end(); ++it) {
   872         _writer_bits::writeToken(*_os, it->first) << ' ';
   873         _writer_bits::writeToken(*_os, it->second->get());
   874         *_os << std::endl;
   875       }
   876     }
   877 
   878   public:
   879 
   880     /// \name Execution of the writer
   881     /// @{
   882 
   883     /// \brief Start the batch processing
   884     ///
   885     /// This function starts the batch processing.
   886     void run() {
   887       if (!_skip_nodes) {
   888         writeNodes();
   889       } else {
   890         createNodeIndex();
   891       }
   892       if (!_skip_arcs) {
   893         writeArcs();
   894       } else {
   895         createArcIndex();
   896       }
   897       writeAttributes();
   898     }
   899 
   900     /// \brief Give back the stream of the writer
   901     ///
   902     /// Give back the stream of the writer.
   903     std::ostream& ostream() {
   904       return *_os;
   905     }
   906 
   907     /// @}
   908   };
   909 
   910   /// \brief Return a \ref DigraphWriter class
   911   ///
   912   /// This function just returns a \ref DigraphWriter class.
   913   /// \relates DigraphWriter
   914   template <typename Digraph>
   915   DigraphWriter<Digraph> digraphWriter(std::ostream& os,
   916                                        const Digraph& digraph) {
   917     DigraphWriter<Digraph> tmp(os, digraph);
   918     return tmp;
   919   }
   920 
   921   /// \brief Return a \ref DigraphWriter class
   922   ///
   923   /// This function just returns a \ref DigraphWriter class.
   924   /// \relates DigraphWriter
   925   template <typename Digraph>
   926   DigraphWriter<Digraph> digraphWriter(const std::string& fn,
   927                                        const Digraph& digraph) {
   928     DigraphWriter<Digraph> tmp(fn, digraph);
   929     return tmp;
   930   }
   931 
   932   /// \brief Return a \ref DigraphWriter class
   933   ///
   934   /// This function just returns a \ref DigraphWriter class.
   935   /// \relates DigraphWriter
   936   template <typename Digraph>
   937   DigraphWriter<Digraph> digraphWriter(const char* fn,
   938                                        const Digraph& digraph) {
   939     DigraphWriter<Digraph> tmp(fn, digraph);
   940     return tmp;
   941   }
   942 
   943   template <typename Graph>
   944   class GraphWriter;
   945 
   946   template <typename Graph>
   947   GraphWriter<Graph> graphWriter(std::ostream& os, const Graph& graph);
   948 
   949   template <typename Graph>
   950   GraphWriter<Graph> graphWriter(const std::string& fn, const Graph& graph);
   951 
   952   template <typename Graph>
   953   GraphWriter<Graph> graphWriter(const char *fn, const Graph& graph);
   954 
   955   /// \ingroup lemon_io
   956   ///
   957   /// \brief \ref lgf-format "LGF" writer for directed graphs
   958   ///
   959   /// This utility writes an \ref lgf-format "LGF" file.
   960   ///
   961   /// It can be used almost the same way as \c DigraphWriter.
   962   /// The only difference is that this class can handle edges and
   963   /// edge maps as well as arcs and arc maps.
   964   ///
   965   /// The arc maps are written into the file as two columns, the
   966   /// caption of the columns are the name of the map prefixed with \c
   967   /// '+' and \c '-'. The arcs are written into the \c \@attributes
   968   /// section as a \c '+' or a \c '-' prefix (depends on the direction
   969   /// of the arc) and the label of corresponding edge.
   970   template <typename _Graph>
   971   class GraphWriter {
   972   public:
   973 
   974     typedef _Graph Graph;
   975     TEMPLATE_GRAPH_TYPEDEFS(Graph);
   976 
   977   private:
   978 
   979 
   980     std::ostream* _os;
   981     bool local_os;
   982 
   983     const Graph& _graph;
   984 
   985     std::string _nodes_caption;
   986     std::string _edges_caption;
   987     std::string _attributes_caption;
   988 
   989     typedef std::map<Node, std::string> NodeIndex;
   990     NodeIndex _node_index;
   991     typedef std::map<Edge, std::string> EdgeIndex;
   992     EdgeIndex _edge_index;
   993 
   994     typedef std::vector<std::pair<std::string,
   995       _writer_bits::MapStorageBase<Node>* > > NodeMaps;
   996     NodeMaps _node_maps;
   997 
   998     typedef std::vector<std::pair<std::string,
   999       _writer_bits::MapStorageBase<Edge>* > >EdgeMaps;
  1000     EdgeMaps _edge_maps;
  1001 
  1002     typedef std::vector<std::pair<std::string,
  1003       _writer_bits::ValueStorageBase*> > Attributes;
  1004     Attributes _attributes;
  1005 
  1006     bool _skip_nodes;
  1007     bool _skip_edges;
  1008 
  1009   public:
  1010 
  1011     /// \brief Constructor
  1012     ///
  1013     /// Construct a directed graph writer, which writes to the given
  1014     /// output stream.
  1015     GraphWriter(std::ostream& is, const Graph& graph)
  1016       : _os(&is), local_os(false), _graph(graph),
  1017         _skip_nodes(false), _skip_edges(false) {}
  1018 
  1019     /// \brief Constructor
  1020     ///
  1021     /// Construct a directed graph writer, which writes to the given
  1022     /// output file.
  1023     GraphWriter(const std::string& fn, const Graph& graph)
  1024       : _os(new std::ofstream(fn.c_str())), local_os(true), _graph(graph),
  1025         _skip_nodes(false), _skip_edges(false) {
  1026       if (!(*_os)) throw IoError(fn, "Cannot write file");
  1027     }
  1028 
  1029     /// \brief Constructor
  1030     ///
  1031     /// Construct a directed graph writer, which writes to the given
  1032     /// output file.
  1033     GraphWriter(const char* fn, const Graph& graph)
  1034       : _os(new std::ofstream(fn)), local_os(true), _graph(graph),
  1035         _skip_nodes(false), _skip_edges(false) {
  1036       if (!(*_os)) throw IoError(fn, "Cannot write file");
  1037     }
  1038 
  1039     /// \brief Destructor
  1040     ~GraphWriter() {
  1041       for (typename NodeMaps::iterator it = _node_maps.begin();
  1042            it != _node_maps.end(); ++it) {
  1043         delete it->second;
  1044       }
  1045 
  1046       for (typename EdgeMaps::iterator it = _edge_maps.begin();
  1047            it != _edge_maps.end(); ++it) {
  1048         delete it->second;
  1049       }
  1050 
  1051       for (typename Attributes::iterator it = _attributes.begin();
  1052            it != _attributes.end(); ++it) {
  1053         delete it->second;
  1054       }
  1055 
  1056       if (local_os) {
  1057         delete _os;
  1058       }
  1059     }
  1060 
  1061   private:
  1062 
  1063     friend GraphWriter<Graph> graphWriter<>(std::ostream& os,
  1064                                             const Graph& graph);
  1065     friend GraphWriter<Graph> graphWriter<>(const std::string& fn,
  1066                                             const Graph& graph);
  1067     friend GraphWriter<Graph> graphWriter<>(const char *fn,
  1068                                             const Graph& graph);
  1069 
  1070     GraphWriter(GraphWriter& other)
  1071       : _os(other._os), local_os(other.local_os), _graph(other._graph),
  1072         _skip_nodes(other._skip_nodes), _skip_edges(other._skip_edges) {
  1073 
  1074       other._os = 0;
  1075       other.local_os = false;
  1076 
  1077       _node_index.swap(other._node_index);
  1078       _edge_index.swap(other._edge_index);
  1079 
  1080       _node_maps.swap(other._node_maps);
  1081       _edge_maps.swap(other._edge_maps);
  1082       _attributes.swap(other._attributes);
  1083 
  1084       _nodes_caption = other._nodes_caption;
  1085       _edges_caption = other._edges_caption;
  1086       _attributes_caption = other._attributes_caption;
  1087     }
  1088 
  1089     GraphWriter& operator=(const GraphWriter&);
  1090 
  1091   public:
  1092 
  1093     /// \name Writing rules
  1094     /// @{
  1095 
  1096     /// \brief Node map writing rule
  1097     ///
  1098     /// Add a node map writing rule to the writer.
  1099     template <typename Map>
  1100     GraphWriter& nodeMap(const std::string& caption, const Map& map) {
  1101       checkConcept<concepts::ReadMap<Node, typename Map::Value>, Map>();
  1102       _writer_bits::MapStorageBase<Node>* storage =
  1103         new _writer_bits::MapStorage<Node, Map>(map);
  1104       _node_maps.push_back(std::make_pair(caption, storage));
  1105       return *this;
  1106     }
  1107 
  1108     /// \brief Node map writing rule
  1109     ///
  1110     /// Add a node map writing rule with specialized converter to the
  1111     /// writer.
  1112     template <typename Map, typename Converter>
  1113     GraphWriter& nodeMap(const std::string& caption, const Map& map,
  1114                            const Converter& converter = Converter()) {
  1115       checkConcept<concepts::ReadMap<Node, typename Map::Value>, Map>();
  1116       _writer_bits::MapStorageBase<Node>* storage =
  1117         new _writer_bits::MapStorage<Node, Map, Converter>(map, converter);
  1118       _node_maps.push_back(std::make_pair(caption, storage));
  1119       return *this;
  1120     }
  1121 
  1122     /// \brief Edge map writing rule
  1123     ///
  1124     /// Add an edge map writing rule to the writer.
  1125     template <typename Map>
  1126     GraphWriter& edgeMap(const std::string& caption, const Map& map) {
  1127       checkConcept<concepts::ReadMap<Edge, typename Map::Value>, Map>();
  1128       _writer_bits::MapStorageBase<Edge>* storage =
  1129         new _writer_bits::MapStorage<Edge, Map>(map);
  1130       _edge_maps.push_back(std::make_pair(caption, storage));
  1131       return *this;
  1132     }
  1133 
  1134     /// \brief Edge map writing rule
  1135     ///
  1136     /// Add an edge map writing rule with specialized converter to the
  1137     /// writer.
  1138     template <typename Map, typename Converter>
  1139     GraphWriter& edgeMap(const std::string& caption, const Map& map,
  1140                           const Converter& converter = Converter()) {
  1141       checkConcept<concepts::ReadMap<Edge, typename Map::Value>, Map>();
  1142       _writer_bits::MapStorageBase<Edge>* storage =
  1143         new _writer_bits::MapStorage<Edge, Map, Converter>(map, converter);
  1144       _edge_maps.push_back(std::make_pair(caption, storage));
  1145       return *this;
  1146     }
  1147 
  1148     /// \brief Arc map writing rule
  1149     ///
  1150     /// Add an arc map writing rule to the writer.
  1151     template <typename Map>
  1152     GraphWriter& arcMap(const std::string& caption, const Map& map) {
  1153       checkConcept<concepts::ReadMap<Arc, typename Map::Value>, Map>();
  1154       _writer_bits::MapStorageBase<Edge>* forward_storage =
  1155         new _writer_bits::GraphArcMapStorage<Graph, true, Map>(_graph, map);
  1156       _edge_maps.push_back(std::make_pair('+' + caption, forward_storage));
  1157       _writer_bits::MapStorageBase<Edge>* backward_storage =
  1158         new _writer_bits::GraphArcMapStorage<Graph, false, Map>(_graph, map);
  1159       _edge_maps.push_back(std::make_pair('-' + caption, backward_storage));
  1160       return *this;
  1161     }
  1162 
  1163     /// \brief Arc map writing rule
  1164     ///
  1165     /// Add an arc map writing rule with specialized converter to the
  1166     /// writer.
  1167     template <typename Map, typename Converter>
  1168     GraphWriter& arcMap(const std::string& caption, const Map& map,
  1169                           const Converter& converter = Converter()) {
  1170       checkConcept<concepts::ReadMap<Arc, typename Map::Value>, Map>();
  1171       _writer_bits::MapStorageBase<Edge>* forward_storage =
  1172         new _writer_bits::GraphArcMapStorage<Graph, true, Map, Converter>
  1173         (_graph, map, converter);
  1174       _edge_maps.push_back(std::make_pair('+' + caption, forward_storage));
  1175       _writer_bits::MapStorageBase<Edge>* backward_storage =
  1176         new _writer_bits::GraphArcMapStorage<Graph, false, Map, Converter>
  1177         (_graph, map, converter);
  1178       _edge_maps.push_back(std::make_pair('-' + caption, backward_storage));
  1179       return *this;
  1180     }
  1181 
  1182     /// \brief Attribute writing rule
  1183     ///
  1184     /// Add an attribute writing rule to the writer.
  1185     template <typename Value>
  1186     GraphWriter& attribute(const std::string& caption, const Value& value) {
  1187       _writer_bits::ValueStorageBase* storage =
  1188         new _writer_bits::ValueStorage<Value>(value);
  1189       _attributes.push_back(std::make_pair(caption, storage));
  1190       return *this;
  1191     }
  1192 
  1193     /// \brief Attribute writing rule
  1194     ///
  1195     /// Add an attribute writing rule with specialized converter to the
  1196     /// writer.
  1197     template <typename Value, typename Converter>
  1198     GraphWriter& attribute(const std::string& caption, const Value& value,
  1199                              const Converter& converter = Converter()) {
  1200       _writer_bits::ValueStorageBase* storage =
  1201         new _writer_bits::ValueStorage<Value, Converter>(value, converter);
  1202       _attributes.push_back(std::make_pair(caption, storage));
  1203       return *this;
  1204     }
  1205 
  1206     /// \brief Node writing rule
  1207     ///
  1208     /// Add a node writing rule to the writer.
  1209     GraphWriter& node(const std::string& caption, const Node& node) {
  1210       typedef _writer_bits::MapLookUpConverter<Node> Converter;
  1211       Converter converter(_node_index);
  1212       _writer_bits::ValueStorageBase* storage =
  1213         new _writer_bits::ValueStorage<Node, Converter>(node, converter);
  1214       _attributes.push_back(std::make_pair(caption, storage));
  1215       return *this;
  1216     }
  1217 
  1218     /// \brief Edge writing rule
  1219     ///
  1220     /// Add an edge writing rule to writer.
  1221     GraphWriter& edge(const std::string& caption, const Edge& edge) {
  1222       typedef _writer_bits::MapLookUpConverter<Edge> Converter;
  1223       Converter converter(_edge_index);
  1224       _writer_bits::ValueStorageBase* storage =
  1225         new _writer_bits::ValueStorage<Edge, Converter>(edge, converter);
  1226       _attributes.push_back(std::make_pair(caption, storage));
  1227       return *this;
  1228     }
  1229 
  1230     /// \brief Arc writing rule
  1231     ///
  1232     /// Add an arc writing rule to writer.
  1233     GraphWriter& arc(const std::string& caption, const Arc& arc) {
  1234       typedef _writer_bits::GraphArcLookUpConverter<Graph> Converter;
  1235       Converter converter(_graph, _edge_index);
  1236       _writer_bits::ValueStorageBase* storage =
  1237         new _writer_bits::ValueStorage<Arc, Converter>(arc, converter);
  1238       _attributes.push_back(std::make_pair(caption, storage));
  1239       return *this;
  1240     }
  1241 
  1242     /// \name Section captions
  1243     /// @{
  1244 
  1245     /// \brief Add an additional caption to the \c \@nodes section
  1246     ///
  1247     /// Add an additional caption to the \c \@nodes section.
  1248     GraphWriter& nodes(const std::string& caption) {
  1249       _nodes_caption = caption;
  1250       return *this;
  1251     }
  1252 
  1253     /// \brief Add an additional caption to the \c \@arcs section
  1254     ///
  1255     /// Add an additional caption to the \c \@arcs section.
  1256     GraphWriter& edges(const std::string& caption) {
  1257       _edges_caption = caption;
  1258       return *this;
  1259     }
  1260 
  1261     /// \brief Add an additional caption to the \c \@attributes section
  1262     ///
  1263     /// Add an additional caption to the \c \@attributes section.
  1264     GraphWriter& attributes(const std::string& caption) {
  1265       _attributes_caption = caption;
  1266       return *this;
  1267     }
  1268 
  1269     /// \name Skipping section
  1270     /// @{
  1271 
  1272     /// \brief Skip writing the node set
  1273     ///
  1274     /// The \c \@nodes section will not be written to the stream.
  1275     GraphWriter& skipNodes() {
  1276       LEMON_ASSERT(!_skip_nodes, "Multiple usage of skipNodes() member");
  1277       _skip_nodes = true;
  1278       return *this;
  1279     }
  1280 
  1281     /// \brief Skip writing edge set
  1282     ///
  1283     /// The \c \@edges section will not be written to the stream.
  1284     GraphWriter& skipEdges() {
  1285       LEMON_ASSERT(!_skip_edges, "Multiple usage of skipEdges() member");
  1286       _skip_edges = true;
  1287       return *this;
  1288     }
  1289 
  1290     /// @}
  1291 
  1292   private:
  1293 
  1294     void writeNodes() {
  1295       _writer_bits::MapStorageBase<Node>* label = 0;
  1296       for (typename NodeMaps::iterator it = _node_maps.begin();
  1297            it != _node_maps.end(); ++it) {
  1298         if (it->first == "label") {
  1299           label = it->second;
  1300           break;
  1301         }
  1302       }
  1303 
  1304       *_os << "@nodes";
  1305       if (!_nodes_caption.empty()) {
  1306         _writer_bits::writeToken(*_os << ' ', _nodes_caption);
  1307       }
  1308       *_os << std::endl;
  1309 
  1310       if (label == 0) {
  1311         *_os << "label" << '\t';
  1312       }
  1313       for (typename NodeMaps::iterator it = _node_maps.begin();
  1314            it != _node_maps.end(); ++it) {
  1315         _writer_bits::writeToken(*_os, it->first) << '\t';
  1316       }
  1317       *_os << std::endl;
  1318 
  1319       std::vector<Node> nodes;
  1320       for (NodeIt n(_graph); n != INVALID; ++n) {
  1321         nodes.push_back(n);
  1322       }
  1323 
  1324       if (label == 0) {
  1325         IdMap<Graph, Node> id_map(_graph);
  1326         _writer_bits::MapLess<IdMap<Graph, Node> > id_less(id_map);
  1327         std::sort(nodes.begin(), nodes.end(), id_less);
  1328       } else {
  1329         label->sort(nodes);
  1330       }
  1331 
  1332       for (int i = 0; i < static_cast<int>(nodes.size()); ++i) {
  1333         Node n = nodes[i];
  1334         if (label == 0) {
  1335           std::ostringstream os;
  1336           os << _graph.id(n);
  1337           _writer_bits::writeToken(*_os, os.str());
  1338           *_os << '\t';
  1339           _node_index.insert(std::make_pair(n, os.str()));
  1340         }
  1341         for (typename NodeMaps::iterator it = _node_maps.begin();
  1342              it != _node_maps.end(); ++it) {
  1343           std::string value = it->second->get(n);
  1344           _writer_bits::writeToken(*_os, value);
  1345           if (it->first == "label") {
  1346             _node_index.insert(std::make_pair(n, value));
  1347           }
  1348           *_os << '\t';
  1349         }
  1350         *_os << std::endl;
  1351       }
  1352     }
  1353 
  1354     void createNodeIndex() {
  1355       _writer_bits::MapStorageBase<Node>* label = 0;
  1356       for (typename NodeMaps::iterator it = _node_maps.begin();
  1357            it != _node_maps.end(); ++it) {
  1358         if (it->first == "label") {
  1359           label = it->second;
  1360           break;
  1361         }
  1362       }
  1363 
  1364       if (label == 0) {
  1365         for (NodeIt n(_graph); n != INVALID; ++n) {
  1366           std::ostringstream os;
  1367           os << _graph.id(n);
  1368           _node_index.insert(std::make_pair(n, os.str()));
  1369         }
  1370       } else {
  1371         for (NodeIt n(_graph); n != INVALID; ++n) {
  1372           std::string value = label->get(n);
  1373           _node_index.insert(std::make_pair(n, value));
  1374         }
  1375       }
  1376     }
  1377 
  1378     void writeEdges() {
  1379       _writer_bits::MapStorageBase<Edge>* label = 0;
  1380       for (typename EdgeMaps::iterator it = _edge_maps.begin();
  1381            it != _edge_maps.end(); ++it) {
  1382         if (it->first == "label") {
  1383           label = it->second;
  1384           break;
  1385         }
  1386       }
  1387 
  1388       *_os << "@edges";
  1389       if (!_edges_caption.empty()) {
  1390         _writer_bits::writeToken(*_os << ' ', _edges_caption);
  1391       }
  1392       *_os << std::endl;
  1393 
  1394       *_os << '\t' << '\t';
  1395       if (label == 0) {
  1396         *_os << "label" << '\t';
  1397       }
  1398       for (typename EdgeMaps::iterator it = _edge_maps.begin();
  1399            it != _edge_maps.end(); ++it) {
  1400         _writer_bits::writeToken(*_os, it->first) << '\t';
  1401       }
  1402       *_os << std::endl;
  1403 
  1404       std::vector<Edge> edges;
  1405       for (EdgeIt n(_graph); n != INVALID; ++n) {
  1406         edges.push_back(n);
  1407       }
  1408 
  1409       if (label == 0) {
  1410         IdMap<Graph, Edge> id_map(_graph);
  1411         _writer_bits::MapLess<IdMap<Graph, Edge> > id_less(id_map);
  1412         std::sort(edges.begin(), edges.end(), id_less);
  1413       } else {
  1414         label->sort(edges);
  1415       }
  1416 
  1417       for (int i = 0; i < static_cast<int>(edges.size()); ++i) {
  1418         Edge e = edges[i];
  1419         _writer_bits::writeToken(*_os, _node_index.
  1420                                  find(_graph.u(e))->second);
  1421         *_os << '\t';
  1422         _writer_bits::writeToken(*_os, _node_index.
  1423                                  find(_graph.v(e))->second);
  1424         *_os << '\t';
  1425         if (label == 0) {
  1426           std::ostringstream os;
  1427           os << _graph.id(e);
  1428           _writer_bits::writeToken(*_os, os.str());
  1429           *_os << '\t';
  1430           _edge_index.insert(std::make_pair(e, os.str()));
  1431         }
  1432         for (typename EdgeMaps::iterator it = _edge_maps.begin();
  1433              it != _edge_maps.end(); ++it) {
  1434           std::string value = it->second->get(e);
  1435           _writer_bits::writeToken(*_os, value);
  1436           if (it->first == "label") {
  1437             _edge_index.insert(std::make_pair(e, value));
  1438           }
  1439           *_os << '\t';
  1440         }
  1441         *_os << std::endl;
  1442       }
  1443     }
  1444 
  1445     void createEdgeIndex() {
  1446       _writer_bits::MapStorageBase<Edge>* label = 0;
  1447       for (typename EdgeMaps::iterator it = _edge_maps.begin();
  1448            it != _edge_maps.end(); ++it) {
  1449         if (it->first == "label") {
  1450           label = it->second;
  1451           break;
  1452         }
  1453       }
  1454 
  1455       if (label == 0) {
  1456         for (EdgeIt e(_graph); e != INVALID; ++e) {
  1457           std::ostringstream os;
  1458           os << _graph.id(e);
  1459           _edge_index.insert(std::make_pair(e, os.str()));
  1460         }
  1461       } else {
  1462         for (EdgeIt e(_graph); e != INVALID; ++e) {
  1463           std::string value = label->get(e);
  1464           _edge_index.insert(std::make_pair(e, value));
  1465         }
  1466       }
  1467     }
  1468 
  1469     void writeAttributes() {
  1470       if (_attributes.empty()) return;
  1471       *_os << "@attributes";
  1472       if (!_attributes_caption.empty()) {
  1473         _writer_bits::writeToken(*_os << ' ', _attributes_caption);
  1474       }
  1475       *_os << std::endl;
  1476       for (typename Attributes::iterator it = _attributes.begin();
  1477            it != _attributes.end(); ++it) {
  1478         _writer_bits::writeToken(*_os, it->first) << ' ';
  1479         _writer_bits::writeToken(*_os, it->second->get());
  1480         *_os << std::endl;
  1481       }
  1482     }
  1483 
  1484   public:
  1485 
  1486     /// \name Execution of the writer
  1487     /// @{
  1488 
  1489     /// \brief Start the batch processing
  1490     ///
  1491     /// This function starts the batch processing.
  1492     void run() {
  1493       if (!_skip_nodes) {
  1494         writeNodes();
  1495       } else {
  1496         createNodeIndex();
  1497       }
  1498       if (!_skip_edges) {
  1499         writeEdges();
  1500       } else {
  1501         createEdgeIndex();
  1502       }
  1503       writeAttributes();
  1504     }
  1505 
  1506     /// \brief Give back the stream of the writer
  1507     ///
  1508     /// Give back the stream of the writer
  1509     std::ostream& ostream() {
  1510       return *_os;
  1511     }
  1512 
  1513     /// @}
  1514   };
  1515 
  1516   /// \brief Return a \ref GraphWriter class
  1517   ///
  1518   /// This function just returns a \ref GraphWriter class.
  1519   /// \relates GraphWriter
  1520   template <typename Graph>
  1521   GraphWriter<Graph> graphWriter(std::ostream& os, const Graph& graph) {
  1522     GraphWriter<Graph> tmp(os, graph);
  1523     return tmp;
  1524   }
  1525 
  1526   /// \brief Return a \ref GraphWriter class
  1527   ///
  1528   /// This function just returns a \ref GraphWriter class.
  1529   /// \relates GraphWriter
  1530   template <typename Graph>
  1531   GraphWriter<Graph> graphWriter(const std::string& fn, const Graph& graph) {
  1532     GraphWriter<Graph> tmp(fn, graph);
  1533     return tmp;
  1534   }
  1535 
  1536   /// \brief Return a \ref GraphWriter class
  1537   ///
  1538   /// This function just returns a \ref GraphWriter class.
  1539   /// \relates GraphWriter
  1540   template <typename Graph>
  1541   GraphWriter<Graph> graphWriter(const char* fn, const Graph& graph) {
  1542     GraphWriter<Graph> tmp(fn, graph);
  1543     return tmp;
  1544   }
  1545 
  1546   class SectionWriter;
  1547 
  1548   SectionWriter sectionWriter(std::istream& is);
  1549   SectionWriter sectionWriter(const std::string& fn);
  1550   SectionWriter sectionWriter(const char* fn);
  1551 
  1552   /// \ingroup lemon_io
  1553   ///
  1554   /// \brief Section writer class
  1555   ///
  1556   /// In the \ref lgf-format "LGF" file extra sections can be placed,
  1557   /// which contain any data in arbitrary format. Such sections can be
  1558   /// written with this class. A writing rule can be added to the
  1559   /// class with two different functions. With the \c sectionLines()
  1560   /// function a generator can write the section line-by-line, while
  1561   /// with the \c sectionStream() member the section can be written to
  1562   /// an output stream.
  1563   class SectionWriter {
  1564   private:
  1565 
  1566     std::ostream* _os;
  1567     bool local_os;
  1568 
  1569     typedef std::vector<std::pair<std::string, _writer_bits::Section*> >
  1570     Sections;
  1571 
  1572     Sections _sections;
  1573 
  1574   public:
  1575 
  1576     /// \brief Constructor
  1577     ///
  1578     /// Construct a section writer, which writes to the given output
  1579     /// stream.
  1580     SectionWriter(std::ostream& os)
  1581       : _os(&os), local_os(false) {}
  1582 
  1583     /// \brief Constructor
  1584     ///
  1585     /// Construct a section writer, which writes into the given file.
  1586     SectionWriter(const std::string& fn)
  1587       : _os(new std::ofstream(fn.c_str())), local_os(true) {
  1588       if (!(*_os)) throw IoError(fn, "Cannot write file");
  1589     }
  1590 
  1591     /// \brief Constructor
  1592     ///
  1593     /// Construct a section writer, which writes into the given file.
  1594     SectionWriter(const char* fn)
  1595       : _os(new std::ofstream(fn)), local_os(true) {
  1596       if (!(*_os)) throw IoError(fn, "Cannot write file");
  1597     }
  1598 
  1599     /// \brief Destructor
  1600     ~SectionWriter() {
  1601       for (Sections::iterator it = _sections.begin();
  1602            it != _sections.end(); ++it) {
  1603         delete it->second;
  1604       }
  1605 
  1606       if (local_os) {
  1607         delete _os;
  1608       }
  1609 
  1610     }
  1611 
  1612   private:
  1613 
  1614     friend SectionWriter sectionWriter(std::ostream& os);
  1615     friend SectionWriter sectionWriter(const std::string& fn);
  1616     friend SectionWriter sectionWriter(const char* fn);
  1617 
  1618     SectionWriter(SectionWriter& other)
  1619       : _os(other._os), local_os(other.local_os) {
  1620 
  1621       other._os = 0;
  1622       other.local_os = false;
  1623 
  1624       _sections.swap(other._sections);
  1625     }
  1626 
  1627     SectionWriter& operator=(const SectionWriter&);
  1628 
  1629   public:
  1630 
  1631     /// \name Section writers
  1632     /// @{
  1633 
  1634     /// \brief Add a section writer with line oriented writing
  1635     ///
  1636     /// The first parameter is the type descriptor of the section, the
  1637     /// second is a generator with std::string values. At the writing
  1638     /// process, the returned \c std::string will be written into the
  1639     /// output file until it is an empty string.
  1640     ///
  1641     /// For example, an integer vector is written into a section.
  1642     ///\code
  1643     ///  @numbers
  1644     ///  12 45 23 78
  1645     ///  4 28 38 28
  1646     ///  23 6 16
  1647     ///\endcode
  1648     ///
  1649     /// The generator is implemented as a struct.
  1650     ///\code
  1651     ///  struct NumberSection {
  1652     ///    std::vector<int>::const_iterator _it, _end;
  1653     ///    NumberSection(const std::vector<int>& data)
  1654     ///      : _it(data.begin()), _end(data.end()) {}
  1655     ///    std::string operator()() {
  1656     ///      int rem_in_line = 4;
  1657     ///      std::ostringstream ls;
  1658     ///      while (rem_in_line > 0 && _it != _end) {
  1659     ///        ls << *(_it++) << ' ';
  1660     ///        --rem_in_line;
  1661     ///      }
  1662     ///      return ls.str();
  1663     ///    }
  1664     ///  };
  1665     ///
  1666     ///  // ...
  1667     ///
  1668     ///  writer.sectionLines("numbers", NumberSection(vec));
  1669     ///\endcode
  1670     template <typename Functor>
  1671     SectionWriter& sectionLines(const std::string& type, Functor functor) {
  1672       LEMON_ASSERT(!type.empty(), "Type is empty.");
  1673       _sections.push_back(std::make_pair(type,
  1674         new _writer_bits::LineSection<Functor>(functor)));
  1675       return *this;
  1676     }
  1677 
  1678 
  1679     /// \brief Add a section writer with stream oriented writing
  1680     ///
  1681     /// The first parameter is the type of the section, the second is
  1682     /// a functor, which takes a \c std::ostream& parameter. The
  1683     /// functor writes the section to the output stream.
  1684     /// \warning The last line must be closed with end-line character.
  1685     template <typename Functor>
  1686     SectionWriter& sectionStream(const std::string& type, Functor functor) {
  1687       LEMON_ASSERT(!type.empty(), "Type is empty.");
  1688       _sections.push_back(std::make_pair(type,
  1689          new _writer_bits::StreamSection<Functor>(functor)));
  1690       return *this;
  1691     }
  1692 
  1693     /// @}
  1694 
  1695   public:
  1696 
  1697 
  1698     /// \name Execution of the writer
  1699     /// @{
  1700 
  1701     /// \brief Start the batch processing
  1702     ///
  1703     /// This function starts the batch processing.
  1704     void run() {
  1705 
  1706       LEMON_ASSERT(_os != 0, "This writer is assigned to an other writer");
  1707 
  1708       for (Sections::iterator it = _sections.begin();
  1709            it != _sections.end(); ++it) {
  1710         (*_os) << '@' << it->first << std::endl;
  1711         it->second->process(*_os);
  1712       }
  1713     }
  1714 
  1715     /// \brief Give back the stream of the writer
  1716     ///
  1717     /// Returns the stream of the writer
  1718     std::ostream& ostream() {
  1719       return *_os;
  1720     }
  1721 
  1722     /// @}
  1723 
  1724   };
  1725 
  1726   /// \brief Return a \ref SectionWriter class
  1727   ///
  1728   /// This function just returns a \ref SectionWriter class.
  1729   /// \relates SectionWriter
  1730   inline SectionWriter sectionWriter(std::ostream& os) {
  1731     SectionWriter tmp(os);
  1732     return tmp;
  1733   }
  1734 
  1735   /// \brief Return a \ref SectionWriter class
  1736   ///
  1737   /// This function just returns a \ref SectionWriter class.
  1738   /// \relates SectionWriter
  1739   inline SectionWriter sectionWriter(const std::string& fn) {
  1740     SectionWriter tmp(fn);
  1741     return tmp;
  1742   }
  1743 
  1744   /// \brief Return a \ref SectionWriter class
  1745   ///
  1746   /// This function just returns a \ref SectionWriter class.
  1747   /// \relates SectionWriter
  1748   inline SectionWriter sectionWriter(const char* fn) {
  1749     SectionWriter tmp(fn);
  1750     return tmp;
  1751   }
  1752 }
  1753 
  1754 #endif