Progress.hpp

#ifndef BKCRACK_PROGRESS_HPP
#define BKCRACK_PROGRESS_HPP
 
#include <atomic>
#include <concepts>
#include <iostream>
#include <mutex>

class Progress
{
public:
    enum class State
    {
        Normal,   
        Canceled, 
        EarlyExit 
    };

    explicit Progress(std::ostream& os);

    virtual ~Progress() = default;

    template <std::invocable<std::ostream&> F>
    void log(F f)
    {
        const auto lock = std::scoped_lock{m_os_mutex};
        beforeLog(m_os);
        f(m_os);
    }
 
    std::atomic<State> state = State::Normal; 
    std::atomic<int>   done  = 0;             
    std::atomic<int>   total = 0;             
 
protected:
    virtual void beforeLog(std::ostream&) {}
 
    std::mutex    m_os_mutex;
    std::ostream& m_os;
};
 
#endif // BKCRACK_PROGRESS_HPP