Rico Schüller : d3dcompiler: Add initial RDEF parsing.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: a2a7000eeb8180468f189d8b5d3c0b13abd2e8db
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=a2a7000eeb8180468f189d8b5d3c0b13abd2e8db

Author: Rico Schüller <kgbricola at web.de>
Date:   Tue Dec 28 11:50:05 2010 +0100

d3dcompiler: Add initial RDEF parsing.

---

 dlls/d3dcompiler_43/d3dcompiler_private.h |    5 ++
 dlls/d3dcompiler_43/reflection.c          |   77 +++++++++++++++++++++++++++++
 2 files changed, 82 insertions(+), 0 deletions(-)

diff --git a/dlls/d3dcompiler_43/d3dcompiler_private.h b/dlls/d3dcompiler_43/d3dcompiler_private.h
index 9355c44..91cce9e 100644
--- a/dlls/d3dcompiler_43/d3dcompiler_private.h
+++ b/dlls/d3dcompiler_43/d3dcompiler_private.h
@@ -70,6 +70,11 @@ struct d3dcompiler_shader_reflection
     ID3D11ShaderReflection ID3D11ShaderReflection_iface;
     LONG refcount;
 
+    char *creator;
+    UINT flags;
+    UINT bound_resource_count;
+    UINT constant_buffer_count;
+
     UINT mov_instruction_count;
     UINT conversion_instruction_count;
     UINT instruction_count;
diff --git a/dlls/d3dcompiler_43/reflection.c b/dlls/d3dcompiler_43/reflection.c
index d50e18c..eb01db8 100644
--- a/dlls/d3dcompiler_43/reflection.c
+++ b/dlls/d3dcompiler_43/reflection.c
@@ -25,6 +25,30 @@
 
 WINE_DEFAULT_DEBUG_CHANNEL(d3dcompiler);
 
+static BOOL copy_name(const char *ptr, char **name)
+{
+    size_t name_len;
+
+    if (!ptr) return TRUE;
+
+    name_len = strlen(ptr) + 1;
+    if (name_len == 1)
+    {
+        return TRUE;
+    }
+
+    *name = HeapAlloc(GetProcessHeap(), 0, name_len);
+    if (!*name)
+    {
+        ERR("Failed to allocate name memory.\n");
+        return FALSE;
+    }
+
+    memcpy(*name, ptr, name_len);
+
+    return TRUE;
+}
+
 static void free_signature(struct d3dcompiler_shader_signature *sig)
 {
     TRACE("Free signature %p\n", sig);
@@ -54,6 +78,8 @@ static void reflection_cleanup(struct d3dcompiler_shader_reflection *ref)
         free_signature(ref->pcsg);
         HeapFree(GetProcessHeap(), 0, ref->pcsg);
     }
+
+    HeapFree(GetProcessHeap(), 0, ref->creator);
 }
 
 static inline struct d3dcompiler_shader_reflection *impl_from_ID3D11ShaderReflection(ID3D11ShaderReflection *iface)
@@ -423,6 +449,48 @@ static HRESULT d3dcompiler_parse_stat(struct d3dcompiler_shader_reflection *r, c
     return E_FAIL;
 }
 
+static HRESULT d3dcompiler_parse_rdef(struct d3dcompiler_shader_reflection *r, const char *data, DWORD data_size)
+{
+    const char *ptr = data;
+    DWORD size = data_size >> 2;
+    DWORD offset;
+
+    TRACE("Size %u\n", size);
+
+    read_dword(&ptr, &r->constant_buffer_count);
+    TRACE("Constant buffer count: %u\n", r->constant_buffer_count);
+
+    read_dword(&ptr, &offset);
+    TRACE("Constant buffer offset: %x\n", offset);
+
+    read_dword(&ptr, &r->bound_resource_count);
+    TRACE("Bound resource count: %u\n", r->bound_resource_count);
+
+    read_dword(&ptr, &offset);
+    TRACE("Bound resource offset: %x\n", offset);
+
+    skip_dword_unknown(&ptr, 1);
+
+    read_dword(&ptr, &r->flags);
+    TRACE("Flags: %u\n", r->flags);
+
+    read_dword(&ptr, &offset);
+    TRACE("Creator at offset %#x.\n", offset);
+
+    if (!copy_name(data + offset, &r->creator))
+    {
+        ERR("Failed to copy name.\n");
+        return E_OUTOFMEMORY;
+    }
+    TRACE("Creator: %s.\n", debugstr_a(r->creator));
+
+    /* todo: Parse D3D11_SHADER_INPUT_BIND_DESC Structure */
+
+    /* todo: Parse Constant buffers */
+
+    return S_OK;
+}
+
 HRESULT d3dcompiler_parse_signature(struct d3dcompiler_shader_signature *s, const char *data, DWORD data_size)
 {
     D3D11_SIGNATURE_PARAMETER_DESC *d;
@@ -520,6 +588,15 @@ HRESULT d3dcompiler_shader_reflection_init(struct d3dcompiler_shader_reflection
                 }
                 break;
 
+            case TAG_RDEF:
+                hr = d3dcompiler_parse_rdef(reflection, section->data, section->data_size);
+                if (FAILED(hr))
+                {
+                    WARN("Failed to parse RDEF section.\n");
+                    goto err_out;
+                }
+                break;
+
             case TAG_ISGN:
                 reflection->isgn = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*reflection->isgn));
                 if (!reflection->isgn)