FTXUI/ftxui/src/ftxui/screen/screen.cpp

#include "ftxui/dom/node.hpp"
#include "ftxui/screen/screen.hpp"
#include "ftxui/screen/string.hpp"
#include "ftxui/screen/terminal.hpp"

#include <sstream>

namespace ftxui {

namespace {
static const wchar_t* BOLD_SET = L"\e[1m";
static const wchar_t* BOLD_RESET = L"\e[22m"; // Can't use 21 here.

static const wchar_t* DIM_SET = L"\e[2m";
static const wchar_t* DIM_RESET = L"\e[22m";

static const wchar_t* UNDERLINED_SET = L"\e[4m";
static const wchar_t* UNDERLINED_RESET = L"\e[24m";

static const wchar_t* BLINK_SET = L"\e[5m";
static const wchar_t* BLINK_RESET = L"\e[25m";

static const wchar_t* INVERTED_SET = L"\e[7m";
static const wchar_t* INVERTED_RESET = L"\e[27m";

static const char* MOVE_LEFT = "\r";
static const char* MOVE_UP = "\e[1A";
static const char* CLEAR_LINE = "\e[2K";

bool In(const Box& stencil, int x, int y) {
  return stencil.x_min <= x && x <= stencil.x_max &&  //
         stencil.y_min <= y && y <= stencil.y_max;
}

Pixel dev_null_pixel;

} // namespace

Screen::Screen(size_t dimx, size_t dimy)
    : stencil({0, int(dimx) - 1, 0, int(dimy) - 1}),
      dimx_(dimx),
      dimy_(dimy),
      pixels_(dimy, std::vector<Pixel>(dimx)) {}

void UpdatePixelStyle(std::wstringstream& ss, Pixel& previous, Pixel& next) {
  if (next.bold != previous.bold)
    ss << (next.bold ? BOLD_SET : BOLD_RESET);

  if (next.dim != previous.dim)
    ss << (next.dim ? DIM_SET : DIM_RESET);

  if (next.underlined != previous.underlined)
    ss << (next.underlined ? UNDERLINED_SET : UNDERLINED_RESET);

  if (next.blink != previous.blink)
    ss << (next.blink ? BLINK_SET : BLINK_RESET);

  if (next.inverted != previous.inverted)
    ss << (next.inverted ? INVERTED_SET : INVERTED_RESET);

  if (next.foreground_color != previous.foreground_color ||
      next.background_color != previous.background_color) {
    ss << L"\e[" + to_wstring(std::to_string((uint8_t)next.foreground_color)) + L"m";
    ss << L"\e[" + to_wstring(std::to_string(10 + (uint8_t)next.background_color)) + L"m";
  }

  previous = next;
}

std::string Screen::ToString() {
  std::wstringstream ss;

  Pixel previous_pixel;

  for (size_t y = 0; y < dimy_; ++y) {
    if (y != 0)
      ss << '\n';
    for (size_t x = 0; x < dimx_; ++x) {
      UpdatePixelStyle(ss, previous_pixel, pixels_[y][x]);
      ss << pixels_[y][x].character;
    }

  }

  Pixel final_pixel;
  UpdatePixelStyle(ss, previous_pixel, final_pixel);

  return to_string(ss.str());
}

wchar_t& Screen::at(size_t x, size_t y) {
  return PixelAt(x,y).character;
}

Pixel& Screen::PixelAt(size_t x, size_t y) {
  return In(stencil, x, y) ? pixels_[y][x] : dev_null_pixel;
}

// static
Screen Screen::TerminalFullscreen() {
  Terminal::Dimensions size = Terminal::Size();
  return Screen(size.dimx, size.dimy);
}

// static
Screen Screen::TerminalOutput(std::unique_ptr<Node>& element) {
  element->ComputeRequirement();
  Terminal::Dimensions size = Terminal::Size();
  return Screen(size.dimx, element->requirement().min.y);
}

// static
Screen Screen::FitDocument(std::unique_ptr<Node>& element) {
  element->ComputeRequirement();
  Terminal::Dimensions size = Terminal::Size();
  return
    Screen(
      std::min(size.dimx, element->requirement().min.x),
      std::min(size.dimy, element->requirement().min.y)
    );
}

std::string Screen::ResetPosition() {
  std::stringstream ss;
  ss << MOVE_LEFT << CLEAR_LINE;
  for (size_t y = 1; y < dimy_; ++y) {
    ss << MOVE_UP << CLEAR_LINE;
  }
  return ss.str();
}

void Screen::Clear() {
  pixels_ = std::vector<std::vector<Pixel>>(dimy_,
                                            std::vector<Pixel>(dimx_, Pixel()));
}

void Screen::ApplyShader() {

  // Merge box characters togethers.
  for(size_t y = 1; y<dimy_; ++y) {
    for(size_t x = 1; x<dimx_; ++x) {
      wchar_t& left = at(x - 1, y);
      wchar_t& top = at(x, y - 1);
      wchar_t& cur = at(x, y);

      // Left vs current
      if (cur== U'│' && left == U'─')
        cur= U'┤';
      if (cur== U'─' && left == U'│')
        left = U'├';
      if (cur== U'├' && left == U'─')
        cur= U'┼';
      if (cur== U'─' && left == U'┤')
        left = U'┼';

      // Top vs current
      if (cur== U'─' && top == U'│')
        cur= U'┴';
      if (cur== U'│' && top == U'─')
        top = U'┬';
      if (cur== U'┬' && top == U'│')
        cur= U'┼';
      if (cur== U'│' && top == U'┴')
        top = U'┼';
    }
  }
}


};  // namespace ftxui