Scheduler.hpp

Go to the documentation of this file.

/*
    Copyright (C) 2014-2016 Leosac
 
    This file is part of Leosac.
 
    Leosac is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.
 
    Leosac is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    GNU Affero General Public License for more details.
 
    You should have received a copy of the GNU Affero General Public License
    along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
 
#pragma once
 
#include "LeosacFwd.hpp"
#include "core/tasks/GenericTask.hpp"
#include <map>
#include <mutex>
#include <queue>
#include <thread>
 
namespace Leosac
{
 
enum class TargetThread
{
    MAIN,
    POOL,
};
 
class Scheduler
{
  public:
    Scheduler(Kernel *kptr);
 
    Scheduler(const Scheduler &) = delete;
    Scheduler(Scheduler &&)      = delete;
    Scheduler &operator=(const Scheduler &) = delete;
    Scheduler &operator=(Scheduler &&) = delete;
 
    template <typename Callable>
    typename std::enable_if<
        !std::is_convertible<Callable, std::shared_ptr<Tasks::Task>>::value,
        void>::type
    enqueue(const Callable &call, TargetThread policy)
    {
        enqueue(Tasks::GenericTask::build(call), policy);
    }
 
    void enqueue(Tasks::TaskPtr t, TargetThread policy);
 
    void update(TargetThread me) noexcept;
 
    void register_thread(TargetThread me);
 
    Kernel &kernel();
 
  private:
    using TaskQueue    = std::queue<Tasks::TaskPtr>;
    using TaskQueueMap = std::map<TargetThread, TaskQueue>;
 
    TaskQueueMap queues_;
 
    Kernel *kptr_;
    mutable std::mutex mutex_;
};
}