dlls/d3dx9_36/shader.c

   1 /*
   2  * Copyright 2008 Luis Busquets
   3  * Copyright 2009 Matteo Bruni
   4  * Copyright 2011 Travis Athougies
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22 #include "wine/port.h"
  23
  24 #define NONAMELESSUNION
  25 #define NONAMELESSSTRUCT
  26 #include "wine/debug.h"
  27 #include "wine/unicode.h"
  28 #include "windef.h"
  29 #include "wingdi.h"
  30 #include "objbase.h"
  31 #include "d3dcommon.h"
  32 #include "d3dcompiler.h"
  33 #include "d3dx9_36_private.h"
  34
  35 WINE_DEFAULT_DEBUG_CHANNEL(d3dx);
  36
  37 /* This function is not declared in the SDK headers yet */
  38 HRESULT WINAPI D3DAssemble(LPCVOID data, SIZE_T datasize, LPCSTR filename,
  39                            const D3D_SHADER_MACRO *defines, ID3DInclude *include,
  40                            UINT flags,
  41                            ID3DBlob **shader, ID3DBlob **error_messages);
  42
  43 LPCSTR WINAPI D3DXGetPixelShaderProfile(LPDIRECT3DDEVICE9 device)
  44 {
  45     D3DCAPS9 caps;
  46
  47     TRACE("device %p\n", device);
  48
  49     if (!device) return NULL;
  50
  51     IDirect3DDevice9_GetDeviceCaps(device,&caps);
  52
  53     switch (caps.PixelShaderVersion)
  54     {
  55     case D3DPS_VERSION(1, 1):
  56         return "ps_1_1";
  57
  58     case D3DPS_VERSION(1, 2):
  59         return "ps_1_2";
  60
  61     case D3DPS_VERSION(1, 3):
  62         return "ps_1_3";
  63
  64     case D3DPS_VERSION(1, 4):
  65         return "ps_1_4";
  66
  67     case D3DPS_VERSION(2, 0):
  68         if ((caps.PS20Caps.NumTemps>=22)                          &&
  69             (caps.PS20Caps.Caps&D3DPS20CAPS_ARBITRARYSWIZZLE)     &&
  70             (caps.PS20Caps.Caps&D3DPS20CAPS_GRADIENTINSTRUCTIONS) &&
  71             (caps.PS20Caps.Caps&D3DPS20CAPS_PREDICATION)          &&
  72             (caps.PS20Caps.Caps&D3DPS20CAPS_NODEPENDENTREADLIMIT) &&
  73             (caps.PS20Caps.Caps&D3DPS20CAPS_NOTEXINSTRUCTIONLIMIT))
  74         {
  75             return "ps_2_a";
  76         }
  77         if ((caps.PS20Caps.NumTemps>=32)                          &&
  78             (caps.PS20Caps.Caps&D3DPS20CAPS_NOTEXINSTRUCTIONLIMIT))
  79         {
  80             return "ps_2_b";
  81         }
  82         return "ps_2_0";
  83
  84     case D3DPS_VERSION(3, 0):
  85         return "ps_3_0";
  86     }
  87
  88     return NULL;
  89 }
  90
  91 UINT WINAPI D3DXGetShaderSize(const DWORD *byte_code)
  92 {
  93     const DWORD *ptr = byte_code;
  94
  95     TRACE("byte_code %p\n", byte_code);
  96
  97     if (!ptr) return 0;
  98
  99     /* Look for the END token, skipping the VERSION token */
 100     while (*++ptr != D3DSIO_END)
 101     {
 102         /* Skip comments */
 103         if ((*ptr & D3DSI_OPCODE_MASK) == D3DSIO_COMMENT)
 104         {
 105             ptr += ((*ptr & D3DSI_COMMENTSIZE_MASK) >> D3DSI_COMMENTSIZE_SHIFT);
 106         }
 107     }
 108     ++ptr;
 109
 110     /* Return the shader size in bytes */
 111     return (ptr - byte_code) * sizeof(*ptr);
 112 }
 113
 114 DWORD WINAPI D3DXGetShaderVersion(const DWORD *byte_code)
 115 {
 116     TRACE("byte_code %p\n", byte_code);
 117
 118     return byte_code ? *byte_code : 0;
 119 }
 120
 121 LPCSTR WINAPI D3DXGetVertexShaderProfile(LPDIRECT3DDEVICE9 device)
 122 {
 123     D3DCAPS9 caps;
 124
 125     TRACE("device %p\n", device);
 126
 127     if (!device) return NULL;
 128
 129     IDirect3DDevice9_GetDeviceCaps(device,&caps);
 130
 131     switch (caps.VertexShaderVersion)
 132     {
 133     case D3DVS_VERSION(1, 1):
 134         return "vs_1_1";
 135     case D3DVS_VERSION(2, 0):
 136         if ((caps.VS20Caps.NumTemps>=13) &&
 137             (caps.VS20Caps.DynamicFlowControlDepth==24) &&
 138             (caps.VS20Caps.Caps&D3DPS20CAPS_PREDICATION))
 139         {
 140             return "vs_2_a";
 141         }
 142         return "vs_2_0";
 143     case D3DVS_VERSION(3, 0):
 144         return "vs_3_0";
 145     }
 146
 147     return NULL;
 148 }
 149
 150 HRESULT WINAPI D3DXFindShaderComment(CONST DWORD* byte_code, DWORD fourcc, LPCVOID* data, UINT* size)
 151 {
 152     CONST DWORD *ptr = byte_code;
 153
 154     TRACE("(%p, %x, %p, %p)\n", byte_code, fourcc, data, size);
 155
 156     if (data) *data = NULL;
 157     if (size) *size = 0;
 158
 159     if (!byte_code)
 160         return D3DERR_INVALIDCALL;
 161
 162     while (*++ptr != D3DSIO_END)
 163     {
 164         /* Check if it is a comment */
 165         if ((*ptr & D3DSI_OPCODE_MASK) == D3DSIO_COMMENT)
 166         {
 167             DWORD comment_size = (*ptr & D3DSI_COMMENTSIZE_MASK) >> D3DSI_COMMENTSIZE_SHIFT;
 168
 169             /* Check if this is the comment we are looking for */
 170             if (*(ptr + 1) == fourcc)
 171             {
 172                 UINT ctab_size = (comment_size - 1) * sizeof(DWORD);
 173                 LPCVOID ctab_data = ptr + 2;
 174                 if (size)
 175                     *size = ctab_size;
 176                 if (data)
 177                     *data = ctab_data;
 178                 TRACE("Returning comment data at %p with size %d\n", ctab_data, ctab_size);
 179                 return D3D_OK;
 180             }
 181             ptr += comment_size;
 182         }
 183     }
 184
 185     return S_FALSE;
 186 }
 187
 188 HRESULT WINAPI D3DXAssembleShader(LPCSTR data,
 189                                   UINT data_len,
 190                                   CONST D3DXMACRO* defines,
 191                                   LPD3DXINCLUDE include,
 192                                   DWORD flags,
 193                                   LPD3DXBUFFER* shader,
 194                                   LPD3DXBUFFER* error_messages)
 195 {
 196     /* Forward to d3dcompiler: the parameter types aren't really different,
 197        the actual data types are equivalent */
 198     HRESULT hr = D3DAssemble(data, data_len, NULL, (D3D_SHADER_MACRO *)defines,
 199                              (ID3DInclude *)include, flags, (ID3DBlob **)shader,
 200                              (ID3DBlob **)error_messages);
 201
 202     if(hr == E_FAIL) hr = D3DXERR_INVALIDDATA;
 203     return hr;
 204 }
 205
 206 /* D3DXInclude private implementation, used to implement
 207    D3DXAssembleShaderFromFile from D3DXAssembleShader */
 208 /* To be able to correctly resolve include search paths we have to store
 209    the pathname of each include file. We store the pathname pointer right
 210    before the file data. */
 211 static HRESULT WINAPI d3dincludefromfile_open(ID3DXInclude *iface,
 212                                               D3DXINCLUDE_TYPE include_type,
 213                                               LPCSTR filename, LPCVOID parent_data,
 214                                               LPCVOID *data, UINT *bytes) {
 215     const char *p, *parent_name = "";
 216     char *pathname = NULL;
 217     char **buffer = NULL;
 218     HANDLE file;
 219     UINT size;
 220
 221     if(parent_data != NULL)
 222         parent_name = *((const char **)parent_data - 1);
 223
 224     TRACE("Looking up for include file %s, parent %s\n", debugstr_a(filename), debugstr_a(parent_name));
 225
 226     if ((p = strrchr(parent_name, '\\')) || (p = strrchr(parent_name, '/'))) p++;
 227     else p = parent_name;
 228     pathname = HeapAlloc(GetProcessHeap(), 0, (p - parent_name) + strlen(filename) + 1);
 229     if(!pathname)
 230         return HRESULT_FROM_WIN32(GetLastError());
 231
 232     memcpy(pathname, parent_name, p - parent_name);
 233     strcpy(pathname + (p - parent_name), filename);
 234
 235     file = CreateFileA(pathname, GENERIC_READ, FILE_SHARE_READ, 0, OPEN_EXISTING, 0, 0);
 236     if(file == INVALID_HANDLE_VALUE)
 237         goto error;
 238
 239     TRACE("Include file found at pathname = %s\n", debugstr_a(pathname));
 240
 241     size = GetFileSize(file, NULL);
 242     if(size == INVALID_FILE_SIZE)
 243         goto error;
 244
 245     buffer = HeapAlloc(GetProcessHeap(), 0, size + sizeof(char *));
 246     if(!buffer)
 247         goto error;
 248     *buffer = pathname;
 249     if(!ReadFile(file, buffer + 1, size, bytes, NULL))
 250         goto error;
 251
 252     *data = buffer + 1;
 253
 254     CloseHandle(file);
 255     return S_OK;
 256
 257 error:
 258     CloseHandle(file);
 259     HeapFree(GetProcessHeap(), 0, pathname);
 260     HeapFree(GetProcessHeap(), 0, buffer);
 261     return HRESULT_FROM_WIN32(GetLastError());
 262 }
 263
 264 static HRESULT WINAPI d3dincludefromfile_close(ID3DXInclude *iface, LPCVOID data) {
 265     HeapFree(GetProcessHeap(), 0, *((char **)data - 1));
 266     HeapFree(GetProcessHeap(), 0, (char **)data - 1);
 267     return S_OK;
 268 }
 269
 270 static const struct ID3DXIncludeVtbl D3DXInclude_Vtbl = {
 271     d3dincludefromfile_open,
 272     d3dincludefromfile_close
 273 };
 274
 275 struct D3DXIncludeImpl {
 276     ID3DXInclude ID3DXInclude_iface;
 277 };
 278
 279 HRESULT WINAPI D3DXAssembleShaderFromFileA(LPCSTR filename,
 280                                            CONST D3DXMACRO* defines,
 281                                            LPD3DXINCLUDE include,
 282                                            DWORD flags,
 283                                            LPD3DXBUFFER* shader,
 284                                            LPD3DXBUFFER* error_messages)
 285 {
 286     LPWSTR filename_w = NULL;
 287     DWORD len;
 288     HRESULT ret;
 289
 290     if (!filename) return D3DXERR_INVALIDDATA;
 291
 292     len = MultiByteToWideChar(CP_ACP, 0, filename, -1, NULL, 0);
 293     filename_w = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 294     if (!filename_w) return E_OUTOFMEMORY;
 295     MultiByteToWideChar(CP_ACP, 0, filename, -1, filename_w, len);
 296
 297     ret = D3DXAssembleShaderFromFileW(filename_w, defines, include, flags, shader, error_messages);
 298
 299     HeapFree(GetProcessHeap(), 0, filename_w);
 300     return ret;
 301 }
 302
 303 HRESULT WINAPI D3DXAssembleShaderFromFileW(LPCWSTR filename,
 304                                            CONST D3DXMACRO* defines,
 305                                            LPD3DXINCLUDE include,
 306                                            DWORD flags,
 307                                            LPD3DXBUFFER* shader,
 308                                            LPD3DXBUFFER* error_messages)
 309 {
 310     void *buffer;
 311     DWORD len;
 312     HRESULT hr;
 313     struct D3DXIncludeImpl includefromfile;
 314
 315     if(FAILED(map_view_of_file(filename, &buffer, &len)))
 316         return D3DXERR_INVALIDDATA;
 317
 318     if(!include)
 319     {
 320         includefromfile.ID3DXInclude_iface.lpVtbl = &D3DXInclude_Vtbl;
 321         include = &includefromfile.ID3DXInclude_iface;
 322     }
 323
 324     hr = D3DXAssembleShader(buffer, len, defines, include, flags,
 325                             shader, error_messages);
 326
 327     UnmapViewOfFile(buffer);
 328     return hr;
 329 }
 330
 331 HRESULT WINAPI D3DXAssembleShaderFromResourceA(HMODULE module,
 332                                                LPCSTR resource,
 333                                                CONST D3DXMACRO* defines,
 334                                                LPD3DXINCLUDE include,
 335                                                DWORD flags,
 336                                                LPD3DXBUFFER* shader,
 337                                                LPD3DXBUFFER* error_messages)
 338 {
 339     HRSRC res;
 340     LPCSTR buffer;
 341     DWORD len;
 342
 343     if (!(res = FindResourceA(module, resource, (LPCSTR)RT_RCDATA)))
 344         return D3DXERR_INVALIDDATA;
 345     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 346         return D3DXERR_INVALIDDATA;
 347     return D3DXAssembleShader(buffer, len, defines, include, flags,
 348                               shader, error_messages);
 349 }
 350
 351 HRESULT WINAPI D3DXAssembleShaderFromResourceW(HMODULE module,
 352                                                LPCWSTR resource,
 353                                                CONST D3DXMACRO* defines,
 354                                                LPD3DXINCLUDE include,
 355                                                DWORD flags,
 356                                                LPD3DXBUFFER* shader,
 357                                                LPD3DXBUFFER* error_messages)
 358 {
 359     HRSRC res;
 360     LPCSTR buffer;
 361     DWORD len;
 362
 363     if (!(res = FindResourceW(module, resource, (LPCWSTR)RT_RCDATA)))
 364         return D3DXERR_INVALIDDATA;
 365     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 366         return D3DXERR_INVALIDDATA;
 367     return D3DXAssembleShader(buffer, len, defines, include, flags,
 368                               shader, error_messages);
 369 }
 370
 371 HRESULT WINAPI D3DXCompileShader(LPCSTR pSrcData,
 372                                  UINT srcDataLen,
 373                                  CONST D3DXMACRO* pDefines,
 374                                  LPD3DXINCLUDE pInclude,
 375                                  LPCSTR pFunctionName,
 376                                  LPCSTR pProfile,
 377                                  DWORD Flags,
 378                                  LPD3DXBUFFER* ppShader,
 379                                  LPD3DXBUFFER* ppErrorMsgs,
 380                                  LPD3DXCONSTANTTABLE * ppConstantTable)
 381 {
 382     HRESULT hr = D3DCompile(pSrcData, srcDataLen, NULL,
 383                             (D3D_SHADER_MACRO *)pDefines, (ID3DInclude *)pInclude,
 384                             pFunctionName, pProfile, Flags, 0,
 385                             (ID3DBlob **)ppShader, (ID3DBlob **)ppErrorMsgs);
 386
 387     if(SUCCEEDED(hr) && ppConstantTable)
 388         return D3DXGetShaderConstantTable(ID3DXBuffer_GetBufferPointer(*ppShader),
 389                                           ppConstantTable);
 390     return hr;
 391 }
 392
 393 HRESULT WINAPI D3DXCompileShaderFromFileA(LPCSTR filename,
 394                                           CONST D3DXMACRO* defines,
 395                                           LPD3DXINCLUDE include,
 396                                           LPCSTR entrypoint,
 397                                           LPCSTR profile,
 398                                           DWORD flags,
 399                                           LPD3DXBUFFER* shader,
 400                                           LPD3DXBUFFER* error_messages,
 401                                           LPD3DXCONSTANTTABLE* constant_table)
 402 {
 403     LPWSTR filename_w = NULL;
 404     DWORD len;
 405     HRESULT ret;
 406
 407     if (!filename) return D3DXERR_INVALIDDATA;
 408
 409     len = MultiByteToWideChar(CP_ACP, 0, filename, -1, NULL, 0);
 410     filename_w = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 411     if (!filename_w) return E_OUTOFMEMORY;
 412     MultiByteToWideChar(CP_ACP, 0, filename, -1, filename_w, len);
 413
 414     ret = D3DXCompileShaderFromFileW(filename_w, defines, include,
 415                                      entrypoint, profile, flags,
 416                                      shader, error_messages, constant_table);
 417
 418     HeapFree(GetProcessHeap(), 0, filename_w);
 419     return ret;
 420 }
 421
 422 HRESULT WINAPI D3DXCompileShaderFromFileW(LPCWSTR filename,
 423                                           CONST D3DXMACRO* defines,
 424                                           LPD3DXINCLUDE include,
 425                                           LPCSTR entrypoint,
 426                                           LPCSTR profile,
 427                                           DWORD flags,
 428                                           LPD3DXBUFFER* shader,
 429                                           LPD3DXBUFFER* error_messages,
 430                                           LPD3DXCONSTANTTABLE* constant_table)
 431 {
 432     void *buffer;
 433     DWORD len, filename_len;
 434     HRESULT hr;
 435     struct D3DXIncludeImpl includefromfile;
 436     char *filename_a;
 437
 438     if (FAILED(map_view_of_file(filename, &buffer, &len)))
 439         return D3DXERR_INVALIDDATA;
 440
 441     if (!include)
 442     {
 443         includefromfile.ID3DXInclude_iface.lpVtbl = &D3DXInclude_Vtbl;
 444         include = &includefromfile.ID3DXInclude_iface;
 445     }
 446
 447     filename_len = WideCharToMultiByte(CP_ACP, 0, filename, -1, NULL, 0, NULL, NULL);
 448     filename_a = HeapAlloc(GetProcessHeap(), 0, filename_len * sizeof(char));
 449     if (!filename_a)
 450     {
 451         UnmapViewOfFile(buffer);
 452         return E_OUTOFMEMORY;
 453     }
 454     WideCharToMultiByte(CP_ACP, 0, filename, -1, filename_a, filename_len, NULL, NULL);
 455
 456     hr = D3DCompile(buffer, len, filename_a, (const D3D_SHADER_MACRO *)defines,
 457                     (ID3DInclude *)include, entrypoint, profile, flags, 0,
 458                     (ID3DBlob **)shader, (ID3DBlob **)error_messages);
 459
 460     if (SUCCEEDED(hr) && constant_table)
 461         hr = D3DXGetShaderConstantTable(ID3DXBuffer_GetBufferPointer(*shader),
 462                                         constant_table);
 463
 464     HeapFree(GetProcessHeap(), 0, filename_a);
 465     UnmapViewOfFile(buffer);
 466     return hr;
 467 }
 468
 469 HRESULT WINAPI D3DXCompileShaderFromResourceA(HMODULE module,
 470                                               LPCSTR resource,
 471                                               CONST D3DXMACRO* defines,
 472                                               LPD3DXINCLUDE include,
 473                                               LPCSTR entrypoint,
 474                                               LPCSTR profile,
 475                                               DWORD flags,
 476                                               LPD3DXBUFFER* shader,
 477                                               LPD3DXBUFFER* error_messages,
 478                                               LPD3DXCONSTANTTABLE* constant_table)
 479 {
 480     HRSRC res;
 481     LPCSTR buffer;
 482     DWORD len;
 483
 484     if (!(res = FindResourceA(module, resource, (LPCSTR)RT_RCDATA)))
 485         return D3DXERR_INVALIDDATA;
 486     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 487         return D3DXERR_INVALIDDATA;
 488     return D3DXCompileShader(buffer, len, defines, include, entrypoint, profile,
 489                              flags, shader, error_messages, constant_table);
 490 }
 491
 492 HRESULT WINAPI D3DXCompileShaderFromResourceW(HMODULE module,
 493                                               LPCWSTR resource,
 494                                               CONST D3DXMACRO* defines,
 495                                               LPD3DXINCLUDE include,
 496                                               LPCSTR entrypoint,
 497                                               LPCSTR profile,
 498                                               DWORD flags,
 499                                               LPD3DXBUFFER* shader,
 500                                               LPD3DXBUFFER* error_messages,
 501                                               LPD3DXCONSTANTTABLE* constant_table)
 502 {
 503     HRSRC res;
 504     LPCSTR buffer;
 505     DWORD len;
 506
 507     if (!(res = FindResourceW(module, resource, (LPCWSTR)RT_RCDATA)))
 508         return D3DXERR_INVALIDDATA;
 509     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 510         return D3DXERR_INVALIDDATA;
 511     return D3DXCompileShader(buffer, len, defines, include, entrypoint, profile,
 512                              flags, shader, error_messages, constant_table);
 513 }
 514
 515 HRESULT WINAPI D3DXPreprocessShader(LPCSTR data,
 516                                     UINT data_len,
 517                                     CONST D3DXMACRO* defines,
 518                                     LPD3DXINCLUDE include,
 519                                     LPD3DXBUFFER* shader,
 520                                     LPD3DXBUFFER* error_messages)
 521 {
 522     TRACE("Forward to D3DPreprocess\n");
 523     return D3DPreprocess(data, data_len, NULL,
 524                          (const D3D_SHADER_MACRO *)defines, (ID3DInclude *)include,
 525                          (ID3DBlob **)shader, (ID3DBlob **)error_messages);
 526 }
 527
 528 HRESULT WINAPI D3DXPreprocessShaderFromFileA(LPCSTR filename,
 529                                              CONST D3DXMACRO* defines,
 530                                              LPD3DXINCLUDE include,
 531                                              LPD3DXBUFFER* shader,
 532                                              LPD3DXBUFFER* error_messages)
 533 {
 534     WCHAR *filename_w = NULL;
 535     DWORD len;
 536     HRESULT ret;
 537
 538     if (!filename) return D3DXERR_INVALIDDATA;
 539
 540     len = MultiByteToWideChar(CP_ACP, 0, filename, -1, NULL, 0);
 541     filename_w = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 542     if (!filename_w) return E_OUTOFMEMORY;
 543     MultiByteToWideChar(CP_ACP, 0, filename, -1, filename_w, len);
 544
 545     ret = D3DXPreprocessShaderFromFileW(filename_w, defines, include, shader, error_messages);
 546
 547     HeapFree(GetProcessHeap(), 0, filename_w);
 548     return ret;
 549 }
 550
 551 HRESULT WINAPI D3DXPreprocessShaderFromFileW(LPCWSTR filename,
 552                                              CONST D3DXMACRO* defines,
 553                                              LPD3DXINCLUDE include,
 554                                              LPD3DXBUFFER* shader,
 555                                              LPD3DXBUFFER* error_messages)
 556 {
 557     void *buffer;
 558     DWORD len;
 559     HRESULT hr;
 560     struct D3DXIncludeImpl includefromfile;
 561
 562     if (FAILED(map_view_of_file(filename, &buffer, &len)))
 563         return D3DXERR_INVALIDDATA;
 564
 565     if (!include)
 566     {
 567         includefromfile.ID3DXInclude_iface.lpVtbl = &D3DXInclude_Vtbl;
 568         include = &includefromfile.ID3DXInclude_iface;
 569     }
 570
 571     hr = D3DPreprocess(buffer, len, NULL,
 572                        (const D3D_SHADER_MACRO *)defines,
 573                        (ID3DInclude *) include,
 574                        (ID3DBlob **)shader, (ID3DBlob **)error_messages);
 575
 576     UnmapViewOfFile(buffer);
 577     return hr;
 578 }
 579
 580 HRESULT WINAPI D3DXPreprocessShaderFromResourceA(HMODULE module,
 581                                                  LPCSTR resource,
 582                                                  CONST D3DXMACRO* defines,
 583                                                  LPD3DXINCLUDE include,
 584                                                  LPD3DXBUFFER* shader,
 585                                                  LPD3DXBUFFER* error_messages)
 586 {
 587     HRSRC res;
 588     const char *buffer;
 589     DWORD len;
 590
 591     if (!(res = FindResourceA(module, resource, (LPCSTR)RT_RCDATA)))
 592         return D3DXERR_INVALIDDATA;
 593     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 594         return D3DXERR_INVALIDDATA;
 595     return D3DXPreprocessShader(buffer, len, defines, include,
 596                                 shader, error_messages);
 597 }
 598
 599 HRESULT WINAPI D3DXPreprocessShaderFromResourceW(HMODULE module,
 600                                                  LPCWSTR resource,
 601                                                  CONST D3DXMACRO* defines,
 602                                                  LPD3DXINCLUDE include,
 603                                                  LPD3DXBUFFER* shader,
 604                                                  LPD3DXBUFFER* error_messages)
 605 {
 606     HRSRC res;
 607     const char *buffer;
 608     DWORD len;
 609
 610     if (!(res = FindResourceW(module, resource, (const WCHAR *)RT_RCDATA)))
 611         return D3DXERR_INVALIDDATA;
 612     if (FAILED(load_resource_into_memory(module, res, (void **)&buffer, &len)))
 613         return D3DXERR_INVALIDDATA;
 614     return D3DXPreprocessShader(buffer, len, defines, include,
 615                                 shader, error_messages);
 616
 617 }
 618
 619 struct ctab_constant {
 620     D3DXCONSTANT_DESC desc;
 621     struct ctab_constant *members;
 622 };
 623
 624 static const struct ID3DXConstantTableVtbl ID3DXConstantTable_Vtbl;
 625
 626 struct ID3DXConstantTableImpl {
 627     ID3DXConstantTable ID3DXConstantTable_iface;
 628     LONG ref;
 629     char *ctab;
 630     DWORD size;
 631     D3DXCONSTANTTABLE_DESC desc;
 632     struct ctab_constant *constants;
 633 };
 634
 635 static void free_constant_table(struct ID3DXConstantTableImpl *table)
 636 {
 637     HeapFree(GetProcessHeap(), 0, table->constants);
 638     HeapFree(GetProcessHeap(), 0, table->ctab);
 639 }
 640
 641 static inline struct ID3DXConstantTableImpl *impl_from_ID3DXConstantTable(ID3DXConstantTable *iface)
 642 {
 643     return CONTAINING_RECORD(iface, struct ID3DXConstantTableImpl, ID3DXConstantTable_iface);
 644 }
 645
 646 static inline int is_vertex_shader(DWORD version)
 647 {
 648     return (version & 0xFFFF0000) == 0xFFFE0000;
 649 }
 650
 651 static DWORD calc_bytes(D3DXCONSTANT_DESC *desc)
 652 {
 653     if (desc->RegisterSet != D3DXRS_FLOAT4 && desc->RegisterSet != D3DXRS_SAMPLER)
 654         FIXME("Don't know how to calculate Bytes for constants of type %d\n",
 655                 desc->RegisterSet);
 656
 657     return 4 * desc->Elements * desc->Rows * desc->Columns;
 658 }
 659
 660 static inline struct ctab_constant *constant_from_handle(D3DXHANDLE handle)
 661 {
 662     return (struct ctab_constant *)handle;
 663 }
 664
 665 static inline D3DXHANDLE handle_from_constant(struct ctab_constant *constant)
 666 {
 667     return (D3DXHANDLE)constant;
 668 }
 669
 670 static inline struct ctab_constant *is_valid_constant(struct ID3DXConstantTableImpl *table, D3DXHANDLE handle)
 671 {
 672     struct ctab_constant *c = constant_from_handle(handle);
 673     UINT i;
 674
 675     if (!c) return NULL;
 676
 677     for (i = 0; i < table->desc.Constants; ++i)
 678     {
 679         if (&table->constants[i] == c) return c;
 680     }
 681
 682     return NULL;
 683 }
 684
 685 static inline void set_float_shader_constant(struct ID3DXConstantTableImpl *table, IDirect3DDevice9 *device,
 686                                              UINT register_index, const FLOAT *data, UINT count)
 687 {
 688     if (is_vertex_shader(table->desc.Version))
 689         IDirect3DDevice9_SetVertexShaderConstantF(device, register_index, data, count);
 690     else
 691         IDirect3DDevice9_SetPixelShaderConstantF(device, register_index, data, count);
 692 }
 693
 694 /*** IUnknown methods ***/
 695 static HRESULT WINAPI ID3DXConstantTableImpl_QueryInterface(ID3DXConstantTable *iface, REFIID riid, void **out)
 696 {
 697     TRACE("iface %p, riid %s, out %p.\n", iface, debugstr_guid(riid), out);
 698
 699     if (IsEqualGUID(riid, &IID_IUnknown) ||
 700         IsEqualGUID(riid, &IID_ID3DXBuffer) ||
 701         IsEqualGUID(riid, &IID_ID3DXConstantTable))
 702     {
 703         ID3DXConstantTable_AddRef(iface);
 704         *out = iface;
 705         return S_OK;
 706     }
 707
 708     WARN("Interface %s not found.\n", debugstr_guid(riid));
 709
 710     return E_NOINTERFACE;
 711 }
 712
 713 static ULONG WINAPI ID3DXConstantTableImpl_AddRef(ID3DXConstantTable *iface)
 714 {
 715     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 716
 717     TRACE("(%p)->(): AddRef from %d\n", This, This->ref);
 718
 719     return InterlockedIncrement(&This->ref);
 720 }
 721
 722 static ULONG WINAPI ID3DXConstantTableImpl_Release(ID3DXConstantTable *iface)
 723 {
 724     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 725     ULONG ref = InterlockedDecrement(&This->ref);
 726
 727     TRACE("(%p)->(): Release from %d\n", This, ref + 1);
 728
 729     if (!ref)
 730     {
 731         free_constant_table(This);
 732         HeapFree(GetProcessHeap(), 0, This);
 733     }
 734
 735     return ref;
 736 }
 737
 738 /*** ID3DXBuffer methods ***/
 739 static LPVOID WINAPI ID3DXConstantTableImpl_GetBufferPointer(ID3DXConstantTable *iface)
 740 {
 741     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 742
 743     TRACE("(%p)->()\n", This);
 744
 745     return This->ctab;
 746 }
 747
 748 static DWORD WINAPI ID3DXConstantTableImpl_GetBufferSize(ID3DXConstantTable *iface)
 749 {
 750     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 751
 752     TRACE("(%p)->()\n", This);
 753
 754     return This->size;
 755 }
 756
 757 /*** ID3DXConstantTable methods ***/
 758 static HRESULT WINAPI ID3DXConstantTableImpl_GetDesc(ID3DXConstantTable *iface, D3DXCONSTANTTABLE_DESC *desc)
 759 {
 760     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 761
 762     TRACE("(%p)->(%p)\n", This, desc);
 763
 764     if (!desc)
 765         return D3DERR_INVALIDCALL;
 766
 767     memcpy(desc, &This->desc, sizeof(This->desc));
 768
 769     return D3D_OK;
 770 }
 771
 772 static HRESULT WINAPI ID3DXConstantTableImpl_GetConstantDesc(ID3DXConstantTable *iface, D3DXHANDLE constant,
 773                                                              D3DXCONSTANT_DESC *desc, UINT *count)
 774 {
 775     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 776     struct ctab_constant *constant_info;
 777
 778     TRACE("(%p)->(%p, %p, %p)\n", This, constant, desc, count);
 779
 780     if (!constant)
 781         return D3DERR_INVALIDCALL;
 782
 783     /* Applications can pass the name of the constant in place of the handle */
 784     if (!is_valid_constant(This, constant))
 785     {
 786         constant = ID3DXConstantTable_GetConstantByName(iface, NULL, constant);
 787         if (!constant)
 788             return D3DERR_INVALIDCALL;
 789     }
 790
 791     constant_info = constant_from_handle(constant);
 792
 793     if (desc)
 794         *desc = constant_info->desc;
 795     if (count)
 796         *count = 1;
 797
 798     return D3D_OK;
 799 }
 800
 801 static UINT WINAPI ID3DXConstantTableImpl_GetSamplerIndex(ID3DXConstantTable *iface, D3DXHANDLE constant)
 802 {
 803     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 804     D3DXCONSTANT_DESC desc;
 805     UINT count = 1;
 806     HRESULT res;
 807
 808     TRACE("(%p)->(%p)\n", This, constant);
 809
 810     res = ID3DXConstantTable_GetConstantDesc(iface, constant, &desc, &count);
 811     if (FAILED(res))
 812         return (UINT)-1;
 813
 814     if (desc.RegisterSet != D3DXRS_SAMPLER)
 815         return (UINT)-1;
 816
 817     return desc.RegisterIndex;
 818 }
 819
 820 static D3DXHANDLE WINAPI ID3DXConstantTableImpl_GetConstant(ID3DXConstantTable *iface, D3DXHANDLE constant, UINT index)
 821 {
 822     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 823
 824     TRACE("(%p)->(%p, %d)\n", This, constant, index);
 825
 826     if (constant)
 827     {
 828         FIXME("Only top level constants supported\n");
 829         return NULL;
 830     }
 831
 832     if (index >= This->desc.Constants)
 833         return NULL;
 834
 835     return handle_from_constant(&This->constants[index]);
 836 }
 837
 838 static D3DXHANDLE WINAPI ID3DXConstantTableImpl_GetConstantByName(ID3DXConstantTable *iface, D3DXHANDLE constant, LPCSTR name)
 839 {
 840     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 841     UINT i;
 842
 843     TRACE("(%p)->(%p, %s)\n", This, constant, name);
 844
 845     if (!name)
 846         return NULL;
 847
 848     if (constant)
 849     {
 850         FIXME("Only top level constants supported\n");
 851         return NULL;
 852     }
 853
 854     for (i = 0; i < This->desc.Constants; i++)
 855         if (!strcmp(This->constants[i].desc.Name, name))
 856             return handle_from_constant(&This->constants[i]);
 857
 858     return NULL;
 859 }
 860
 861 static D3DXHANDLE WINAPI ID3DXConstantTableImpl_GetConstantElement(ID3DXConstantTable *iface, D3DXHANDLE constant, UINT index)
 862 {
 863     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 864
 865     FIXME("(%p)->(%p, %d): stub\n", This, constant, index);
 866
 867     return NULL;
 868 }
 869
 870 static HRESULT set_scalar_array(ID3DXConstantTable *iface, IDirect3DDevice9 *device, D3DXHANDLE constant, const void *data,
 871                                UINT count, D3DXPARAMETER_TYPE type)
 872 {
 873     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 874     D3DXCONSTANT_DESC desc;
 875     HRESULT hr;
 876     UINT i, desc_count = 1;
 877     float row[4] = {0.0f, 0.0f, 0.0f, 0.0f};
 878
 879     hr = ID3DXConstantTable_GetConstantDesc(iface, constant, &desc, &desc_count);
 880     if (FAILED(hr))
 881     {
 882         TRACE("ID3DXConstantTable_GetConstantDesc failed: %08x\n", hr);
 883         return D3DERR_INVALIDCALL;
 884     }
 885
 886     if (desc.Class != D3DXPC_SCALAR)
 887         return D3D_OK;
 888
 889     switch (desc.RegisterSet)
 890     {
 891         case D3DXRS_FLOAT4:
 892             for (i = 0; i < min(count, desc.RegisterCount); i++)
 893             {
 894                 /* We need the for loop since each IDirect3DDevice9_Set*ShaderConstantF expects a float4 */
 895                 switch(type)
 896                 {
 897                     case D3DXPT_FLOAT:
 898                         row[0] = ((float *)data)[i];
 899                         break;
 900                     case D3DXPT_INT:
 901                         row[0] = (float)((int *)data)[i];
 902                         break;
 903                     case D3DXPT_BOOL:
 904                         row[0] = ((BOOL *)data)[i] ? 1.0f : 0.0f;
 905                         break;
 906                     default:
 907                         FIXME("Unhandled type %#x\n", type);
 908                         return D3DERR_INVALIDCALL;
 909                 }
 910                 set_float_shader_constant(This, device, desc.RegisterIndex + i, row, 1);
 911             }
 912             break;
 913         default:
 914             FIXME("Handle other register sets\n");
 915             return E_NOTIMPL;
 916     }
 917
 918     return D3D_OK;
 919 }
 920
 921 static HRESULT set_vector_array(ID3DXConstantTable *iface, IDirect3DDevice9 *device, D3DXHANDLE constant, const void *data,
 922                                 UINT count, D3DXPARAMETER_TYPE type)
 923 {
 924     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
 925     D3DXCONSTANT_DESC desc;
 926     HRESULT hr;
 927     UINT i, j, desc_count = 1;
 928     float vec[4] = {0.0f, 0.0f, 0.0f, 0.0f};
 929
 930     hr = ID3DXConstantTable_GetConstantDesc(iface, constant, &desc, &desc_count);
 931     if (FAILED(hr))
 932     {
 933         TRACE("ID3DXConstantTable_GetConstantDesc failed: %08x\n", hr);
 934         return D3DERR_INVALIDCALL;
 935     }
 936
 937     if (desc.Class == D3DXPC_MATRIX_ROWS || desc.Class == D3DXPC_MATRIX_COLUMNS)
 938         return D3D_OK;
 939
 940     switch (desc.RegisterSet)
 941     {
 942         case D3DXRS_FLOAT4:
 943             for (i = 0; i < min(count, desc.RegisterCount); i++)
 944             {
 945                 switch (type)
 946                 {
 947                     case D3DXPT_FLOAT:
 948                         memcpy(vec, ((float *)data) + i * desc.Columns, desc.Columns * sizeof(float));
 949                         break;
 950                     case D3DXPT_INT:
 951                         for (j = 0; j < desc.Columns; j++)
 952                             vec[j] = (float)((int *)data)[i * desc.Columns + j];
 953                         break;
 954                     case D3DXPT_BOOL:
 955                         for (j = 0; j < desc.Columns; j++)
 956                             vec[j] = ((BOOL *)data)[i * desc.Columns + j] ? 1.0f : 0.0f;
 957                         break;
 958                     default:
 959                         FIXME("Unhandled type %#x\n", type);
 960                         return D3DERR_INVALIDCALL;
 961                 }
 962
 963                 set_float_shader_constant(This, device, desc.RegisterIndex + i, vec, 1);
 964             }
 965             break;
 966         default:
 967             FIXME("Unhandled register set %#x\n", desc.RegisterSet);
 968             return E_NOTIMPL;
 969     }
 970
 971     return D3D_OK;
 972 }
 973
 974 static HRESULT set_float_matrix(FLOAT *matrix, const D3DXCONSTANT_DESC *desc,
 975                                 UINT row_offset, UINT column_offset, UINT rows, UINT columns,
 976                                 const void *data, D3DXPARAMETER_TYPE type, UINT src_columns)
 977 {
 978     UINT i, j;
 979
 980     switch (type)
 981     {
 982         case D3DXPT_FLOAT:
 983             for (i = 0; i < rows; i++)
 984             {
 985                 for (j = 0; j < columns; j++)
 986                     matrix[i * row_offset + j * column_offset] = ((FLOAT *)data)[i * src_columns + j];
 987             }
 988             break;
 989         case D3DXPT_INT:
 990             for (i = 0; i < rows; i++)
 991             {
 992                 for (j = 0; j < columns; j++)
 993                     matrix[i * row_offset + j * column_offset] = ((INT *)data)[i * src_columns + j];
 994             }
 995             break;
 996         default:
 997             FIXME("Unhandled type %#x\n", type);
 998             return D3DERR_INVALIDCALL;
 999     }
1000
1001     return D3D_OK;
1002 }
1003
1004 static HRESULT set_matrix_array(ID3DXConstantTable *iface, IDirect3DDevice9 *device, D3DXHANDLE constant, const void *data,
1005                                 UINT count, D3DXPARAMETER_CLASS class, D3DXPARAMETER_TYPE type, UINT rows, UINT columns)
1006 {
1007     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1008     D3DXCONSTANT_DESC desc;
1009     HRESULT hr;
1010     UINT registers_per_matrix;
1011     UINT i, desc_count = 1;
1012     UINT num_rows, num_columns;
1013     UINT row_offset, column_offset;
1014     const DWORD *data_ptr;
1015     FLOAT matrix[16] = {0.0f, 0.0f, 0.0f, 0.0f,
1016                         0.0f, 0.0f, 0.0f, 0.0f,
1017                         0.0f, 0.0f, 0.0f, 0.0f,
1018                         0.0f, 0.0f, 0.0f, 0.0f};
1019
1020     hr = ID3DXConstantTable_GetConstantDesc(iface, constant, &desc, &desc_count);
1021     if (FAILED(hr))
1022     {
1023         TRACE("ID3DXConstantTable_GetConstantDesc failed: %08x\n", hr);
1024         return D3DERR_INVALIDCALL;
1025     }
1026
1027     if (desc.Class == D3DXPC_MATRIX_ROWS || desc.Class == D3DXPC_MATRIX_COLUMNS)
1028     {
1029         if (desc.Class == class)
1030         {
1031             column_offset = 1;
1032             row_offset = 4;
1033         }
1034         else
1035         {
1036             column_offset = 4;
1037             row_offset = 1;
1038         }
1039
1040         if (class == D3DXPC_MATRIX_ROWS)
1041         {
1042             num_rows = min(desc.Rows, rows);
1043             num_columns = min(desc.Columns, columns);
1044         }
1045         else
1046         {
1047             num_rows = min(desc.Columns, columns);
1048             num_columns = min(desc.Rows, rows);
1049         }
1050
1051         registers_per_matrix = (desc.Class == D3DXPC_MATRIX_ROWS) ? desc.Rows : desc.Columns;
1052     }
1053     else if (desc.Class == D3DXPC_SCALAR)
1054     {
1055         column_offset = 1;
1056         row_offset = 1;
1057         num_rows = min(desc.Rows, rows);
1058         num_columns = min(desc.Columns, columns);
1059         registers_per_matrix = 1;
1060     }
1061     else
1062     {
1063         FIXME("Unhandled variable class %#x\n", desc.Class);
1064         return D3D_OK;
1065     }
1066
1067     switch (desc.RegisterSet)
1068     {
1069         case D3DXRS_FLOAT4:
1070             data_ptr = data;
1071             for (i = 0; i < count; i++)
1072             {
1073                 if (registers_per_matrix * (i + 1) > desc.RegisterCount)
1074                     break;
1075
1076                 hr = set_float_matrix(matrix, &desc, row_offset, column_offset, num_rows, num_columns, data_ptr, type, columns);
1077                 if (FAILED(hr)) return hr;
1078
1079                 set_float_shader_constant(This, device, desc.RegisterIndex + i * registers_per_matrix, matrix, registers_per_matrix);
1080
1081                 data_ptr += rows * columns;
1082             }
1083             break;
1084         default:
1085             FIXME("Unhandled register set %#x\n", desc.RegisterSet);
1086             return E_NOTIMPL;
1087     }
1088
1089     return D3D_OK;
1090 }
1091
1092 static HRESULT set_matrix_pointer_array(ID3DXConstantTable *iface, IDirect3DDevice9 *device, D3DXHANDLE constant,
1093                                 const D3DXMATRIX **data, UINT count, D3DXPARAMETER_CLASS class)
1094 {
1095     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1096     D3DXCONSTANT_DESC desc;
1097     HRESULT hr;
1098     UINT registers_per_matrix;
1099     UINT i, desc_count = 1;
1100     UINT num_rows, num_columns;
1101     UINT row_offset, column_offset;
1102     FLOAT matrix[16] = {0.0f, 0.0f, 0.0f, 0.0f,
1103                         0.0f, 0.0f, 0.0f, 0.0f,
1104                         0.0f, 0.0f, 0.0f, 0.0f,
1105                         0.0f, 0.0f, 0.0f, 0.0f};
1106
1107     hr = ID3DXConstantTable_GetConstantDesc(iface, constant, &desc, &desc_count);
1108     if (FAILED(hr))
1109     {
1110         TRACE("ID3DXConstantTable_GetConstantDesc failed: %08x\n", hr);
1111         return D3DERR_INVALIDCALL;
1112     }
1113
1114     if (desc.Class == D3DXPC_MATRIX_ROWS || desc.Class == D3DXPC_MATRIX_COLUMNS)
1115     {
1116         if (desc.Class == class)
1117         {
1118             column_offset = 1;
1119             row_offset = 4;
1120         }
1121         else
1122         {
1123             column_offset = 4;
1124             row_offset = 1;
1125         }
1126
1127         if (class == D3DXPC_MATRIX_ROWS)
1128         {
1129             num_rows = desc.Rows;
1130             num_columns = desc.Columns;
1131         }
1132         else
1133         {
1134             num_rows = desc.Columns;
1135             num_columns = desc.Rows;
1136         }
1137
1138         registers_per_matrix = (desc.Class == D3DXPC_MATRIX_ROWS) ? desc.Rows : desc.Columns;
1139     }
1140     else if (desc.Class == D3DXPC_SCALAR)
1141     {
1142         registers_per_matrix = 1;
1143         column_offset = 1;
1144         row_offset = 1;
1145         num_rows = desc.Rows;
1146         num_columns = desc.Columns;
1147     }
1148     else if (desc.Class == D3DXPC_VECTOR)
1149     {
1150         registers_per_matrix = 1;
1151
1152         if (class == D3DXPC_MATRIX_ROWS)
1153         {
1154             column_offset = 1;
1155             row_offset = 4;
1156             num_rows = desc.Rows;
1157             num_columns = desc.Columns;
1158         }
1159         else
1160         {
1161             column_offset = 4;
1162             row_offset = 1;
1163             num_rows = desc.Columns;
1164             num_columns = desc.Rows;
1165         }
1166     }
1167     else
1168     {
1169         FIXME("Unhandled variable class %#x\n", desc.Class);
1170         return D3D_OK;
1171     }
1172
1173     switch (desc.RegisterSet)
1174     {
1175         case D3DXRS_FLOAT4:
1176             for (i = 0; i < count; i++)
1177             {
1178                 if (registers_per_matrix * (i + 1) > desc.RegisterCount)
1179                     break;
1180
1181                 hr = set_float_matrix(matrix, &desc, row_offset, column_offset, num_rows, num_columns, *data, D3DXPT_FLOAT, 4);
1182                 if (FAILED(hr)) return hr;
1183
1184                 set_float_shader_constant(This, device, desc.RegisterIndex + i * registers_per_matrix, matrix, registers_per_matrix);
1185
1186                 data++;
1187             }
1188             break;
1189         default:
1190             FIXME("Unhandled register set %#x\n", desc.RegisterSet);
1191             return E_NOTIMPL;
1192     }
1193
1194     return D3D_OK;
1195 }
1196
1197 static HRESULT WINAPI ID3DXConstantTableImpl_SetDefaults(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device)
1198 {
1199     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1200     UINT i;
1201
1202     TRACE("(%p)->(%p)\n", This, device);
1203
1204     if (!device)
1205         return D3DERR_INVALIDCALL;
1206
1207     for (i = 0; i < This->desc.Constants; i++)
1208     {
1209         D3DXCONSTANT_DESC *desc = &This->constants[i].desc;
1210
1211         if (!desc->DefaultValue)
1212             continue;
1213
1214         set_float_shader_constant(This, device, desc->RegisterIndex, desc->DefaultValue, desc->RegisterCount);
1215     }
1216
1217     return D3D_OK;
1218 }
1219
1220 static HRESULT WINAPI ID3DXConstantTableImpl_SetValue(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1221                                                       D3DXHANDLE constant, LPCVOID data, UINT bytes)
1222 {
1223     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1224     HRESULT hr;
1225     UINT elements;
1226     UINT count = 1;
1227     D3DXCONSTANT_DESC desc;
1228
1229     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, data, bytes);
1230
1231     if (!device || !constant || !data)
1232         return D3DERR_INVALIDCALL;
1233
1234     hr = ID3DXConstantTable_GetConstantDesc(iface, constant, &desc, &count);
1235     if (FAILED(hr))
1236         return hr;
1237
1238     elements = bytes / (desc.Bytes / desc.Elements);
1239
1240     switch (desc.Class)
1241     {
1242         case D3DXPC_SCALAR:
1243             return set_scalar_array(iface, device, constant, data, elements, desc.Type);
1244         case D3DXPC_VECTOR:
1245             return set_vector_array(iface, device, constant, data,elements, desc.Type);
1246         case D3DXPC_MATRIX_ROWS:
1247         case D3DXPC_MATRIX_COLUMNS:
1248             return set_matrix_array(iface, device, constant, data, elements,
1249                     D3DXPC_MATRIX_ROWS, desc.Type, desc.Rows, desc.Columns);
1250         default:
1251             FIXME("Unhandled parameter class %#x\n", desc.Class);
1252             return D3DERR_INVALIDCALL;
1253     }
1254 }
1255
1256 static HRESULT WINAPI ID3DXConstantTableImpl_SetBool(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1257                                                      D3DXHANDLE constant, BOOL b)
1258 {
1259     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1260
1261     TRACE("(%p)->(%p, %p, %d)\n", This, device, constant, b);
1262
1263     return set_scalar_array(iface, device, constant, &b, 1, D3DXPT_BOOL);
1264 }
1265
1266 static HRESULT WINAPI ID3DXConstantTableImpl_SetBoolArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1267                                                           D3DXHANDLE constant, CONST BOOL* b, UINT count)
1268 {
1269     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1270
1271     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, b, count);
1272
1273     return set_scalar_array(iface, device, constant, b, count, D3DXPT_BOOL);
1274 }
1275
1276 static HRESULT WINAPI ID3DXConstantTableImpl_SetInt(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device, D3DXHANDLE constant, INT n)
1277 {
1278     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1279
1280     TRACE("(%p)->(%p, %p, %d)\n", This, device, constant, n);
1281
1282     return set_scalar_array(iface, device, constant, &n, 1, D3DXPT_INT);
1283 }
1284
1285 static HRESULT WINAPI ID3DXConstantTableImpl_SetIntArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1286                                                          D3DXHANDLE constant, CONST INT* n, UINT count)
1287 {
1288     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1289
1290     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, n, count);
1291
1292     return set_scalar_array(iface, device, constant, n, count, D3DXPT_INT);
1293 }
1294
1295 static HRESULT WINAPI ID3DXConstantTableImpl_SetFloat(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1296                                                       D3DXHANDLE constant, FLOAT f)
1297 {
1298     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1299
1300     TRACE("(%p)->(%p, %p, %f)\n", This, device, constant, f);
1301
1302     return set_scalar_array(iface, device, constant, &f, 1, D3DXPT_FLOAT);
1303 }
1304
1305 static HRESULT WINAPI ID3DXConstantTableImpl_SetFloatArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1306                                                            D3DXHANDLE constant, CONST FLOAT *f, UINT count)
1307 {
1308     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1309
1310     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, f, count);
1311
1312     return set_scalar_array(iface, device, constant, f, count, D3DXPT_FLOAT);
1313 }
1314
1315 static HRESULT WINAPI ID3DXConstantTableImpl_SetVector(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1316                                                        D3DXHANDLE constant, CONST D3DXVECTOR4 *vector)
1317 {
1318     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1319
1320     TRACE("(%p)->(%p, %p, %p)\n", This, device, constant, vector);
1321
1322     return set_vector_array(iface, device, constant, vector, 1, D3DXPT_FLOAT);
1323 }
1324
1325 static HRESULT WINAPI ID3DXConstantTableImpl_SetVectorArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1326                                                             D3DXHANDLE constant, CONST D3DXVECTOR4 *vector, UINT count)
1327 {
1328     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1329
1330     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, vector, count);
1331
1332     return set_vector_array(iface, device, constant, vector, count, D3DXPT_FLOAT);
1333 }
1334
1335 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrix(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1336                                                        D3DXHANDLE constant, CONST D3DXMATRIX *matrix)
1337 {
1338     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1339
1340     TRACE("(%p)->(%p, %p, %p)\n", This, device, constant, matrix);
1341
1342     return set_matrix_array(iface, device, constant, matrix, 1, D3DXPC_MATRIX_ROWS, D3DXPT_FLOAT, 4, 4);
1343 }
1344
1345 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1346                                                             D3DXHANDLE constant, CONST D3DXMATRIX *matrix, UINT count)
1347 {
1348     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1349
1350     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, matrix, count);
1351
1352     return set_matrix_array(iface, device, constant, matrix, count, D3DXPC_MATRIX_ROWS, D3DXPT_FLOAT, 4, 4);
1353 }
1354
1355 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixPointerArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1356                                                                    D3DXHANDLE constant, CONST D3DXMATRIX **matrix, UINT count)
1357 {
1358     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1359
1360     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, matrix, count);
1361
1362     return set_matrix_pointer_array(iface, device, constant, matrix, count, D3DXPC_MATRIX_ROWS);
1363 }
1364
1365 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixTranspose(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1366                                                                 D3DXHANDLE constant, CONST D3DXMATRIX *matrix)
1367 {
1368     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1369
1370     TRACE("(%p)->(%p, %p, %p)\n", This, device, constant, matrix);
1371
1372     return set_matrix_array(iface, device, constant, matrix, 1, D3DXPC_MATRIX_COLUMNS, D3DXPT_FLOAT, 4, 4);
1373 }
1374
1375 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixTransposeArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1376                                                                      D3DXHANDLE constant, CONST D3DXMATRIX *matrix, UINT count)
1377 {
1378     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1379
1380     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, matrix, count);
1381
1382     return set_matrix_array(iface, device, constant, matrix, count, D3DXPC_MATRIX_COLUMNS, D3DXPT_FLOAT, 4, 4);
1383 }
1384
1385 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixTransposePointerArray(ID3DXConstantTable *iface, LPDIRECT3DDEVICE9 device,
1386                                                                             D3DXHANDLE constant, CONST D3DXMATRIX **matrix, UINT count)
1387 {
1388     struct ID3DXConstantTableImpl *This = impl_from_ID3DXConstantTable(iface);
1389
1390     TRACE("(%p)->(%p, %p, %p, %d)\n", This, device, constant, matrix, count);
1391
1392     return set_matrix_pointer_array(iface, device, constant, matrix, count, D3DXPC_MATRIX_COLUMNS);
1393 }
1394
1395 static const struct ID3DXConstantTableVtbl ID3DXConstantTable_Vtbl =
1396 {
1397     /*** IUnknown methods ***/
1398     ID3DXConstantTableImpl_QueryInterface,
1399     ID3DXConstantTableImpl_AddRef,
1400     ID3DXConstantTableImpl_Release,
1401     /*** ID3DXBuffer methods ***/
1402     ID3DXConstantTableImpl_GetBufferPointer,
1403     ID3DXConstantTableImpl_GetBufferSize,
1404     /*** ID3DXConstantTable methods ***/
1405     ID3DXConstantTableImpl_GetDesc,
1406     ID3DXConstantTableImpl_GetConstantDesc,
1407     ID3DXConstantTableImpl_GetSamplerIndex,
1408     ID3DXConstantTableImpl_GetConstant,
1409     ID3DXConstantTableImpl_GetConstantByName,
1410     ID3DXConstantTableImpl_GetConstantElement,
1411     ID3DXConstantTableImpl_SetDefaults,
1412     ID3DXConstantTableImpl_SetValue,
1413     ID3DXConstantTableImpl_SetBool,
1414     ID3DXConstantTableImpl_SetBoolArray,
1415     ID3DXConstantTableImpl_SetInt,
1416     ID3DXConstantTableImpl_SetIntArray,
1417     ID3DXConstantTableImpl_SetFloat,
1418     ID3DXConstantTableImpl_SetFloatArray,
1419     ID3DXConstantTableImpl_SetVector,
1420     ID3DXConstantTableImpl_SetVectorArray,
1421     ID3DXConstantTableImpl_SetMatrix,
1422     ID3DXConstantTableImpl_SetMatrixArray,
1423     ID3DXConstantTableImpl_SetMatrixPointerArray,
1424     ID3DXConstantTableImpl_SetMatrixTranspose,
1425     ID3DXConstantTableImpl_SetMatrixTransposeArray,
1426     ID3DXConstantTableImpl_SetMatrixTransposePointerArray
1427 };
1428
1429 static HRESULT parse_ctab_constant_type(const D3DXSHADER_TYPEINFO *type, struct ctab_constant *constant)
1430 {
1431     constant->desc.Class = type->Class;
1432     constant->desc.Type = type->Type;
1433     constant->desc.Rows = type->Rows;
1434     constant->desc.Columns = type->Columns;
1435     constant->desc.Elements = type->Elements;
1436     constant->desc.StructMembers = type->StructMembers;
1437     constant->desc.Bytes = calc_bytes(&constant->desc);
1438
1439     TRACE("class = %d, type = %d, rows = %d, columns = %d, elements = %d, struct_members = %d\n",
1440           constant->desc.Class, constant->desc.Type, constant->desc.Rows,
1441           constant->desc.Columns, constant->desc.Elements, constant->desc.StructMembers);
1442
1443     if ((constant->desc.Class == D3DXPC_STRUCT) && constant->desc.StructMembers)
1444     {
1445         FIXME("Struct not supported yet\n");
1446         return E_NOTIMPL;
1447     }
1448
1449     return D3D_OK;
1450 }
1451
1452 HRESULT WINAPI D3DXGetShaderConstantTableEx(CONST DWORD *byte_code,
1453                                             DWORD flags,
1454                                             LPD3DXCONSTANTTABLE *constant_table)
1455 {
1456     struct ID3DXConstantTableImpl *object = NULL;
1457     HRESULT hr;
1458     LPCVOID data;
1459     UINT size;
1460     const D3DXSHADER_CONSTANTTABLE* ctab_header;
1461     D3DXSHADER_CONSTANTINFO* constant_info;
1462     DWORD i;
1463
1464     TRACE("(%p, %x, %p)\n", byte_code, flags, constant_table);
1465
1466     if (!byte_code || !constant_table)
1467     {
1468         WARN("Invalid argument specified.\n");
1469         return D3DERR_INVALIDCALL;
1470     }
1471
1472     if (flags) FIXME("Flags (%#x) are not handled, yet!\n", flags);
1473
1474     hr = D3DXFindShaderComment(byte_code, MAKEFOURCC('C','T','A','B'), &data, &size);
1475     if (hr != D3D_OK)
1476     {
1477         WARN("CTAB not found.\n");
1478         return D3DXERR_INVALIDDATA;
1479     }
1480
1481     if (size < sizeof(D3DXSHADER_CONSTANTTABLE))
1482     {
1483         WARN("Invalid CTAB size.\n");
1484         return D3DXERR_INVALIDDATA;
1485     }
1486
1487     ctab_header = (const D3DXSHADER_CONSTANTTABLE *)data;
1488     if (ctab_header->Size != sizeof(D3DXSHADER_CONSTANTTABLE))
1489     {
1490         WARN("Invalid D3DXSHADER_CONSTANTTABLE size.\n");
1491         return D3DXERR_INVALIDDATA;
1492     }
1493
1494     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object));
1495     if (!object)
1496     {
1497         ERR("Out of memory\n");
1498         return E_OUTOFMEMORY;
1499     }
1500
1501     object->ID3DXConstantTable_iface.lpVtbl = &ID3DXConstantTable_Vtbl;
1502     object->ref = 1;
1503
1504     object->ctab = HeapAlloc(GetProcessHeap(), 0, size);
1505     if (!object->ctab)
1506     {
1507         ERR("Out of memory\n");
1508         HeapFree(GetProcessHeap(), 0, object);
1509         return E_OUTOFMEMORY;
1510     }
1511     object->size = size;
1512     memcpy(object->ctab, data, object->size);
1513
1514     object->desc.Creator = ctab_header->Creator ? object->ctab + ctab_header->Creator : NULL;
1515     object->desc.Version = ctab_header->Version;
1516     object->desc.Constants = ctab_header->Constants;
1517     TRACE("Creator %s, Version %x, Constants %u, Target %s\n",
1518             debugstr_a(object->desc.Creator), object->desc.Version, object->desc.Constants,
1519             debugstr_a(ctab_header->Target ? object->ctab + ctab_header->Target : NULL));
1520
1521     if (object->desc.Constants > 65535)
1522     {
1523         FIXME("Too many constants (%u)\n", object->desc.Constants);
1524         hr = E_NOTIMPL;
1525         goto error;
1526     }
1527
1528     object->constants = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
1529                                   sizeof(*object->constants) * object->desc.Constants);
1530     if (!object->constants)
1531     {
1532          ERR("Out of memory\n");
1533          hr = E_OUTOFMEMORY;
1534          goto error;
1535     }
1536
1537     constant_info = (LPD3DXSHADER_CONSTANTINFO)(object->ctab + ctab_header->ConstantInfo);
1538     for (i = 0; i < ctab_header->Constants; i++)
1539     {
1540         TRACE("name = %s\n", object->ctab + constant_info[i].Name);
1541         object->constants[i].desc.Name = object->ctab + constant_info[i].Name;
1542         object->constants[i].desc.RegisterSet = constant_info[i].RegisterSet;
1543         object->constants[i].desc.RegisterIndex = constant_info[i].RegisterIndex;
1544         object->constants[i].desc.RegisterCount = constant_info[i].RegisterCount;
1545         object->constants[i].desc.DefaultValue = constant_info[i].DefaultValue
1546                 ? object->ctab + constant_info[i].DefaultValue : NULL;
1547
1548         hr = parse_ctab_constant_type((LPD3DXSHADER_TYPEINFO)(object->ctab + constant_info[i].TypeInfo),
1549              &object->constants[i]);
1550         if (hr != D3D_OK)
1551             goto error;
1552     }
1553
1554     *constant_table = &object->ID3DXConstantTable_iface;
1555
1556     return D3D_OK;
1557
1558 error:
1559
1560     free_constant_table(object);
1561     HeapFree(GetProcessHeap(), 0, object);
1562
1563     return hr;
1564 }
1565
1566 HRESULT WINAPI D3DXGetShaderConstantTable(CONST DWORD* byte_code,
1567                                           LPD3DXCONSTANTTABLE* constant_table)
1568 {
1569     TRACE("(%p, %p): Forwarded to D3DXGetShaderConstantTableEx\n", byte_code, constant_table);
1570
1571     return D3DXGetShaderConstantTableEx(byte_code, 0, constant_table);
1572 }
1573
1574 HRESULT WINAPI D3DXGetShaderSamplers(CONST DWORD *byte_code, LPCSTR *samplers, UINT *count)
1575 {
1576     HRESULT hr;
1577     UINT i, sampler_count = 0;
1578     UINT size;
1579     LPCSTR data;
1580     const D3DXSHADER_CONSTANTTABLE *ctab_header;
1581     const D3DXSHADER_CONSTANTINFO *constant_info;
1582
1583     TRACE("byte_code %p, samplers %p, count %p\n", byte_code, samplers, count);
1584
1585     if (count) *count = 0;
1586
1587     hr = D3DXFindShaderComment(byte_code, MAKEFOURCC('C','T','A','B'), (LPCVOID *)&data, &size);
1588     if (hr != D3D_OK) return D3D_OK;
1589
1590     if (size < sizeof(D3DXSHADER_CONSTANTTABLE)) return D3D_OK;
1591
1592     ctab_header = (const D3DXSHADER_CONSTANTTABLE *)data;
1593     if (ctab_header->Size != sizeof(*ctab_header)) return D3D_OK;
1594
1595     constant_info = (D3DXSHADER_CONSTANTINFO *)(data + ctab_header->ConstantInfo);
1596     for (i = 0; i < ctab_header->Constants; i++)
1597     {
1598         const D3DXSHADER_TYPEINFO *type;
1599
1600         TRACE("name = %s\n", data + constant_info[i].Name);
1601
1602         type = (D3DXSHADER_TYPEINFO *)(data + constant_info[i].TypeInfo);
1603
1604         if (type->Type == D3DXPT_SAMPLER
1605                 || type->Type == D3DXPT_SAMPLER1D
1606                 || type->Type == D3DXPT_SAMPLER2D
1607                 || type->Type == D3DXPT_SAMPLER3D
1608                 || type->Type == D3DXPT_SAMPLERCUBE)
1609         {
1610             if (samplers) samplers[sampler_count] = data + constant_info[i].Name;
1611
1612             ++sampler_count;
1613         }
1614     }
1615
1616     TRACE("Found %u samplers\n", sampler_count);
1617
1618     if (count) *count = sampler_count;
1619
1620     return D3D_OK;
1621 }