d3drm: Handle texture associated with the material when loading a mesh.
[wine] / dlls / windowscodecs / metadatahandler.c
1 /*
2  * Copyright 2012 Vincent Povirk for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include "config.h"
20
21 #include <stdarg.h>
22
23 #define COBJMACROS
24 #define NONAMELESSUNION
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "objbase.h"
29 #include "wincodec.h"
30 #include "wincodecsdk.h"
31
32 #include "wincodecs_private.h"
33
34 #include "wine/debug.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(wincodecs);
37
38 typedef struct MetadataHandler {
39     IWICMetadataWriter IWICMetadataWriter_iface;
40     LONG ref;
41     IWICPersistStream IWICPersistStream_iface;
42     const MetadataHandlerVtbl *vtable;
43     MetadataItem *items;
44     DWORD item_count;
45     CRITICAL_SECTION lock;
46 } MetadataHandler;
47
48 static inline MetadataHandler *impl_from_IWICMetadataWriter(IWICMetadataWriter *iface)
49 {
50     return CONTAINING_RECORD(iface, MetadataHandler, IWICMetadataWriter_iface);
51 }
52
53 static inline MetadataHandler *impl_from_IWICPersistStream(IWICPersistStream *iface)
54 {
55     return CONTAINING_RECORD(iface, MetadataHandler, IWICPersistStream_iface);
56 }
57
58 static void MetadataHandler_FreeItems(MetadataHandler *This)
59 {
60     int i;
61
62     for (i=0; i<This->item_count; i++)
63     {
64         PropVariantClear(&This->items[i].schema);
65         PropVariantClear(&This->items[i].id);
66         PropVariantClear(&This->items[i].value);
67     }
68
69     HeapFree(GetProcessHeap(), 0, This->items);
70 }
71
72 static HRESULT MetadataHandlerEnum_Create(MetadataHandler *parent, DWORD index,
73     IWICEnumMetadataItem **ppIEnumMetadataItem);
74
75 static HRESULT WINAPI MetadataHandler_QueryInterface(IWICMetadataWriter *iface, REFIID iid,
76     void **ppv)
77 {
78     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
79     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
80
81     if (!ppv) return E_INVALIDARG;
82
83     if (IsEqualIID(&IID_IUnknown, iid) ||
84         IsEqualIID(&IID_IWICMetadataReader, iid) ||
85         (IsEqualIID(&IID_IWICMetadataWriter, iid) && This->vtable->is_writer))
86     {
87         *ppv = &This->IWICMetadataWriter_iface;
88     }
89     else if (IsEqualIID(&IID_IPersist, iid) ||
90              IsEqualIID(&IID_IPersistStream, iid) ||
91              IsEqualIID(&IID_IWICPersistStream, iid))
92     {
93         *ppv = &This->IWICPersistStream_iface;
94     }
95     else
96     {
97         *ppv = NULL;
98         return E_NOINTERFACE;
99     }
100
101     IUnknown_AddRef((IUnknown*)*ppv);
102     return S_OK;
103 }
104
105 static ULONG WINAPI MetadataHandler_AddRef(IWICMetadataWriter *iface)
106 {
107     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
108     ULONG ref = InterlockedIncrement(&This->ref);
109
110     TRACE("(%p) refcount=%u\n", iface, ref);
111
112     return ref;
113 }
114
115 static ULONG WINAPI MetadataHandler_Release(IWICMetadataWriter *iface)
116 {
117     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
118     ULONG ref = InterlockedDecrement(&This->ref);
119
120     TRACE("(%p) refcount=%u\n", iface, ref);
121
122     if (ref == 0)
123     {
124         MetadataHandler_FreeItems(This);
125         This->lock.DebugInfo->Spare[0] = 0;
126         DeleteCriticalSection(&This->lock);
127         HeapFree(GetProcessHeap(), 0, This);
128     }
129
130     return ref;
131 }
132
133 static HRESULT WINAPI MetadataHandler_GetMetadataFormat(IWICMetadataWriter *iface,
134     GUID *pguidMetadataFormat)
135 {
136     if (!pguidMetadataFormat) return E_INVALIDARG;
137
138     FIXME("(%p,%s): stub\n", iface, debugstr_guid(pguidMetadataFormat));
139     return E_NOTIMPL;
140 }
141
142 static HRESULT WINAPI MetadataHandler_GetMetadataHandlerInfo(IWICMetadataWriter *iface,
143     IWICMetadataHandlerInfo **ppIHandler)
144 {
145     FIXME("(%p,%p): stub\n", iface, ppIHandler);
146     return E_NOTIMPL;
147 }
148
149 static HRESULT WINAPI MetadataHandler_GetCount(IWICMetadataWriter *iface,
150     UINT *pcCount)
151 {
152     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
153
154     TRACE("%p,%p\n", iface, pcCount);
155
156     if (!pcCount) return E_INVALIDARG;
157
158     *pcCount = This->item_count;
159     return S_OK;
160 }
161
162 static HRESULT WINAPI MetadataHandler_GetValueByIndex(IWICMetadataWriter *iface,
163     UINT nIndex, PROPVARIANT *pvarSchema, PROPVARIANT *pvarId, PROPVARIANT *pvarValue)
164 {
165     FIXME("(%p,%u,%p,%p,%p): stub\n", iface, nIndex, pvarSchema, pvarId, pvarValue);
166     return E_NOTIMPL;
167 }
168
169 static HRESULT WINAPI MetadataHandler_GetValue(IWICMetadataWriter *iface,
170     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, PROPVARIANT *pvarValue)
171 {
172     FIXME("(%p,%p,%p,%p): stub\n", iface, pvarSchema, pvarId, pvarValue);
173     return E_NOTIMPL;
174 }
175
176 static HRESULT WINAPI MetadataHandler_GetEnumerator(IWICMetadataWriter *iface,
177     IWICEnumMetadataItem **ppIEnumMetadata)
178 {
179     MetadataHandler *This = impl_from_IWICMetadataWriter(iface);
180     TRACE("(%p,%p)\n", iface, ppIEnumMetadata);
181     return MetadataHandlerEnum_Create(This, 0, ppIEnumMetadata);
182 }
183
184 static HRESULT WINAPI MetadataHandler_SetValue(IWICMetadataWriter *iface,
185     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, const PROPVARIANT *pvarValue)
186 {
187     FIXME("(%p,%p,%p,%p): stub\n", iface, pvarSchema, pvarId, pvarValue);
188     return E_NOTIMPL;
189 }
190
191 static HRESULT WINAPI MetadataHandler_SetValueByIndex(IWICMetadataWriter *iface,
192     UINT nIndex, const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId, const PROPVARIANT *pvarValue)
193 {
194     FIXME("(%p,%u,%p,%p,%p): stub\n", iface, nIndex, pvarSchema, pvarId, pvarValue);
195     return E_NOTIMPL;
196 }
197
198 static HRESULT WINAPI MetadataHandler_RemoveValue(IWICMetadataWriter *iface,
199     const PROPVARIANT *pvarSchema, const PROPVARIANT *pvarId)
200 {
201     FIXME("(%p,%p,%p): stub\n", iface, pvarSchema, pvarId);
202     return E_NOTIMPL;
203 }
204
205 static HRESULT WINAPI MetadataHandler_RemoveValueByIndex(IWICMetadataWriter *iface,
206     UINT nIndex)
207 {
208     FIXME("(%p,%u): stub\n", iface, nIndex);
209     return E_NOTIMPL;
210 }
211
212 static const IWICMetadataWriterVtbl MetadataHandler_Vtbl = {
213     MetadataHandler_QueryInterface,
214     MetadataHandler_AddRef,
215     MetadataHandler_Release,
216     MetadataHandler_GetMetadataFormat,
217     MetadataHandler_GetMetadataHandlerInfo,
218     MetadataHandler_GetCount,
219     MetadataHandler_GetValueByIndex,
220     MetadataHandler_GetValue,
221     MetadataHandler_GetEnumerator,
222     MetadataHandler_SetValue,
223     MetadataHandler_SetValueByIndex,
224     MetadataHandler_RemoveValue,
225     MetadataHandler_RemoveValueByIndex
226 };
227
228 static HRESULT WINAPI MetadataHandler_PersistStream_QueryInterface(IWICPersistStream *iface,
229     REFIID iid, void **ppv)
230 {
231     MetadataHandler *This = impl_from_IWICPersistStream(iface);
232     return IWICMetadataWriter_QueryInterface(&This->IWICMetadataWriter_iface, iid, ppv);
233 }
234
235 static ULONG WINAPI MetadataHandler_PersistStream_AddRef(IWICPersistStream *iface)
236 {
237     MetadataHandler *This = impl_from_IWICPersistStream(iface);
238     return IWICMetadataWriter_AddRef(&This->IWICMetadataWriter_iface);
239 }
240
241 static ULONG WINAPI MetadataHandler_PersistStream_Release(IWICPersistStream *iface)
242 {
243     MetadataHandler *This = impl_from_IWICPersistStream(iface);
244     return IWICMetadataWriter_Release(&This->IWICMetadataWriter_iface);
245 }
246
247 static HRESULT WINAPI MetadataHandler_GetClassID(IWICPersistStream *iface,
248     CLSID *pClassID)
249 {
250     FIXME("(%p,%p): stub\n", iface, pClassID);
251     return E_NOTIMPL;
252 }
253
254 static HRESULT WINAPI MetadataHandler_IsDirty(IWICPersistStream *iface)
255 {
256     FIXME("(%p): stub\n", iface);
257     return E_NOTIMPL;
258 }
259
260 static HRESULT WINAPI MetadataHandler_Load(IWICPersistStream *iface,
261     IStream *pStm)
262 {
263     FIXME("(%p,%p): stub\n", iface, pStm);
264     return E_NOTIMPL;
265 }
266
267 static HRESULT WINAPI MetadataHandler_Save(IWICPersistStream *iface,
268     IStream *pStm, BOOL fClearDirty)
269 {
270     FIXME("(%p,%p,%i): stub\n", iface, pStm, fClearDirty);
271     return E_NOTIMPL;
272 }
273
274 static HRESULT WINAPI MetadataHandler_GetSizeMax(IWICPersistStream *iface,
275     ULARGE_INTEGER *pcbSize)
276 {
277     FIXME("(%p,%p): stub\n", iface, pcbSize);
278     return E_NOTIMPL;
279 }
280
281 static HRESULT WINAPI MetadataHandler_LoadEx(IWICPersistStream *iface,
282     IStream *pIStream, const GUID *pguidPreferredVendor, DWORD dwPersistOptions)
283 {
284     MetadataHandler *This = impl_from_IWICPersistStream(iface);
285     HRESULT hr;
286     MetadataItem *new_items=NULL;
287     DWORD item_count=0;
288
289     TRACE("(%p,%p,%s,%x)\n", iface, pIStream, debugstr_guid(pguidPreferredVendor), dwPersistOptions);
290
291     EnterCriticalSection(&This->lock);
292
293     hr = This->vtable->fnLoad(pIStream, pguidPreferredVendor, dwPersistOptions,
294         &new_items, &item_count);
295
296     if (SUCCEEDED(hr))
297     {
298         MetadataHandler_FreeItems(This);
299         This->items = new_items;
300         This->item_count = item_count;
301     }
302
303     LeaveCriticalSection(&This->lock);
304
305     return hr;
306 }
307
308 static HRESULT WINAPI MetadataHandler_SaveEx(IWICPersistStream *iface,
309     IStream *pIStream, DWORD dwPersistOptions, BOOL fClearDirty)
310 {
311     FIXME("(%p,%p,%x,%i): stub\n", iface, pIStream, dwPersistOptions, fClearDirty);
312     return E_NOTIMPL;
313 }
314
315 static const IWICPersistStreamVtbl MetadataHandler_PersistStream_Vtbl = {
316     MetadataHandler_PersistStream_QueryInterface,
317     MetadataHandler_PersistStream_AddRef,
318     MetadataHandler_PersistStream_Release,
319     MetadataHandler_GetClassID,
320     MetadataHandler_IsDirty,
321     MetadataHandler_Load,
322     MetadataHandler_Save,
323     MetadataHandler_GetSizeMax,
324     MetadataHandler_LoadEx,
325     MetadataHandler_SaveEx
326 };
327
328 static HRESULT MetadataReader_Create(const MetadataHandlerVtbl *vtable, IUnknown *pUnkOuter, REFIID iid, void** ppv)
329 {
330     MetadataHandler *This;
331     HRESULT hr;
332
333     TRACE("%s\n", debugstr_guid(vtable->clsid));
334
335     *ppv = NULL;
336
337     if (pUnkOuter) return CLASS_E_NOAGGREGATION;
338
339     This = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataHandler));
340     if (!This) return E_OUTOFMEMORY;
341
342     This->IWICMetadataWriter_iface.lpVtbl = &MetadataHandler_Vtbl;
343     This->IWICPersistStream_iface.lpVtbl = &MetadataHandler_PersistStream_Vtbl;
344     This->ref = 1;
345     This->vtable = vtable;
346     This->items = NULL;
347     This->item_count = 0;
348
349     InitializeCriticalSection(&This->lock);
350     This->lock.DebugInfo->Spare[0] = (DWORD_PTR)(__FILE__ ": MetadataHandler.lock");
351
352     hr = IWICMetadataWriter_QueryInterface(&This->IWICMetadataWriter_iface, iid, ppv);
353
354     IWICMetadataWriter_Release(&This->IWICMetadataWriter_iface);
355
356     return hr;
357 }
358
359 typedef struct MetadataHandlerEnum {
360     IWICEnumMetadataItem IWICEnumMetadataItem_iface;
361     LONG ref;
362     MetadataHandler *parent;
363     DWORD index;
364 } MetadataHandlerEnum;
365
366 static inline MetadataHandlerEnum *impl_from_IWICEnumMetadataItem(IWICEnumMetadataItem *iface)
367 {
368     return CONTAINING_RECORD(iface, MetadataHandlerEnum, IWICEnumMetadataItem_iface);
369 }
370
371 static HRESULT WINAPI MetadataHandlerEnum_QueryInterface(IWICEnumMetadataItem *iface, REFIID iid,
372     void **ppv)
373 {
374     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
375     TRACE("(%p,%s,%p)\n", iface, debugstr_guid(iid), ppv);
376
377     if (!ppv) return E_INVALIDARG;
378
379     if (IsEqualIID(&IID_IUnknown, iid) ||
380         IsEqualIID(&IID_IWICEnumMetadataItem, iid))
381     {
382         *ppv = &This->IWICEnumMetadataItem_iface;
383     }
384     else
385     {
386         *ppv = NULL;
387         return E_NOINTERFACE;
388     }
389
390     IUnknown_AddRef((IUnknown*)*ppv);
391     return S_OK;
392 }
393
394 static ULONG WINAPI MetadataHandlerEnum_AddRef(IWICEnumMetadataItem *iface)
395 {
396     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
397     ULONG ref = InterlockedIncrement(&This->ref);
398
399     TRACE("(%p) refcount=%u\n", iface, ref);
400
401     return ref;
402 }
403
404 static ULONG WINAPI MetadataHandlerEnum_Release(IWICEnumMetadataItem *iface)
405 {
406     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
407     ULONG ref = InterlockedDecrement(&This->ref);
408
409     TRACE("(%p) refcount=%u\n", iface, ref);
410
411     if (ref == 0)
412     {
413         IWICMetadataWriter_Release(&This->parent->IWICMetadataWriter_iface);
414         HeapFree(GetProcessHeap(), 0, This);
415     }
416
417     return ref;
418 }
419
420 static HRESULT WINAPI MetadataHandlerEnum_Next(IWICEnumMetadataItem *iface,
421     ULONG celt, PROPVARIANT *rgeltSchema, PROPVARIANT *rgeltId,
422     PROPVARIANT *rgeltValue, ULONG *pceltFetched)
423 {
424     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
425     ULONG new_index;
426     HRESULT hr=S_FALSE;
427     int i;
428
429     TRACE("(%p,%i)\n", iface, celt);
430
431     EnterCriticalSection(&This->parent->lock);
432
433     if (This->index >= This->parent->item_count)
434     {
435         *pceltFetched = 0;
436         LeaveCriticalSection(&This->parent->lock);
437         return S_FALSE;
438     }
439
440     new_index = min(This->parent->item_count, This->index + celt);
441     *pceltFetched = new_index - This->index;
442
443     if (rgeltSchema)
444     {
445         for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
446             hr = PropVariantCopy(&rgeltSchema[i], &This->parent->items[i+This->index].schema);
447     }
448
449     for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
450         hr = PropVariantCopy(&rgeltId[i], &This->parent->items[i+This->index].id);
451
452     if (rgeltValue)
453     {
454         for (i=0; SUCCEEDED(hr) && i < *pceltFetched; i++)
455             hr = PropVariantCopy(&rgeltValue[i], &This->parent->items[i+This->index].value);
456     }
457
458     if (SUCCEEDED(hr))
459     {
460         This->index = new_index;
461     }
462
463     LeaveCriticalSection(&This->parent->lock);
464
465     return hr;
466 }
467
468 static HRESULT WINAPI MetadataHandlerEnum_Skip(IWICEnumMetadataItem *iface,
469     ULONG celt)
470 {
471     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
472
473     EnterCriticalSection(&This->parent->lock);
474
475     This->index += celt;
476
477     LeaveCriticalSection(&This->parent->lock);
478
479     return S_OK;
480 }
481
482 static HRESULT WINAPI MetadataHandlerEnum_Reset(IWICEnumMetadataItem *iface)
483 {
484     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
485
486     EnterCriticalSection(&This->parent->lock);
487
488     This->index = 0;
489
490     LeaveCriticalSection(&This->parent->lock);
491
492     return S_OK;
493 }
494
495 static HRESULT WINAPI MetadataHandlerEnum_Clone(IWICEnumMetadataItem *iface,
496     IWICEnumMetadataItem **ppIEnumMetadataItem)
497 {
498     MetadataHandlerEnum *This = impl_from_IWICEnumMetadataItem(iface);
499     HRESULT hr;
500
501     EnterCriticalSection(&This->parent->lock);
502
503     hr = MetadataHandlerEnum_Create(This->parent, This->index, ppIEnumMetadataItem);
504
505     LeaveCriticalSection(&This->parent->lock);
506
507     return hr;
508 }
509
510 static const IWICEnumMetadataItemVtbl MetadataHandlerEnum_Vtbl = {
511     MetadataHandlerEnum_QueryInterface,
512     MetadataHandlerEnum_AddRef,
513     MetadataHandlerEnum_Release,
514     MetadataHandlerEnum_Next,
515     MetadataHandlerEnum_Skip,
516     MetadataHandlerEnum_Reset,
517     MetadataHandlerEnum_Clone
518 };
519
520 static HRESULT MetadataHandlerEnum_Create(MetadataHandler *parent, DWORD index,
521     IWICEnumMetadataItem **ppIEnumMetadataItem)
522 {
523     MetadataHandlerEnum *This;
524
525     if (!ppIEnumMetadataItem) return E_INVALIDARG;
526
527     *ppIEnumMetadataItem = NULL;
528
529     This = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataHandlerEnum));
530     if (!This) return E_OUTOFMEMORY;
531
532     IWICMetadataWriter_AddRef(&parent->IWICMetadataWriter_iface);
533
534     This->IWICEnumMetadataItem_iface.lpVtbl = &MetadataHandlerEnum_Vtbl;
535     This->ref = 1;
536     This->parent = parent;
537     This->index = index;
538
539     *ppIEnumMetadataItem = &This->IWICEnumMetadataItem_iface;
540
541     return S_OK;
542 }
543
544 static HRESULT LoadUnknownMetadata(IStream *input, const GUID *preferred_vendor,
545     DWORD persist_options, MetadataItem **items, DWORD *item_count)
546 {
547     HRESULT hr;
548     MetadataItem *result;
549     STATSTG stat;
550     BYTE *data;
551     ULONG bytesread;
552
553     TRACE("\n");
554
555     hr = IStream_Stat(input, &stat, STATFLAG_NONAME);
556     if (FAILED(hr))
557         return hr;
558
559     data = HeapAlloc(GetProcessHeap(), 0, stat.cbSize.QuadPart);
560     if (!data) return E_OUTOFMEMORY;
561
562     hr = IStream_Read(input, data, stat.cbSize.QuadPart, &bytesread);
563     if (FAILED(hr))
564     {
565         HeapFree(GetProcessHeap(), 0, data);
566         return hr;
567     }
568
569     result = HeapAlloc(GetProcessHeap(), 0, sizeof(MetadataItem));
570     if (!result)
571     {
572         HeapFree(GetProcessHeap(), 0, data);
573         return E_OUTOFMEMORY;
574     }
575
576     PropVariantInit(&result[0].schema);
577     PropVariantInit(&result[0].id);
578     PropVariantInit(&result[0].value);
579
580     result[0].value.vt = VT_BLOB;
581     result[0].value.u.blob.cbSize = bytesread;
582     result[0].value.u.blob.pBlobData = data;
583
584     *items = result;
585     *item_count = 1;
586
587     return S_OK;
588 }
589
590 static const MetadataHandlerVtbl UnknownMetadataReader_Vtbl = {
591     0,
592     &CLSID_WICUnknownMetadataReader,
593     LoadUnknownMetadata
594 };
595
596 HRESULT UnknownMetadataReader_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void** ppv)
597 {
598     return MetadataReader_Create(&UnknownMetadataReader_Vtbl, pUnkOuter, iid, ppv);
599 }
600
601 static HRESULT LoadIfdMetadata(IStream *input, const GUID *preferred_vendor,
602     DWORD persist_options, MetadataItem **items, DWORD *item_count)
603 {
604     FIXME("stub\n");
605
606     *items = NULL;
607     *item_count = 0;
608
609     return S_OK;
610 }
611
612 static const MetadataHandlerVtbl IfdMetadataReader_Vtbl = {
613     0,
614     &CLSID_WICIfdMetadataReader,
615     LoadIfdMetadata
616 };
617
618 HRESULT IfdMetadataReader_CreateInstance(IUnknown *pUnkOuter, REFIID iid, void **ppv)
619 {
620     return MetadataReader_Create(&IfdMetadataReader_Vtbl, pUnkOuter, iid, ppv);
621 }