ProgressMonitor.hh

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <chrono>
#include <stack>
#include <map>
#include <vector>
#include <functional>
 
 
class ProgressMonitor {
    public:
        ProgressMonitor(std::function<void(const std::string&, int, int)> report = nullptr, int report_level = 2);
        virtual ~ProgressMonitor();
 
        void group(std::string name = "", int total = 0, int level = -1);
 
        void progress(); // Increment current step by 1
        void progress(int n);
        void progress(int n, int total);
 
 
        void message(const std::string&);
        
        
        void print() const;
 
        std::function<void(const std::string&, int, int)> report;
 
        // Level above which to report progress updates and not only add to totals
        int report_level;
 
        
        class Total {
            public:
                Total();
 
                std::string               name;
                size_t                    call_count;
                std::chrono::milliseconds time_spent;
                int                       total_steps;
                std::vector<std::string>  messages;
 
                long                      time_spent_as_long() const;
 
                bool operator==(const Total& other) const;
 
                std::string               str() const;
            };
 
        std::vector<Total> totals() const;
 
    private:
        
        class Block {
            public:
                Block(const std::string& name, int level);
 
                std::string               name;
                std::chrono::milliseconds started;
                int                       step, total_steps;
                std::vector<std::string>  messages;
                int                       level;
            };
 
        std::stack<Block>            call_stack;
        std::map<std::string, Total> call_totals;
    };
 
 
class ScopedProgressGroup
{
public:
    ScopedProgressGroup(ProgressMonitor* pm, const std::string& name, int total = 0, int level = -1);
    ~ScopedProgressGroup();
 
    void progress();
    void progress(int n);
    void progress(int n, int total);
 
private:
    ProgressMonitor* pm;
};