logging.hpp

Go to the documentation of this file.

#pragma once
#include <string>
#include <iostream>
#include <iomanip>
 
namespace ppl {
namespace util {
 
struct ProgressLogger 
{
 
    ProgressLogger(size_t max, 
                   const std::string& name, 
                   std::ostream& os = std::cout)
        : max_(max)
        , name_(name)
        , os_(os)
    {};
 
    ~ProgressLogger() { os_ << std::endl; }
 
    void printProgress(size_t step) {
        ++step;
        size_t denom = max_ <= 100 ? max_ : 100;
        if (step % (max_ / denom) == 0) {
            int percent = static_cast<double>(step * 100) /
                            static_cast<double>(max_);
            os_ << '\r' << name_ << " ["
                << std::string(percent, '=')
                << std::string(100 - percent, ' ')
                << "] (" << std::setw(2)
                << percent
                << "%)" << std::flush;
        }
    }
 
private:
    size_t max_;        // maximum vaue to count progress towards
    std::string name_;  // name of the algorithm being measured, printed with progress bar
    std::ostream& os_;  // output stream to print to
};
 
} // namespace util
} // namespace ppl