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