progress_reporter.h

#pragma once
 
#include <fstream>
#include <memory>
#include <mutex>
 
namespace FDR
{
typedef int64_t TaskId;
 
class ProgressReporter
{
public:
    ProgressReporter();
    virtual ~ProgressReporter();
 
    enum
    {
        INVALID_TASK = 0,
    };
 
    virtual void activate();
 
    virtual void deactivate();
 
    virtual void task_started(TaskId task_id, TaskId parent_task_id, const std::string& task_name);
 
    virtual void task_finished(TaskId task_id);
 
    virtual void task_message_output(TaskId task_id, const std::string& message);
 
    virtual void task_status_updated(TaskId task_id, const std::string& new_status);
 
private:
    struct Data;
    std::unique_ptr<Data> data;
};
 
class FileProgressReporter : public ProgressReporter
{
public:
    FileProgressReporter(const std::string& file_name);
 
    void activate() override;
    void deactivate() override;
    void task_started(TaskId task_id, TaskId parent_task_id, const std::string& task_name) override;
    void task_finished(TaskId task_id) override;
    void task_message_output(TaskId task_id, const std::string& message) override;
    void task_status_updated(TaskId task_id, const std::string& new_status) override;
 
private:
    std::mutex mutex;
    std::ofstream stream;
};
 
class StreamProgressReporter : public ProgressReporter
{
public:
    enum class StreamIdentifier
    {
        StdOut,
        StdErr,
    };
 
    StreamProgressReporter(StreamIdentifier output_stream_identifier);
 
    StreamProgressReporter(std::ostream& output_stream);
 
    void activate() override;
    void deactivate() override;
    void task_started(TaskId task_id, TaskId parent_task_id, const std::string& task_name) override;
    void task_finished(TaskId task_id) override;
    void task_message_output(TaskId task_id, const std::string& message) override;
    void task_status_updated(TaskId task_id, const std::string& new_status) override;
 
private:
    std::mutex mutex;
    std::ostream& stream;
};
 
}  // end FDR