vbscript: Added support for title and type arguments of MsgBox.
[wine.git] / dlls / d3d10core / shader.c
blob007dfc2458bf617ac88f3bc4b702b21a2fc61e69
1 /*
2 * Copyright 2009 Henri Verbeet for CodeWeavers
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.
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.
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
20 #include "config.h"
21 #include "wine/port.h"
23 #include "d3d10core_private.h"
25 WINE_DEFAULT_DEBUG_CHANNEL(d3d10core);
27 static HRESULT shdr_handler(const char *data, DWORD data_size, DWORD tag, void *ctx)
29 struct d3d10_shader_info *shader_info = ctx;
30 HRESULT hr;
32 switch(tag)
34 case TAG_OSGN:
35 hr = shader_parse_signature(data, data_size, shader_info->output_signature);
36 if (FAILED(hr)) return hr;
37 break;
39 case TAG_SHDR:
40 shader_info->shader_code = (const DWORD *)data;
41 break;
43 default:
44 FIXME("Unhandled chunk %s\n", debugstr_an((const char *)&tag, 4));
45 break;
48 return S_OK;
51 static HRESULT shader_extract_from_dxbc(const void *dxbc, SIZE_T dxbc_length, struct d3d10_shader_info *shader_info)
53 HRESULT hr;
55 shader_info->shader_code = NULL;
56 memset(shader_info->output_signature, 0, sizeof(*shader_info->output_signature));
58 hr = parse_dxbc(dxbc, dxbc_length, shdr_handler, shader_info);
59 if (!shader_info->shader_code) hr = E_INVALIDARG;
61 if (FAILED(hr))
63 ERR("Failed to parse shader, hr %#x\n", hr);
64 shader_free_signature(shader_info->output_signature);
67 return hr;
70 HRESULT shader_parse_signature(const char *data, DWORD data_size, struct wined3d_shader_signature *s)
72 struct wined3d_shader_signature_element *e;
73 unsigned int string_data_offset;
74 unsigned int string_data_size;
75 const char *ptr = data;
76 char *string_data;
77 unsigned int i;
78 DWORD count;
80 read_dword(&ptr, &count);
81 TRACE("%u elements\n", count);
83 skip_dword_unknown(&ptr, 1);
85 e = HeapAlloc(GetProcessHeap(), 0, count * sizeof(*e));
86 if (!e)
88 ERR("Failed to allocate input signature memory.\n");
89 return E_OUTOFMEMORY;
92 /* 2 DWORDs for the header, 6 for each element. */
93 string_data_offset = 2 * sizeof(DWORD) + count * 6 * sizeof(DWORD);
94 string_data_size = data_size - string_data_offset;
95 string_data = HeapAlloc(GetProcessHeap(), 0, string_data_size);
96 if (!string_data)
98 ERR("Failed to allocate string data memory.\n");
99 HeapFree(GetProcessHeap(), 0, e);
100 return E_OUTOFMEMORY;
102 memcpy(string_data, data + string_data_offset, string_data_size);
104 for (i = 0; i < count; ++i)
106 UINT name_offset;
108 read_dword(&ptr, &name_offset);
109 e[i].semantic_name = string_data + (name_offset - string_data_offset);
110 read_dword(&ptr, &e[i].semantic_idx);
111 read_dword(&ptr, &e[i].sysval_semantic);
112 read_dword(&ptr, &e[i].component_type);
113 read_dword(&ptr, &e[i].register_idx);
114 read_dword(&ptr, &e[i].mask);
116 TRACE("semantic: %s, semantic idx: %u, sysval_semantic %#x, "
117 "type %u, register idx: %u, use_mask %#x, input_mask %#x\n",
118 debugstr_a(e[i].semantic_name), e[i].semantic_idx, e[i].sysval_semantic,
119 e[i].component_type, e[i].register_idx, (e[i].mask >> 8) & 0xff, e[i].mask & 0xff);
122 s->elements = e;
123 s->element_count = count;
124 s->string_data = string_data;
126 return S_OK;
129 void shader_free_signature(struct wined3d_shader_signature *s)
131 HeapFree(GetProcessHeap(), 0, s->string_data);
132 HeapFree(GetProcessHeap(), 0, s->elements);
135 static inline struct d3d10_vertex_shader *impl_from_ID3D10VertexShader(ID3D10VertexShader *iface)
137 return CONTAINING_RECORD(iface, struct d3d10_vertex_shader, ID3D10VertexShader_iface);
140 /* IUnknown methods */
142 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_QueryInterface(ID3D10VertexShader *iface,
143 REFIID riid, void **object)
145 TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object);
147 if (IsEqualGUID(riid, &IID_ID3D10VertexShader)
148 || IsEqualGUID(riid, &IID_ID3D10DeviceChild)
149 || IsEqualGUID(riid, &IID_IUnknown))
151 IUnknown_AddRef(iface);
152 *object = iface;
153 return S_OK;
156 WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid));
158 *object = NULL;
159 return E_NOINTERFACE;
162 static ULONG STDMETHODCALLTYPE d3d10_vertex_shader_AddRef(ID3D10VertexShader *iface)
164 struct d3d10_vertex_shader *This = impl_from_ID3D10VertexShader(iface);
165 ULONG refcount = InterlockedIncrement(&This->refcount);
167 TRACE("%p increasing refcount to %u\n", This, refcount);
169 if (refcount == 1)
171 ID3D10Device1_AddRef(This->device);
172 wined3d_shader_incref(This->wined3d_shader);
175 return refcount;
178 static ULONG STDMETHODCALLTYPE d3d10_vertex_shader_Release(ID3D10VertexShader *iface)
180 struct d3d10_vertex_shader *This = impl_from_ID3D10VertexShader(iface);
181 ULONG refcount = InterlockedDecrement(&This->refcount);
183 TRACE("%p decreasing refcount to %u\n", This, refcount);
185 if (!refcount)
187 ID3D10Device1 *device = This->device;
189 wined3d_shader_decref(This->wined3d_shader);
190 /* Release the device last, it may cause the wined3d device to be
191 * destroyed. */
192 ID3D10Device1_Release(device);
195 return refcount;
198 /* ID3D10DeviceChild methods */
200 static void STDMETHODCALLTYPE d3d10_vertex_shader_GetDevice(ID3D10VertexShader *iface, ID3D10Device **device)
202 struct d3d10_vertex_shader *shader = impl_from_ID3D10VertexShader(iface);
204 TRACE("iface %p, device %p.\n", iface, device);
206 *device = (ID3D10Device *)shader->device;
207 ID3D10Device_AddRef(*device);
210 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_GetPrivateData(ID3D10VertexShader *iface,
211 REFGUID guid, UINT *data_size, void *data)
213 FIXME("iface %p, guid %s, data_size %p, data %p stub!\n",
214 iface, debugstr_guid(guid), data_size, data);
216 return E_NOTIMPL;
219 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_SetPrivateData(ID3D10VertexShader *iface,
220 REFGUID guid, UINT data_size, const void *data)
222 FIXME("iface %p, guid %s, data_size %u, data %p stub!\n",
223 iface, debugstr_guid(guid), data_size, data);
225 return E_NOTIMPL;
228 static HRESULT STDMETHODCALLTYPE d3d10_vertex_shader_SetPrivateDataInterface(ID3D10VertexShader *iface,
229 REFGUID guid, const IUnknown *data)
231 FIXME("iface %p, guid %s, data %p stub!\n", iface, debugstr_guid(guid), data);
233 return E_NOTIMPL;
236 static const struct ID3D10VertexShaderVtbl d3d10_vertex_shader_vtbl =
238 /* IUnknown methods */
239 d3d10_vertex_shader_QueryInterface,
240 d3d10_vertex_shader_AddRef,
241 d3d10_vertex_shader_Release,
242 /* ID3D10DeviceChild methods */
243 d3d10_vertex_shader_GetDevice,
244 d3d10_vertex_shader_GetPrivateData,
245 d3d10_vertex_shader_SetPrivateData,
246 d3d10_vertex_shader_SetPrivateDataInterface,
249 static void STDMETHODCALLTYPE d3d10_vertex_shader_wined3d_object_destroyed(void *parent)
251 struct d3d10_vertex_shader *shader = parent;
252 shader_free_signature(&shader->output_signature);
253 HeapFree(GetProcessHeap(), 0, shader);
256 static const struct wined3d_parent_ops d3d10_vertex_shader_wined3d_parent_ops =
258 d3d10_vertex_shader_wined3d_object_destroyed,
261 HRESULT d3d10_vertex_shader_init(struct d3d10_vertex_shader *shader, struct d3d10_device *device,
262 const void *byte_code, SIZE_T byte_code_length)
264 struct d3d10_shader_info shader_info;
265 HRESULT hr;
267 shader->ID3D10VertexShader_iface.lpVtbl = &d3d10_vertex_shader_vtbl;
268 shader->refcount = 1;
270 shader_info.output_signature = &shader->output_signature;
271 hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info);
272 if (FAILED(hr))
274 ERR("Failed to extract shader, hr %#x.\n", hr);
275 return hr;
278 hr = wined3d_shader_create_vs(device->wined3d_device, shader_info.shader_code,
279 &shader->output_signature, shader, &d3d10_vertex_shader_wined3d_parent_ops, &shader->wined3d_shader, 4);
280 if (FAILED(hr))
282 WARN("Failed to create wined3d vertex shader, hr %#x.\n", hr);
283 shader_free_signature(&shader->output_signature);
284 hr = E_INVALIDARG;
285 return hr;
288 shader->device = &device->ID3D10Device1_iface;
289 ID3D10Device1_AddRef(shader->device);
291 return S_OK;
294 struct d3d10_vertex_shader *unsafe_impl_from_ID3D10VertexShader(ID3D10VertexShader *iface)
296 if (!iface)
297 return NULL;
298 assert(iface->lpVtbl == &d3d10_vertex_shader_vtbl);
300 return impl_from_ID3D10VertexShader(iface);
303 static inline struct d3d10_geometry_shader *impl_from_ID3D10GeometryShader(ID3D10GeometryShader *iface)
305 return CONTAINING_RECORD(iface, struct d3d10_geometry_shader, ID3D10GeometryShader_iface);
308 /* IUnknown methods */
310 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_QueryInterface(ID3D10GeometryShader *iface,
311 REFIID riid, void **object)
313 TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object);
315 if (IsEqualGUID(riid, &IID_ID3D10GeometryShader)
316 || IsEqualGUID(riid, &IID_ID3D10DeviceChild)
317 || IsEqualGUID(riid, &IID_IUnknown))
319 IUnknown_AddRef(iface);
320 *object = iface;
321 return S_OK;
324 WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid));
326 *object = NULL;
327 return E_NOINTERFACE;
330 static ULONG STDMETHODCALLTYPE d3d10_geometry_shader_AddRef(ID3D10GeometryShader *iface)
332 struct d3d10_geometry_shader *This = impl_from_ID3D10GeometryShader(iface);
333 ULONG refcount = InterlockedIncrement(&This->refcount);
335 TRACE("%p increasing refcount to %u\n", This, refcount);
337 return refcount;
340 static ULONG STDMETHODCALLTYPE d3d10_geometry_shader_Release(ID3D10GeometryShader *iface)
342 struct d3d10_geometry_shader *This = impl_from_ID3D10GeometryShader(iface);
343 ULONG refcount = InterlockedDecrement(&This->refcount);
345 TRACE("%p decreasing refcount to %u\n", This, refcount);
347 if (!refcount)
348 wined3d_shader_decref(This->wined3d_shader);
350 return refcount;
353 /* ID3D10DeviceChild methods */
355 static void STDMETHODCALLTYPE d3d10_geometry_shader_GetDevice(ID3D10GeometryShader *iface, ID3D10Device **device)
357 FIXME("iface %p, device %p stub!\n", iface, device);
360 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_GetPrivateData(ID3D10GeometryShader *iface,
361 REFGUID guid, UINT *data_size, void *data)
363 FIXME("iface %p, guid %s, data_size %p, data %p stub!\n",
364 iface, debugstr_guid(guid), data_size, data);
366 return E_NOTIMPL;
369 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_SetPrivateData(ID3D10GeometryShader *iface,
370 REFGUID guid, UINT data_size, const void *data)
372 FIXME("iface %p, guid %s, data_size %u, data %p stub!\n",
373 iface, debugstr_guid(guid), data_size, data);
375 return E_NOTIMPL;
378 static HRESULT STDMETHODCALLTYPE d3d10_geometry_shader_SetPrivateDataInterface(ID3D10GeometryShader *iface,
379 REFGUID guid, const IUnknown *data)
381 FIXME("iface %p, guid %s, data %p stub!\n", iface, debugstr_guid(guid), data);
383 return E_NOTIMPL;
386 static const struct ID3D10GeometryShaderVtbl d3d10_geometry_shader_vtbl =
388 /* IUnknown methods */
389 d3d10_geometry_shader_QueryInterface,
390 d3d10_geometry_shader_AddRef,
391 d3d10_geometry_shader_Release,
392 /* ID3D10DeviceChild methods */
393 d3d10_geometry_shader_GetDevice,
394 d3d10_geometry_shader_GetPrivateData,
395 d3d10_geometry_shader_SetPrivateData,
396 d3d10_geometry_shader_SetPrivateDataInterface,
399 static void STDMETHODCALLTYPE d3d10_geometry_shader_wined3d_object_destroyed(void *parent)
401 struct d3d10_geometry_shader *shader = parent;
402 shader_free_signature(&shader->output_signature);
403 HeapFree(GetProcessHeap(), 0, shader);
406 static const struct wined3d_parent_ops d3d10_geometry_shader_wined3d_parent_ops =
408 d3d10_geometry_shader_wined3d_object_destroyed,
411 HRESULT d3d10_geometry_shader_init(struct d3d10_geometry_shader *shader, struct d3d10_device *device,
412 const void *byte_code, SIZE_T byte_code_length)
414 struct d3d10_shader_info shader_info;
415 HRESULT hr;
417 shader->ID3D10GeometryShader_iface.lpVtbl = &d3d10_geometry_shader_vtbl;
418 shader->refcount = 1;
420 shader_info.output_signature = &shader->output_signature;
421 hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info);
422 if (FAILED(hr))
424 ERR("Failed to extract shader, hr %#x.\n", hr);
425 return hr;
428 hr = wined3d_shader_create_gs(device->wined3d_device, shader_info.shader_code,
429 &shader->output_signature, shader, &d3d10_geometry_shader_wined3d_parent_ops, &shader->wined3d_shader, 4);
430 if (FAILED(hr))
432 WARN("Failed to create wined3d geometry shader, hr %#x.\n", hr);
433 shader_free_signature(&shader->output_signature);
434 hr = E_INVALIDARG;
435 return hr;
438 return S_OK;
441 struct d3d10_geometry_shader *unsafe_impl_from_ID3D10GeometryShader(ID3D10GeometryShader *iface)
443 if (!iface)
444 return NULL;
445 assert(iface->lpVtbl == &d3d10_geometry_shader_vtbl);
447 return impl_from_ID3D10GeometryShader(iface);
450 static inline struct d3d10_pixel_shader *impl_from_ID3D10PixelShader(ID3D10PixelShader *iface)
452 return CONTAINING_RECORD(iface, struct d3d10_pixel_shader, ID3D10PixelShader_iface);
455 /* IUnknown methods */
457 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_QueryInterface(ID3D10PixelShader *iface,
458 REFIID riid, void **object)
460 TRACE("iface %p, riid %s, object %p\n", iface, debugstr_guid(riid), object);
462 if (IsEqualGUID(riid, &IID_ID3D10PixelShader)
463 || IsEqualGUID(riid, &IID_ID3D10DeviceChild)
464 || IsEqualGUID(riid, &IID_IUnknown))
466 IUnknown_AddRef(iface);
467 *object = iface;
468 return S_OK;
471 WARN("%s not implemented, returning E_NOINTERFACE\n", debugstr_guid(riid));
473 *object = NULL;
474 return E_NOINTERFACE;
477 static ULONG STDMETHODCALLTYPE d3d10_pixel_shader_AddRef(ID3D10PixelShader *iface)
479 struct d3d10_pixel_shader *This = impl_from_ID3D10PixelShader(iface);
480 ULONG refcount = InterlockedIncrement(&This->refcount);
482 TRACE("%p increasing refcount to %u\n", This, refcount);
484 if (refcount == 1)
486 ID3D10Device1_AddRef(This->device);
487 wined3d_shader_incref(This->wined3d_shader);
490 return refcount;
493 static ULONG STDMETHODCALLTYPE d3d10_pixel_shader_Release(ID3D10PixelShader *iface)
495 struct d3d10_pixel_shader *This = impl_from_ID3D10PixelShader(iface);
496 ULONG refcount = InterlockedDecrement(&This->refcount);
498 TRACE("%p decreasing refcount to %u\n", This, refcount);
500 if (!refcount)
502 ID3D10Device1 *device = This->device;
504 wined3d_shader_decref(This->wined3d_shader);
505 /* Release the device last, it may cause the wined3d device to be
506 * destroyed. */
507 ID3D10Device1_Release(device);
510 return refcount;
513 /* ID3D10DeviceChild methods */
515 static void STDMETHODCALLTYPE d3d10_pixel_shader_GetDevice(ID3D10PixelShader *iface, ID3D10Device **device)
517 struct d3d10_pixel_shader *shader = impl_from_ID3D10PixelShader(iface);
519 TRACE("iface %p, device %p.\n", iface, device);
521 *device = (ID3D10Device *)shader->device;
522 ID3D10Device_AddRef(*device);
525 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_GetPrivateData(ID3D10PixelShader *iface,
526 REFGUID guid, UINT *data_size, void *data)
528 FIXME("iface %p, guid %s, data_size %p, data %p stub!\n",
529 iface, debugstr_guid(guid), data_size, data);
531 return E_NOTIMPL;
534 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_SetPrivateData(ID3D10PixelShader *iface,
535 REFGUID guid, UINT data_size, const void *data)
537 FIXME("iface %p, guid %s, data_size %u, data %p stub!\n",
538 iface, debugstr_guid(guid), data_size, data);
540 return E_NOTIMPL;
543 static HRESULT STDMETHODCALLTYPE d3d10_pixel_shader_SetPrivateDataInterface(ID3D10PixelShader *iface,
544 REFGUID guid, const IUnknown *data)
546 FIXME("iface %p, guid %s, data %p stub!\n", iface, debugstr_guid(guid), data);
548 return E_NOTIMPL;
551 static const struct ID3D10PixelShaderVtbl d3d10_pixel_shader_vtbl =
553 /* IUnknown methods */
554 d3d10_pixel_shader_QueryInterface,
555 d3d10_pixel_shader_AddRef,
556 d3d10_pixel_shader_Release,
557 /* ID3D10DeviceChild methods */
558 d3d10_pixel_shader_GetDevice,
559 d3d10_pixel_shader_GetPrivateData,
560 d3d10_pixel_shader_SetPrivateData,
561 d3d10_pixel_shader_SetPrivateDataInterface,
564 static void STDMETHODCALLTYPE d3d10_pixel_shader_wined3d_object_destroyed(void *parent)
566 struct d3d10_pixel_shader *shader = parent;
567 shader_free_signature(&shader->output_signature);
568 HeapFree(GetProcessHeap(), 0, shader);
571 static const struct wined3d_parent_ops d3d10_pixel_shader_wined3d_parent_ops =
573 d3d10_pixel_shader_wined3d_object_destroyed,
576 HRESULT d3d10_pixel_shader_init(struct d3d10_pixel_shader *shader, struct d3d10_device *device,
577 const void *byte_code, SIZE_T byte_code_length)
579 struct d3d10_shader_info shader_info;
580 HRESULT hr;
582 shader->ID3D10PixelShader_iface.lpVtbl = &d3d10_pixel_shader_vtbl;
583 shader->refcount = 1;
585 shader_info.output_signature = &shader->output_signature;
586 hr = shader_extract_from_dxbc(byte_code, byte_code_length, &shader_info);
587 if (FAILED(hr))
589 ERR("Failed to extract shader, hr %#x.\n", hr);
590 return hr;
593 hr = wined3d_shader_create_ps(device->wined3d_device, shader_info.shader_code,
594 &shader->output_signature, shader, &d3d10_pixel_shader_wined3d_parent_ops, &shader->wined3d_shader, 4);
595 if (FAILED(hr))
597 WARN("Failed to create wined3d pixel shader, hr %#x.\n", hr);
598 shader_free_signature(&shader->output_signature);
599 hr = E_INVALIDARG;
600 return hr;
603 shader->device = &device->ID3D10Device1_iface;
604 ID3D10Device1_AddRef(shader->device);
606 return S_OK;
609 struct d3d10_pixel_shader *unsafe_impl_from_ID3D10PixelShader(ID3D10PixelShader *iface)
611 if (!iface)
612 return NULL;
613 assert(iface->lpVtbl == &d3d10_pixel_shader_vtbl);
615 return impl_from_ID3D10PixelShader(iface);