src/ToGraphvizWriter.cpp

   1 //! \file ToGraphvizWriter.cpp
   2
   3 #include "ToGraphvizWriter.h"
   4
   5 #include <string>
   6 // uh-oh...
   7 #include <regex>
   8
   9 #include "Article.h"
  10
  11 namespace WikiWalker
  12 {
  13   static void writeHeader(std::ostream& os)
  14   {
  15     os << "digraph G {";
  16     os << std::endl;
  17   }
  18
  19   static void writeFooter(std::ostream& os)
  20   {
  21     os << "}";
  22     os << std::endl;
  23   }
  24
  25   /*! \bug When writing an #Article only, attributes are only set on the article
  26    * itself. Attributes won't be written on linked articles.
  27    * However, when writing an #ArticleCollection, all articles are included, so
  28    * all attibutes will be written.
  29    */
  30   static void writeArticle(const Article* a, std::ostream& os)
  31   {
  32     std::string atitle = a->title();
  33
  34     // search for quotes
  35     std::regex re(R"(")");
  36     // replace by escaped quote
  37     atitle = std::regex_replace(atitle, re, R"(\")");
  38
  39     // marked articles are printed as box
  40     if(a->marked()) {
  41       os << R"(")" << atitle << R"(" [shape=box];)" << std::endl;
  42     }
  43
  44     if(!a->analyzed()) {
  45       os << R"(")" << atitle << R"(" [fillcolor=gray,style=filled];)"
  46          << std::endl;
  47     }
  48
  49     // unanalyzed articles are printed greyed out
  50     for(auto al = a->linkBegin(); al != a->linkEnd(); al++) {
  51       auto lck_article = al->lock();
  52       if(lck_article != nullptr) {
  53         std::string alinkedtitle = lck_article->title();
  54         os << R"(")" << atitle << R"(" -> ")"
  55            << std::regex_replace(alinkedtitle, re, R"(\")") << R"(";)"
  56            << std::endl;
  57       }
  58     }
  59   }
  60
  61   void ToGraphvizWriter::output(const Article* a, std::ostream& os)
  62   {
  63     writeHeader(os);
  64
  65     writeArticle(a, os);
  66
  67     writeFooter(os);
  68   }
  69
  70   void ToGraphvizWriter::output(const CollectionUtils::ArticleCollection& ac,
  71                                 std::ostream& os)
  72   {
  73     writeHeader(os);
  74
  75     for(auto a : ac) {
  76       writeArticle(a.second.get(), os);
  77     }
  78
  79     writeFooter(os);
  80   }
  81 }  // namespace WikiWalker