source: lemon-0.x/src/work/deba/graph_writer.h @ 1059:bd97feae7d90

/* -*- C++ -*-
 * src/lemon/graph_writer.h - Part of LEMON, a generic C++ optimization library
 *
 * Copyright (C) 2004 Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
 * (Egervary Combinatorial Optimization Research Group, 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 gio
///\file
///\brief Graph writer.


#include <iostream>
#include <sstream>

#include <map>
#include <vector>

#include <memory>

#include <lemon/invalid.h>
#include <lemon/error.h>


namespace lemon {

  struct DefaultWriterTraits {

    template <typename _Value>
    struct Writer {
      typedef _Value Value;

      void write(std::ostream& os, const Value& value) {
        os << value << '\t';
      }
    };

  };


  class QuotedStringWriter {
  public:
    typedef std::string Value;

    QuotedStringWriter(bool _escaped = true) : escaped(_escaped) {}

    void write(std::ostream& os, const std::string& value) {
      char c;
      os << "\"";
      if (escaped) {
        ostringstream ls;
        for (int i = 0; i < value.size(); ++i) {
          writeEscape(ls, value[i]);
        }
        os << ls.str();
      } else {
        os << value;
      }
      os << "\"";
    }

  private:
    
    static void writeEscape(std::ostream& os, char c) {
      switch (c) {
      case '\\':
        os << "\\\\";
        return;
      case '\"':
        os << "\\\"";
        return;
      case '\'':
        os << "\\\'";
        return;
      case '\?':
        os << "\\\?";
        return;
      case '\a':
        os << "\\a";
        return;
      case '\b':
        os << "\\b";
        return;
      case '\f':
        os << "\\f";
        return;
      case '\r':
        os << "\\r";
        return;
      case '\n':
        os << "\\n";
        return;
      case '\t':
        os << "\\t";
        return;
      case '\v':
        os << "\\v";
        return;
      default:
        if (c < 0x20) {
          os << '\\' << oct << (int)c;
        } else {
          os << c;
        }
        return;
      }     
    }
  private:
    bool escaped;
  };

  // Graph writer
  
  template <typename _Graph, typename _WriterTraits = DefaultWriterTraits> 
  class GraphWriter {
  public:
    
    typedef _Graph Graph;
    typedef typename Graph::Node Node;
    typedef typename Graph::NodeIt NodeIt;
    typedef typename Graph::Edge Edge;
    typedef typename Graph::EdgeIt EdgeIt;

    typedef _WriterTraits WriterTraits;

    GraphWriter(std::ostream& _os, Graph& _graph) : os(_os), graph(_graph) {}


    ~GraphWriter() {

      for (typename NodeMapWriters::iterator it = node_map_writers.begin(); it != node_map_writers.end(); ++it) {
        delete it->second;
      }

      for (typename EdgeMapWriters::iterator it = edge_map_writers.begin(); it != edge_map_writers.end(); ++it) {
        delete it->second;
      }

    }

    // Node map rules

    template <typename Map>
    GraphWriter& writeNodeMap(std::string name, const Map& map) {
      return writeNodeMap<typename WriterTraits::template Writer<typename Map::Value>, Map>(name, map);
    }

    template <typename Writer, typename Map>
    GraphWriter& writeNodeMap(std::string name, const Map& map, const Writer& writer = Writer()) {
      //      if (node_map_writers.find(name) != node_map_writers.end()) {
      //        throw Exception() << "Multiple write rule for node map: " << name;
      //      }
      node_map_writers.push_back(make_pair(name, new MapWriter<Node, Map, Writer>(map, writer)));
      return *this;
    }

    // Edge map rules

    template <typename Map>
    GraphWriter& writeEdgeMap(std::string name, const Map& map) { 
      return writeEdgeMap<typename WriterTraits::template Writer<typename Map::Value>, Map>(name, map);
    }


    template <typename Writer, typename Map>
    GraphWriter& writeEdgeMap(std::string name, const Map& map, const Writer& writer = Writer()) {
      //      if (edge_map_writers.find(name) != edge_map_writers.end()) {
      //        throw Exception() << "Multiple write rule for edge map: " << name;
      //      }
      edge_map_writers.push_back(make_pair(name, new MapWriter<Edge, Map, Writer>(map, writer)));
      return *this;
    }

    // Node rules
    GraphWriter& writeNode(std::string name, const Node& node) {
      //      if (node_writers.find(name) != node_writers.end()) {
      //        throw Exception() << "Multiple write rule for node";
      //      }
      node_writers.push_back(make_pair(name, node));
    }

    // Edge rules

    GraphWriter& writeEdge(std::string name, const Edge& edge) {
      //      if (edge_writers.find(name) != edge_writers.end()) {
      //        throw Exception() << "Multiple write rule for edge";
      //      }
      edge_writers.push_back(make_pair(name, edge));
    }

    void write() {   
      writeNodeSet();
      writeEdgeSet();
      writeNodes();
      writeEdges();
      os << "@end" << std::endl;
    }

  private:

    void writeNodeSet() {
      if (node_map_writers.size() == 0) return;
      os << "@nodeset" << std::endl;
      for (int i = 0; i < node_map_writers.size(); ++i) {
        os << node_map_writers[i].first << '\t';
      } 
      os << std::endl;
      for (NodeIt it(graph); it != INVALID; ++it) {
        for (int i = 0; i < node_map_writers.size(); ++i) {
          node_map_writers[i].second->write(os, it);
        }
        os << std::endl;
      }

    }

    void writeEdgeSet() {
      if (edge_map_writers.size() == 0) return;
      if (node_map_writers.size() == 0) {
        throw Exception() << "Missing node id map";
      }
      os << "@edgeset" << std::endl;
      os << "\t\t";
      for (int i = 0; i < edge_map_writers.size(); ++i) {
        os << edge_map_writers[i].first << '\t';
      } 
      os << std::endl;
      for (EdgeIt it(graph); it != INVALID; ++it) {
        node_map_writers[0].second->write(os, graph.source(it));
        node_map_writers[0].second->write(os, graph.target(it));
        for (int i = 0; i < edge_map_writers.size(); ++i) {
          edge_map_writers[i].second->write(os, it);
        }
        os << std::endl;
      }
    }

    void writeNodes() {
      if (node_writers.size() == 0) return;
      if (node_map_writers.size() == 0) {
        throw Exception() << "Missing node id map";
      }
      os << "@nodes" << std::endl;
      for (int i = 0; i < node_writers.size(); ++i) {
        os << node_writers[i].first << '\t';
        node_map_writers[0].second->write(os, node_writers[i].second);
        os << std::endl;
      } 
    }

    void writeEdges() {
      if (edge_writers.size() == 0) return;
      if (edge_map_writers.size() == 0) {
        throw Exception() << "Missing edge id map";
      }
      os << "@edges" << std::endl;
      for (int i = 0; i < edge_writers.size(); ++i) {
        os << edge_writers[i].first << '\t';
        edge_map_writers[0].second->write(os, edge_writers[i].second);
        os << std::endl;
      } 
    }
    
    // Writers

    template <class _Item>
    class WriterBase {
    public:
      typedef _Item Item;
      virtual void write(std::ostream&, const Item&) = 0;
    };

    template <class _Item, typename _Map, typename _Writer>
    class MapWriter : public WriterBase<_Item> {
    public:
      typedef _Map Map;
      typedef _Writer Writer;
      typedef typename Writer::Value Value;
      typedef _Item Item;
      
      const Map& map;
      Writer writer;

      MapWriter(const Map& _map, const Writer& _writer) 
        : map(_map), writer(_writer) {}


      virtual void write(std::ostream& os, const Item& item) {
        Value value;
        writer.write(os, map[item]);
      }

    };



    typedef std::vector< std::pair<std::string, WriterBase<Node>*> > NodeMapWriters;
    NodeMapWriters node_map_writers;

    typedef std::vector< std::pair<std::string, WriterBase<Edge>*> > EdgeMapWriters;
    EdgeMapWriters edge_map_writers;

    typedef std::vector<std::pair<std::string, Node> > NodeWriters;
    NodeWriters node_writers;

    typedef std::vector<std::pair<std::string, Edge> > EdgeWriters;
    EdgeWriters edge_writers;

    std::ostream& os;
    Graph& graph;

  };


}