forked from mirrors/gecko-dev
ba23af9d48
Inspired by emilio's suggestion in the shader module API patch. This tries to be the most straightforward way to go from the strings coming from IPC to the ones consumed by wgpu. Differential Revision: https://phabricator.services.mozilla.com/D151024
108 lines
3.2 KiB
C++
108 lines
3.2 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "mozilla/dom/WebGPUBinding.h"
|
|
#include "ComputePassEncoder.h"
|
|
#include "BindGroup.h"
|
|
#include "ComputePipeline.h"
|
|
#include "CommandEncoder.h"
|
|
|
|
#include "mozilla/webgpu/ffi/wgpu.h"
|
|
|
|
namespace mozilla::webgpu {
|
|
|
|
GPU_IMPL_CYCLE_COLLECTION(ComputePassEncoder, mParent, mUsedBindGroups,
|
|
mUsedPipelines)
|
|
GPU_IMPL_JS_WRAP(ComputePassEncoder)
|
|
|
|
ffi::WGPUComputePass* ScopedFfiComputeTraits::empty() { return nullptr; }
|
|
|
|
void ScopedFfiComputeTraits::release(ffi::WGPUComputePass* raw) {
|
|
if (raw) {
|
|
ffi::wgpu_compute_pass_destroy(raw);
|
|
}
|
|
}
|
|
|
|
ffi::WGPUComputePass* BeginComputePass(
|
|
RawId aEncoderId, const dom::GPUComputePassDescriptor& aDesc) {
|
|
ffi::WGPUComputePassDescriptor desc = {};
|
|
|
|
webgpu::StringHelper label(aDesc.mLabel);
|
|
desc.label = label.Get();
|
|
|
|
return ffi::wgpu_command_encoder_begin_compute_pass(aEncoderId, &desc);
|
|
}
|
|
|
|
ComputePassEncoder::ComputePassEncoder(
|
|
CommandEncoder* const aParent, const dom::GPUComputePassDescriptor& aDesc)
|
|
: ChildOf(aParent), mPass(BeginComputePass(aParent->mId, aDesc)) {}
|
|
|
|
ComputePassEncoder::~ComputePassEncoder() {
|
|
if (mValid) {
|
|
mValid = false;
|
|
}
|
|
}
|
|
|
|
void ComputePassEncoder::SetBindGroup(
|
|
uint32_t aSlot, const BindGroup& aBindGroup,
|
|
const dom::Sequence<uint32_t>& aDynamicOffsets) {
|
|
if (mValid) {
|
|
mUsedBindGroups.AppendElement(&aBindGroup);
|
|
ffi::wgpu_compute_pass_set_bind_group(mPass, aSlot, aBindGroup.mId,
|
|
aDynamicOffsets.Elements(),
|
|
aDynamicOffsets.Length());
|
|
}
|
|
}
|
|
|
|
void ComputePassEncoder::SetPipeline(const ComputePipeline& aPipeline) {
|
|
if (mValid) {
|
|
mUsedPipelines.AppendElement(&aPipeline);
|
|
ffi::wgpu_compute_pass_set_pipeline(mPass, aPipeline.mId);
|
|
}
|
|
}
|
|
|
|
void ComputePassEncoder::DispatchWorkgroups(uint32_t x, uint32_t y,
|
|
uint32_t z) {
|
|
if (mValid) {
|
|
ffi::wgpu_compute_pass_dispatch_workgroups(mPass, x, y, z);
|
|
}
|
|
}
|
|
|
|
void ComputePassEncoder::DispatchWorkgroupsIndirect(
|
|
const Buffer& aIndirectBuffer, uint64_t aIndirectOffset) {
|
|
if (mValid) {
|
|
ffi::wgpu_compute_pass_dispatch_workgroups_indirect(
|
|
mPass, aIndirectBuffer.mId, aIndirectOffset);
|
|
}
|
|
}
|
|
|
|
void ComputePassEncoder::PushDebugGroup(const nsAString& aString) {
|
|
if (mValid) {
|
|
const NS_ConvertUTF16toUTF8 utf8(aString);
|
|
ffi::wgpu_compute_pass_push_debug_group(mPass, utf8.get(), 0);
|
|
}
|
|
}
|
|
void ComputePassEncoder::PopDebugGroup() {
|
|
if (mValid) {
|
|
ffi::wgpu_compute_pass_pop_debug_group(mPass);
|
|
}
|
|
}
|
|
void ComputePassEncoder::InsertDebugMarker(const nsAString& aString) {
|
|
if (mValid) {
|
|
const NS_ConvertUTF16toUTF8 utf8(aString);
|
|
ffi::wgpu_compute_pass_insert_debug_marker(mPass, utf8.get(), 0);
|
|
}
|
|
}
|
|
|
|
void ComputePassEncoder::EndPass(ErrorResult& aRv) {
|
|
if (mValid) {
|
|
mValid = false;
|
|
auto* pass = mPass.forget();
|
|
MOZ_ASSERT(pass);
|
|
mParent->EndComputePass(*pass, aRv);
|
|
}
|
|
}
|
|
|
|
} // namespace mozilla::webgpu
|