HTML.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#include <framework/utilities/HTML.h>
 
#include <boost/algorithm/string/replace.hpp>
#include <regex>
 
#include <TMatrixFBase.h>
#include <TMatrixD.h>
#include <TVector3.h>
 
#include <framework/gearbox/Unit.h>
 
#include <iomanip>
#include <sstream>
 
using namespace Belle2;
using namespace boost::algorithm;
 
std::string HTML::getString(const TMatrixFBase& matrix, int precision, bool color)
{
  const int nr = matrix.GetNrows();
  const int nc = matrix.GetNcols();
 
  TMatrixD matrix_double(nr, nc);
  for (int ir = 0; ir < nr; ir++)
    for (int ic = 0; ic < nc; ic++)
      matrix_double(ir, ic) = matrix(ir, ic);
  return getString(matrix_double, precision, color);
}
 
std::string HTML::getString(const TMatrixDBase& matrix, int precision, bool color)
{
  std::stringstream stream;
  stream.precision(precision);
 
  stream << "<table padding=0>";
  const double max = matrix.Max();
  const double min = matrix.Min();
  for (int i = 0; i < matrix.GetNrows(); i++) {
    stream << "<tr>";
    for (int k = 0; k < matrix.GetNcols(); k++) {
      const double value = matrix(i, k);
      stream << "<td align=right";
      if (color and value != 0.0) {
        //0 white, min cyan, max red
        int b = 255;
        int r = 255;
        int g = 255;
        if (value > 0)
          b = g = 255 - 180 * value / max;
        else
          r = 255 - 200 * value / min;
 
        stream << " bgcolor=" << std::setfill('0') << std::hex;
        stream << std::setw(2) << r;
        stream << std::setw(2) << g;
        stream << std::setw(2) << b;
        stream << std::dec << std::setfill(' ');
      }
      stream << ">" << std::setw(11) << value;
      stream << "</td>";
    }
    stream << "</tr>";
  }
  stream << "</table>";
  stream << "<br>";
 
  return stream.str();
}
 
std::string HTML::getString(const TVector3& vec, int precision)
{
  std::stringstream stream;
  stream.precision(precision);
 
  stream << std::fixed << "(" << vec.x() << ", " <<  vec.y() << ", " <<  vec.z() << ")";
  return stream.str();
}
 
std::string HTML::getStringConvertToUnit(const TVector3& vec, int precision, const std::string& unitType)
{
  std::stringstream stream;
  stream.precision(precision);
  std::string displayedUnitType = unitType;
 
  if (unitType == "um")
    displayedUnitType = "\x0b5m";
 
 
  stream << std::fixed << "(" << Unit::convertValueToUnit(vec.x(), unitType) << ", "
         <<  Unit::convertValueToUnit(vec.y(), unitType) << ", " <<  Unit::convertValueToUnit(vec.z(), unitType)
         << ") " << displayedUnitType;
 
 
  return stream.str();
}
std::string HTML::chooseUnitOfLength(const TVector3& vec)
{
  double xyz [3];
  std::string unitType;
  vec.GetXYZ(xyz);
  double max = 0;
 
  for (auto entry : xyz)
    if (std::abs(entry) > max)
      max = std::abs(entry);
 
  // choose specific range for that the unit is useful
  if (max < 0.1)
    unitType = "um";
  else
    unitType = "cm";
  return unitType;
}
 
std::string HTML::getHexDump(const int* buf, int length)
{
  std::stringstream stream;
  stream << "<small><tt>";
  char str[10];
  //print everything in blocks of 4 bytes, 4 blocks per line
  for (int i = 0; i < length; i++) {
    //byte = 2letters
    //4byte = 8 letters
    snprintf(str, sizeof(str), "%08x ", buf[i]);
    stream << str;
  }
  stream << "</tt></small>";
  return stream.str();
}
 
 
std::string HTML::htmlToPlainText(const std::string& html)
{
  std::string tmp = html;
  //conversions to plaintext
  replace_all(tmp, "<br>", "\n");
  replace_all(tmp, "</p>", "\n");
  replace_all(tmp, "</tr>", "\n");
  replace_all(tmp, "&nbsp;", " ");
 
  //remove non-ascii \mu
  replace_all(tmp, "\x0b5", "u");
 
  //remove all unknown tags (non-greedy match)
  const static std::regex tagRegex("<.*?>");
  tmp = std::regex_replace(tmp, tagRegex, "");
 
  tmp = unescape(tmp);
 
  return tmp;
}
 
std::string HTML::escape(const std::string& str)
{
  std::string tmp = str;
  replace_all(tmp, "&", "&amp;"); //must be first
  replace_all(tmp, ">", "&gt;");
  replace_all(tmp, "<", "&lt;");
  replace_all(tmp, "\"", "&quot;");
  return tmp;
}
 
std::string HTML::unescape(const std::string& str)
{
  std::string tmp = str;
  //replace entities (important ones at least)
  replace_all(tmp, "&quot;", "\"");
  replace_all(tmp, "&gt;", ">");
  replace_all(tmp, "&lt;", "<");
  replace_all(tmp, "&amp;", "&"); //must be last
  return tmp;
}