ComputeShader.hpp

#ifndef COMPUTE_SHADER_H
#define COMPUTE_SHADER_H

#include <string>
#if RENDERER_METAL
#import <Metal/Metal.h>
#endif
#if RENDERER_D3D12
#include <d3dcompiler.h>
#endif
#if RENDERER_VULKAN
#include <vulkan/vulkan.h>
#include <cstdint>
#endif

namespace ae3d
{
    namespace FileSystem
    {
        struct FileContentsData;
    }
    
    class ComputeShader
    {
    public:
        void Dispatch( unsigned groupCountX, unsigned groupCountY, unsigned groupCountZ );

        void Load( const char* source );

#if RENDERER_VULKAN
        void LoadSPIRV( const FileSystem::FileContentsData& contents );
#endif
        void Load( const char* metalShaderName,
                   const FileSystem::FileContentsData& dataHLSL,
                   const FileSystem::FileContentsData& dataSPIRV );
        
#if RENDERER_METAL
        void LoadFromLibrary( const char* vertexShaderName, const char* fragmentShaderName );
#endif  
#if RENDERER_D3D12
        ID3DBlob* blobShader = nullptr;
#endif
#if RENDERER_VULKAN
        VkPipelineShaderStageCreateInfo& GetVertexInfo() { return info; }
#endif

    private:
        std::string path;

#if RENDERER_VULKAN
        VkPipelineShaderStageCreateInfo info;
#endif
#if RENDERER_METAL
        id <MTLFunction> function;
        id<MTLComputePipelineState> pipeline;
#endif
    };
}
#endif