tvg: added task support that runs on a threadpool.

this patch adds an async() function that takes a shared_task
and runs asyncronously in a threadpool.

Change-Id: I02a47df6938656828f924fbf5e2bc075073b329b
This commit is contained in:
Subhransu Mohanty 2020-08-13 13:01:53 +09:00
parent 9dd3cc5edc
commit 275c2f9d79
3 changed files with 229 additions and 0 deletions

View file

@ -35,6 +35,7 @@ source_file = [
'tvgScene.cpp',
'tvgShape.cpp',
'tvgSwCanvas.cpp',
'tvgTask.cpp',
]
common_dep = declare_dependency(

156
src/lib/tvgTask.cpp Normal file
View file

@ -0,0 +1,156 @@
/*
* Copyright (c) 2020 Samsung Electronics Co., Ltd All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
#include <tvgTask.h>
#include <deque>
template <typename Task>
class TaskQueue {
using lock_t = std::unique_lock<std::mutex>;
std::deque<Task> _q;
bool _done{false};
std::mutex _mutex;
std::condition_variable _ready;
public:
bool try_pop(Task &task)
{
lock_t lock{_mutex, std::try_to_lock};
if (!lock || _q.empty()) return false;
task = std::move(_q.front());
_q.pop_front();
return true;
}
bool try_push(Task &&task)
{
{
lock_t lock{_mutex, std::try_to_lock};
if (!lock) return false;
_q.push_back(std::move(task));
}
_ready.notify_one();
return true;
}
void done()
{
{
lock_t lock{_mutex};
_done = true;
}
_ready.notify_all();
}
bool pop(Task &task)
{
lock_t lock{_mutex};
while (_q.empty() && !_done) _ready.wait(lock);
if (_q.empty()) return false;
task = std::move(_q.front());
_q.pop_front();
return true;
}
void push(Task &&task)
{
{
lock_t lock{_mutex};
_q.push_back(std::move(task));
}
_ready.notify_one();
}
};
#include <thread>
#include <vector>
namespace tvg
{
class Executor
{
const unsigned _count{std::thread::hardware_concurrency()};
std::vector<std::thread> _threads;
std::vector<TaskQueue<shared_task>> _q{_count};
std::atomic<unsigned> _index{0};
void run(unsigned i)
{
// Task Loop
shared_task task;
while (true) {
bool success = false;
for (unsigned n = 0; n != _count * 2; ++n) {
if (_q[(i + n) % _count].try_pop(task)) {
success = true;
break;
}
}
if (!success && !_q[i].pop(task)) break;
(*task)();
}
}
Executor()
{
for (unsigned n = 0; n != _count; ++n) {
_threads.emplace_back([&, n] { run(n); });
}
}
~Executor()
{
for (auto &e : _q) e.done();
for (auto &e : _threads) e.join();
}
public:
static Executor& instance() {
static Executor singleton;
return singleton;
}
void post(shared_task task)
{
task->prepare();
auto i = _index++;
for (unsigned n = 0; n != _count; ++n) {
if (_q[(i + n) % _count].try_push(std::move(task))) return;
}
if (_count > 0) {
_q[i % _count].push(std::move(task));
}
}
};
void async(shared_task task)
{
Executor::instance().post(std::move(task));
}
}

72
src/lib/tvgTask.h Normal file
View file

@ -0,0 +1,72 @@
/*
* Copyright (c) 2020 Samsung Electronics Co., Ltd All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
#ifndef _TVG_TASK_H_
#define _TVG_TASK_H_
#include <memory>
#include <future>
namespace tvg
{
/*
Task Interface.
Able to run a task in the thread pool. derive from the
task interface and implement run method.
To get the result call task->get() which will return immidiately if the
task is already finishd otherwise will wait till task completion.
*/
class Task
{
public:
virtual ~Task() = default;
void get() { if (_receiver.valid()) _receiver.get(); }
protected:
virtual void run() = 0;
private:
void operator()()
{
run();
_sender.set_value();
}
void prepare()
{
_sender = std::promise<void>();
_receiver = _sender.get_future();
}
friend class Executor;
std::promise<void> _sender;
std::future<void> _receiver;
};
using shared_task = std::shared_ptr<Task>;
/*
async() function takes a shared task and runs it in
a thread pool asyncronously. call get() on the shared_task
to get the ressult out of the shared_task.
*/
void async(shared_task task);
}
#endif //_TVG_TASK_H_