forked from mirrors/gecko-dev
e88f442492
This makes sure we never create an invalid (zero) ID after incorrect usage of a command encoder. It also simplifies the code. The JS object should not do any validaion (per spec) and simply forward the commands to the parent process where all of the validation is done. Differential Revision: https://phabricator.services.mozilla.com/D192839
76 lines
2.1 KiB
C++
76 lines
2.1 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/. */
|
|
|
|
#ifndef GPU_ComputePassEncoder_H_
|
|
#define GPU_ComputePassEncoder_H_
|
|
|
|
#include "mozilla/Scoped.h"
|
|
#include "mozilla/dom/TypedArray.h"
|
|
#include "ObjectModel.h"
|
|
|
|
namespace mozilla {
|
|
class ErrorResult;
|
|
|
|
namespace dom {
|
|
struct GPUComputePassDescriptor;
|
|
}
|
|
|
|
namespace webgpu {
|
|
namespace ffi {
|
|
struct WGPUComputePass;
|
|
} // namespace ffi
|
|
|
|
class BindGroup;
|
|
class Buffer;
|
|
class CommandEncoder;
|
|
class ComputePipeline;
|
|
|
|
struct ScopedFfiComputeTraits {
|
|
using type = ffi::WGPUComputePass*;
|
|
static type empty();
|
|
static void release(type raw);
|
|
};
|
|
|
|
class ComputePassEncoder final : public ObjectBase,
|
|
public ChildOf<CommandEncoder> {
|
|
public:
|
|
GPU_DECL_CYCLE_COLLECTION(ComputePassEncoder)
|
|
GPU_DECL_JS_WRAP(ComputePassEncoder)
|
|
|
|
ComputePassEncoder(CommandEncoder* const aParent,
|
|
const dom::GPUComputePassDescriptor& aDesc);
|
|
|
|
private:
|
|
virtual ~ComputePassEncoder();
|
|
void Cleanup() {}
|
|
|
|
Scoped<ScopedFfiComputeTraits> mPass;
|
|
// keep all the used objects alive while the pass is recorded
|
|
nsTArray<RefPtr<const BindGroup>> mUsedBindGroups;
|
|
nsTArray<RefPtr<const ComputePipeline>> mUsedPipelines;
|
|
|
|
public:
|
|
// programmable pass encoder
|
|
void SetBindGroup(uint32_t aSlot, const BindGroup& aBindGroup,
|
|
const dom::Sequence<uint32_t>& aDynamicOffsets);
|
|
// self
|
|
void SetPipeline(const ComputePipeline& aPipeline);
|
|
|
|
void DispatchWorkgroups(uint32_t workgroupCountX, uint32_t workgroupCountY,
|
|
uint32_t workgroupCountZ);
|
|
void DispatchWorkgroupsIndirect(const Buffer& aIndirectBuffer,
|
|
uint64_t aIndirectOffset);
|
|
|
|
void PushDebugGroup(const nsAString& aString);
|
|
void PopDebugGroup();
|
|
void InsertDebugMarker(const nsAString& aString);
|
|
|
|
void End();
|
|
};
|
|
|
|
} // namespace webgpu
|
|
} // namespace mozilla
|
|
|
|
#endif // GPU_ComputePassEncoder_H_
|