yum-mirror/slang

Making it easier to work with shaders

git clone https://git.yummers.dev/yum-mirror/slang

Gangzheng TongUse wide char version of Windows API (#8390)3aff764c2

master
branchesmasterall refs →
1.9 KiB62 linesraw
1#include "slang-rt.h"
2
3#include "../core/slang-basic.h"
4#include "../core/slang-shared-library.h"
5
6#if SLANG_WINDOWS_FAMILY
7#include <windows.h>
8#endif
9
10using namespace Slang;
11Dictionary<String, ComPtr<ISlangSharedLibrary>> slangRT_loadedLibraries;
12
13extern "C"
14{
15    SLANG_RT_API void SLANG_MCALL _slang_rt_abort(Slang::String errorMessage)
16    {
17        fprintf(stderr, "%s", errorMessage.getBuffer());
18#if SLANG_WINDOWS_FAMILY
19        MessageBoxW(0, errorMessage.toWString(), L"Slang Runtime Error", MB_ICONERROR);
20#endif
21        abort();
22    }
23
24    SLANG_RT_API void* SLANG_MCALL _slang_rt_load_dll(Slang::String modulePath)
25    {
26        ComPtr<ISlangSharedLibrary> lib;
27        if (!slangRT_loadedLibraries.tryGetValue(modulePath, lib))
28        {
29            if (DefaultSharedLibraryLoader::getSingleton()->loadSharedLibrary(
30                    modulePath.getBuffer(),
31                    lib.writeRef()) != SLANG_OK)
32            {
33                _slang_rt_abort("Failed to load DLL \"" + modulePath + "\"");
34            }
35            slangRT_loadedLibraries[modulePath] = lib;
36        }
37        return lib.get();
38    }
39
40    SLANG_RT_API void* SLANG_MCALL
41    _slang_rt_load_dll_func(void* moduleHandle, Slang::String funcName, uint32_t argSize)
42    {
43        if (moduleHandle == nullptr)
44        {
45            moduleHandle = _slang_rt_load_dll("");
46        }
47        auto lib = static_cast<ISlangSharedLibrary*>(moduleHandle);
48        auto funcPtr = lib->findFuncByName(funcName.getBuffer());
49        if (!funcPtr)
50        {
51            // If failed, try stdcall mangled name.
52            StringBuilder sb;
53            sb << "_" << funcName << "@" << argSize;
54            funcPtr = lib->findFuncByName(sb.toString().getBuffer());
55        }
56        if (!funcPtr)
57        {
58            _slang_rt_abort("Cannot find function \"" + funcName + "\" in loaded library.");
59        }
60        return (void*)funcPtr;
61    }
62}