/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* 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_