blob: a86ee3a0b601347a0c6d8a51dc2e3370bf60c14a [file] [log] [blame]
// Copyright 2021 The Dawn Authors
//
// 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 DAWN_NODE_BINDING_ASYNC_RUNNER_H_
#define DAWN_NODE_BINDING_ASYNC_RUNNER_H_
#include <stdint.h>
#include <memory>
#include "dawn/webgpu_cpp.h"
#include "napi.h"
namespace wgpu { namespace binding {
// AsyncRunner is used to poll a wgpu::Device with calls to Tick() while there are asynchronous
// tasks in flight.
class AsyncRunner {
public:
AsyncRunner(Napi::Env env, wgpu::Device device);
// Begin() should be called when a new asynchronous task is started.
// If the number of executing asynchronous tasks transitions from 0 to 1, then a function
// will be scheduled on the main JavaScript thread to call wgpu::Device::Tick() whenever the
// thread is idle. This will be repeatedly called until the number of executing asynchronous
// tasks reaches 0 again.
void Begin();
// End() should be called once the asynchronous task has finished.
// Every call to Begin() should eventually result in a call to End().
void End();
private:
void QueueTick();
Napi::Env env_;
wgpu::Device const device_;
uint64_t count_ = 0;
bool tick_queued_ = false;
};
// AsyncTask is a RAII helper for calling AsyncRunner::Begin() on construction, and
// AsyncRunner::End() on destruction.
class AsyncTask {
public:
inline AsyncTask(AsyncTask&&) = default;
// Constructor.
// Calls AsyncRunner::Begin()
inline AsyncTask(std::shared_ptr<AsyncRunner> runner) : runner_(std::move(runner)) {
runner_->Begin();
};
// Destructor.
// Calls AsyncRunner::End()
inline ~AsyncTask() {
runner_->End();
}
private:
AsyncTask(const AsyncTask&) = delete;
AsyncTask& operator=(const AsyncTask&) = delete;
std::shared_ptr<AsyncRunner> runner_;
};
}} // namespace wgpu::binding
#endif // DAWN_NODE_BINDING_ASYNC_RUNNER_H_