TaskRunner.h

/* Distributed under the Apache License, Version 2.0.
   See accompanying NOTICE file for details.*/

#pragma once

#include "TaskQueue.h"

#include <queue>
#include <memory>
#include <mutex>

//--------------------------------------------------------------------------------------------------
//--------------------------------------------------------------------------------------------------
class CDM_DECL TaskRunner : public TaskQueue
{
public:
    TaskRunner();
    TaskRunner(unsigned int threadCount);
    virtual ~TaskRunner() = default;

    TaskRunner(const TaskRunner& other) = delete;
    TaskRunner& operator=(const TaskRunner& other) = delete;

    void AddTask(std::unique_ptr<Task> task);
    void Run();

    virtual std::unique_ptr<Task> GetNextTask() override;

    void EchoProgress(bool echo) { m_echoPercentComplete = echo; }
    virtual void ReportTaskComplete() override;

protected:
    void PrintProgress();

    mutable std::recursive_mutex m_taskQueueMutex;  
    std::queue<std::unique_ptr<Task>> m_tasks;      

    unsigned int m_threadCount = 1;                 

    bool m_echoPercentComplete = false;             
    size_t m_totalTaskCount = 0;                    
    unsigned int m_tasksComplete = 0;               
};