dlls/wined3d/shader_sm4.c

   1 /*
   2  * Copyright 2009 Henri Verbeet for CodeWeavers
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include "config.h"
  20 #include "wine/port.h"
  21
  22 #include "wined3d_private.h"
  23
  24 WINE_DEFAULT_DEBUG_CHANNEL(d3d_shader);
  25
  26 #define WINED3D_SM4_INSTRUCTION_MODIFIER        (1 << 31)
  27
  28 #define WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT    24
  29 #define WINED3D_SM4_INSTRUCTION_LENGTH_MASK     (0xf << WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT)
  30
  31 #define WINED3D_SM4_OPCODE_MASK                 0xff
  32
  33 #define WINED3D_SM4_REGISTER_MODIFIER           (1 << 31)
  34
  35 #define WINED3D_SM4_REGISTER_ORDER_SHIFT        20
  36 #define WINED3D_SM4_REGISTER_ORDER_MASK         (0x3 << WINED3D_SM4_REGISTER_ORDER_SHIFT)
  37
  38 #define WINED3D_SM4_REGISTER_TYPE_SHIFT         12
  39 #define WINED3D_SM4_REGISTER_TYPE_MASK          (0xf << WINED3D_SM4_REGISTER_TYPE_SHIFT)
  40
  41 #define WINED3D_SM4_IMMCONST_TYPE_SHIFT         0
  42 #define WINED3D_SM4_IMMCONST_TYPE_MASK          (0x3 << WINED3D_SM4_IMMCONST_TYPE_SHIFT)
  43
  44 #define WINED3D_SM4_WRITEMASK_SHIFT             4
  45 #define WINED3D_SM4_WRITEMASK_MASK              (0xf << WINED3D_SM4_WRITEMASK_SHIFT)
  46
  47 #define WINED3D_SM4_SWIZZLE_SHIFT               4
  48 #define WINED3D_SM4_SWIZZLE_MASK                (0xff << WINED3D_SM4_SWIZZLE_SHIFT)
  49
  50 #define WINED3D_SM4_VERSION_MAJOR(version)      (((version) >> 4) & 0xf)
  51 #define WINED3D_SM4_VERSION_MINOR(version)      (((version) >> 0) & 0xf)
  52
  53 enum wined3d_sm4_opcode
  54 {
  55     WINED3D_SM4_OP_ADD          = 0x00,
  56     WINED3D_SM4_OP_AND          = 0x01,
  57     WINED3D_SM4_OP_BREAK        = 0x02,
  58     WINED3D_SM4_OP_BREAKC       = 0x03,
  59     WINED3D_SM4_OP_CUT          = 0x09,
  60     WINED3D_SM4_OP_DIV          = 0x0e,
  61     WINED3D_SM4_OP_DP3          = 0x10,
  62     WINED3D_SM4_OP_DP4          = 0x11,
  63     WINED3D_SM4_OP_EMIT         = 0x13,
  64     WINED3D_SM4_OP_ENDIF        = 0x15,
  65     WINED3D_SM4_OP_ENDLOOP      = 0x16,
  66     WINED3D_SM4_OP_EXP          = 0x19,
  67     WINED3D_SM4_OP_FRC          = 0x1a,
  68     WINED3D_SM4_OP_FTOI         = 0x1b,
  69     WINED3D_SM4_OP_IADD         = 0x1e,
  70     WINED3D_SM4_OP_IF           = 0x1f,
  71     WINED3D_SM4_OP_IEQ          = 0x20,
  72     WINED3D_SM4_OP_IGE          = 0x21,
  73     WINED3D_SM4_OP_IMUL         = 0x26,
  74     WINED3D_SM4_OP_ITOF         = 0x2b,
  75     WINED3D_SM4_OP_LD           = 0x2d,
  76     WINED3D_SM4_OP_LOG          = 0x2f,
  77     WINED3D_SM4_OP_LOOP         = 0x30,
  78     WINED3D_SM4_OP_LT           = 0x31,
  79     WINED3D_SM4_OP_MAD          = 0x32,
  80     WINED3D_SM4_OP_MIN          = 0x33,
  81     WINED3D_SM4_OP_MAX          = 0x34,
  82     WINED3D_SM4_OP_MOV          = 0x36,
  83     WINED3D_SM4_OP_MOVC         = 0x37,
  84     WINED3D_SM4_OP_MUL          = 0x38,
  85     WINED3D_SM4_OP_RET          = 0x3e,
  86     WINED3D_SM4_OP_RSQ          = 0x44,
  87     WINED3D_SM4_OP_SAMPLE       = 0x45,
  88     WINED3D_SM4_OP_SAMPLE_LOD   = 0x48,
  89     WINED3D_SM4_OP_SINCOS       = 0x4d,
  90     WINED3D_SM4_OP_UTOF         = 0x56,
  91 };
  92
  93 enum wined3d_sm4_register_type
  94 {
  95     WINED3D_SM4_RT_TEMP         = 0x0,
  96     WINED3D_SM4_RT_INPUT        = 0x1,
  97     WINED3D_SM4_RT_OUTPUT       = 0x2,
  98     WINED3D_SM4_RT_IMMCONST     = 0x4,
  99     WINED3D_SM4_RT_SAMPLER      = 0x6,
 100     WINED3D_SM4_RT_CONSTBUFFER  = 0x8,
 101     WINED3D_SM4_RT_NULL         = 0xd,
 102 };
 103
 104 enum wined3d_sm4_immconst_type
 105 {
 106     WINED3D_SM4_IMMCONST_SCALAR = 0x1,
 107     WINED3D_SM4_IMMCONST_VEC4   = 0x2,
 108 };
 109
 110 struct wined3d_sm4_data
 111 {
 112     struct wined3d_shader_version shader_version;
 113     const DWORD *end;
 114     const struct wined3d_shader_signature *output_signature;
 115 };
 116
 117 struct wined3d_sm4_opcode_info
 118 {
 119     enum wined3d_sm4_opcode opcode;
 120     enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx;
 121     UINT dst_count;
 122     UINT src_count;
 123 };
 124
 125 struct sysval_map
 126 {
 127     enum wined3d_sysval_semantic sysval;
 128     WINED3DSHADER_PARAM_REGISTER_TYPE register_type;
 129     UINT register_idx;
 130 };
 131
 132 static const struct wined3d_sm4_opcode_info opcode_table[] =
 133 {
 134     {WINED3D_SM4_OP_ADD,        WINED3DSIH_ADD,         1,  2},
 135     {WINED3D_SM4_OP_AND,        WINED3DSIH_AND,         1,  2},
 136     {WINED3D_SM4_OP_BREAK,      WINED3DSIH_BREAK,       0,  0},
 137     {WINED3D_SM4_OP_BREAKC,     WINED3DSIH_BREAKP,      0,  1},
 138     {WINED3D_SM4_OP_CUT,        WINED3DSIH_CUT,         0,  0},
 139     {WINED3D_SM4_OP_DIV,        WINED3DSIH_DIV,         1,  2},
 140     {WINED3D_SM4_OP_DP3,        WINED3DSIH_DP3,         1,  2},
 141     {WINED3D_SM4_OP_DP4,        WINED3DSIH_DP4,         1,  2},
 142     {WINED3D_SM4_OP_EMIT,       WINED3DSIH_EMIT,        0,  0},
 143     {WINED3D_SM4_OP_ENDIF,      WINED3DSIH_ENDIF,       0,  0},
 144     {WINED3D_SM4_OP_ENDLOOP,    WINED3DSIH_ENDLOOP,     0,  0},
 145     {WINED3D_SM4_OP_EXP,        WINED3DSIH_EXP,         1,  1},
 146     {WINED3D_SM4_OP_FRC,        WINED3DSIH_FRC,         1,  1},
 147     {WINED3D_SM4_OP_FTOI,       WINED3DSIH_FTOI,        1,  1},
 148     {WINED3D_SM4_OP_IADD,       WINED3DSIH_IADD,        1,  2},
 149     {WINED3D_SM4_OP_IF,         WINED3DSIH_IF,          0,  1},
 150     {WINED3D_SM4_OP_IEQ,        WINED3DSIH_IEQ,         1,  2},
 151     {WINED3D_SM4_OP_IGE,        WINED3DSIH_IGE,         1,  2},
 152     {WINED3D_SM4_OP_IMUL,       WINED3DSIH_IMUL,        2,  2},
 153     {WINED3D_SM4_OP_ITOF,       WINED3DSIH_ITOF,        1,  1},
 154     {WINED3D_SM4_OP_LD,         WINED3DSIH_LD,          1,  2},
 155     {WINED3D_SM4_OP_LOG,        WINED3DSIH_LOG,         1,  1},
 156     {WINED3D_SM4_OP_LOOP,       WINED3DSIH_LOOP,        0,  0},
 157     {WINED3D_SM4_OP_LT,         WINED3DSIH_LT,          1,  2},
 158     {WINED3D_SM4_OP_MAD,        WINED3DSIH_MAD,         1,  3},
 159     {WINED3D_SM4_OP_MIN,        WINED3DSIH_MIN,         1,  2},
 160     {WINED3D_SM4_OP_MAX,        WINED3DSIH_MAX,         1,  2},
 161     {WINED3D_SM4_OP_MOV,        WINED3DSIH_MOV,         1,  1},
 162     {WINED3D_SM4_OP_MOVC,       WINED3DSIH_MOVC,        1,  3},
 163     {WINED3D_SM4_OP_MUL,        WINED3DSIH_MUL,         1,  2},
 164     {WINED3D_SM4_OP_RET,        WINED3DSIH_RET,         0,  0},
 165     {WINED3D_SM4_OP_RSQ,        WINED3DSIH_RSQ,         1,  1},
 166     {WINED3D_SM4_OP_SAMPLE,     WINED3DSIH_SAMPLE,      1,  3},
 167     {WINED3D_SM4_OP_SAMPLE_LOD, WINED3DSIH_SAMPLE_LOD,  1,  4},
 168     {WINED3D_SM4_OP_SINCOS,     WINED3DSIH_SINCOS,      2,  1},
 169     {WINED3D_SM4_OP_UTOF,       WINED3DSIH_UTOF,        1,  1},
 170 };
 171
 172 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table[] =
 173 {
 174     /* WINED3D_SM4_RT_TEMP */           WINED3DSPR_TEMP,
 175     /* WINED3D_SM4_RT_INPUT */          WINED3DSPR_INPUT,
 176     /* WINED3D_SM4_RT_OUTPUT */         WINED3DSPR_OUTPUT,
 177     /* UNKNOWN */                       0,
 178     /* WINED3D_SM4_RT_IMMCONST */       WINED3DSPR_IMMCONST,
 179     /* UNKNOWN */                       0,
 180     /* WINED3D_SM4_RT_SAMPLER */        WINED3DSPR_SAMPLER,
 181     /* WINED3D_SM4_RT_RESOURCE */       WINED3DSPR_RESOURCE,
 182     /* WINED3D_SM4_RT_CONSTBUFFER */    WINED3DSPR_CONSTBUFFER,
 183     /* UNKNOWN */                       0,
 184     /* UNKNOWN */                       0,
 185     /* UNKNOWN */                       0,
 186     /* UNKNOWN */                       0,
 187     /* WINED3D_SM4_RT_NULL */           WINED3DSPR_NULL,
 188 };
 189
 190 static const struct sysval_map sysval_map[] =
 191 {
 192     {WINED3D_SV_DEPTH,      WINED3DSPR_DEPTHOUT,    0},
 193     {WINED3D_SV_TARGET0,    WINED3DSPR_COLOROUT,    0},
 194     {WINED3D_SV_TARGET1,    WINED3DSPR_COLOROUT,    1},
 195     {WINED3D_SV_TARGET2,    WINED3DSPR_COLOROUT,    2},
 196     {WINED3D_SV_TARGET3,    WINED3DSPR_COLOROUT,    3},
 197     {WINED3D_SV_TARGET4,    WINED3DSPR_COLOROUT,    4},
 198     {WINED3D_SV_TARGET5,    WINED3DSPR_COLOROUT,    5},
 199     {WINED3D_SV_TARGET6,    WINED3DSPR_COLOROUT,    6},
 200     {WINED3D_SV_TARGET7,    WINED3DSPR_COLOROUT,    7},
 201 };
 202
 203 static const struct wined3d_sm4_opcode_info *get_opcode_info(enum wined3d_sm4_opcode opcode)
 204 {
 205     unsigned int i;
 206
 207     for (i = 0; i < sizeof(opcode_table) / sizeof(*opcode_table); ++i)
 208     {
 209         if (opcode == opcode_table[i].opcode) return &opcode_table[i];
 210     }
 211
 212     return NULL;
 213 }
 214
 215 static void map_sysval(enum wined3d_sysval_semantic sysval, struct wined3d_shader_register *reg)
 216 {
 217     unsigned int i;
 218
 219     for (i = 0; i < sizeof(sysval_map) / sizeof(*sysval_map); ++i)
 220     {
 221         if (sysval == sysval_map[i].sysval)
 222         {
 223             reg->type = sysval_map[i].register_type;
 224             reg->idx = sysval_map[i].register_idx;
 225         }
 226     }
 227 }
 228
 229 static void map_register(struct wined3d_sm4_data *priv, struct wined3d_shader_register *reg)
 230 {
 231     switch (priv->shader_version.type)
 232     {
 233         case WINED3D_SHADER_TYPE_PIXEL:
 234             if (reg->type == WINED3DSPR_OUTPUT)
 235             {
 236                 unsigned int i;
 237                 const struct wined3d_shader_signature *s = priv->output_signature;
 238
 239                 if (!s)
 240                 {
 241                     ERR("Shader has no output signature, unable to map register.\n");
 242                     break;
 243                 }
 244
 245                 for (i = 0; i < s->element_count; ++i)
 246                 {
 247                     if (s->elements[i].register_idx == reg->idx)
 248                     {
 249                         map_sysval(s->elements[i].sysval_semantic, reg);
 250                         break;
 251                     }
 252                 }
 253             }
 254             break;
 255
 256         default:
 257             break;
 258     }
 259 }
 260
 261 static void *shader_sm4_init(const DWORD *byte_code, const struct wined3d_shader_signature *output_signature)
 262 {
 263     struct wined3d_sm4_data *priv = HeapAlloc(GetProcessHeap(), 0, sizeof(*priv));
 264     if (!priv)
 265     {
 266         ERR("Failed to allocate private data\n");
 267         return NULL;
 268     }
 269
 270     priv->output_signature = output_signature;
 271
 272     return priv;
 273 }
 274
 275 static void shader_sm4_free(void *data)
 276 {
 277     HeapFree(GetProcessHeap(), 0, data);
 278 }
 279
 280 static void shader_sm4_read_header(void *data, const DWORD **ptr, struct wined3d_shader_version *shader_version)
 281 {
 282     struct wined3d_sm4_data *priv = data;
 283     DWORD version_token;
 284
 285     priv->end = *ptr;
 286
 287     version_token = *(*ptr)++;
 288     TRACE("version: 0x%08x\n", version_token);
 289
 290     TRACE("token count: %u\n", **ptr);
 291     priv->end += *(*ptr)++;
 292
 293     switch (version_token >> 16)
 294     {
 295         case WINED3D_SM4_PS:
 296             priv->shader_version.type = WINED3D_SHADER_TYPE_PIXEL;
 297             break;
 298
 299         case WINED3D_SM4_VS:
 300             priv->shader_version.type = WINED3D_SHADER_TYPE_VERTEX;
 301             break;
 302
 303         case WINED3D_SM4_GS:
 304             priv->shader_version.type = WINED3D_SHADER_TYPE_GEOMETRY;
 305             break;
 306
 307         default:
 308             FIXME("Unrecognized shader type %#x\n", version_token >> 16);
 309     }
 310     priv->shader_version.major = WINED3D_SM4_VERSION_MAJOR(version_token);
 311     priv->shader_version.minor = WINED3D_SM4_VERSION_MINOR(version_token);
 312
 313     *shader_version = priv->shader_version;
 314 }
 315
 316 static void shader_sm4_read_opcode(void *data, const DWORD **ptr, struct wined3d_shader_instruction *ins,
 317         UINT *param_size)
 318 {
 319     const struct wined3d_sm4_opcode_info *opcode_info;
 320     DWORD token = *(*ptr)++;
 321     DWORD opcode = token & WINED3D_SM4_OPCODE_MASK;
 322
 323     *param_size = ((token & WINED3D_SM4_INSTRUCTION_LENGTH_MASK) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT) - 1;
 324
 325     opcode_info = get_opcode_info(opcode);
 326     if (!opcode_info)
 327     {
 328         FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode, token);
 329         ins->handler_idx = WINED3DSIH_TABLE_SIZE;
 330         return;
 331     }
 332
 333     ins->handler_idx = opcode_info->handler_idx;
 334     ins->flags = 0;
 335     ins->coissue = 0;
 336     ins->predicate = 0;
 337     ins->dst_count = opcode_info->dst_count;
 338     ins->src_count = opcode_info->src_count;
 339
 340     if (token & WINED3D_SM4_INSTRUCTION_MODIFIER)
 341     {
 342         DWORD modifier = *(*ptr)++;
 343         FIXME("Skipping modifier 0x%08x.\n", modifier);
 344     }
 345 }
 346
 347 static void shader_sm4_read_src_param(void *data, const DWORD **ptr, struct wined3d_shader_src_param *src_param,
 348         struct wined3d_shader_src_param *src_rel_addr)
 349 {
 350     struct wined3d_sm4_data *priv = data;
 351     DWORD token = *(*ptr)++;
 352     enum wined3d_sm4_register_type register_type;
 353     DWORD order;
 354
 355     register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
 356     if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
 357     {
 358         FIXME("Unhandled register type %#x\n", register_type);
 359         src_param->reg.type = WINED3DSPR_TEMP;
 360     }
 361     else
 362     {
 363         src_param->reg.type = register_type_table[register_type];
 364     }
 365
 366     if (token & WINED3D_SM4_REGISTER_MODIFIER)
 367     {
 368         DWORD modifier = *(*ptr)++;
 369         FIXME("Skipping modifier 0x%08x.\n", modifier);
 370     }
 371
 372     order = (token & WINED3D_SM4_REGISTER_ORDER_MASK) >> WINED3D_SM4_REGISTER_ORDER_SHIFT;
 373
 374     if (order < 1) src_param->reg.idx = ~0U;
 375     else src_param->reg.idx = *(*ptr)++;
 376
 377     if (order < 2) src_param->reg.array_idx = ~0U;
 378     else src_param->reg.array_idx = *(*ptr)++;
 379
 380     if (order > 2) FIXME("Unhandled order %u.\n", order);
 381
 382     if (register_type == WINED3D_SM4_RT_IMMCONST)
 383     {
 384         enum wined3d_sm4_immconst_type immconst_type =
 385                 (token & WINED3D_SM4_IMMCONST_TYPE_MASK) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT;
 386         src_param->swizzle = WINED3DSP_NOSWIZZLE;
 387
 388         switch(immconst_type)
 389         {
 390             case WINED3D_SM4_IMMCONST_SCALAR:
 391                 src_param->reg.immconst_type = WINED3D_IMMCONST_SCALAR;
 392                 memcpy(src_param->reg.immconst_data, *ptr, 1 * sizeof(DWORD));
 393                 *ptr += 1;
 394                 break;
 395
 396             case WINED3D_SM4_IMMCONST_VEC4:
 397                 src_param->reg.immconst_type = WINED3D_IMMCONST_VEC4;
 398                 memcpy(src_param->reg.immconst_data, *ptr, 4 * sizeof(DWORD));
 399                 *ptr += 4;
 400                 break;
 401
 402             default:
 403                 FIXME("Unhandled immediate constant type %#x\n", immconst_type);
 404                 break;
 405         }
 406     }
 407     else
 408     {
 409         src_param->swizzle = (token & WINED3D_SM4_SWIZZLE_MASK) >> WINED3D_SM4_SWIZZLE_SHIFT;
 410     }
 411
 412     src_param->modifiers = 0;
 413     src_param->reg.rel_addr = NULL;
 414
 415     map_register(priv, &src_param->reg);
 416 }
 417
 418 static void shader_sm4_read_dst_param(void *data, const DWORD **ptr, struct wined3d_shader_dst_param *dst_param,
 419         struct wined3d_shader_src_param *dst_rel_addr)
 420 {
 421     struct wined3d_sm4_data *priv = data;
 422     DWORD token = *(*ptr)++;
 423     enum wined3d_sm4_register_type register_type;
 424     DWORD order;
 425
 426     register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
 427     if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
 428     {
 429         FIXME("Unhandled register type %#x\n", register_type);
 430         dst_param->reg.type = WINED3DSPR_TEMP;
 431     }
 432     else
 433     {
 434         dst_param->reg.type = register_type_table[register_type];
 435     }
 436
 437     order = (token & WINED3D_SM4_REGISTER_ORDER_MASK) >> WINED3D_SM4_REGISTER_ORDER_SHIFT;
 438
 439     if (order < 1) dst_param->reg.idx = ~0U;
 440     else dst_param->reg.idx = *(*ptr)++;
 441
 442     if (order < 2) dst_param->reg.array_idx = ~0U;
 443     else dst_param->reg.array_idx = *(*ptr)++;
 444
 445     if (order > 2) FIXME("Unhandled order %u.\n", order);
 446
 447     dst_param->write_mask = (token & WINED3D_SM4_WRITEMASK_MASK) >> WINED3D_SM4_WRITEMASK_SHIFT;
 448     dst_param->modifiers = 0;
 449     dst_param->shift = 0;
 450     dst_param->reg.rel_addr = NULL;
 451
 452     map_register(priv, &dst_param->reg);
 453 }
 454
 455 static void shader_sm4_read_semantic(const DWORD **ptr, struct wined3d_shader_semantic *semantic)
 456 {
 457     FIXME("ptr %p, semantic %p stub!\n", ptr, semantic);
 458 }
 459
 460 static void shader_sm4_read_comment(const DWORD **ptr, const char **comment, UINT *comment_size)
 461 {
 462     FIXME("ptr %p, comment %p, comment_size %p stub!\n", ptr, comment, comment_size);
 463     *comment = NULL;
 464 }
 465
 466 static BOOL shader_sm4_is_end(void *data, const DWORD **ptr)
 467 {
 468     struct wined3d_sm4_data *priv = data;
 469     return *ptr == priv->end;
 470 }
 471
 472 const struct wined3d_shader_frontend sm4_shader_frontend =
 473 {
 474     shader_sm4_init,
 475     shader_sm4_free,
 476     shader_sm4_read_header,
 477     shader_sm4_read_opcode,
 478     shader_sm4_read_src_param,
 479     shader_sm4_read_dst_param,
 480     shader_sm4_read_semantic,
 481     shader_sm4_read_comment,
 482     shader_sm4_is_end,
 483 };