Get entry_point from IRShaderReflection

Author: andrejnau

src/FlyCube/HLSLCompiler/MetalShaderConverter.cpp

@@ -29,7 +29,9 @@ IRShaderStage GetShaderStage(ShaderType type)
 
 } // namespace
 
-std::vector<uint8_t> ConvertToMetalLibBytecode(ShaderType shader_type, const std::vector<uint8_t>& blob)
+std::vector<uint8_t> ConvertToMetalLibBytecode(ShaderType shader_type,
+                                               const std::vector<uint8_t>& blob,
+                                               std::string& entry_point)
 {
     IRCompiler* compiler = IRCompilerCreate();
     IRObject* dxil_obj = IRObjectCreateFromDXIL(blob.data(), blob.size(), IRBytecodeOwnershipNone);
@@ -55,6 +57,11 @@ std::vector<uint8_t> ConvertToMetalLibBytecode(ShaderType shader_type, const std
     std::vector<uint8_t> metal_lib_bytecode(metal_lib_size);
     IRMetalLibGetBytecode(metal_lib, metal_lib_bytecode.data());
 
+    IRShaderReflection* reflection = IRShaderReflectionCreate();
+    IRObjectGetReflection(metal_ir, GetShaderStage(shader_type), reflection);
+    entry_point = IRShaderReflectionGetEntryPointFunctionName(reflection);
+
+    IRShaderReflectionDestroy(reflection);
     IRMetalLibBinaryDestroy(metal_lib);
     IRObjectDestroy(metal_ir);
     IRObjectDestroy(dxil_obj);

src/FlyCube/HLSLCompiler/MetalShaderConverter.h

@@ -1,4 +1,6 @@
 #pragma once
 #include "Instance/BaseTypes.h"
 
-std::vector<uint8_t> ConvertToMetalLibBytecode(ShaderType shader_type, const std::vector<uint8_t>& blob);
+std::vector<uint8_t> ConvertToMetalLibBytecode(ShaderType shader_type,
+                                               const std::vector<uint8_t>& blob,
+                                               std::string& entry_point);

src/FlyCube/Shader/MTShader.mm

@@ -13,8 +13,8 @@
     : ShaderBase(blob, blob_type, shader_type)
 {
 #if defined(USE_METAL_SHADER_CONVERTER)
-    std::string entry_point = "main";
-    auto metal_lib_bytecode = ConvertToMetalLibBytecode(shader_type, blob);
+    std::string entry_point;
+    auto metal_lib_bytecode = ConvertToMetalLibBytecode(shader_type, blob, entry_point);
     dispatch_data_t metal_lib_data = dispatch_data_create(metal_lib_bytecode.data(), metal_lib_bytecode.size(), nullptr,
                                                           DISPATCH_DATA_DESTRUCTOR_DEFAULT);
     NSError* error = nullptr;