cjit include loader

Author: nahiim

include/nbl/asset/utils/CHLSLCompiler.h

@@ -55,6 +55,8 @@ class NBL_API2 CHLSLCompiler final : public IShaderCompiler
 		// when Nabla is used as a lib
 		nbl::asset::hlsl::impl::DXC* m_dxcCompilerTypes;
 
+		IShaderCompiler::CIncludeFinder* m_defaultIncludeFinder;
+
 		static CHLSLCompiler::SOptions option_cast(const IShaderCompiler::SCompilerOptions& options)
 		{
 			CHLSLCompiler::SOptions ret = {};

include/nbl/asset/utils/CJITIncludeLoader.h

@@ -0,0 +1,18 @@
+#ifndef _NBL_ASSET_C_JIT_INCLUDE_LOADER_H_INCLUDED_
+#define _NBL_ASSET_C_JIT_INCLUDE_LOADER_H_INCLUDED_
+
+#include <string>
+#include <unordered_map>
+
+class CJITIncludeLoader
+{
+public:
+    CJITIncludeLoader();
+    std::string loadInclude(const std::string path);
+
+private:
+    std::unordered_map<std::string, std::string> m_includes;
+    std::string collectDeviceCaps();
+};
+
+#endif // CJITINCLUDELOADER_H

include/nbl/asset/utils/IShaderCompiler.h

@@ -13,6 +13,7 @@
 
 #include "nbl/asset/ICPUSpecializedShader.h"
 #include "nbl/asset/utils/ISPIRVOptimizer.h"
+#include "nbl/asset/utils/CJITIncludeLoader.h"
 
 namespace nbl::asset
 {
@@ -293,6 +294,8 @@ class NBL_API2 IShaderCompiler : public core::IReferenceCounted
 		core::smart_refctd_ptr<system::ISystem> m_system;
 	private:
 		core::smart_refctd_ptr<CIncludeFinder> m_defaultIncludeFinder;
+
+		CJITIncludeLoader m_JITIncludeLoader;
 };
 
 NBL_ENUM_ADD_BITWISE_OPERATORS(IShaderCompiler::E_DEBUG_INFO_FLAGS)

include/nbl/builtin/hlsl/jit/device_capabilities.hlsl

@@ -46,6 +46,8 @@ namespace nbl::asset::hlsl::impl
 CHLSLCompiler::CHLSLCompiler(core::smart_refctd_ptr<system::ISystem>&& system)
     : IShaderCompiler(std::move(system))
 {
+    m_defaultIncludeFinder->addSearchPath("", core::make_smart_refctd_ptr<CJITIncludeLoader>());
+
     ComPtr<IDxcUtils> utils;
     auto res = DxcCreateInstance(CLSID_DxcUtils, IID_PPV_ARGS(utils.GetAddressOf()));
     assert(SUCCEEDED(res));

src/nbl/asset/utils/CHLSLCompiler.cpp

@@ -0,0 +1,61 @@
+
+#include "CJITIncludeLoader.h"
+
+
+CJITIncludeLoader::CJITIncludeLoader()
+{
+    m_includes["device_capabilities.hlsl"] = collectDeviceCaps();
+}
+
+std::string CJITIncludeLoader::loadInclude(const std::string path)
+{
+    const std::string prefix = "nbl/builtin/hlsl/jit/";
+    if (path.compare(0, prefix.size(), prefix) == 0)
+    {
+        std::string includeFileName = path.substr(prefix.size());
+
+        // Look up the content in m_includes map
+        auto it = m_includes.find(includeFileName);
+        if (it != m_includes.end())
+        {
+            // Return the content of the specified include file
+            return it->second;
+        }
+        else
+        {
+            // Handle error: include file not found
+            std::cerr << "Error: Include file '" << path << "' not found!" << std::endl;
+            return "";
+        }
+    }
+    else
+    {
+        // Handle error: invalid include path
+        std::cerr << "Error: Invalid include path '" << path << "'!" << std::endl;
+        return "";
+    }
+}
+
+
+std::string CJITIncludeLoader::collectDeviceCaps()
+{
+    std::ostringstream content;
+    content << "#ifndef _NBL_BUILTIN_HLSL_JIT_DEVICE_CAPABILITIES_INCLUDED_" << std::endl;
+    content << "#define _NBL_BUILTIN_HLSL_JIT_DEVICE_CAPABILITIES_INCLUDED_" << std::endl;
+    content << "namespace nbl {" << std::endl;
+    content << "namespace hlsl {" << std::endl;
+    content << "namespace jit {" << std::endl;
+    content << "struct device_capabilities {" << std::endl;
+    content << "  NBL_CONSTEXPR_STATIC_INLINE bool shaderFloat64 = false" << std::endl;
+    content << "  NBL_CONSTEXPR_STATIC_INLINE bool shaderDrawParameters = false" << std::endl;
+    content << "  NBL_CONSTEXPR_STATIC_INLINE bool subgroupArithmetic = false" << std::endl;
+    content << "  NBL_CONSTEXPR_STATIC_INLINE bool fragmentShaderPixelInterlock = false" << std::endl;
+    content << std::endl;
+    content << "  NBL_CONSTEXPR_STATIC_INLINE uint16_t maxOptimallyResidentWorkgroupInvocations = 0" << std::endl;
+    content << "};" << std::endl;
+    content << "}" << std::endl;
+    content << "}" << std::endl;
+    content << "#endif" << std::endl;
+
+    return content.str();
+}