Output.hpp

Go to the documentation of this file.

#pragma once
 
#include "io/Color.hpp"
 
#include <fmt/format.h>
#include <cstdio>
#include <string>
 
namespace ppforest2::io {
  struct Output {
    bool quiet;
    int indent_level = 0;
 
    explicit Output(bool quiet)
        : quiet(quiet) {}
 
    // -- stdout (quiet-aware, indented) ----------------------------------

    template<typename... Args> void println(fmt::format_string<Args...> fmt_str, Args&&... args) const {
      if (!quiet) {
        print_indent();
        fmt::print(fmt_str, std::forward<Args>(args)...);
        fmt::print("\n");
      }
    }

    template<typename... Args> void print(fmt::format_string<Args...> fmt_str, Args&&... args) const {
      if (!quiet) {
        print_indent();
        fmt::print(fmt_str, std::forward<Args>(args)...);
      }
    }

    void newline() const {
      if (!quiet) {
        fmt::print("\n");
      }
    }
 
    // -- indentation -----------------------------------------------------
 
    void indent() { ++indent_level; }
 
    void dedent() {
      if (indent_level > 0) {
        --indent_level;
      }
    }
 
    // -- high-level patterns ---------------------------------------------

    void saved(std::string const& label, std::string const& path) const { println("{} saved to {}", label, path); }

    void progress(int current, int total, int bar_width = 50) const {
      if (quiet) {
        return;
      }
 
      float pct = static_cast<float>(current) / static_cast<float>(total);
      int pos   = static_cast<int>(static_cast<float>(bar_width) * pct);
 
      std::string bar_tpl = style::emphasis("{} |");
      std::string bar     = std::string(pos, '-') + std::string(bar_width - pos, ' ');
 
      if (current == total) {
        bar_tpl = style::success(bar_tpl);
      } else {
        bar_tpl = style::info(bar_tpl);
      }
 
      fmt::print(
          "\r{:{}}" + bar_tpl + " {}/{} ({}%)     ",
          "",
          indent_level * 2,
          bar,
          current,
          total,
          static_cast<int>(pct * 100.0)
      );
      std::fflush(stdout);
 
      if (current == total) {
        fmt::print("\n");
      }
    }
 
  private:
    void print_indent() const {
      if (indent_level > 0) {
        fmt::print("{:{}}", "", indent_level * 2);
      }
    }
  };
}