GPU_worker.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2025 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include "BLI_vector.hh"
#include "GPU_context.hh"
 
#include <atomic>
#include <condition_variable>
#include <functional>
#include <memory>
#include <mutex>
#include <thread>
 
namespace blender::gpu {

class GPUWorker {
 private:
  Vector<std::unique_ptr<std::thread>> threads_;
  std::condition_variable condition_var_;
  std::mutex &mutex_;
  bool terminate_ = false;
 
 public:
  enum class ContextType {
    Main,
    PerThread,
  };

  GPUWorker(uint32_t threads_count,
            ContextType context_type,
            std::mutex &mutex,
            std::function<void *()> pop_work,
            std::function<void(void *)> do_work);
  ~GPUWorker();
 
  /* Wake up a single thread. */
  void wake_up()
  {
    condition_var_.notify_one();
  }
 
 private:
  void run(std::shared_ptr<GPUSecondaryContext> context,
           std::function<void *()> pop_work,
           std::function<void(void *)> do_work);
};
 
}  // namespace blender::gpu