async_work_task.h

#pragma once
 
#include <tess/ops/async_work.h>
#include <tess/sim/schedule.h>
 
namespace tess {

template <typename T>
class ResumableWorkTask {
 public:
  explicit ResumableWorkTask(ResumableWorkQueue<T>& queue) noexcept
      : queue_(&queue) {}
 
  ResumableWorkTask(const ResumableWorkTask&) = delete;
  auto operator=(const ResumableWorkTask&) -> ResumableWorkTask& = delete;
  ResumableWorkTask(ResumableWorkTask&&) = delete;
  auto operator=(ResumableWorkTask&&) -> ResumableWorkTask& = delete;
 
  [[nodiscard]] auto operator()(const ScheduleTaskContext& context)
      -> ScheduleTaskResult {
    const auto stats = queue_->advance(AsyncWorkBudget{context.budget_items});
    return ScheduleTaskResult{{}, stats.items_done, stats.pending != 0, 0};
  }
 
 private:
  ResumableWorkQueue<T>* queue_ = nullptr;
};
 
}  // namespace tess