summaryrefslogtreecommitdiff
path: root/tools/gfx/metal/metal-pipeline-state.h
blob: 71897925ad10861c6d04fb08a0a56f90cee45ddf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// metal-pipeline-state.h
#pragma once

#include "metal-base.h"

namespace gfx
{

using namespace Slang;

namespace metal
{

class PipelineStateImpl : public PipelineStateBase
{
public:
    DeviceImpl* m_device;
    NS::SharedPtr<MTL::RenderPipelineState> m_renderPipelineState;
    NS::SharedPtr<MTL::DepthStencilState> m_depthStencilState;
    NS::SharedPtr<MTL::ComputePipelineState> m_computePipelineState;
    MTL::Size m_threadGroupSize;
    NS::UInteger m_vertexBufferOffset;

    PipelineStateImpl(DeviceImpl* device);
    ~PipelineStateImpl();

    void init(const GraphicsPipelineStateDesc& desc);
    void init(const ComputePipelineStateDesc& desc);
    void init(const RayTracingPipelineStateDesc& desc);

    Result createMetalComputePipelineState();
    Result createMetalRenderPipelineState();

    virtual Result ensureAPIPipelineStateCreated() override;

    virtual SLANG_NO_THROW Result SLANG_MCALL getNativeHandle(InteropHandle* outHandle) override;
};

class RayTracingPipelineStateImpl : public PipelineStateImpl
{
public:
    Dictionary<String, Index> shaderGroupNameToIndex;
    Int shaderGroupCount;

    RayTracingPipelineStateImpl(DeviceImpl* device);

    virtual Result ensureAPIPipelineStateCreated() override;

    virtual SLANG_NO_THROW Result SLANG_MCALL getNativeHandle(InteropHandle* outHandle) override;
};

} // namespace metal
} // namespace gfx