winegstreamer: Return S_FALSE from wg_parser_stream_get_event() if the stream is EOS.

Instead of using WG_PARSER_EVENT_EOS.

Signed-off-by: Zebediah Figura <zfigura@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>

dlls/winegstreamer/gst_private.h

@@ -80,7 +80,7 @@ void wg_parser_stream_get_preferred_format(struct wg_parser_stream *stream, stru
 void wg_parser_stream_enable(struct wg_parser_stream *stream, const struct wg_format *format);
 void wg_parser_stream_disable(struct wg_parser_stream *stream);
 
-void wg_parser_stream_get_event(struct wg_parser_stream *stream, struct wg_parser_event *event);
+bool wg_parser_stream_get_event(struct wg_parser_stream *stream, struct wg_parser_event *event);
 bool wg_parser_stream_copy_buffer(struct wg_parser_stream *stream,
         void *data, uint32_t offset, uint32_t size);
 void wg_parser_stream_release_buffer(struct wg_parser_stream *stream);

dlls/winegstreamer/main.c

@@ -172,7 +172,7 @@ void wg_parser_stream_disable(struct wg_parser_stream *stream)
     __wine_unix_call(unix_handle, unix_wg_parser_stream_disable, stream);
 }
 
-void wg_parser_stream_get_event(struct wg_parser_stream *stream, struct wg_parser_event *event)
+bool wg_parser_stream_get_event(struct wg_parser_stream *stream, struct wg_parser_event *event)
 {
     struct wg_parser_stream_get_event_params params =
     {
@@ -180,7 +180,7 @@ void wg_parser_stream_get_event(struct wg_parser_stream *stream, struct wg_parse
         .event = event,
     };
 
-    __wine_unix_call(unix_handle, unix_wg_parser_stream_get_event, &params);
+    return !__wine_unix_call(unix_handle, unix_wg_parser_stream_get_event, &params);
 }
 
 bool wg_parser_stream_copy_buffer(struct wg_parser_stream *stream,

dlls/winegstreamer/media_source.c

@@ -532,33 +532,15 @@ static void wait_on_sample(struct media_stream *stream, IUnknown *token)
 
     TRACE("%p, %p\n", stream, token);
 
-    if (stream->eos)
+    if (wg_parser_stream_get_event(stream->wg_stream, &event))
     {
-        IMFMediaEventQueue_QueueEventParamVar(stream->event_queue, MEError, &GUID_NULL, MF_E_END_OF_STREAM, &empty_var);
-        return;
-    }
-
-    for (;;)
-    {
-        wg_parser_stream_get_event(stream->wg_stream, &event);
-
-        TRACE("Got event of type %#x.\n", event.type);
-
-        switch (event.type)
-        {
-            case WG_PARSER_EVENT_BUFFER:
         send_buffer(stream, &event, token);
-                return;
+    }
-
+    else
-            case WG_PARSER_EVENT_EOS:
+    {
         stream->eos = TRUE;
         IMFMediaEventQueue_QueueEventParamVar(stream->event_queue, MEEndOfStream, &GUID_NULL, S_OK, &empty_var);
         dispatch_end_of_presentation(stream->parent_source);
-                return;
-
-            case WG_PARSER_EVENT_NONE:
-                assert(0);
-        }
     }
 }
 

dlls/winegstreamer/quartz_parser.c

@@ -833,25 +833,19 @@ static DWORD CALLBACK stream_thread(void *arg)
             continue;
         }
 
-        wg_parser_stream_get_event(pin->wg_stream, &event);
+        if (wg_parser_stream_get_event(pin->wg_stream, &event))
-
-        TRACE("Got event of type %#x.\n", event.type);
-
-        switch (event.type)
         {
-            case WG_PARSER_EVENT_BUFFER:
             send_buffer(pin, &event);
-                break;
+        }
-
+        else
-            case WG_PARSER_EVENT_EOS:
+        {
+            TRACE("Got EOS.\n");
             IPin_EndOfStream(pin->pin.pin.peer);
             pin->eos = true;
-                break;
-
-            case WG_PARSER_EVENT_NONE:
-                assert(0);
         }
 
+        TRACE("Got event of type %#x.\n", event.type);
+
         LeaveCriticalSection(&pin->flushing_cs);
     }
 

dlls/winegstreamer/unixlib.h

@@ -107,7 +107,6 @@ enum wg_parser_event_type
 {
     WG_PARSER_EVENT_NONE = 0,
     WG_PARSER_EVENT_BUFFER,
-    WG_PARSER_EVENT_EOS,
 };
 
 struct wg_parser_event

dlls/winegstreamer/wg_parser.c

@@ -259,19 +259,20 @@ static NTSTATUS wg_parser_stream_get_event(void *args)
 
     pthread_mutex_lock(&parser->mutex);
 
-    while (stream->event.type == WG_PARSER_EVENT_NONE)
+    while (!stream->eos && stream->event.type == WG_PARSER_EVENT_NONE)
         pthread_cond_wait(&stream->event_cond, &parser->mutex);
 
-    *params->event = stream->event;
+    /* Note that we can both have a buffer and stream->eos, in which case we
-
+     * must return the buffer. */
-    if (stream->event.type != WG_PARSER_EVENT_BUFFER)
+    if (stream->event.type != WG_PARSER_EVENT_NONE)
     {
-        stream->event.type = WG_PARSER_EVENT_NONE;
+        *params->event = stream->event;
-        pthread_cond_signal(&stream->event_empty_cond);
-    }
         pthread_mutex_unlock(&parser->mutex);
-
         return S_OK;
+    }
+
+    pthread_mutex_unlock(&parser->mutex);
+    return S_FALSE;
 }
 
 static NTSTATUS wg_parser_stream_copy_buffer(void *args)
@@ -434,8 +435,7 @@ static GstFlowReturn queue_stream_event(struct wg_parser_stream *stream,
         GST_DEBUG("Filter is flushing; discarding event.");
         return GST_FLOW_FLUSHING;
     }
-    if (buffer)
+
-    {
     assert(GST_IS_BUFFER(buffer));
     if (!gst_buffer_map(buffer, &stream->map_info, GST_MAP_READ))
     {
@@ -443,7 +443,7 @@ static GstFlowReturn queue_stream_event(struct wg_parser_stream *stream,
         GST_ERROR("Failed to map buffer.\n");
         return GST_FLOW_ERROR;
     }
-    }
+
     stream->event = *event;
     stream->buffer = buffer;
     pthread_mutex_unlock(&parser->mutex);
@@ -479,20 +479,13 @@ static gboolean sink_event_cb(GstPad *pad, GstObject *parent, GstEvent *event)
             break;
 
         case GST_EVENT_EOS:
-            if (stream->enabled)
-            {
-                struct wg_parser_event stream_event;
-
-                stream_event.type = WG_PARSER_EVENT_EOS;
-                queue_stream_event(stream, &stream_event, NULL);
-            }
-            else
-            {
             pthread_mutex_lock(&parser->mutex);
             stream->eos = true;
             pthread_mutex_unlock(&parser->mutex);
+            if (stream->enabled)
+                pthread_cond_signal(&stream->event_cond);
+            else
                 pthread_cond_signal(&parser->init_cond);
-            }
             break;
 
         case GST_EVENT_FLUSH_START:
@@ -524,12 +517,13 @@ static gboolean sink_event_cb(GstPad *pad, GstObject *parent, GstEvent *event)
             if (reset_time)
                 gst_segment_init(&stream->segment, GST_FORMAT_UNDEFINED);
 
-            if (stream->enabled)
-            {
             pthread_mutex_lock(&parser->mutex);
+
+            stream->eos = false;
+            if (stream->enabled)
                 stream->flushing = false;
+
             pthread_mutex_unlock(&parser->mutex);
-            }
             break;
         }
 

dlls/winegstreamer/wm_reader.c

@@ -1826,6 +1826,10 @@ HRESULT wm_reader_get_stream_sample(struct wm_stream *stream,
     IWMReaderCallbackAdvanced *callback_advanced = stream->reader->callback_advanced;
     struct wg_parser_stream *wg_stream = stream->wg_stream;
     struct wg_parser_event event;
+    DWORD size, capacity;
+    INSSBuffer *sample;
+    HRESULT hr;
+    BYTE *data;
 
     if (stream->selection == WMT_OFF)
         return NS_E_INVALID_REQUEST;
@@ -1835,19 +1839,14 @@ HRESULT wm_reader_get_stream_sample(struct wm_stream *stream,
 
     for (;;)
     {
-        wg_parser_stream_get_event(wg_stream, &event);
+        if (!wg_parser_stream_get_event(wg_stream, &event))
-
-        TRACE("Got event of type %#x for %s stream %p.\n", event.type,
-                get_major_type_string(stream->format.major_type), stream);
-
-        switch (event.type)
         {
-            case WG_PARSER_EVENT_BUFFER:
+            stream->eos = true;
-            {
+            TRACE("End of stream.\n");
-                DWORD size, capacity;
+            return NS_E_NO_MORE_SAMPLES;
-                INSSBuffer *sample;
+        }
-                HRESULT hr;
+
-                BYTE *data;
+        TRACE("Got buffer for '%s' stream %p.\n", get_major_type_string(stream->format.major_type), stream);
 
         if (callback_advanced && stream->read_compressed && stream->allocate_stream)
         {
@@ -1893,14 +1892,13 @@ HRESULT wm_reader_get_stream_sample(struct wm_stream *stream,
         if (FAILED(hr = INSSBuffer_GetMaxLength(sample, &capacity)))
             ERR("Failed to get capacity, hr %#lx.\n", hr);
         if (event.u.buffer.size > capacity)
-                    ERR("Returned capacity %lu is less than requested capacity %u.\n",
+            ERR("Returned capacity %lu is less than requested capacity %u.\n", capacity, event.u.buffer.size);
-                            capacity, event.u.buffer.size);
 
         if (!wg_parser_stream_copy_buffer(wg_stream, data, 0, event.u.buffer.size))
         {
             /* The GStreamer pin has been flushed. */
             INSSBuffer_Release(sample);
-                    break;
+            continue;
         }
 
         if (FAILED(hr = INSSBuffer_SetLength(sample, event.u.buffer.size)))
@@ -1924,16 +1922,6 @@ HRESULT wm_reader_get_stream_sample(struct wm_stream *stream,
         *ret_sample = sample;
         return S_OK;
     }
-
-            case WG_PARSER_EVENT_EOS:
-                stream->eos = true;
-                TRACE("End of stream.\n");
-                return NS_E_NO_MORE_SAMPLES;
-
-            case WG_PARSER_EVENT_NONE:
-                assert(0);
-        }
-    }
 }
 
 void wm_reader_seek(struct wm_reader *reader, QWORD start, LONGLONG duration)