src/base/closure.h

@@ -12,11 +12,12 @@
 
 // Helper for logging location info, e.g. LOG << LOCATION(from)
 #define LOCATION(from)                                                 \
-  std::get<0>(from) << "() [" << [](std::string path) -> std::string { \
-    size_t last_slash_pos = path.find_last_of("\\/");                  \
+  std::get<0>(from) << "() [" << [](const char* path) -> std::string { \
+    std::string file_name(path);                                       \
+    size_t last_slash_pos = file_name.find_last_of("\\/");             \
     if (last_slash_pos != std::string::npos)                           \
-      path = path.substr(last_slash_pos + 1);                          \
-    return path;                                                       \
+      file_name = file_name.substr(last_slash_pos + 1);                \
+    return file_name;                                                  \
   }(std::get<1>(from)) << ":"                                          \
                        << std::get<2>(from) << "]"
 
@@ -35,7 +36,7 @@ using Closure = std::function<void()>;
 
 // Provides location info (function name, file name and line number) where of a
 // Closure was constructed.
-using Location = std::tuple<std::string, std::string, int>;
+using Location = std::tuple<const char*, const char*, int>;
 
 #else
 

src/base/task_runner.cc

@@ -4,14 +4,12 @@
 
 #include "base/log.h"
 
-namespace base {
-
 namespace {
 
-void PostTaskAndReplyRelay(Location from,
-                           Closure task_cb,
-                           Closure reply_cb,
-                           std::shared_ptr<TaskRunner> destination) {
+void PostTaskAndReplyRelay(base::Location from,
+                           base::Closure task_cb,
+                           base::Closure reply_cb,
+                           base::TaskRunner* destination) {
   task_cb();
 
   if (reply_cb)
@@ -20,23 +18,21 @@ void PostTaskAndReplyRelay(Location from,
 
 }  // namespace
 
-// The task runner that belongs to the thread it's created in. Tasks to be run
-// on a specific thread can be posted to this task runner.
-// TaskRunner::GetThreadLocalTaskRunner()->RunTasks() is expected to be
-// periodically called.
-thread_local std::shared_ptr<TaskRunner> TaskRunner::thread_local_task_runner;
+namespace base {
+
+thread_local std::unique_ptr<TaskRunner> TaskRunner::thread_local_task_runner;
 
 void TaskRunner::CreateThreadLocalTaskRunner() {
   DCHECK(!thread_local_task_runner);
 
-  thread_local_task_runner = std::make_shared<TaskRunner>();
+  thread_local_task_runner = std::make_unique<TaskRunner>();
 }
 
-std::shared_ptr<TaskRunner> TaskRunner::GetThreadLocalTaskRunner() {
-  return thread_local_task_runner;
+TaskRunner* TaskRunner::GetThreadLocalTaskRunner() {
+  return thread_local_task_runner.get();
 }
 
-void TaskRunner::PostTask(Location from, Closure task) {
+void TaskRunner::PostTask(const Location& from, Closure task) {
   DCHECK(task) << LOCATION(from);
 
   task_count_.fetch_add(1, std::memory_order_relaxed);
@@ -44,28 +40,19 @@ void TaskRunner::PostTask(Location from, Closure task) {
   queue_.emplace_back(from, std::move(task));
 }
 
-void TaskRunner::PostTaskAndReply(Location from, Closure task, Closure reply) {
+void TaskRunner::PostTaskAndReply(const Location& from,
+                                  Closure task,
+                                  Closure reply) {
   DCHECK(task) << LOCATION(from);
   DCHECK(reply) << LOCATION(from);
   DCHECK(thread_local_task_runner) << LOCATION(from);
 
-  auto relay = std::bind(PostTaskAndReplyRelay, from, std::move(task),
-                         std::move(reply), thread_local_task_runner);
+  auto relay = std::bind(::PostTaskAndReplyRelay, from, std::move(task),
+                         std::move(reply), thread_local_task_runner.get());
   PostTask(from, std::move(relay));
 }
 
-void TaskRunner::CancelTasks() {
-  std::lock_guard<std::mutex> scoped_lock(lock_);
-  task_count_.fetch_sub(queue_.size(), std::memory_order_release);
-  queue_.clear();
-}
-
-void TaskRunner::WaitForCompletion() {
-  while (task_count_.load(std::memory_order_acquire) > 0)
-    std::this_thread::yield();
-}
-
-void TaskRunner::RunTasks() {
+void TaskRunner::MultiConsumerRun() {
   for (;;) {
     Task task;
     {
@@ -84,7 +71,55 @@ void TaskRunner::RunTasks() {
 
     task_cb();
     task_count_.fetch_sub(1, std::memory_order_release);
+
+    if (cancel_tasks_.load(std::memory_order_relaxed)) {
+      CancelTasksInternal();
+      break;
+    }
   }
 }
 
+void TaskRunner::SingleConsumerRun() {
+  std::deque<Task> queue;
+  {
+    std::lock_guard<std::mutex> scoped_lock(lock_);
+    if (queue_.empty())
+      return;
+    queue.swap(queue_);
+  }
+
+  while (!queue.empty()) {
+    auto [from, task_cb] = queue.front();
+    queue.pop_front();
+
+#if 0
+    LOG << __func__ << " from: " << LOCATION(from);
+#endif
+
+    task_cb();
+    task_count_.fetch_sub(1, std::memory_order_release);
+
+    if (cancel_tasks_.load(std::memory_order_relaxed)) {
+      CancelTasksInternal();
+      break;
+    }
+  }
+}
+
+void TaskRunner::CancelTasks() {
+  cancel_tasks_.store(true, std::memory_order_relaxed);
+}
+
+void TaskRunner::WaitForCompletion() {
+  while (task_count_.load(std::memory_order_acquire) > 0)
+    std::this_thread::yield();
+}
+
+void TaskRunner::CancelTasksInternal() {
+  cancel_tasks_.store(false, std::memory_order_relaxed);
+  task_count_.store(0, std::memory_order_relaxed);
+  std::lock_guard<std::mutex> scoped_lock(lock_);
+  queue_.clear();
+}
+
 }  // namespace base

src/base/task_runner.h

@@ -26,7 +26,7 @@ void ReturnAsParamAdapter(std::function<ReturnType()> func,
 template <typename ReturnType>
 void ReplyAdapter(std::function<void(ReturnType)> callback,
                   ReturnType* result) {
-  callback(std::move(*result));
+  callback(*result);
   delete result;
 }
 
@@ -34,23 +34,20 @@ void ReplyAdapter(std::function<void(ReturnType)> callback,
 
 // Runs queued tasks (in the form of Closure objects). All methods are
 // thread-safe and can be called on any thread.
-// Tasks run in FIFO order when consumed by a single thread. When consumed
-// concurrently by multiple threads, it doesn't guarantee whether tasks overlap,
-// or whether they run on a particular thread.
+// Tasks run in FIFO order. When consumed concurrently by multiple threads, it
+// doesn't guarantee whether tasks overlap, or whether they run on a particular
+// thread.
 class TaskRunner {
  public:
   TaskRunner() = default;
   ~TaskRunner() = default;
 
-  static void CreateThreadLocalTaskRunner();
-  static std::shared_ptr<TaskRunner> GetThreadLocalTaskRunner();
+  void PostTask(const Location& from, Closure task);
 
-  void PostTask(Location from, Closure task);
-
-  void PostTaskAndReply(Location from, Closure task, Closure reply);
+  void PostTaskAndReply(const Location& from, Closure task, Closure reply);
 
   template <typename ReturnType>
-  void PostTaskAndReplyWithResult(Location from,
+  void PostTaskAndReplyWithResult(const Location& from,
                                   std::function<ReturnType()> task,
                                   std::function<void(ReturnType)> reply) {
     auto* result = new ReturnType;
@@ -62,11 +59,14 @@ class TaskRunner {
                   result));
   }
 
-  void CancelTasks();
+  void MultiConsumerRun();
+  void SingleConsumerRun();
 
+  void CancelTasks();
   void WaitForCompletion();
 
-  void RunTasks();
+  static void CreateThreadLocalTaskRunner();
+  static TaskRunner* GetThreadLocalTaskRunner();
 
  private:
   using Task = std::tuple<Location, Closure>;
@@ -74,8 +74,11 @@ class TaskRunner {
   std::deque<Task> queue_;
   mutable std::mutex lock_;
   std::atomic<size_t> task_count_{0};
+  std::atomic<bool> cancel_tasks_{false};
 
-  static thread_local std::shared_ptr<TaskRunner> thread_local_task_runner;
+  static thread_local std::unique_ptr<TaskRunner> thread_local_task_runner;
+
+  void CancelTasksInternal();
 
   TaskRunner(TaskRunner const&) = delete;
   TaskRunner& operator=(TaskRunner const&) = delete;

src/base/thread_pool.cc

@@ -40,31 +40,30 @@ void ThreadPool::Shutdown() {
   threads_.clear();
 }
 
-void ThreadPool::PostTask(Location from, Closure task) {
+void ThreadPool::PostTask(const Location& from, Closure task) {
   DCHECK((!threads_.empty()));
 
   task_runner_.PostTask(from, std::move(task));
   semaphore_.release();
 }
 
-void ThreadPool::PostTaskAndReply(Location from, Closure task, Closure reply) {
+void ThreadPool::PostTaskAndReply(const Location& from,
+                                  Closure task,
+                                  Closure reply) {
   DCHECK((!threads_.empty()));
 
   task_runner_.PostTaskAndReply(from, std::move(task), std::move(reply));
   semaphore_.release();
 }
 
-void ThreadPool::CancelTasks() {
-  task_runner_.CancelTasks();
-}
-
 void ThreadPool::WorkerMain() {
   for (;;) {
     semaphore_.acquire();
+
     if (quit_.load(std::memory_order_relaxed))
       return;
 
-    task_runner_.RunTasks();
+    task_runner_.MultiConsumerRun();
   }
 }
 

src/base/thread_pool.h

@@ -11,6 +11,8 @@
 
 namespace base {
 
+class TaskRunner;
+
 // Feed the ThreadPool tasks (in the form of Closure objects) and they will be
 // called on any thread from the pool.
 class ThreadPool {
@@ -24,12 +26,12 @@ class ThreadPool {
 
   void Shutdown();
 
-  void PostTask(Location from, Closure task);
+  void PostTask(const Location& from, Closure task);
 
-  void PostTaskAndReply(Location from, Closure task, Closure reply);
+  void PostTaskAndReply(const Location& from, Closure task, Closure reply);
 
   template <typename ReturnType>
-  void PostTaskAndReplyWithResult(Location from,
+  void PostTaskAndReplyWithResult(const Location& from,
                                   std::function<ReturnType()> task,
                                   std::function<void(ReturnType)> reply) {
     task_runner_.PostTaskAndReplyWithResult(from, std::move(task),
@@ -37,8 +39,6 @@ class ThreadPool {
     semaphore_.release();
   }
 
-  void CancelTasks();
-
  private:
   std::vector<std::thread> threads_;
 

src/demo/menu.cc

@@ -201,6 +201,7 @@ bool Menu::Initialize() {
                                          : RendererType::kOpenGL);
         renderer_type_.SetEnabled(
             (Engine::Get().GetRendererType() == RendererType::kVulkan));
+        Engine::Get().ConsumeInputEvents();
       },
       true, Engine::Get().GetRendererType() == RendererType::kVulkan,
       kColorFadeOut, {Vector4f{1, 1, 1, 1}, Vector4f{1, 1, 1, 1}});

src/engine/audio/audio_mixer.h

@@ -16,6 +16,7 @@ class TaskRunner;
 
 namespace eng {
 
+class AudioSink;
 class AudioBus;
 
 // Mix and render audio with low overhead. A platform specific AudioSink
@@ -84,13 +85,13 @@ class AudioMixer : public AudioSink::Delegate {
 
   std::list<std::shared_ptr<Resource>> end_list_;
 
-  std::shared_ptr<base::TaskRunner> main_thread_task_runner_;
+  base::TaskRunner* main_thread_task_runner_;
 
   std::unique_ptr<AudioSink> audio_sink_;
 
   bool audio_enabled_ = true;
 
-  // AudioSink::Delegate interface
+  // AudioSink::Delegate implementation
   int GetChannelCount() final { return kChannelCount; }
   void RenderAudio(float* output_buffer, size_t num_frames) final;
 

src/engine/engine.cc

@@ -50,9 +50,6 @@ Engine::Engine(Platform* platform)
 Engine::~Engine() {
   LOG << "Shutting down engine.";
 
-  thread_pool_.CancelTasks();
-  thread_pool_.Shutdown();
-
   game_.reset();
   stats_.reset();
   textures_.clear();
@@ -76,7 +73,7 @@ void Engine::Run() {
   float frame_frac = 0.0f;
 
   for (;;) {
-    TaskRunner::GetThreadLocalTaskRunner()->RunTasks();
+    TaskRunner::GetThreadLocalTaskRunner()->SingleConsumerRun();
 
     platform_->Update();
     if (platform_->should_exit())
@@ -293,13 +290,11 @@ void Engine::RefreshImage(const std::string& asset_name) {
     return;
   }
 
-  if (it->second.persistent || it->second.use_count > 0) {
   auto image = it->second.create_image();
   if (image)
     it->second.texture->Update(std::move(image));
   else
     it->second.texture->Destroy();
-  }
 }
 
 Texture* Engine::AcquireTexture(const std::string& asset_name) {
@@ -309,9 +304,9 @@ Texture* Engine::AcquireTexture(const std::string& asset_name) {
     return nullptr;
   }
 
-  it->second.use_count++;
   if (!it->second.texture->IsValid())
     RefreshImage(it->first);
+  it->second.use_count++;
   return it->second.texture.get();
 }
 
@@ -401,6 +396,10 @@ std::unique_ptr<InputEvent> Engine::GetNextInputEvent() {
   return event;
 }
 
+void Engine::ConsumeInputEvents() {
+  input_queue_.clear();
+}
+
 void Engine::StartRecording(const Json::Value& payload) {
   if (!replaying_ && !recording_) {
     recording_ = true;
@@ -636,8 +635,6 @@ void Engine::ContextLost() {
 
   if (game_)
     game_->ContextLost();
-
-  input_queue_.clear();
 }
 
 void Engine::SetStatsVisible(bool visible) {

src/engine/engine.h

@@ -84,6 +84,7 @@ class Engine : public PlatformObserver {
   void RemoveCustomShader(const std::string& asset_name);
 
   std::unique_ptr<InputEvent> GetNextInputEvent();
+  void ConsumeInputEvents();
 
   void StartRecording(const Json::Value& payload);
   void EndRecording(const std::string file_name);

src/engine/renderer/opengl/renderer_opengl.h

@@ -144,7 +144,7 @@ class RendererOpenGL final : public Renderer {
 
   std::counting_semaphore<> draw_complete_semaphore_{0};
 
-  std::shared_ptr<base::TaskRunner> main_thread_task_runner_;
+  base::TaskRunner* main_thread_task_runner_;
 #endif  // THREADED_RENDERING
 
   // Stats.

src/engine/renderer/vulkan/renderer_vulkan.cc

@@ -987,7 +987,6 @@ void RendererVulkan::Shutdown() {
     return;
 
   LOG << "Shutting down renderer.";
-  task_runner_.CancelTasks();
   quit_.store(true, std::memory_order_relaxed);
   semaphore_.release();
   setup_thread_.join();
@@ -2006,7 +2005,7 @@ void RendererVulkan::SetupThreadMain(int preallocate) {
     if (quit_.load(std::memory_order_relaxed))
       break;
 
-    task_runner_.RunTasks();
+    task_runner_.SingleConsumerRun();
   }
 
   for (size_t i = 0; i < staging_buffers_.size(); i++) {