/* 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 {

/* Abstracts the creation and management of secondary threads with GPU contexts.
 * Must be created from the main thread.
 * Threads and their context remain alive until destruction. */
class GPUWorker {
 private:
  Vector<std::unique_ptr<std::thread>> threads_;
  std::condition_variable condition_var_;
  std::mutex mutex_;
  std::atomic<bool> terminate_ = false;

 public:
  /**
   * \param threads_count: Number of threads to span.
   * \param share_context: If true, all threads will use the same secondary GPUContext,
   *  otherwise each thread will have its own unique GPUContext.
   * \param run_cb: The callback function that will be called by a thread on `wake_up()`.
   */
  GPUWorker(uint32_t threads_count, bool share_context, std::function<void()> run_cb);
  ~GPUWorker();

  /* Wake up a single thread. */
  void wake_up()
  {
    condition_var_.notify_one();
  }

 private:
  void run(std::shared_ptr<GPUSecondaryContext> context, std::function<void()> run_cb);
};

}  // namespace blender::gpu