?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
mf: Avoid leaking nodes in IMFTopoLoader_Load.
[wine.git] / dlls / mf / topology.c
blob8d64c10b5c7f5a65c998a73ac623f483d2cc1305
1 /*
2 * Copyright 2017 Nikolay Sivov
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
20 #include <stdarg.h>
21
22 #define COBJMACROS
23
24 #include "windef.h"
25 #include "winbase.h"
26
27 #undef INITGUID
28 #include <guiddef.h>
29 #include "mfidl.h"
30
31 #include "wine/debug.h"
32
33 #include "mf_private.h"
34
35 WINE_DEFAULT_DEBUG_CHANNEL(mfplat);
36
37 static LONG next_node_id;
38 static TOPOID next_topology_id;
39
40 struct node_stream
41 {
42 IMFMediaType *preferred_type;
43 struct topology_node *connection;
44 DWORD connection_stream;
45 };
46
47 struct node_streams
48 {
49 struct node_stream *streams;
50 size_t size;
51 size_t count;
52 };
53
54 struct topology_node
55 {
56 IMFTopologyNode IMFTopologyNode_iface;
57 LONG refcount;
58 IMFAttributes *attributes;
59 MF_TOPOLOGY_TYPE node_type;
60 TOPOID id;
61 IUnknown *object;
62 IMFMediaType *input_type; /* Only for tee nodes. */
63 struct node_streams inputs;
64 struct node_streams outputs;
65 CRITICAL_SECTION cs;
66 };
67
68 struct topology
69 {
70 IMFTopology IMFTopology_iface;
71 LONG refcount;
72 IMFAttributes *attributes;
73 struct
74 {
75 struct topology_node **nodes;
76 size_t size;
77 size_t count;
78 } nodes;
79 TOPOID id;
80 };
81
82 struct topology_loader
83 {
84 IMFTopoLoader IMFTopoLoader_iface;
85 LONG refcount;
86 };
87
88 struct seq_source
89 {
90 IMFSequencerSource IMFSequencerSource_iface;
91 IMFMediaSourceTopologyProvider IMFMediaSourceTopologyProvider_iface;
92 LONG refcount;
93 };
94
95 static inline struct topology *impl_from_IMFTopology(IMFTopology *iface)
96 {
97 return CONTAINING_RECORD(iface, struct topology, IMFTopology_iface);
98 }
99
100 static struct topology_node *impl_from_IMFTopologyNode(IMFTopologyNode *iface)
101 {
102 return CONTAINING_RECORD(iface, struct topology_node, IMFTopologyNode_iface);
103 }
104
105 static const IMFTopologyNodeVtbl topologynodevtbl;
106
107 static struct topology_node *unsafe_impl_from_IMFTopologyNode(IMFTopologyNode *iface)
108 {
109 if (!iface || iface->lpVtbl != &topologynodevtbl)
110 return NULL;
111 return impl_from_IMFTopologyNode(iface);
112 }
113
114 static HRESULT create_topology_node(MF_TOPOLOGY_TYPE node_type, struct topology_node **node);
115 static HRESULT topology_node_connect_output(struct topology_node *node, DWORD output_index,
116 struct topology_node *connection, DWORD input_index);
117
118 static struct topology *unsafe_impl_from_IMFTopology(IMFTopology *iface);
119
120 static struct topology_loader *impl_from_IMFTopoLoader(IMFTopoLoader *iface)
121 {
122 return CONTAINING_RECORD(iface, struct topology_loader, IMFTopoLoader_iface);
123 }
124
125 static struct seq_source *impl_from_IMFSequencerSource(IMFSequencerSource *iface)
126 {
127 return CONTAINING_RECORD(iface, struct seq_source, IMFSequencerSource_iface);
128 }
129
130 static struct seq_source *impl_from_IMFMediaSourceTopologyProvider(IMFMediaSourceTopologyProvider *iface)
131 {
132 return CONTAINING_RECORD(iface, struct seq_source, IMFMediaSourceTopologyProvider_iface);
133 }
134
135 static HRESULT topology_node_reserve_streams(struct node_streams *streams, DWORD index)
136 {
137 if (!mf_array_reserve((void **)&streams->streams, &streams->size, index + 1, sizeof(*streams->streams)))
138 return E_OUTOFMEMORY;
139
140 if (index >= streams->count)
141 {
142 memset(&streams->streams[streams->count], 0, (index - streams->count + 1) * sizeof(*streams->streams));
143 streams->count = index + 1;
144 }
145
146 return S_OK;
147 }
148
149 static HRESULT WINAPI topology_QueryInterface(IMFTopology *iface, REFIID riid, void **out)
150 {
151 struct topology *topology = impl_from_IMFTopology(iface);
152
153 TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
154
155 if (IsEqualIID(riid, &IID_IMFTopology) ||
156 IsEqualIID(riid, &IID_IMFAttributes) ||
157 IsEqualIID(riid, &IID_IUnknown))
158 {
159 *out = &topology->IMFTopology_iface;
160 }
161 else
162 {
163 FIXME("(%s, %p)\n", debugstr_guid(riid), out);
164 *out = NULL;
165 return E_NOINTERFACE;
166 }
167
168 IUnknown_AddRef((IUnknown*)*out);
169 return S_OK;
170 }
171
172 static ULONG WINAPI topology_AddRef(IMFTopology *iface)
173 {
174 struct topology *topology = impl_from_IMFTopology(iface);
175 ULONG refcount = InterlockedIncrement(&topology->refcount);
176
177 TRACE("%p, refcount %lu.\n", iface, refcount);
178
179 return refcount;
180 }
181
182 static HRESULT topology_node_disconnect_output(struct topology_node *node, DWORD output_index)
183 {
184 struct topology_node *connection = NULL;
185 struct node_stream *stream;
186 DWORD connection_stream;
187 HRESULT hr = S_OK;
188
189 EnterCriticalSection(&node->cs);
190
191 if (output_index < node->outputs.count)
192 {
193 stream = &node->outputs.streams[output_index];
194
195 if (stream->connection)
196 {
197 connection = stream->connection;
198 connection_stream = stream->connection_stream;
199 stream->connection = NULL;
200 stream->connection_stream = 0;
201 }
202 else
203 hr = MF_E_NOT_FOUND;
204 }
205 else
206 hr = E_INVALIDARG;
207
208 LeaveCriticalSection(&node->cs);
209
210 if (connection)
211 {
212 EnterCriticalSection(&connection->cs);
213
214 if (connection_stream < connection->inputs.count)
215 {
216 stream = &connection->inputs.streams[connection_stream];
217
218 if (stream->connection)
219 {
220 stream->connection = NULL;
221 stream->connection_stream = 0;
222 }
223 }
224
225 LeaveCriticalSection(&connection->cs);
226
227 IMFTopologyNode_Release(&connection->IMFTopologyNode_iface);
228 IMFTopologyNode_Release(&node->IMFTopologyNode_iface);
229 }
230
231 return hr;
232 }
233
234 static void topology_node_disconnect(struct topology_node *node)
235 {
236 struct node_stream *stream;
237 size_t i;
238
239 for (i = 0; i < node->outputs.count; ++i)
240 topology_node_disconnect_output(node, i);
241
242 for (i = 0; i < node->inputs.count; ++i)
243 {
244 stream = &node->inputs.streams[i];
245 if (stream->connection)
246 topology_node_disconnect_output(stream->connection, stream->connection_stream);
247 }
248 }
249
250 static void topology_clear(struct topology *topology)
251 {
252 size_t i;
253
254 for (i = 0; i < topology->nodes.count; ++i)
255 {
256 topology_node_disconnect(topology->nodes.nodes[i]);
257 IMFTopologyNode_Release(&topology->nodes.nodes[i]->IMFTopologyNode_iface);
258 }
259 free(topology->nodes.nodes);
260 topology->nodes.nodes = NULL;
261 topology->nodes.count = 0;
262 topology->nodes.size = 0;
263 }
264
265 static ULONG WINAPI topology_Release(IMFTopology *iface)
266 {
267 struct topology *topology = impl_from_IMFTopology(iface);
268 ULONG refcount = InterlockedDecrement(&topology->refcount);
269
270 TRACE("%p, refcount %lu.\n", iface, refcount);
271
272 if (!refcount)
273 {
274 if (topology->attributes)
275 IMFAttributes_Release(topology->attributes);
276 topology_clear(topology);
277 free(topology);
278 }
279
280 return refcount;
281 }
282
283 static HRESULT WINAPI topology_GetItem(IMFTopology *iface, REFGUID key, PROPVARIANT *value)
284 {
285 struct topology *topology = impl_from_IMFTopology(iface);
286
287 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
288
289 return IMFAttributes_GetItem(topology->attributes, key, value);
290 }
291
292 static HRESULT WINAPI topology_GetItemType(IMFTopology *iface, REFGUID key, MF_ATTRIBUTE_TYPE *type)
293 {
294 struct topology *topology = impl_from_IMFTopology(iface);
295
296 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), type);
297
298 return IMFAttributes_GetItemType(topology->attributes, key, type);
299 }
300
301 static HRESULT WINAPI topology_CompareItem(IMFTopology *iface, REFGUID key, REFPROPVARIANT value, BOOL *result)
302 {
303 struct topology *topology = impl_from_IMFTopology(iface);
304
305 TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), value, result);
306
307 return IMFAttributes_CompareItem(topology->attributes, key, value, result);
308 }
309
310 static HRESULT WINAPI topology_Compare(IMFTopology *iface, IMFAttributes *theirs, MF_ATTRIBUTES_MATCH_TYPE type,
311 BOOL *result)
312 {
313 struct topology *topology = impl_from_IMFTopology(iface);
314
315 TRACE("%p, %p, %d, %p.\n", iface, theirs, type, result);
316
317 return IMFAttributes_Compare(topology->attributes, theirs, type, result);
318 }
319
320 static HRESULT WINAPI topology_GetUINT32(IMFTopology *iface, REFGUID key, UINT32 *value)
321 {
322 struct topology *topology = impl_from_IMFTopology(iface);
323
324 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
325
326 return IMFAttributes_GetUINT32(topology->attributes, key, value);
327 }
328
329 static HRESULT WINAPI topology_GetUINT64(IMFTopology *iface, REFGUID key, UINT64 *value)
330 {
331 struct topology *topology = impl_from_IMFTopology(iface);
332
333 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
334
335 return IMFAttributes_GetUINT64(topology->attributes, key, value);
336 }
337
338 static HRESULT WINAPI topology_GetDouble(IMFTopology *iface, REFGUID key, double *value)
339 {
340 struct topology *topology = impl_from_IMFTopology(iface);
341
342 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
343
344 return IMFAttributes_GetDouble(topology->attributes, key, value);
345 }
346
347 static HRESULT WINAPI topology_GetGUID(IMFTopology *iface, REFGUID key, GUID *value)
348 {
349 struct topology *topology = impl_from_IMFTopology(iface);
350
351 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
352
353 return IMFAttributes_GetGUID(topology->attributes, key, value);
354 }
355
356 static HRESULT WINAPI topology_GetStringLength(IMFTopology *iface, REFGUID key, UINT32 *length)
357 {
358 struct topology *topology = impl_from_IMFTopology(iface);
359
360 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), length);
361
362 return IMFAttributes_GetStringLength(topology->attributes, key, length);
363 }
364
365 static HRESULT WINAPI topology_GetString(IMFTopology *iface, REFGUID key, WCHAR *value,
366 UINT32 size, UINT32 *length)
367 {
368 struct topology *topology = impl_from_IMFTopology(iface);
369
370 TRACE("%p, %s, %p, %d, %p.\n", iface, debugstr_guid(key), value, size, length);
371
372 return IMFAttributes_GetString(topology->attributes, key, value, size, length);
373 }
374
375 static HRESULT WINAPI topology_GetAllocatedString(IMFTopology *iface, REFGUID key,
376 WCHAR **value, UINT32 *length)
377 {
378 struct topology *topology = impl_from_IMFTopology(iface);
379
380 TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), value, length);
381
382 return IMFAttributes_GetAllocatedString(topology->attributes, key, value, length);
383 }
384
385 static HRESULT WINAPI topology_GetBlobSize(IMFTopology *iface, REFGUID key, UINT32 *size)
386 {
387 struct topology *topology = impl_from_IMFTopology(iface);
388
389 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), size);
390
391 return IMFAttributes_GetBlobSize(topology->attributes, key, size);
392 }
393
394 static HRESULT WINAPI topology_GetBlob(IMFTopology *iface, REFGUID key, UINT8 *buf,
395 UINT32 bufsize, UINT32 *blobsize)
396 {
397 struct topology *topology = impl_from_IMFTopology(iface);
398
399 TRACE("%p, %s, %p, %d, %p.\n", iface, debugstr_guid(key), buf, bufsize, blobsize);
400
401 return IMFAttributes_GetBlob(topology->attributes, key, buf, bufsize, blobsize);
402 }
403
404 static HRESULT WINAPI topology_GetAllocatedBlob(IMFTopology *iface, REFGUID key, UINT8 **buf, UINT32 *size)
405 {
406 struct topology *topology = impl_from_IMFTopology(iface);
407
408 TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), buf, size);
409
410 return IMFAttributes_GetAllocatedBlob(topology->attributes, key, buf, size);
411 }
412
413 static HRESULT WINAPI topology_GetUnknown(IMFTopology *iface, REFGUID key, REFIID riid, void **ppv)
414 {
415 struct topology *topology = impl_from_IMFTopology(iface);
416
417 TRACE("%p, %s, %s, %p.\n", iface, debugstr_guid(key), debugstr_guid(riid), ppv);
418
419 return IMFAttributes_GetUnknown(topology->attributes, key, riid, ppv);
420 }
421
422 static HRESULT WINAPI topology_SetItem(IMFTopology *iface, REFGUID key, REFPROPVARIANT value)
423 {
424 struct topology *topology = impl_from_IMFTopology(iface);
425
426 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
427
428 return IMFAttributes_SetItem(topology->attributes, key, value);
429 }
430
431 static HRESULT WINAPI topology_DeleteItem(IMFTopology *iface, REFGUID key)
432 {
433 struct topology *topology = impl_from_IMFTopology(iface);
434
435 TRACE("%p, %s.\n", topology, debugstr_guid(key));
436
437 return IMFAttributes_DeleteItem(topology->attributes, key);
438 }
439
440 static HRESULT WINAPI topology_DeleteAllItems(IMFTopology *iface)
441 {
442 struct topology *topology = impl_from_IMFTopology(iface);
443
444 TRACE("%p.\n", iface);
445
446 return IMFAttributes_DeleteAllItems(topology->attributes);
447 }
448
449 static HRESULT WINAPI topology_SetUINT32(IMFTopology *iface, REFGUID key, UINT32 value)
450 {
451 struct topology *topology = impl_from_IMFTopology(iface);
452
453 TRACE("%p, %s, %d.\n", iface, debugstr_guid(key), value);
454
455 return IMFAttributes_SetUINT32(topology->attributes, key, value);
456 }
457
458 static HRESULT WINAPI topology_SetUINT64(IMFTopology *iface, REFGUID key, UINT64 value)
459 {
460 struct topology *topology = impl_from_IMFTopology(iface);
461
462 TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), wine_dbgstr_longlong(value));
463
464 return IMFAttributes_SetUINT64(topology->attributes, key, value);
465 }
466
467 static HRESULT WINAPI topology_SetDouble(IMFTopology *iface, REFGUID key, double value)
468 {
469 struct topology *topology = impl_from_IMFTopology(iface);
470
471 TRACE("%p, %s, %f.\n", iface, debugstr_guid(key), value);
472
473 return IMFAttributes_SetDouble(topology->attributes, key, value);
474 }
475
476 static HRESULT WINAPI topology_SetGUID(IMFTopology *iface, REFGUID key, REFGUID value)
477 {
478 struct topology *topology = impl_from_IMFTopology(iface);
479
480 TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), debugstr_guid(value));
481
482 return IMFAttributes_SetGUID(topology->attributes, key, value);
483 }
484
485 static HRESULT WINAPI topology_SetString(IMFTopology *iface, REFGUID key, const WCHAR *value)
486 {
487 struct topology *topology = impl_from_IMFTopology(iface);
488
489 TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), debugstr_w(value));
490
491 return IMFAttributes_SetString(topology->attributes, key, value);
492 }
493
494 static HRESULT WINAPI topology_SetBlob(IMFTopology *iface, REFGUID key, const UINT8 *buf, UINT32 size)
495 {
496 struct topology *topology = impl_from_IMFTopology(iface);
497
498 TRACE("%p, %s, %p, %d.\n", iface, debugstr_guid(key), buf, size);
499
500 return IMFAttributes_SetBlob(topology->attributes, key, buf, size);
501 }
502
503 static HRESULT WINAPI topology_SetUnknown(IMFTopology *iface, REFGUID key, IUnknown *unknown)
504 {
505 struct topology *topology = impl_from_IMFTopology(iface);
506
507 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), unknown);
508
509 return IMFAttributes_SetUnknown(topology->attributes, key, unknown);
510 }
511
512 static HRESULT WINAPI topology_LockStore(IMFTopology *iface)
513 {
514 struct topology *topology = impl_from_IMFTopology(iface);
515
516 TRACE("%p.\n", iface);
517
518 return IMFAttributes_LockStore(topology->attributes);
519 }
520
521 static HRESULT WINAPI topology_UnlockStore(IMFTopology *iface)
522 {
523 struct topology *topology = impl_from_IMFTopology(iface);
524
525 TRACE("%p.\n", iface);
526
527 return IMFAttributes_UnlockStore(topology->attributes);
528 }
529
530 static HRESULT WINAPI topology_GetCount(IMFTopology *iface, UINT32 *count)
531 {
532 struct topology *topology = impl_from_IMFTopology(iface);
533
534 TRACE("%p, %p.\n", iface, count);
535
536 return IMFAttributes_GetCount(topology->attributes, count);
537 }
538
539 static HRESULT WINAPI topology_GetItemByIndex(IMFTopology *iface, UINT32 index, GUID *key, PROPVARIANT *value)
540 {
541 struct topology *topology = impl_from_IMFTopology(iface);
542
543 TRACE("%p, %u, %p, %p.\n", iface, index, key, value);
544
545 return IMFAttributes_GetItemByIndex(topology->attributes, index, key, value);
546 }
547
548 static HRESULT WINAPI topology_CopyAllItems(IMFTopology *iface, IMFAttributes *dest)
549 {
550 struct topology *topology = impl_from_IMFTopology(iface);
551
552 TRACE("%p, %p.\n", iface, dest);
553
554 return IMFAttributes_CopyAllItems(topology->attributes, dest);
555 }
556
557 static HRESULT WINAPI topology_GetTopologyID(IMFTopology *iface, TOPOID *id)
558 {
559 struct topology *topology = impl_from_IMFTopology(iface);
560
561 TRACE("%p, %p.\n", iface, id);
562
563 if (!id)
564 return E_POINTER;
565
566 *id = topology->id;
567
568 return S_OK;
569 }
570
571 static HRESULT topology_get_node_by_id(const struct topology *topology, TOPOID id, struct topology_node **node)
572 {
573 size_t i = 0;
574
575 for (i = 0; i < topology->nodes.count; ++i)
576 {
577 if (topology->nodes.nodes[i]->id == id)
578 {
579 *node = topology->nodes.nodes[i];
580 return S_OK;
581 }
582 }
583
584 return MF_E_NOT_FOUND;
585 }
586
587 static HRESULT topology_add_node(struct topology *topology, struct topology_node *node)
588 {
589 struct topology_node *match;
590
591 if (!node)
592 return E_POINTER;
593
594 if (SUCCEEDED(topology_get_node_by_id(topology, node->id, &match)))
595 return E_INVALIDARG;
596
597 if (!mf_array_reserve((void **)&topology->nodes.nodes, &topology->nodes.size, topology->nodes.count + 1,
598 sizeof(*topology->nodes.nodes)))
599 {
600 return E_OUTOFMEMORY;
601 }
602
603 topology->nodes.nodes[topology->nodes.count++] = node;
604 IMFTopologyNode_AddRef(&node->IMFTopologyNode_iface);
605
606 return S_OK;
607 }
608
609 static HRESULT WINAPI topology_AddNode(IMFTopology *iface, IMFTopologyNode *node_iface)
610 {
611 struct topology *topology = impl_from_IMFTopology(iface);
612 struct topology_node *node = unsafe_impl_from_IMFTopologyNode(node_iface);
613
614 TRACE("%p, %p.\n", iface, node_iface);
615
616 return topology_add_node(topology, node);
617 }
618
619 static HRESULT WINAPI topology_RemoveNode(IMFTopology *iface, IMFTopologyNode *node)
620 {
621 struct topology *topology = impl_from_IMFTopology(iface);
622 size_t i, count;
623
624 TRACE("%p, %p.\n", iface, node);
625
626 for (i = 0; i < topology->nodes.count; ++i)
627 {
628 if (&topology->nodes.nodes[i]->IMFTopologyNode_iface == node)
629 {
630 topology_node_disconnect(topology->nodes.nodes[i]);
631 IMFTopologyNode_Release(&topology->nodes.nodes[i]->IMFTopologyNode_iface);
632 count = topology->nodes.count - i - 1;
633 if (count)
634 {
635 memmove(&topology->nodes.nodes[i], &topology->nodes.nodes[i + 1],
636 count * sizeof(*topology->nodes.nodes));
637 }
638 topology->nodes.count--;
639 return S_OK;
640 }
641 }
642
643 return E_INVALIDARG;
644 }
645
646 static HRESULT WINAPI topology_GetNodeCount(IMFTopology *iface, WORD *count)
647 {
648 struct topology *topology = impl_from_IMFTopology(iface);
649
650 TRACE("%p, %p.\n", iface, count);
651
652 if (!count)
653 return E_POINTER;
654
655 *count = topology->nodes.count;
656
657 return S_OK;
658 }
659
660 static HRESULT WINAPI topology_GetNode(IMFTopology *iface, WORD index, IMFTopologyNode **node)
661 {
662 struct topology *topology = impl_from_IMFTopology(iface);
663
664 TRACE("%p, %u, %p.\n", iface, index, node);
665
666 if (!node)
667 return E_POINTER;
668
669 if (index >= topology->nodes.count)
670 return MF_E_INVALIDINDEX;
671
672 *node = &topology->nodes.nodes[index]->IMFTopologyNode_iface;
673 IMFTopologyNode_AddRef(*node);
674
675 return S_OK;
676 }
677
678 static HRESULT WINAPI topology_Clear(IMFTopology *iface)
679 {
680 struct topology *topology = impl_from_IMFTopology(iface);
681
682 TRACE("%p.\n", iface);
683
684 topology_clear(topology);
685 return S_OK;
686 }
687
688 static HRESULT WINAPI topology_CloneFrom(IMFTopology *iface, IMFTopology *src)
689 {
690 struct topology *topology = impl_from_IMFTopology(iface);
691 struct topology *src_topology = unsafe_impl_from_IMFTopology(src);
692 struct topology_node *node;
693 size_t i, j;
694 HRESULT hr;
695
696 TRACE("%p, %p.\n", iface, src);
697
698 topology_clear(topology);
699
700 /* Clone nodes. */
701 for (i = 0; i < src_topology->nodes.count; ++i)
702 {
703 if (FAILED(hr = create_topology_node(src_topology->nodes.nodes[i]->node_type, &node)))
704 {
705 WARN("Failed to create a node, hr %#lx.\n", hr);
706 break;
707 }
708
709 if (SUCCEEDED(hr = IMFTopologyNode_CloneFrom(&node->IMFTopologyNode_iface,
710 &src_topology->nodes.nodes[i]->IMFTopologyNode_iface)))
711 {
712 topology_add_node(topology, node);
713 }
714
715 IMFTopologyNode_Release(&node->IMFTopologyNode_iface);
716 }
717
718 /* Clone connections. */
719 for (i = 0; i < src_topology->nodes.count; ++i)
720 {
721 const struct node_streams *outputs = &src_topology->nodes.nodes[i]->outputs;
722
723 for (j = 0; j < outputs->count; ++j)
724 {
725 DWORD input_index = outputs->streams[j].connection_stream;
726 TOPOID id = outputs->streams[j].connection->id;
727
728 /* Skip node lookup in destination topology, assuming same node order. */
729 if (SUCCEEDED(hr = topology_get_node_by_id(topology, id, &node)))
730 topology_node_connect_output(topology->nodes.nodes[i], j, node, input_index);
731 }
732 }
733
734 /* Copy attributes and id. */
735 hr = IMFTopology_CopyAllItems(src, (IMFAttributes *)&topology->IMFTopology_iface);
736 if (SUCCEEDED(hr))
737 topology->id = src_topology->id;
738
739 return S_OK;
740 }
741
742 static HRESULT WINAPI topology_GetNodeByID(IMFTopology *iface, TOPOID id, IMFTopologyNode **ret)
743 {
744 struct topology *topology = impl_from_IMFTopology(iface);
745 struct topology_node *node;
746 HRESULT hr;
747
748 TRACE("%p, %p.\n", iface, ret);
749
750 if (SUCCEEDED(hr = topology_get_node_by_id(topology, id, &node)))
751 {
752 *ret = &node->IMFTopologyNode_iface;
753 IMFTopologyNode_AddRef(*ret);
754 }
755 else
756 *ret = NULL;
757
758 return hr;
759 }
760
761 static HRESULT topology_get_node_collection(const struct topology *topology, MF_TOPOLOGY_TYPE node_type,
762 IMFCollection **collection)
763 {
764 HRESULT hr;
765 size_t i;
766
767 if (!collection)
768 return E_POINTER;
769
770 if (FAILED(hr = MFCreateCollection(collection)))
771 return hr;
772
773 for (i = 0; i < topology->nodes.count; ++i)
774 {
775 if (topology->nodes.nodes[i]->node_type == node_type)
776 {
777 if (FAILED(hr = IMFCollection_AddElement(*collection,
778 (IUnknown *)&topology->nodes.nodes[i]->IMFTopologyNode_iface)))
779 {
780 IMFCollection_Release(*collection);
781 *collection = NULL;
782 break;
783 }
784 }
785 }
786
787 return hr;
788 }
789
790 static HRESULT WINAPI topology_GetSourceNodeCollection(IMFTopology *iface, IMFCollection **collection)
791 {
792 struct topology *topology = impl_from_IMFTopology(iface);
793
794 TRACE("%p, %p.\n", iface, collection);
795
796 return topology_get_node_collection(topology, MF_TOPOLOGY_SOURCESTREAM_NODE, collection);
797 }
798
799 static HRESULT WINAPI topology_GetOutputNodeCollection(IMFTopology *iface, IMFCollection **collection)
800 {
801 struct topology *topology = impl_from_IMFTopology(iface);
802
803 TRACE("%p, %p.\n", iface, collection);
804
805 return topology_get_node_collection(topology, MF_TOPOLOGY_OUTPUT_NODE, collection);
806 }
807
808 static const IMFTopologyVtbl topologyvtbl =
809 {
810 topology_QueryInterface,
811 topology_AddRef,
812 topology_Release,
813 topology_GetItem,
814 topology_GetItemType,
815 topology_CompareItem,
816 topology_Compare,
817 topology_GetUINT32,
818 topology_GetUINT64,
819 topology_GetDouble,
820 topology_GetGUID,
821 topology_GetStringLength,
822 topology_GetString,
823 topology_GetAllocatedString,
824 topology_GetBlobSize,
825 topology_GetBlob,
826 topology_GetAllocatedBlob,
827 topology_GetUnknown,
828 topology_SetItem,
829 topology_DeleteItem,
830 topology_DeleteAllItems,
831 topology_SetUINT32,
832 topology_SetUINT64,
833 topology_SetDouble,
834 topology_SetGUID,
835 topology_SetString,
836 topology_SetBlob,
837 topology_SetUnknown,
838 topology_LockStore,
839 topology_UnlockStore,
840 topology_GetCount,
841 topology_GetItemByIndex,
842 topology_CopyAllItems,
843 topology_GetTopologyID,
844 topology_AddNode,
845 topology_RemoveNode,
846 topology_GetNodeCount,
847 topology_GetNode,
848 topology_Clear,
849 topology_CloneFrom,
850 topology_GetNodeByID,
851 topology_GetSourceNodeCollection,
852 topology_GetOutputNodeCollection,
853 };
854
855 static struct topology *unsafe_impl_from_IMFTopology(IMFTopology *iface)
856 {
857 if (!iface || iface->lpVtbl != &topologyvtbl)
858 return NULL;
859 return impl_from_IMFTopology(iface);
860 }
861
862 static TOPOID topology_generate_id(void)
863 {
864 TOPOID old;
865
866 do
867 {
868 old = next_topology_id;
869 }
870 while (InterlockedCompareExchange64((LONG64 *)&next_topology_id, old + 1, old) != old);
871
872 return next_topology_id;
873 }
874
875 /***********************************************************************
876 * MFCreateTopology (mf.@)
877 */
878 HRESULT WINAPI MFCreateTopology(IMFTopology **topology)
879 {
880 struct topology *object;
881 HRESULT hr;
882
883 TRACE("%p.\n", topology);
884
885 if (!topology)
886 return E_POINTER;
887
888 if (!(object = calloc(1, sizeof(*object))))
889 return E_OUTOFMEMORY;
890
891 object->IMFTopology_iface.lpVtbl = &topologyvtbl;
892 object->refcount = 1;
893
894 hr = MFCreateAttributes(&object->attributes, 0);
895 if (FAILED(hr))
896 {
897 IMFTopology_Release(&object->IMFTopology_iface);
898 return hr;
899 }
900
901 object->id = topology_generate_id();
902
903 *topology = &object->IMFTopology_iface;
904
905 return S_OK;
906 }
907
908 static HRESULT WINAPI topology_node_QueryInterface(IMFTopologyNode *iface, REFIID riid, void **out)
909 {
910 struct topology_node *node = impl_from_IMFTopologyNode(iface);
911
912 TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
913
914 if (IsEqualIID(riid, &IID_IMFTopologyNode) ||
915 IsEqualIID(riid, &IID_IMFAttributes) ||
916 IsEqualIID(riid, &IID_IUnknown))
917 {
918 *out = &node->IMFTopologyNode_iface;
919 IMFTopologyNode_AddRef(iface);
920 return S_OK;
921 }
922
923 WARN("Unsupported interface %s.\n", debugstr_guid(riid));
924 *out = NULL;
925
926 return E_NOINTERFACE;
927 }
928
929 static ULONG WINAPI topology_node_AddRef(IMFTopologyNode *iface)
930 {
931 struct topology_node *node = impl_from_IMFTopologyNode(iface);
932 ULONG refcount = InterlockedIncrement(&node->refcount);
933
934 TRACE("%p, refcount %lu.\n", iface, refcount);
935
936 return refcount;
937 }
938
939 static ULONG WINAPI topology_node_Release(IMFTopologyNode *iface)
940 {
941 struct topology_node *node = impl_from_IMFTopologyNode(iface);
942 ULONG refcount = InterlockedDecrement(&node->refcount);
943 unsigned int i;
944
945 TRACE("%p, refcount %lu.\n", iface, refcount);
946
947 if (!refcount)
948 {
949 if (node->object)
950 IUnknown_Release(node->object);
951 if (node->input_type)
952 IMFMediaType_Release(node->input_type);
953 for (i = 0; i < node->inputs.count; ++i)
954 {
955 if (node->inputs.streams[i].preferred_type)
956 IMFMediaType_Release(node->inputs.streams[i].preferred_type);
957 }
958 for (i = 0; i < node->outputs.count; ++i)
959 {
960 if (node->outputs.streams[i].preferred_type)
961 IMFMediaType_Release(node->outputs.streams[i].preferred_type);
962 }
963 free(node->inputs.streams);
964 free(node->outputs.streams);
965 IMFAttributes_Release(node->attributes);
966 DeleteCriticalSection(&node->cs);
967 free(node);
968 }
969
970 return refcount;
971 }
972
973 static HRESULT WINAPI topology_node_GetItem(IMFTopologyNode *iface, REFGUID key, PROPVARIANT *value)
974 {
975 struct topology_node *node = impl_from_IMFTopologyNode(iface);
976
977 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
978
979 return IMFAttributes_GetItem(node->attributes, key, value);
980 }
981
982 static HRESULT WINAPI topology_node_GetItemType(IMFTopologyNode *iface, REFGUID key, MF_ATTRIBUTE_TYPE *type)
983 {
984 struct topology_node *node = impl_from_IMFTopologyNode(iface);
985
986 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), type);
987
988 return IMFAttributes_GetItemType(node->attributes, key, type);
989 }
990
991 static HRESULT WINAPI topology_node_CompareItem(IMFTopologyNode *iface, REFGUID key, REFPROPVARIANT value, BOOL *result)
992 {
993 struct topology_node *node = impl_from_IMFTopologyNode(iface);
994
995 TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), value, result);
996
997 return IMFAttributes_CompareItem(node->attributes, key, value, result);
998 }
999
1000 static HRESULT WINAPI topology_node_Compare(IMFTopologyNode *iface, IMFAttributes *theirs,
1001 MF_ATTRIBUTES_MATCH_TYPE type, BOOL *result)
1002 {
1003 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1004
1005 TRACE("%p, %p, %d, %p.\n", iface, theirs, type, result);
1006
1007 return IMFAttributes_Compare(node->attributes, theirs, type, result);
1008 }
1009
1010 static HRESULT WINAPI topology_node_GetUINT32(IMFTopologyNode *iface, REFGUID key, UINT32 *value)
1011 {
1012 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1013
1014 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1015
1016 return IMFAttributes_GetUINT32(node->attributes, key, value);
1017 }
1018
1019 static HRESULT WINAPI topology_node_GetUINT64(IMFTopologyNode *iface, REFGUID key, UINT64 *value)
1020 {
1021 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1022
1023 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1024
1025 return IMFAttributes_GetUINT64(node->attributes, key, value);
1026 }
1027
1028 static HRESULT WINAPI topology_node_GetDouble(IMFTopologyNode *iface, REFGUID key, double *value)
1029 {
1030 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1031
1032 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1033
1034 return IMFAttributes_GetDouble(node->attributes, key, value);
1035 }
1036
1037 static HRESULT WINAPI topology_node_GetGUID(IMFTopologyNode *iface, REFGUID key, GUID *value)
1038 {
1039 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1040
1041 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1042
1043 return IMFAttributes_GetGUID(node->attributes, key, value);
1044 }
1045
1046 static HRESULT WINAPI topology_node_GetStringLength(IMFTopologyNode *iface, REFGUID key, UINT32 *length)
1047 {
1048 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1049
1050 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), length);
1051
1052 return IMFAttributes_GetStringLength(node->attributes, key, length);
1053 }
1054
1055 static HRESULT WINAPI topology_node_GetString(IMFTopologyNode *iface, REFGUID key, WCHAR *value,
1056 UINT32 size, UINT32 *length)
1057 {
1058 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1059
1060 TRACE("%p, %s, %p, %d, %p.\n", iface, debugstr_guid(key), value, size, length);
1061
1062 return IMFAttributes_GetString(node->attributes, key, value, size, length);
1063 }
1064
1065 static HRESULT WINAPI topology_node_GetAllocatedString(IMFTopologyNode *iface, REFGUID key,
1066 WCHAR **value, UINT32 *length)
1067 {
1068 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1069
1070 TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), value, length);
1071
1072 return IMFAttributes_GetAllocatedString(node->attributes, key, value, length);
1073 }
1074
1075 static HRESULT WINAPI topology_node_GetBlobSize(IMFTopologyNode *iface, REFGUID key, UINT32 *size)
1076 {
1077 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1078
1079 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), size);
1080
1081 return IMFAttributes_GetBlobSize(node->attributes, key, size);
1082 }
1083
1084 static HRESULT WINAPI topology_node_GetBlob(IMFTopologyNode *iface, REFGUID key, UINT8 *buf,
1085 UINT32 bufsize, UINT32 *blobsize)
1086 {
1087 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1088
1089 TRACE("%p, %s, %p, %d, %p.\n", iface, debugstr_guid(key), buf, bufsize, blobsize);
1090
1091 return IMFAttributes_GetBlob(node->attributes, key, buf, bufsize, blobsize);
1092 }
1093
1094 static HRESULT WINAPI topology_node_GetAllocatedBlob(IMFTopologyNode *iface, REFGUID key, UINT8 **buf, UINT32 *size)
1095 {
1096 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1097
1098 TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), buf, size);
1099
1100 return IMFAttributes_GetAllocatedBlob(node->attributes, key, buf, size);
1101 }
1102
1103 static HRESULT WINAPI topology_node_GetUnknown(IMFTopologyNode *iface, REFGUID key, REFIID riid, void **ppv)
1104 {
1105 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1106
1107 TRACE("%p, %s, %s, %p.\n", iface, debugstr_guid(key), debugstr_guid(riid), ppv);
1108
1109 return IMFAttributes_GetUnknown(node->attributes, key, riid, ppv);
1110 }
1111
1112 static HRESULT WINAPI topology_node_SetItem(IMFTopologyNode *iface, REFGUID key, REFPROPVARIANT value)
1113 {
1114 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1115
1116 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1117
1118 return IMFAttributes_SetItem(node->attributes, key, value);
1119 }
1120
1121 static HRESULT WINAPI topology_node_DeleteItem(IMFTopologyNode *iface, REFGUID key)
1122 {
1123 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1124
1125 TRACE("%p, %s.\n", iface, debugstr_guid(key));
1126
1127 return IMFAttributes_DeleteItem(node->attributes, key);
1128 }
1129
1130 static HRESULT WINAPI topology_node_DeleteAllItems(IMFTopologyNode *iface)
1131 {
1132 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1133
1134 TRACE("%p.\n", iface);
1135
1136 return IMFAttributes_DeleteAllItems(node->attributes);
1137 }
1138
1139 static HRESULT WINAPI topology_node_SetUINT32(IMFTopologyNode *iface, REFGUID key, UINT32 value)
1140 {
1141 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1142
1143 TRACE("%p, %s, %d.\n", iface, debugstr_guid(key), value);
1144
1145 return IMFAttributes_SetUINT32(node->attributes, key, value);
1146 }
1147
1148 static HRESULT WINAPI topology_node_SetUINT64(IMFTopologyNode *iface, REFGUID key, UINT64 value)
1149 {
1150 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1151
1152 TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), wine_dbgstr_longlong(value));
1153
1154 return IMFAttributes_SetUINT64(node->attributes, key, value);
1155 }
1156
1157 static HRESULT WINAPI topology_node_SetDouble(IMFTopologyNode *iface, REFGUID key, double value)
1158 {
1159 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1160
1161 TRACE("%p, %s, %f.\n", iface, debugstr_guid(key), value);
1162
1163 return IMFAttributes_SetDouble(node->attributes, key, value);
1164 }
1165
1166 static HRESULT WINAPI topology_node_SetGUID(IMFTopologyNode *iface, REFGUID key, REFGUID value)
1167 {
1168 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1169
1170 TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), debugstr_guid(value));
1171
1172 return IMFAttributes_SetGUID(node->attributes, key, value);
1173 }
1174
1175 static HRESULT WINAPI topology_node_SetString(IMFTopologyNode *iface, REFGUID key, const WCHAR *value)
1176 {
1177 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1178
1179 TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), debugstr_w(value));
1180
1181 return IMFAttributes_SetString(node->attributes, key, value);
1182 }
1183
1184 static HRESULT WINAPI topology_node_SetBlob(IMFTopologyNode *iface, REFGUID key, const UINT8 *buf, UINT32 size)
1185 {
1186 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1187
1188 TRACE("%p, %s, %p, %d.\n", iface, debugstr_guid(key), buf, size);
1189
1190 return IMFAttributes_SetBlob(node->attributes, key, buf, size);
1191 }
1192
1193 static HRESULT WINAPI topology_node_SetUnknown(IMFTopologyNode *iface, REFGUID key, IUnknown *unknown)
1194 {
1195 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1196
1197 TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), unknown);
1198
1199 return IMFAttributes_SetUnknown(node->attributes, key, unknown);
1200 }
1201
1202 static HRESULT WINAPI topology_node_LockStore(IMFTopologyNode *iface)
1203 {
1204 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1205
1206 TRACE("%p.\n", iface);
1207
1208 return IMFAttributes_LockStore(node->attributes);
1209 }
1210
1211 static HRESULT WINAPI topology_node_UnlockStore(IMFTopologyNode *iface)
1212 {
1213 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1214
1215 TRACE("%p.\n", iface);
1216
1217 return IMFAttributes_UnlockStore(node->attributes);
1218 }
1219
1220 static HRESULT WINAPI topology_node_GetCount(IMFTopologyNode *iface, UINT32 *count)
1221 {
1222 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1223
1224 TRACE("%p, %p.\n", iface, count);
1225
1226 return IMFAttributes_GetCount(node->attributes, count);
1227 }
1228
1229 static HRESULT WINAPI topology_node_GetItemByIndex(IMFTopologyNode *iface, UINT32 index, GUID *key, PROPVARIANT *value)
1230 {
1231 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1232
1233 TRACE("%p, %u, %p, %p.\n", iface, index, key, value);
1234
1235 return IMFAttributes_GetItemByIndex(node->attributes, index, key, value);
1236 }
1237
1238 static HRESULT WINAPI topology_node_CopyAllItems(IMFTopologyNode *iface, IMFAttributes *dest)
1239 {
1240 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1241
1242 TRACE("%p, %p.\n", iface, dest);
1243
1244 return IMFAttributes_CopyAllItems(node->attributes, dest);
1245 }
1246
1247 static HRESULT topology_node_set_object(struct topology_node *node, IUnknown *object)
1248 {
1249 static const GUID *iids[3] = { &IID_IPersist, &IID_IPersistStorage, &IID_IPersistPropertyBag };
1250 IPersist *persist = NULL;
1251 BOOL has_object_id;
1252 GUID object_id;
1253 unsigned int i;
1254 HRESULT hr;
1255
1256 has_object_id = IMFAttributes_GetGUID(node->attributes, &MF_TOPONODE_TRANSFORM_OBJECTID, &object_id) == S_OK;
1257
1258 if (object && !has_object_id)
1259 {
1260 for (i = 0; i < ARRAY_SIZE(iids); ++i)
1261 {
1262 persist = NULL;
1263 if (SUCCEEDED(hr = IUnknown_QueryInterface(object, iids[i], (void **)&persist)))
1264 break;
1265 }
1266
1267 if (persist)
1268 {
1269 if (FAILED(hr = IPersist_GetClassID(persist, &object_id)))
1270 {
1271 IPersist_Release(persist);
1272 persist = NULL;
1273 }
1274 }
1275 }
1276
1277 EnterCriticalSection(&node->cs);
1278
1279 if (node->object)
1280 IUnknown_Release(node->object);
1281 node->object = object;
1282 if (node->object)
1283 IUnknown_AddRef(node->object);
1284
1285 if (persist)
1286 IMFAttributes_SetGUID(node->attributes, &MF_TOPONODE_TRANSFORM_OBJECTID, &object_id);
1287
1288 LeaveCriticalSection(&node->cs);
1289
1290 if (persist)
1291 IPersist_Release(persist);
1292
1293 return S_OK;
1294 }
1295
1296 static HRESULT WINAPI topology_node_SetObject(IMFTopologyNode *iface, IUnknown *object)
1297 {
1298 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1299
1300 TRACE("%p, %p.\n", iface, object);
1301
1302 return topology_node_set_object(node, object);
1303 }
1304
1305 static HRESULT WINAPI topology_node_GetObject(IMFTopologyNode *iface, IUnknown **object)
1306 {
1307 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1308
1309 TRACE("%p, %p.\n", iface, object);
1310
1311 if (!object)
1312 return E_POINTER;
1313
1314 EnterCriticalSection(&node->cs);
1315
1316 *object = node->object;
1317 if (*object)
1318 IUnknown_AddRef(*object);
1319
1320 LeaveCriticalSection(&node->cs);
1321
1322 return *object ? S_OK : E_FAIL;
1323 }
1324
1325 static HRESULT WINAPI topology_node_GetNodeType(IMFTopologyNode *iface, MF_TOPOLOGY_TYPE *node_type)
1326 {
1327 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1328
1329 TRACE("%p, %p.\n", iface, node_type);
1330
1331 *node_type = node->node_type;
1332
1333 return S_OK;
1334 }
1335
1336 static HRESULT WINAPI topology_node_GetTopoNodeID(IMFTopologyNode *iface, TOPOID *id)
1337 {
1338 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1339
1340 TRACE("%p, %p.\n", iface, id);
1341
1342 *id = node->id;
1343
1344 return S_OK;
1345 }
1346
1347 static HRESULT WINAPI topology_node_SetTopoNodeID(IMFTopologyNode *iface, TOPOID id)
1348 {
1349 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1350
1351 TRACE("%p, %s.\n", iface, wine_dbgstr_longlong(id));
1352
1353 node->id = id;
1354
1355 return S_OK;
1356 }
1357
1358 static HRESULT WINAPI topology_node_GetInputCount(IMFTopologyNode *iface, DWORD *count)
1359 {
1360 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1361
1362 TRACE("%p, %p.\n", iface, count);
1363
1364 *count = node->inputs.count;
1365
1366 return S_OK;
1367 }
1368
1369 static HRESULT WINAPI topology_node_GetOutputCount(IMFTopologyNode *iface, DWORD *count)
1370 {
1371 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1372
1373 TRACE("%p, %p.\n", iface, count);
1374
1375 *count = node->outputs.count;
1376
1377 return S_OK;
1378 }
1379
1380 static void topology_node_set_stream_type(struct node_stream *stream, IMFMediaType *mediatype)
1381 {
1382 if (stream->preferred_type)
1383 IMFMediaType_Release(stream->preferred_type);
1384 stream->preferred_type = mediatype;
1385 if (stream->preferred_type)
1386 IMFMediaType_AddRef(stream->preferred_type);
1387 }
1388
1389 static HRESULT topology_node_connect_output(struct topology_node *node, DWORD output_index,
1390 struct topology_node *connection, DWORD input_index)
1391 {
1392 struct node_stream *stream;
1393 HRESULT hr;
1394
1395 if (node->node_type == MF_TOPOLOGY_OUTPUT_NODE || connection->node_type == MF_TOPOLOGY_SOURCESTREAM_NODE)
1396 return E_FAIL;
1397
1398 EnterCriticalSection(&node->cs);
1399 EnterCriticalSection(&connection->cs);
1400
1401 topology_node_disconnect_output(node, output_index);
1402 if (input_index < connection->inputs.count)
1403 {
1404 stream = &connection->inputs.streams[input_index];
1405 if (stream->connection)
1406 topology_node_disconnect_output(stream->connection, stream->connection_stream);
1407 }
1408
1409 hr = topology_node_reserve_streams(&node->outputs, output_index);
1410 if (SUCCEEDED(hr))
1411 {
1412 size_t old_count = connection->inputs.count;
1413 hr = topology_node_reserve_streams(&connection->inputs, input_index);
1414 if (SUCCEEDED(hr) && !old_count && connection->input_type)
1415 {
1416 topology_node_set_stream_type(connection->inputs.streams, connection->input_type);
1417 IMFMediaType_Release(connection->input_type);
1418 connection->input_type = NULL;
1419 }
1420 }
1421
1422 if (SUCCEEDED(hr))
1423 {
1424 node->outputs.streams[output_index].connection = connection;
1425 IMFTopologyNode_AddRef(&node->outputs.streams[output_index].connection->IMFTopologyNode_iface);
1426 node->outputs.streams[output_index].connection_stream = input_index;
1427 connection->inputs.streams[input_index].connection = node;
1428 IMFTopologyNode_AddRef(&connection->inputs.streams[input_index].connection->IMFTopologyNode_iface);
1429 connection->inputs.streams[input_index].connection_stream = output_index;
1430 }
1431
1432 LeaveCriticalSection(&connection->cs);
1433 LeaveCriticalSection(&node->cs);
1434
1435 return hr;
1436 }
1437
1438 static HRESULT WINAPI topology_node_ConnectOutput(IMFTopologyNode *iface, DWORD output_index,
1439 IMFTopologyNode *peer, DWORD input_index)
1440 {
1441 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1442 struct topology_node *connection = unsafe_impl_from_IMFTopologyNode(peer);
1443
1444 TRACE("%p, %lu, %p, %lu.\n", iface, output_index, peer, input_index);
1445
1446 if (!connection)
1447 {
1448 WARN("External node implementations are not supported.\n");
1449 return E_UNEXPECTED;
1450 }
1451
1452 return topology_node_connect_output(node, output_index, connection, input_index);
1453 }
1454
1455 static HRESULT WINAPI topology_node_DisconnectOutput(IMFTopologyNode *iface, DWORD output_index)
1456 {
1457 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1458
1459 TRACE("%p, %lu.\n", iface, output_index);
1460
1461 return topology_node_disconnect_output(node, output_index);
1462 }
1463
1464 static HRESULT WINAPI topology_node_GetInput(IMFTopologyNode *iface, DWORD input_index, IMFTopologyNode **ret,
1465 DWORD *output_index)
1466 {
1467 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1468 HRESULT hr = S_OK;
1469
1470 TRACE("%p, %lu, %p, %p.\n", iface, input_index, ret, output_index);
1471
1472 EnterCriticalSection(&node->cs);
1473
1474 if (input_index < node->inputs.count)
1475 {
1476 const struct node_stream *stream = &node->inputs.streams[input_index];
1477
1478 if (stream->connection)
1479 {
1480 *ret = &stream->connection->IMFTopologyNode_iface;
1481 IMFTopologyNode_AddRef(*ret);
1482 *output_index = stream->connection_stream;
1483 }
1484 else
1485 hr = MF_E_NOT_FOUND;
1486 }
1487 else
1488 hr = E_INVALIDARG;
1489
1490 LeaveCriticalSection(&node->cs);
1491
1492 return hr;
1493 }
1494
1495 static HRESULT WINAPI topology_node_GetOutput(IMFTopologyNode *iface, DWORD output_index, IMFTopologyNode **ret,
1496 DWORD *input_index)
1497 {
1498 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1499 HRESULT hr = S_OK;
1500
1501 TRACE("%p, %lu, %p, %p.\n", iface, output_index, ret, input_index);
1502
1503 EnterCriticalSection(&node->cs);
1504
1505 if (output_index < node->outputs.count)
1506 {
1507 const struct node_stream *stream = &node->outputs.streams[output_index];
1508
1509 if (stream->connection)
1510 {
1511 *ret = &stream->connection->IMFTopologyNode_iface;
1512 IMFTopologyNode_AddRef(*ret);
1513 *input_index = stream->connection_stream;
1514 }
1515 else
1516 hr = MF_E_NOT_FOUND;
1517 }
1518 else
1519 hr = E_INVALIDARG;
1520
1521 LeaveCriticalSection(&node->cs);
1522
1523 return hr;
1524 }
1525
1526 static HRESULT WINAPI topology_node_SetOutputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType *mediatype)
1527 {
1528 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1529 HRESULT hr = S_OK;
1530
1531 TRACE("%p, %lu, %p.\n", iface, index, mediatype);
1532
1533 EnterCriticalSection(&node->cs);
1534
1535 if (node->node_type != MF_TOPOLOGY_OUTPUT_NODE)
1536 {
1537 if (SUCCEEDED(hr = topology_node_reserve_streams(&node->outputs, index)))
1538 topology_node_set_stream_type(&node->outputs.streams[index], mediatype);
1539 }
1540 else
1541 hr = E_NOTIMPL;
1542
1543 LeaveCriticalSection(&node->cs);
1544
1545 return hr;
1546 }
1547
1548 static HRESULT topology_node_get_pref_type(struct node_streams *streams, unsigned int index, IMFMediaType **mediatype)
1549 {
1550 *mediatype = streams->streams[index].preferred_type;
1551 if (*mediatype)
1552 {
1553 IMFMediaType_AddRef(*mediatype);
1554 return S_OK;
1555 }
1556
1557 return E_FAIL;
1558 }
1559
1560 static HRESULT WINAPI topology_node_GetOutputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType **mediatype)
1561 {
1562 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1563 HRESULT hr = S_OK;
1564
1565 TRACE("%p, %lu, %p.\n", iface, index, mediatype);
1566
1567 EnterCriticalSection(&node->cs);
1568
1569 if (index < node->outputs.count)
1570 hr = topology_node_get_pref_type(&node->outputs, index, mediatype);
1571 else
1572 hr = E_INVALIDARG;
1573
1574 LeaveCriticalSection(&node->cs);
1575
1576 return hr;
1577 }
1578
1579 static HRESULT WINAPI topology_node_SetInputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType *mediatype)
1580 {
1581 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1582 HRESULT hr = S_OK;
1583
1584 TRACE("%p, %lu, %p.\n", iface, index, mediatype);
1585
1586 EnterCriticalSection(&node->cs);
1587
1588 switch (node->node_type)
1589 {
1590 case MF_TOPOLOGY_TEE_NODE:
1591 if (index)
1592 {
1593 hr = MF_E_INVALIDTYPE;
1594 break;
1595 }
1596 if (node->inputs.count)
1597 topology_node_set_stream_type(&node->inputs.streams[index], mediatype);
1598 else
1599 {
1600 if (node->input_type)
1601 IMFMediaType_Release(node->input_type);
1602 node->input_type = mediatype;
1603 if (node->input_type)
1604 IMFMediaType_AddRef(node->input_type);
1605 }
1606 break;
1607 case MF_TOPOLOGY_SOURCESTREAM_NODE:
1608 hr = E_NOTIMPL;
1609 break;
1610 default:
1611 if (SUCCEEDED(hr = topology_node_reserve_streams(&node->inputs, index)))
1612 topology_node_set_stream_type(&node->inputs.streams[index], mediatype);
1613 }
1614
1615 LeaveCriticalSection(&node->cs);
1616
1617 return hr;
1618 }
1619
1620 static HRESULT WINAPI topology_node_GetInputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType **mediatype)
1621 {
1622 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1623 HRESULT hr = S_OK;
1624
1625 TRACE("%p, %lu, %p.\n", iface, index, mediatype);
1626
1627 EnterCriticalSection(&node->cs);
1628
1629 if (index < node->inputs.count)
1630 {
1631 hr = topology_node_get_pref_type(&node->inputs, index, mediatype);
1632 }
1633 else if (node->node_type == MF_TOPOLOGY_TEE_NODE && node->input_type)
1634 {
1635 *mediatype = node->input_type;
1636 IMFMediaType_AddRef(*mediatype);
1637 }
1638 else
1639 hr = E_INVALIDARG;
1640
1641 LeaveCriticalSection(&node->cs);
1642
1643 return hr;
1644 }
1645
1646 static HRESULT WINAPI topology_node_CloneFrom(IMFTopologyNode *iface, IMFTopologyNode *src_node)
1647 {
1648 struct topology_node *node = impl_from_IMFTopologyNode(iface);
1649 MF_TOPOLOGY_TYPE node_type;
1650 IMFMediaType *mediatype;
1651 IUnknown *object;
1652 DWORD count, i;
1653 TOPOID topoid;
1654 HRESULT hr;
1655
1656 TRACE("%p, %p.\n", iface, src_node);
1657
1658 if (FAILED(hr = IMFTopologyNode_GetNodeType(src_node, &node_type)))
1659 return hr;
1660
1661 if (node->node_type != node_type)
1662 return MF_E_INVALIDREQUEST;
1663
1664 if (FAILED(hr = IMFTopologyNode_GetTopoNodeID(src_node, &topoid)))
1665 return hr;
1666
1667 object = NULL;
1668 IMFTopologyNode_GetObject(src_node, &object);
1669
1670 EnterCriticalSection(&node->cs);
1671
1672 hr = IMFTopologyNode_CopyAllItems(src_node, node->attributes);
1673
1674 if (SUCCEEDED(hr))
1675 hr = topology_node_set_object(node, object);
1676
1677 if (SUCCEEDED(hr))
1678 node->id = topoid;
1679
1680 if (SUCCEEDED(IMFTopologyNode_GetInputCount(src_node, &count)))
1681 {
1682 for (i = 0; i < count; ++i)
1683 {
1684 if (SUCCEEDED(IMFTopologyNode_GetInputPrefType(src_node, i, &mediatype)))
1685 {
1686 IMFTopologyNode_SetInputPrefType(iface, i, mediatype);
1687 IMFMediaType_Release(mediatype);
1688 }
1689 }
1690 }
1691
1692 if (SUCCEEDED(IMFTopologyNode_GetOutputCount(src_node, &count)))
1693 {
1694 for (i = 0; i < count; ++i)
1695 {
1696 if (SUCCEEDED(IMFTopologyNode_GetOutputPrefType(src_node, i, &mediatype)))
1697 {
1698 IMFTopologyNode_SetOutputPrefType(iface, i, mediatype);
1699 IMFMediaType_Release(mediatype);
1700 }
1701 }
1702 }
1703
1704 LeaveCriticalSection(&node->cs);
1705
1706 if (object)
1707 IUnknown_Release(object);
1708
1709 return hr;
1710 }
1711
1712 static const IMFTopologyNodeVtbl topologynodevtbl =
1713 {
1714 topology_node_QueryInterface,
1715 topology_node_AddRef,
1716 topology_node_Release,
1717 topology_node_GetItem,
1718 topology_node_GetItemType,
1719 topology_node_CompareItem,
1720 topology_node_Compare,
1721 topology_node_GetUINT32,
1722 topology_node_GetUINT64,
1723 topology_node_GetDouble,
1724 topology_node_GetGUID,
1725 topology_node_GetStringLength,
1726 topology_node_GetString,
1727 topology_node_GetAllocatedString,
1728 topology_node_GetBlobSize,
1729 topology_node_GetBlob,
1730 topology_node_GetAllocatedBlob,
1731 topology_node_GetUnknown,
1732 topology_node_SetItem,
1733 topology_node_DeleteItem,
1734 topology_node_DeleteAllItems,
1735 topology_node_SetUINT32,
1736 topology_node_SetUINT64,
1737 topology_node_SetDouble,
1738 topology_node_SetGUID,
1739 topology_node_SetString,
1740 topology_node_SetBlob,
1741 topology_node_SetUnknown,
1742 topology_node_LockStore,
1743 topology_node_UnlockStore,
1744 topology_node_GetCount,
1745 topology_node_GetItemByIndex,
1746 topology_node_CopyAllItems,
1747 topology_node_SetObject,
1748 topology_node_GetObject,
1749 topology_node_GetNodeType,
1750 topology_node_GetTopoNodeID,
1751 topology_node_SetTopoNodeID,
1752 topology_node_GetInputCount,
1753 topology_node_GetOutputCount,
1754 topology_node_ConnectOutput,
1755 topology_node_DisconnectOutput,
1756 topology_node_GetInput,
1757 topology_node_GetOutput,
1758 topology_node_SetOutputPrefType,
1759 topology_node_GetOutputPrefType,
1760 topology_node_SetInputPrefType,
1761 topology_node_GetInputPrefType,
1762 topology_node_CloneFrom,
1763 };
1764
1765 static HRESULT create_topology_node(MF_TOPOLOGY_TYPE node_type, struct topology_node **node)
1766 {
1767 HRESULT hr;
1768
1769 if (!(*node = calloc(1, sizeof(**node))))
1770 return E_OUTOFMEMORY;
1771
1772 (*node)->IMFTopologyNode_iface.lpVtbl = &topologynodevtbl;
1773 (*node)->refcount = 1;
1774 (*node)->node_type = node_type;
1775 hr = MFCreateAttributes(&(*node)->attributes, 0);
1776 if (FAILED(hr))
1777 {
1778 free(*node);
1779 return hr;
1780 }
1781 (*node)->id = ((TOPOID)GetCurrentProcessId() << 32) | InterlockedIncrement(&next_node_id);
1782 InitializeCriticalSection(&(*node)->cs);
1783
1784 return S_OK;
1785 }
1786
1787 HRESULT topology_node_get_object(IMFTopologyNode *node, REFIID riid, void **obj)
1788 {
1789 IUnknown *unk;
1790 HRESULT hr;
1791
1792 *obj = NULL;
1793
1794 if (SUCCEEDED(hr = IMFTopologyNode_GetObject(node, &unk)))
1795 {
1796 hr = IUnknown_QueryInterface(unk, riid, obj);
1797 IUnknown_Release(unk);
1798 }
1799
1800 return hr;
1801 }
1802
1803 /***********************************************************************
1804 * MFCreateTopologyNode (mf.@)
1805 */
1806 HRESULT WINAPI MFCreateTopologyNode(MF_TOPOLOGY_TYPE node_type, IMFTopologyNode **node)
1807 {
1808 struct topology_node *object;
1809 HRESULT hr;
1810
1811 TRACE("%d, %p.\n", node_type, node);
1812
1813 if (!node)
1814 return E_POINTER;
1815
1816 hr = create_topology_node(node_type, &object);
1817 if (SUCCEEDED(hr))
1818 *node = &object->IMFTopologyNode_iface;
1819
1820 return hr;
1821 }
1822
1823 /***********************************************************************
1824 * MFGetTopoNodeCurrentType (mf.@)
1825 */
1826 HRESULT WINAPI MFGetTopoNodeCurrentType(IMFTopologyNode *node, DWORD stream, BOOL output, IMFMediaType **type)
1827 {
1828 IMFMediaTypeHandler *type_handler;
1829 MF_TOPOLOGY_TYPE node_type;
1830 IMFStreamSink *stream_sink;
1831 IMFStreamDescriptor *sd;
1832 IMFTransform *transform;
1833 UINT32 primary_output;
1834 HRESULT hr;
1835
1836 TRACE("%p, %lu, %d, %p.\n", node, stream, output, type);
1837
1838 if (FAILED(hr = IMFTopologyNode_GetNodeType(node, &node_type)))
1839 return hr;
1840
1841 switch (node_type)
1842 {
1843 case MF_TOPOLOGY_OUTPUT_NODE:
1844 if (SUCCEEDED(topology_node_get_object(node, &IID_IMFStreamSink, (void **)&stream_sink)))
1845 {
1846 hr = IMFStreamSink_GetMediaTypeHandler(stream_sink, &type_handler);
1847 IMFStreamSink_Release(stream_sink);
1848
1849 if (SUCCEEDED(hr))
1850 {
1851 hr = IMFMediaTypeHandler_GetCurrentMediaType(type_handler, type);
1852 IMFMediaTypeHandler_Release(type_handler);
1853 }
1854 }
1855 break;
1856 case MF_TOPOLOGY_SOURCESTREAM_NODE:
1857 if (FAILED(hr = IMFTopologyNode_GetUnknown(node, &MF_TOPONODE_STREAM_DESCRIPTOR, &IID_IMFStreamDescriptor,
1858 (void **)&sd)))
1859 {
1860 return hr;
1861 }
1862
1863 hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, &type_handler);
1864 IMFStreamDescriptor_Release(sd);
1865 if (SUCCEEDED(hr))
1866 {
1867 hr = IMFMediaTypeHandler_GetCurrentMediaType(type_handler, type);
1868 IMFMediaTypeHandler_Release(type_handler);
1869 }
1870 break;
1871 case MF_TOPOLOGY_TRANSFORM_NODE:
1872 if (SUCCEEDED(hr = topology_node_get_object(node, &IID_IMFTransform, (void **)&transform)))
1873 {
1874 if (output)
1875 hr = IMFTransform_GetOutputCurrentType(transform, stream, type);
1876 else
1877 hr = IMFTransform_GetInputCurrentType(transform, stream, type);
1878 IMFTransform_Release(transform);
1879 }
1880 break;
1881 case MF_TOPOLOGY_TEE_NODE:
1882 if (SUCCEEDED(hr = IMFTopologyNode_GetInputPrefType(node, 0, type)))
1883 break;
1884
1885 if (FAILED(IMFTopologyNode_GetUINT32(node, &MF_TOPONODE_PRIMARYOUTPUT, &primary_output)))
1886 primary_output = 0;
1887
1888 hr = IMFTopologyNode_GetOutputPrefType(node, primary_output, type);
1889 break;
1890 default:
1891 ;
1892 }
1893
1894 return hr;
1895 }
1896
1897 static HRESULT WINAPI topology_loader_QueryInterface(IMFTopoLoader *iface, REFIID riid, void **out)
1898 {
1899 TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
1900
1901 if (IsEqualIID(riid, &IID_IMFTopoLoader) ||
1902 IsEqualIID(riid, &IID_IUnknown))
1903 {
1904 *out = iface;
1905 IMFTopoLoader_AddRef(iface);
1906 return S_OK;
1907 }
1908
1909 WARN("Unsupported %s.\n", debugstr_guid(riid));
1910 *out = NULL;
1911 return E_NOINTERFACE;
1912 }
1913
1914 static ULONG WINAPI topology_loader_AddRef(IMFTopoLoader *iface)
1915 {
1916 struct topology_loader *loader = impl_from_IMFTopoLoader(iface);
1917 ULONG refcount = InterlockedIncrement(&loader->refcount);
1918
1919 TRACE("%p, refcount %lu.\n", iface, refcount);
1920
1921 return refcount;
1922 }
1923
1924 static ULONG WINAPI topology_loader_Release(IMFTopoLoader *iface)
1925 {
1926 struct topology_loader *loader = impl_from_IMFTopoLoader(iface);
1927 ULONG refcount = InterlockedDecrement(&loader->refcount);
1928
1929 TRACE("%p, refcount %lu.\n", iface, refcount);
1930
1931 if (!refcount)
1932 free(loader);
1933
1934 return refcount;
1935 }
1936
1937 struct topoloader_context
1938 {
1939 IMFTopology *input_topology;
1940 IMFTopology *output_topology;
1941 unsigned int marker;
1942 GUID key;
1943 };
1944
1945 static IMFTopologyNode *topology_loader_get_node_for_marker(struct topoloader_context *context, TOPOID *id)
1946 {
1947 IMFTopologyNode *node;
1948 unsigned short i = 0;
1949 unsigned int value;
1950
1951 while (SUCCEEDED(IMFTopology_GetNode(context->output_topology, i++, &node)))
1952 {
1953 if (SUCCEEDED(IMFTopologyNode_GetUINT32(node, &context->key, &value)) && value == context->marker)
1954 {
1955 IMFTopologyNode_GetTopoNodeID(node, id);
1956 return node;
1957 }
1958 IMFTopologyNode_Release(node);
1959 }
1960
1961 *id = 0;
1962 return NULL;
1963 }
1964
1965 static HRESULT topology_loader_clone_node(struct topoloader_context *context, IMFTopologyNode *node,
1966 IMFTopologyNode **ret, unsigned int marker)
1967 {
1968 IMFTopologyNode *cloned_node;
1969 MF_TOPOLOGY_TYPE node_type;
1970 HRESULT hr;
1971
1972 if (ret) *ret = NULL;
1973
1974 IMFTopologyNode_GetNodeType(node, &node_type);
1975
1976 if (FAILED(hr = MFCreateTopologyNode(node_type, &cloned_node)))
1977 return hr;
1978
1979 if (SUCCEEDED(hr = IMFTopologyNode_CloneFrom(cloned_node, node)))
1980 hr = IMFTopologyNode_SetUINT32(cloned_node, &context->key, marker);
1981
1982 if (SUCCEEDED(hr))
1983 hr = IMFTopology_AddNode(context->output_topology, cloned_node);
1984
1985 if (SUCCEEDED(hr) && ret)
1986 {
1987 *ret = cloned_node;
1988 IMFTopologyNode_AddRef(*ret);
1989 }
1990
1991 IMFTopologyNode_Release(cloned_node);
1992
1993 return hr;
1994 }
1995
1996 struct transform_output_type
1997 {
1998 IMFMediaType *type;
1999 IMFTransform *transform;
2000 IMFActivate *activate;
2001 };
2002
2003 struct connect_context
2004 {
2005 struct topoloader_context *context;
2006 IMFTopologyNode *upstream_node;
2007 IMFTopologyNode *sink;
2008 IMFMediaTypeHandler *sink_handler;
2009 unsigned int output_index;
2010 unsigned int input_index;
2011 GUID converter_category;
2012 GUID decoder_category;
2013 };
2014
2015 typedef HRESULT (*p_connect_func)(struct transform_output_type *output_type, struct connect_context *context);
2016
2017 static void topology_loader_release_transforms(IMFActivate **activates, unsigned int count)
2018 {
2019 unsigned int i;
2020
2021 for (i = 0; i < count; ++i)
2022 IMFActivate_Release(activates[i]);
2023 CoTaskMemFree(activates);
2024 }
2025
2026 static HRESULT topology_loader_enumerate_output_types(const GUID *category, IMFMediaType *input_type,
2027 p_connect_func connect_func, struct connect_context *context)
2028 {
2029 MFT_REGISTER_TYPE_INFO mft_typeinfo;
2030 IMFActivate **activates;
2031 unsigned int i, count;
2032 HRESULT hr;
2033
2034 if (FAILED(hr = IMFMediaType_GetMajorType(input_type, &mft_typeinfo.guidMajorType)))
2035 return hr;
2036
2037 if (FAILED(hr = IMFMediaType_GetGUID(input_type, &MF_MT_SUBTYPE, &mft_typeinfo.guidSubtype)))
2038 return hr;
2039
2040 if (FAILED(hr = MFTEnumEx(*category, MFT_ENUM_FLAG_ALL, &mft_typeinfo, NULL, &activates, &count)))
2041 return hr;
2042
2043 hr = E_FAIL;
2044
2045 for (i = 0; i < count; ++i)
2046 {
2047 IMFTransform *transform;
2048
2049 if (FAILED(IMFActivate_ActivateObject(activates[i], &IID_IMFTransform, (void **)&transform)))
2050 {
2051 WARN("Failed to create a transform.\n");
2052 continue;
2053 }
2054
2055 if (SUCCEEDED(hr = IMFTransform_SetInputType(transform, 0, input_type, 0)))
2056 {
2057 struct transform_output_type output_type;
2058 unsigned int output_count = 0;
2059
2060 output_type.transform = transform;
2061 output_type.activate = activates[i];
2062 while (SUCCEEDED(IMFTransform_GetOutputAvailableType(transform, 0, output_count++, &output_type.type)))
2063 {
2064 hr = connect_func(&output_type, context);
2065 IMFMediaType_Release(output_type.type);
2066 if (SUCCEEDED(hr))
2067 {
2068 topology_loader_release_transforms(activates, count);
2069 return hr;
2070 }
2071 }
2072 }
2073
2074 IMFActivate_ShutdownObject(activates[i]);
2075 }
2076
2077 topology_loader_release_transforms(activates, count);
2078
2079 return hr;
2080 }
2081
2082 static HRESULT topology_loader_create_transform(const struct transform_output_type *output_type,
2083 IMFTopologyNode **node)
2084 {
2085 HRESULT hr;
2086 GUID guid;
2087
2088 if (FAILED(hr = MFCreateTopologyNode(MF_TOPOLOGY_TRANSFORM_NODE, node)))
2089 return hr;
2090
2091 IMFTopologyNode_SetObject(*node, (IUnknown *)output_type->transform);
2092
2093 if (SUCCEEDED(IMFActivate_GetGUID(output_type->activate, &MF_TRANSFORM_CATEGORY_Attribute, &guid)) &&
2094 (IsEqualGUID(&guid, &MFT_CATEGORY_AUDIO_DECODER) || IsEqualGUID(&guid, &MFT_CATEGORY_VIDEO_DECODER)))
2095 {
2096 IMFTopologyNode_SetUINT32(*node, &MF_TOPONODE_DECODER, 1);
2097 }
2098
2099 if (SUCCEEDED(IMFActivate_GetGUID(output_type->activate, &MFT_TRANSFORM_CLSID_Attribute, &guid)))
2100 IMFTopologyNode_SetGUID(*node, &MF_TOPONODE_TRANSFORM_OBJECTID, &guid);
2101
2102 return hr;
2103 }
2104
2105 static HRESULT connect_to_sink(struct transform_output_type *output_type, struct connect_context *context)
2106 {
2107 IMFTopologyNode *node;
2108 HRESULT hr;
2109
2110 if (FAILED(IMFMediaTypeHandler_IsMediaTypeSupported(context->sink_handler, output_type->type, NULL)))
2111 return MF_E_TRANSFORM_NOT_POSSIBLE_FOR_CURRENT_MEDIATYPE_COMBINATION;
2112
2113 if (FAILED(hr = topology_loader_create_transform(output_type, &node)))
2114 return hr;
2115
2116 IMFTopology_AddNode(context->context->output_topology, node);
2117 IMFTopologyNode_ConnectOutput(context->upstream_node, 0, node, 0);
2118 IMFTopologyNode_ConnectOutput(node, 0, context->sink, 0);
2119
2120 IMFTopologyNode_Release(node);
2121
2122 hr = IMFMediaTypeHandler_SetCurrentMediaType(context->sink_handler, output_type->type);
2123 if (SUCCEEDED(hr))
2124 hr = IMFTransform_SetOutputType(output_type->transform, 0, output_type->type, 0);
2125
2126 return S_OK;
2127 }
2128
2129 static HRESULT connect_to_converter(struct transform_output_type *output_type, struct connect_context *context)
2130 {
2131 struct connect_context sink_ctx;
2132 IMFTopologyNode *node;
2133 HRESULT hr;
2134
2135 if (SUCCEEDED(connect_to_sink(output_type, context)))
2136 return S_OK;
2137
2138 if (FAILED(hr = topology_loader_create_transform(output_type, &node)))
2139 return hr;
2140
2141 sink_ctx = *context;
2142 sink_ctx.upstream_node = node;
2143
2144 if (SUCCEEDED(hr = topology_loader_enumerate_output_types(&context->converter_category, output_type->type,
2145 connect_to_sink, &sink_ctx)))
2146 {
2147 hr = IMFTopology_AddNode(context->context->output_topology, node);
2148 }
2149 IMFTopologyNode_Release(node);
2150
2151 if (SUCCEEDED(hr))
2152 {
2153 IMFTopology_AddNode(context->context->output_topology, node);
2154 IMFTopologyNode_ConnectOutput(context->upstream_node, 0, node, 0);
2155
2156 hr = IMFTransform_SetOutputType(output_type->transform, 0, output_type->type, 0);
2157 }
2158
2159 return hr;
2160 }
2161
2162 typedef HRESULT (*p_topology_loader_connect_func)(struct topoloader_context *context, IMFTopologyNode *upstream_node,
2163 unsigned int output_index, IMFTopologyNode *downstream_node, unsigned int input_index);
2164
2165 static HRESULT topology_loader_get_node_type_handler(IMFTopologyNode *node, IMFMediaTypeHandler **handler)
2166 {
2167 MF_TOPOLOGY_TYPE node_type;
2168 IMFStreamSink *stream_sink;
2169 IMFStreamDescriptor *sd;
2170 HRESULT hr;
2171
2172 if (FAILED(hr = IMFTopologyNode_GetNodeType(node, &node_type)))
2173 return hr;
2174
2175 switch (node_type)
2176 {
2177 case MF_TOPOLOGY_OUTPUT_NODE:
2178 if (SUCCEEDED(hr = topology_node_get_object(node, &IID_IMFStreamSink, (void **)&stream_sink)))
2179 {
2180 hr = IMFStreamSink_GetMediaTypeHandler(stream_sink, handler);
2181 IMFStreamSink_Release(stream_sink);
2182 }
2183 break;
2184 case MF_TOPOLOGY_SOURCESTREAM_NODE:
2185 if (SUCCEEDED(hr = IMFTopologyNode_GetUnknown(node, &MF_TOPONODE_STREAM_DESCRIPTOR,
2186 &IID_IMFStreamDescriptor, (void **)&sd)))
2187 {
2188 hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, handler);
2189 IMFStreamDescriptor_Release(sd);
2190 }
2191 break;
2192 default:
2193 WARN("Unexpected node type %u.\n", node_type);
2194 return MF_E_UNEXPECTED;
2195 }
2196
2197 return hr;
2198 }
2199
2200 static HRESULT topology_loader_get_mft_categories(IMFMediaTypeHandler *handler, GUID *decode_cat, GUID *convert_cat)
2201 {
2202 GUID major;
2203 HRESULT hr;
2204
2205 if (FAILED(hr = IMFMediaTypeHandler_GetMajorType(handler, &major)))
2206 return hr;
2207
2208 if (IsEqualGUID(&major, &MFMediaType_Audio))
2209 {
2210 *decode_cat = MFT_CATEGORY_AUDIO_DECODER;
2211 *convert_cat = MFT_CATEGORY_AUDIO_EFFECT;
2212 }
2213 else if (IsEqualGUID(&major, &MFMediaType_Video))
2214 {
2215 *decode_cat = MFT_CATEGORY_VIDEO_DECODER;
2216 *convert_cat = MFT_CATEGORY_VIDEO_EFFECT;
2217 }
2218 else
2219 {
2220 WARN("Unexpected major type %s.\n", debugstr_guid(&major));
2221 return MF_E_INVALIDTYPE;
2222 }
2223
2224 return S_OK;
2225 }
2226
2227 static HRESULT topology_loader_connect(IMFMediaTypeHandler *sink_handler, unsigned int sink_method,
2228 struct connect_context *sink_ctx, struct connect_context *convert_ctx, IMFMediaType *media_type)
2229 {
2230 HRESULT hr;
2231
2232 if (SUCCEEDED(hr = IMFMediaTypeHandler_IsMediaTypeSupported(sink_handler, media_type, NULL))
2233 && SUCCEEDED(hr = IMFMediaTypeHandler_SetCurrentMediaType(sink_handler, media_type)))
2234 {
2235 hr = IMFTopologyNode_ConnectOutput(sink_ctx->upstream_node, sink_ctx->output_index, sink_ctx->sink, sink_ctx->input_index);
2236 }
2237
2238 if (FAILED(hr) && sink_method & MF_CONNECT_ALLOW_CONVERTER)
2239 {
2240 hr = topology_loader_enumerate_output_types(&convert_ctx->converter_category, media_type, connect_to_sink, sink_ctx);
2241 }
2242
2243 if (FAILED(hr) && sink_method & MF_CONNECT_ALLOW_DECODER)
2244 {
2245 hr = topology_loader_enumerate_output_types(&convert_ctx->decoder_category, media_type,
2246 connect_to_converter, convert_ctx);
2247 }
2248
2249 return hr;
2250 }
2251
2252 static HRESULT topology_loader_foreach_source_type(IMFMediaTypeHandler *sink_handler, IMFMediaTypeHandler *source_handler,
2253 unsigned int sink_method, struct connect_context *sink_ctx, struct connect_context *convert_ctx)
2254 {
2255 unsigned int index = 0;
2256 IMFMediaType *media_type;
2257 HRESULT hr;
2258
2259 while (SUCCEEDED(hr = IMFMediaTypeHandler_GetMediaTypeByIndex(source_handler, index++, &media_type)))
2260 {
2261 hr = topology_loader_connect(sink_handler, sink_method, sink_ctx, convert_ctx, media_type);
2262 if (SUCCEEDED(hr)) break;
2263 IMFMediaType_Release(media_type);
2264 media_type = NULL;
2265 }
2266
2267 if (media_type)
2268 {
2269 hr = IMFMediaTypeHandler_SetCurrentMediaType(source_handler, media_type);
2270 IMFMediaType_Release(media_type);
2271 }
2272
2273 return hr;
2274 }
2275
2276 static HRESULT topology_loader_connect_source_to_sink(struct topoloader_context *context, IMFTopologyNode *source,
2277 unsigned int output_index, IMFTopologyNode *sink, unsigned int input_index)
2278 {
2279 IMFMediaTypeHandler *source_handler = NULL, *sink_handler = NULL;
2280 struct connect_context convert_ctx, sink_ctx;
2281 MF_CONNECT_METHOD source_method, sink_method;
2282 unsigned int enumerate_source_types = 0;
2283 IMFMediaType *media_type;
2284 HRESULT hr;
2285
2286 TRACE("attempting to connect %p:%u to %p:%u\n", source, output_index, sink, input_index);
2287
2288 if (FAILED(hr = topology_loader_get_node_type_handler(source, &source_handler)))
2289 goto done;
2290
2291 if (FAILED(hr = topology_loader_get_node_type_handler(sink, &sink_handler)))
2292 goto done;
2293
2294 if (FAILED(IMFTopologyNode_GetUINT32(source, &MF_TOPONODE_CONNECT_METHOD, &source_method)))
2295 source_method = MF_CONNECT_DIRECT;
2296 if (FAILED(IMFTopologyNode_GetUINT32(sink, &MF_TOPONODE_CONNECT_METHOD, &sink_method)))
2297 sink_method = MF_CONNECT_ALLOW_DECODER;
2298
2299 sink_ctx.context = context;
2300 sink_ctx.upstream_node = source;
2301 sink_ctx.sink = sink;
2302 sink_ctx.sink_handler = sink_handler;
2303 sink_ctx.output_index = output_index;
2304 sink_ctx.input_index = input_index;
2305
2306 convert_ctx = sink_ctx;
2307 if (FAILED(hr = topology_loader_get_mft_categories(source_handler, &convert_ctx.decoder_category,
2308 &convert_ctx.converter_category)))
2309 goto done;
2310
2311 IMFTopology_GetUINT32(context->output_topology, &MF_TOPOLOGY_ENUMERATE_SOURCE_TYPES, &enumerate_source_types);
2312
2313 if (enumerate_source_types)
2314 {
2315 if (source_method & MF_CONNECT_RESOLVE_INDEPENDENT_OUTPUTTYPES)
2316 {
2317 hr = topology_loader_foreach_source_type(sink_handler, source_handler, MF_CONNECT_ALLOW_DECODER,
2318 &sink_ctx, &convert_ctx);
2319 }
2320 else
2321 {
2322 hr = topology_loader_foreach_source_type(sink_handler, source_handler, MF_CONNECT_DIRECT,
2323 &sink_ctx, &convert_ctx);
2324 if (FAILED(hr))
2325 hr = topology_loader_foreach_source_type(sink_handler, source_handler, MF_CONNECT_ALLOW_CONVERTER,
2326 &sink_ctx, &convert_ctx);
2327 if (FAILED(hr))
2328 hr = topology_loader_foreach_source_type(sink_handler, source_handler, MF_CONNECT_ALLOW_DECODER,
2329 &sink_ctx, &convert_ctx);
2330 }
2331 }
2332 else
2333 {
2334 if (SUCCEEDED(hr = IMFMediaTypeHandler_GetCurrentMediaType(source_handler, &media_type)))
2335 {
2336 hr = topology_loader_connect(sink_handler, sink_method, &sink_ctx, &convert_ctx, media_type);
2337 IMFMediaType_Release(media_type);
2338 }
2339 }
2340
2341 done:
2342 if (source_handler)
2343 IMFMediaTypeHandler_Release(source_handler);
2344 if (sink_handler)
2345 IMFMediaTypeHandler_Release(sink_handler);
2346
2347 return hr;
2348 }
2349
2350 static HRESULT topology_loader_resolve_branch(struct topoloader_context *context, IMFTopologyNode *upstream_node,
2351 unsigned int output_index, IMFTopologyNode *downstream_node, unsigned input_index)
2352 {
2353 static const p_topology_loader_connect_func connectors[MF_TOPOLOGY_TEE_NODE+1][MF_TOPOLOGY_TEE_NODE+1] =
2354 {
2355 /* OUTPUT */ { NULL },
2356 /* SOURCESTREAM */ { topology_loader_connect_source_to_sink, NULL, NULL, NULL },
2357 /* TRANSFORM */ { NULL },
2358 /* TEE */ { NULL },
2359 };
2360 MF_TOPOLOGY_TYPE u_type, d_type;
2361 IMFTopologyNode *node;
2362 HRESULT hr;
2363 TOPOID id;
2364
2365 /* Downstream node might have already been cloned. */
2366 IMFTopologyNode_GetTopoNodeID(downstream_node, &id);
2367 if (FAILED(IMFTopology_GetNodeByID(context->output_topology, id, &node)))
2368 topology_loader_clone_node(context, downstream_node, &node, context->marker + 1);
2369
2370 IMFTopologyNode_GetNodeType(upstream_node, &u_type);
2371 IMFTopologyNode_GetNodeType(downstream_node, &d_type);
2372
2373 if (!connectors[u_type][d_type])
2374 {
2375 WARN("Unsupported branch kind %d -> %d.\n", u_type, d_type);
2376 IMFTopologyNode_Release(node);
2377 return E_FAIL;
2378 }
2379
2380 hr = connectors[u_type][d_type](context, upstream_node, output_index, node, input_index);
2381 IMFTopologyNode_Release(node);
2382 return hr;
2383 }
2384
2385 static HRESULT topology_loader_resolve_nodes(struct topoloader_context *context, unsigned int *layer_size)
2386 {
2387 IMFTopologyNode *downstream_node, *node, *orig_node;
2388 MF_TOPOLOGY_TYPE node_type;
2389 unsigned int size = 0;
2390 DWORD input_index;
2391 HRESULT hr = S_OK;
2392 TOPOID id;
2393
2394 while ((node = topology_loader_get_node_for_marker(context, &id)))
2395 {
2396 ++size;
2397
2398 IMFTopologyNode_GetNodeType(node, &node_type);
2399 switch (node_type)
2400 {
2401 case MF_TOPOLOGY_SOURCESTREAM_NODE:
2402 if (SUCCEEDED(hr = IMFTopology_GetNodeByID(context->input_topology, id, &orig_node)))
2403 {
2404 hr = IMFTopologyNode_GetOutput(orig_node, 0, &downstream_node, &input_index);
2405 IMFTopologyNode_Release(orig_node);
2406 }
2407
2408 if (FAILED(hr))
2409 {
2410 IMFTopology_RemoveNode(context->output_topology, node);
2411 IMFTopologyNode_Release(node);
2412 continue;
2413 }
2414
2415 hr = topology_loader_resolve_branch(context, node, 0, downstream_node, input_index);
2416 IMFTopologyNode_Release(downstream_node);
2417 break;
2418 case MF_TOPOLOGY_TRANSFORM_NODE:
2419 case MF_TOPOLOGY_TEE_NODE:
2420 FIXME("Unsupported node type %d.\n", node_type);
2421 break;
2422 default:
2423 WARN("Unexpected node type %d.\n", node_type);
2424 }
2425
2426 IMFTopologyNode_DeleteItem(node, &context->key);
2427 IMFTopologyNode_Release(node);
2428
2429 if (FAILED(hr))
2430 break;
2431 }
2432
2433 *layer_size = size;
2434
2435 return hr;
2436 }
2437
2438 static BOOL topology_loader_is_node_d3d_aware(IMFTopologyNode *node)
2439 {
2440 IMFAttributes *attributes;
2441 unsigned int d3d_aware = 0;
2442 IMFTransform *transform;
2443
2444 if (FAILED(topology_node_get_object(node, &IID_IMFAttributes, (void **)&attributes)))
2445 return FALSE;
2446
2447 IMFAttributes_GetUINT32(attributes, &MF_SA_D3D_AWARE, &d3d_aware);
2448 IMFAttributes_Release(attributes);
2449
2450 if (!d3d_aware && SUCCEEDED(topology_node_get_object(node, &IID_IMFTransform, (void **)&transform)))
2451 {
2452 d3d_aware = mf_is_sample_copier_transform(transform);
2453 IMFTransform_Release(transform);
2454 }
2455
2456 return !!d3d_aware;
2457 }
2458
2459 static HRESULT topology_loader_create_copier(IMFTopologyNode *upstream_node, DWORD upstream_output,
2460 IMFTopologyNode *downstream_node, unsigned int downstream_input, IMFTransform **copier)
2461 {
2462 IMFMediaType *input_type = NULL, *output_type = NULL;
2463 IMFTransform *transform;
2464 HRESULT hr;
2465
2466 if (FAILED(hr = MFCreateSampleCopierMFT(&transform)))
2467 return hr;
2468
2469 if (FAILED(hr = MFGetTopoNodeCurrentType(upstream_node, upstream_output, TRUE, &input_type)))
2470 WARN("Failed to get upstream media type hr %#lx.\n", hr);
2471
2472 if (SUCCEEDED(hr) && FAILED(hr = MFGetTopoNodeCurrentType(downstream_node, downstream_input, FALSE, &output_type)))
2473 WARN("Failed to get downstream media type hr %#lx.\n", hr);
2474
2475 if (SUCCEEDED(hr) && FAILED(hr = IMFTransform_SetInputType(transform, 0, input_type, 0)))
2476 WARN("Input type wasn't accepted, hr %#lx.\n", hr);
2477
2478 if (SUCCEEDED(hr) && FAILED(hr = IMFTransform_SetOutputType(transform, 0, output_type, 0)))
2479 WARN("Output type wasn't accepted, hr %#lx.\n", hr);
2480
2481 if (SUCCEEDED(hr))
2482 {
2483 *copier = transform;
2484 IMFTransform_AddRef(*copier);
2485 }
2486
2487 if (input_type)
2488 IMFMediaType_Release(input_type);
2489 if (output_type)
2490 IMFMediaType_Release(output_type);
2491
2492 IMFTransform_Release(transform);
2493
2494 return hr;
2495 }
2496
2497 static HRESULT topology_loader_connect_copier(struct topoloader_context *context, IMFTopologyNode *upstream_node,
2498 DWORD upstream_output, IMFTopologyNode *downstream_node, DWORD downstream_input, IMFTransform *copier)
2499 {
2500 IMFTopologyNode *copier_node;
2501 HRESULT hr;
2502
2503 if (FAILED(hr = MFCreateTopologyNode(MF_TOPOLOGY_TRANSFORM_NODE, &copier_node)))
2504 return hr;
2505
2506 IMFTopologyNode_SetObject(copier_node, (IUnknown *)copier);
2507 IMFTopology_AddNode(context->output_topology, copier_node);
2508 IMFTopologyNode_ConnectOutput(upstream_node, upstream_output, copier_node, 0);
2509 IMFTopologyNode_ConnectOutput(copier_node, 0, downstream_node, downstream_input);
2510
2511 IMFTopologyNode_Release(copier_node);
2512
2513 return S_OK;
2514 }
2515
2516 /* Right now this should be used for output nodes only. */
2517 static HRESULT topology_loader_connect_d3d_aware_input(struct topoloader_context *context,
2518 IMFTopologyNode *node)
2519 {
2520 IMFTopologyNode *upstream_node;
2521 IMFTransform *copier = NULL;
2522 IMFStreamSink *stream_sink;
2523 DWORD upstream_output;
2524 HRESULT hr;
2525
2526 if (FAILED(hr = topology_node_get_object(node, &IID_IMFStreamSink, (void **)&stream_sink)))
2527 return hr;
2528
2529 if (topology_loader_is_node_d3d_aware(node))
2530 {
2531 if (SUCCEEDED(IMFTopologyNode_GetInput(node, 0, &upstream_node, &upstream_output)))
2532 {
2533 if (!topology_loader_is_node_d3d_aware(upstream_node))
2534 {
2535 if (SUCCEEDED(hr = topology_loader_create_copier(upstream_node, upstream_output, node, 0, &copier)))
2536 {
2537 hr = topology_loader_connect_copier(context, upstream_node, upstream_output, node, 0, copier);
2538 IMFTransform_Release(copier);
2539 }
2540 }
2541 IMFTopologyNode_Release(upstream_node);
2542 }
2543 }
2544
2545 IMFStreamSink_Release(stream_sink);
2546
2547 return hr;
2548 }
2549
2550 static void topology_loader_resolve_complete(struct topoloader_context *context)
2551 {
2552 MF_TOPOLOGY_TYPE node_type;
2553 IMFTopologyNode *node;
2554 WORD i, node_count;
2555 HRESULT hr;
2556
2557 IMFTopology_GetNodeCount(context->output_topology, &node_count);
2558
2559 for (i = 0; i < node_count; ++i)
2560 {
2561 if (SUCCEEDED(IMFTopology_GetNode(context->output_topology, i, &node)))
2562 {
2563 IMFTopologyNode_GetNodeType(node, &node_type);
2564
2565 if (node_type == MF_TOPOLOGY_OUTPUT_NODE)
2566 {
2567 /* Set MF_TOPONODE_STREAMID for all outputs. */
2568 if (FAILED(IMFTopologyNode_GetItem(node, &MF_TOPONODE_STREAMID, NULL)))
2569 IMFTopologyNode_SetUINT32(node, &MF_TOPONODE_STREAMID, 0);
2570
2571 if (FAILED(hr = topology_loader_connect_d3d_aware_input(context, node)))
2572 WARN("Failed to connect D3D-aware input, hr %#lx.\n", hr);
2573 }
2574 else if (node_type == MF_TOPOLOGY_SOURCESTREAM_NODE)
2575 {
2576 /* Set MF_TOPONODE_MEDIASTART for all sources. */
2577 if (FAILED(IMFTopologyNode_GetItem(node, &MF_TOPONODE_MEDIASTART, NULL)))
2578 IMFTopologyNode_SetUINT64(node, &MF_TOPONODE_MEDIASTART, 0);
2579 }
2580
2581 IMFTopologyNode_Release(node);
2582 }
2583 }
2584 }
2585
2586 static HRESULT WINAPI topology_loader_Load(IMFTopoLoader *iface, IMFTopology *input_topology,
2587 IMFTopology **ret_topology, IMFTopology *current_topology)
2588 {
2589 struct topoloader_context context = { 0 };
2590 IMFTopology *output_topology;
2591 MF_TOPOLOGY_TYPE node_type;
2592 unsigned int layer_size;
2593 IMFTopologyNode *node;
2594 unsigned short i = 0;
2595 IMFStreamSink *sink;
2596 IUnknown *object;
2597 HRESULT hr = E_FAIL;
2598
2599 FIXME("%p, %p, %p, %p.\n", iface, input_topology, ret_topology, current_topology);
2600
2601 if (current_topology)
2602 FIXME("Current topology instance is ignored.\n");
2603
2604 /* Basic sanity checks for input topology:
2605
2606 - source nodes must have stream descriptor set;
2607 - sink nodes must be resolved to stream sink objects;
2608 */
2609 while (SUCCEEDED(IMFTopology_GetNode(input_topology, i++, &node)))
2610 {
2611 IMFTopologyNode_GetNodeType(node, &node_type);
2612
2613 switch (node_type)
2614 {
2615 case MF_TOPOLOGY_OUTPUT_NODE:
2616 if (SUCCEEDED(hr = IMFTopologyNode_GetObject(node, &object)))
2617 {
2618 /* Sinks must be bound beforehand. */
2619 if (FAILED(IUnknown_QueryInterface(object, &IID_IMFStreamSink, (void **)&sink)))
2620 hr = MF_E_TOPO_SINK_ACTIVATES_UNSUPPORTED;
2621 else if (sink)
2622 IMFStreamSink_Release(sink);
2623 IUnknown_Release(object);
2624 }
2625 break;
2626 case MF_TOPOLOGY_SOURCESTREAM_NODE:
2627 hr = IMFTopologyNode_GetItem(node, &MF_TOPONODE_STREAM_DESCRIPTOR, NULL);
2628 break;
2629 default:
2630 ;
2631 }
2632
2633 IMFTopologyNode_Release(node);
2634 if (FAILED(hr))
2635 return hr;
2636 }
2637
2638 if (FAILED(hr = MFCreateTopology(&output_topology)))
2639 return hr;
2640
2641 IMFTopology_CopyAllItems(input_topology, (IMFAttributes *)output_topology);
2642
2643 context.input_topology = input_topology;
2644 context.output_topology = output_topology;
2645 memset(&context.key, 0xff, sizeof(context.key));
2646
2647 /* Clone source nodes, use initial marker value. */
2648 i = 0;
2649 while (SUCCEEDED(IMFTopology_GetNode(input_topology, i++, &node)))
2650 {
2651 IMFTopologyNode_GetNodeType(node, &node_type);
2652
2653 if (node_type == MF_TOPOLOGY_SOURCESTREAM_NODE)
2654 {
2655 if (FAILED(hr = topology_loader_clone_node(&context, node, NULL, 0)))
2656 WARN("Failed to clone source node, hr %#lx.\n", hr);
2657 }
2658
2659 IMFTopologyNode_Release(node);
2660 }
2661
2662 for (context.marker = 0;; ++context.marker)
2663 {
2664 if (FAILED(hr = topology_loader_resolve_nodes(&context, &layer_size)))
2665 {
2666 WARN("Failed to resolve for marker %u, hr %#lx.\n", context.marker, hr);
2667 break;
2668 }
2669
2670 /* Reached last marker value. */
2671 if (!layer_size)
2672 break;
2673 }
2674
2675 if (FAILED(hr))
2676 IMFTopology_Release(output_topology);
2677 else
2678 {
2679 topology_loader_resolve_complete(&context);
2680 *ret_topology = output_topology;
2681 }
2682
2683 return hr;
2684 }
2685
2686 static const IMFTopoLoaderVtbl topologyloadervtbl =
2687 {
2688 topology_loader_QueryInterface,
2689 topology_loader_AddRef,
2690 topology_loader_Release,
2691 topology_loader_Load,
2692 };
2693
2694 /***********************************************************************
2695 * MFCreateTopoLoader (mf.@)
2696 */
2697 HRESULT WINAPI MFCreateTopoLoader(IMFTopoLoader **loader)
2698 {
2699 struct topology_loader *object;
2700
2701 TRACE("%p.\n", loader);
2702
2703 if (!loader)
2704 return E_POINTER;
2705
2706 if (!(object = calloc(1, sizeof(*object))))
2707 return E_OUTOFMEMORY;
2708
2709 object->IMFTopoLoader_iface.lpVtbl = &topologyloadervtbl;
2710 object->refcount = 1;
2711
2712 *loader = &object->IMFTopoLoader_iface;
2713
2714 return S_OK;
2715 }
2716
2717 static HRESULT WINAPI seq_source_QueryInterface(IMFSequencerSource *iface, REFIID riid, void **out)
2718 {
2719 struct seq_source *seq_source = impl_from_IMFSequencerSource(iface);
2720
2721 TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
2722
2723 *out = NULL;
2724
2725 if (IsEqualIID(riid, &IID_IMFSequencerSource) ||
2726 IsEqualIID(riid, &IID_IUnknown))
2727 {
2728 *out = &seq_source->IMFSequencerSource_iface;
2729 }
2730 else if (IsEqualIID(riid, &IID_IMFMediaSourceTopologyProvider))
2731 {
2732 *out = &seq_source->IMFMediaSourceTopologyProvider_iface;
2733 }
2734 else
2735 {
2736 WARN("Unimplemented %s.\n", debugstr_guid(riid));
2737 return E_NOINTERFACE;
2738 }
2739
2740 if (*out)
2741 IUnknown_AddRef((IUnknown *)*out);
2742
2743 return S_OK;
2744 }
2745
2746 static ULONG WINAPI seq_source_AddRef(IMFSequencerSource *iface)
2747 {
2748 struct seq_source *seq_source = impl_from_IMFSequencerSource(iface);
2749 ULONG refcount = InterlockedIncrement(&seq_source->refcount);
2750
2751 TRACE("%p, refcount %lu.\n", iface, refcount);
2752
2753 return refcount;
2754 }
2755
2756 static ULONG WINAPI seq_source_Release(IMFSequencerSource *iface)
2757 {
2758 struct seq_source *seq_source = impl_from_IMFSequencerSource(iface);
2759 ULONG refcount = InterlockedDecrement(&seq_source->refcount);
2760
2761 TRACE("%p, refcount %lu.\n", iface, refcount);
2762
2763 if (!refcount)
2764 free(seq_source);
2765
2766 return refcount;
2767 }
2768
2769 static HRESULT WINAPI seq_source_AppendTopology(IMFSequencerSource *iface, IMFTopology *topology,
2770 DWORD flags, MFSequencerElementId *id)
2771 {
2772 FIXME("%p, %p, %lx, %p.\n", iface, topology, flags, id);
2773
2774 return E_NOTIMPL;
2775 }
2776
2777 static HRESULT WINAPI seq_source_DeleteTopology(IMFSequencerSource *iface, MFSequencerElementId id)
2778 {
2779 FIXME("%p, %#lx.\n", iface, id);
2780
2781 return E_NOTIMPL;
2782 }
2783
2784 static HRESULT WINAPI seq_source_GetPresentationContext(IMFSequencerSource *iface,
2785 IMFPresentationDescriptor *descriptor, MFSequencerElementId *id, IMFTopology **topology)
2786 {
2787 FIXME("%p, %p, %p, %p.\n", iface, descriptor, id, topology);
2788
2789 return E_NOTIMPL;
2790 }
2791
2792 static HRESULT WINAPI seq_source_UpdateTopology(IMFSequencerSource *iface, MFSequencerElementId id,
2793 IMFTopology *topology)
2794 {
2795 FIXME("%p, %#lx, %p.\n", iface, id, topology);
2796
2797 return E_NOTIMPL;
2798 }
2799
2800 static HRESULT WINAPI seq_source_UpdateTopologyFlags(IMFSequencerSource *iface, MFSequencerElementId id, DWORD flags)
2801 {
2802 FIXME("%p, %#lx, %#lx.\n", iface, id, flags);
2803
2804 return E_NOTIMPL;
2805 }
2806
2807 static HRESULT WINAPI seq_source_topology_provider_QueryInterface(IMFMediaSourceTopologyProvider *iface, REFIID riid,
2808 void **obj)
2809 {
2810 struct seq_source *seq_source = impl_from_IMFMediaSourceTopologyProvider(iface);
2811 return IMFSequencerSource_QueryInterface(&seq_source->IMFSequencerSource_iface, riid, obj);
2812 }
2813
2814 static ULONG WINAPI seq_source_topology_provider_AddRef(IMFMediaSourceTopologyProvider *iface)
2815 {
2816 struct seq_source *seq_source = impl_from_IMFMediaSourceTopologyProvider(iface);
2817 return IMFSequencerSource_AddRef(&seq_source->IMFSequencerSource_iface);
2818 }
2819
2820 static ULONG WINAPI seq_source_topology_provider_Release(IMFMediaSourceTopologyProvider *iface)
2821 {
2822 struct seq_source *seq_source = impl_from_IMFMediaSourceTopologyProvider(iface);
2823 return IMFSequencerSource_Release(&seq_source->IMFSequencerSource_iface);
2824 }
2825
2826 static HRESULT WINAPI seq_source_topology_provider_GetMediaSourceTopology(IMFMediaSourceTopologyProvider *iface,
2827 IMFPresentationDescriptor *pd, IMFTopology **topology)
2828 {
2829 FIXME("%p, %p, %p.\n", iface, pd, topology);
2830
2831 return E_NOTIMPL;
2832 }
2833
2834 static const IMFMediaSourceTopologyProviderVtbl seq_source_topology_provider_vtbl =
2835 {
2836 seq_source_topology_provider_QueryInterface,
2837 seq_source_topology_provider_AddRef,
2838 seq_source_topology_provider_Release,
2839 seq_source_topology_provider_GetMediaSourceTopology,
2840 };
2841
2842 static const IMFSequencerSourceVtbl seqsourcevtbl =
2843 {
2844 seq_source_QueryInterface,
2845 seq_source_AddRef,
2846 seq_source_Release,
2847 seq_source_AppendTopology,
2848 seq_source_DeleteTopology,
2849 seq_source_GetPresentationContext,
2850 seq_source_UpdateTopology,
2851 seq_source_UpdateTopologyFlags,
2852 };
2853
2854 /***********************************************************************
2855 * MFCreateSequencerSource (mf.@)
2856 */
2857 HRESULT WINAPI MFCreateSequencerSource(IUnknown *reserved, IMFSequencerSource **seq_source)
2858 {
2859 struct seq_source *object;
2860
2861 TRACE("%p, %p.\n", reserved, seq_source);
2862
2863 if (!seq_source)
2864 return E_POINTER;
2865
2866 if (!(object = calloc(1, sizeof(*object))))
2867 return E_OUTOFMEMORY;
2868
2869 object->IMFSequencerSource_iface.lpVtbl = &seqsourcevtbl;
2870 object->IMFMediaSourceTopologyProvider_iface.lpVtbl = &seq_source_topology_provider_vtbl;
2871 object->refcount = 1;
2872
2873 *seq_source = &object->IMFSequencerSource_iface;
2874
2875 return S_OK;
2876 }
Windows API implementation