src/lib/tvgTaskScheduler.h

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All rights reserved.
   3
   4  * Permission is hereby granted, free of charge, to any person obtaining a copy
   5  * of this software and associated documentation files (the "Software"), to deal
   6  * in the Software without restriction, including without limitation the rights
   7  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   8  * copies of the Software, and to permit persons to whom the Software is
   9  * furnished to do so, subject to the following conditions:
  10
  11  * The above copyright notice and this permission notice shall be included in all
  12  * copies or substantial portions of the Software.
  13
  14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  20  * SOFTWARE.
  21  */
  22 #ifndef _TVG_TASK_SCHEDULER_H_
  23 #define _TVG_TASK_SCHEDULER_H_
  24
  25 #include "tvgCommon.h"
  26
  27 namespace tvg
  28 {
  29
  30 struct Task
  31 {
  32 private:
  33     std::promise<void> sender;
  34     std::future<void>  receiver;
  35
  36 public:
  37     virtual ~Task() = default;
  38
  39     void get()
  40     {
  41         if (receiver.valid()) receiver.get();
  42     }
  43
  44 protected:
  45     virtual void run() = 0;
  46
  47 private:
  48     void operator()()
  49     {
  50         run();
  51         sender.set_value();
  52     }
  53
  54     void prepare()
  55     {
  56         sender = std::promise<void>();
  57         receiver = sender.get_future();
  58     }
  59
  60     friend class TaskSchedulerImpl;
  61 };
  62
  63 struct TaskScheduler
  64 {
  65     static void init(unsigned threads);
  66     static void term();
  67     static void request(shared_ptr<Task> task);
  68 };
  69
  70 }
  71
  72 #endif //_TVG_TASK_SCHEDULER_H_