dlls/d3dx9_36/shader.c

   1 /*
   2  * Copyright 2008 Luis Busquets
   3  * Copyright 2009 Matteo Bruni
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include "config.h"
  21 #include "wine/port.h"
  22 #include "wine/debug.h"
  23 #include "wine/unicode.h"
  24 #include "windef.h"
  25 #include "wingdi.h"
  26 #include "wine/wpp.h"
  27 #include "d3dx9_36_private.h"
  28
  29 WINE_DEFAULT_DEBUG_CHANNEL(d3dx);
  30
  31 LPCSTR WINAPI D3DXGetPixelShaderProfile(LPDIRECT3DDEVICE9 device)
  32 {
  33     D3DCAPS9 caps;
  34
  35     TRACE("device %p\n", device);
  36
  37     if (!device) return NULL;
  38
  39     IDirect3DDevice9_GetDeviceCaps(device,&caps);
  40
  41     switch (caps.PixelShaderVersion)
  42     {
  43     case D3DPS_VERSION(1, 1):
  44         return "ps_1_1";
  45
  46     case D3DPS_VERSION(1, 2):
  47         return "ps_1_2";
  48
  49     case D3DPS_VERSION(1, 3):
  50         return "ps_1_3";
  51
  52     case D3DPS_VERSION(1, 4):
  53         return "ps_1_4";
  54
  55     case D3DPS_VERSION(2, 0):
  56         if ((caps.PS20Caps.NumTemps>=22)                          &&
  57             (caps.PS20Caps.Caps&D3DPS20CAPS_ARBITRARYSWIZZLE)     &&
  58             (caps.PS20Caps.Caps&D3DPS20CAPS_GRADIENTINSTRUCTIONS) &&
  59             (caps.PS20Caps.Caps&D3DPS20CAPS_PREDICATION)          &&
  60             (caps.PS20Caps.Caps&D3DPS20CAPS_NODEPENDENTREADLIMIT) &&
  61             (caps.PS20Caps.Caps&D3DPS20CAPS_NOTEXINSTRUCTIONLIMIT))
  62         {
  63             return "ps_2_a";
  64         }
  65         if ((caps.PS20Caps.NumTemps>=32)                          &&
  66             (caps.PS20Caps.Caps&D3DPS20CAPS_NOTEXINSTRUCTIONLIMIT))
  67         {
  68             return "ps_2_b";
  69         }
  70         return "ps_2_0";
  71
  72     case D3DPS_VERSION(3, 0):
  73         return "ps_3_0";
  74     }
  75
  76     return NULL;
  77 }
  78
  79 UINT WINAPI D3DXGetShaderSize(const DWORD *byte_code)
  80 {
  81     const DWORD *ptr = byte_code;
  82
  83     TRACE("byte_code %p\n", byte_code);
  84
  85     if (!ptr) return 0;
  86
  87     /* Look for the END token, skipping the VERSION token */
  88     while (*++ptr != D3DSIO_END)
  89     {
  90         /* Skip comments */
  91         if ((*ptr & D3DSI_OPCODE_MASK) == D3DSIO_COMMENT)
  92         {
  93             ptr += ((*ptr & D3DSI_COMMENTSIZE_MASK) >> D3DSI_COMMENTSIZE_SHIFT);
  94         }
  95     }
  96     ++ptr;
  97
  98     /* Return the shader size in bytes */
  99     return (ptr - byte_code) * sizeof(*ptr);
 100 }
 101
 102 DWORD WINAPI D3DXGetShaderVersion(const DWORD *byte_code)
 103 {
 104     TRACE("byte_code %p\n", byte_code);
 105
 106     return byte_code ? *byte_code : 0;
 107 }
 108
 109 LPCSTR WINAPI D3DXGetVertexShaderProfile(LPDIRECT3DDEVICE9 device)
 110 {
 111     D3DCAPS9 caps;
 112
 113     TRACE("device %p\n", device);
 114
 115     if (!device) return NULL;
 116
 117     IDirect3DDevice9_GetDeviceCaps(device,&caps);
 118
 119     switch (caps.VertexShaderVersion)
 120     {
 121     case D3DVS_VERSION(1, 1):
 122         return "vs_1_1";
 123     case D3DVS_VERSION(2, 0):
 124         if ((caps.VS20Caps.NumTemps>=13) &&
 125             (caps.VS20Caps.DynamicFlowControlDepth==24) &&
 126             (caps.VS20Caps.Caps&D3DPS20CAPS_PREDICATION))
 127         {
 128             return "vs_2_a";
 129         }
 130         return "vs_2_0";
 131     case D3DVS_VERSION(3, 0):
 132         return "vs_3_0";
 133     }
 134
 135     return NULL;
 136 }
 137
 138 HRESULT WINAPI D3DXFindShaderComment(CONST DWORD* byte_code, DWORD fourcc, LPCVOID* data, UINT* size)
 139 {
 140     CONST DWORD *ptr = byte_code;
 141
 142     TRACE("(%p, %x, %p, %p)\n", byte_code, fourcc, data, size);
 143
 144     if (!byte_code)
 145         return D3DERR_INVALIDCALL;
 146
 147     while (*++ptr != D3DSIO_END)
 148     {
 149         /* Check if it is a comment */
 150         if ((*ptr & D3DSI_OPCODE_MASK) == D3DSIO_COMMENT)
 151         {
 152             DWORD comment_size = (*ptr & D3DSI_COMMENTSIZE_MASK) >> D3DSI_COMMENTSIZE_SHIFT;
 153
 154             /* Check if this is the comment we are looking for */
 155             if (*(ptr + 1) == fourcc)
 156             {
 157                 UINT ctab_size = (comment_size - 1) * sizeof(DWORD);
 158                 LPCVOID ctab_data = ptr + 2;
 159                 if (size)
 160                     *size = ctab_size;
 161                 if (data)
 162                     *data = ctab_data;
 163                 TRACE("Returning comment data at %p with size %d\n", ctab_data, ctab_size);
 164                 return D3D_OK;
 165             }
 166             ptr += comment_size;
 167         }
 168     }
 169
 170     return S_FALSE;
 171 }
 172
 173 #define BUFFER_INITIAL_CAPACITY 256
 174
 175 struct mem_file_desc
 176 {
 177     const char *buffer;
 178     unsigned int size;
 179     unsigned int pos;
 180 };
 181
 182 struct mem_file_desc current_shader;
 183 LPD3DXINCLUDE current_include;
 184 char *wpp_output;
 185 int wpp_output_capacity, wpp_output_size;
 186
 187 char *wpp_messages;
 188 int wpp_messages_capacity, wpp_messages_size;
 189
 190 /* Mutex used to guarantee a single invocation
 191    of the D3DXAssembleShader function (or its variants) at a time.
 192    This is needed as wpp isn't thread-safe */
 193 static CRITICAL_SECTION wpp_mutex;
 194 static CRITICAL_SECTION_DEBUG wpp_mutex_debug =
 195 {
 196     0, 0, &wpp_mutex,
 197     { &wpp_mutex_debug.ProcessLocksList,
 198       &wpp_mutex_debug.ProcessLocksList },
 199       0, 0, { (DWORD_PTR)(__FILE__ ": wpp_mutex") }
 200 };
 201 static CRITICAL_SECTION wpp_mutex = { &wpp_mutex_debug, -1, 0, 0, 0, 0 };
 202
 203 /* Preprocessor error reporting functions */
 204 static void wpp_write_message(const char *fmt, va_list args)
 205 {
 206     char* newbuffer;
 207     int rc, newsize;
 208
 209     if(wpp_messages_capacity == 0)
 210     {
 211         wpp_messages = HeapAlloc(GetProcessHeap(), 0, MESSAGEBUFFER_INITIAL_SIZE);
 212         if(wpp_messages == NULL)
 213         {
 214             ERR("Error allocating memory for parser messages\n");
 215             return;
 216         }
 217         wpp_messages_capacity = MESSAGEBUFFER_INITIAL_SIZE;
 218     }
 219
 220     while(1)
 221     {
 222         rc = vsnprintf(wpp_messages + wpp_messages_size,
 223                        wpp_messages_capacity - wpp_messages_size, fmt, args);
 224
 225         if (rc < 0 ||                                           /* C89 */
 226             rc >= wpp_messages_capacity - wpp_messages_size) {  /* C99 */
 227             /* Resize the buffer */
 228             newsize = wpp_messages_capacity * 2;
 229             newbuffer = HeapReAlloc(GetProcessHeap(), 0, wpp_messages, newsize);
 230             if(newbuffer == NULL)
 231             {
 232                 ERR("Error reallocating memory for parser messages\n");
 233                 return;
 234             }
 235             wpp_messages = newbuffer;
 236             wpp_messages_capacity = newsize;
 237         }
 238         else
 239         {
 240             wpp_messages_size += rc;
 241             return;
 242         }
 243     }
 244 }
 245
 246 static void PRINTF_ATTR(1,2) wpp_write_message_var(const char *fmt, ...)
 247 {
 248     va_list args;
 249
 250     va_start(args, fmt);
 251     wpp_write_message(fmt, args);
 252     va_end(args);
 253 }
 254
 255 static void wpp_error(const char *file, int line, int col, const char *near,
 256                       const char *msg, va_list ap)
 257 {
 258     wpp_write_message_var("%s:%d:%d: %s: ", file ? file : "'main file'",
 259                           line, col, "Error");
 260     wpp_write_message(msg, ap);
 261     wpp_write_message_var("\n");
 262 }
 263
 264 static void wpp_warning(const char *file, int line, int col, const char *near,
 265                         const char *msg, va_list ap)
 266 {
 267     wpp_write_message_var("%s:%d:%d: %s: ", file ? file : "'main file'",
 268                           line, col, "Warning");
 269     wpp_write_message(msg, ap);
 270     wpp_write_message_var("\n");
 271 }
 272
 273 static char *wpp_lookup_mem(const char *filename, const char *parent_name,
 274                             char **include_path, int include_path_count)
 275 {
 276     /* Here we return always ok. We will maybe fail on the next wpp_open_mem */
 277     char *path;
 278
 279     path = malloc(strlen(filename) + 1);
 280     if(!path) return NULL;
 281     memcpy(path, filename, strlen(filename) + 1);
 282     return path;
 283 }
 284
 285 static void *wpp_open_mem(const char *filename, int type)
 286 {
 287     struct mem_file_desc *desc;
 288     HRESULT hr;
 289
 290     if(filename[0] == '\0') /* "" means to load the initial shader */
 291     {
 292         current_shader.pos = 0;
 293         return &current_shader;
 294     }
 295
 296     if(current_include == NULL) return NULL;
 297     desc = HeapAlloc(GetProcessHeap(), 0, sizeof(*desc));
 298     if(!desc)
 299     {
 300         ERR("Error allocating memory\n");
 301         return NULL;
 302     }
 303     hr = ID3DXInclude_Open(current_include,
 304                            type ? D3DXINC_SYSTEM : D3DXINC_LOCAL,
 305                            filename, NULL, (LPCVOID *)&desc->buffer,
 306                            &desc->size);
 307     if(FAILED(hr))
 308     {
 309         HeapFree(GetProcessHeap(), 0, desc);
 310         return NULL;
 311     }
 312     desc->pos = 0;
 313     return desc;
 314 }
 315
 316 static void wpp_close_mem(void *file)
 317 {
 318     struct mem_file_desc *desc = file;
 319
 320     if(desc != &current_shader)
 321     {
 322         if(current_include)
 323             ID3DXInclude_Close(current_include, desc->buffer);
 324         else
 325             ERR("current_include == NULL, desc == %p, buffer = %s\n",
 326                 desc, desc->buffer);
 327
 328         HeapFree(GetProcessHeap(), 0, desc);
 329     }
 330 }
 331
 332 static int wpp_read_mem(void *file, char *buffer, unsigned int len)
 333 {
 334     struct mem_file_desc *desc = file;
 335
 336     len = min(len, desc->size - desc->pos);
 337     memcpy(buffer, desc->buffer + desc->pos, len);
 338     desc->pos += len;
 339     return len;
 340 }
 341
 342 static void wpp_write_mem(const char *buffer, unsigned int len)
 343 {
 344     char *new_wpp_output;
 345
 346     if(wpp_output_capacity == 0)
 347     {
 348         wpp_output = HeapAlloc(GetProcessHeap(), 0, BUFFER_INITIAL_CAPACITY);
 349         if(!wpp_output)
 350         {
 351             ERR("Error allocating memory\n");
 352             return;
 353         }
 354         wpp_output_capacity = BUFFER_INITIAL_CAPACITY;
 355     }
 356     if(len > wpp_output_capacity - wpp_output_size)
 357     {
 358         while(len > wpp_output_capacity - wpp_output_size)
 359         {
 360             wpp_output_capacity *= 2;
 361         }
 362         new_wpp_output = HeapReAlloc(GetProcessHeap(), 0, wpp_output,
 363                                      wpp_output_capacity);
 364         if(!new_wpp_output)
 365         {
 366             ERR("Error allocating memory\n");
 367             return;
 368         }
 369         wpp_output = new_wpp_output;
 370     }
 371     memcpy(wpp_output + wpp_output_size, buffer, len);
 372     wpp_output_size += len;
 373 }
 374
 375 static int wpp_close_output(void)
 376 {
 377     char *new_wpp_output = HeapReAlloc(GetProcessHeap(), 0, wpp_output,
 378                                        wpp_output_size + 1);
 379     if(!new_wpp_output) return 0;
 380     wpp_output = new_wpp_output;
 381     wpp_output[wpp_output_size]='\0';
 382     return 1;
 383 }
 384
 385 HRESULT assemble_shader(const char *preprocShader, const char *preprocMessages,
 386                         LPD3DXBUFFER* ppShader, LPD3DXBUFFER* ppErrorMsgs)
 387 {
 388     struct bwriter_shader *shader;
 389     char *messages = NULL;
 390     HRESULT hr;
 391     DWORD *res;
 392     LPD3DXBUFFER buffer;
 393     int size;
 394     char *pos;
 395
 396     shader = SlAssembleShader(preprocShader, &messages);
 397
 398     if(messages || preprocMessages)
 399     {
 400         if(preprocMessages)
 401         {
 402             TRACE("Preprocessor messages:\n");
 403             TRACE("%s", preprocMessages);
 404         }
 405         if(messages)
 406         {
 407             TRACE("Assembler messages:\n");
 408             TRACE("%s", messages);
 409         }
 410
 411         TRACE("Shader source:\n");
 412         TRACE("%s\n", debugstr_a(preprocShader));
 413
 414         if(ppErrorMsgs)
 415         {
 416             size = (messages ? strlen(messages) : 0) +
 417                 (preprocMessages ? strlen(preprocMessages) : 0) + 1;
 418             hr = D3DXCreateBuffer(size, &buffer);
 419             if(FAILED(hr))
 420             {
 421                 HeapFree(GetProcessHeap(), 0, messages);
 422                 if(shader) SlDeleteShader(shader);
 423                 return hr;
 424             }
 425             pos = ID3DXBuffer_GetBufferPointer(buffer);
 426             if(preprocMessages)
 427             {
 428                 CopyMemory(pos, preprocMessages, strlen(preprocMessages) + 1);
 429                 pos += strlen(preprocMessages);
 430             }
 431             if(messages)
 432                 CopyMemory(pos, messages, strlen(messages) + 1);
 433
 434             *ppErrorMsgs = buffer;
 435         }
 436
 437         HeapFree(GetProcessHeap(), 0, messages);
 438     }
 439
 440     if(shader == NULL)
 441     {
 442         ERR("Asm reading failed\n");
 443         return D3DXERR_INVALIDDATA;
 444     }
 445
 446     hr = SlWriteBytecode(shader, 9, &res);
 447     SlDeleteShader(shader);
 448     if(FAILED(hr))
 449     {
 450         ERR("SlWriteBytecode failed with 0x%08x\n", hr);
 451         return D3DXERR_INVALIDDATA;
 452     }
 453
 454     if(ppShader)
 455     {
 456         size = HeapSize(GetProcessHeap(), 0, res);
 457         hr = D3DXCreateBuffer(size, &buffer);
 458         if(FAILED(hr))
 459         {
 460             HeapFree(GetProcessHeap(), 0, res);
 461             return hr;
 462         }
 463         CopyMemory(ID3DXBuffer_GetBufferPointer(buffer), res, size);
 464         *ppShader = buffer;
 465     }
 466
 467     HeapFree(GetProcessHeap(), 0, res);
 468
 469     return D3D_OK;
 470 }
 471
 472 HRESULT WINAPI D3DXAssembleShader(LPCSTR data,
 473                                   UINT data_len,
 474                                   CONST D3DXMACRO* defines,
 475                                   LPD3DXINCLUDE include,
 476                                   DWORD flags,
 477                                   LPD3DXBUFFER* shader,
 478                                   LPD3DXBUFFER* error_messages)
 479 {
 480     int ret;
 481     HRESULT hr;
 482     CONST D3DXMACRO* def = defines;
 483
 484     static const struct wpp_callbacks wpp_callbacks = {
 485         wpp_lookup_mem,
 486         wpp_open_mem,
 487         wpp_close_mem,
 488         wpp_read_mem,
 489         wpp_write_mem,
 490         wpp_error,
 491         wpp_warning,
 492     };
 493
 494     EnterCriticalSection(&wpp_mutex);
 495
 496     /* TODO: flags */
 497     if(flags) FIXME("flags: %x\n", flags);
 498
 499     if(def != NULL)
 500     {
 501         while(def->Name != NULL)
 502         {
 503             wpp_add_define(def->Name, def->Definition);
 504             def++;
 505         }
 506     }
 507     current_include = include;
 508
 509     if(shader) *shader = NULL;
 510     if(error_messages) *error_messages = NULL;
 511     wpp_output_size = wpp_output_capacity = 0;
 512     wpp_output = NULL;
 513
 514     /* Preprocess shader */
 515     wpp_set_callbacks(&wpp_callbacks);
 516     wpp_messages_size = wpp_messages_capacity = 0;
 517     wpp_messages = NULL;
 518     current_shader.buffer = data;
 519     current_shader.size = data_len;
 520
 521     ret = wpp_parse("", NULL);
 522     if(!wpp_close_output())
 523         ret = 1;
 524     if(ret)
 525     {
 526         TRACE("Error during shader preprocessing\n");
 527         if(wpp_messages)
 528         {
 529             int size;
 530             LPD3DXBUFFER buffer;
 531
 532             TRACE("Preprocessor messages:\n");
 533             TRACE("%s", wpp_messages);
 534
 535             if(error_messages)
 536             {
 537                 size = strlen(wpp_messages) + 1;
 538                 hr = D3DXCreateBuffer(size, &buffer);
 539                 if(FAILED(hr)) goto cleanup;
 540                 CopyMemory(ID3DXBuffer_GetBufferPointer(buffer), wpp_messages, size);
 541                 *error_messages = buffer;
 542             }
 543         }
 544         if(data)
 545         {
 546             TRACE("Shader source:\n");
 547             TRACE("%s\n", debugstr_an(data, data_len));
 548         }
 549         hr = D3DXERR_INVALIDDATA;
 550         goto cleanup;
 551     }
 552
 553     hr = assemble_shader(wpp_output, wpp_messages, shader, error_messages);
 554
 555 cleanup:
 556     /* Remove the previously added defines */
 557     if(defines != NULL)
 558     {
 559         while(defines->Name != NULL)
 560         {
 561             wpp_del_define(defines->Name);
 562             defines++;
 563         }
 564     }
 565     HeapFree(GetProcessHeap(), 0, wpp_messages);
 566     HeapFree(GetProcessHeap(), 0, wpp_output);
 567     LeaveCriticalSection(&wpp_mutex);
 568     return hr;
 569 }
 570
 571 HRESULT WINAPI D3DXAssembleShaderFromFileA(LPCSTR filename,
 572                                            CONST D3DXMACRO* defines,
 573                                            LPD3DXINCLUDE include,
 574                                            DWORD flags,
 575                                            LPD3DXBUFFER* shader,
 576                                            LPD3DXBUFFER* error_messages)
 577 {
 578     LPWSTR filename_w = NULL;
 579     DWORD len;
 580     HRESULT ret;
 581
 582     if (!filename) return D3DXERR_INVALIDDATA;
 583
 584     len = MultiByteToWideChar(CP_ACP, 0, filename, -1, NULL, 0);
 585     filename_w = HeapAlloc(GetProcessHeap(), 0, len * sizeof(WCHAR));
 586     if (!filename_w) return E_OUTOFMEMORY;
 587     MultiByteToWideChar(CP_ACP, 0, filename, -1, filename_w, len);
 588
 589     ret = D3DXAssembleShaderFromFileW(filename_w, defines, include, flags, shader, error_messages);
 590
 591     HeapFree(GetProcessHeap(), 0, filename_w);
 592     return ret;
 593 }
 594
 595 HRESULT WINAPI D3DXAssembleShaderFromFileW(LPCWSTR filename,
 596                                            CONST D3DXMACRO* defines,
 597                                            LPD3DXINCLUDE include,
 598                                            DWORD flags,
 599                                            LPD3DXBUFFER* shader,
 600                                            LPD3DXBUFFER* error_messages)
 601 {
 602     FIXME("(%s, %p, %p, %x, %p, %p): stub\n", debugstr_w(filename), defines, include, flags, shader, error_messages);
 603     return D3DERR_INVALIDCALL;
 604 }
 605
 606 HRESULT WINAPI D3DXAssembleShaderFromResourceA(HMODULE module,
 607                                                LPCSTR resource,
 608                                                CONST D3DXMACRO* defines,
 609                                                LPD3DXINCLUDE include,
 610                                                DWORD flags,
 611                                                LPD3DXBUFFER* shader,
 612                                                LPD3DXBUFFER* error_messages)
 613 {
 614     HRSRC res;
 615     LPCSTR buffer;
 616     DWORD len;
 617
 618     if (!(res = FindResourceA(module, resource, (LPCSTR)RT_RCDATA)))
 619         return D3DXERR_INVALIDDATA;
 620     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 621         return D3DXERR_INVALIDDATA;
 622     return D3DXAssembleShader(buffer, len, defines, include, flags,
 623                               shader, error_messages);
 624 }
 625
 626 HRESULT WINAPI D3DXAssembleShaderFromResourceW(HMODULE module,
 627                                                LPCWSTR resource,
 628                                                CONST D3DXMACRO* defines,
 629                                                LPD3DXINCLUDE include,
 630                                                DWORD flags,
 631                                                LPD3DXBUFFER* shader,
 632                                                LPD3DXBUFFER* error_messages)
 633 {
 634     HRSRC res;
 635     LPCSTR buffer;
 636     DWORD len;
 637
 638     if (!(res = FindResourceW(module, resource, (LPCWSTR)RT_RCDATA)))
 639         return D3DXERR_INVALIDDATA;
 640     if (FAILED(load_resource_into_memory(module, res, (LPVOID *)&buffer, &len)))
 641         return D3DXERR_INVALIDDATA;
 642     return D3DXAssembleShader(buffer, len, defines, include, flags,
 643                               shader, error_messages);
 644 }
 645
 646 HRESULT WINAPI D3DXCompileShader(LPCSTR pSrcData,
 647                                  UINT srcDataLen,
 648                                  CONST D3DXMACRO* pDefines,
 649                                  LPD3DXINCLUDE pInclude,
 650                                  LPCSTR pFunctionName,
 651                                  LPCSTR pProfile,
 652                                  DWORD Flags,
 653                                  LPD3DXBUFFER* ppShader,
 654                                  LPD3DXBUFFER* ppErrorMsgs,
 655                                  LPD3DXCONSTANTTABLE * ppConstantTable)
 656 {
 657     FIXME("(%p, %d, %p, %p, %s, %s, %x, %p, %p, %p): stub\n",
 658           pSrcData, srcDataLen, pDefines, pInclude, debugstr_a(pFunctionName),
 659           debugstr_a(pProfile), Flags, ppShader, ppErrorMsgs, ppConstantTable);
 660
 661     TRACE("Shader source:\n");
 662     TRACE("%s\n", debugstr_an(pSrcData, srcDataLen));
 663
 664     return D3DERR_INVALIDCALL;
 665 }
 666
 667 static const struct ID3DXConstantTableVtbl ID3DXConstantTable_Vtbl;
 668
 669 typedef struct ID3DXConstantTableImpl {
 670     const ID3DXConstantTableVtbl *lpVtbl;
 671     LONG ref;
 672     LPVOID ctab;
 673     DWORD size;
 674     D3DXCONSTANTTABLE_DESC desc;
 675 } ID3DXConstantTableImpl;
 676
 677 /*** IUnknown methods ***/
 678 static HRESULT WINAPI ID3DXConstantTableImpl_QueryInterface(ID3DXConstantTable* iface, REFIID riid, void** ppvObject)
 679 {
 680     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 681
 682     TRACE("(%p)->(%s, %p)\n", This, debugstr_guid(riid), ppvObject);
 683
 684     if (IsEqualGUID(riid, &IID_IUnknown) ||
 685         IsEqualGUID(riid, &IID_ID3DXBuffer) ||
 686         IsEqualGUID(riid, &IID_ID3DXConstantTable))
 687     {
 688         ID3DXConstantTable_AddRef(iface);
 689         *ppvObject = This;
 690         return S_OK;
 691     }
 692
 693     WARN("Interface %s not found.\n", debugstr_guid(riid));
 694
 695     return E_NOINTERFACE;
 696 }
 697
 698 static ULONG WINAPI ID3DXConstantTableImpl_AddRef(ID3DXConstantTable* iface)
 699 {
 700     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 701
 702     TRACE("(%p)->(): AddRef from %d\n", This, This->ref);
 703
 704     return InterlockedIncrement(&This->ref);
 705 }
 706
 707 static ULONG WINAPI ID3DXConstantTableImpl_Release(ID3DXConstantTable* iface)
 708 {
 709     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 710     ULONG ref = InterlockedDecrement(&This->ref);
 711
 712     TRACE("(%p)->(): Release from %d\n", This, ref + 1);
 713
 714     if (!ref)
 715     {
 716         HeapFree(GetProcessHeap(), 0, This->ctab);
 717         HeapFree(GetProcessHeap(), 0, This);
 718     }
 719
 720     return ref;
 721 }
 722
 723 /*** ID3DXBuffer methods ***/
 724 static LPVOID WINAPI ID3DXConstantTableImpl_GetBufferPointer(ID3DXConstantTable* iface)
 725 {
 726     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 727
 728     TRACE("(%p)->()\n", This);
 729
 730     return This->ctab;
 731 }
 732
 733 static DWORD WINAPI ID3DXConstantTableImpl_GetBufferSize(ID3DXConstantTable* iface)
 734 {
 735     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 736
 737     TRACE("(%p)->()\n", This);
 738
 739     return This->size;
 740 }
 741
 742 /*** ID3DXConstantTable methods ***/
 743 static HRESULT WINAPI ID3DXConstantTableImpl_GetDesc(ID3DXConstantTable* iface, D3DXCONSTANTTABLE_DESC *desc)
 744 {
 745     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 746
 747     TRACE("(%p)->(%p)\n", This, desc);
 748
 749     if (!desc)
 750         return D3DERR_INVALIDCALL;
 751
 752     memcpy(desc, &This->desc, sizeof(This->desc));
 753
 754     return D3D_OK;
 755 }
 756
 757 static HRESULT WINAPI ID3DXConstantTableImpl_GetConstantDesc(ID3DXConstantTable* iface, D3DXHANDLE constant,
 758                                                              D3DXCONSTANT_DESC *desc, UINT *count)
 759 {
 760     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 761
 762     FIXME("(%p)->(%p, %p, %p): stub\n", This, constant, desc, count);
 763
 764     return E_NOTIMPL;
 765 }
 766
 767 static UINT WINAPI ID3DXConstantTableImpl_GetSamplerIndex(LPD3DXCONSTANTTABLE iface, D3DXHANDLE constant)
 768 {
 769     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 770
 771     FIXME("(%p)->(%p): stub\n", This, constant);
 772
 773     return (UINT)-1;
 774 }
 775
 776 static D3DXHANDLE WINAPI ID3DXConstantTableImpl_GetConstant(ID3DXConstantTable* iface, D3DXHANDLE constant, UINT index)
 777 {
 778     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 779
 780     FIXME("(%p)->(%p, %d): stub\n", This, constant, index);
 781
 782     return NULL;
 783 }
 784
 785 static D3DXHANDLE WINAPI ID3DXConstantTableImpl_GetConstantByName(ID3DXConstantTable* iface, D3DXHANDLE constant, LPCSTR name)
 786 {
 787     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 788
 789     FIXME("(%p)->(%p, %s): stub\n", This, constant, name);
 790
 791     return NULL;
 792 }
 793
 794 static D3DXHANDLE WINAPI ID3DXConstantTableImpl_GetConstantElement(ID3DXConstantTable* iface, D3DXHANDLE constant, UINT index)
 795 {
 796     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 797
 798     FIXME("(%p)->(%p, %d): stub\n", This, constant, index);
 799
 800     return NULL;
 801 }
 802
 803 static HRESULT WINAPI ID3DXConstantTableImpl_SetDefaults(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device)
 804 {
 805     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 806
 807     FIXME("(%p)->(%p): stub\n", This, device);
 808
 809     return E_NOTIMPL;
 810 }
 811
 812 static HRESULT WINAPI ID3DXConstantTableImpl_SetValue(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 813                                                       D3DXHANDLE constant, LPCVOID data, UINT bytes)
 814 {
 815     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 816
 817     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, data, bytes);
 818
 819     return E_NOTIMPL;
 820 }
 821
 822 static HRESULT WINAPI ID3DXConstantTableImpl_SetBool(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 823                                                      D3DXHANDLE constant, BOOL b)
 824 {
 825     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 826
 827     FIXME("(%p)->(%p, %p, %d): stub\n", This, device, constant, b);
 828
 829     return E_NOTIMPL;
 830 }
 831
 832 static HRESULT WINAPI ID3DXConstantTableImpl_SetBoolArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 833                                                           D3DXHANDLE constant, CONST BOOL* b, UINT count)
 834 {
 835     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 836
 837     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, b, count);
 838
 839     return E_NOTIMPL;
 840 }
 841
 842 static HRESULT WINAPI ID3DXConstantTableImpl_SetInt(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device, D3DXHANDLE constant, INT n)
 843 {
 844     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 845
 846     FIXME("(%p)->(%p, %p, %d): stub\n", This, device, constant, n);
 847
 848     return E_NOTIMPL;
 849 }
 850
 851 static HRESULT WINAPI ID3DXConstantTableImpl_SetIntArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 852                                                          D3DXHANDLE constant, CONST INT* n, UINT count)
 853 {
 854     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 855
 856     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, n, count);
 857
 858     return E_NOTIMPL;
 859 }
 860
 861 static HRESULT WINAPI ID3DXConstantTableImpl_SetFloat(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 862                                                       D3DXHANDLE constant, FLOAT f)
 863 {
 864     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 865
 866     FIXME("(%p)->(%p, %p, %f): stub\n", This, device, constant, f);
 867
 868     return E_NOTIMPL;
 869 }
 870
 871 static HRESULT WINAPI ID3DXConstantTableImpl_SetFloatArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 872                                                            D3DXHANDLE constant, CONST FLOAT* f, UINT count)
 873 {
 874     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 875
 876     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, f, count);
 877
 878     return E_NOTIMPL;
 879 }
 880
 881 static HRESULT WINAPI ID3DXConstantTableImpl_SetVector(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 882                                                        D3DXHANDLE constant, CONST D3DXVECTOR4* vector)
 883 {
 884     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 885
 886     FIXME("(%p)->(%p, %p, %p): stub\n", This, device, constant, vector);
 887
 888     return E_NOTIMPL;
 889 }
 890
 891 static HRESULT WINAPI ID3DXConstantTableImpl_SetVectorArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 892                                                             D3DXHANDLE constant, CONST D3DXVECTOR4* vector, UINT count)
 893 {
 894     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 895
 896     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, vector, count);
 897
 898     return E_NOTIMPL;
 899 }
 900
 901 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrix(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 902                                                        D3DXHANDLE constant, CONST D3DXMATRIX* matrix)
 903 {
 904     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 905
 906     FIXME("(%p)->(%p, %p, %p): stub\n", This, device, constant, matrix);
 907
 908     return E_NOTIMPL;
 909 }
 910
 911 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 912                                                             D3DXHANDLE constant, CONST D3DXMATRIX* matrix, UINT count)
 913 {
 914     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 915
 916     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, matrix, count);
 917
 918     return E_NOTIMPL;
 919 }
 920
 921 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixPointerArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 922                                                                    D3DXHANDLE constant, CONST D3DXMATRIX** matrix, UINT count)
 923 {
 924     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 925
 926     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, matrix, count);
 927
 928     return E_NOTIMPL;
 929 }
 930
 931 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixTranspose(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 932                                                                 D3DXHANDLE constant, CONST D3DXMATRIX* matrix)
 933 {
 934     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 935
 936     FIXME("(%p)->(%p, %p, %p): stub\n", This, device, constant, matrix);
 937
 938     return E_NOTIMPL;
 939 }
 940
 941 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixTransposeArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 942                                                                      D3DXHANDLE constant, CONST D3DXMATRIX* matrix, UINT count)
 943 {
 944     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 945
 946     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, matrix, count);
 947
 948     return E_NOTIMPL;
 949 }
 950
 951 static HRESULT WINAPI ID3DXConstantTableImpl_SetMatrixTransposePointerArray(ID3DXConstantTable* iface, LPDIRECT3DDEVICE9 device,
 952                                                                             D3DXHANDLE constant, CONST D3DXMATRIX** matrix, UINT count)
 953 {
 954     ID3DXConstantTableImpl *This = (ID3DXConstantTableImpl *)iface;
 955
 956     FIXME("(%p)->(%p, %p, %p, %d): stub\n", This, device, constant, matrix, count);
 957
 958     return E_NOTIMPL;
 959 }
 960
 961 static const struct ID3DXConstantTableVtbl ID3DXConstantTable_Vtbl =
 962 {
 963     /*** IUnknown methods ***/
 964     ID3DXConstantTableImpl_QueryInterface,
 965     ID3DXConstantTableImpl_AddRef,
 966     ID3DXConstantTableImpl_Release,
 967     /*** ID3DXBuffer methods ***/
 968     ID3DXConstantTableImpl_GetBufferPointer,
 969     ID3DXConstantTableImpl_GetBufferSize,
 970     /*** ID3DXConstantTable methods ***/
 971     ID3DXConstantTableImpl_GetDesc,
 972     ID3DXConstantTableImpl_GetConstantDesc,
 973     ID3DXConstantTableImpl_GetSamplerIndex,
 974     ID3DXConstantTableImpl_GetConstant,
 975     ID3DXConstantTableImpl_GetConstantByName,
 976     ID3DXConstantTableImpl_GetConstantElement,
 977     ID3DXConstantTableImpl_SetDefaults,
 978     ID3DXConstantTableImpl_SetValue,
 979     ID3DXConstantTableImpl_SetBool,
 980     ID3DXConstantTableImpl_SetBoolArray,
 981     ID3DXConstantTableImpl_SetInt,
 982     ID3DXConstantTableImpl_SetIntArray,
 983     ID3DXConstantTableImpl_SetFloat,
 984     ID3DXConstantTableImpl_SetFloatArray,
 985     ID3DXConstantTableImpl_SetVector,
 986     ID3DXConstantTableImpl_SetVectorArray,
 987     ID3DXConstantTableImpl_SetMatrix,
 988     ID3DXConstantTableImpl_SetMatrixArray,
 989     ID3DXConstantTableImpl_SetMatrixPointerArray,
 990     ID3DXConstantTableImpl_SetMatrixTranspose,
 991     ID3DXConstantTableImpl_SetMatrixTransposeArray,
 992     ID3DXConstantTableImpl_SetMatrixTransposePointerArray
 993 };
 994
 995 HRESULT WINAPI D3DXGetShaderConstantTableEx(CONST DWORD* byte_code,
 996                                             DWORD flags,
 997                                             LPD3DXCONSTANTTABLE* constant_table)
 998 {
 999     ID3DXConstantTableImpl* object = NULL;
1000     HRESULT hr;
1001     LPCVOID data;
1002     UINT size;
1003     const D3DXSHADER_CONSTANTTABLE* ctab_header;
1004
1005     FIXME("(%p, %x, %p): semi-stub\n", byte_code, flags, constant_table);
1006
1007     if (!byte_code || !constant_table)
1008         return D3DERR_INVALIDCALL;
1009
1010     hr = D3DXFindShaderComment(byte_code, MAKEFOURCC('C','T','A','B'), &data, &size);
1011     if (hr != D3D_OK)
1012         return D3DXERR_INVALIDDATA;
1013
1014     object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(ID3DXConstantTableImpl));
1015     if (!object)
1016     {
1017         ERR("Out of memory\n");
1018         return E_OUTOFMEMORY;
1019     }
1020
1021     object->lpVtbl = &ID3DXConstantTable_Vtbl;
1022     object->ref = 1;
1023
1024     if (size < sizeof(D3DXSHADER_CONSTANTTABLE))
1025         goto error;
1026
1027     object->ctab = HeapAlloc(GetProcessHeap(), 0, size);
1028     if (!object->ctab)
1029     {
1030         HeapFree(GetProcessHeap(), 0, object);
1031         ERR("Out of memory\n");
1032         return E_OUTOFMEMORY;
1033     }
1034     object->size = size;
1035     memcpy(object->ctab, data, object->size);
1036
1037     ctab_header = (const D3DXSHADER_CONSTANTTABLE*)data;
1038     if (ctab_header->Size != sizeof(D3DXSHADER_CONSTANTTABLE))
1039         goto error;
1040     object->desc.Creator = ctab_header->Creator ? (LPCSTR)object->ctab + ctab_header->Creator : NULL;
1041     object->desc.Version = ctab_header->Version;
1042     object->desc.Constants = ctab_header->Constants;
1043
1044     *constant_table = (LPD3DXCONSTANTTABLE)object;
1045
1046     return D3D_OK;
1047
1048 error:
1049
1050     HeapFree(GetProcessHeap(), 0, object->ctab);
1051     HeapFree(GetProcessHeap(), 0, object);
1052
1053     return D3DXERR_INVALIDDATA;
1054 }
1055
1056 HRESULT WINAPI D3DXGetShaderConstantTable(CONST DWORD* byte_code,
1057                                           LPD3DXCONSTANTTABLE* constant_table)
1058 {
1059     TRACE("(%p, %p): Forwarded to D3DXGetShaderConstantTableEx\n", byte_code, constant_table);
1060
1061     return D3DXGetShaderConstantTableEx(byte_code, 0, constant_table);
1062 }