src/lib/tvgTaskScheduler.h

   1 /*
   2  * Copyright (c) 2020 - 2022 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 <mutex>
  26 #include <condition_variable>
  27 #include "tvgCommon.h"
  28
  29 namespace tvg
  30 {
  31
  32 struct Task;
  33
  34 struct TaskScheduler
  35 {
  36     static unsigned threads();
  37     static void init(unsigned threads);
  38     static void term();
  39     static void request(Task* task);
  40 };
  41
  42 struct Task
  43 {
  44 private:
  45     mutex                   mtx;
  46     condition_variable      cv;
  47     bool                    ready{true};
  48     bool                    pending{false};
  49
  50 public:
  51     virtual ~Task() = default;
  52
  53     void done()
  54     {
  55         if (!pending) return;
  56
  57         unique_lock<mutex> lock(mtx);
  58         while (!ready) cv.wait(lock);
  59         pending = false;
  60     }
  61
  62 protected:
  63     virtual void run(unsigned tid) = 0;
  64
  65 private:
  66     void operator()(unsigned tid)
  67     {
  68         run(tid);
  69
  70         lock_guard<mutex> lock(mtx);
  71         ready = true;
  72         cv.notify_one();
  73     }
  74
  75     void prepare()
  76     {
  77         ready = false;
  78         pending = true;
  79     }
  80
  81     friend class TaskSchedulerImpl;
  82 };
  83
  84
  85
  86 }
  87
  88 #endif //_TVG_TASK_SCHEDULER_H_