Output.hpp

Go to the documentation of this file.

#pragma once
 
#include "io/Color.hpp"
 
#include <fmt/format.h>
#include <cstdio>
#include <functional>
#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");
      }
    }

    void flush() const { std::fflush(stdout); }
 
    // -- stderr (always prints, no indent) -------------------------------

    template<typename... Args> void errorln(fmt::format_string<Args...> fmt_str, Args&&... args) const {
      fmt::print(stderr, fmt_str, std::forward<Args>(args)...);
      fmt::print(stderr, "\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) / total;
      int pos   = static_cast<int>(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");
      }
    }

    int try_or_fail(std::function<void()> const& f, std::string const& context = "") const {
      try {
        f();
        return 0;
      } catch (std::exception const& e) {
        if (context.empty()) {
          errorln("{} {}", style::error("Error:"), e.what());
        } else {
          errorln("{} {}: {}", style::error("Error:"), context, e.what());
        }
 
        return 1;
      }
    }
 
  private:
    void print_indent() const {
      if (indent_level > 0) {
        fmt::print("{:{}}", "", indent_level * 2);
      }
    }
  };
}