system/wine
system/wine
1
0
Fork 0
mirror of git://source.winehq.org/git/wine.git synced 2024-07-19 02:13:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

mfreadwrite/writer: Implement AddStream().

Browse source 
Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com>
This commit is contained in:
Nikolay Sivov 2022-08-16 19:41:24 +03:00 committed by Alexandre Julliard
parent b5040bce82
commit 98f0e8bca3
3 changed files with 143 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
dlls/mfreadwrite/mf_private.h
View file

@ -16,7 +16,36 @@
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include "mferror.h"
extern HRESULT create_sink_writer_from_url(const WCHAR *url, IMFByteStream *stream,
IMFAttributes *attributes, REFIID riid, void **out) DECLSPEC_HIDDEN;
extern HRESULT create_sink_writer_from_sink(IMFMediaSink *sink, IMFAttributes *attributes,
REFIID riid, void **out) DECLSPEC_HIDDEN;
static inline BOOL mf_array_reserve(void **elements, size_t *capacity, size_t count, size_t size)
{
size_t new_capacity, max_capacity;
void *new_elements;
if (count <= *capacity)
return TRUE;
max_capacity = ~(SIZE_T)0 / size;
if (count > max_capacity)
return FALSE;
new_capacity = max(4, *capacity);
while (new_capacity < count && new_capacity <= max_capacity / 2)
new_capacity *= 2;
if (new_capacity < count)
new_capacity = max_capacity;
if (!(new_elements = realloc(*elements, new_capacity * size)))
return FALSE;
*elements = new_elements;
*capacity = new_capacity;
return TRUE;
}

1
dlls/mfreadwrite/reader.c
View file

@ -30,7 +30,6 @@
#undef INITGUID
#include <guiddef.h>
#include "mfapi.h"
#include "mferror.h"
#include "mfidl.h"
#include "mfreadwrite.h"
#include "d3d9.h"

119
dlls/mfreadwrite/writer.c
View file

@ -30,10 +30,33 @@
WINE_DEFAULT_DEBUG_CHANNEL(mfplat);
enum writer_state
{
SINK_WRITER_STATE_INITIAL = 0,
};
struct stream
{
IMFStreamSink *stream_sink;
};
struct sink_writer
{
IMFSinkWriter IMFSinkWriter_iface;
LONG refcount;
struct
{
struct stream *items;
size_t count;
size_t capacity;
DWORD next_id;
} streams;
IMFMediaSink *sink;
enum writer_state state;
CRITICAL_SECTION cs;
};
static struct sink_writer *impl_from_IMFSinkWriter(IMFSinkWriter *iface)
@ -72,22 +95,80 @@ static ULONG WINAPI sink_writer_Release(IMFSinkWriter *iface)
{
struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
ULONG refcount = InterlockedDecrement(&writer->refcount);
unsigned int i;
TRACE("%p, %lu.\n", iface, refcount);
if (!refcount)
{
if (writer->sink)
IMFMediaSink_Release(writer->sink);
for (i = 0; i < writer->streams.count; ++i)
{
struct stream *stream = &writer->streams.items[i];
if (stream->stream_sink)
IMFStreamSink_Release(stream->stream_sink);
}
DeleteCriticalSection(&writer->cs);
free(writer);
}
return refcount;
}
static HRESULT WINAPI sink_writer_AddStream(IMFSinkWriter *iface, IMFMediaType *type, DWORD *index)
static HRESULT sink_writer_add_stream(struct sink_writer *writer, IMFStreamSink *stream_sink, DWORD *index)
{
FIXME("%p, %p, %p.\n", iface, type, index);
DWORD id = 0;
HRESULT hr;
return E_NOTIMPL;
if (!mf_array_reserve((void **)&writer->streams.items, &writer->streams.capacity, writer->streams.count + 1,
sizeof(*writer->streams.items)))
{
return E_OUTOFMEMORY;
}
if (FAILED(hr = IMFStreamSink_GetIdentifier(stream_sink, &id))) return hr;
*index = writer->streams.count++;
writer->streams.items[*index].stream_sink = stream_sink;
IMFStreamSink_AddRef(stream_sink);
writer->streams.next_id = max(writer->streams.next_id, id);
return hr;
}
static HRESULT WINAPI sink_writer_AddStream(IMFSinkWriter *iface, IMFMediaType *media_type, DWORD *index)
{
struct sink_writer *writer = impl_from_IMFSinkWriter(iface);
HRESULT hr = MF_E_INVALIDREQUEST;
IMFStreamSink *stream_sink;
DWORD id;
TRACE("%p, %p, %p.\n", iface, media_type, index);
if (!media_type)
return E_INVALIDARG;
if (!index)
return E_POINTER;
EnterCriticalSection(&writer->cs);
if (writer->state == SINK_WRITER_STATE_INITIAL)
{
id = writer->streams.next_id + 1;
if (SUCCEEDED(hr = IMFMediaSink_AddStreamSink(writer->sink, id, media_type, &stream_sink)))
{
if (FAILED(hr = sink_writer_add_stream(writer, stream_sink, index)))
IMFMediaSink_RemoveStreamSink(writer->sink, id);
}
}
LeaveCriticalSection(&writer->cs);
return hr;
}
static HRESULT WINAPI sink_writer_SetInputMediaType(IMFSinkWriter *iface, DWORD index, IMFMediaType *type,
@ -180,6 +261,26 @@ static const IMFSinkWriterVtbl sink_writer_vtbl =
sink_writer_GetStatistics,
};
static HRESULT sink_writer_initialize_existing_streams(struct sink_writer *writer, IMFMediaSink *sink)
{
IMFStreamSink *stream_sink;
DWORD count = 0, i, index;
HRESULT hr;
if (FAILED(hr = IMFMediaSink_GetStreamSinkCount(sink, &count))) return hr;
if (!count) return S_OK;
for (i = 0; i < count; ++i)
{
if (FAILED(hr = IMFMediaSink_GetStreamSinkByIndex(sink, i, &stream_sink))) break;
hr = sink_writer_add_stream(writer, stream_sink, &index);
IMFStreamSink_Release(stream_sink);
if (FAILED(hr)) break;
}
return hr;
}
HRESULT create_sink_writer_from_sink(IMFMediaSink *sink, IMFAttributes *attributes,
REFIID riid, void **out)
{
@ -191,12 +292,20 @@ HRESULT create_sink_writer_from_sink(IMFMediaSink *sink, IMFAttributes *attribut
if (!sink)
return E_INVALIDARG;
object = malloc(sizeof(*object));
if (!object)
if (!(object = calloc(1, sizeof(*object))))
return E_OUTOFMEMORY;
object->IMFSinkWriter_iface.lpVtbl = &sink_writer_vtbl;
object->refcount = 1;
object->sink = sink;
IMFMediaSink_AddRef(sink);
InitializeCriticalSection(&object->cs);
if (FAILED(hr = sink_writer_initialize_existing_streams(object, sink)))
{
IMFSinkWriter_Release(&object->IMFSinkWriter_iface);
return hr;
}
hr = IMFSinkWriter_QueryInterface(&object->IMFSinkWriter_iface, riid, out);
IMFSinkWriter_Release(&object->IMFSinkWriter_iface);