blob: e1c0413d706167196763e3e618cad8320ef397bb [file] [log] [blame] [edit]
// 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.
#include "src/dawn_node/binding/GPUQueue.h"
#include <memory>
#include "src/dawn_node/binding/Converter.h"
#include "src/dawn_node/binding/GPUBuffer.h"
#include "src/dawn_node/binding/GPUCommandBuffer.h"
#include "src/dawn_node/utils/Debug.h"
namespace wgpu { namespace binding {
////////////////////////////////////////////////////////////////////////////////
// wgpu::bindings::GPUQueue
////////////////////////////////////////////////////////////////////////////////
GPUQueue::GPUQueue(wgpu::Queue queue, std::shared_ptr<AsyncRunner> async)
: queue_(std::move(queue)), async_(std::move(async)) {
}
void GPUQueue::submit(
Napi::Env env,
std::vector<interop::Interface<interop::GPUCommandBuffer>> commandBuffers) {
std::vector<wgpu::CommandBuffer> bufs(commandBuffers.size());
for (size_t i = 0; i < commandBuffers.size(); i++) {
bufs[i] = *commandBuffers[i].As<GPUCommandBuffer>();
}
Converter conv(env);
uint32_t bufs_size;
if (!conv(bufs_size, bufs.size())) {
return;
}
queue_.Submit(bufs_size, bufs.data());
}
interop::Promise<void> GPUQueue::onSubmittedWorkDone(Napi::Env env) {
struct Context {
Napi::Env env;
interop::Promise<void> promise;
AsyncTask task;
};
auto ctx = new Context{env, interop::Promise<void>(env, PROMISE_INFO), async_};
auto promise = ctx->promise;
queue_.OnSubmittedWorkDone(
0,
[](WGPUQueueWorkDoneStatus status, void* userdata) {
auto c = std::unique_ptr<Context>(static_cast<Context*>(userdata));
if (status != WGPUQueueWorkDoneStatus::WGPUQueueWorkDoneStatus_Success) {
Napi::Error::New(c->env, "onSubmittedWorkDone() failed")
.ThrowAsJavaScriptException();
}
c->promise.Resolve();
},
ctx);
return promise;
}
void GPUQueue::writeBuffer(Napi::Env env,
interop::Interface<interop::GPUBuffer> buffer,
interop::GPUSize64 bufferOffset,
interop::BufferSource data,
interop::GPUSize64 dataOffset,
std::optional<interop::GPUSize64> size) {
wgpu::Buffer buf = *buffer.As<GPUBuffer>();
Converter::BufferSource src{};
Converter conv(env);
if (!conv(src, data)) {
return;
}
// TODO(crbug.com/dawn/1132): Bounds check
if (src.data) {
src.data = reinterpret_cast<uint8_t*>(src.data) + dataOffset;
}
src.size -= dataOffset;
if (size.has_value()) {
src.size = size.value();
}
queue_.WriteBuffer(buf, bufferOffset, src.data, src.size);
}
void GPUQueue::writeTexture(Napi::Env env,
interop::GPUImageCopyTexture destination,
interop::BufferSource data,
interop::GPUImageDataLayout dataLayout,
interop::GPUExtent3D size) {
wgpu::ImageCopyTexture dst{};
Converter::BufferSource src{};
wgpu::TextureDataLayout layout{};
wgpu::Extent3D sz{};
Converter conv(env);
if (!conv(dst, destination) || //
!conv(src, data) || //
!conv(layout, dataLayout) || //
!conv(sz, size)) {
return;
}
queue_.WriteTexture(&dst, src.data, src.size, &layout, &sz);
}
void GPUQueue::copyExternalImageToTexture(Napi::Env,
interop::GPUImageCopyExternalImage source,
interop::GPUImageCopyTextureTagged destination,
interop::GPUExtent3D copySize) {
UNIMPLEMENTED();
}
std::optional<std::string> GPUQueue::getLabel(Napi::Env) {
UNIMPLEMENTED();
}
void GPUQueue::setLabel(Napi::Env, std::optional<std::string> value) {
UNIMPLEMENTED();
}
}} // namespace wgpu::binding