Inject aux Slang code

Author: corporateshark

lvk/vulkan/VulkanClasses.cpp

@@ -5827,6 +5827,39 @@ lvk::ShaderModuleState lvk::VulkanContext::createShaderModuleFromSlang(ShaderSta
                                                                        const char* source,
                                                                        const char* debugName,
                                                                        Result* outResult) const {
+  std::string sourcePatched;
+
+  if (!source || !*source) {
+    Result::setResult(outResult, Result::Code::ArgumentOutOfRange, "Shader source is empty");
+    return {};
+  }
+
+  auto addCode = [source, &sourcePatched](const char* substr, const char* code) -> void {
+    if (strstr(source, substr)) {
+      sourcePatched.append(code);
+    }
+  };
+
+  // bindless texture and sampler arrays
+  sourcePatched +=
+      "[[vk::binding(0, 0)]] Texture2D    kTextures2D[];\n"
+      "[[vk::binding(0, 1)]] Texture3D    kTextures3D[];\n"
+      "[[vk::binding(0, 2)]] TextureCube  kTexturesCube[];\n"
+      "[[vk::binding(1, 0)]] SamplerState kSamplers[];\n";
+  addCode("textureBindless2D(",
+          "float4 textureBindless2D(uint textureid, uint samplerid, float2 uv) {\n"
+          "  return kTextures2D[NonUniformResourceIndex(textureid)].Sample(\n"
+          "    kSamplers[NonUniformResourceIndex(samplerid)], uv);\n"
+          "}\n");
+  addCode("textureBindlessCube(",
+          "float4 textureBindlessCube(uint textureid, uint samplerid, float3 dir) {\n"
+          "  return kTexturesCube[NonUniformResourceIndex(textureid)].Sample(\n"
+          "    kSamplers[NonUniformResourceIndex(samplerid)], dir);\n"
+          "}\n");
+
+  sourcePatched += source;
+  source = sourcePatched.c_str();
+
   std::vector<uint8_t> spirv;
   lvk::Result::setResult(outResult, lvk::compileShaderSlang(stage, source, &spirv));
 

samples/009_TriplanarMapping.cpp

@@ -68,15 +68,6 @@ void vertexMain(uint vertexID   : SV_VertexID,
   output.normal = normalize(float3(v.x, v.y, v.z)); // object space normal
 }
 
-// bindless texture and sampler arrays
-[[vk::binding(0, 0)]] Texture2D    kTextures2D[];
-[[vk::binding(1, 0)]] SamplerState kSamplers[];
-
-float4 textureBindless2D(uint textureid, uint samplerid, float2 uv) {
-  return kTextures2D[NonUniformResourceIndex(textureid)].Sample(
-    kSamplers[NonUniformResourceIndex(samplerid)], uv);
-}
-
 float4 triplanar(uint tex, float3 worldPos, float3 normal) {
   // generate weights, show texture on both sides of the object (positive and negative)
   float3 weights = abs(normal);