src/ToGraphvizWriter.cpp

   1 //! \file ToGraphvizWriter.cpp
   2
   3 #include "ToGraphvizWriter.h"
   4
   5 #include <string>
   6
   7 #include "Article.h"
   8
   9 void ToGraphvizWriter::writeHeader(std::ostream& os)
  10 {
  11     os << "digraph G {";
  12     os << std::endl;
  13 }
  14
  15 void ToGraphvizWriter::writeFooter(std::ostream& os)
  16 {
  17     os << "}";
  18     os << std::endl;
  19 }
  20
  21 /*! \bug When writing an #Article only, attributes are only set on the article
  22  * itself. Attributes won't be written on linked articles.
  23  * However, when writing an #ArticleCollection, all articles are included, so
  24  * all attibutes will be written.
  25  */
  26 void ToGraphvizWriter::writeArticle(const Article* a, std::ostream& os)
  27 {
  28     // marked articles are printed as box
  29     if(a->isMarked()) {
  30         os << "\"" << a->getTitle() << "\" [shape=box];" << std::endl;
  31     }
  32
  33     if(!a->isAnalyzed()) {
  34         os << "\"" << a->getTitle() << "\" [fillcolor=gray,style=filled];" << std::endl;
  35     }
  36
  37     // unanalyzed articles are printed greyed out
  38     for(auto al = a->linkBegin(); al != a->linkEnd(); al++) {
  39         os << "\"" << a->getTitle() << "\" -> \"" << (*al)->getTitle() << "\";" << std::endl;
  40     }
  41 }
  42
  43 void ToGraphvizWriter::output(const Article* a, std::ostream& os)
  44 {
  45     writeHeader(os);
  46
  47     writeArticle(a, os);
  48
  49     writeFooter(os);
  50 }
  51
  52 void ToGraphvizWriter::output(const ArticleCollection& ac, std::ostream& os)
  53 {
  54     writeHeader(os);
  55
  56     for(auto a : ac) {
  57         writeArticle(a.second, os);
  58     }
  59
  60     writeFooter(os);
  61 }