output-selector: add drain handling

Release the latest buffer, if any

1 files changed, 13 insertions, 0 deletions

diff --git a/plugins/elements/gstoutputselector.c b/plugins/elements/gstoutputselector.c
index 8ad870c98..5d2e01d32 100644
--- a/plugins/elements/gstoutputselector.c
+++ b/plugins/elements/gstoutputselector.c
@@ -563,6 +563,15 @@ gst_output_selector_event (GstPad * pad, GstObject * parent, GstEvent * event)
   return res;
 }
 
+static void
+gst_output_selector_drain (GstOutputSelector * sel)
+{
+  if (sel->latest_buffer) {
+    gst_buffer_unref (sel->latest_buffer);
+    sel->latest_buffer = NULL;
+  }
+}
+
 static gboolean
 gst_output_selector_query (GstPad * pad, GstObject * parent, GstQuery * query)
 {
@@ -595,9 +604,13 @@ gst_output_selector_query (GstPad * pad, GstObject * parent, GstQuery * query)
       }
       break;
     }
+    case GST_QUERY_DRAIN:
+      gst_output_selector_drain (sel);
+      /* fall through */
     default:
       res = gst_pad_query_default (pad, parent, query);
       break;
   }
+
   return res;
 }