dlls/mf/topology.c

   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 %u.\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     heap_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 %u.\n", iface, refcount);
 271
 272     if (!refcount)
 273     {
 274         if (topology->attributes)
 275             IMFAttributes_Release(topology->attributes);
 276         topology_clear(topology);
 277         heap_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 %#x.\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     object = heap_alloc_zero(sizeof(*object));
 889     if (!object)
 890         return E_OUTOFMEMORY;
 891
 892     object->IMFTopology_iface.lpVtbl = &topologyvtbl;
 893     object->refcount = 1;
 894
 895     hr = MFCreateAttributes(&object->attributes, 0);
 896     if (FAILED(hr))
 897     {
 898         IMFTopology_Release(&object->IMFTopology_iface);
 899         return hr;
 900     }
 901
 902     object->id = topology_generate_id();
 903
 904     *topology = &object->IMFTopology_iface;
 905
 906     return S_OK;
 907 }
 908
 909 static HRESULT WINAPI topology_node_QueryInterface(IMFTopologyNode *iface, REFIID riid, void **out)
 910 {
 911     struct topology_node *node = impl_from_IMFTopologyNode(iface);
 912
 913     TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
 914
 915     if (IsEqualIID(riid, &IID_IMFTopologyNode) ||
 916             IsEqualIID(riid, &IID_IMFAttributes) ||
 917             IsEqualIID(riid, &IID_IUnknown))
 918     {
 919         *out = &node->IMFTopologyNode_iface;
 920         IMFTopologyNode_AddRef(iface);
 921         return S_OK;
 922     }
 923
 924     WARN("Unsupported interface %s.\n", debugstr_guid(riid));
 925     *out = NULL;
 926
 927     return E_NOINTERFACE;
 928 }
 929
 930 static ULONG WINAPI topology_node_AddRef(IMFTopologyNode *iface)
 931 {
 932     struct topology_node *node = impl_from_IMFTopologyNode(iface);
 933     ULONG refcount = InterlockedIncrement(&node->refcount);
 934
 935     TRACE("%p, refcount %u.\n", iface, refcount);
 936
 937     return refcount;
 938 }
 939
 940 static ULONG WINAPI topology_node_Release(IMFTopologyNode *iface)
 941 {
 942     struct topology_node *node = impl_from_IMFTopologyNode(iface);
 943     ULONG refcount = InterlockedDecrement(&node->refcount);
 944     unsigned int i;
 945
 946     TRACE("%p, refcount %u.\n", iface, refcount);
 947
 948     if (!refcount)
 949     {
 950         if (node->object)
 951             IUnknown_Release(node->object);
 952         if (node->input_type)
 953             IMFMediaType_Release(node->input_type);
 954         for (i = 0; i < node->inputs.count; ++i)
 955         {
 956             if (node->inputs.streams[i].preferred_type)
 957                 IMFMediaType_Release(node->inputs.streams[i].preferred_type);
 958         }
 959         for (i = 0; i < node->outputs.count; ++i)
 960         {
 961             if (node->outputs.streams[i].preferred_type)
 962                 IMFMediaType_Release(node->outputs.streams[i].preferred_type);
 963         }
 964         heap_free(node->inputs.streams);
 965         heap_free(node->outputs.streams);
 966         IMFAttributes_Release(node->attributes);
 967         DeleteCriticalSection(&node->cs);
 968         heap_free(node);
 969     }
 970
 971     return refcount;
 972 }
 973
 974 static HRESULT WINAPI topology_node_GetItem(IMFTopologyNode *iface, REFGUID key, PROPVARIANT *value)
 975 {
 976     struct topology_node *node = impl_from_IMFTopologyNode(iface);
 977
 978     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
 979
 980     return IMFAttributes_GetItem(node->attributes, key, value);
 981 }
 982
 983 static HRESULT WINAPI topology_node_GetItemType(IMFTopologyNode *iface, REFGUID key, MF_ATTRIBUTE_TYPE *type)
 984 {
 985     struct topology_node *node = impl_from_IMFTopologyNode(iface);
 986
 987     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), type);
 988
 989     return IMFAttributes_GetItemType(node->attributes, key, type);
 990 }
 991
 992 static HRESULT WINAPI topology_node_CompareItem(IMFTopologyNode *iface, REFGUID key, REFPROPVARIANT value, BOOL *result)
 993 {
 994     struct topology_node *node = impl_from_IMFTopologyNode(iface);
 995
 996     TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), value, result);
 997
 998     return IMFAttributes_CompareItem(node->attributes, key, value, result);
 999 }
1000
1001 static HRESULT WINAPI topology_node_Compare(IMFTopologyNode *iface, IMFAttributes *theirs,
1002         MF_ATTRIBUTES_MATCH_TYPE type, BOOL *result)
1003 {
1004     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1005
1006     TRACE("%p, %p, %d, %p.\n", iface, theirs, type, result);
1007
1008     return IMFAttributes_Compare(node->attributes, theirs, type, result);
1009 }
1010
1011 static HRESULT WINAPI topology_node_GetUINT32(IMFTopologyNode *iface, REFGUID key, UINT32 *value)
1012 {
1013     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1014
1015     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1016
1017     return IMFAttributes_GetUINT32(node->attributes, key, value);
1018 }
1019
1020 static HRESULT WINAPI topology_node_GetUINT64(IMFTopologyNode *iface, REFGUID key, UINT64 *value)
1021 {
1022     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1023
1024     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1025
1026     return IMFAttributes_GetUINT64(node->attributes, key, value);
1027 }
1028
1029 static HRESULT WINAPI topology_node_GetDouble(IMFTopologyNode *iface, REFGUID key, double *value)
1030 {
1031     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1032
1033     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1034
1035     return IMFAttributes_GetDouble(node->attributes, key, value);
1036 }
1037
1038 static HRESULT WINAPI topology_node_GetGUID(IMFTopologyNode *iface, REFGUID key, GUID *value)
1039 {
1040     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1041
1042     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1043
1044     return IMFAttributes_GetGUID(node->attributes, key, value);
1045 }
1046
1047 static HRESULT WINAPI topology_node_GetStringLength(IMFTopologyNode *iface, REFGUID key, UINT32 *length)
1048 {
1049     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1050
1051     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), length);
1052
1053     return IMFAttributes_GetStringLength(node->attributes, key, length);
1054 }
1055
1056 static HRESULT WINAPI topology_node_GetString(IMFTopologyNode *iface, REFGUID key, WCHAR *value,
1057         UINT32 size, UINT32 *length)
1058 {
1059     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1060
1061     TRACE("%p, %s, %p, %d, %p.\n", iface, debugstr_guid(key), value, size, length);
1062
1063     return IMFAttributes_GetString(node->attributes, key, value, size, length);
1064 }
1065
1066 static HRESULT WINAPI topology_node_GetAllocatedString(IMFTopologyNode *iface, REFGUID key,
1067         WCHAR **value, UINT32 *length)
1068 {
1069     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1070
1071     TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), value, length);
1072
1073     return IMFAttributes_GetAllocatedString(node->attributes, key, value, length);
1074 }
1075
1076 static HRESULT WINAPI topology_node_GetBlobSize(IMFTopologyNode *iface, REFGUID key, UINT32 *size)
1077 {
1078     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1079
1080     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), size);
1081
1082     return IMFAttributes_GetBlobSize(node->attributes, key, size);
1083 }
1084
1085 static HRESULT WINAPI topology_node_GetBlob(IMFTopologyNode *iface, REFGUID key, UINT8 *buf,
1086         UINT32 bufsize, UINT32 *blobsize)
1087 {
1088     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1089
1090     TRACE("%p, %s, %p, %d, %p.\n", iface, debugstr_guid(key), buf, bufsize, blobsize);
1091
1092     return IMFAttributes_GetBlob(node->attributes, key, buf, bufsize, blobsize);
1093 }
1094
1095 static HRESULT WINAPI topology_node_GetAllocatedBlob(IMFTopologyNode *iface, REFGUID key, UINT8 **buf, UINT32 *size)
1096 {
1097     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1098
1099     TRACE("%p, %s, %p, %p.\n", iface, debugstr_guid(key), buf, size);
1100
1101     return IMFAttributes_GetAllocatedBlob(node->attributes, key, buf, size);
1102 }
1103
1104 static HRESULT WINAPI topology_node_GetUnknown(IMFTopologyNode *iface, REFGUID key, REFIID riid, void **ppv)
1105 {
1106     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1107
1108     TRACE("%p, %s, %s, %p.\n", iface, debugstr_guid(key), debugstr_guid(riid), ppv);
1109
1110     return IMFAttributes_GetUnknown(node->attributes, key, riid, ppv);
1111 }
1112
1113 static HRESULT WINAPI topology_node_SetItem(IMFTopologyNode *iface, REFGUID key, REFPROPVARIANT value)
1114 {
1115     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1116
1117     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), value);
1118
1119     return IMFAttributes_SetItem(node->attributes, key, value);
1120 }
1121
1122 static HRESULT WINAPI topology_node_DeleteItem(IMFTopologyNode *iface, REFGUID key)
1123 {
1124     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1125
1126     TRACE("%p, %s.\n", iface, debugstr_guid(key));
1127
1128     return IMFAttributes_DeleteItem(node->attributes, key);
1129 }
1130
1131 static HRESULT WINAPI topology_node_DeleteAllItems(IMFTopologyNode *iface)
1132 {
1133     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1134
1135     TRACE("%p.\n", iface);
1136
1137     return IMFAttributes_DeleteAllItems(node->attributes);
1138 }
1139
1140 static HRESULT WINAPI topology_node_SetUINT32(IMFTopologyNode *iface, REFGUID key, UINT32 value)
1141 {
1142     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1143
1144     TRACE("%p, %s, %d.\n", iface, debugstr_guid(key), value);
1145
1146     return IMFAttributes_SetUINT32(node->attributes, key, value);
1147 }
1148
1149 static HRESULT WINAPI topology_node_SetUINT64(IMFTopologyNode *iface, REFGUID key, UINT64 value)
1150 {
1151     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1152
1153     TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), wine_dbgstr_longlong(value));
1154
1155     return IMFAttributes_SetUINT64(node->attributes, key, value);
1156 }
1157
1158 static HRESULT WINAPI topology_node_SetDouble(IMFTopologyNode *iface, REFGUID key, double value)
1159 {
1160     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1161
1162     TRACE("%p, %s, %f.\n", iface, debugstr_guid(key), value);
1163
1164     return IMFAttributes_SetDouble(node->attributes, key, value);
1165 }
1166
1167 static HRESULT WINAPI topology_node_SetGUID(IMFTopologyNode *iface, REFGUID key, REFGUID value)
1168 {
1169     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1170
1171     TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), debugstr_guid(value));
1172
1173     return IMFAttributes_SetGUID(node->attributes, key, value);
1174 }
1175
1176 static HRESULT WINAPI topology_node_SetString(IMFTopologyNode *iface, REFGUID key, const WCHAR *value)
1177 {
1178     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1179
1180     TRACE("%p, %s, %s.\n", iface, debugstr_guid(key), debugstr_w(value));
1181
1182     return IMFAttributes_SetString(node->attributes, key, value);
1183 }
1184
1185 static HRESULT WINAPI topology_node_SetBlob(IMFTopologyNode *iface, REFGUID key, const UINT8 *buf, UINT32 size)
1186 {
1187     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1188
1189     TRACE("%p, %s, %p, %d.\n", iface, debugstr_guid(key), buf, size);
1190
1191     return IMFAttributes_SetBlob(node->attributes, key, buf, size);
1192 }
1193
1194 static HRESULT WINAPI topology_node_SetUnknown(IMFTopologyNode *iface, REFGUID key, IUnknown *unknown)
1195 {
1196     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1197
1198     TRACE("%p, %s, %p.\n", iface, debugstr_guid(key), unknown);
1199
1200     return IMFAttributes_SetUnknown(node->attributes, key, unknown);
1201 }
1202
1203 static HRESULT WINAPI topology_node_LockStore(IMFTopologyNode *iface)
1204 {
1205     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1206
1207     TRACE("%p.\n", iface);
1208
1209     return IMFAttributes_LockStore(node->attributes);
1210 }
1211
1212 static HRESULT WINAPI topology_node_UnlockStore(IMFTopologyNode *iface)
1213 {
1214     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1215
1216     TRACE("%p.\n", iface);
1217
1218     return IMFAttributes_UnlockStore(node->attributes);
1219 }
1220
1221 static HRESULT WINAPI topology_node_GetCount(IMFTopologyNode *iface, UINT32 *count)
1222 {
1223     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1224
1225     TRACE("%p, %p.\n", iface, count);
1226
1227     return IMFAttributes_GetCount(node->attributes, count);
1228 }
1229
1230 static HRESULT WINAPI topology_node_GetItemByIndex(IMFTopologyNode *iface, UINT32 index, GUID *key, PROPVARIANT *value)
1231 {
1232     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1233
1234     TRACE("%p, %u, %p, %p.\n", iface, index, key, value);
1235
1236     return IMFAttributes_GetItemByIndex(node->attributes, index, key, value);
1237 }
1238
1239 static HRESULT WINAPI topology_node_CopyAllItems(IMFTopologyNode *iface, IMFAttributes *dest)
1240 {
1241     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1242
1243     TRACE("%p, %p.\n", iface, dest);
1244
1245     return IMFAttributes_CopyAllItems(node->attributes, dest);
1246 }
1247
1248 static HRESULT topology_node_set_object(struct topology_node *node, IUnknown *object)
1249 {
1250     static const GUID *iids[3] = { &IID_IPersist, &IID_IPersistStorage, &IID_IPersistPropertyBag };
1251     IPersist *persist = NULL;
1252     BOOL has_object_id;
1253     GUID object_id;
1254     unsigned int i;
1255     HRESULT hr;
1256
1257     has_object_id = IMFAttributes_GetGUID(node->attributes, &MF_TOPONODE_TRANSFORM_OBJECTID, &object_id) == S_OK;
1258
1259     if (object && !has_object_id)
1260     {
1261         for (i = 0; i < ARRAY_SIZE(iids); ++i)
1262         {
1263             persist = NULL;
1264             if (SUCCEEDED(hr = IUnknown_QueryInterface(object, iids[i], (void **)&persist)))
1265                 break;
1266         }
1267
1268         if (persist)
1269         {
1270             if (FAILED(hr = IPersist_GetClassID(persist, &object_id)))
1271             {
1272                 IPersist_Release(persist);
1273                 persist = NULL;
1274             }
1275         }
1276     }
1277
1278     EnterCriticalSection(&node->cs);
1279
1280     if (node->object)
1281         IUnknown_Release(node->object);
1282     node->object = object;
1283     if (node->object)
1284         IUnknown_AddRef(node->object);
1285
1286     if (persist)
1287         IMFAttributes_SetGUID(node->attributes, &MF_TOPONODE_TRANSFORM_OBJECTID, &object_id);
1288
1289     LeaveCriticalSection(&node->cs);
1290
1291     if (persist)
1292         IPersist_Release(persist);
1293
1294     return S_OK;
1295 }
1296
1297 static HRESULT WINAPI topology_node_SetObject(IMFTopologyNode *iface, IUnknown *object)
1298 {
1299     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1300
1301     TRACE("%p, %p.\n", iface, object);
1302
1303     return topology_node_set_object(node, object);
1304 }
1305
1306 static HRESULT WINAPI topology_node_GetObject(IMFTopologyNode *iface, IUnknown **object)
1307 {
1308     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1309
1310     TRACE("%p, %p.\n", iface, object);
1311
1312     if (!object)
1313         return E_POINTER;
1314
1315     EnterCriticalSection(&node->cs);
1316
1317     *object = node->object;
1318     if (*object)
1319         IUnknown_AddRef(*object);
1320
1321     LeaveCriticalSection(&node->cs);
1322
1323     return *object ? S_OK : E_FAIL;
1324 }
1325
1326 static HRESULT WINAPI topology_node_GetNodeType(IMFTopologyNode *iface, MF_TOPOLOGY_TYPE *node_type)
1327 {
1328     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1329
1330     TRACE("%p, %p.\n", iface, node_type);
1331
1332     *node_type = node->node_type;
1333
1334     return S_OK;
1335 }
1336
1337 static HRESULT WINAPI topology_node_GetTopoNodeID(IMFTopologyNode *iface, TOPOID *id)
1338 {
1339     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1340
1341     TRACE("%p, %p.\n", iface, id);
1342
1343     *id = node->id;
1344
1345     return S_OK;
1346 }
1347
1348 static HRESULT WINAPI topology_node_SetTopoNodeID(IMFTopologyNode *iface, TOPOID id)
1349 {
1350     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1351
1352     TRACE("%p, %s.\n", iface, wine_dbgstr_longlong(id));
1353
1354     node->id = id;
1355
1356     return S_OK;
1357 }
1358
1359 static HRESULT WINAPI topology_node_GetInputCount(IMFTopologyNode *iface, DWORD *count)
1360 {
1361     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1362
1363     TRACE("%p, %p.\n", iface, count);
1364
1365     *count = node->inputs.count;
1366
1367     return S_OK;
1368 }
1369
1370 static HRESULT WINAPI topology_node_GetOutputCount(IMFTopologyNode *iface, DWORD *count)
1371 {
1372     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1373
1374     TRACE("%p, %p.\n", iface, count);
1375
1376     *count = node->outputs.count;
1377
1378     return S_OK;
1379 }
1380
1381 static void topology_node_set_stream_type(struct node_stream *stream, IMFMediaType *mediatype)
1382 {
1383     if (stream->preferred_type)
1384         IMFMediaType_Release(stream->preferred_type);
1385     stream->preferred_type = mediatype;
1386     if (stream->preferred_type)
1387         IMFMediaType_AddRef(stream->preferred_type);
1388 }
1389
1390 static HRESULT topology_node_connect_output(struct topology_node *node, DWORD output_index,
1391         struct topology_node *connection, DWORD input_index)
1392 {
1393     struct node_stream *stream;
1394     HRESULT hr;
1395
1396     if (node->node_type == MF_TOPOLOGY_OUTPUT_NODE || connection->node_type == MF_TOPOLOGY_SOURCESTREAM_NODE)
1397          return E_FAIL;
1398
1399     EnterCriticalSection(&node->cs);
1400     EnterCriticalSection(&connection->cs);
1401
1402     topology_node_disconnect_output(node, output_index);
1403     if (input_index < connection->inputs.count)
1404     {
1405         stream = &connection->inputs.streams[input_index];
1406         if (stream->connection)
1407             topology_node_disconnect_output(stream->connection, stream->connection_stream);
1408     }
1409
1410     hr = topology_node_reserve_streams(&node->outputs, output_index);
1411     if (SUCCEEDED(hr))
1412     {
1413         size_t old_count = connection->inputs.count;
1414         hr = topology_node_reserve_streams(&connection->inputs, input_index);
1415         if (SUCCEEDED(hr) && !old_count && connection->input_type)
1416         {
1417             topology_node_set_stream_type(connection->inputs.streams, connection->input_type);
1418             IMFMediaType_Release(connection->input_type);
1419             connection->input_type = NULL;
1420         }
1421     }
1422
1423     if (SUCCEEDED(hr))
1424     {
1425         node->outputs.streams[output_index].connection = connection;
1426         IMFTopologyNode_AddRef(&node->outputs.streams[output_index].connection->IMFTopologyNode_iface);
1427         node->outputs.streams[output_index].connection_stream = input_index;
1428         connection->inputs.streams[input_index].connection = node;
1429         IMFTopologyNode_AddRef(&connection->inputs.streams[input_index].connection->IMFTopologyNode_iface);
1430         connection->inputs.streams[input_index].connection_stream = output_index;
1431     }
1432
1433     LeaveCriticalSection(&connection->cs);
1434     LeaveCriticalSection(&node->cs);
1435
1436     return hr;
1437 }
1438
1439 static HRESULT WINAPI topology_node_ConnectOutput(IMFTopologyNode *iface, DWORD output_index,
1440         IMFTopologyNode *peer, DWORD input_index)
1441 {
1442     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1443     struct topology_node *connection = unsafe_impl_from_IMFTopologyNode(peer);
1444
1445     TRACE("%p, %u, %p, %u.\n", iface, output_index, peer, input_index);
1446
1447     if (!connection)
1448     {
1449         WARN("External node implementations are not supported.\n");
1450         return E_UNEXPECTED;
1451     }
1452
1453     return topology_node_connect_output(node, output_index, connection, input_index);
1454 }
1455
1456 static HRESULT WINAPI topology_node_DisconnectOutput(IMFTopologyNode *iface, DWORD output_index)
1457 {
1458     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1459
1460     TRACE("%p, %u.\n", iface, output_index);
1461
1462     return topology_node_disconnect_output(node, output_index);
1463 }
1464
1465 static HRESULT WINAPI topology_node_GetInput(IMFTopologyNode *iface, DWORD input_index, IMFTopologyNode **ret,
1466         DWORD *output_index)
1467 {
1468     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1469     HRESULT hr = S_OK;
1470
1471     TRACE("%p, %u, %p, %p.\n", iface, input_index, ret, output_index);
1472
1473     EnterCriticalSection(&node->cs);
1474
1475     if (input_index < node->inputs.count)
1476     {
1477         const struct node_stream *stream = &node->inputs.streams[input_index];
1478
1479         if (stream->connection)
1480         {
1481             *ret = &stream->connection->IMFTopologyNode_iface;
1482             IMFTopologyNode_AddRef(*ret);
1483             *output_index = stream->connection_stream;
1484         }
1485         else
1486             hr = MF_E_NOT_FOUND;
1487     }
1488     else
1489         hr = E_INVALIDARG;
1490
1491     LeaveCriticalSection(&node->cs);
1492
1493     return hr;
1494 }
1495
1496 static HRESULT WINAPI topology_node_GetOutput(IMFTopologyNode *iface, DWORD output_index, IMFTopologyNode **ret,
1497         DWORD *input_index)
1498 {
1499     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1500     HRESULT hr = S_OK;
1501
1502     TRACE("%p, %u, %p, %p.\n", iface, output_index, ret, input_index);
1503
1504     EnterCriticalSection(&node->cs);
1505
1506     if (output_index < node->outputs.count)
1507     {
1508         const struct node_stream *stream = &node->outputs.streams[output_index];
1509
1510         if (stream->connection)
1511         {
1512             *ret = &stream->connection->IMFTopologyNode_iface;
1513             IMFTopologyNode_AddRef(*ret);
1514             *input_index = stream->connection_stream;
1515         }
1516         else
1517             hr = MF_E_NOT_FOUND;
1518     }
1519     else
1520         hr = E_INVALIDARG;
1521
1522     LeaveCriticalSection(&node->cs);
1523
1524     return hr;
1525 }
1526
1527 static HRESULT WINAPI topology_node_SetOutputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType *mediatype)
1528 {
1529     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1530     HRESULT hr = S_OK;
1531
1532     TRACE("%p, %u, %p.\n", iface, index, mediatype);
1533
1534     EnterCriticalSection(&node->cs);
1535
1536     if (node->node_type != MF_TOPOLOGY_OUTPUT_NODE)
1537     {
1538         if (SUCCEEDED(hr = topology_node_reserve_streams(&node->outputs, index)))
1539             topology_node_set_stream_type(&node->outputs.streams[index], mediatype);
1540     }
1541     else
1542         hr = E_NOTIMPL;
1543
1544     LeaveCriticalSection(&node->cs);
1545
1546     return hr;
1547 }
1548
1549 static HRESULT topology_node_get_pref_type(struct node_streams *streams, unsigned int index, IMFMediaType **mediatype)
1550 {
1551     *mediatype = streams->streams[index].preferred_type;
1552     if (*mediatype)
1553     {
1554         IMFMediaType_AddRef(*mediatype);
1555         return S_OK;
1556     }
1557
1558     return E_FAIL;
1559 }
1560
1561 static HRESULT WINAPI topology_node_GetOutputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType **mediatype)
1562 {
1563     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1564     HRESULT hr = S_OK;
1565
1566     TRACE("%p, %u, %p.\n", iface, index, mediatype);
1567
1568     EnterCriticalSection(&node->cs);
1569
1570     if (index < node->outputs.count)
1571         hr = topology_node_get_pref_type(&node->outputs, index, mediatype);
1572     else
1573         hr = E_INVALIDARG;
1574
1575     LeaveCriticalSection(&node->cs);
1576
1577     return hr;
1578 }
1579
1580 static HRESULT WINAPI topology_node_SetInputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType *mediatype)
1581 {
1582     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1583     HRESULT hr = S_OK;
1584
1585     TRACE("%p, %u, %p.\n", iface, index, mediatype);
1586
1587     EnterCriticalSection(&node->cs);
1588
1589     switch (node->node_type)
1590     {
1591         case MF_TOPOLOGY_TEE_NODE:
1592             if (index)
1593             {
1594                 hr = MF_E_INVALIDTYPE;
1595                 break;
1596             }
1597             if (node->inputs.count)
1598                 topology_node_set_stream_type(&node->inputs.streams[index], mediatype);
1599             else
1600             {
1601                 if (node->input_type)
1602                     IMFMediaType_Release(node->input_type);
1603                 node->input_type = mediatype;
1604                 if (node->input_type)
1605                     IMFMediaType_AddRef(node->input_type);
1606             }
1607             break;
1608         case MF_TOPOLOGY_SOURCESTREAM_NODE:
1609             hr = E_NOTIMPL;
1610             break;
1611         default:
1612             if (SUCCEEDED(hr = topology_node_reserve_streams(&node->inputs, index)))
1613                 topology_node_set_stream_type(&node->inputs.streams[index], mediatype);
1614     }
1615
1616     LeaveCriticalSection(&node->cs);
1617
1618     return hr;
1619 }
1620
1621 static HRESULT WINAPI topology_node_GetInputPrefType(IMFTopologyNode *iface, DWORD index, IMFMediaType **mediatype)
1622 {
1623     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1624     HRESULT hr = S_OK;
1625
1626     TRACE("%p, %u, %p.\n", iface, index, mediatype);
1627
1628     EnterCriticalSection(&node->cs);
1629
1630     if (index < node->inputs.count)
1631     {
1632         hr = topology_node_get_pref_type(&node->inputs, index, mediatype);
1633     }
1634     else if (node->node_type == MF_TOPOLOGY_TEE_NODE && node->input_type)
1635     {
1636         *mediatype = node->input_type;
1637         IMFMediaType_AddRef(*mediatype);
1638     }
1639     else
1640         hr = E_INVALIDARG;
1641
1642     LeaveCriticalSection(&node->cs);
1643
1644     return hr;
1645 }
1646
1647 static HRESULT WINAPI topology_node_CloneFrom(IMFTopologyNode *iface, IMFTopologyNode *src_node)
1648 {
1649     struct topology_node *node = impl_from_IMFTopologyNode(iface);
1650     MF_TOPOLOGY_TYPE node_type;
1651     IMFMediaType *mediatype;
1652     IUnknown *object;
1653     DWORD count, i;
1654     TOPOID topoid;
1655     HRESULT hr;
1656
1657     TRACE("%p, %p.\n", iface, src_node);
1658
1659     if (FAILED(hr = IMFTopologyNode_GetNodeType(src_node, &node_type)))
1660         return hr;
1661
1662     if (node->node_type != node_type)
1663         return MF_E_INVALIDREQUEST;
1664
1665     if (FAILED(hr = IMFTopologyNode_GetTopoNodeID(src_node, &topoid)))
1666         return hr;
1667
1668     object = NULL;
1669     IMFTopologyNode_GetObject(src_node, &object);
1670
1671     EnterCriticalSection(&node->cs);
1672
1673     hr = IMFTopologyNode_CopyAllItems(src_node, node->attributes);
1674
1675     if (SUCCEEDED(hr))
1676         hr = topology_node_set_object(node, object);
1677
1678     if (SUCCEEDED(hr))
1679         node->id = topoid;
1680
1681     if (SUCCEEDED(IMFTopologyNode_GetInputCount(src_node, &count)))
1682     {
1683         for (i = 0; i < count; ++i)
1684         {
1685             if (SUCCEEDED(IMFTopologyNode_GetInputPrefType(src_node, i, &mediatype)))
1686             {
1687                 IMFTopologyNode_SetInputPrefType(iface, i, mediatype);
1688                 IMFMediaType_Release(mediatype);
1689             }
1690         }
1691     }
1692
1693     if (SUCCEEDED(IMFTopologyNode_GetOutputCount(src_node, &count)))
1694     {
1695         for (i = 0; i < count; ++i)
1696         {
1697             if (SUCCEEDED(IMFTopologyNode_GetOutputPrefType(src_node, i, &mediatype)))
1698             {
1699                 IMFTopologyNode_SetOutputPrefType(iface, i, mediatype);
1700                 IMFMediaType_Release(mediatype);
1701             }
1702         }
1703     }
1704
1705     LeaveCriticalSection(&node->cs);
1706
1707     if (object)
1708         IUnknown_Release(object);
1709
1710     return hr;
1711 }
1712
1713 static const IMFTopologyNodeVtbl topologynodevtbl =
1714 {
1715     topology_node_QueryInterface,
1716     topology_node_AddRef,
1717     topology_node_Release,
1718     topology_node_GetItem,
1719     topology_node_GetItemType,
1720     topology_node_CompareItem,
1721     topology_node_Compare,
1722     topology_node_GetUINT32,
1723     topology_node_GetUINT64,
1724     topology_node_GetDouble,
1725     topology_node_GetGUID,
1726     topology_node_GetStringLength,
1727     topology_node_GetString,
1728     topology_node_GetAllocatedString,
1729     topology_node_GetBlobSize,
1730     topology_node_GetBlob,
1731     topology_node_GetAllocatedBlob,
1732     topology_node_GetUnknown,
1733     topology_node_SetItem,
1734     topology_node_DeleteItem,
1735     topology_node_DeleteAllItems,
1736     topology_node_SetUINT32,
1737     topology_node_SetUINT64,
1738     topology_node_SetDouble,
1739     topology_node_SetGUID,
1740     topology_node_SetString,
1741     topology_node_SetBlob,
1742     topology_node_SetUnknown,
1743     topology_node_LockStore,
1744     topology_node_UnlockStore,
1745     topology_node_GetCount,
1746     topology_node_GetItemByIndex,
1747     topology_node_CopyAllItems,
1748     topology_node_SetObject,
1749     topology_node_GetObject,
1750     topology_node_GetNodeType,
1751     topology_node_GetTopoNodeID,
1752     topology_node_SetTopoNodeID,
1753     topology_node_GetInputCount,
1754     topology_node_GetOutputCount,
1755     topology_node_ConnectOutput,
1756     topology_node_DisconnectOutput,
1757     topology_node_GetInput,
1758     topology_node_GetOutput,
1759     topology_node_SetOutputPrefType,
1760     topology_node_GetOutputPrefType,
1761     topology_node_SetInputPrefType,
1762     topology_node_GetInputPrefType,
1763     topology_node_CloneFrom,
1764 };
1765
1766 static HRESULT create_topology_node(MF_TOPOLOGY_TYPE node_type, struct topology_node **node)
1767 {
1768     HRESULT hr;
1769
1770     *node = heap_alloc_zero(sizeof(**node));
1771     if (!*node)
1772         return E_OUTOFMEMORY;
1773
1774     (*node)->IMFTopologyNode_iface.lpVtbl = &topologynodevtbl;
1775     (*node)->refcount = 1;
1776     (*node)->node_type = node_type;
1777     hr = MFCreateAttributes(&(*node)->attributes, 0);
1778     if (FAILED(hr))
1779     {
1780         heap_free(*node);
1781         return hr;
1782     }
1783     (*node)->id = ((TOPOID)GetCurrentProcessId() << 32) | InterlockedIncrement(&next_node_id);
1784     InitializeCriticalSection(&(*node)->cs);
1785
1786     return S_OK;
1787 }
1788
1789 HRESULT topology_node_get_object(IMFTopologyNode *node, REFIID riid, void **obj)
1790 {
1791     IUnknown *unk;
1792     HRESULT hr;
1793
1794     *obj = NULL;
1795
1796     if (SUCCEEDED(hr = IMFTopologyNode_GetObject(node, &unk)))
1797     {
1798         hr = IUnknown_QueryInterface(unk, riid, obj);
1799         IUnknown_Release(unk);
1800     }
1801
1802     return hr;
1803 }
1804
1805 /***********************************************************************
1806  *      MFCreateTopologyNode (mf.@)
1807  */
1808 HRESULT WINAPI MFCreateTopologyNode(MF_TOPOLOGY_TYPE node_type, IMFTopologyNode **node)
1809 {
1810     struct topology_node *object;
1811     HRESULT hr;
1812
1813     TRACE("%d, %p.\n", node_type, node);
1814
1815     if (!node)
1816         return E_POINTER;
1817
1818     hr = create_topology_node(node_type, &object);
1819     if (SUCCEEDED(hr))
1820         *node = &object->IMFTopologyNode_iface;
1821
1822     return hr;
1823 }
1824
1825 /***********************************************************************
1826  *      MFGetTopoNodeCurrentType (mf.@)
1827  */
1828 HRESULT WINAPI MFGetTopoNodeCurrentType(IMFTopologyNode *node, DWORD stream, BOOL output, IMFMediaType **type)
1829 {
1830     IMFMediaTypeHandler *type_handler;
1831     MF_TOPOLOGY_TYPE node_type;
1832     IMFStreamSink *stream_sink;
1833     IMFStreamDescriptor *sd;
1834     IMFTransform *transform;
1835     UINT32 primary_output;
1836     HRESULT hr;
1837
1838     TRACE("%p, %u, %d, %p.\n", node, stream, output, type);
1839
1840     if (FAILED(hr = IMFTopologyNode_GetNodeType(node, &node_type)))
1841         return hr;
1842
1843     switch (node_type)
1844     {
1845         case MF_TOPOLOGY_OUTPUT_NODE:
1846             if (SUCCEEDED(topology_node_get_object(node, &IID_IMFStreamSink, (void **)&stream_sink)))
1847             {
1848                 hr = IMFStreamSink_GetMediaTypeHandler(stream_sink, &type_handler);
1849                 IMFStreamSink_Release(stream_sink);
1850
1851                 if (SUCCEEDED(hr))
1852                 {
1853                     hr = IMFMediaTypeHandler_GetCurrentMediaType(type_handler, type);
1854                     IMFMediaTypeHandler_Release(type_handler);
1855                 }
1856             }
1857             break;
1858         case MF_TOPOLOGY_SOURCESTREAM_NODE:
1859             if (FAILED(hr = IMFTopologyNode_GetUnknown(node, &MF_TOPONODE_STREAM_DESCRIPTOR, &IID_IMFStreamDescriptor,
1860                     (void **)&sd)))
1861             {
1862                 return hr;
1863             }
1864
1865             hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, &type_handler);
1866             IMFStreamDescriptor_Release(sd);
1867             if (SUCCEEDED(hr))
1868             {
1869                 hr = IMFMediaTypeHandler_GetCurrentMediaType(type_handler, type);
1870                 IMFMediaTypeHandler_Release(type_handler);
1871             }
1872             break;
1873         case MF_TOPOLOGY_TRANSFORM_NODE:
1874             if (SUCCEEDED(hr = topology_node_get_object(node, &IID_IMFTransform, (void **)&transform)))
1875             {
1876                 if (output)
1877                     hr = IMFTransform_GetOutputCurrentType(transform, stream, type);
1878                 else
1879                     hr = IMFTransform_GetInputCurrentType(transform, stream, type);
1880                 IMFTransform_Release(transform);
1881             }
1882             break;
1883         case MF_TOPOLOGY_TEE_NODE:
1884             if (SUCCEEDED(hr = IMFTopologyNode_GetInputPrefType(node, 0, type)))
1885                 break;
1886
1887             if (FAILED(IMFTopologyNode_GetUINT32(node, &MF_TOPONODE_PRIMARYOUTPUT, &primary_output)))
1888                 primary_output = 0;
1889
1890             hr = IMFTopologyNode_GetOutputPrefType(node, primary_output, type);
1891             break;
1892         default:
1893             ;
1894     }
1895
1896     return hr;
1897 }
1898
1899 static HRESULT WINAPI topology_loader_QueryInterface(IMFTopoLoader *iface, REFIID riid, void **out)
1900 {
1901     TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
1902
1903     if (IsEqualIID(riid, &IID_IMFTopoLoader) ||
1904             IsEqualIID(riid, &IID_IUnknown))
1905     {
1906         *out = iface;
1907         IMFTopoLoader_AddRef(iface);
1908         return S_OK;
1909     }
1910
1911     WARN("Unsupported %s.\n", debugstr_guid(riid));
1912     *out = NULL;
1913     return E_NOINTERFACE;
1914 }
1915
1916 static ULONG WINAPI topology_loader_AddRef(IMFTopoLoader *iface)
1917 {
1918     struct topology_loader *loader = impl_from_IMFTopoLoader(iface);
1919     ULONG refcount = InterlockedIncrement(&loader->refcount);
1920
1921     TRACE("%p, refcount %u.\n", iface, refcount);
1922
1923     return refcount;
1924 }
1925
1926 static ULONG WINAPI topology_loader_Release(IMFTopoLoader *iface)
1927 {
1928     struct topology_loader *loader = impl_from_IMFTopoLoader(iface);
1929     ULONG refcount = InterlockedDecrement(&loader->refcount);
1930
1931     TRACE("%p, refcount %u.\n", iface, refcount);
1932
1933     if (!refcount)
1934     {
1935         heap_free(loader);
1936     }
1937
1938     return refcount;
1939 }
1940
1941 struct topoloader_context
1942 {
1943     IMFTopology *input_topology;
1944     IMFTopology *output_topology;
1945     unsigned int marker;
1946     GUID key;
1947 };
1948
1949 static IMFTopologyNode *topology_loader_get_node_for_marker(struct topoloader_context *context, TOPOID *id)
1950 {
1951     IMFTopologyNode *node;
1952     unsigned short i = 0;
1953     unsigned int value;
1954
1955     while (SUCCEEDED(IMFTopology_GetNode(context->output_topology, i++, &node)))
1956     {
1957         if (SUCCEEDED(IMFTopologyNode_GetUINT32(node, &context->key, &value)) && value == context->marker)
1958         {
1959             IMFTopologyNode_GetTopoNodeID(node, id);
1960             return node;
1961         }
1962         IMFTopologyNode_Release(node);
1963     }
1964
1965     *id = 0;
1966     return NULL;
1967 }
1968
1969 static HRESULT topology_loader_clone_node(struct topoloader_context *context, IMFTopologyNode *node,
1970         IMFTopologyNode **ret, unsigned int marker)
1971 {
1972     IMFTopologyNode *cloned_node;
1973     MF_TOPOLOGY_TYPE node_type;
1974     HRESULT hr;
1975
1976     if (ret) *ret = NULL;
1977
1978     IMFTopologyNode_GetNodeType(node, &node_type);
1979
1980     if (FAILED(hr = MFCreateTopologyNode(node_type, &cloned_node)))
1981         return hr;
1982
1983     if (SUCCEEDED(hr = IMFTopologyNode_CloneFrom(cloned_node, node)))
1984         hr = IMFTopologyNode_SetUINT32(cloned_node, &context->key, marker);
1985
1986     if (SUCCEEDED(hr))
1987         hr = IMFTopology_AddNode(context->output_topology, cloned_node);
1988
1989     if (SUCCEEDED(hr) && ret)
1990     {
1991         *ret = cloned_node;
1992         IMFTopologyNode_AddRef(*ret);
1993     }
1994
1995     IMFTopologyNode_Release(cloned_node);
1996
1997     return hr;
1998 }
1999
2000 struct transform_output_type
2001 {
2002     IMFMediaType *type;
2003     IMFTransform *transform;
2004     IMFActivate *activate;
2005 };
2006
2007 struct connect_context
2008 {
2009     struct topoloader_context *context;
2010     IMFTopologyNode *upstream_node;
2011     IMFTopologyNode *sink;
2012     IMFMediaTypeHandler *sink_handler;
2013     const GUID *converter_category;
2014 };
2015
2016 typedef HRESULT (*p_connect_func)(struct transform_output_type *output_type, struct connect_context *context);
2017
2018 static void topology_loader_release_transforms(IMFActivate **activates, unsigned int count)
2019 {
2020     unsigned int i;
2021
2022     for (i = 0; i < count; ++i)
2023         IMFActivate_Release(activates[i]);
2024     CoTaskMemFree(activates);
2025 }
2026
2027 static HRESULT topology_loader_enumerate_output_types(const GUID *category, IMFMediaType *input_type,
2028         p_connect_func connect_func, struct connect_context *context)
2029 {
2030     MFT_REGISTER_TYPE_INFO mft_typeinfo;
2031     IMFActivate **activates;
2032     unsigned int i, count;
2033     HRESULT hr;
2034
2035     if (FAILED(hr = IMFMediaType_GetMajorType(input_type, &mft_typeinfo.guidMajorType)))
2036         return hr;
2037
2038     if (FAILED(hr = IMFMediaType_GetGUID(input_type, &MF_MT_SUBTYPE, &mft_typeinfo.guidSubtype)))
2039         return hr;
2040
2041     if (FAILED(hr = MFTEnumEx(*category, MFT_ENUM_FLAG_ALL, &mft_typeinfo, NULL, &activates, &count)))
2042         return hr;
2043
2044     hr = E_FAIL;
2045
2046     for (i = 0; i < count; ++i)
2047     {
2048         IMFTransform *transform;
2049
2050         if (FAILED(IMFActivate_ActivateObject(activates[i], &IID_IMFTransform, (void **)&transform)))
2051         {
2052             WARN("Failed to create a transform.\n");
2053             continue;
2054         }
2055
2056         if (SUCCEEDED(hr = IMFTransform_SetInputType(transform, 0, input_type, 0)))
2057         {
2058             struct transform_output_type output_type;
2059             unsigned int output_count = 0;
2060
2061             output_type.transform = transform;
2062             output_type.activate = activates[i];
2063             while (SUCCEEDED(IMFTransform_GetOutputAvailableType(transform, 0, output_count++, &output_type.type)))
2064             {
2065                 hr = connect_func(&output_type, context);
2066                 IMFMediaType_Release(output_type.type);
2067                 if (SUCCEEDED(hr))
2068                 {
2069                     topology_loader_release_transforms(activates, count);
2070                     return hr;
2071                 }
2072             }
2073         }
2074
2075         IMFActivate_ShutdownObject(activates[i]);
2076     }
2077
2078     topology_loader_release_transforms(activates, count);
2079
2080     return hr;
2081 }
2082
2083 static HRESULT topology_loader_create_transform(const struct transform_output_type *output_type,
2084         IMFTopologyNode **node)
2085 {
2086     HRESULT hr;
2087     GUID guid;
2088
2089     if (FAILED(hr = MFCreateTopologyNode(MF_TOPOLOGY_TRANSFORM_NODE, node)))
2090         return hr;
2091
2092     IMFTopologyNode_SetObject(*node, (IUnknown *)output_type->transform);
2093
2094     if (SUCCEEDED(IMFActivate_GetGUID(output_type->activate, &MF_TRANSFORM_CATEGORY_Attribute, &guid)) &&
2095             (IsEqualGUID(&guid, &MFT_CATEGORY_AUDIO_DECODER) || IsEqualGUID(&guid, &MFT_CATEGORY_VIDEO_DECODER)))
2096     {
2097         IMFTopologyNode_SetUINT32(*node, &MF_TOPONODE_DECODER, 1);
2098     }
2099
2100     if (SUCCEEDED(IMFActivate_GetGUID(output_type->activate, &MFT_TRANSFORM_CLSID_Attribute, &guid)))
2101         IMFTopologyNode_SetGUID(*node, &MF_TOPONODE_TRANSFORM_OBJECTID, &guid);
2102
2103     return hr;
2104 }
2105
2106 static HRESULT connect_to_sink(struct transform_output_type *output_type, struct connect_context *context)
2107 {
2108     IMFTopologyNode *node;
2109     HRESULT hr;
2110
2111     if (FAILED(IMFMediaTypeHandler_IsMediaTypeSupported(context->sink_handler, output_type->type, NULL)))
2112         return MF_E_TRANSFORM_NOT_POSSIBLE_FOR_CURRENT_MEDIATYPE_COMBINATION;
2113
2114     if (FAILED(hr = topology_loader_create_transform(output_type, &node)))
2115         return hr;
2116
2117     IMFTopology_AddNode(context->context->output_topology, node);
2118     IMFTopologyNode_ConnectOutput(context->upstream_node, 0, node, 0);
2119     IMFTopologyNode_ConnectOutput(node, 0, context->sink, 0);
2120
2121     IMFTopologyNode_Release(node);
2122
2123     hr = IMFMediaTypeHandler_SetCurrentMediaType(context->sink_handler, output_type->type);
2124     if (SUCCEEDED(hr))
2125         hr = IMFTransform_SetOutputType(output_type->transform, 0, output_type->type, 0);
2126
2127     return S_OK;
2128 }
2129
2130 static HRESULT connect_to_converter(struct transform_output_type *output_type, struct connect_context *context)
2131 {
2132     struct connect_context sink_ctx;
2133     IMFTopologyNode *node;
2134     HRESULT hr;
2135
2136     if (SUCCEEDED(connect_to_sink(output_type, context)))
2137         return S_OK;
2138
2139     if (FAILED(hr = topology_loader_create_transform(output_type, &node)))
2140         return hr;
2141
2142     sink_ctx = *context;
2143     sink_ctx.upstream_node = node;
2144
2145     if (SUCCEEDED(hr = topology_loader_enumerate_output_types(context->converter_category, output_type->type,
2146             connect_to_sink, &sink_ctx)))
2147     {
2148         hr = IMFTopology_AddNode(context->context->output_topology, node);
2149     }
2150     IMFTopologyNode_Release(node);
2151
2152     if (SUCCEEDED(hr))
2153     {
2154         IMFTopology_AddNode(context->context->output_topology, node);
2155         IMFTopologyNode_ConnectOutput(context->upstream_node, 0, node, 0);
2156
2157         hr = IMFTransform_SetOutputType(output_type->transform, 0, output_type->type, 0);
2158     }
2159
2160     return hr;
2161 }
2162
2163 typedef HRESULT (*p_topology_loader_connect_func)(struct topoloader_context *context, IMFTopologyNode *upstream_node,
2164         unsigned int output_index, IMFTopologyNode *downstream_node, unsigned int input_index);
2165
2166 static HRESULT topology_loader_get_node_type_handler(IMFTopologyNode *node, IMFMediaTypeHandler **handler)
2167 {
2168     MF_TOPOLOGY_TYPE node_type;
2169     IMFStreamSink *stream_sink;
2170     IMFStreamDescriptor *sd;
2171     HRESULT hr;
2172
2173     if (FAILED(hr = IMFTopologyNode_GetNodeType(node, &node_type)))
2174          return hr;
2175
2176     switch (node_type)
2177     {
2178         case MF_TOPOLOGY_OUTPUT_NODE:
2179             if (SUCCEEDED(hr = topology_node_get_object(node, &IID_IMFStreamSink, (void **)&stream_sink)))
2180             {
2181                 hr = IMFStreamSink_GetMediaTypeHandler(stream_sink, handler);
2182                 IMFStreamSink_Release(stream_sink);
2183             }
2184             break;
2185         case MF_TOPOLOGY_SOURCESTREAM_NODE:
2186             if (SUCCEEDED(hr = IMFTopologyNode_GetUnknown(node, &MF_TOPONODE_STREAM_DESCRIPTOR,
2187                     &IID_IMFStreamDescriptor, (void **)&sd)))
2188             {
2189                 hr = IMFStreamDescriptor_GetMediaTypeHandler(sd, handler);
2190                 IMFStreamDescriptor_Release(sd);
2191             }
2192             break;
2193         default:
2194             WARN("Unexpected node type %u.\n", node_type);
2195             return MF_E_UNEXPECTED;
2196     }
2197
2198     return hr;
2199 }
2200
2201 static HRESULT topology_loader_get_mft_categories(IMFMediaTypeHandler *handler, GUID *decode_cat, GUID *convert_cat)
2202 {
2203     GUID major;
2204     HRESULT hr;
2205
2206     if (FAILED(hr = IMFMediaTypeHandler_GetMajorType(handler, &major)))
2207         return hr;
2208
2209     if (IsEqualGUID(&major, &MFMediaType_Audio))
2210     {
2211         *decode_cat = MFT_CATEGORY_AUDIO_DECODER;
2212         *convert_cat = MFT_CATEGORY_AUDIO_EFFECT;
2213     }
2214     else if (IsEqualGUID(&major, &MFMediaType_Video))
2215     {
2216         *decode_cat = MFT_CATEGORY_VIDEO_DECODER;
2217         *convert_cat = MFT_CATEGORY_VIDEO_EFFECT;
2218     }
2219     else
2220     {
2221         WARN("Unexpected major type %s.\n", debugstr_guid(&major));
2222         return MF_E_INVALIDTYPE;
2223     }
2224
2225     return S_OK;
2226 }
2227
2228 static HRESULT topology_loader_connect_source_to_sink(struct topoloader_context *context, IMFTopologyNode *source,
2229         unsigned int output_index, IMFTopologyNode *sink, unsigned int input_index)
2230 {
2231     IMFMediaTypeHandler *source_handler = NULL, *sink_handler = NULL;
2232     struct connect_context convert_ctx, sink_ctx;
2233     MF_CONNECT_METHOD source_method, sink_method;
2234     GUID decode_cat, convert_cat;
2235     IMFMediaType *media_type;
2236     HRESULT hr;
2237
2238     TRACE("attempting to connect %p:%u to %p:%u\n", source, output_index, sink, input_index);
2239
2240     if (FAILED(hr = topology_loader_get_node_type_handler(source, &source_handler)))
2241         goto done;
2242
2243     if (FAILED(hr = topology_loader_get_node_type_handler(sink, &sink_handler)))
2244         goto done;
2245
2246     if (FAILED(IMFTopologyNode_GetUINT32(source, &MF_TOPONODE_CONNECT_METHOD, &source_method)))
2247         source_method = MF_CONNECT_DIRECT;
2248     if (FAILED(IMFTopologyNode_GetUINT32(sink, &MF_TOPONODE_CONNECT_METHOD, &sink_method)))
2249         sink_method = MF_CONNECT_ALLOW_DECODER;
2250
2251     if (FAILED(hr = topology_loader_get_mft_categories(source_handler, &decode_cat, &convert_cat)))
2252         goto done;
2253
2254     sink_ctx.context = context;
2255     sink_ctx.upstream_node = source;
2256     sink_ctx.sink = sink;
2257     sink_ctx.sink_handler = sink_handler;
2258
2259     convert_ctx = sink_ctx;
2260     convert_ctx.converter_category = &convert_cat;
2261
2262     if (SUCCEEDED(hr = IMFMediaTypeHandler_GetCurrentMediaType(source_handler, &media_type)))
2263     {
2264         /* Direct connection. */
2265         if (SUCCEEDED(hr = IMFMediaTypeHandler_IsMediaTypeSupported(sink_handler, media_type, NULL))
2266                 && SUCCEEDED(hr = IMFMediaTypeHandler_SetCurrentMediaType(sink_handler, media_type)))
2267         {
2268             hr = IMFTopologyNode_ConnectOutput(source, output_index, sink, input_index);
2269         }
2270         if (FAILED(hr) && sink_method & MF_CONNECT_ALLOW_CONVERTER)
2271         {
2272             hr = topology_loader_enumerate_output_types(&convert_cat, media_type, connect_to_sink, &sink_ctx);
2273         }
2274         if (FAILED(hr) && sink_method & MF_CONNECT_ALLOW_DECODER)
2275         {
2276             hr = topology_loader_enumerate_output_types(&decode_cat, media_type, connect_to_converter, &convert_ctx);
2277         }
2278
2279         IMFMediaType_Release(media_type);
2280     }
2281
2282 done:
2283     if (source_handler)
2284         IMFMediaTypeHandler_Release(source_handler);
2285     if (sink_handler)
2286         IMFMediaTypeHandler_Release(sink_handler);
2287
2288     return hr;
2289 }
2290
2291 static HRESULT topology_loader_resolve_branch(struct topoloader_context *context, IMFTopologyNode *upstream_node,
2292         unsigned int output_index, IMFTopologyNode *downstream_node, unsigned input_index)
2293 {
2294     static const p_topology_loader_connect_func connectors[MF_TOPOLOGY_TEE_NODE+1][MF_TOPOLOGY_TEE_NODE+1] =
2295     {
2296           /* OUTPUT */ { NULL },
2297     /* SOURCESTREAM */ { topology_loader_connect_source_to_sink, NULL, NULL, NULL },
2298        /* TRANSFORM */ { NULL },
2299              /* TEE */ { NULL },
2300     };
2301     MF_TOPOLOGY_TYPE u_type, d_type;
2302     IMFTopologyNode *node;
2303     TOPOID id;
2304
2305     /* Downstream node might have already been cloned. */
2306     IMFTopologyNode_GetTopoNodeID(downstream_node, &id);
2307     if (FAILED(IMFTopology_GetNodeByID(context->output_topology, id, &node)))
2308         topology_loader_clone_node(context, downstream_node, &node, context->marker + 1);
2309
2310     IMFTopologyNode_GetNodeType(upstream_node, &u_type);
2311     IMFTopologyNode_GetNodeType(downstream_node, &d_type);
2312
2313     if (!connectors[u_type][d_type])
2314     {
2315         WARN("Unsupported branch kind %d -> %d.\n", u_type, d_type);
2316         return E_FAIL;
2317     }
2318
2319     return connectors[u_type][d_type](context, upstream_node, output_index, node, input_index);
2320 }
2321
2322 static HRESULT topology_loader_resolve_nodes(struct topoloader_context *context, unsigned int *layer_size)
2323 {
2324     IMFTopologyNode *downstream_node, *node, *orig_node;
2325     unsigned int input_index, size = 0;
2326     MF_TOPOLOGY_TYPE node_type;
2327     HRESULT hr = S_OK;
2328     TOPOID id;
2329
2330     while ((node = topology_loader_get_node_for_marker(context, &id)))
2331     {
2332         ++size;
2333
2334         IMFTopology_GetNodeByID(context->input_topology, id, &orig_node);
2335
2336         IMFTopologyNode_GetNodeType(node, &node_type);
2337         switch (node_type)
2338         {
2339             case MF_TOPOLOGY_SOURCESTREAM_NODE:
2340                 if (FAILED(IMFTopologyNode_GetOutput(orig_node, 0, &downstream_node, &input_index)))
2341                 {
2342                     IMFTopology_RemoveNode(context->output_topology, node);
2343                     continue;
2344                 }
2345
2346                 hr = topology_loader_resolve_branch(context, node, 0, downstream_node, input_index);
2347                 break;
2348             case MF_TOPOLOGY_TRANSFORM_NODE:
2349             case MF_TOPOLOGY_TEE_NODE:
2350                 FIXME("Unsupported node type %d.\n", node_type);
2351                 break;
2352             default:
2353                 WARN("Unexpected node type %d.\n", node_type);
2354         }
2355
2356         IMFTopologyNode_DeleteItem(node, &context->key);
2357
2358         if (FAILED(hr))
2359             break;
2360     }
2361
2362     *layer_size = size;
2363
2364     return hr;
2365 }
2366
2367 static BOOL topology_loader_is_node_d3d_aware(IMFTopologyNode *node)
2368 {
2369     IMFAttributes *attributes;
2370     unsigned int d3d_aware = 0;
2371     IMFTransform *transform;
2372
2373     if (FAILED(topology_node_get_object(node, &IID_IMFAttributes, (void **)&attributes)))
2374         return FALSE;
2375
2376     IMFAttributes_GetUINT32(attributes, &MF_SA_D3D_AWARE, &d3d_aware);
2377     IMFAttributes_Release(attributes);
2378
2379     if (!d3d_aware && SUCCEEDED(topology_node_get_object(node, &IID_IMFTransform, (void **)&transform)))
2380     {
2381         d3d_aware = mf_is_sample_copier_transform(transform);
2382         IMFTransform_Release(transform);
2383     }
2384
2385     return !!d3d_aware;
2386 }
2387
2388 static HRESULT topology_loader_create_copier(IMFTopologyNode *upstream_node, unsigned int upstream_output,
2389         IMFTopologyNode *downstream_node, unsigned int downstream_input, IMFTransform **copier)
2390 {
2391     IMFMediaType *input_type = NULL, *output_type = NULL;
2392     IMFTransform *transform;
2393     HRESULT hr;
2394
2395     if (FAILED(hr = MFCreateSampleCopierMFT(&transform)))
2396         return hr;
2397
2398     if (FAILED(hr = MFGetTopoNodeCurrentType(upstream_node, upstream_output, TRUE, &input_type)))
2399         WARN("Failed to get upstream media type hr %#x.\n", hr);
2400
2401     if (SUCCEEDED(hr) && FAILED(hr = MFGetTopoNodeCurrentType(downstream_node, downstream_input, FALSE, &output_type)))
2402         WARN("Failed to get downstream media type hr %#x.\n", hr);
2403
2404     if (SUCCEEDED(hr) && FAILED(hr = IMFTransform_SetInputType(transform, 0, input_type, 0)))
2405         WARN("Input type wasn't accepted, hr %#x.\n", hr);
2406
2407     if (SUCCEEDED(hr) && FAILED(hr = IMFTransform_SetOutputType(transform, 0, output_type, 0)))
2408         WARN("Output type wasn't accepted, hr %#x.\n", hr);
2409
2410     if (SUCCEEDED(hr))
2411     {
2412         *copier = transform;
2413         IMFTransform_AddRef(*copier);
2414     }
2415
2416     if (input_type)
2417         IMFMediaType_Release(input_type);
2418     if (output_type)
2419         IMFMediaType_Release(output_type);
2420
2421     IMFTransform_Release(transform);
2422
2423     return hr;
2424 }
2425
2426 static HRESULT topology_loader_connect_copier(struct topoloader_context *context, IMFTopologyNode *upstream_node,
2427         unsigned int upstream_output, IMFTopologyNode *downstream_node, unsigned int downstream_input, IMFTransform *copier)
2428 {
2429     IMFTopologyNode *copier_node;
2430     HRESULT hr;
2431
2432     if (FAILED(hr = MFCreateTopologyNode(MF_TOPOLOGY_TRANSFORM_NODE, &copier_node)))
2433         return hr;
2434
2435     IMFTopologyNode_SetObject(copier_node, (IUnknown *)copier);
2436     IMFTopology_AddNode(context->output_topology, copier_node);
2437     IMFTopologyNode_ConnectOutput(upstream_node, upstream_output, copier_node, 0);
2438     IMFTopologyNode_ConnectOutput(copier_node, 0, downstream_node, downstream_input);
2439
2440     IMFTopologyNode_Release(copier_node);
2441
2442     return S_OK;
2443 }
2444
2445 /* Right now this should be used for output nodes only. */
2446 static HRESULT topology_loader_connect_d3d_aware_input(struct topoloader_context *context,
2447         IMFTopologyNode *node)
2448 {
2449     IMFTopologyNode *upstream_node;
2450     unsigned int upstream_output;
2451     IMFStreamSink *stream_sink;
2452     IMFTransform *copier = NULL;
2453     HRESULT hr;
2454
2455     if (FAILED(hr = topology_node_get_object(node, &IID_IMFStreamSink, (void **)&stream_sink)))
2456         return hr;
2457
2458     if (topology_loader_is_node_d3d_aware(node))
2459     {
2460         if (SUCCEEDED(IMFTopologyNode_GetInput(node, 0, &upstream_node, &upstream_output)))
2461         {
2462             if (!topology_loader_is_node_d3d_aware(upstream_node))
2463             {
2464                 if (SUCCEEDED(hr = topology_loader_create_copier(upstream_node, upstream_output, node, 0, &copier)))
2465                 {
2466                     hr = topology_loader_connect_copier(context, upstream_node, upstream_output, node, 0, copier);
2467                     IMFTransform_Release(copier);
2468                 }
2469             }
2470             IMFTopologyNode_Release(upstream_node);
2471         }
2472     }
2473
2474     IMFStreamSink_Release(stream_sink);
2475
2476     return hr;
2477 }
2478
2479 static void topology_loader_resolve_complete(struct topoloader_context *context)
2480 {
2481     MF_TOPOLOGY_TYPE node_type;
2482     IMFTopologyNode *node;
2483     WORD i, node_count;
2484     HRESULT hr;
2485
2486     IMFTopology_GetNodeCount(context->output_topology, &node_count);
2487
2488     for (i = 0; i < node_count; ++i)
2489     {
2490         if (SUCCEEDED(IMFTopology_GetNode(context->output_topology, i, &node)))
2491         {
2492             IMFTopologyNode_GetNodeType(node, &node_type);
2493
2494             if (node_type == MF_TOPOLOGY_OUTPUT_NODE)
2495             {
2496                 /* Set MF_TOPONODE_STREAMID for all outputs. */
2497                 if (FAILED(IMFTopologyNode_GetItem(node, &MF_TOPONODE_STREAMID, NULL)))
2498                     IMFTopologyNode_SetUINT32(node, &MF_TOPONODE_STREAMID, 0);
2499
2500                 if (FAILED(hr = topology_loader_connect_d3d_aware_input(context, node)))
2501                     WARN("Failed to connect D3D-aware input, hr %#x.\n", hr);
2502             }
2503             else if (node_type == MF_TOPOLOGY_SOURCESTREAM_NODE)
2504             {
2505                 /* Set MF_TOPONODE_MEDIASTART for all sources. */
2506                 if (FAILED(IMFTopologyNode_GetItem(node, &MF_TOPONODE_MEDIASTART, NULL)))
2507                     IMFTopologyNode_SetUINT64(node, &MF_TOPONODE_MEDIASTART, 0);
2508             }
2509
2510             IMFTopologyNode_Release(node);
2511         }
2512     }
2513 }
2514
2515 static HRESULT WINAPI topology_loader_Load(IMFTopoLoader *iface, IMFTopology *input_topology,
2516         IMFTopology **ret_topology, IMFTopology *current_topology)
2517 {
2518     struct topoloader_context context = { 0 };
2519     IMFTopology *output_topology;
2520     MF_TOPOLOGY_TYPE node_type;
2521     unsigned int layer_size;
2522     IMFTopologyNode *node;
2523     unsigned short i = 0;
2524     IMFStreamSink *sink;
2525     IUnknown *object;
2526     HRESULT hr = E_FAIL;
2527
2528     FIXME("%p, %p, %p, %p.\n", iface, input_topology, ret_topology, current_topology);
2529
2530     if (current_topology)
2531         FIXME("Current topology instance is ignored.\n");
2532
2533     /* Basic sanity checks for input topology:
2534
2535        - source nodes must have stream descriptor set;
2536        - sink nodes must be resolved to stream sink objects;
2537     */
2538     while (SUCCEEDED(IMFTopology_GetNode(input_topology, i++, &node)))
2539     {
2540         IMFTopologyNode_GetNodeType(node, &node_type);
2541
2542         switch (node_type)
2543         {
2544             case MF_TOPOLOGY_OUTPUT_NODE:
2545                 if (SUCCEEDED(hr = IMFTopologyNode_GetObject(node, &object)))
2546                 {
2547                     /* Sinks must be bound beforehand. */
2548                     if (FAILED(IUnknown_QueryInterface(object, &IID_IMFStreamSink, (void **)&sink)))
2549                         hr = MF_E_TOPO_SINK_ACTIVATES_UNSUPPORTED;
2550                     else if (sink)
2551                         IMFStreamSink_Release(sink);
2552                     IUnknown_Release(object);
2553                 }
2554                 break;
2555             case MF_TOPOLOGY_SOURCESTREAM_NODE:
2556                 hr = IMFTopologyNode_GetItem(node, &MF_TOPONODE_STREAM_DESCRIPTOR, NULL);
2557                 break;
2558             default:
2559                 ;
2560         }
2561
2562         IMFTopologyNode_Release(node);
2563         if (FAILED(hr))
2564             return hr;
2565     }
2566
2567     if (FAILED(hr = MFCreateTopology(&output_topology)))
2568         return hr;
2569
2570     context.input_topology = input_topology;
2571     context.output_topology = output_topology;
2572     memset(&context.key, 0xff, sizeof(context.key));
2573
2574     /* Clone source nodes, use initial marker value. */
2575     i = 0;
2576     while (SUCCEEDED(IMFTopology_GetNode(input_topology, i++, &node)))
2577     {
2578         IMFTopologyNode_GetNodeType(node, &node_type);
2579
2580         if (node_type == MF_TOPOLOGY_SOURCESTREAM_NODE)
2581         {
2582             if (FAILED(hr = topology_loader_clone_node(&context, node, NULL, 0)))
2583                 WARN("Failed to clone source node, hr %#x.\n", hr);
2584         }
2585
2586         IMFTopologyNode_Release(node);
2587     }
2588
2589     for (context.marker = 0;; ++context.marker)
2590     {
2591         if (FAILED(hr = topology_loader_resolve_nodes(&context, &layer_size)))
2592         {
2593             WARN("Failed to resolve for marker %u, hr %#x.\n", context.marker, hr);
2594             break;
2595         }
2596
2597         /* Reached last marker value. */
2598         if (!layer_size)
2599             break;
2600     }
2601
2602     if (SUCCEEDED(hr))
2603         topology_loader_resolve_complete(&context);
2604
2605     *ret_topology = output_topology;
2606
2607     return hr;
2608 }
2609
2610 static const IMFTopoLoaderVtbl topologyloadervtbl =
2611 {
2612     topology_loader_QueryInterface,
2613     topology_loader_AddRef,
2614     topology_loader_Release,
2615     topology_loader_Load,
2616 };
2617
2618 /***********************************************************************
2619  *      MFCreateTopoLoader (mf.@)
2620  */
2621 HRESULT WINAPI MFCreateTopoLoader(IMFTopoLoader **loader)
2622 {
2623     struct topology_loader *object;
2624
2625     TRACE("%p.\n", loader);
2626
2627     if (!loader)
2628         return E_POINTER;
2629
2630     object = heap_alloc(sizeof(*object));
2631     if (!object)
2632         return E_OUTOFMEMORY;
2633
2634     object->IMFTopoLoader_iface.lpVtbl = &topologyloadervtbl;
2635     object->refcount = 1;
2636
2637     *loader = &object->IMFTopoLoader_iface;
2638
2639     return S_OK;
2640 }
2641
2642 static HRESULT WINAPI seq_source_QueryInterface(IMFSequencerSource *iface, REFIID riid, void **out)
2643 {
2644     struct seq_source *seq_source = impl_from_IMFSequencerSource(iface);
2645
2646     TRACE("%p, %s, %p.\n", iface, debugstr_guid(riid), out);
2647
2648     *out = NULL;
2649
2650     if (IsEqualIID(riid, &IID_IMFSequencerSource) ||
2651             IsEqualIID(riid, &IID_IUnknown))
2652     {
2653         *out = &seq_source->IMFSequencerSource_iface;
2654     }
2655     else if (IsEqualIID(riid, &IID_IMFMediaSourceTopologyProvider))
2656     {
2657         *out = &seq_source->IMFMediaSourceTopologyProvider_iface;
2658     }
2659     else
2660     {
2661         WARN("Unimplemented %s.\n", debugstr_guid(riid));
2662         return E_NOINTERFACE;
2663     }
2664
2665     if (*out)
2666         IUnknown_AddRef((IUnknown *)*out);
2667
2668     return S_OK;
2669 }
2670
2671 static ULONG WINAPI seq_source_AddRef(IMFSequencerSource *iface)
2672 {
2673     struct seq_source *seq_source = impl_from_IMFSequencerSource(iface);
2674     ULONG refcount = InterlockedIncrement(&seq_source->refcount);
2675
2676     TRACE("%p, refcount %u.\n", iface, refcount);
2677
2678     return refcount;
2679 }
2680
2681 static ULONG WINAPI seq_source_Release(IMFSequencerSource *iface)
2682 {
2683     struct seq_source *seq_source = impl_from_IMFSequencerSource(iface);
2684     ULONG refcount = InterlockedDecrement(&seq_source->refcount);
2685
2686     TRACE("%p, refcount %u.\n", iface, refcount);
2687
2688     if (!refcount)
2689     {
2690         heap_free(seq_source);
2691     }
2692
2693     return refcount;
2694 }
2695
2696 static HRESULT WINAPI seq_source_AppendTopology(IMFSequencerSource *iface, IMFTopology *topology,
2697         DWORD flags, MFSequencerElementId *id)
2698 {
2699     FIXME("%p, %p, %x, %p.\n", iface, topology, flags, id);
2700
2701     return E_NOTIMPL;
2702 }
2703
2704 static HRESULT WINAPI seq_source_DeleteTopology(IMFSequencerSource *iface, MFSequencerElementId id)
2705 {
2706     FIXME("%p, %#x.\n", iface, id);
2707
2708     return E_NOTIMPL;
2709 }
2710
2711 static HRESULT WINAPI seq_source_GetPresentationContext(IMFSequencerSource *iface,
2712         IMFPresentationDescriptor *descriptor, MFSequencerElementId *id, IMFTopology **topology)
2713 {
2714     FIXME("%p, %p, %p, %p.\n", iface, descriptor, id, topology);
2715
2716     return E_NOTIMPL;
2717 }
2718
2719 static HRESULT WINAPI seq_source_UpdateTopology(IMFSequencerSource *iface, MFSequencerElementId id,
2720         IMFTopology *topology)
2721 {
2722     FIXME("%p, %#x, %p.\n", iface, id, topology);
2723
2724     return E_NOTIMPL;
2725 }
2726
2727 static HRESULT WINAPI seq_source_UpdateTopologyFlags(IMFSequencerSource *iface, MFSequencerElementId id, DWORD flags)
2728 {
2729     FIXME("%p, %#x, %#x.\n", iface, id, flags);
2730
2731     return E_NOTIMPL;
2732 }
2733
2734 static HRESULT WINAPI seq_source_topology_provider_QueryInterface(IMFMediaSourceTopologyProvider *iface, REFIID riid,
2735         void **obj)
2736 {
2737     struct seq_source *seq_source = impl_from_IMFMediaSourceTopologyProvider(iface);
2738     return IMFSequencerSource_QueryInterface(&seq_source->IMFSequencerSource_iface, riid, obj);
2739 }
2740
2741 static ULONG WINAPI seq_source_topology_provider_AddRef(IMFMediaSourceTopologyProvider *iface)
2742 {
2743     struct seq_source *seq_source = impl_from_IMFMediaSourceTopologyProvider(iface);
2744     return IMFSequencerSource_AddRef(&seq_source->IMFSequencerSource_iface);
2745 }
2746
2747 static ULONG WINAPI seq_source_topology_provider_Release(IMFMediaSourceTopologyProvider *iface)
2748 {
2749     struct seq_source *seq_source = impl_from_IMFMediaSourceTopologyProvider(iface);
2750     return IMFSequencerSource_Release(&seq_source->IMFSequencerSource_iface);
2751 }
2752
2753 static HRESULT WINAPI seq_source_topology_provider_GetMediaSourceTopology(IMFMediaSourceTopologyProvider *iface,
2754         IMFPresentationDescriptor *pd, IMFTopology **topology)
2755 {
2756     FIXME("%p, %p, %p.\n", iface, pd, topology);
2757
2758     return E_NOTIMPL;
2759 }
2760
2761 static const IMFMediaSourceTopologyProviderVtbl seq_source_topology_provider_vtbl =
2762 {
2763     seq_source_topology_provider_QueryInterface,
2764     seq_source_topology_provider_AddRef,
2765     seq_source_topology_provider_Release,
2766     seq_source_topology_provider_GetMediaSourceTopology,
2767 };
2768
2769 static const IMFSequencerSourceVtbl seqsourcevtbl =
2770 {
2771     seq_source_QueryInterface,
2772     seq_source_AddRef,
2773     seq_source_Release,
2774     seq_source_AppendTopology,
2775     seq_source_DeleteTopology,
2776     seq_source_GetPresentationContext,
2777     seq_source_UpdateTopology,
2778     seq_source_UpdateTopologyFlags,
2779 };
2780
2781 /***********************************************************************
2782  *      MFCreateSequencerSource (mf.@)
2783  */
2784 HRESULT WINAPI MFCreateSequencerSource(IUnknown *reserved, IMFSequencerSource **seq_source)
2785 {
2786     struct seq_source *object;
2787
2788     TRACE("%p, %p.\n", reserved, seq_source);
2789
2790     if (!seq_source)
2791         return E_POINTER;
2792
2793     object = heap_alloc(sizeof(*object));
2794     if (!object)
2795         return E_OUTOFMEMORY;
2796
2797     object->IMFSequencerSource_iface.lpVtbl = &seqsourcevtbl;
2798     object->IMFMediaSourceTopologyProvider_iface.lpVtbl = &seq_source_topology_provider_vtbl;
2799     object->refcount = 1;
2800
2801     *seq_source = &object->IMFSequencerSource_iface;
2802
2803     return S_OK;
2804 }