forked from mirrors/gecko-dev
83 lines
2.7 KiB
C++
83 lines
2.7 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 WEBGPU_CommandEncoder_H_
|
|
#define WEBGPU_CommandEncoder_H_
|
|
|
|
#include "mozilla/dom/TypedArray.h"
|
|
#include "nsWrapperCache.h"
|
|
#include "ObjectModel.h"
|
|
|
|
namespace mozilla {
|
|
namespace dom {
|
|
template <typename T>
|
|
class Sequence;
|
|
class WebGPUComputePipelineOrWebGPURenderPipeline;
|
|
struct WebGPURenderPassDescriptor;
|
|
} // namespace dom
|
|
namespace webgpu {
|
|
|
|
class BindGroup;
|
|
class Buffer;
|
|
class CommandBuffer;
|
|
class Device;
|
|
|
|
class CommandEncoder final : public ChildOf<Device> {
|
|
public:
|
|
WEBGPU_DECL_GOOP(CommandEncoder)
|
|
|
|
private:
|
|
CommandEncoder() = delete;
|
|
virtual ~CommandEncoder();
|
|
|
|
public:
|
|
already_AddRefed<CommandBuffer> FinishEncoding() const;
|
|
|
|
// Commands allowed outside of "passes"
|
|
void CopyBufferToBuffer(const Buffer& src, uint32_t srcOffset,
|
|
const Buffer& dst, uint32_t dstOffset,
|
|
uint32_t size) const;
|
|
// TODO figure out all the arguments required for these
|
|
void CopyBufferToTexture() const;
|
|
void CopyTextureToBuffer() const;
|
|
void CopyTextureToTexture() const;
|
|
void Blit() const;
|
|
|
|
void TransitionBuffer(const Buffer& b, uint32_t flags) const;
|
|
|
|
// Allowed in both compute and render passes
|
|
void SetPushConstants(uint32_t stageFlags, uint32_t offset, uint32_t count,
|
|
const dom::ArrayBuffer& data) const;
|
|
void SetBindGroup(uint32_t index, const BindGroup& bindGroup) const;
|
|
void SetPipeline(
|
|
const dom::WebGPUComputePipelineOrWebGPURenderPipeline& pipeline) const;
|
|
|
|
// Compute pass commands
|
|
void BeginComputePass() const;
|
|
void EndComputePass() const;
|
|
|
|
void Dispatch(uint32_t x, uint32_t y, uint32_t z) const;
|
|
|
|
// Render pass commands
|
|
void BeginRenderPass(const dom::WebGPURenderPassDescriptor& desc) const;
|
|
void EndRenderPass() const;
|
|
|
|
void SetBlendColor(float r, float g, float b, float a) const;
|
|
void SetIndexBuffer(const Buffer& buffer, uint32_t offset) const;
|
|
void SetVertexBuffers(uint32_t startSlot,
|
|
const dom::Sequence<OwningNonNull<Buffer>>& buffers,
|
|
const dom::Sequence<uint32_t>& offsets) const;
|
|
|
|
void Draw(uint32_t vertexCount, uint32_t instanceCount, uint32_t firstVertex,
|
|
uint32_t firstInstance) const;
|
|
void DrawIndexed(uint32_t indexCount, uint32_t instanceCount,
|
|
uint32_t firstIndex, uint32_t firstInstance,
|
|
uint32_t firstVertex) const;
|
|
};
|
|
|
|
} // namespace webgpu
|
|
} // namespace mozilla
|
|
|
|
#endif // WEBGPU_CommandEncoder_H_
|