/*
 * Copyright 2021 Google LLC
 *
 * Use of this source code is governed by a BSD-style license that can be
 * found in the LICENSE file.
 */

#ifndef skgpu_MtlGraphicsPipeline_DEFINED
#define skgpu_MtlGraphicsPipeline_DEFINED

#include "experimental/graphite/src/GraphicsPipeline.h"
#include "include/core/SkRefCnt.h"
#include "include/ports/SkCFObject.h"
#include <memory>

#import <Metal/Metal.h>

namespace skgpu {
class Context;
class GraphicsPipelineDesc;
struct RenderPassDesc;
} // namespace skgpu

namespace skgpu::mtl {
class Gpu;

class GraphicsPipeline final : public skgpu::GraphicsPipeline {
public:
    inline static constexpr unsigned int kIntrinsicUniformBufferIndex = 0;
    inline static constexpr unsigned int kRenderStepUniformBufferIndex = 1;
    inline static constexpr unsigned int kPaintUniformBufferIndex = 2;
    inline static constexpr unsigned int kVertexBufferIndex = 3;
    inline static constexpr unsigned int kInstanceBufferIndex = 4;

    static sk_sp<GraphicsPipeline> Make(const Context*,
                                        const Gpu*,
                                        const skgpu::GraphicsPipelineDesc&,
                                        const skgpu::RenderPassDesc&);
    ~GraphicsPipeline() override {}

    id<MTLRenderPipelineState> mtlPipelineState() const { return fPipelineState.get(); }
    id<MTLDepthStencilState> mtlDepthStencilState() const { return fDepthStencilState; }
    uint32_t stencilReferenceValue() const { return fStencilReferenceValue; }
    size_t vertexStride() const { return fVertexStride; }
    size_t instanceStride() const { return fInstanceStride; }

private:
    GraphicsPipeline(const skgpu::Gpu* gpu,
                     sk_cfp<id<MTLRenderPipelineState>> pso,
                     id<MTLDepthStencilState> dss,
                     uint32_t refValue,
                     size_t vertexStride,
                     size_t instanceStride)
        : skgpu::GraphicsPipeline(gpu)
        , fPipelineState(std::move(pso))
        , fDepthStencilState(dss)
        , fStencilReferenceValue(refValue)
        , fVertexStride(vertexStride)
        , fInstanceStride(instanceStride) {}

    void onFreeGpuData() override;

    sk_cfp<id<MTLRenderPipelineState>> fPipelineState;
    id<MTLDepthStencilState> fDepthStencilState;
    uint32_t fStencilReferenceValue;
    size_t fVertexStride = 0;
    size_t fInstanceStride = 0;
};

} // namespace skgpu::mtl

#endif // skgpu_MtlGraphicsPipeline_DEFINED