| /* |
| * Copyright 2009 Henri Verbeet for CodeWeavers |
| * Copyright 2010 Rico Schüller |
| * |
| * 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 "d3dcompiler_private.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(d3dcompiler); |
| |
| enum D3DCOMPILER_SIGNATURE_ELEMENT_SIZE |
| { |
| D3DCOMPILER_SIGNATURE_ELEMENT_SIZE6 = 6, |
| D3DCOMPILER_SIGNATURE_ELEMENT_SIZE7 = 7, |
| }; |
| |
| #define D3DCOMPILER_SHADER_TARGET_VERSION_MASK 0xffff |
| #define D3DCOMPILER_SHADER_TARGET_SHADERTYPE_MASK 0xffff0000 |
| |
| struct d3dcompiler_shader_signature |
| { |
| D3D11_SIGNATURE_PARAMETER_DESC *elements; |
| UINT element_count; |
| char *string_data; |
| }; |
| |
| struct d3dcompiler_shader_reflection_type |
| { |
| ID3D11ShaderReflectionType ID3D11ShaderReflectionType_iface; |
| |
| DWORD id; |
| struct wine_rb_entry entry; |
| |
| struct d3dcompiler_shader_reflection *reflection; |
| |
| D3D11_SHADER_TYPE_DESC desc; |
| struct d3dcompiler_shader_reflection_type_member *members; |
| }; |
| |
| struct d3dcompiler_shader_reflection_type_member |
| { |
| char *name; |
| DWORD offset; |
| struct d3dcompiler_shader_reflection_type *type; |
| }; |
| |
| struct d3dcompiler_shader_reflection_variable |
| { |
| ID3D11ShaderReflectionVariable ID3D11ShaderReflectionVariable_iface; |
| |
| struct d3dcompiler_shader_reflection_constant_buffer *constant_buffer; |
| struct d3dcompiler_shader_reflection_type *type; |
| |
| char *name; |
| UINT start_offset; |
| UINT size; |
| UINT flags; |
| void *default_value; |
| }; |
| |
| struct d3dcompiler_shader_reflection_constant_buffer |
| { |
| ID3D11ShaderReflectionConstantBuffer ID3D11ShaderReflectionConstantBuffer_iface; |
| |
| struct d3dcompiler_shader_reflection *reflection; |
| |
| char *name; |
| D3D_CBUFFER_TYPE type; |
| UINT variable_count; |
| UINT size; |
| UINT flags; |
| |
| struct d3dcompiler_shader_reflection_variable *variables; |
| }; |
| |
| /* ID3D11ShaderReflection */ |
| struct d3dcompiler_shader_reflection |
| { |
| ID3D11ShaderReflection ID3D11ShaderReflection_iface; |
| LONG refcount; |
| |
| DWORD target; |
| char *creator; |
| UINT flags; |
| UINT version; |
| UINT bound_resource_count; |
| UINT constant_buffer_count; |
| |
| UINT mov_instruction_count; |
| UINT conversion_instruction_count; |
| UINT instruction_count; |
| UINT emit_instruction_count; |
| D3D_PRIMITIVE_TOPOLOGY gs_output_topology; |
| UINT gs_max_output_vertex_count; |
| D3D_PRIMITIVE input_primitive; |
| UINT cut_instruction_count; |
| UINT dcl_count; |
| UINT static_flow_control_count; |
| UINT float_instruction_count; |
| UINT temp_register_count; |
| UINT int_instruction_count; |
| UINT uint_instruction_count; |
| UINT temp_array_count; |
| UINT array_instruction_count; |
| UINT texture_normal_instructions; |
| UINT texture_load_instructions; |
| UINT texture_comp_instructions; |
| UINT texture_bias_instructions; |
| UINT texture_gradient_instructions; |
| UINT dynamic_flow_control_count; |
| UINT c_control_points; |
| D3D_TESSELLATOR_OUTPUT_PRIMITIVE hs_output_primitive; |
| D3D_TESSELLATOR_PARTITIONING hs_prtitioning; |
| D3D_TESSELLATOR_DOMAIN tessellator_domain; |
| |
| struct d3dcompiler_shader_signature *isgn; |
| struct d3dcompiler_shader_signature *osgn; |
| struct d3dcompiler_shader_signature *pcsg; |
| char *resource_string; |
| D3D11_SHADER_INPUT_BIND_DESC *bound_resources; |
| struct d3dcompiler_shader_reflection_constant_buffer *constant_buffers; |
| struct wine_rb_tree types; |
| }; |
| |
| static struct d3dcompiler_shader_reflection_type *get_reflection_type(struct d3dcompiler_shader_reflection *reflection, const char *data, DWORD offset); |
| |
| static const struct ID3D11ShaderReflectionConstantBufferVtbl d3dcompiler_shader_reflection_constant_buffer_vtbl; |
| static const struct ID3D11ShaderReflectionVariableVtbl d3dcompiler_shader_reflection_variable_vtbl; |
| static const struct ID3D11ShaderReflectionTypeVtbl d3dcompiler_shader_reflection_type_vtbl; |
| |
| /* null objects - needed for invalid calls */ |
| static struct d3dcompiler_shader_reflection_constant_buffer null_constant_buffer = {{&d3dcompiler_shader_reflection_constant_buffer_vtbl}}; |
| static struct d3dcompiler_shader_reflection_type null_type = {{&d3dcompiler_shader_reflection_type_vtbl}}; |
| static struct d3dcompiler_shader_reflection_variable null_variable = {{&d3dcompiler_shader_reflection_variable_vtbl}, |
| &null_constant_buffer, &null_type}; |
| |
| 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 BOOL copy_value(const char *ptr, void **value, DWORD size) |
| { |
| if (!ptr || !size) return TRUE; |
| |
| *value = HeapAlloc(GetProcessHeap(), 0, size); |
| if (!*value) |
| { |
| ERR("Failed to allocate value memory.\n"); |
| return FALSE; |
| } |
| |
| memcpy(*value, ptr, size); |
| |
| return TRUE; |
| } |
| |
| static void *d3dcompiler_rb_alloc(size_t size) |
| { |
| return HeapAlloc(GetProcessHeap(), 0, size); |
| } |
| |
| static void *d3dcompiler_rb_realloc(void *ptr, size_t size) |
| { |
| return HeapReAlloc(GetProcessHeap(), 0, ptr, size); |
| } |
| |
| static void d3dcompiler_rb_free(void *ptr) |
| { |
| HeapFree(GetProcessHeap(), 0, ptr); |
| } |
| |
| static int d3dcompiler_shader_reflection_type_compare(const void *key, const struct wine_rb_entry *entry) |
| { |
| const struct d3dcompiler_shader_reflection_type *t = WINE_RB_ENTRY_VALUE(entry, const struct d3dcompiler_shader_reflection_type, entry); |
| const DWORD *id = key; |
| |
| return *id - t->id; |
| } |
| |
| static void free_type_member(struct d3dcompiler_shader_reflection_type_member *member) |
| { |
| if (member) |
| { |
| HeapFree(GetProcessHeap(), 0, member->name); |
| } |
| } |
| |
| static void d3dcompiler_shader_reflection_type_destroy(struct wine_rb_entry *entry, void *context) |
| { |
| struct d3dcompiler_shader_reflection_type *t = WINE_RB_ENTRY_VALUE(entry, struct d3dcompiler_shader_reflection_type, entry); |
| unsigned int i; |
| |
| TRACE("reflection type %p.\n", t); |
| |
| if (t->members) |
| { |
| for (i = 0; i < t->desc.Members; ++i) |
| { |
| free_type_member(&t->members[i]); |
| } |
| HeapFree(GetProcessHeap(), 0, t->members); |
| } |
| |
| HeapFree(GetProcessHeap(), 0, t); |
| } |
| |
| static const struct wine_rb_functions d3dcompiler_shader_reflection_type_rb_functions = |
| { |
| d3dcompiler_rb_alloc, |
| d3dcompiler_rb_realloc, |
| d3dcompiler_rb_free, |
| d3dcompiler_shader_reflection_type_compare, |
| }; |
| |
| static void free_signature(struct d3dcompiler_shader_signature *sig) |
| { |
| TRACE("Free signature %p\n", sig); |
| |
| HeapFree(GetProcessHeap(), 0, sig->elements); |
| HeapFree(GetProcessHeap(), 0, sig->string_data); |
| } |
| |
| static void free_variable(struct d3dcompiler_shader_reflection_variable *var) |
| { |
| if (var) |
| { |
| HeapFree(GetProcessHeap(), 0, var->name); |
| HeapFree(GetProcessHeap(), 0, var->default_value); |
| } |
| } |
| |
| static void free_constant_buffer(struct d3dcompiler_shader_reflection_constant_buffer *cb) |
| { |
| if (cb->variables) |
| { |
| unsigned int i; |
| |
| for (i = 0; i < cb->variable_count; ++i) |
| { |
| free_variable(&cb->variables[i]); |
| } |
| HeapFree(GetProcessHeap(), 0, cb->variables); |
| } |
| |
| HeapFree(GetProcessHeap(), 0, cb->name); |
| } |
| |
| static void reflection_cleanup(struct d3dcompiler_shader_reflection *ref) |
| { |
| TRACE("Cleanup %p\n", ref); |
| |
| if (ref->isgn) |
| { |
| free_signature(ref->isgn); |
| HeapFree(GetProcessHeap(), 0, ref->isgn); |
| } |
| |
| if (ref->osgn) |
| { |
| free_signature(ref->osgn); |
| HeapFree(GetProcessHeap(), 0, ref->osgn); |
| } |
| |
| if (ref->pcsg) |
| { |
| free_signature(ref->pcsg); |
| HeapFree(GetProcessHeap(), 0, ref->pcsg); |
| } |
| |
| if (ref->constant_buffers) |
| { |
| unsigned int i; |
| |
| for (i = 0; i < ref->constant_buffer_count; ++i) |
| { |
| free_constant_buffer(&ref->constant_buffers[i]); |
| } |
| } |
| |
| wine_rb_destroy(&ref->types, d3dcompiler_shader_reflection_type_destroy, NULL); |
| HeapFree(GetProcessHeap(), 0, ref->constant_buffers); |
| HeapFree(GetProcessHeap(), 0, ref->bound_resources); |
| HeapFree(GetProcessHeap(), 0, ref->resource_string); |
| HeapFree(GetProcessHeap(), 0, ref->creator); |
| } |
| |
| /* IUnknown methods */ |
| |
| static inline struct d3dcompiler_shader_reflection *impl_from_ID3D11ShaderReflection(ID3D11ShaderReflection *iface) |
| { |
| return CONTAINING_RECORD(iface, struct d3dcompiler_shader_reflection, ID3D11ShaderReflection_iface); |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_QueryInterface(ID3D11ShaderReflection *iface, REFIID riid, void **object) |
| { |
| TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object); |
| |
| if (IsEqualGUID(riid, &IID_ID3D11ShaderReflection) |
| || IsEqualGUID(riid, &IID_IUnknown)) |
| { |
| IUnknown_AddRef(iface); |
| *object = iface; |
| return S_OK; |
| } |
| |
| WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid)); |
| |
| *object = NULL; |
| return E_NOINTERFACE; |
| } |
| |
| static ULONG STDMETHODCALLTYPE d3dcompiler_shader_reflection_AddRef(ID3D11ShaderReflection *iface) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| ULONG refcount = InterlockedIncrement(&This->refcount); |
| |
| TRACE("%p increasing refcount to %u\n", This, refcount); |
| |
| return refcount; |
| } |
| |
| static ULONG STDMETHODCALLTYPE d3dcompiler_shader_reflection_Release(ID3D11ShaderReflection *iface) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| ULONG refcount = InterlockedDecrement(&This->refcount); |
| |
| TRACE("%p decreasing refcount to %u\n", This, refcount); |
| |
| if (!refcount) |
| { |
| reflection_cleanup(This); |
| HeapFree(GetProcessHeap(), 0, This); |
| } |
| |
| return refcount; |
| } |
| |
| /* ID3D11ShaderReflection methods */ |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetDesc(ID3D11ShaderReflection *iface, D3D11_SHADER_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| FIXME("iface %p, desc %p partial stub!\n", iface, desc); |
| |
| if (!desc) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_FAIL; |
| } |
| |
| desc->Version = This->version; |
| desc->Creator = This->creator; |
| desc->Flags = This->flags; |
| desc->ConstantBuffers = This->constant_buffer_count; |
| desc->BoundResources = This->bound_resource_count; |
| desc->InputParameters = This->isgn ? This->isgn->element_count : 0; |
| desc->OutputParameters = This->osgn ? This->osgn->element_count : 0; |
| desc->InstructionCount = This->instruction_count; |
| desc->TempRegisterCount = This->temp_register_count; |
| desc->TempArrayCount = This->temp_array_count; |
| desc->DefCount = 0; |
| desc->DclCount = This->dcl_count; |
| desc->TextureNormalInstructions = This->texture_normal_instructions; |
| desc->TextureLoadInstructions = This->texture_load_instructions; |
| desc->TextureCompInstructions = This->texture_comp_instructions; |
| desc->TextureBiasInstructions = This->texture_bias_instructions; |
| desc->TextureGradientInstructions = This->texture_gradient_instructions; |
| desc->FloatInstructionCount = This->float_instruction_count; |
| desc->IntInstructionCount = This->int_instruction_count; |
| desc->UintInstructionCount = This->uint_instruction_count; |
| desc->StaticFlowControlCount = This->static_flow_control_count; |
| desc->DynamicFlowControlCount = This->dynamic_flow_control_count; |
| desc->MacroInstructionCount = 0; |
| desc->ArrayInstructionCount = This->array_instruction_count; |
| desc->CutInstructionCount = This->cut_instruction_count; |
| desc->EmitInstructionCount = This->emit_instruction_count; |
| desc->GSOutputTopology = This->gs_output_topology; |
| desc->GSMaxOutputVertexCount = This->gs_max_output_vertex_count; |
| desc->InputPrimitive = This->input_primitive; |
| desc->PatchConstantParameters = This->pcsg ? This->pcsg->element_count : 0; |
| desc->cGSInstanceCount = 0; |
| desc->cControlPoints = This->c_control_points; |
| desc->HSOutputPrimitive = This->hs_output_primitive; |
| desc->HSPartitioning = This->hs_prtitioning; |
| desc->TessellatorDomain = This->tessellator_domain; |
| desc->cBarrierInstructions = 0; |
| desc->cInterlockedInstructions = 0; |
| desc->cTextureStoreInstructions = 0; |
| |
| return S_OK; |
| } |
| |
| static struct ID3D11ShaderReflectionConstantBuffer * STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetConstantBufferByIndex( |
| ID3D11ShaderReflection *iface, UINT index) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p, index %u\n", iface, index); |
| |
| if (index >= This->constant_buffer_count) |
| { |
| WARN("Invalid argument specified\n"); |
| return &null_constant_buffer.ID3D11ShaderReflectionConstantBuffer_iface; |
| } |
| |
| return &This->constant_buffers[index].ID3D11ShaderReflectionConstantBuffer_iface; |
| } |
| |
| static struct ID3D11ShaderReflectionConstantBuffer * STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetConstantBufferByName( |
| ID3D11ShaderReflection *iface, const char *name) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| unsigned int i; |
| |
| TRACE("iface %p, name %s\n", iface, debugstr_a(name)); |
| |
| if (!name) |
| { |
| WARN("Invalid argument specified\n"); |
| return &null_constant_buffer.ID3D11ShaderReflectionConstantBuffer_iface; |
| } |
| |
| for (i = 0; i < This->constant_buffer_count; ++i) |
| { |
| struct d3dcompiler_shader_reflection_constant_buffer *d = &This->constant_buffers[i]; |
| |
| if (!strcmp(d->name, name)) |
| { |
| TRACE("Returning ID3D11ShaderReflectionConstantBuffer %p.\n", d); |
| return &d->ID3D11ShaderReflectionConstantBuffer_iface; |
| } |
| } |
| |
| WARN("Invalid name specified\n"); |
| |
| return &null_constant_buffer.ID3D11ShaderReflectionConstantBuffer_iface; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetResourceBindingDesc( |
| ID3D11ShaderReflection *iface, UINT index, D3D11_SHADER_INPUT_BIND_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p, index %u, desc %p\n", iface, index, desc); |
| |
| if (!desc || index >= This->bound_resource_count) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_INVALIDARG; |
| } |
| |
| *desc = This->bound_resources[index]; |
| |
| return S_OK; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetInputParameterDesc( |
| ID3D11ShaderReflection *iface, UINT index, D3D11_SIGNATURE_PARAMETER_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p, index %u, desc %p\n", iface, index, desc); |
| |
| if (!desc || !This->isgn || index >= This->isgn->element_count) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_INVALIDARG; |
| } |
| |
| *desc = This->isgn->elements[index]; |
| |
| return S_OK; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetOutputParameterDesc( |
| ID3D11ShaderReflection *iface, UINT index, D3D11_SIGNATURE_PARAMETER_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p, index %u, desc %p\n", iface, index, desc); |
| |
| if (!desc || !This->osgn || index >= This->osgn->element_count) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_INVALIDARG; |
| } |
| |
| *desc = This->osgn->elements[index]; |
| |
| return S_OK; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetPatchConstantParameterDesc( |
| ID3D11ShaderReflection *iface, UINT index, D3D11_SIGNATURE_PARAMETER_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p, index %u, desc %p\n", iface, index, desc); |
| |
| if (!desc || !This->pcsg || index >= This->pcsg->element_count) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_INVALIDARG; |
| } |
| |
| *desc = This->pcsg->elements[index]; |
| |
| return S_OK; |
| } |
| |
| static struct ID3D11ShaderReflectionVariable * STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetVariableByName( |
| ID3D11ShaderReflection *iface, const char *name) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| unsigned int i, k; |
| |
| TRACE("iface %p, name %s\n", iface, debugstr_a(name)); |
| |
| if (!name) |
| { |
| WARN("Invalid name specified\n"); |
| return &null_variable.ID3D11ShaderReflectionVariable_iface; |
| } |
| |
| for (i = 0; i < This->constant_buffer_count; ++i) |
| { |
| struct d3dcompiler_shader_reflection_constant_buffer *cb = &This->constant_buffers[i]; |
| |
| for (k = 0; k < cb->variable_count; ++k) |
| { |
| struct d3dcompiler_shader_reflection_variable *v = &cb->variables[k]; |
| |
| if (!strcmp(v->name, name)) |
| { |
| TRACE("Returning ID3D11ShaderReflectionVariable %p.\n", v); |
| return &v->ID3D11ShaderReflectionVariable_iface; |
| } |
| } |
| } |
| |
| WARN("Invalid name specified\n"); |
| |
| return &null_variable.ID3D11ShaderReflectionVariable_iface; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetResourceBindingDescByName( |
| ID3D11ShaderReflection *iface, const char *name, D3D11_SHADER_INPUT_BIND_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| unsigned int i; |
| |
| TRACE("iface %p, name %s, desc %p\n", iface, debugstr_a(name), desc); |
| |
| if (!desc || !name) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_INVALIDARG; |
| } |
| |
| for (i = 0; i < This->bound_resource_count; ++i) |
| { |
| D3D11_SHADER_INPUT_BIND_DESC *d = &This->bound_resources[i]; |
| |
| if (!strcmp(d->Name, name)) |
| { |
| TRACE("Returning D3D11_SHADER_INPUT_BIND_DESC %p.\n", d); |
| *desc = *d; |
| return S_OK; |
| } |
| } |
| |
| WARN("Invalid name specified\n"); |
| |
| return E_INVALIDARG; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetMovInstructionCount( |
| ID3D11ShaderReflection *iface) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p\n", iface); |
| |
| return This->mov_instruction_count; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetMovcInstructionCount( |
| ID3D11ShaderReflection *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return 0; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetConversionInstructionCount( |
| ID3D11ShaderReflection *iface) |
| { |
| struct d3dcompiler_shader_reflection *This = impl_from_ID3D11ShaderReflection(iface); |
| |
| TRACE("iface %p\n", iface); |
| |
| return This->conversion_instruction_count; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetBitwiseInstructionCount( |
| ID3D11ShaderReflection *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return 0; |
| } |
| |
| static D3D_PRIMITIVE STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetGSInputPrimitive( |
| ID3D11ShaderReflection *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return 0; |
| } |
| |
| static BOOL STDMETHODCALLTYPE d3dcompiler_shader_reflection_IsSampleFrequencyShader( |
| ID3D11ShaderReflection *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return FALSE; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetNumInterfaceSlots( |
| ID3D11ShaderReflection *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return 0; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetMinFeatureLevel( |
| ID3D11ShaderReflection *iface, D3D_FEATURE_LEVEL *level) |
| { |
| FIXME("iface %p, level %p stub!\n", iface, level); |
| |
| return E_NOTIMPL; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_GetThreadGroupSize( |
| ID3D11ShaderReflection *iface, UINT *sizex, UINT *sizey, UINT *sizez) |
| { |
| FIXME("iface %p, sizex %p, sizey %p, sizez %p stub!\n", iface, sizex, sizey, sizez); |
| |
| return 0; |
| } |
| |
| static const struct ID3D11ShaderReflectionVtbl d3dcompiler_shader_reflection_vtbl = |
| { |
| /* IUnknown methods */ |
| d3dcompiler_shader_reflection_QueryInterface, |
| d3dcompiler_shader_reflection_AddRef, |
| d3dcompiler_shader_reflection_Release, |
| /* ID3D11ShaderReflection methods */ |
| d3dcompiler_shader_reflection_GetDesc, |
| d3dcompiler_shader_reflection_GetConstantBufferByIndex, |
| d3dcompiler_shader_reflection_GetConstantBufferByName, |
| d3dcompiler_shader_reflection_GetResourceBindingDesc, |
| d3dcompiler_shader_reflection_GetInputParameterDesc, |
| d3dcompiler_shader_reflection_GetOutputParameterDesc, |
| d3dcompiler_shader_reflection_GetPatchConstantParameterDesc, |
| d3dcompiler_shader_reflection_GetVariableByName, |
| d3dcompiler_shader_reflection_GetResourceBindingDescByName, |
| d3dcompiler_shader_reflection_GetMovInstructionCount, |
| d3dcompiler_shader_reflection_GetMovcInstructionCount, |
| d3dcompiler_shader_reflection_GetConversionInstructionCount, |
| d3dcompiler_shader_reflection_GetBitwiseInstructionCount, |
| d3dcompiler_shader_reflection_GetGSInputPrimitive, |
| d3dcompiler_shader_reflection_IsSampleFrequencyShader, |
| d3dcompiler_shader_reflection_GetNumInterfaceSlots, |
| d3dcompiler_shader_reflection_GetMinFeatureLevel, |
| d3dcompiler_shader_reflection_GetThreadGroupSize, |
| }; |
| |
| /* ID3D11ShaderReflectionConstantBuffer methods */ |
| |
| static inline struct d3dcompiler_shader_reflection_constant_buffer *impl_from_ID3D11ShaderReflectionConstantBuffer(ID3D11ShaderReflectionConstantBuffer *iface) |
| { |
| return CONTAINING_RECORD(iface, struct d3dcompiler_shader_reflection_constant_buffer, ID3D11ShaderReflectionConstantBuffer_iface); |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_constant_buffer_GetDesc( |
| ID3D11ShaderReflectionConstantBuffer *iface, D3D11_SHADER_BUFFER_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection_constant_buffer *This = impl_from_ID3D11ShaderReflectionConstantBuffer(iface); |
| |
| TRACE("iface %p, desc %p\n", iface, desc); |
| |
| if (This == &null_constant_buffer) |
| { |
| WARN("Null constant buffer specified\n"); |
| return E_FAIL; |
| } |
| |
| if (!desc) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_FAIL; |
| } |
| |
| desc->Name = This->name; |
| desc->Type = This->type; |
| desc->Variables = This->variable_count; |
| desc->Size = This->size; |
| desc->uFlags = This->flags; |
| |
| return S_OK; |
| } |
| |
| static ID3D11ShaderReflectionVariable * STDMETHODCALLTYPE d3dcompiler_shader_reflection_constant_buffer_GetVariableByIndex( |
| ID3D11ShaderReflectionConstantBuffer *iface, UINT index) |
| { |
| struct d3dcompiler_shader_reflection_constant_buffer *This = impl_from_ID3D11ShaderReflectionConstantBuffer(iface); |
| |
| TRACE("iface %p, index %u\n", iface, index); |
| |
| if (index >= This->variable_count) |
| { |
| WARN("Invalid index specified\n"); |
| return &null_variable.ID3D11ShaderReflectionVariable_iface; |
| } |
| |
| return &This->variables[index].ID3D11ShaderReflectionVariable_iface; |
| } |
| |
| static ID3D11ShaderReflectionVariable * STDMETHODCALLTYPE d3dcompiler_shader_reflection_constant_buffer_GetVariableByName( |
| ID3D11ShaderReflectionConstantBuffer *iface, const char *name) |
| { |
| struct d3dcompiler_shader_reflection_constant_buffer *This = impl_from_ID3D11ShaderReflectionConstantBuffer(iface); |
| unsigned int i; |
| |
| TRACE("iface %p, name %s\n", iface, debugstr_a(name)); |
| |
| if (!name) |
| { |
| WARN("Invalid argument specified\n"); |
| return &null_variable.ID3D11ShaderReflectionVariable_iface; |
| } |
| |
| for (i = 0; i < This->variable_count; ++i) |
| { |
| struct d3dcompiler_shader_reflection_variable *v = &This->variables[i]; |
| |
| if (!strcmp(v->name, name)) |
| { |
| TRACE("Returning ID3D11ShaderReflectionVariable %p.\n", v); |
| return &v->ID3D11ShaderReflectionVariable_iface; |
| } |
| } |
| |
| WARN("Invalid name specified\n"); |
| |
| return &null_variable.ID3D11ShaderReflectionVariable_iface; |
| } |
| |
| static const struct ID3D11ShaderReflectionConstantBufferVtbl d3dcompiler_shader_reflection_constant_buffer_vtbl = |
| { |
| /* ID3D11ShaderReflectionConstantBuffer methods */ |
| d3dcompiler_shader_reflection_constant_buffer_GetDesc, |
| d3dcompiler_shader_reflection_constant_buffer_GetVariableByIndex, |
| d3dcompiler_shader_reflection_constant_buffer_GetVariableByName, |
| }; |
| |
| /* ID3D11ShaderReflectionVariable methods */ |
| |
| static inline struct d3dcompiler_shader_reflection_variable *impl_from_ID3D11ShaderReflectionVariable(ID3D11ShaderReflectionVariable *iface) |
| { |
| return CONTAINING_RECORD(iface, struct d3dcompiler_shader_reflection_variable, ID3D11ShaderReflectionVariable_iface); |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_variable_GetDesc( |
| ID3D11ShaderReflectionVariable *iface, D3D11_SHADER_VARIABLE_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection_variable *This = impl_from_ID3D11ShaderReflectionVariable(iface); |
| |
| TRACE("iface %p, desc %p\n", iface, desc); |
| |
| if (This == &null_variable) |
| { |
| WARN("Null variable specified\n"); |
| return E_FAIL; |
| } |
| |
| if (!desc) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_FAIL; |
| } |
| |
| desc->Name = This->name; |
| desc->StartOffset = This->start_offset; |
| desc->Size = This->size; |
| desc->uFlags = This->flags; |
| desc->DefaultValue = This->default_value; |
| |
| return S_OK; |
| } |
| |
| static ID3D11ShaderReflectionType * STDMETHODCALLTYPE d3dcompiler_shader_reflection_variable_GetType( |
| ID3D11ShaderReflectionVariable *iface) |
| { |
| struct d3dcompiler_shader_reflection_variable *This = impl_from_ID3D11ShaderReflectionVariable(iface); |
| |
| TRACE("iface %p\n", iface); |
| |
| return &This->type->ID3D11ShaderReflectionType_iface; |
| } |
| |
| static ID3D11ShaderReflectionConstantBuffer * STDMETHODCALLTYPE d3dcompiler_shader_reflection_variable_GetBuffer( |
| ID3D11ShaderReflectionVariable *iface) |
| { |
| struct d3dcompiler_shader_reflection_variable *This = impl_from_ID3D11ShaderReflectionVariable(iface); |
| |
| TRACE("iface %p\n", iface); |
| |
| return &This->constant_buffer->ID3D11ShaderReflectionConstantBuffer_iface; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_variable_GetInterfaceSlot( |
| ID3D11ShaderReflectionVariable *iface, UINT index) |
| { |
| FIXME("iface %p, index %u stub!\n", iface, index); |
| |
| return 0; |
| } |
| |
| static const struct ID3D11ShaderReflectionVariableVtbl d3dcompiler_shader_reflection_variable_vtbl = |
| { |
| /* ID3D11ShaderReflectionVariable methods */ |
| d3dcompiler_shader_reflection_variable_GetDesc, |
| d3dcompiler_shader_reflection_variable_GetType, |
| d3dcompiler_shader_reflection_variable_GetBuffer, |
| d3dcompiler_shader_reflection_variable_GetInterfaceSlot, |
| }; |
| |
| /* ID3D11ShaderReflectionType methods */ |
| |
| static inline struct d3dcompiler_shader_reflection_type *impl_from_ID3D11ShaderReflectionType(ID3D11ShaderReflectionType *iface) |
| { |
| return CONTAINING_RECORD(iface, struct d3dcompiler_shader_reflection_type, ID3D11ShaderReflectionType_iface); |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetDesc( |
| ID3D11ShaderReflectionType *iface, D3D11_SHADER_TYPE_DESC *desc) |
| { |
| struct d3dcompiler_shader_reflection_type *This = impl_from_ID3D11ShaderReflectionType(iface); |
| |
| TRACE("iface %p, desc %p\n", iface, desc); |
| |
| if (This == &null_type) |
| { |
| WARN("Null type specified\n"); |
| return E_FAIL; |
| } |
| |
| if (!desc) |
| { |
| WARN("Invalid argument specified\n"); |
| return E_FAIL; |
| } |
| |
| *desc = This->desc; |
| |
| return S_OK; |
| } |
| |
| static ID3D11ShaderReflectionType * STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetMemberTypeByIndex( |
| ID3D11ShaderReflectionType *iface, UINT index) |
| { |
| struct d3dcompiler_shader_reflection_type *This = impl_from_ID3D11ShaderReflectionType(iface); |
| |
| TRACE("iface %p, index %u\n", iface, index); |
| |
| if (index >= This->desc.Members) |
| { |
| WARN("Invalid index specified\n"); |
| return &null_type.ID3D11ShaderReflectionType_iface; |
| } |
| |
| return &This->members[index].type->ID3D11ShaderReflectionType_iface; |
| } |
| |
| static ID3D11ShaderReflectionType * STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetMemberTypeByName( |
| ID3D11ShaderReflectionType *iface, const char *name) |
| { |
| struct d3dcompiler_shader_reflection_type *This = impl_from_ID3D11ShaderReflectionType(iface); |
| unsigned int i; |
| |
| TRACE("iface %p, name %s\n", iface, debugstr_a(name)); |
| |
| if (!name) |
| { |
| WARN("Invalid argument specified\n"); |
| return &null_type.ID3D11ShaderReflectionType_iface; |
| } |
| |
| for (i = 0; i < This->desc.Members; ++i) |
| { |
| struct d3dcompiler_shader_reflection_type_member *member = &This->members[i]; |
| |
| if (!strcmp(member->name, name)) |
| { |
| TRACE("Returning ID3D11ShaderReflectionType %p.\n", member->type); |
| return &member->type->ID3D11ShaderReflectionType_iface; |
| } |
| } |
| |
| WARN("Invalid name specified\n"); |
| |
| return &null_type.ID3D11ShaderReflectionType_iface; |
| } |
| |
| static const char * STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetMemberTypeName( |
| ID3D11ShaderReflectionType *iface, UINT index) |
| { |
| struct d3dcompiler_shader_reflection_type *This = impl_from_ID3D11ShaderReflectionType(iface); |
| |
| TRACE("iface %p, index %u\n", iface, index); |
| |
| if (This == &null_type) |
| { |
| WARN("Null type specified\n"); |
| return "$Invalid"; |
| } |
| |
| if (index >= This->desc.Members) |
| { |
| WARN("Invalid index specified\n"); |
| return NULL; |
| } |
| |
| return This->members[index].name; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_IsEqual( |
| ID3D11ShaderReflectionType *iface, ID3D11ShaderReflectionType *type) |
| { |
| struct d3dcompiler_shader_reflection_type *This = impl_from_ID3D11ShaderReflectionType(iface); |
| |
| TRACE("iface %p, type %p\n", iface, type); |
| |
| if (This == &null_type) |
| { |
| WARN("Null type specified\n"); |
| return E_FAIL; |
| } |
| |
| if (iface == type) |
| return S_OK; |
| |
| return S_FALSE; |
| } |
| |
| static ID3D11ShaderReflectionType * STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetSubType( |
| ID3D11ShaderReflectionType *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return NULL; |
| } |
| |
| static ID3D11ShaderReflectionType * STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetBaseClass( |
| ID3D11ShaderReflectionType *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return NULL; |
| } |
| |
| static UINT STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetNumInterfaces( |
| ID3D11ShaderReflectionType *iface) |
| { |
| FIXME("iface %p stub!\n", iface); |
| |
| return 0; |
| } |
| |
| static ID3D11ShaderReflectionType * STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_GetInterfaceByIndex( |
| ID3D11ShaderReflectionType *iface, UINT index) |
| { |
| FIXME("iface %p, index %u stub!\n", iface, index); |
| |
| return NULL; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_IsOfType( |
| ID3D11ShaderReflectionType *iface, ID3D11ShaderReflectionType *type) |
| { |
| FIXME("iface %p, type %p stub!\n", iface, type); |
| |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT STDMETHODCALLTYPE d3dcompiler_shader_reflection_type_ImplementsInterface( |
| ID3D11ShaderReflectionType *iface, ID3D11ShaderReflectionType *base) |
| { |
| FIXME("iface %p, base %p stub!\n", iface, base); |
| |
| return E_NOTIMPL; |
| } |
| |
| static const struct ID3D11ShaderReflectionTypeVtbl d3dcompiler_shader_reflection_type_vtbl = |
| { |
| /* ID3D11ShaderReflectionType methods */ |
| d3dcompiler_shader_reflection_type_GetDesc, |
| d3dcompiler_shader_reflection_type_GetMemberTypeByIndex, |
| d3dcompiler_shader_reflection_type_GetMemberTypeByName, |
| d3dcompiler_shader_reflection_type_GetMemberTypeName, |
| d3dcompiler_shader_reflection_type_IsEqual, |
| d3dcompiler_shader_reflection_type_GetSubType, |
| d3dcompiler_shader_reflection_type_GetBaseClass, |
| d3dcompiler_shader_reflection_type_GetNumInterfaces, |
| d3dcompiler_shader_reflection_type_GetInterfaceByIndex, |
| d3dcompiler_shader_reflection_type_IsOfType, |
| d3dcompiler_shader_reflection_type_ImplementsInterface, |
| }; |
| |
| static HRESULT d3dcompiler_parse_stat(struct d3dcompiler_shader_reflection *r, const char *data, DWORD data_size) |
| { |
| const char *ptr = data; |
| DWORD size = data_size >> 2; |
| |
| TRACE("Size %u\n", size); |
| |
| read_dword(&ptr, &r->instruction_count); |
| TRACE("InstructionCount: %u\n", r->instruction_count); |
| |
| read_dword(&ptr, &r->temp_register_count); |
| TRACE("TempRegisterCount: %u\n", r->temp_register_count); |
| |
| skip_dword_unknown(&ptr, 1); |
| |
| read_dword(&ptr, &r->dcl_count); |
| TRACE("DclCount: %u\n", r->dcl_count); |
| |
| read_dword(&ptr, &r->float_instruction_count); |
| TRACE("FloatInstructionCount: %u\n", r->float_instruction_count); |
| |
| read_dword(&ptr, &r->int_instruction_count); |
| TRACE("IntInstructionCount: %u\n", r->int_instruction_count); |
| |
| read_dword(&ptr, &r->uint_instruction_count); |
| TRACE("UintInstructionCount: %u\n", r->uint_instruction_count); |
| |
| read_dword(&ptr, &r->static_flow_control_count); |
| TRACE("StaticFlowControlCount: %u\n", r->static_flow_control_count); |
| |
| read_dword(&ptr, &r->dynamic_flow_control_count); |
| TRACE("DynamicFlowControlCount: %u\n", r->dynamic_flow_control_count); |
| |
| skip_dword_unknown(&ptr, 1); |
| |
| read_dword(&ptr, &r->temp_array_count); |
| TRACE("TempArrayCount: %u\n", r->temp_array_count); |
| |
| read_dword(&ptr, &r->array_instruction_count); |
| TRACE("ArrayInstructionCount: %u\n", r->array_instruction_count); |
| |
| read_dword(&ptr, &r->cut_instruction_count); |
| TRACE("CutInstructionCount: %u\n", r->cut_instruction_count); |
| |
| read_dword(&ptr, &r->emit_instruction_count); |
| TRACE("EmitInstructionCount: %u\n", r->emit_instruction_count); |
| |
| read_dword(&ptr, &r->texture_normal_instructions); |
| TRACE("TextureNormalInstructions: %u\n", r->texture_normal_instructions); |
| |
| read_dword(&ptr, &r->texture_load_instructions); |
| TRACE("TextureLoadInstructions: %u\n", r->texture_load_instructions); |
| |
| read_dword(&ptr, &r->texture_comp_instructions); |
| TRACE("TextureCompInstructions: %u\n", r->texture_comp_instructions); |
| |
| read_dword(&ptr, &r->texture_bias_instructions); |
| TRACE("TextureBiasInstructions: %u\n", r->texture_bias_instructions); |
| |
| read_dword(&ptr, &r->texture_gradient_instructions); |
| TRACE("TextureGradientInstructions: %u\n", r->texture_gradient_instructions); |
| |
| read_dword(&ptr, &r->mov_instruction_count); |
| TRACE("MovInstructionCount: %u\n", r->mov_instruction_count); |
| |
| skip_dword_unknown(&ptr, 1); |
| |
| read_dword(&ptr, &r->conversion_instruction_count); |
| TRACE("ConversionInstructionCount: %u\n", r->conversion_instruction_count); |
| |
| skip_dword_unknown(&ptr, 1); |
| |
| read_dword(&ptr, &r->input_primitive); |
| TRACE("InputPrimitive: %x\n", r->input_primitive); |
| |
| read_dword(&ptr, &r->gs_output_topology); |
| TRACE("GSOutputTopology: %x\n", r->gs_output_topology); |
| |
| read_dword(&ptr, &r->gs_max_output_vertex_count); |
| TRACE("GSMaxOutputVertexCount: %u\n", r->gs_max_output_vertex_count); |
| |
| skip_dword_unknown(&ptr, 3); |
| |
| /* dx10 stat size */ |
| if (size == 29) return S_OK; |
| |
| skip_dword_unknown(&ptr, 1); |
| |
| read_dword(&ptr, &r->c_control_points); |
| TRACE("cControlPoints: %u\n", r->c_control_points); |
| |
| read_dword(&ptr, &r->hs_output_primitive); |
| TRACE("HSOutputPrimitive: %x\n", r->hs_output_primitive); |
| |
| read_dword(&ptr, &r->hs_prtitioning); |
| TRACE("HSPartitioning: %x\n", r->hs_prtitioning); |
| |
| read_dword(&ptr, &r->tessellator_domain); |
| TRACE("TessellatorDomain: %x\n", r->tessellator_domain); |
| |
| skip_dword_unknown(&ptr, 3); |
| |
| /* dx11 stat size */ |
| if (size == 37) return S_OK; |
| |
| FIXME("Unhandled size %u\n", size); |
| |
| return E_FAIL; |
| } |
| |
| static HRESULT d3dcompiler_parse_type_members(struct d3dcompiler_shader_reflection *ref, |
| struct d3dcompiler_shader_reflection_type_member *member, const char *data, const char **ptr) |
| { |
| DWORD offset; |
| |
| read_dword(ptr, &offset); |
| if (!copy_name(data + offset, &member->name)) |
| { |
| ERR("Failed to copy name.\n"); |
| return E_OUTOFMEMORY; |
| } |
| TRACE("Member name: %s.\n", debugstr_a(member->name)); |
| |
| read_dword(ptr, &offset); |
| TRACE("Member type offset: %x\n", offset); |
| |
| member->type = get_reflection_type(ref, data, offset); |
| if (!member->type) |
| { |
| ERR("Failed to get member type\n"); |
| HeapFree(GetProcessHeap(), 0, member->name); |
| return E_FAIL; |
| } |
| |
| read_dword(ptr, &member->offset); |
| TRACE("Member offset %x\n", member->offset); |
| |
| return S_OK; |
| } |
| |
| static HRESULT d3dcompiler_parse_type(struct d3dcompiler_shader_reflection_type *type, const char *data, DWORD offset) |
| { |
| const char *ptr = data + offset; |
| DWORD temp; |
| D3D11_SHADER_TYPE_DESC *desc; |
| unsigned int i; |
| struct d3dcompiler_shader_reflection_type_member *members = NULL; |
| HRESULT hr; |
| DWORD member_offset; |
| |
| desc = &type->desc; |
| |
| read_dword(&ptr, &temp); |
| desc->Class = temp & 0xffff; |
| desc->Type = temp >> 16; |
| TRACE("Class %s, Type %s\n", debug_d3dcompiler_shader_variable_class(desc->Class), |
| debug_d3dcompiler_shader_variable_type(desc->Type)); |
| |
| read_dword(&ptr, &temp); |
| desc->Rows = temp & 0xffff; |
| desc->Columns = temp >> 16; |
| TRACE("Rows %u, Columns %u\n", desc->Rows, desc->Columns); |
| |
| read_dword(&ptr, &temp); |
| desc->Elements = temp & 0xffff; |
| desc->Members = temp >> 16; |
| TRACE("Elements %u, Members %u\n", desc->Elements, desc->Members); |
| |
| read_dword(&ptr, &member_offset); |
| TRACE("Member Offset %u\n", member_offset); |
| |
| if ((type->reflection->target & D3DCOMPILER_SHADER_TARGET_VERSION_MASK) >= 0x500) |
| skip_dword_unknown(&ptr, 4); |
| |
| if (desc->Members) |
| { |
| const char *ptr2 = data + member_offset; |
| |
| members = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*members) * desc->Members); |
| if (!members) |
| { |
| ERR("Failed to allocate type memory.\n"); |
| return E_OUTOFMEMORY; |
| } |
| |
| for (i = 0; i < desc->Members; ++i) |
| { |
| hr = d3dcompiler_parse_type_members(type->reflection, &members[i], data, &ptr2); |
| if (hr != S_OK) |
| { |
| FIXME("Failed to parse type members.\n"); |
| goto err_out; |
| } |
| } |
| } |
| |
| type->members = members; |
| |
| return S_OK; |
| |
| err_out: |
| for (i = 0; i < desc->Members; ++i) |
| { |
| free_type_member(&members[i]); |
| } |
| HeapFree(GetProcessHeap(), 0, members); |
| return hr; |
| } |
| |
| static struct d3dcompiler_shader_reflection_type *get_reflection_type(struct d3dcompiler_shader_reflection *reflection, const char *data, DWORD offset) |
| { |
| struct d3dcompiler_shader_reflection_type *type; |
| struct wine_rb_entry *entry; |
| HRESULT hr; |
| |
| entry = wine_rb_get(&reflection->types, &offset); |
| if (entry) |
| { |
| TRACE("Returning existing type.\n"); |
| return WINE_RB_ENTRY_VALUE(entry, struct d3dcompiler_shader_reflection_type, entry); |
| } |
| |
| type = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*type)); |
| if (!type) |
| return NULL; |
| |
| type->ID3D11ShaderReflectionType_iface.lpVtbl = &d3dcompiler_shader_reflection_type_vtbl; |
| type->id = offset; |
| type->reflection = reflection; |
| |
| hr = d3dcompiler_parse_type(type, data, offset); |
| if (FAILED(hr)) |
| { |
| ERR("Failed to parse type info, hr %#x.\n", hr); |
| HeapFree(GetProcessHeap(), 0, type); |
| return NULL; |
| } |
| |
| if (wine_rb_put(&reflection->types, &offset, &type->entry) == -1) |
| { |
| ERR("Failed to insert type entry.\n"); |
| HeapFree(GetProcessHeap(), 0, type); |
| return NULL; |
| } |
| |
| return type; |
| } |
| |
| static HRESULT d3dcompiler_parse_variables(struct d3dcompiler_shader_reflection_constant_buffer *cb, |
| const char *data, DWORD data_size, const char *ptr) |
| { |
| struct d3dcompiler_shader_reflection_variable *variables; |
| unsigned int i; |
| HRESULT hr; |
| |
| variables = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, cb->variable_count * sizeof(*variables)); |
| if (!variables) |
| { |
| ERR("Failed to allocate variables memory.\n"); |
| return E_OUTOFMEMORY; |
| } |
| |
| for (i = 0; i < cb->variable_count; i++) |
| { |
| struct d3dcompiler_shader_reflection_variable *v = &variables[i]; |
| DWORD offset; |
| |
| v->ID3D11ShaderReflectionVariable_iface.lpVtbl = &d3dcompiler_shader_reflection_variable_vtbl; |
| v->constant_buffer = cb; |
| |
| read_dword(&ptr, &offset); |
| if (!copy_name(data + offset, &v->name)) |
| { |
| ERR("Failed to copy name.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| TRACE("Variable name: %s.\n", debugstr_a(v->name)); |
| |
| read_dword(&ptr, &v->start_offset); |
| TRACE("Variable offset: %u\n", v->start_offset); |
| |
| read_dword(&ptr, &v->size); |
| TRACE("Variable size: %u\n", v->size); |
| |
| read_dword(&ptr, &v->flags); |
| TRACE("Variable flags: %u\n", v->flags); |
| |
| read_dword(&ptr, &offset); |
| TRACE("Variable type offset: %x\n", offset); |
| v->type = get_reflection_type(cb->reflection, data, offset); |
| if (!v->type) |
| { |
| ERR("Failed to get type.\n"); |
| hr = E_FAIL; |
| goto err_out; |
| } |
| |
| read_dword(&ptr, &offset); |
| TRACE("Variable default value offset: %x\n", offset); |
| if (!copy_value(data + offset, &v->default_value, offset ? v->size : 0)) |
| { |
| ERR("Failed to copy name.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| |
| if ((cb->reflection->target & D3DCOMPILER_SHADER_TARGET_VERSION_MASK) >= 0x500) |
| skip_dword_unknown(&ptr, 4); |
| } |
| |
| cb->variables = variables; |
| |
| return S_OK; |
| |
| err_out: |
| for (i = 0; i < cb->variable_count; i++) |
| { |
| free_variable(&variables[i]); |
| } |
| HeapFree(GetProcessHeap(), 0, variables); |
| return hr; |
| } |
| |
| 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, cbuffer_offset, resource_offset, creator_offset; |
| unsigned int i, string_data_offset, string_data_size; |
| char *string_data = NULL, *creator = NULL; |
| D3D11_SHADER_INPUT_BIND_DESC *bound_resources = NULL; |
| struct d3dcompiler_shader_reflection_constant_buffer *constant_buffers = NULL; |
| HRESULT hr; |
| |
| 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, &cbuffer_offset); |
| TRACE("Constant buffer offset: %#x\n", cbuffer_offset); |
| |
| read_dword(&ptr, &r->bound_resource_count); |
| TRACE("Bound resource count: %u\n", r->bound_resource_count); |
| |
| read_dword(&ptr, &resource_offset); |
| TRACE("Bound resource offset: %#x\n", resource_offset); |
| |
| read_dword(&ptr, &r->target); |
| TRACE("Target: %#x\n", r->target); |
| |
| read_dword(&ptr, &r->flags); |
| TRACE("Flags: %u\n", r->flags); |
| |
| read_dword(&ptr, &creator_offset); |
| TRACE("Creator at offset %#x.\n", creator_offset); |
| |
| if (!copy_name(data + creator_offset, &creator)) |
| { |
| ERR("Failed to copy name.\n"); |
| return E_OUTOFMEMORY; |
| } |
| TRACE("Creator: %s.\n", debugstr_a(creator)); |
| |
| /* todo: Parse RD11 */ |
| if ((r->target & D3DCOMPILER_SHADER_TARGET_VERSION_MASK) >= 0x500) |
| { |
| skip_dword_unknown(&ptr, 8); |
| } |
| |
| if (r->bound_resource_count) |
| { |
| /* 8 for each bind desc */ |
| string_data_offset = resource_offset + r->bound_resource_count * 8 * sizeof(DWORD); |
| string_data_size = (cbuffer_offset ? cbuffer_offset : creator_offset) - string_data_offset; |
| |
| string_data = HeapAlloc(GetProcessHeap(), 0, string_data_size); |
| if (!string_data) |
| { |
| ERR("Failed to allocate string data memory.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| memcpy(string_data, data + string_data_offset, string_data_size); |
| |
| bound_resources = HeapAlloc(GetProcessHeap(), 0, r->bound_resource_count * sizeof(*bound_resources)); |
| if (!bound_resources) |
| { |
| ERR("Failed to allocate resources memory.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| |
| ptr = data + resource_offset; |
| for (i = 0; i < r->bound_resource_count; i++) |
| { |
| D3D11_SHADER_INPUT_BIND_DESC *desc = &bound_resources[i]; |
| |
| read_dword(&ptr, &offset); |
| desc->Name = string_data + (offset - string_data_offset); |
| TRACE("Input bind Name: %s\n", debugstr_a(desc->Name)); |
| |
| read_dword(&ptr, &desc->Type); |
| TRACE("Input bind Type: %#x\n", desc->Type); |
| |
| read_dword(&ptr, &desc->ReturnType); |
| TRACE("Input bind ReturnType: %#x\n", desc->ReturnType); |
| |
| read_dword(&ptr, &desc->Dimension); |
| TRACE("Input bind Dimension: %#x\n", desc->Dimension); |
| |
| read_dword(&ptr, &desc->NumSamples); |
| TRACE("Input bind NumSamples: %u\n", desc->NumSamples); |
| |
| read_dword(&ptr, &desc->BindPoint); |
| TRACE("Input bind BindPoint: %u\n", desc->BindPoint); |
| |
| read_dword(&ptr, &desc->BindCount); |
| TRACE("Input bind BindCount: %u\n", desc->BindCount); |
| |
| read_dword(&ptr, &desc->uFlags); |
| TRACE("Input bind uFlags: %u\n", desc->uFlags); |
| } |
| } |
| |
| if (r->constant_buffer_count) |
| { |
| constant_buffers = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, r->constant_buffer_count * sizeof(*constant_buffers)); |
| if (!constant_buffers) |
| { |
| ERR("Failed to allocate constant buffer memory.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| |
| ptr = data + cbuffer_offset; |
| for (i = 0; i < r->constant_buffer_count; i++) |
| { |
| struct d3dcompiler_shader_reflection_constant_buffer *cb = &constant_buffers[i]; |
| |
| cb->ID3D11ShaderReflectionConstantBuffer_iface.lpVtbl = &d3dcompiler_shader_reflection_constant_buffer_vtbl; |
| cb->reflection = r; |
| |
| read_dword(&ptr, &offset); |
| if (!copy_name(data + offset, &cb->name)) |
| { |
| ERR("Failed to copy name.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| TRACE("Name: %s.\n", debugstr_a(cb->name)); |
| |
| read_dword(&ptr, &cb->variable_count); |
| TRACE("Variable count: %u\n", cb->variable_count); |
| |
| read_dword(&ptr, &offset); |
| TRACE("Variable offset: %x\n", offset); |
| |
| hr = d3dcompiler_parse_variables(cb, data, data_size, data + offset); |
| if (hr != S_OK) |
| { |
| FIXME("Failed to parse variables.\n"); |
| goto err_out; |
| } |
| |
| read_dword(&ptr, &cb->size); |
| TRACE("Cbuffer size: %u\n", cb->size); |
| |
| read_dword(&ptr, &cb->flags); |
| TRACE("Cbuffer flags: %u\n", cb->flags); |
| |
| read_dword(&ptr, &cb->type); |
| TRACE("Cbuffer type: %#x\n", cb->type); |
| } |
| } |
| |
| r->creator = creator; |
| r->resource_string = string_data; |
| r->bound_resources = bound_resources; |
| r->constant_buffers = constant_buffers; |
| |
| return S_OK; |
| |
| err_out: |
| for (i = 0; i < r->constant_buffer_count; ++i) |
| { |
| free_constant_buffer(&constant_buffers[i]); |
| } |
| HeapFree(GetProcessHeap(), 0, constant_buffers); |
| HeapFree(GetProcessHeap(), 0, bound_resources); |
| HeapFree(GetProcessHeap(), 0, string_data); |
| HeapFree(GetProcessHeap(), 0, creator); |
| |
| return hr; |
| } |
| |
| static HRESULT d3dcompiler_parse_signature(struct d3dcompiler_shader_signature *s, struct dxbc_section *section, DWORD target) |
| { |
| D3D11_SIGNATURE_PARAMETER_DESC *d; |
| unsigned int string_data_offset; |
| unsigned int string_data_size; |
| const char *ptr = section->data; |
| char *string_data; |
| unsigned int i; |
| DWORD count; |
| enum D3DCOMPILER_SIGNATURE_ELEMENT_SIZE element_size; |
| |
| switch (section->tag) |
| { |
| case TAG_OSG5: |
| element_size = D3DCOMPILER_SIGNATURE_ELEMENT_SIZE7; |
| break; |
| |
| case TAG_ISGN: |
| case TAG_OSGN: |
| case TAG_PCSG: |
| element_size = D3DCOMPILER_SIGNATURE_ELEMENT_SIZE6; |
| break; |
| |
| default: |
| FIXME("Unhandled section %s!\n", debugstr_an((const char *)§ion->tag, 4)); |
| element_size = D3DCOMPILER_SIGNATURE_ELEMENT_SIZE6; |
| break; |
| } |
| |
| read_dword(&ptr, &count); |
| TRACE("%u elements\n", count); |
| |
| skip_dword_unknown(&ptr, 1); |
| |
| d = HeapAlloc(GetProcessHeap(), 0, count * sizeof(*d)); |
| if (!d) |
| { |
| ERR("Failed to allocate signature memory.\n"); |
| return E_OUTOFMEMORY; |
| } |
| |
| /* 2 DWORDs for the header, element_size for each element. */ |
| string_data_offset = 2 * sizeof(DWORD) + count * element_size * sizeof(DWORD); |
| string_data_size = section->data_size - string_data_offset; |
| |
| string_data = HeapAlloc(GetProcessHeap(), 0, string_data_size); |
| if (!string_data) |
| { |
| ERR("Failed to allocate string data memory.\n"); |
| HeapFree(GetProcessHeap(), 0, d); |
| return E_OUTOFMEMORY; |
| } |
| memcpy(string_data, section->data + string_data_offset, string_data_size); |
| |
| for (i = 0; i < count; ++i) |
| { |
| UINT name_offset; |
| DWORD mask; |
| |
| if (element_size == D3DCOMPILER_SIGNATURE_ELEMENT_SIZE7) |
| { |
| read_dword(&ptr, &d[i].Stream); |
| } |
| else |
| { |
| d[i].Stream = 0; |
| } |
| |
| read_dword(&ptr, &name_offset); |
| d[i].SemanticName = string_data + (name_offset - string_data_offset); |
| read_dword(&ptr, &d[i].SemanticIndex); |
| read_dword(&ptr, &d[i].SystemValueType); |
| read_dword(&ptr, &d[i].ComponentType); |
| read_dword(&ptr, &d[i].Register); |
| read_dword(&ptr, &mask); |
| d[i].ReadWriteMask = (mask >> 8) & 0xff; |
| d[i].Mask = mask & 0xff; |
| |
| /* pixel shaders have a special handling for SystemValueType in the output signature */ |
| if (((target & D3DCOMPILER_SHADER_TARGET_SHADERTYPE_MASK) == 0xffff0000) && (section->tag == TAG_OSG5 || section->tag == TAG_OSGN)) |
| { |
| TRACE("Pixelshader output signature fixup.\n"); |
| |
| if (d[i].Register == 0xffffffff) |
| { |
| if (!strcasecmp(d[i].SemanticName, "sv_depth")) d[i].SystemValueType = D3D_NAME_DEPTH; |
| if (!strcasecmp(d[i].SemanticName, "sv_coverage")) d[i].SystemValueType = D3D_NAME_COVERAGE; |
| if (!strcasecmp(d[i].SemanticName, "sv_depthgreaterequal")) d[i].SystemValueType = D3D_NAME_DEPTH_GREATER_EQUAL; |
| if (!strcasecmp(d[i].SemanticName, "sv_depthlessequal")) d[i].SystemValueType = D3D_NAME_DEPTH_LESS_EQUAL; |
| } |
| else |
| { |
| d[i].SystemValueType = D3D_NAME_TARGET; |
| } |
| } |
| |
| TRACE("semantic: %s, semantic idx: %u, sysval_semantic %#x, " |
| "type %u, register idx: %u, use_mask %#x, input_mask %#x, stream %u\n", |
| debugstr_a(d[i].SemanticName), d[i].SemanticIndex, d[i].SystemValueType, |
| d[i].ComponentType, d[i].Register, d[i].Mask, d[i].ReadWriteMask, d[i].Stream); |
| } |
| |
| s->elements = d; |
| s->element_count = count; |
| s->string_data = string_data; |
| |
| return S_OK; |
| } |
| |
| static HRESULT d3dcompiler_parse_shdr(struct d3dcompiler_shader_reflection *r, const char *data, DWORD data_size) |
| { |
| const char *ptr = data; |
| |
| read_dword(&ptr, &r->version); |
| TRACE("Shader version: %u\n", r->version); |
| |
| /* todo: Check if anything else is needed from the shdr or shex blob. */ |
| |
| return S_OK; |
| } |
| |
| static HRESULT d3dcompiler_shader_reflection_init(struct d3dcompiler_shader_reflection *reflection, |
| const void *data, SIZE_T data_size) |
| { |
| struct dxbc src_dxbc; |
| HRESULT hr; |
| unsigned int i; |
| |
| reflection->ID3D11ShaderReflection_iface.lpVtbl = &d3dcompiler_shader_reflection_vtbl; |
| reflection->refcount = 1; |
| |
| if (wine_rb_init(&reflection->types, &d3dcompiler_shader_reflection_type_rb_functions) == -1) |
| { |
| ERR("Failed to initialize type rbtree.\n"); |
| return E_FAIL; |
| } |
| |
| hr = dxbc_parse(data, data_size, &src_dxbc); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to parse reflection\n"); |
| return hr; |
| } |
| |
| for (i = 0; i < src_dxbc.count; ++i) |
| { |
| struct dxbc_section *section = &src_dxbc.sections[i]; |
| |
| switch (section->tag) |
| { |
| 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) |
| { |
| ERR("Failed to allocate ISGN memory.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| |
| hr = d3dcompiler_parse_signature(reflection->isgn, section, reflection->target); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to parse section ISGN.\n"); |
| goto err_out; |
| } |
| break; |
| |
| case TAG_OSG5: |
| case TAG_OSGN: |
| reflection->osgn = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*reflection->osgn)); |
| if (!reflection->osgn) |
| { |
| ERR("Failed to allocate OSGN memory.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| |
| hr = d3dcompiler_parse_signature(reflection->osgn, section, reflection->target); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to parse section OSGN.\n"); |
| goto err_out; |
| } |
| break; |
| |
| case TAG_PCSG: |
| reflection->pcsg = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*reflection->pcsg)); |
| if (!reflection->pcsg) |
| { |
| ERR("Failed to allocate PCSG memory.\n"); |
| hr = E_OUTOFMEMORY; |
| goto err_out; |
| } |
| |
| hr = d3dcompiler_parse_signature(reflection->pcsg, section, reflection->target); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to parse section PCSG.\n"); |
| goto err_out; |
| } |
| break; |
| |
| case TAG_SHEX: |
| case TAG_SHDR: |
| hr = d3dcompiler_parse_shdr(reflection, section->data, section->data_size); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to parse SHDR section.\n"); |
| goto err_out; |
| } |
| break; |
| |
| case TAG_STAT: |
| hr = d3dcompiler_parse_stat(reflection, section->data, section->data_size); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to parse section STAT.\n"); |
| goto err_out; |
| } |
| break; |
| |
| default: |
| FIXME("Unhandled section %s!\n", debugstr_an((const char *)§ion->tag, 4)); |
| break; |
| } |
| } |
| |
| dxbc_destroy(&src_dxbc); |
| |
| return hr; |
| |
| err_out: |
| reflection_cleanup(reflection); |
| dxbc_destroy(&src_dxbc); |
| |
| return hr; |
| } |
| |
| HRESULT WINAPI D3DReflect(const void *data, SIZE_T data_size, REFIID riid, void **reflector) |
| { |
| struct d3dcompiler_shader_reflection *object; |
| HRESULT hr; |
| const DWORD *temp = data; |
| |
| TRACE("data %p, data_size %lu, riid %s, blob %p\n", data, data_size, debugstr_guid(riid), reflector); |
| |
| if (!data || data_size < 32) |
| { |
| WARN("Invalid argument supplied.\n"); |
| return D3DERR_INVALIDCALL; |
| } |
| |
| if (temp[6] != data_size) |
| { |
| WARN("Wrong size supplied.\n"); |
| return E_FAIL; |
| } |
| |
| if (!IsEqualGUID(riid, &IID_ID3D11ShaderReflection)) |
| { |
| WARN("Wrong riid %s, accept only %s!\n", debugstr_guid(riid), debugstr_guid(&IID_ID3D11ShaderReflection)); |
| return E_NOINTERFACE; |
| } |
| |
| object = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*object)); |
| if (!object) |
| return E_OUTOFMEMORY; |
| |
| hr = d3dcompiler_shader_reflection_init(object, data, data_size); |
| if (FAILED(hr)) |
| { |
| WARN("Failed to initialize shader reflection\n"); |
| HeapFree(GetProcessHeap(), 0, object); |
| return hr; |
| } |
| |
| *reflector = object; |
| |
| TRACE("Created ID3D11ShaderReflection %p\n", object); |
| |
| return S_OK; |
| } |