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_LENGTH_SHIFT    24
  27 #define WINED3D_SM4_INSTRUCTION_LENGTH_MASK     (0xf << WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT)
  28
  29 #define WINED3D_SM4_OPCODE_MASK                 0xff
  30
  31 #define WINED3D_SM4_REGISTER_TYPE_SHIFT         12
  32 #define WINED3D_SM4_REGISTER_TYPE_MASK          (0xf << WINED3D_SM4_REGISTER_TYPE_SHIFT)
  33
  34 #define WINED3D_SM4_IMMCONST_TYPE_SHIFT         0
  35 #define WINED3D_SM4_IMMCONST_TYPE_MASK          (0x3 << WINED3D_SM4_IMMCONST_TYPE_SHIFT)
  36
  37 #define WINED3D_SM4_WRITEMASK_SHIFT             4
  38 #define WINED3D_SM4_WRITEMASK_MASK              (0xf << WINED3D_SM4_WRITEMASK_SHIFT)
  39
  40 #define WINED3D_SM4_SWIZZLE_SHIFT               4
  41 #define WINED3D_SM4_SWIZZLE_MASK                (0xff << WINED3D_SM4_SWIZZLE_SHIFT)
  42
  43 #define WINED3D_SM4_VERSION_MAJOR(version)      (((version) >> 4) & 0xf)
  44 #define WINED3D_SM4_VERSION_MINOR(version)      (((version) >> 0) & 0xf)
  45
  46 enum wined3d_sm4_opcode
  47 {
  48     WINED3D_SM4_OP_ADD      = 0x00,
  49     WINED3D_SM4_OP_EXP      = 0x19,
  50     WINED3D_SM4_OP_MOV      = 0x36,
  51     WINED3D_SM4_OP_MUL      = 0x38,
  52     WINED3D_SM4_OP_RET      = 0x3e,
  53     WINED3D_SM4_OP_SINCOS   = 0x4d,
  54 };
  55
  56 enum wined3d_sm4_register_type
  57 {
  58     WINED3D_SM4_RT_TEMP     = 0x0,
  59     WINED3D_SM4_RT_INPUT    = 0x1,
  60     WINED3D_SM4_RT_OUTPUT   = 0x2,
  61     WINED3D_SM4_RT_IMMCONST = 0x4,
  62 };
  63
  64 enum wined3d_sm4_immconst_type
  65 {
  66     WINED3D_SM4_IMMCONST_FLOAT  = 0x1,
  67     WINED3D_SM4_IMMCONST_FLOAT4 = 0x2,
  68 };
  69
  70 struct wined3d_sm4_data
  71 {
  72     struct wined3d_shader_version shader_version;
  73     const DWORD *end;
  74 };
  75
  76 struct wined3d_sm4_opcode_info
  77 {
  78     enum wined3d_sm4_opcode opcode;
  79     enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx;
  80     UINT dst_count;
  81     UINT src_count;
  82 };
  83
  84 static const struct wined3d_sm4_opcode_info opcode_table[] =
  85 {
  86     {WINED3D_SM4_OP_ADD,    WINED3DSIH_ADD,         1,  2},
  87     {WINED3D_SM4_OP_EXP,    WINED3DSIH_EXP,         1,  1},
  88     {WINED3D_SM4_OP_MOV,    WINED3DSIH_MOV,         1,  1},
  89     {WINED3D_SM4_OP_MUL,    WINED3DSIH_MUL,         1,  2},
  90     {WINED3D_SM4_OP_RET,    WINED3DSIH_RET,         0,  0},
  91     {WINED3D_SM4_OP_SINCOS, WINED3DSIH_SINCOS,      1,  2},
  92 };
  93
  94 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table[] =
  95 {
  96     /* WINED3D_SM4_RT_TEMP */       WINED3DSPR_TEMP,
  97     /* WINED3D_SM4_RT_INPUT */      WINED3DSPR_INPUT,
  98     /* WINED3D_SM4_RT_OUTPUT */     WINED3DSPR_OUTPUT,
  99     /* UNKNOWN */                   0,
 100     /* WINED3D_SM4_RT_IMMCONST */   WINED3DSPR_IMMCONST,
 101 };
 102
 103 static const struct wined3d_sm4_opcode_info *get_opcode_info(enum wined3d_sm4_opcode opcode)
 104 {
 105     unsigned int i;
 106
 107     for (i = 0; i < sizeof(opcode_table) / sizeof(*opcode_table); ++i)
 108     {
 109         if (opcode == opcode_table[i].opcode) return &opcode_table[i];
 110     }
 111
 112     return NULL;
 113 }
 114
 115 static void *shader_sm4_init(const DWORD *byte_code)
 116 {
 117     struct wined3d_sm4_data *priv = HeapAlloc(GetProcessHeap(), 0, sizeof(*priv));
 118     if (!priv)
 119     {
 120         ERR("Failed to allocate private data\n");
 121         return NULL;
 122     }
 123
 124     return priv;
 125 }
 126
 127 static void shader_sm4_free(void *data)
 128 {
 129     HeapFree(GetProcessHeap(), 0, data);
 130 }
 131
 132 static void shader_sm4_read_header(void *data, const DWORD **ptr, struct wined3d_shader_version *shader_version)
 133 {
 134     struct wined3d_sm4_data *priv = data;
 135     DWORD version_token;
 136
 137     priv->end = *ptr;
 138
 139     version_token = *(*ptr)++;
 140     TRACE("version: 0x%08x\n", version_token);
 141
 142     TRACE("token count: %u\n", **ptr);
 143     priv->end += *(*ptr)++;
 144
 145     switch (version_token >> 16)
 146     {
 147         case WINED3D_SM4_PS:
 148             priv->shader_version.type = WINED3D_SHADER_TYPE_PIXEL;
 149             break;
 150
 151         case WINED3D_SM4_VS:
 152             priv->shader_version.type = WINED3D_SHADER_TYPE_VERTEX;
 153             break;
 154
 155         case WINED3D_SM4_GS:
 156             priv->shader_version.type = WINED3D_SHADER_TYPE_GEOMETRY;
 157             break;
 158
 159         default:
 160             FIXME("Unrecognized shader type %#x\n", version_token >> 16);
 161     }
 162     priv->shader_version.major = WINED3D_SM4_VERSION_MAJOR(version_token);
 163     priv->shader_version.minor = WINED3D_SM4_VERSION_MINOR(version_token);
 164
 165     *shader_version = priv->shader_version;
 166 }
 167
 168 static void shader_sm4_read_opcode(void *data, const DWORD **ptr, struct wined3d_shader_instruction *ins,
 169         UINT *param_size)
 170 {
 171     const struct wined3d_sm4_opcode_info *opcode_info;
 172     DWORD token = *(*ptr)++;
 173     DWORD opcode = token & WINED3D_SM4_OPCODE_MASK;
 174
 175     *param_size = ((token & WINED3D_SM4_INSTRUCTION_LENGTH_MASK) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT) - 1;
 176
 177     opcode_info = get_opcode_info(opcode);
 178     if (!opcode_info)
 179     {
 180         FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode, token);
 181         ins->handler_idx = WINED3DSIH_TABLE_SIZE;
 182         return;
 183     }
 184
 185     ins->handler_idx = opcode_info->handler_idx;
 186     ins->flags = 0;
 187     ins->coissue = 0;
 188     ins->predicate = 0;
 189     ins->dst_count = opcode_info->dst_count;
 190     ins->src_count = opcode_info->src_count;
 191 }
 192
 193 static void shader_sm4_read_src_param(void *data, const DWORD **ptr, struct wined3d_shader_src_param *src_param,
 194         struct wined3d_shader_src_param *src_rel_addr)
 195 {
 196     DWORD token = *(*ptr)++;
 197     enum wined3d_sm4_register_type register_type;
 198
 199     register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
 200     if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
 201     {
 202         FIXME("Unhandled register type %#x\n", register_type);
 203         src_param->reg.type = WINED3DSPR_TEMP;
 204     }
 205     else
 206     {
 207         src_param->reg.type = register_type_table[register_type];
 208     }
 209
 210     if (register_type == WINED3D_SM4_RT_IMMCONST)
 211     {
 212         enum wined3d_sm4_immconst_type immconst_type =
 213                 (token & WINED3D_SM4_IMMCONST_TYPE_MASK) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT;
 214         src_param->swizzle = WINED3DSP_NOSWIZZLE;
 215
 216         switch(immconst_type)
 217         {
 218             case WINED3D_SM4_IMMCONST_FLOAT:
 219                 src_param->reg.immconst_type = WINED3D_IMMCONST_FLOAT;
 220                 memcpy(src_param->reg.immconst_data, *ptr, 1 * sizeof(DWORD));
 221                 *ptr += 1;
 222                 break;
 223
 224             case WINED3D_SM4_IMMCONST_FLOAT4:
 225                 src_param->reg.immconst_type = WINED3D_IMMCONST_FLOAT4;
 226                 memcpy(src_param->reg.immconst_data, *ptr, 4 * sizeof(DWORD));
 227                 *ptr += 4;
 228                 break;
 229
 230             default:
 231                 FIXME("Unhandled immediate constant type %#x\n", immconst_type);
 232                 break;
 233         }
 234     }
 235     else
 236     {
 237         src_param->reg.idx = *(*ptr)++;
 238         src_param->swizzle = (token & WINED3D_SM4_SWIZZLE_MASK) >> WINED3D_SM4_SWIZZLE_SHIFT;
 239     }
 240
 241     src_param->modifiers = 0;
 242     src_param->reg.rel_addr = NULL;
 243 }
 244
 245 static void shader_sm4_read_dst_param(void *data, const DWORD **ptr, struct wined3d_shader_dst_param *dst_param,
 246         struct wined3d_shader_src_param *dst_rel_addr)
 247 {
 248     DWORD token = *(*ptr)++;
 249     UINT register_idx = *(*ptr)++;
 250     enum wined3d_sm4_register_type register_type;
 251
 252     register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
 253     if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
 254     {
 255         FIXME("Unhandled register type %#x\n", register_type);
 256         dst_param->reg.type = WINED3DSPR_TEMP;
 257     }
 258     else
 259     {
 260         dst_param->reg.type = register_type_table[register_type];
 261     }
 262
 263     dst_param->reg.idx = register_idx;
 264     dst_param->write_mask = (token & WINED3D_SM4_WRITEMASK_MASK) >> WINED3D_SM4_WRITEMASK_SHIFT;
 265     dst_param->modifiers = 0;
 266     dst_param->shift = 0;
 267     dst_param->reg.rel_addr = NULL;
 268 }
 269
 270 static void shader_sm4_read_semantic(const DWORD **ptr, struct wined3d_shader_semantic *semantic)
 271 {
 272     FIXME("ptr %p, semantic %p stub!\n", ptr, semantic);
 273 }
 274
 275 static void shader_sm4_read_comment(const DWORD **ptr, const char **comment)
 276 {
 277     FIXME("ptr %p, comment %p stub!\n", ptr, comment);
 278     *comment = NULL;
 279 }
 280
 281 static BOOL shader_sm4_is_end(void *data, const DWORD **ptr)
 282 {
 283     struct wined3d_sm4_data *priv = data;
 284     return *ptr == priv->end;
 285 }
 286
 287 const struct wined3d_shader_frontend sm4_shader_frontend =
 288 {
 289     shader_sm4_init,
 290     shader_sm4_free,
 291     shader_sm4_read_header,
 292     shader_sm4_read_opcode,
 293     shader_sm4_read_src_param,
 294     shader_sm4_read_dst_param,
 295     shader_sm4_read_semantic,
 296     shader_sm4_read_comment,
 297     shader_sm4_is_end,
 298 };