[PATCH 2/6] d3dx9: add preshaders to effect.

Paul Gofman gofmanp at gmail.com
Wed Mar 9 07:38:28 CST 2016


Signed-off-by: Paul Gofman <gofmanp at gmail.com>
---
 dlls/d3dx9_36/Makefile.in        |   3 +-
 dlls/d3dx9_36/d3dx9_36_private.h |  36 +++
 dlls/d3dx9_36/effect.c           |  80 +++--
 dlls/d3dx9_36/preshader.c        | 671 +++++++++++++++++++++++++++++++++++++++
 4 files changed, 757 insertions(+), 33 deletions(-)
 create mode 100644 dlls/d3dx9_36/preshader.c

diff --git a/dlls/d3dx9_36/Makefile.in b/dlls/d3dx9_36/Makefile.in
index 95e3045..b71b38e 100644
--- a/dlls/d3dx9_36/Makefile.in
+++ b/dlls/d3dx9_36/Makefile.in
@@ -19,6 +19,7 @@ C_SRCS = \
 	texture.c \
 	util.c \
 	volume.c \
-	xfile.c
+	xfile.c \
+	preshader.c
 
 RC_SRCS = version.rc
diff --git a/dlls/d3dx9_36/d3dx9_36_private.h b/dlls/d3dx9_36/d3dx9_36_private.h
index 79f3b76..f3da79d 100644
--- a/dlls/d3dx9_36/d3dx9_36_private.h
+++ b/dlls/d3dx9_36/d3dx9_36_private.h
@@ -110,4 +110,40 @@ const char *debug_d3dxparameter_registerset(D3DXREGISTER_SET r) DECLSPEC_HIDDEN;
 void set_number(void *outdata, D3DXPARAMETER_TYPE outtype,
         const void *indata, D3DXPARAMETER_TYPE intype) DECLSPEC_HIDDEN;
 
+struct d3dx_preshader;
+
+struct d3dx_parameter
+{
+    char *name;
+    char *semantic;
+    void *data;
+    D3DXPARAMETER_CLASS class;
+    D3DXPARAMETER_TYPE  type;
+    UINT rows;
+    UINT columns;
+    UINT element_count;
+    UINT annotation_count;
+    UINT member_count;
+    DWORD flags;
+    UINT bytes;
+    DWORD object_id;
+
+    D3DXHANDLE handle;
+
+    struct d3dx_parameter *annotations;
+    struct d3dx_parameter *members;
+
+    struct d3dx_parameter *referenced_param;
+    struct d3dx_preshader *pres;
+};
+
+struct d3dx9_base_effect;
+
+struct d3dx_parameter *get_parameter_by_name(struct d3dx9_base_effect *base,
+        struct d3dx_parameter *parameter, const char *name) DECLSPEC_HIDDEN;
+
+HRESULT d3dx_create_preshader(struct d3dx9_base_effect *base_effect, void *byte_code,
+        unsigned int byte_code_size, D3DXPARAMETER_TYPE type, struct d3dx_preshader **pres) DECLSPEC_HIDDEN;
+void d3dx_free_preshader(struct d3dx_preshader *pres) DECLSPEC_HIDDEN;
+
 #endif /* __WINE_D3DX9_36_PRIVATE_H */
diff --git a/dlls/d3dx9_36/effect.c b/dlls/d3dx9_36/effect.c
index e42cf86..32241f0 100644
--- a/dlls/d3dx9_36/effect.c
+++ b/dlls/d3dx9_36/effect.c
@@ -99,30 +99,6 @@ enum STATE_TYPE
     ST_ARRAY_SELECTOR,
 };
 
-struct d3dx_parameter
-{
-    char *name;
-    char *semantic;
-    void *data;
-    D3DXPARAMETER_CLASS class;
-    D3DXPARAMETER_TYPE  type;
-    UINT rows;
-    UINT columns;
-    UINT element_count;
-    UINT annotation_count;
-    UINT member_count;
-    DWORD flags;
-    UINT bytes;
-    DWORD object_id;
-
-    D3DXHANDLE handle;
-
-    struct d3dx_parameter *annotations;
-    struct d3dx_parameter *members;
-
-    struct d3dx_parameter *referenced_param;
-};
-
 struct d3dx_object
 {
     UINT size;
@@ -211,8 +187,6 @@ struct ID3DXEffectCompilerImpl
     struct d3dx9_base_effect base_effect;
 };
 
-static struct d3dx_parameter *get_parameter_by_name(struct d3dx9_base_effect *base,
-        struct d3dx_parameter *parameter, const char *name);
 static struct d3dx_parameter *get_annotation_by_name(UINT count, struct d3dx_parameter *parameters,
         const char *name);
 static HRESULT d3dx9_parse_state(struct d3dx9_base_effect *base, struct d3dx_state *state,
@@ -543,6 +517,12 @@ static void free_parameter(struct d3dx_parameter *param, BOOL element, BOOL chil
     if (!param)
         return;
 
+    if (param->pres)
+    {
+        d3dx_free_preshader(param->pres);
+        param->pres = NULL;
+    }
+
     if (param->annotations)
     {
         for (i = 0; i < param->annotation_count; ++i)
@@ -894,7 +874,7 @@ static struct d3dx_parameter *get_annotation_by_name(UINT count, struct d3dx_par
     return NULL;
 }
 
-static struct d3dx_parameter *get_parameter_by_name(struct d3dx9_base_effect *base,
+struct d3dx_parameter *get_parameter_by_name(struct d3dx9_base_effect *base,
         struct d3dx_parameter *parameter, const char *name)
 {
     UINT i, count, length;
@@ -5427,6 +5407,7 @@ static HRESULT d3dx9_parse_array_selector(struct d3dx9_base_effect *base, struct
     DWORD string_size;
     struct d3dx_object *object = &base->objects[param->object_id];
     char *ptr = object->data;
+    HRESULT ret;
 
     TRACE("Parsing array entry selection state for parameter %p.\n", param);
 
@@ -5443,9 +5424,30 @@ static HRESULT d3dx9_parse_array_selector(struct d3dx9_base_effect *base, struct
     }
     TRACE("Unknown DWORD: 0x%.8x.\n", *(DWORD *)(ptr + string_size));
 
-    FIXME("Parse preshader.\n");
+    d3dx_create_preshader(base, (DWORD *)(ptr + string_size) + 1, object->size - (string_size + 1),
+            D3DXPT_INT, &param->pres);
+    ret = D3D_OK;
+    param = param->referenced_param;
+    if (param->type == D3DXPT_VERTEXSHADER || param->type == D3DXPT_PIXELSHADER)
+    {
+        UINT i;
 
-    return D3D_OK;
+        for (i = 0; i < param->element_count; i++)
+        {
+            if (param->members[i].type != param->type)
+            {
+                FIXME("Unexpected member parameter type %u (need %u).\n", param->members[i].type, param->type);
+                return D3DXERR_INVALIDDATA;
+            }
+            if (!param->members[i].pres)
+            {
+                TRACE("Creating preshader for object %u.\n", param->members[i].object_id);
+                object = &base->objects[param->members[i].object_id];
+                d3dx_create_preshader(base, object->data, object->size, param->type, &param->members[i].pres);
+            }
+        }
+    }
+    return ret;
 }
 
 static HRESULT d3dx9_parse_resource(struct d3dx9_base_effect *base, const char *data, const char **ptr)
@@ -5551,7 +5553,11 @@ static HRESULT d3dx9_parse_resource(struct d3dx9_base_effect *base, const char *
                         return hr;
 
                     if (object->data)
-                        hr = d3dx9_create_object(base, object);
+                    {
+                        if (FAILED(hr = d3dx9_create_object(base, object)))
+                            return hr;
+                        d3dx_create_preshader(base, object->data, object->size, param->type, &param->pres);
+                    }
                     break;
 
                 case D3DXPT_BOOL:
@@ -5559,7 +5565,9 @@ static HRESULT d3dx9_parse_resource(struct d3dx9_base_effect *base, const char *
                 case D3DXPT_FLOAT:
                 case D3DXPT_STRING:
                     state->type = ST_FXLC;
-                    hr = d3dx9_copy_data(&base->objects[param->object_id], ptr);
+                    if (FAILED(hr = d3dx9_copy_data(&base->objects[param->object_id], ptr)))
+                        return hr;
+                    d3dx_create_preshader(base, object->data, object->size, param->type, &param->pres);
                     break;
 
                 default:
@@ -5577,7 +5585,15 @@ static HRESULT d3dx9_parse_resource(struct d3dx9_base_effect *base, const char *
             param->referenced_param = get_parameter_by_name(base, NULL, object->data);
             if (param->referenced_param)
             {
-                TRACE("Mapping to parameter %p.\n", param->referenced_param);
+                struct d3dx_parameter *refpar = param->referenced_param;
+
+                TRACE("Mapping to parameter %p, having object id %u.\n", refpar, refpar->object_id);
+                if (refpar->type == D3DXPT_VERTEXSHADER || refpar->type == D3DXPT_PIXELSHADER)
+                {
+                    struct d3dx_object *refobj = &base->objects[refpar->object_id];
+
+                    d3dx_create_preshader(base, refobj->data, refobj->size, refpar->type, &refpar->pres);
+                }
             }
             else
             {
diff --git a/dlls/d3dx9_36/preshader.c b/dlls/d3dx9_36/preshader.c
new file mode 100644
index 0000000..536514a
--- /dev/null
+++ b/dlls/d3dx9_36/preshader.c
@@ -0,0 +1,671 @@
+/*
+ * FX preshader parsing & execution, setting shader constants
+ *     from effect parameters.
+ *
+ * Copyright 2016 Paul Gofman
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include "config.h"
+#include "wine/port.h"
+
+#include "windef.h"
+#include "wingdi.h"
+#include "d3dx9_36_private.h"
+
+#include "wine/debug.h"
+
+#include <float.h>
+
+WINE_DEFAULT_DEBUG_CHANNEL(d3dx);
+
+#define MIN(a, b) ((a) < (b) ? (a) : (b))
+#define MAX(a, b) ((a) > (b) ? (a) : (b))
+
+enum preshader_type
+{
+    PRES_TYPE_FXLC,
+    PRES_TYPE_SHADER
+};
+
+enum PRES_OPS
+{
+    PO_NOP,
+    PO_MOV,
+    PO_OP_MAX
+};
+
+typedef float (*pres_op_func)(float *args, int ncomp);
+
+static float pres_mov(float *args, int ncomp) {return args[0];}
+
+#define PRES_OPCODE_MASK 0x7FF00000
+#define PRES_OPCODE_SHIFT 20
+#define PRES_SCALAR_FLAG 0x80000000
+#define PRES_NCOMP_MASK  0x0000FFFF
+
+#define FOURCC_PRES 0x53455250
+#define FOURCC_CLIT 0x54494C43
+#define FOURCC_FXLC 0x434C5846
+#define FOURCC_PRSI 0x49535250
+#define PRES_SIGN 0x46580000
+
+struct op_info
+{
+    DWORD opcode;
+    char mnem[8];
+    unsigned int ninp_args;
+    BOOL func_all_comps;
+    pres_op_func func;
+};
+
+static struct op_info pres_op_info[] =
+{
+    {0x000, "nop", 0, 0, NULL    }, /* PO_NOP */
+    {0x100, "mov", 1, 0, pres_mov}, /* PO_MOV */
+};
+
+enum PRES_VALUE_TYPE
+{
+    PRES_VT_FLOAT,
+    PRES_VT_DOUBLE,
+    PRES_VT_INT,
+    PRES_VT_BOOL
+};
+
+enum PRES_TABLES
+{
+    PRES_TAB_NONE,
+    PRES_TAB_IMMED,   /* immediate double constants from CLIT */
+    PRES_TAB_CONST,   /* preshader input float constants */
+    PRES_TAB_VERTEX,  /* not used */
+    PRES_TAB_OCONST,  /* shader input float constants */
+    PRES_TAB_OBCONST, /* shader input bool constants */
+    PRES_TAB_OICONST, /* shader input int constants */
+    PRES_TAB_REG,     /* registers */
+    PRES_TAB_MAX
+};
+
+static const struct
+{
+    unsigned int value_size;
+    unsigned int reg_value_cnt;
+    enum PRES_VALUE_TYPE type;
+}
+table_info[] =
+{
+    {0, 0},
+    {sizeof(double), 1, PRES_VT_DOUBLE}, /* PRES_TAB_IMMED */
+    {sizeof(float),  4, PRES_VT_FLOAT }, /* PRES_TAB_CONST */
+    {sizeof(float),  4, PRES_VT_FLOAT }, /* PRES_TAB_VERTEX */
+    {sizeof(float),  4, PRES_VT_FLOAT }, /* PRES_TAB_OCONST */
+    {sizeof(BOOL),   1, PRES_VT_BOOL  }, /* PRES_TAB_OBCONST */
+    {sizeof(int),    4, PRES_VT_INT,  }, /* PRES_TAB_OICONST */
+    {sizeof(float),  4, PRES_VT_FLOAT }  /* PRES_TAB_REG */
+};
+
+static const char *table_symbol[] =
+{
+    "(null)", "imm", "c", "v", "oc", "ob", "oi", "r"
+};
+
+static const char *xyzw_str = "xyzw";
+
+#define OFFSET2REG(table, offset) ((offset) / table_info[table].reg_value_cnt)
+
+static const enum PRES_TABLES pres_regset2table[] =
+{
+    PRES_TAB_OBCONST,  /* D3DXRS_BOOL */
+    PRES_TAB_OICONST,  /* D3DXRS_INT4 */
+    PRES_TAB_CONST,    /* D3DXRS_FLOAT4 */
+    PRES_TAB_NONE,     /* D3DXRS_SAMPLER */
+};
+
+static const enum PRES_TABLES shad_regset2table[] =
+{
+    PRES_TAB_OBCONST,  /* D3DXRS_BOOL */
+    PRES_TAB_OICONST,  /* D3DXRS_INT4 */
+    PRES_TAB_OCONST,   /* D3DXRS_FLOAT4 */
+    PRES_TAB_NONE,     /* D3DXRS_SAMPLER */
+};
+
+struct d3dx_pres_operand
+{
+    enum PRES_TABLES table;
+    unsigned int offset; /* value index, not register index */
+};
+
+struct d3dx_pres_ins
+{
+    enum PRES_OPS op;
+    BOOL scalar_op; /* first input argument is scalar,
+                       first scalar component is propagated */
+    unsigned int ncomps;
+    unsigned int ninp_args;
+    struct d3dx_pres_operand inputs[3];
+    struct d3dx_pres_operand output;
+};
+
+#define PRES_VS_BITS_PER_WORD (sizeof(DWORD)*8)
+
+struct d3dx_const_tab
+{
+    unsigned int ninputs;
+    D3DXCONSTANT_DESC *inputs;
+    struct d3dx_parameter **inputs_param;
+    ID3DXConstantTable *ctab;
+    /* TODO: do not keep input constant structure
+       (use it only at the parse stage) */
+    const enum PRES_TABLES *regset2table;
+};
+
+struct d3dx_fxlc
+{
+    void *tables[PRES_TAB_MAX];
+    unsigned int table_sizes[PRES_TAB_MAX]; /* registers count */
+    DWORD *table_value_set[PRES_TAB_MAX];
+
+    unsigned int nins;
+    struct d3dx_pres_ins *ins;
+
+    struct d3dx_const_tab inputs;
+};
+
+struct d3dx_preshader
+{
+    enum preshader_type pres_type;
+    D3DXPARAMETER_TYPE param_type;
+
+    struct d3dx_fxlc fxlc;
+    struct d3dx_const_tab shader_inputs;
+};
+
+static void val_set_reg(struct d3dx_fxlc *fxlc, unsigned int table, unsigned int reg_idx)
+{
+    fxlc->table_value_set[table][reg_idx / PRES_VS_BITS_PER_WORD] |=
+            1 << (reg_idx % PRES_VS_BITS_PER_WORD);
+}
+
+static void dump_shader_bytecode(void *data, unsigned int size)
+{
+    DWORD *words = (DWORD *)data;
+    unsigned int i, j, n;
+
+    size /= 4;
+    i = 0;
+    while (i < size)
+    {
+        n = MIN(size - i, 15);
+        for (j = 0; j < n; j++)
+            MESSAGE("0x%08X,", words[i + j]);
+        i += n;
+        MESSAGE("\n");
+    }
+}
+
+static DWORD *find_bytecode_comment(DWORD *ptr, DWORD fourcc)
+{
+    while ((*ptr & 0xFFFF) == 0xFFFE)
+    {
+        if (*(ptr + 1) == fourcc)
+            return ptr + 2;
+        ptr += 1 + (*ptr >> 16);
+    }
+    return NULL;
+}
+
+static DWORD *parse_pres_arg(DWORD *ptr, struct d3dx_pres_operand *opr)
+{
+    if (*ptr != 0)
+    {
+        FIXME("Relative addressing not supported yet, word %#x.\n", *ptr);
+        return NULL;
+    }
+    ptr++;
+
+    if (*ptr < 1 || *ptr >= PRES_TAB_MAX || *ptr == PRES_TAB_VERTEX)
+    {
+        FIXME("Unsupported argument table: %#x.\n", *ptr);
+        return NULL;
+    }
+    opr->table = *ptr;
+    ++ptr;
+    opr->offset = *ptr;
+    ++ptr;
+
+    if (opr->table == PRES_TAB_OBCONST)
+        opr->offset /= 4;
+    return ptr;
+}
+
+static DWORD *parse_pres_ins(DWORD *ptr, struct d3dx_pres_ins *ins)
+{
+    DWORD ins_code, ins_raw;
+    unsigned int ninp_args;
+    unsigned int i;
+
+    ins_raw = *ptr;
+    ins_code = (ins_raw & PRES_OPCODE_MASK) >> PRES_OPCODE_SHIFT;
+    ins->ncomps = ins_raw & PRES_NCOMP_MASK;
+    ins->scalar_op = (ins_raw & PRES_SCALAR_FLAG) != 0;
+
+    if (ins->ncomps < 1 || ins->ncomps > 4)
+    {
+        FIXME("Unsupported number of components: %u.\n", ins->ncomps);
+        return NULL;
+    }
+    ++ptr;
+    ninp_args = *ptr++;
+    for (i = 0; i < PO_OP_MAX; i++)
+        if (ins_code == pres_op_info[i].opcode)
+            break;
+    if (i == PO_OP_MAX)
+    {
+        FIXME("Unknown opcode: %#x (raw %#x).\n", ins_code, ins_raw);
+        return NULL;
+    }
+    ins->op = i;
+    if (ninp_args != pres_op_info[i].ninp_args)
+    {
+        FIXME("Actual input args %u, expected %u, instruction %s.\n", ninp_args,
+                pres_op_info[i].ninp_args, pres_op_info[i].mnem);
+        return NULL;
+    }
+    ins->ninp_args = ninp_args;
+    for (i = 0; i < ninp_args; i++)
+    {
+        ptr = parse_pres_arg(ptr, &ins->inputs[i]);
+        if (ptr == NULL)
+            return NULL;
+    }
+    ptr = parse_pres_arg(ptr, &ins->output);
+    return ptr;
+}
+
+static HRESULT get_constants_desc(DWORD *byte_code, struct d3dx_const_tab *out, struct d3dx9_base_effect *base)
+{
+    ID3DXConstantTable *ctab;
+    D3DXCONSTANT_DESC *cdesc;
+    struct d3dx_parameter **inputs_param;
+    D3DXCONSTANTTABLE_DESC desc;
+    HRESULT hr;
+    D3DXHANDLE hc;
+    unsigned int i;
+    unsigned int cnt;
+
+    out->inputs = cdesc = NULL;
+    out->ctab = NULL;
+    out->inputs_param = NULL;
+    out->ninputs = 0;
+    inputs_param = NULL;
+    hr = D3DXGetShaderConstantTable(byte_code, &ctab);
+    if (FAILED(hr) || !ctab)
+    {
+        TRACE("Could not get CTAB data, hr %#x.\n", hr);
+        /* returnin OK is not a typo */
+        return D3D_OK;
+    }
+    hr = ID3DXConstantTable_GetDesc(ctab, &desc);
+    if (FAILED(hr))
+    {
+        FIXME("Could not get CTAB desc, hr %#x.\n", hr);
+        goto err_out;
+    }
+
+    cdesc = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(D3DXCONSTANT_DESC) * desc.Constants);
+    inputs_param = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(struct d3dx_parameter **) * desc.Constants);
+    for (i = 0; i < desc.Constants; i++)
+    {
+        hc = ID3DXConstantTable_GetConstant(ctab, NULL, i);
+        if (hc == NULL)
+        {
+            FIXME("Null constant handle.\n");
+            goto err_out;
+        }
+        cnt = 1;
+        hr = ID3DXConstantTable_GetConstantDesc(ctab, hc, &cdesc[i], &cnt);
+        if (FAILED(hr))
+        {
+            FIXME("Could not get constant desc, hr %#x.\n", hr);
+            goto err_out;
+        }
+        inputs_param[i] = get_parameter_by_name(base, NULL, cdesc[i].Name);
+        if (cdesc[i].Class == D3DXPC_OBJECT)
+            TRACE("Object %s, parameter %p.\n", cdesc[i].Name, inputs_param[i]);
+        else if (inputs_param[i] == NULL)
+            ERR("Could not find parameter %s in effect.\n", cdesc[i].Name);
+    }
+    out->ninputs = desc.Constants;
+    out->inputs = cdesc;
+    out->inputs_param = inputs_param;
+    out->ctab = ctab;
+    return D3D_OK;
+
+err_out:
+    if (cdesc)
+        HeapFree(GetProcessHeap(), 0, cdesc);
+    if (inputs_param)
+        HeapFree(GetProcessHeap(), 0, inputs_param);
+    if (ctab)
+        ID3DXConstantTable_Release(ctab);
+    return hr;
+}
+
+static void update_table_sizes_consts(unsigned int *table_sizes, struct d3dx_const_tab *ctab)
+{
+    unsigned int i, table, max_offset;
+
+    for (i = 0; i < ctab->ninputs; i++)
+    {
+        max_offset = ctab->inputs[i].RegisterIndex + ctab->inputs[i].RegisterCount;
+        table = ctab->regset2table[ctab->inputs[i].RegisterSet];
+        table_sizes[table] = MAX(table_sizes[table], max_offset);
+    }
+}
+
+static void dump_arg(struct d3dx_fxlc *fxlc, struct d3dx_pres_operand *arg, int ncomps)
+{
+    unsigned int i, table;
+
+    table = arg->table;
+    if (table == PRES_TAB_IMMED)
+    {
+        TRACE("(");
+        for (i = 0; i < ncomps; i++)
+            TRACE(i < ncomps - 1 ? "%lf, " : "%lf", ((double *)fxlc->tables[PRES_TAB_IMMED])[arg->offset + i]);
+        TRACE(")");
+    }
+    else
+    {
+        TRACE("%s%u.", table_symbol[table], OFFSET2REG(table, arg->offset));
+        for (i = 0; i < ncomps; i++)
+            TRACE("%c", xyzw_str[(arg->offset + i) % 4]);
+    }
+}
+
+static void dump_registers(struct d3dx_const_tab *ctab)
+{
+    unsigned int table, i;
+
+    for (i = 0; i < ctab->ninputs; i++)
+    {
+        table = ctab->regset2table[ctab->inputs[i].RegisterSet];
+        TRACE("//   %-12s %s%-4u %u\n", ctab->inputs_param[i] ? ctab->inputs_param[i]->name : "(nil)",
+                table_symbol[table], ctab->inputs[i].RegisterIndex, ctab->inputs[i].RegisterCount);
+    }
+}
+
+static void dump_ins(struct d3dx_fxlc *fxlc, struct d3dx_pres_ins *ins)
+{
+    unsigned int i;
+
+    TRACE("    %s ", pres_op_info[ins->op].mnem);
+    dump_arg(fxlc, &ins->output, pres_op_info[ins->op].func_all_comps ? 1 : ins->ncomps);
+    for (i = 0; i < ins->ninp_args; i++)
+    {
+        TRACE(",");
+        dump_arg(fxlc, &ins->inputs[i], ins->scalar_op && !i ? 1 : ins->ncomps);
+    }
+    TRACE("\n");
+}
+
+static void dump_preshader(struct d3dx_fxlc *fxlc)
+{
+    unsigned int i;
+
+    TRACE("// Preshader registers:\n");
+    dump_registers(&fxlc->inputs);
+    TRACE("    preshader\n");
+    for (i = 0; i < fxlc->nins; i++)
+        dump_ins(fxlc, &fxlc->ins[i]);
+}
+
+static HRESULT parse_preshader(struct d3dx_fxlc *fxlc, DWORD *ptr, struct d3dx9_base_effect *base)
+{
+    DWORD *p;
+    unsigned int i, j, nconst, max_offset;
+    double *dconst;
+    HRESULT hr;
+    DWORD saved_word;
+
+    TRACE("Preshader version: %#x.\n", *ptr & 0xFFFF);
+
+    p = find_bytecode_comment(ptr + 1, FOURCC_CLIT);
+    if (p)
+    {
+        nconst = *p;
+        dconst = (double *)(p + 1);
+    }
+    else
+    {
+        nconst = 0;
+        dconst = NULL;
+    }
+    TRACE("%u double constants.\n", nconst);
+
+    p = find_bytecode_comment(ptr + 1, FOURCC_FXLC);
+    if (!p)
+    {
+        WARN("Could not find preshader code.\n");
+        return D3D_OK;
+    }
+
+    fxlc->nins = *p;
+    TRACE("%u instructions.\n", fxlc->nins);
+    ++p;
+    fxlc->ins = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(struct d3dx_pres_ins) * fxlc->nins);
+    for (i = 0; i < fxlc->nins; i++)
+    {
+        p = parse_pres_ins(p, &fxlc->ins[i]);
+        if (p == NULL)
+            return D3DXERR_INVALIDDATA;
+    }
+
+    saved_word = *ptr;
+    *ptr = 0xfffe0000;
+    hr = get_constants_desc(ptr, &fxlc->inputs, base);
+    *ptr = saved_word;
+    if (FAILED(hr))
+        return hr;
+
+    fxlc->inputs.regset2table = pres_regset2table;
+
+    fxlc->table_sizes[PRES_TAB_IMMED] = nconst;
+
+    for (i = 0; i < fxlc->nins; i++)
+    {
+        unsigned int table;
+
+        for (j = 0; j < fxlc->ins[i].ninp_args; j++)
+        {
+            table = fxlc->ins[i].inputs[j].table;
+            max_offset = OFFSET2REG(table, fxlc->ins[i].inputs[j].offset + fxlc->ins[i].ncomps - 1);
+            fxlc->table_sizes[table] = MAX(fxlc->table_sizes[table], max_offset + 1);
+        }
+        table = fxlc->ins[i].output.table;
+        max_offset = OFFSET2REG(table, fxlc->ins[i].output.offset + fxlc->ins[i].ncomps - 1);
+        fxlc->table_sizes[table] = MAX(fxlc->table_sizes[table], max_offset + 1);
+    }
+    update_table_sizes_consts(fxlc->table_sizes, &fxlc->inputs);
+
+    if (fxlc->table_sizes[PRES_TAB_IMMED])
+    {
+        unsigned int sz;
+
+        sz = sizeof(double) * fxlc->table_sizes[PRES_TAB_IMMED];
+        fxlc->tables[PRES_TAB_IMMED] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sz);
+        fxlc->table_value_set[PRES_TAB_IMMED] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
+                    (sz + PRES_VS_BITS_PER_WORD - 1)/PRES_VS_BITS_PER_WORD);
+        if (!fxlc->tables[PRES_TAB_IMMED] || !fxlc->table_value_set[PRES_TAB_IMMED])
+            return E_OUTOFMEMORY;
+        memcpy(fxlc->tables[PRES_TAB_IMMED], dconst, sizeof(double) * nconst);
+        for (i = 0; i < nconst; i++)
+            val_set_reg(fxlc, PRES_TAB_IMMED, i);
+    }
+
+    return D3D_OK;
+}
+
+HRESULT d3dx_create_preshader(struct d3dx9_base_effect *base_effect, void *byte_code, unsigned int byte_code_size,
+        D3DXPARAMETER_TYPE type, struct d3dx_preshader **pres_out)
+{
+    struct d3dx_preshader *pres;
+    HRESULT hr;
+    DWORD *ptr;
+    unsigned int i;
+
+    TRACE("base_effect %p, byte_code %p, byte_code_size %u, type %u, pres_out %p.\n",
+            base_effect, byte_code, byte_code_size, type, pres_out);
+
+    if (byte_code == NULL)
+    {
+        *pres_out = NULL;
+        return D3DERR_INVALIDCALL;
+    }
+
+    hr = E_FAIL;
+    pres = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*pres));
+    if (!pres)
+    {
+        hr = E_OUTOFMEMORY;
+        goto err_out;
+    }
+
+    pres->param_type = type;
+    switch (type)
+    {
+        case D3DXPT_VERTEXSHADER:
+        case D3DXPT_PIXELSHADER:
+            pres->pres_type = PRES_TYPE_SHADER;
+            break;
+        default:
+            pres->pres_type = PRES_TYPE_FXLC;
+            break;
+    }
+    pres->shader_inputs.regset2table = shad_regset2table;
+
+    ptr = (DWORD *)byte_code;
+    if (pres->pres_type == PRES_TYPE_SHADER)
+    {
+        if ((*ptr & 0xFFFE0000) != 0xFFFE0000)
+        {
+            FIXME("Invalid shader signature: %#x.\n", *ptr);
+            hr = D3DXERR_INVALIDDATA;
+            goto err_out;
+        }
+        TRACE("Shader version: %#x.\n", *ptr & 0xFFFF);
+
+        hr = get_constants_desc(ptr, &pres->shader_inputs, base_effect);
+        if (FAILED(hr))
+        {
+            FIXME("Could not get shader constant table, hr %#x.\n", hr);
+            goto err_out;
+        }
+        update_table_sizes_consts(pres->fxlc.table_sizes, &pres->shader_inputs);
+        ptr++;
+        ptr = find_bytecode_comment(ptr, FOURCC_PRES);
+        if (ptr == NULL)
+            TRACE("No preshader found.\n");
+    }
+    if (ptr)
+    {
+        hr = parse_preshader(&pres->fxlc, ptr, base_effect);
+        if (FAILED(hr))
+        {
+            FIXME("Failed parsing preshader: byte code for analysis follows.\n");
+            dump_shader_bytecode(byte_code, byte_code_size);
+            goto err_out;
+        }
+    }
+
+    for (i = PRES_TAB_IMMED + 1; i < PRES_TAB_MAX; i++)
+    {
+        unsigned int sz;
+
+        TRACE("table_sizes[%u] %u\n", i, pres->fxlc.table_sizes[i]);
+        sz = table_info[i].value_size * table_info[i].reg_value_cnt * pres->fxlc.table_sizes[i];
+        if (sz != 0)
+        {
+            pres->fxlc.tables[i] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sz);
+            pres->fxlc.table_value_set[i] = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
+                    (sz + PRES_VS_BITS_PER_WORD - 1)/PRES_VS_BITS_PER_WORD);
+            if (!pres->fxlc.tables[i] || !pres->fxlc.table_value_set[i])
+            {
+                hr = E_OUTOFMEMORY;
+                goto err_out;
+            }
+        }
+    }
+
+    if (TRACE_ON(d3dx))
+    {
+        dump_shader_bytecode(byte_code, byte_code_size);
+        dump_preshader(&pres->fxlc);
+        if (pres->pres_type == PRES_TYPE_SHADER)
+        {
+            TRACE("// Shader registers:\n");
+            dump_registers(&pres->shader_inputs);
+        }
+    }
+    *pres_out = pres;
+    return D3D_OK;
+
+err_out:
+    FIXME("Error creating preshader.\n");
+    d3dx_free_preshader(pres);
+    *pres_out = NULL;
+    return hr;
+}
+
+void d3dx_free_const_tab(struct d3dx_const_tab *ctab)
+{
+    if (ctab->inputs)
+        HeapFree(GetProcessHeap(), 0, ctab->inputs);
+    if (ctab->inputs_param)
+        HeapFree(GetProcessHeap(), 0, ctab->inputs_param);
+    if (ctab->ctab)
+        ID3DXConstantTable_Release(ctab->ctab);
+}
+
+void d3dx_free_fxlc(struct d3dx_fxlc *fxlc)
+{
+    unsigned int i;
+
+    if (fxlc->ins)
+        HeapFree(GetProcessHeap(), 0, fxlc->ins);
+
+    for (i = PRES_TAB_IMMED; i < PRES_TAB_MAX; i++)
+    {
+        if (fxlc->tables[i])
+            HeapFree(GetProcessHeap(), 0, fxlc->tables[i]);
+        if (fxlc->table_value_set[i])
+            HeapFree(GetProcessHeap(), 0, fxlc->table_value_set[i]);
+    }
+    d3dx_free_const_tab(&fxlc->inputs);
+}
+
+void d3dx_free_preshader(struct d3dx_preshader *pres)
+{
+    TRACE("pres %p.\n", pres);
+
+    if (!pres)
+        return ;
+
+    d3dx_free_fxlc(&pres->fxlc);
+    d3dx_free_const_tab(&pres->shader_inputs);
+    HeapFree(GetProcessHeap(), 0, pres);
+}
+
-- 
2.5.0




More information about the wine-patches mailing list