2 files changed, 19 insertions, 4 deletions
diff --git a/gst/vaapi/gstvaapivideocontext.c b/gst/vaapi/gstvaapivideocontext.c
index f049273e..c90ede7d 100644
--- a/gst/vaapi/gstvaapivideocontext.c
+++ b/gst/vaapi/gstvaapivideocontext.c
@@ -52,17 +52,27 @@ _init_context_debug (void)
 #endif
 }
 
+void
+gst_vaapi_video_context_set_display (GstContext * context,
+    GstVaapiDisplay * display)
+{
+  GstStructure *structure;
+
+  g_return_if_fail (context != NULL);
+
+  structure = gst_context_writable_structure (context);
+  gst_structure_set (structure, GST_VAAPI_DISPLAY_CONTEXT_TYPE_NAME,
+      GST_VAAPI_TYPE_DISPLAY, display, NULL);
+}
+
 GstContext *
 gst_vaapi_video_context_new_with_display (GstVaapiDisplay * display,
     gboolean persistent)
 {
   GstContext *context;
-  GstStructure *structure;
 
   context = gst_context_new (GST_VAAPI_DISPLAY_CONTEXT_TYPE_NAME, persistent);
-  structure = gst_context_writable_structure (context);
-  gst_structure_set (structure, GST_VAAPI_DISPLAY_CONTEXT_TYPE_NAME,
-      GST_VAAPI_TYPE_DISPLAY, display, NULL);
+  gst_vaapi_video_context_set_display (context, display);
   return context;
 }
 
diff --git a/gst/vaapi/gstvaapivideocontext.h b/gst/vaapi/gstvaapivideocontext.h
index 95e134b6..2d56183d 100644
--- a/gst/vaapi/gstvaapivideocontext.h
+++ b/gst/vaapi/gstvaapivideocontext.h
@@ -32,6 +32,11 @@
 #define GST_VAAPI_DISPLAY_CONTEXT_TYPE_NAME "gst.vaapi.Display"
 
 G_GNUC_INTERNAL
+void
+gst_vaapi_video_context_set_display (GstContext * context,
+    GstVaapiDisplay * display);
+
+G_GNUC_INTERNAL
 GstContext *
 gst_vaapi_video_context_new_with_display (GstVaapiDisplay * display,
     gboolean persistent);