html_utils.h

Go to the documentation of this file.

#pragma once
#include <iomanip> // precision
#include <sstream>
#include <stdexcept>
#include <string>

struct Colors {
  static constexpr const char *red = "\"#DC143C\"";
  static constexpr const char *green = "\"#7FFF00\"";
  static constexpr const char *blue = "\"#4169E1\"";
  static constexpr const char *grey = "\"#D3D3D3\"";
  static constexpr const char *yellow = "\"#FFFF94\"";
  static constexpr const char *white = "\"#FFF\"";
  static constexpr const char *black = "\"#000\"";
};

class HtmlDivisionWriter {
public:
  HtmlDivisionWriter() {
    division_string_stream << std::fixed << std::setprecision(2);
    division_string_stream << "<div>";
  }
  void addHeader(std::string header, int level = 1) {
    std::string h_name = "h" + std::to_string(level);
    division_string_stream << "<" << h_name << " align=\"center\" >" << header
                           << "</" << h_name << ">";
  }

  void addText(std::string text) {
    division_string_stream << text << std::endl;
  }

  std::string getDivisionText() {
    return division_string_stream.str() + "</div>";
  }

private:
  std::stringstream division_string_stream; 
};

class HtmlTableWriter {
public:
  HtmlTableWriter(int width = 120) : width_(width), row_ended(true) {
    table_string_stream << std::fixed << std::setprecision(2);
    table_string_stream << "<table border=\"1\">";
  }

  template <typename DataT>
  void addCell(const DataT &data, std::string header = "",
               std::string bg_color = Colors::white, int colspan = 1) {
    if (row_ended) {
      throw std::runtime_error("Cannot add cell without beginning row");
    }
    std::string width_text = "\"" + std::to_string(colspan * width_) + "\"";
    table_string_stream << "<td width=" << width_text << " colspan=\""
                        << std::to_string(colspan) << "\" bgcolor=" << bg_color
                        << " >";
    if (!header.empty()) {
      table_string_stream << header << ": ";
    }
    table_string_stream << data;
    table_string_stream << "</td>";
  }

  void beginRow() {
    if (!row_ended) {
      endRow();
    }
    table_string_stream << "<tr>";
    row_ended = false;
  }

  void addHeader(std::string header, std::string text_color = Colors::black,
                 int colspan = 1) {
    if (row_ended) {
      throw std::runtime_error("Cannot add cell without beginning row");
    }
    std::string width_text = "\"" + std::to_string(colspan * width_) + "\"";
    table_string_stream << "<th width=" << width_text << " colspan=\""
                        << std::to_string(colspan) << "\" >";
    table_string_stream << "<font color=" << text_color << ">" << header
                        << "</font>";
    table_string_stream << "</th>";
  }

  std::string getTableString() {
    if (!row_ended) {
      endRow();
    }
    return table_string_stream.str() + "</table>";
  }

private:
  std::stringstream table_string_stream; 
  int width_;                            
  bool row_ended;                        

  void endRow() {
    row_ended = true;
    table_string_stream << "</tr>";
  }
};