runner.h

Go to the documentation of this file.

// Copyright (c) 2021, Aalto University, developed by Erik Schultheis
// All rights reserved.
//
// SPDX-License-Identifier: MIT
 
#ifndef DISMEC_RUNNER_H
#define DISMEC_RUNNER_H
 
#include <functional>
#include <memory>
#include <chrono>
#include "spdlog/spdlog.h"
 
namespace dismec::parallel {
    class TaskGenerator;
 
    struct RunResult {
        bool IsFinished = false;        
        long NextTask = -1;             
        // timing info
        std::chrono::seconds Duration;  
    };
 
    class ParallelRunner {
    public:
        explicit ParallelRunner(long num_threads, long chunk_size=1);
 
        void set_chunk_size(long chunk_size);
        void set_time_limit(std::chrono::milliseconds time_limit);
 
        void set_logger(std::shared_ptr<spdlog::logger> logger);
 
        [[nodiscard]] RunResult run(TaskGenerator& tasks, long start=0);
 
    private:
 
        // helpers
        void log_start(long begin, long end);
        void log_finished(long begin, long end);
 
        long m_NumThreads;
        long m_ChunkSize = 1;
        std::chrono::milliseconds m_TimeLimit;
        std::shared_ptr<spdlog::logger> m_Logger;
 
        bool m_BindThreads = true;
    };
}
 
#endif //DISMEC_RUNNER_H