dlls/d3dx10_43/compiler.c

   1 /*
   2  * Copyright 2021 Nikolay Sivov for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  *
  18  */
  19
  20 #include "wine/debug.h"
  21
  22 #define COBJMACROS
  23
  24 #include "d3d10_1.h"
  25 #include "d3dx10.h"
  26 #include "d3dcompiler.h"
  27 #include "dxhelpers.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(d3dx);
  30
  31 #define D3DERR_INVALIDCALL 0x8876086c
  32
  33 static HRESULT create_effect(const void *data, SIZE_T datasize, const char *filename,
  34         const D3D10_SHADER_MACRO *defines, ID3D10Include *include, const char *profile,
  35         UINT shader_flags, UINT effect_flags, ID3D10Device *device, ID3D10EffectPool *effect_pool,
  36         ID3D10Effect **effect, ID3D10Blob **errors)
  37 {
  38     const char dxbc[] = {'D','X','B','C'};
  39     ID3D10Blob *code = NULL;
  40     void *buffer;
  41     SIZE_T size;
  42     HRESULT hr;
  43
  44     if (!data || !device)
  45         return E_FAIL;
  46
  47     if (errors)
  48         *errors = NULL;
  49
  50     buffer = (void *)data;
  51     size = datasize;
  52
  53     /* Effect is not compiled. */
  54     if (datasize < sizeof(dxbc) || memcmp(dxbc, data, sizeof(dxbc)))
  55     {
  56         if (!profile)
  57             return D3DERR_INVALIDCALL;
  58         if (FAILED(hr = D3DCompile(data, datasize, filename, defines, include, "main", profile,
  59                 shader_flags, effect_flags, &code, errors)))
  60         {
  61             WARN("Effect compilation failed, hr %#lx.\n", hr);
  62             return hr;
  63         }
  64         buffer = ID3D10Blob_GetBufferPointer(code);
  65         size = ID3D10Blob_GetBufferSize(code);
  66     }
  67
  68     hr = D3D10CreateEffectFromMemory(buffer, size, effect_flags, device, effect_pool, effect);
  69
  70     if (code)
  71         ID3D10Blob_Release(code);
  72     return hr;
  73 }
  74
  75 HRESULT WINAPI D3DX10CreateEffectFromMemory(const void *data, SIZE_T datasize, const char *filename,
  76         const D3D10_SHADER_MACRO *defines, ID3D10Include *include, const char *profile,
  77         UINT shader_flags, UINT effect_flags, ID3D10Device *device, ID3D10EffectPool *effect_pool,
  78         ID3DX10ThreadPump *pump, ID3D10Effect **effect, ID3D10Blob **errors, HRESULT *hresult)
  79 {
  80     TRACE("data %p, datasize %Iu, filename %s, defines %p, include %p, profile %s, shader_flags %#x,"
  81             "effect_flags %#x, device %p, effect_pool %p, pump %p, effect %p, errors %p, hresult %p.\n",
  82             data, datasize, debugstr_a(filename), defines, include, debugstr_a(profile),
  83             shader_flags, effect_flags, device, effect_pool, pump, effect, errors, hresult);
  84
  85     if (pump)
  86         FIXME("Asynchronous mode is not supported.\n");
  87
  88     if (!include)
  89         include = D3D_COMPILE_STANDARD_FILE_INCLUDE;
  90
  91     return create_effect(data, datasize, filename, defines, include, profile,
  92             shader_flags, effect_flags, device, effect_pool, effect, errors);
  93 }
  94
  95 HRESULT WINAPI D3DX10CreateEffectFromFileW(const WCHAR *filename, const D3D10_SHADER_MACRO *defines,
  96         ID3D10Include *include, const char *profile, UINT shader_flags, UINT effect_flags,
  97         ID3D10Device *device, ID3D10EffectPool *effect_pool, ID3DX10ThreadPump *pump,
  98         ID3D10Effect **effect, ID3D10Blob **errors, HRESULT *hresult)
  99 {
 100     char filename_a[MAX_PATH];
 101     void *buffer = NULL;
 102     DWORD size = 0;
 103     HRESULT hr;
 104
 105     TRACE("filename %s, defines %p, include %p, profile %s, shader_flags %#x, effect_flags %#x, "
 106             "device %p, effect_pool %p, pump %p, effect %p, errors %p, hresult %p.\n",
 107             debugstr_w(filename), defines, include, debugstr_a(profile), shader_flags, effect_flags,
 108             device, effect_pool, pump, effect, errors, hresult);
 109
 110     if (!filename)
 111         return E_FAIL;
 112
 113     if (pump)
 114         FIXME("Asynchronous mode is not supported.\n");
 115
 116     if (!include)
 117         include = D3D_COMPILE_STANDARD_FILE_INCLUDE;
 118
 119     if (SUCCEEDED((hr = load_file(filename, &buffer, &size))))
 120     {
 121         WideCharToMultiByte(CP_ACP, 0, filename, -1, filename_a, sizeof(filename_a), NULL, NULL);
 122         hr = create_effect(buffer, size, filename_a, defines, include, profile,
 123                 shader_flags, effect_flags, device, effect_pool, effect, errors);
 124         free(buffer);
 125     }
 126
 127     return hr;
 128 }
 129
 130 HRESULT WINAPI D3DX10CreateEffectFromFileA(const char *filename, const D3D10_SHADER_MACRO *defines,
 131         ID3D10Include *include, const char *profile, UINT shader_flags, UINT effect_flags,
 132         ID3D10Device *device, ID3D10EffectPool *effect_pool, ID3DX10ThreadPump *pump,
 133         ID3D10Effect **effect, ID3D10Blob **errors, HRESULT *hresult)
 134 {
 135     WCHAR *filenameW;
 136     HRESULT hr;
 137     int len;
 138
 139     TRACE("filename %s, defines %p, include %p, profile %s, shader_flags %#x, effect_flags %#x, "
 140             "device %p, effect_pool %p, pump %p, effect %p, errors %p, hresult %p.\n",
 141             debugstr_a(filename), defines, include, debugstr_a(profile), shader_flags, effect_flags,
 142             device, effect_pool, pump, effect, errors, hresult);
 143
 144     if (!filename)
 145         return E_FAIL;
 146
 147     len = MultiByteToWideChar(CP_ACP, 0, filename, -1, NULL, 0);
 148     if (!(filenameW = malloc(len * sizeof(*filenameW))))
 149         return E_OUTOFMEMORY;
 150     MultiByteToWideChar(CP_ACP, 0, filename, -1, filenameW, len);
 151
 152     hr = D3DX10CreateEffectFromFileW(filenameW, defines, include, profile, shader_flags,
 153             effect_flags, device, effect_pool, pump, effect, errors, hresult);
 154     free(filenameW);
 155
 156     return hr;
 157 }
 158
 159 HRESULT WINAPI D3DX10CreateEffectFromResourceA(HMODULE module, const char *resource_name,
 160         const char *filename, const D3D10_SHADER_MACRO *defines, ID3D10Include *include,
 161         const char *profile, UINT shader_flags, UINT effect_flags, ID3D10Device *device,
 162         ID3D10EffectPool *effect_pool, ID3DX10ThreadPump *pump, ID3D10Effect **effect,
 163         ID3D10Blob **errors, HRESULT *hresult)
 164 {
 165     void *data;
 166     DWORD size;
 167     HRESULT hr;
 168
 169     TRACE("module %p, resource_name %s, filename %s, defines %p, include %p, profile %s, "
 170             "shader_flags %#x, effect_flags %#x, device %p, effect_pool %p, pump %p, effect %p, "
 171             "errors %p, hresult %p.\n", module, debugstr_a(resource_name), debugstr_a(filename),
 172             defines, include, debugstr_a(profile), shader_flags, effect_flags,
 173             device, effect_pool, pump, effect, errors, hresult);
 174
 175     hr = load_resourceA(module, resource_name, &data, &size);
 176     if (FAILED(hr))
 177         return hr;
 178
 179     return create_effect(data, size, filename, defines, include, profile,
 180             shader_flags, effect_flags, device, effect_pool, effect, errors);
 181 }
 182
 183 HRESULT WINAPI D3DX10CreateEffectFromResourceW(HMODULE module, const WCHAR *resource_name,
 184         const WCHAR *filenameW, const D3D10_SHADER_MACRO *defines, ID3D10Include *include,
 185         const char *profile, UINT shader_flags, UINT effect_flags, ID3D10Device *device,
 186         ID3D10EffectPool *effect_pool, ID3DX10ThreadPump *pump, ID3D10Effect **effect,
 187         ID3D10Blob **errors, HRESULT *hresult)
 188 {
 189     char *filename = NULL;
 190     void *data;
 191     DWORD size;
 192     HRESULT hr;
 193     int len;
 194
 195     TRACE("module %p, resource_name %s, filename %s, defines %p, include %p, profile %s, "
 196             "shader_flags %#x, effect_flags %#x, device %p, effect_pool %p, pump %p, effect %p, "
 197             "errors %p, hresult %p.\n", module, debugstr_w(resource_name), debugstr_w(filenameW),
 198             defines, include, debugstr_a(profile), shader_flags, effect_flags,
 199             device, effect_pool, pump, effect, errors, hresult);
 200
 201     hr = load_resourceW(module, resource_name, &data, &size);
 202     if (FAILED(hr))
 203         return hr;
 204
 205     if (filenameW)
 206     {
 207         len = WideCharToMultiByte(CP_ACP, 0, filenameW, -1, NULL, 0, NULL, NULL);
 208         if (!(filename = malloc(len)))
 209             return E_OUTOFMEMORY;
 210         WideCharToMultiByte(CP_ACP, 0, filenameW, -1, filename, len, NULL, NULL);
 211     }
 212
 213     hr = create_effect(data, size, filename, defines, include, profile,
 214             shader_flags, effect_flags, device, effect_pool, effect, errors);
 215     free(filename);
 216     return hr;
 217 }
 218
 219 HRESULT WINAPI D3DX10PreprocessShaderFromMemory(const char *data, SIZE_T data_size, const char *filename,
 220         const D3D10_SHADER_MACRO *defines, ID3DInclude *include, ID3DX10ThreadPump *pump, ID3D10Blob **shader_text,
 221         ID3D10Blob **errors, HRESULT *hresult)
 222 {
 223     HRESULT hr;
 224
 225     TRACE("data %s, data_size %Iu, filename %s, defines %p, include %p, pump %p, shader_text %p, "
 226             "errors %p, hresult %p.\n",
 227             debugstr_an(data, data_size), data_size, debugstr_a(filename), defines, include, pump,
 228             shader_text, errors, hresult);
 229
 230     if (!data)
 231         return E_FAIL;
 232
 233     if (pump)
 234         FIXME("Unimplemented ID3DX10ThreadPump handling.\n");
 235
 236     hr = D3DPreprocess(data, data_size, filename, defines, include, shader_text, errors);
 237     if (hresult)
 238         *hresult = hr;
 239     return hr;
 240 }