text_wrap.cpp

#include <cstdarg>
#include <vector>
#include <string>
#include <sstream>
#include "esphome/components/display/display_buffer.h"
#include "esphome/components/font/font.h"
#include "text_wrap.h"

using namespace esphome;
using namespace esphome::display;

void wrap_text_box2(Display *it, int x, int y, int width, int height, BaseFont *font, TextAlign align,
                    float line_height, const char *format, ...) {
  va_list arg;
  va_start(arg, format);

  // Calculate the bounds of a single space character
  int space_x1, space_y1, space_width, space_height;
  it->get_text_bounds(x, y, " ", font, align, &space_x1, &space_y1, &space_width, &space_height);

  char buffer[256];
  int ret = vsnprintf(buffer, sizeof(buffer), format, arg);
  if (ret == 0)
    return;

  // Break text into words
  std::vector<std::string> words;
  std::stringstream ss(buffer);
  std::string word;
  while (std::getline(ss, word, ' ')) {
    words.push_back(word);
  }

  // Initialize variables for the wrapped text
  int line_x = x;
  int line_y = y;
  int x_max = x + width;
  int y_max = y + height;
  int word_x1, word_y1, word_width, word_height;
  bool last = false;
  std::string line;

  // Iterate through the words and wrap them
  for (const auto &w : words) {
    it->get_text_bounds(line_x, line_y, w.c_str(), font, align, &word_x1, &word_y1, &word_width, &word_height);

    if (line_x + word_width >= x_max) {
      // If the next line would overspill the height box, get ready to stop.
      if (line_y + 2 * static_cast<int>(word_height * line_height) > y_max) {
        last = true;
        size_t length = line.length();
        const std::string cont = "...";
        if (length >= cont.length()) {
          line.replace(length - cont.length(), cont.length(), cont);
        }
      }
      // Print the current line and move to the next line
      it->printf(x, line_y, font, align, "%s", line.c_str());
      if (last) {
        break;
      }
      line_y += static_cast<int>(word_height * line_height);
      line_x = x;
      // Clear the line buffer
      line.clear();
    }

    // Add the word to the line buffer and move the cursor
    line += w;
    line_x += word_width + space_width;

    // If it's not the last word, add a space
    if (!line.empty() && &w != &words.back()) {
      line += " ";
    }
  }

  // Print the last line
  if (!line.empty()) {
    it->printf(x, line_y, font, align, "%s", line.c_str());
  }
}