reactos/dll/directx/wine/d3d8/vertexdeclaration.c

   1 /*
   2  * IDirect3DVertexDeclaration8 implementation
   3  *
   4  * Copyright 2007 Henri Verbeet
   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 /* IDirect3DVertexDeclaration8 is internal to our implementation.
  22  * It's not visible in the API. */
  23
  24 #include "config.h"
  25 #include "d3d8_private.h"
  26
  27 WINE_DEFAULT_DEBUG_CHANNEL(d3d8);
  28
  29 /* IUnknown */
  30 static HRESULT WINAPI IDirect3DVertexDeclaration8Impl_QueryInterface(IDirect3DVertexDeclaration8 *iface, REFIID riid, void **obj_ptr)
  31 {
  32     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), obj_ptr);
  33
  34     if (IsEqualGUID(riid, &IID_IUnknown)
  35             || IsEqualGUID(riid, &IID_IDirect3DVertexDeclaration8))
  36     {
  37         IUnknown_AddRef(iface);
  38         *obj_ptr = iface;
  39         return S_OK;
  40     }
  41
  42     *obj_ptr = NULL;
  43     return E_NOINTERFACE;
  44 }
  45
  46 static ULONG WINAPI IDirect3DVertexDeclaration8Impl_AddRef(IDirect3DVertexDeclaration8 *iface)
  47 {
  48     IDirect3DVertexDeclaration8Impl *This = (IDirect3DVertexDeclaration8Impl *)iface;
  49
  50     ULONG ref_count = InterlockedIncrement(&This->ref_count);
  51     TRACE("(%p) : AddRef increasing to %d\n", This, ref_count);
  52
  53     return ref_count;
  54 }
  55
  56 static ULONG WINAPI IDirect3DVertexDeclaration8Impl_Release(IDirect3DVertexDeclaration8 *iface)
  57 {
  58     IDirect3DVertexDeclaration8Impl *This = (IDirect3DVertexDeclaration8Impl *)iface;
  59
  60     ULONG ref_count = InterlockedDecrement(&This->ref_count);
  61     TRACE("(%p) : Releasing to %d\n", This, ref_count);
  62
  63     if (!ref_count) {
  64         EnterCriticalSection(&d3d8_cs);
  65         IWineD3DVertexDeclaration_Release(This->wined3d_vertex_declaration);
  66         LeaveCriticalSection(&d3d8_cs);
  67         HeapFree(GetProcessHeap(), 0, This->elements);
  68         HeapFree(GetProcessHeap(), 0, This);
  69     }
  70
  71     return ref_count;
  72 }
  73
  74 static const char *debug_d3dvsdt_type(D3DVSDT_TYPE d3dvsdt_type)
  75 {
  76     switch (d3dvsdt_type)
  77     {
  78 #define D3DVSDT_TYPE_TO_STR(u) case u: return #u
  79         D3DVSDT_TYPE_TO_STR(D3DVSDT_FLOAT1);
  80         D3DVSDT_TYPE_TO_STR(D3DVSDT_FLOAT2);
  81         D3DVSDT_TYPE_TO_STR(D3DVSDT_FLOAT3);
  82         D3DVSDT_TYPE_TO_STR(D3DVSDT_FLOAT4);
  83         D3DVSDT_TYPE_TO_STR(D3DVSDT_D3DCOLOR);
  84         D3DVSDT_TYPE_TO_STR(D3DVSDT_UBYTE4);
  85         D3DVSDT_TYPE_TO_STR(D3DVSDT_SHORT2);
  86         D3DVSDT_TYPE_TO_STR(D3DVSDT_SHORT4);
  87 #undef D3DVSDT_TYPE_TO_STR
  88         default:
  89             FIXME("Unrecognized D3DVSDT_TYPE %#x\n", d3dvsdt_type);
  90             return "unrecognized";
  91     }
  92 }
  93
  94 static const char *debug_d3dvsde_register(D3DVSDE_REGISTER d3dvsde_register)
  95 {
  96     switch (d3dvsde_register)
  97     {
  98 #define D3DVSDE_REGISTER_TO_STR(u) case u: return #u
  99         D3DVSDE_REGISTER_TO_STR(D3DVSDE_POSITION);
 100         D3DVSDE_REGISTER_TO_STR(D3DVSDE_BLENDWEIGHT);
 101         D3DVSDE_REGISTER_TO_STR(D3DVSDE_BLENDINDICES);
 102         D3DVSDE_REGISTER_TO_STR(D3DVSDE_NORMAL);
 103         D3DVSDE_REGISTER_TO_STR(D3DVSDE_PSIZE);
 104         D3DVSDE_REGISTER_TO_STR(D3DVSDE_DIFFUSE);
 105         D3DVSDE_REGISTER_TO_STR(D3DVSDE_SPECULAR);
 106         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD0);
 107         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD1);
 108         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD2);
 109         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD3);
 110         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD4);
 111         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD5);
 112         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD6);
 113         D3DVSDE_REGISTER_TO_STR(D3DVSDE_TEXCOORD7);
 114         D3DVSDE_REGISTER_TO_STR(D3DVSDE_POSITION2);
 115         D3DVSDE_REGISTER_TO_STR(D3DVSDE_NORMAL2);
 116 #undef D3DVSDE_REGISTER_TO_STR
 117         default:
 118             FIXME("Unrecognized D3DVSDE_REGISTER %#x\n", d3dvsde_register);
 119             return "unrecognized";
 120     }
 121 }
 122
 123 size_t parse_token(const DWORD* pToken)
 124 {
 125     const DWORD token = *pToken;
 126     size_t tokenlen = 1;
 127
 128     switch ((token & D3DVSD_TOKENTYPEMASK) >> D3DVSD_TOKENTYPESHIFT) { /* maybe a macro to inverse ... */
 129         case D3DVSD_TOKEN_NOP:
 130             TRACE(" 0x%08x NOP()\n", token);
 131             break;
 132
 133         case D3DVSD_TOKEN_STREAM:
 134             if (token & D3DVSD_STREAMTESSMASK)
 135             {
 136                 TRACE(" 0x%08x STREAM_TESS()\n", token);
 137             } else {
 138                 TRACE(" 0x%08x STREAM(%u)\n", token, ((token & D3DVSD_STREAMNUMBERMASK) >> D3DVSD_STREAMNUMBERSHIFT));
 139             }
 140             break;
 141
 142         case D3DVSD_TOKEN_STREAMDATA:
 143             if (token & 0x10000000)
 144             {
 145                 TRACE(" 0x%08x SKIP(%u)\n", token, ((token & D3DVSD_SKIPCOUNTMASK) >> D3DVSD_SKIPCOUNTSHIFT));
 146             } else {
 147                 DWORD type = ((token & D3DVSD_DATATYPEMASK)  >> D3DVSD_DATATYPESHIFT);
 148                 DWORD reg = ((token & D3DVSD_VERTEXREGMASK) >> D3DVSD_VERTEXREGSHIFT);
 149                 TRACE(" 0x%08x REG(%s, %s)\n", token, debug_d3dvsde_register(reg), debug_d3dvsdt_type(type));
 150             }
 151             break;
 152
 153         case D3DVSD_TOKEN_TESSELLATOR:
 154             if (token & 0x10000000)
 155             {
 156                 DWORD type = ((token & D3DVSD_DATATYPEMASK)  >> D3DVSD_DATATYPESHIFT);
 157                 DWORD reg = ((token & D3DVSD_VERTEXREGMASK) >> D3DVSD_VERTEXREGSHIFT);
 158                 TRACE(" 0x%08x TESSUV(%s) as %s\n", token, debug_d3dvsde_register(reg), debug_d3dvsdt_type(type));
 159             } else {
 160                 DWORD type = ((token & D3DVSD_DATATYPEMASK)    >> D3DVSD_DATATYPESHIFT);
 161                 DWORD regout = ((token & D3DVSD_VERTEXREGMASK)   >> D3DVSD_VERTEXREGSHIFT);
 162                 DWORD regin = ((token & D3DVSD_VERTEXREGINMASK) >> D3DVSD_VERTEXREGINSHIFT);
 163                 TRACE(" 0x%08x TESSNORMAL(%s, %s) as %s\n", token, debug_d3dvsde_register(regin),
 164                         debug_d3dvsde_register(regout), debug_d3dvsdt_type(type));
 165             }
 166             break;
 167
 168         case D3DVSD_TOKEN_CONSTMEM:
 169             {
 170                 DWORD count = ((token & D3DVSD_CONSTCOUNTMASK)   >> D3DVSD_CONSTCOUNTSHIFT);
 171                 tokenlen = (4 * count) + 1;
 172             }
 173             break;
 174
 175         case D3DVSD_TOKEN_EXT:
 176             {
 177                 DWORD count = ((token & D3DVSD_CONSTCOUNTMASK) >> D3DVSD_CONSTCOUNTSHIFT);
 178                 DWORD extinfo = ((token & D3DVSD_EXTINFOMASK)    >> D3DVSD_EXTINFOSHIFT);
 179                 TRACE(" 0x%08x EXT(%u, %u)\n", token, count, extinfo);
 180                 /* todo ... print extension */
 181                 tokenlen = count + 1;
 182             }
 183             break;
 184
 185         case D3DVSD_TOKEN_END:
 186             TRACE(" 0x%08x END()\n", token);
 187             break;
 188
 189         default:
 190             TRACE(" 0x%08x UNKNOWN\n", token);
 191             /* argg error */
 192     }
 193
 194     return tokenlen;
 195 }
 196
 197 void load_local_constants(const DWORD *d3d8_elements, IWineD3DVertexShader *wined3d_vertex_shader)
 198 {
 199     const DWORD *token = d3d8_elements;
 200
 201     while (*token != D3DVSD_END())
 202     {
 203         if (((*token & D3DVSD_TOKENTYPEMASK) >> D3DVSD_TOKENTYPESHIFT) == D3DVSD_TOKEN_CONSTMEM)
 204         {
 205             DWORD count = ((*token & D3DVSD_CONSTCOUNTMASK) >> D3DVSD_CONSTCOUNTSHIFT);
 206             DWORD constant_idx = ((*token & D3DVSD_CONSTADDRESSMASK) >> D3DVSD_CONSTADDRESSSHIFT);
 207             HRESULT hr;
 208
 209             if (TRACE_ON(d3d8))
 210             {
 211                 DWORD i;
 212                 for (i = 0; i < count; ++i)
 213                 {
 214                     TRACE("c[%u] = (%8f, %8f, %8f, %8f)\n",
 215                             constant_idx,
 216                             *(const float *)(token + i * 4 + 1),
 217                             *(const float *)(token + i * 4 + 2),
 218                             *(const float *)(token + i * 4 + 3),
 219                             *(const float *)(token + i * 4 + 4));
 220                 }
 221             }
 222             hr = IWineD3DVertexShader_SetLocalConstantsF(wined3d_vertex_shader, constant_idx, (const float *)token+1, count);
 223             if (FAILED(hr)) ERR("Failed setting shader constants\n");
 224         }
 225
 226         token += parse_token(token);
 227     }
 228 }
 229
 230 /* NOTE: Make sure these are in the correct numerical order. (see /include/wined3d_types.h) */
 231 static const size_t wined3d_type_sizes[] =
 232 {
 233     /*WINED3DDECLTYPE_FLOAT1*/    1 * sizeof(float),
 234     /*WINED3DDECLTYPE_FLOAT2*/    2 * sizeof(float),
 235     /*WINED3DDECLTYPE_FLOAT3*/    3 * sizeof(float),
 236     /*WINED3DDECLTYPE_FLOAT4*/    4 * sizeof(float),
 237     /*WINED3DDECLTYPE_D3DCOLOR*/  4 * sizeof(BYTE),
 238     /*WINED3DDECLTYPE_UBYTE4*/    4 * sizeof(BYTE),
 239     /*WINED3DDECLTYPE_SHORT2*/    2 * sizeof(short int),
 240     /*WINED3DDECLTYPE_SHORT4*/    4 * sizeof(short int),
 241     /*WINED3DDECLTYPE_UBYTE4N*/   4 * sizeof(BYTE),
 242     /*WINED3DDECLTYPE_SHORT2N*/   2 * sizeof(short int),
 243     /*WINED3DDECLTYPE_SHORT4N*/   4 * sizeof(short int),
 244     /*WINED3DDECLTYPE_USHORT2N*/  2 * sizeof(short int),
 245     /*WINED3DDECLTYPE_USHORT4N*/  4 * sizeof(short int),
 246     /*WINED3DDECLTYPE_UDEC3*/     3 * sizeof(short int),
 247     /*WINED3DDECLTYPE_DEC3N*/     3 * sizeof(short int),
 248     /*WINED3DDECLTYPE_FLOAT16_2*/ 2 * sizeof(short int),
 249     /*WINED3DDECLTYPE_FLOAT16_4*/ 4 * sizeof(short int)
 250 };
 251
 252 static const WINED3DFORMAT wined3d_format_lookup[] =
 253 {
 254     /*WINED3DDECLTYPE_FLOAT1*/    WINED3DFMT_R32_FLOAT,
 255     /*WINED3DDECLTYPE_FLOAT2*/    WINED3DFMT_R32G32_FLOAT,
 256     /*WINED3DDECLTYPE_FLOAT3*/    WINED3DFMT_R32G32B32_FLOAT,
 257     /*WINED3DDECLTYPE_FLOAT4*/    WINED3DFMT_R32G32B32A32_FLOAT,
 258     /*WINED3DDECLTYPE_D3DCOLOR*/  WINED3DFMT_A8R8G8B8,
 259     /*WINED3DDECLTYPE_UBYTE4*/    WINED3DFMT_R8G8B8A8_UINT,
 260     /*WINED3DDECLTYPE_SHORT2*/    WINED3DFMT_R16G16_SINT,
 261     /*WINED3DDECLTYPE_SHORT4*/    WINED3DFMT_R16G16B16A16_SINT,
 262     /*WINED3DDECLTYPE_UBYTE4N*/   WINED3DFMT_R8G8B8A8_UNORM,
 263     /*WINED3DDECLTYPE_SHORT2N*/   WINED3DFMT_R16G16_SNORM,
 264     /*WINED3DDECLTYPE_SHORT4N*/   WINED3DFMT_R16G16B16A16_SNORM,
 265     /*WINED3DDECLTYPE_USHORT2N*/  WINED3DFMT_R16G16_UNORM,
 266     /*WINED3DDECLTYPE_USHORT4N*/  WINED3DFMT_R16G16B16A16_UNORM,
 267     /*WINED3DDECLTYPE_UDEC3*/     WINED3DFMT_R10G10B10A2_UINT,
 268     /*WINED3DDECLTYPE_DEC3N*/     WINED3DFMT_R10G10B10A2_SNORM,
 269     /*WINED3DDECLTYPE_FLOAT16_2*/ WINED3DFMT_R16G16_FLOAT,
 270     /*WINED3DDECLTYPE_FLOAT16_4*/ WINED3DFMT_R16G16B16A16_FLOAT,
 271 };
 272
 273 typedef struct {
 274     BYTE usage;
 275     BYTE usage_idx;
 276 } wined3d_usage_t;
 277
 278 static const wined3d_usage_t wined3d_usage_lookup[] = {
 279     /*D3DVSDE_POSITION*/     {WINED3DDECLUSAGE_POSITION,     0},
 280     /*D3DVSDE_BLENDWEIGHT*/  {WINED3DDECLUSAGE_BLENDWEIGHT,  0},
 281     /*D3DVSDE_BLENDINDICES*/ {WINED3DDECLUSAGE_BLENDINDICES, 0},
 282     /*D3DVSDE_NORMAL*/       {WINED3DDECLUSAGE_NORMAL,       0},
 283     /*D3DVSDE_PSIZE*/        {WINED3DDECLUSAGE_PSIZE,        0},
 284     /*D3DVSDE_DIFFUSE*/      {WINED3DDECLUSAGE_COLOR,        0},
 285     /*D3DVSDE_SPECULAR*/     {WINED3DDECLUSAGE_COLOR,        1},
 286     /*D3DVSDE_TEXCOORD0*/    {WINED3DDECLUSAGE_TEXCOORD,     0},
 287     /*D3DVSDE_TEXCOORD1*/    {WINED3DDECLUSAGE_TEXCOORD,     1},
 288     /*D3DVSDE_TEXCOORD2*/    {WINED3DDECLUSAGE_TEXCOORD,     2},
 289     /*D3DVSDE_TEXCOORD3*/    {WINED3DDECLUSAGE_TEXCOORD,     3},
 290     /*D3DVSDE_TEXCOORD4*/    {WINED3DDECLUSAGE_TEXCOORD,     4},
 291     /*D3DVSDE_TEXCOORD5*/    {WINED3DDECLUSAGE_TEXCOORD,     5},
 292     /*D3DVSDE_TEXCOORD6*/    {WINED3DDECLUSAGE_TEXCOORD,     6},
 293     /*D3DVSDE_TEXCOORD7*/    {WINED3DDECLUSAGE_TEXCOORD,     7},
 294     /*D3DVSDE_POSITION2*/    {WINED3DDECLUSAGE_POSITION,     1},
 295     /*D3DVSDE_NORMAL2*/      {WINED3DDECLUSAGE_NORMAL,       1},
 296 };
 297
 298 /* TODO: find out where rhw (or positionT) is for declaration8 */
 299 UINT convert_to_wined3d_declaration(const DWORD *d3d8_elements, DWORD *d3d8_elements_size, WINED3DVERTEXELEMENT **wined3d_elements)
 300 {
 301     const DWORD *token = d3d8_elements;
 302     WINED3DVERTEXELEMENT *element;
 303     D3DVSD_TOKENTYPE token_type;
 304     unsigned int element_count = 0;
 305     DWORD stream = 0;
 306     int offset = 0;
 307
 308     TRACE("d3d8_elements %p, wined3d_elements %p\n", d3d8_elements, wined3d_elements);
 309
 310     /* 128 should be enough for anyone... */
 311     *wined3d_elements = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, 128 * sizeof(WINED3DVERTEXELEMENT));
 312     while (D3DVSD_END() != *token)
 313     {
 314         token_type = ((*token & D3DVSD_TOKENTYPEMASK) >> D3DVSD_TOKENTYPESHIFT);
 315
 316         if (token_type == D3DVSD_TOKEN_STREAM && !(*token & D3DVSD_STREAMTESSMASK))
 317         {
 318             stream = ((*token & D3DVSD_STREAMNUMBERMASK) >> D3DVSD_STREAMNUMBERSHIFT);
 319             offset = 0;
 320         } else if (token_type == D3DVSD_TOKEN_STREAMDATA && !(token_type & 0x10000000)) {
 321             DWORD type = ((*token & D3DVSD_DATATYPEMASK) >> D3DVSD_DATATYPESHIFT);
 322             DWORD reg  = ((*token & D3DVSD_VERTEXREGMASK) >> D3DVSD_VERTEXREGSHIFT);
 323
 324             TRACE("Adding element %d:\n", element_count);
 325
 326             element = *wined3d_elements + element_count++;
 327             element->format = wined3d_format_lookup[type];
 328             element->input_slot = stream;
 329             element->offset = offset;
 330             element->output_slot = reg;
 331             element->method = WINED3DDECLMETHOD_DEFAULT;
 332             element->usage = wined3d_usage_lookup[reg].usage;
 333             element->usage_idx = wined3d_usage_lookup[reg].usage_idx;
 334
 335             offset += wined3d_type_sizes[type];
 336         } else if (token_type == D3DVSD_TOKEN_STREAMDATA && (token_type & 0x10000000)) {
 337             TRACE(" 0x%08x SKIP(%u)\n", token_type, ((token_type & D3DVSD_SKIPCOUNTMASK) >> D3DVSD_SKIPCOUNTSHIFT));
 338             offset += sizeof(DWORD) * ((token_type & D3DVSD_SKIPCOUNTMASK) >> D3DVSD_SKIPCOUNTSHIFT);
 339         }
 340
 341         if (element_count >= 127) {
 342             ERR("More than 127 elements?\n");
 343             break;
 344         }
 345
 346         token += parse_token(token);
 347     }
 348
 349     *d3d8_elements_size = (++token - d3d8_elements) * sizeof(DWORD);
 350
 351     return element_count;
 352 }
 353
 354 const IDirect3DVertexDeclaration8Vtbl Direct3DVertexDeclaration8_Vtbl =
 355 {
 356     IDirect3DVertexDeclaration8Impl_QueryInterface,
 357     IDirect3DVertexDeclaration8Impl_AddRef,
 358     IDirect3DVertexDeclaration8Impl_Release
 359 };