ComputeShaderMTL.h

Go to the documentation of this file.

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2025 Jorrit Rouwe
// SPDX-License-Identifier: MIT
 
#pragma once
 
#ifdef JPH_USE_MTL
 
#include <MetalKit/MetalKit.h>
 
#include <Jolt/Compute/ComputeShader.h>
#include <Jolt/Core/UnorderedMap.h>
 
JPH_NAMESPACE_BEGIN
 
class JPH_EXPORT ComputeShaderMTL : public ComputeShader
{
public:
    JPH_OVERRIDE_NEW_DELETE
 
                                ComputeShaderMTL(id<MTLComputePipelineState> inPipelineState, MTLComputePipelineReflection *inReflection, uint32 inGroupSizeX, uint32 inGroupSizeY, uint32 inGroupSizeZ);
    virtual                     ~ComputeShaderMTL() override                    { [mPipelineState release]; }
 
    id<MTLComputePipelineState> GetPipelineState() const                        { return mPipelineState; }
 
    uint                        NameToBindingIndex(const char *inName) const;
 
private:
    id<MTLComputePipelineState> mPipelineState;
    UnorderedMap<String, uint>  mNameToBindingIndex;
};
 
JPH_NAMESPACE_END
 
#endif // JPH_USE_MTL