sys/msdk/gstmsdkvpp.c

   1 /* GStreamer Intel MSDK plugin
   2  * Copyright (c) 2018, Intel Corporation
   3  * All rights reserved.
   4  *
   5  * Author: Sreerenj Balachaandran <sreerenj.balachandran@intel.com>
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions are met:
   9  *
  10  * 1. Redistributions of source code must retain the above copyright notice,
  11  *    this list of conditions and the following disclaimer.
  12  *
  13  * 2. Redistributions in binary form must reproduce the above copyright notice,
  14  *    this list of conditions and the following disclaimer in the documentation
  15  *    and/or other materials provided with the distribution.
  16  *
  17  * 3. Neither the name of the copyright holder nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  23  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  24  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
  25  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  26  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  27  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  28  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  29  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
  30  * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  31  * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #ifdef HAVE_CONFIG_H
  35 #  include <config.h>
  36 #endif
  37
  38 #include <stdlib.h>
  39
  40 #include "gstmsdkvpp.h"
  41 #include "gstmsdkbufferpool.h"
  42 #include "gstmsdkvideomemory.h"
  43 #include "gstmsdksystemmemory.h"
  44 #include "gstmsdkcontextutil.h"
  45 #include "gstmsdkvpputil.h"
  46
  47 #ifndef _WIN32
  48 #include "gstmsdkallocator_libva.h"
  49 #endif
  50
  51 GST_DEBUG_CATEGORY_EXTERN (gst_msdkvpp_debug);
  52 #define GST_CAT_DEFAULT gst_msdkvpp_debug
  53
  54 static GstStaticPadTemplate gst_msdkvpp_sink_factory =
  55     GST_STATIC_PAD_TEMPLATE ("sink",
  56     GST_PAD_SINK,
  57     GST_PAD_ALWAYS,
  58     GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE
  59         ("{ NV12, YV12, I420, YUY2, UYVY, BGRA, BGRx }")
  60         ", " "interlace-mode = (string){ progressive, interleaved, mixed }" ";"
  61         GST_VIDEO_CAPS_MAKE_WITH_FEATURES (GST_CAPS_FEATURE_MEMORY_DMABUF,
  62             "{ NV12, BGRA, YUY2}")));
  63
  64 static GstStaticPadTemplate gst_msdkvpp_src_factory =
  65     GST_STATIC_PAD_TEMPLATE ("src",
  66     GST_PAD_SRC,
  67     GST_PAD_ALWAYS,
  68     GST_STATIC_CAPS (GST_VIDEO_CAPS_MAKE_WITH_FEATURES
  69         (GST_CAPS_FEATURE_MEMORY_DMABUF,
  70             "{ BGRA, YUY2, NV12}") ";"
  71         GST_VIDEO_CAPS_MAKE ("{ NV12, YUY2, BGRA, BGRx }") ", "
  72         "interlace-mode = (string){ progressive, interleaved, mixed }" ";"));
  73
  74 enum
  75 {
  76   PROP_0,
  77   PROP_HARDWARE,
  78   PROP_ASYNC_DEPTH,
  79   PROP_DENOISE,
  80   PROP_ROTATION,
  81   PROP_DEINTERLACE_MODE,
  82   PROP_DEINTERLACE_METHOD,
  83   PROP_HUE,
  84   PROP_SATURATION,
  85   PROP_BRIGHTNESS,
  86   PROP_CONTRAST,
  87   PROP_DETAIL,
  88   PROP_MIRRORING,
  89   PROP_SCALING_MODE,
  90   PROP_FORCE_ASPECT_RATIO,
  91   PROP_FRC_ALGORITHM,
  92   PROP_N,
  93 };
  94
  95 #define PROP_HARDWARE_DEFAULT            TRUE
  96 #define PROP_ASYNC_DEPTH_DEFAULT         1
  97 #define PROP_DENOISE_DEFAULT             0
  98 #define PROP_ROTATION_DEFAULT            MFX_ANGLE_0
  99 #define PROP_DEINTERLACE_MODE_DEFAULT    GST_MSDKVPP_DEINTERLACE_MODE_AUTO
 100 #define PROP_DEINTERLACE_METHOD_DEFAULT  MFX_DEINTERLACING_BOB
 101 #define PROP_HUE_DEFAULT                 0
 102 #define PROP_SATURATION_DEFAULT          1
 103 #define PROP_BRIGHTNESS_DEFAULT          0
 104 #define PROP_CONTRAST_DEFAULT            1
 105 #define PROP_DETAIL_DEFAULT              0
 106 #define PROP_MIRRORING_DEFAULT           MFX_MIRRORING_DISABLED
 107 #define PROP_SCALING_MODE_DEFAULT        MFX_SCALING_MODE_DEFAULT
 108 #define PROP_FORCE_ASPECT_RATIO_DEFAULT  TRUE
 109 #define PROP_FRC_ALGORITHM_DEFAULT       _MFX_FRC_ALGORITHM_NONE
 110
 111 #define gst_msdkvpp_parent_class parent_class
 112 G_DEFINE_TYPE (GstMsdkVPP, gst_msdkvpp, GST_TYPE_BASE_TRANSFORM);
 113
 114 typedef struct
 115 {
 116   mfxFrameSurface1 *surface;
 117   GstBuffer *buf;
 118 } MsdkSurface;
 119
 120 static void
 121 free_msdk_surface (MsdkSurface * surface)
 122 {
 123   if (surface->buf)
 124     gst_buffer_unref (surface->buf);
 125   g_slice_free (MsdkSurface, surface);
 126 }
 127
 128 static void
 129 gst_msdkvpp_add_extra_param (GstMsdkVPP * thiz, mfxExtBuffer * param)
 130 {
 131   if (thiz->num_extra_params < MAX_EXTRA_PARAMS) {
 132     thiz->extra_params[thiz->num_extra_params] = param;
 133     thiz->num_extra_params++;
 134   }
 135 }
 136
 137 static gboolean
 138 ensure_context (GstBaseTransform * trans)
 139 {
 140   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
 141
 142   if (gst_msdk_context_prepare (GST_ELEMENT_CAST (thiz), &thiz->context)) {
 143     GST_INFO_OBJECT (thiz, "Found context from neighbour %" GST_PTR_FORMAT,
 144         thiz->context);
 145
 146     if (gst_msdk_context_get_job_type (thiz->context) & GST_MSDK_JOB_VPP) {
 147       GstMsdkContext *parent_context;
 148
 149       parent_context = thiz->context;
 150       thiz->context = gst_msdk_context_new_with_parent (parent_context);
 151       gst_object_unref (parent_context);
 152
 153       GST_INFO_OBJECT (thiz,
 154           "Creating new context %" GST_PTR_FORMAT " with joined session",
 155           thiz->context);
 156     } else {
 157       gst_msdk_context_add_job_type (thiz->context, GST_MSDK_JOB_VPP);
 158     }
 159   } else {
 160     if (!gst_msdk_context_ensure_context (GST_ELEMENT_CAST (thiz),
 161             thiz->hardware, GST_MSDK_JOB_VPP))
 162       return FALSE;
 163     GST_INFO_OBJECT (thiz, "Creating new context %" GST_PTR_FORMAT,
 164         thiz->context);
 165   }
 166
 167   gst_msdk_context_add_shared_async_depth (thiz->context, thiz->async_depth);
 168
 169   return TRUE;
 170 }
 171
 172 static GstBuffer *
 173 create_output_buffer (GstMsdkVPP * thiz)
 174 {
 175   GstBuffer *outbuf;
 176   GstFlowReturn ret;
 177   GstBufferPool *pool = thiz->srcpad_buffer_pool;
 178
 179   g_return_val_if_fail (pool != NULL, NULL);
 180
 181   if (!gst_buffer_pool_is_active (pool) &&
 182       !gst_buffer_pool_set_active (pool, TRUE))
 183     goto error_activate_pool;
 184
 185   outbuf = NULL;
 186   ret = gst_buffer_pool_acquire_buffer (pool, &outbuf, NULL);
 187   if (ret != GST_FLOW_OK || !outbuf)
 188     goto error_create_buffer;
 189
 190   return outbuf;
 191
 192   /* ERRORS */
 193 error_activate_pool:
 194   {
 195     GST_ERROR_OBJECT (thiz, "failed to activate output video buffer pool");
 196     return NULL;
 197   }
 198 error_create_buffer:
 199   {
 200     GST_ERROR_OBJECT (thiz, "failed to create output video buffer");
 201     return NULL;
 202   }
 203 }
 204
 205 static GstFlowReturn
 206 gst_msdkvpp_prepare_output_buffer (GstBaseTransform * trans,
 207     GstBuffer * inbuf, GstBuffer ** outbuf_ptr)
 208 {
 209   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
 210
 211   if (gst_base_transform_is_passthrough (trans)) {
 212     *outbuf_ptr = inbuf;
 213     return GST_FLOW_OK;
 214   }
 215
 216   *outbuf_ptr = create_output_buffer (thiz);
 217   return *outbuf_ptr ? GST_FLOW_OK : GST_FLOW_ERROR;
 218 }
 219
 220 static GstBufferPool *
 221 gst_msdkvpp_create_buffer_pool (GstMsdkVPP * thiz, GstPadDirection direction,
 222     GstCaps * caps, guint min_num_buffers)
 223 {
 224   GstBufferPool *pool = NULL;
 225   GstStructure *config;
 226   GstAllocator *allocator = NULL;
 227   GstVideoInfo info;
 228   GstVideoInfo *pool_info = NULL;
 229   GstVideoAlignment align;
 230   GstAllocationParams params = { 0, 31, 0, 0, };
 231   mfxFrameAllocResponse *alloc_resp = NULL;
 232   gboolean use_dmabuf = FALSE;
 233
 234   if (direction == GST_PAD_SINK) {
 235     alloc_resp = &thiz->in_alloc_resp;
 236     pool_info = &thiz->sinkpad_buffer_pool_info;
 237     use_dmabuf = thiz->use_sinkpad_dmabuf;
 238   } else if (direction == GST_PAD_SRC) {
 239     alloc_resp = &thiz->out_alloc_resp;
 240     pool_info = &thiz->srcpad_buffer_pool_info;
 241     use_dmabuf = thiz->use_srcpad_dmabuf;
 242   }
 243
 244   pool = gst_msdk_buffer_pool_new (thiz->context, alloc_resp);
 245   if (!pool)
 246     goto error_no_pool;
 247
 248   if (!gst_video_info_from_caps (&info, caps))
 249     goto error_no_video_info;
 250
 251   gst_msdk_set_video_alignment (&info, &align);
 252   gst_video_info_align (&info, &align);
 253
 254   if (use_dmabuf)
 255     allocator =
 256         gst_msdk_dmabuf_allocator_new (thiz->context, &info, alloc_resp);
 257   else if (thiz->use_video_memory)
 258     allocator = gst_msdk_video_allocator_new (thiz->context, &info, alloc_resp);
 259   else
 260     allocator = gst_msdk_system_allocator_new (&info);
 261
 262   if (!allocator)
 263     goto error_no_allocator;
 264
 265   config = gst_buffer_pool_get_config (GST_BUFFER_POOL_CAST (pool));
 266   gst_buffer_pool_config_set_params (config, caps, info.size, min_num_buffers,
 267       0);
 268
 269   gst_buffer_pool_config_add_option (config, GST_BUFFER_POOL_OPTION_VIDEO_META);
 270   gst_buffer_pool_config_add_option (config,
 271       GST_BUFFER_POOL_OPTION_VIDEO_ALIGNMENT);
 272   if (thiz->use_video_memory) {
 273     gst_buffer_pool_config_add_option (config,
 274         GST_BUFFER_POOL_OPTION_MSDK_USE_VIDEO_MEMORY);
 275     if (use_dmabuf)
 276       gst_buffer_pool_config_add_option (config,
 277           GST_BUFFER_POOL_OPTION_MSDK_USE_DMABUF);
 278   }
 279
 280   gst_buffer_pool_config_set_video_alignment (config, &align);
 281   gst_buffer_pool_config_set_allocator (config, allocator, &params);
 282   gst_object_unref (allocator);
 283
 284   if (!gst_buffer_pool_set_config (pool, config))
 285     goto error_pool_config;
 286
 287   /* Updating pool_info with algined info of allocator */
 288   *pool_info = info;
 289
 290   return pool;
 291
 292 error_no_pool:
 293   {
 294     GST_INFO_OBJECT (thiz, "Failed to create bufferpool");
 295     return NULL;
 296   }
 297 error_no_video_info:
 298   {
 299     GST_INFO_OBJECT (thiz, "Failed to get Video info from caps");
 300     return NULL;
 301   }
 302 error_no_allocator:
 303   {
 304     GST_INFO_OBJECT (thiz, "Failed to create allocator");
 305     if (pool)
 306       gst_object_unref (pool);
 307     return NULL;
 308   }
 309 error_pool_config:
 310   {
 311     GST_INFO_OBJECT (thiz, "Failed to set config");
 312     if (pool)
 313       gst_object_unref (pool);
 314     if (allocator)
 315       gst_object_unref (allocator);
 316     return NULL;
 317   }
 318 }
 319
 320 static gboolean
 321 _gst_caps_has_feature (const GstCaps * caps, const gchar * feature)
 322 {
 323   guint i;
 324
 325   for (i = 0; i < gst_caps_get_size (caps); i++) {
 326     GstCapsFeatures *const features = gst_caps_get_features (caps, i);
 327     /* Skip ANY features, we need an exact match for correct evaluation */
 328     if (gst_caps_features_is_any (features))
 329       continue;
 330     if (gst_caps_features_contains (features, feature))
 331       return TRUE;
 332   }
 333   return FALSE;
 334 }
 335
 336 static gboolean
 337 gst_msdkvpp_decide_allocation (GstBaseTransform * trans, GstQuery * query)
 338 {
 339   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
 340   GstVideoInfo info;
 341   GstBufferPool *pool = NULL;
 342   GstStructure *config = NULL;
 343   GstCaps *caps;
 344   guint size = 0, min_buffers = 0, max_buffers = 0;
 345   GstAllocator *allocator = NULL;
 346   GstAllocationParams params;
 347   gboolean update_pool = FALSE;
 348
 349   gst_query_parse_allocation (query, &caps, NULL);
 350   if (!caps) {
 351     GST_ERROR_OBJECT (thiz, "Failed to parse the decide_allocation caps");
 352     return FALSE;
 353   }
 354   if (!gst_video_info_from_caps (&info, caps)) {
 355     GST_ERROR_OBJECT (thiz, "Failed to get video info");
 356     return FALSE;
 357   }
 358   /* if downstream allocation query supports dmabuf-capsfeatures,
 359    * we do allocate dmabuf backed memory */
 360   if (_gst_caps_has_feature (caps, GST_CAPS_FEATURE_MEMORY_DMABUF)) {
 361     GST_INFO_OBJECT (thiz, "MSDK VPP srcpad uses DMABuf memory");
 362     thiz->use_srcpad_dmabuf = TRUE;
 363   }
 364
 365   if (gst_query_find_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL))
 366     thiz->add_video_meta = TRUE;
 367   else
 368     thiz->add_video_meta = FALSE;
 369
 370   /* Check whether the query has pool */
 371   if (gst_query_get_n_allocation_pools (query) > 0)
 372     update_pool = TRUE;
 373
 374   /* increase the min_buffers with number of concurrent vpp operations */
 375   min_buffers += thiz->async_depth;
 376
 377   /* invalidate the cached pool if there is an allocation_query */
 378   if (thiz->srcpad_buffer_pool)
 379     gst_object_unref (thiz->srcpad_buffer_pool);
 380
 381   /* Always create a pool for vpp out buffers. Each of the msdk element
 382    * has to create it's own mfxsurfacepool which is an msdk contraint.
 383    * For eg: Each Msdk component (vpp, dec and enc) will invoke the external
 384    * Frame allocator for video-memory usage.So sharing the pool between
 385    * gst-msdk elements might not be a good idea, rather each element
 386    * can check the buffer type (whether it is from msdk-buffer pool)
 387    * to make sure there is no copy. Since we share the context between
 388    * msdk elements, using buffers from one sdk's framealloator in another
 389    * sdk-components is perfectly fine */
 390   pool = gst_msdkvpp_create_buffer_pool (thiz, GST_PAD_SRC, caps, min_buffers);
 391   thiz->srcpad_buffer_pool = pool;
 392
 393   /* get the configured pool properties inorder to set in query */
 394   config = gst_buffer_pool_get_config (pool);
 395   gst_buffer_pool_config_get_params (config, &caps, &size, &min_buffers,
 396       &max_buffers);
 397   if (gst_buffer_pool_config_get_allocator (config, &allocator, &params))
 398     gst_query_add_allocation_param (query, allocator, &params);
 399   gst_structure_free (config);
 400
 401   if (update_pool)
 402     gst_query_set_nth_allocation_pool (query, 0, pool, size, min_buffers,
 403         max_buffers);
 404   else
 405     gst_query_add_allocation_pool (query, pool, size, min_buffers, max_buffers);
 406
 407   gst_query_add_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
 408
 409   /* Fixme if downstream doesn't have videometa support, msdkvpp should
 410    * copy the output buffers */
 411
 412   return TRUE;
 413 }
 414
 415 static gboolean
 416 gst_msdkvpp_propose_allocation (GstBaseTransform * trans,
 417     GstQuery * decide_query, GstQuery * query)
 418 {
 419   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
 420   GstVideoInfo info;
 421   GstBufferPool *pool = NULL;
 422   GstAllocator *allocator = NULL;
 423   GstCaps *caps;
 424   GstStructure *config;
 425   gboolean need_pool;
 426   GstAllocationParams params;
 427   guint size;
 428   guint min_buffers = thiz->async_depth + 1;
 429
 430   gst_query_parse_allocation (query, &caps, &need_pool);
 431   if (!caps) {
 432     GST_ERROR_OBJECT (thiz, "Failed to parse the allocation caps");
 433     return FALSE;
 434   }
 435
 436   if (!gst_video_info_from_caps (&info, caps)) {
 437     GST_ERROR_OBJECT (thiz, "Failed to get video info");
 438     return FALSE;
 439   }
 440
 441   /* if upstream allocation query supports dmabuf-capsfeatures,
 442    * we do allocate dmabuf backed memory */
 443   if (_gst_caps_has_feature (caps, GST_CAPS_FEATURE_MEMORY_DMABUF)) {
 444     GST_INFO_OBJECT (thiz, "MSDK VPP srcpad uses DMABuf memory");
 445     thiz->use_sinkpad_dmabuf = TRUE;
 446   }
 447
 448   if (need_pool) {
 449     /* alwys provide a new pool for upstream to help re-negotiation
 450      * more info here: https://bugzilla.gnome.org/show_bug.cgi?id=748344 */
 451     pool = gst_msdkvpp_create_buffer_pool (thiz, GST_PAD_SINK, caps,
 452         min_buffers);
 453   }
 454
 455   /* Update the internal pool if any allocation attribute changed */
 456   if (!gst_video_info_is_equal (&thiz->sinkpad_buffer_pool_info, &info)) {
 457     gst_object_unref (thiz->sinkpad_buffer_pool);
 458     thiz->sinkpad_buffer_pool = gst_msdkvpp_create_buffer_pool (thiz,
 459         GST_PAD_SINK, caps, min_buffers);
 460   }
 461
 462   /* get the size and allocator params from configured pool and set it in query */
 463   if (!need_pool)
 464     pool = gst_object_ref (thiz->sinkpad_buffer_pool);
 465   config = gst_buffer_pool_get_config (GST_BUFFER_POOL_CAST (pool));
 466   gst_buffer_pool_config_get_params (config, NULL, &size, NULL, NULL);
 467   if (gst_buffer_pool_config_get_allocator (config, &allocator, &params))
 468     gst_query_add_allocation_param (query, allocator, &params);
 469   gst_structure_free (config);
 470
 471   /* if upstream does't have a pool requirement, set only
 472    *  size, min_buffers and max_buffers in query */
 473   gst_query_add_allocation_pool (query, need_pool ? pool : NULL, size,
 474       min_buffers, 0);
 475   gst_query_add_allocation_meta (query, GST_VIDEO_META_API_TYPE, NULL);
 476
 477   gst_object_unref (pool);
 478
 479   return GST_BASE_TRANSFORM_CLASS (parent_class)->propose_allocation (trans,
 480       decide_query, query);
 481 }
 482
 483 static MsdkSurface *
 484 get_surface_from_pool (GstMsdkVPP * thiz, GstBufferPool * pool,
 485     GstBufferPoolAcquireParams * params)
 486 {
 487   GstBuffer *new_buffer;
 488   mfxFrameSurface1 *new_surface;
 489   MsdkSurface *msdk_surface;
 490
 491   if (!gst_buffer_pool_is_active (pool) &&
 492       !gst_buffer_pool_set_active (pool, TRUE)) {
 493     GST_ERROR_OBJECT (pool, "failed to activate buffer pool");
 494     return NULL;
 495   }
 496
 497   if (gst_buffer_pool_acquire_buffer (pool, &new_buffer, params) != GST_FLOW_OK) {
 498     GST_ERROR_OBJECT (pool, "failed to acquire a buffer from pool");
 499     return NULL;
 500   }
 501
 502   if (gst_msdk_is_msdk_buffer (new_buffer))
 503     new_surface = gst_msdk_get_surface_from_buffer (new_buffer);
 504   else {
 505     GST_ERROR_OBJECT (pool, "the acquired memory is not MSDK memory");
 506     return NULL;
 507   }
 508
 509   msdk_surface = g_slice_new0 (MsdkSurface);
 510   msdk_surface->surface = new_surface;
 511   msdk_surface->buf = new_buffer;
 512
 513   return msdk_surface;
 514 }
 515
 516 static gboolean
 517 import_dmabuf_to_msdk_surface (GstMsdkVPP * thiz, GstBuffer * buf,
 518     MsdkSurface * msdk_surface)
 519 {
 520   GstMemory *mem = NULL;
 521   GstVideoInfo vinfo;
 522   GstVideoMeta *vmeta;
 523   GstMsdkMemoryID *msdk_mid = NULL;
 524   mfxFrameSurface1 *mfx_surface = NULL;
 525   gint fd, i;
 526
 527   mem = gst_buffer_peek_memory (buf, 0);
 528   fd = gst_dmabuf_memory_get_fd (mem);
 529   if (fd < 0)
 530     return FALSE;
 531
 532   vinfo = thiz->sinkpad_info;
 533
 534   /* Update offset/stride/size if there is VideoMeta attached to
 535    * the buffer */
 536   vmeta = gst_buffer_get_video_meta (buf);
 537   if (vmeta) {
 538     if (GST_VIDEO_INFO_FORMAT (&vinfo) != vmeta->format ||
 539         GST_VIDEO_INFO_WIDTH (&vinfo) != vmeta->width ||
 540         GST_VIDEO_INFO_HEIGHT (&vinfo) != vmeta->height ||
 541         GST_VIDEO_INFO_N_PLANES (&vinfo) != vmeta->n_planes) {
 542       GST_ERROR_OBJECT (thiz, "VideoMeta attached to buffer is not matching"
 543           "the negotiated width/height/format");
 544       return FALSE;
 545     }
 546     for (i = 0; i < GST_VIDEO_INFO_N_PLANES (&vinfo); ++i) {
 547       GST_VIDEO_INFO_PLANE_OFFSET (&vinfo, i) = vmeta->offset[i];
 548       GST_VIDEO_INFO_PLANE_STRIDE (&vinfo, i) = vmeta->stride[i];
 549     }
 550     GST_VIDEO_INFO_SIZE (&vinfo) = gst_buffer_get_size (buf);
 551   }
 552
 553   /* Upstream neither accepted the msdk pool nor the msdk buffer size restrictions.
 554    * Current media-driver and GMMLib will fail due to strict memory size restrictions.
 555    * Ideally, media-driver should accept what ever memory coming from other drivers
 556    * in case of dmabuf-import and this is how the intel-vaapi-driver works.
 557    * For now, in order to avoid any crash we check the buffer size and fallback
 558    * to copy frame method.
 559    *
 560    * See this: https://github.com/intel/media-driver/issues/169
 561    * */
 562   if (GST_VIDEO_INFO_SIZE (&vinfo) <
 563       GST_VIDEO_INFO_SIZE (&thiz->sinkpad_buffer_pool_info))
 564     return FALSE;
 565
 566   mfx_surface = msdk_surface->surface;
 567   msdk_mid = (GstMsdkMemoryID *) mfx_surface->Data.MemId;
 568
 569   /* release the internal memory storage of associated mfxSurface */
 570   gst_msdk_replace_mfx_memid (thiz->context, mfx_surface, VA_INVALID_ID);
 571
 572   /* export dmabuf to vasurface */
 573   if (!gst_msdk_export_dmabuf_to_vasurface (thiz->context, &vinfo, fd,
 574           msdk_mid->surface))
 575     return FALSE;
 576
 577   return TRUE;
 578 }
 579
 580 static MsdkSurface *
 581 get_msdk_surface_from_input_buffer (GstMsdkVPP * thiz, GstBuffer * inbuf)
 582 {
 583   GstVideoFrame src_frame, out_frame;
 584   MsdkSurface *msdk_surface;
 585   GstMemory *mem = NULL;
 586
 587   if (gst_msdk_is_msdk_buffer (inbuf)) {
 588     msdk_surface = g_slice_new0 (MsdkSurface);
 589     msdk_surface->surface = gst_msdk_get_surface_from_buffer (inbuf);
 590     msdk_surface->buf = gst_buffer_ref (inbuf);
 591     return msdk_surface;
 592   }
 593
 594   /* If upstream hasn't accpeted the proposed msdk bufferpool,
 595    * just copy frame (if not dmabuf backed) to msdk buffer and
 596    * take a surface from it.   */
 597   if (!(msdk_surface =
 598           get_surface_from_pool (thiz, thiz->sinkpad_buffer_pool, NULL)))
 599     goto error;
 600
 601 #ifndef _WIN32
 602   /************ dmabuf-import ************* */
 603   /* if upstream provided a dmabuf backed memory, but not an msdk
 604    * buffer, we could export the dmabuf to underlined vasurface */
 605   mem = gst_buffer_peek_memory (inbuf, 0);
 606   if (gst_is_dmabuf_memory (mem)) {
 607     if (import_dmabuf_to_msdk_surface (thiz, inbuf, msdk_surface))
 608       return msdk_surface;
 609     else
 610       GST_INFO_OBJECT (thiz, "Upstream dmabuf-backed memory is not imported"
 611           "to the msdk surface, fall back to the copy input frame method");
 612   }
 613 #endif
 614
 615   if (!gst_video_frame_map (&src_frame, &thiz->sinkpad_info, inbuf,
 616           GST_MAP_READ)) {
 617     GST_ERROR_OBJECT (thiz, "failed to map the frame for source");
 618     goto error;
 619   }
 620
 621   if (!gst_video_frame_map (&out_frame, &thiz->sinkpad_buffer_pool_info,
 622           msdk_surface->buf, GST_MAP_WRITE)) {
 623     GST_ERROR_OBJECT (thiz, "failed to map the frame for destination");
 624     gst_video_frame_unmap (&src_frame);
 625     goto error;
 626   }
 627
 628   if (!gst_video_frame_copy (&out_frame, &src_frame)) {
 629     GST_ERROR_OBJECT (thiz, "failed to copy frame");
 630     gst_video_frame_unmap (&out_frame);
 631     gst_video_frame_unmap (&src_frame);
 632     goto error;
 633   }
 634
 635   gst_video_frame_unmap (&out_frame);
 636   gst_video_frame_unmap (&src_frame);
 637
 638   return msdk_surface;
 639
 640 error:
 641   return NULL;
 642 }
 643
 644 static GstFlowReturn
 645 gst_msdkvpp_transform (GstBaseTransform * trans, GstBuffer * inbuf,
 646     GstBuffer * outbuf)
 647 {
 648   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
 649   GstClockTime timestamp;
 650   GstFlowReturn ret = GST_FLOW_OK;
 651   mfxSession session;
 652   mfxSyncPoint sync_point = NULL;
 653   mfxStatus status;
 654   MsdkSurface *in_surface = NULL;
 655   MsdkSurface *out_surface = NULL;
 656
 657   timestamp = GST_BUFFER_TIMESTAMP (inbuf);
 658
 659   in_surface = get_msdk_surface_from_input_buffer (thiz, inbuf);
 660   if (!in_surface)
 661     return GST_FLOW_ERROR;
 662
 663   if (gst_msdk_is_msdk_buffer (outbuf)) {
 664     out_surface = g_slice_new0 (MsdkSurface);
 665     out_surface->surface = gst_msdk_get_surface_from_buffer (outbuf);
 666   } else {
 667     GST_ERROR ("Failed to get msdk outsurface!");
 668     return GST_FLOW_ERROR;
 669   }
 670
 671   session = gst_msdk_context_get_session (thiz->context);
 672
 673   /* outer loop is for handling FrameRate Control and deinterlace use cases */
 674   do {
 675     for (;;) {
 676       status =
 677           MFXVideoVPP_RunFrameVPPAsync (session, in_surface->surface,
 678           out_surface->surface, NULL, &sync_point);
 679       if (status != MFX_WRN_DEVICE_BUSY)
 680         break;
 681       /* If device is busy, wait 1ms and retry, as per MSDK's recommendation */
 682       g_usleep (1000);
 683     };
 684
 685     if (status != MFX_ERR_NONE && status != MFX_ERR_MORE_DATA
 686         && status != MFX_ERR_MORE_SURFACE)
 687       goto vpp_error;
 688
 689     /* No output generated */
 690     if (status == MFX_ERR_MORE_DATA)
 691       goto error_more_data;
 692
 693     if (sync_point)
 694       MFXVideoCORE_SyncOperation (session, sync_point, 10000);
 695
 696     /* More than one output buffers are generated */
 697     if (status == MFX_ERR_MORE_SURFACE) {
 698       GST_BUFFER_TIMESTAMP (outbuf) = timestamp;
 699       GST_BUFFER_DURATION (outbuf) = thiz->buffer_duration;
 700       timestamp += thiz->buffer_duration;
 701       ret = gst_pad_push (GST_BASE_TRANSFORM_SRC_PAD (trans), outbuf);
 702       if (ret != GST_FLOW_OK)
 703         goto error_push_buffer;
 704       outbuf = create_output_buffer (thiz);
 705     } else {
 706       GST_BUFFER_TIMESTAMP (outbuf) = timestamp;
 707       GST_BUFFER_DURATION (outbuf) = thiz->buffer_duration;
 708     }
 709   } while (status == MFX_ERR_MORE_SURFACE);
 710
 711   free_msdk_surface (in_surface);
 712   return ret;
 713
 714 vpp_error:
 715   GST_ERROR_OBJECT (thiz, "MSDK Failed to do VPP");
 716   free_msdk_surface (in_surface);
 717   free_msdk_surface (out_surface);
 718   return GST_FLOW_ERROR;
 719
 720 error_more_data:
 721   GST_WARNING_OBJECT (thiz,
 722       "MSDK Requries additional input for processing, "
 723       "Retruning FLOW_DROPPED since no output buffer was generated");
 724   free_msdk_surface (in_surface);
 725   return GST_BASE_TRANSFORM_FLOW_DROPPED;
 726
 727 error_push_buffer:
 728   {
 729     free_msdk_surface (in_surface);
 730     free_msdk_surface (out_surface);
 731     GST_DEBUG_OBJECT (thiz, "failed to push output buffer: %s",
 732         gst_flow_get_name (ret));
 733     return ret;
 734   }
 735 }
 736
 737 static void
 738 gst_msdkvpp_close (GstMsdkVPP * thiz)
 739 {
 740   mfxStatus status;
 741
 742   if (!thiz->context)
 743     return;
 744
 745   GST_DEBUG_OBJECT (thiz, "Closing VPP 0x%p", thiz->context);
 746   status = MFXVideoVPP_Close (gst_msdk_context_get_session (thiz->context));
 747   if (status != MFX_ERR_NONE && status != MFX_ERR_NOT_INITIALIZED) {
 748     GST_WARNING_OBJECT (thiz, "Encoder close failed (%s)",
 749         msdk_status_to_string (status));
 750   }
 751
 752   if (thiz->context)
 753     gst_object_replace ((GstObject **) & thiz->context, NULL);
 754
 755   memset (&thiz->param, 0, sizeof (thiz->param));
 756
 757   if (thiz->sinkpad_buffer_pool)
 758     gst_object_unref (thiz->sinkpad_buffer_pool);
 759   thiz->sinkpad_buffer_pool = NULL;
 760   if (thiz->srcpad_buffer_pool)
 761     gst_object_unref (thiz->srcpad_buffer_pool);
 762   thiz->srcpad_buffer_pool = NULL;
 763
 764   thiz->buffer_duration = GST_CLOCK_TIME_NONE;
 765   gst_video_info_init (&thiz->sinkpad_info);
 766   gst_video_info_init (&thiz->srcpad_info);
 767 }
 768
 769 static void
 770 ensure_filters (GstMsdkVPP * thiz)
 771 {
 772
 773   /* Denoise */
 774   if (thiz->flags & GST_MSDK_FLAG_DENOISE) {
 775     mfxExtVPPDenoise *mfx_denoise = &thiz->mfx_denoise;
 776     mfx_denoise->Header.BufferId = MFX_EXTBUFF_VPP_DENOISE;
 777     mfx_denoise->Header.BufferSz = sizeof (mfxExtVPPDenoise);
 778     mfx_denoise->DenoiseFactor = thiz->denoise_factor;
 779     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_denoise);
 780   }
 781
 782   /* Rotation */
 783   if (thiz->flags & GST_MSDK_FLAG_ROTATION) {
 784     mfxExtVPPRotation *mfx_rotation = &thiz->mfx_rotation;
 785     mfx_rotation->Header.BufferId = MFX_EXTBUFF_VPP_ROTATION;
 786     mfx_rotation->Header.BufferSz = sizeof (mfxExtVPPRotation);
 787     mfx_rotation->Angle = thiz->rotation;
 788     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_rotation);
 789   }
 790
 791   /* Deinterlace */
 792   if (thiz->flags & GST_MSDK_FLAG_DEINTERLACE) {
 793     mfxExtVPPDeinterlacing *mfx_deinterlace = &thiz->mfx_deinterlace;
 794     mfx_deinterlace->Header.BufferId = MFX_EXTBUFF_VPP_DEINTERLACING;
 795     mfx_deinterlace->Header.BufferSz = sizeof (mfxExtVPPDeinterlacing);
 796     mfx_deinterlace->Mode = thiz->deinterlace_method;
 797     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_deinterlace);
 798   }
 799
 800   /* Colorbalance(ProcAmp) */
 801   if (thiz->flags & (GST_MSDK_FLAG_HUE | GST_MSDK_FLAG_SATURATION |
 802           GST_MSDK_FLAG_BRIGHTNESS | GST_MSDK_FLAG_CONTRAST)) {
 803     mfxExtVPPProcAmp *mfx_procamp = &thiz->mfx_procamp;
 804     mfx_procamp->Header.BufferId = MFX_EXTBUFF_VPP_PROCAMP;
 805     mfx_procamp->Header.BufferSz = sizeof (mfxExtVPPProcAmp);
 806     mfx_procamp->Hue = thiz->hue;
 807     mfx_procamp->Saturation = thiz->saturation;
 808     mfx_procamp->Brightness = thiz->brightness;
 809     mfx_procamp->Contrast = thiz->contrast;
 810     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_procamp);
 811   }
 812
 813   /* Detail/Edge enhancement */
 814   if (thiz->flags & GST_MSDK_FLAG_DETAIL) {
 815     mfxExtVPPDetail *mfx_detail = &thiz->mfx_detail;
 816     mfx_detail->Header.BufferId = MFX_EXTBUFF_VPP_DETAIL;
 817     mfx_detail->Header.BufferSz = sizeof (mfxExtVPPDetail);
 818     mfx_detail->DetailFactor = thiz->detail;
 819     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_detail);
 820   }
 821
 822   /* Mirroring */
 823   if (thiz->flags & GST_MSDK_FLAG_MIRRORING) {
 824     mfxExtVPPMirroring *mfx_mirroring = &thiz->mfx_mirroring;
 825     mfx_mirroring->Header.BufferId = MFX_EXTBUFF_VPP_MIRRORING;
 826     mfx_mirroring->Header.BufferSz = sizeof (mfxExtVPPMirroring);
 827     mfx_mirroring->Type = thiz->mirroring;
 828     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_mirroring);
 829   }
 830
 831   /* Scaling Mode */
 832   if (thiz->flags & GST_MSDK_FLAG_SCALING_MODE) {
 833     mfxExtVPPScaling *mfx_scaling = &thiz->mfx_scaling;
 834     mfx_scaling->Header.BufferId = MFX_EXTBUFF_VPP_SCALING;
 835     mfx_scaling->Header.BufferSz = sizeof (mfxExtVPPScaling);
 836     mfx_scaling->ScalingMode = thiz->scaling_mode;
 837     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_scaling);
 838   }
 839
 840   /* FRC */
 841   if (thiz->flags & GST_MSDK_FLAG_FRC) {
 842     mfxExtVPPFrameRateConversion *mfx_frc = &thiz->mfx_frc;
 843     mfx_frc->Header.BufferId = MFX_EXTBUFF_VPP_FRAME_RATE_CONVERSION;
 844     mfx_frc->Header.BufferSz = sizeof (mfxExtVPPFrameRateConversion);
 845     mfx_frc->Algorithm = thiz->frc_algm;
 846     gst_msdkvpp_add_extra_param (thiz, (mfxExtBuffer *) mfx_frc);
 847   }
 848 }
 849
 850 static void
 851 gst_msdkvpp_set_passthrough (GstMsdkVPP * thiz)
 852 {
 853   gboolean passthrough = TRUE;
 854
 855   /* no passthrough if any of the filter algorithm is enabled */
 856   if (thiz->flags)
 857     passthrough = FALSE;
 858
 859   /* vpp could be needed in some specific circumstances, for eg:
 860    * input surface is dmabuf and output must be videomemory. So far
 861    * the underline iHD driver doesn't seems to support dmabuf mapping,
 862    * so we could explicitly ask msdkvpp to provide non-dambuf videomemory
 863    * surfaces as output thourgh capsfileters */
 864   if (thiz->need_vpp)
 865     passthrough = FALSE;
 866
 867   /* no passthrough if there is change in out width,height or format */
 868   if (GST_VIDEO_INFO_WIDTH (&thiz->sinkpad_info) !=
 869       GST_VIDEO_INFO_WIDTH (&thiz->srcpad_info)
 870       || GST_VIDEO_INFO_HEIGHT (&thiz->sinkpad_info) !=
 871       GST_VIDEO_INFO_HEIGHT (&thiz->srcpad_info)
 872       || GST_VIDEO_INFO_FORMAT (&thiz->sinkpad_info) !=
 873       GST_VIDEO_INFO_FORMAT (&thiz->srcpad_info))
 874     passthrough = FALSE;
 875
 876   gst_base_transform_set_passthrough (GST_BASE_TRANSFORM (thiz), passthrough);
 877 }
 878
 879 static gboolean
 880 gst_msdkvpp_initialize (GstMsdkVPP * thiz)
 881 {
 882   mfxSession session;
 883   mfxStatus status;
 884   mfxFrameAllocRequest request[2];
 885
 886   if (!thiz->context) {
 887     GST_WARNING_OBJECT (thiz, "No MSDK Context");
 888     return FALSE;
 889   }
 890
 891   GST_OBJECT_LOCK (thiz);
 892   session = gst_msdk_context_get_session (thiz->context);
 893
 894   if (thiz->use_video_memory) {
 895     gst_msdk_set_frame_allocator (thiz->context);
 896     thiz->param.IOPattern =
 897         MFX_IOPATTERN_IN_VIDEO_MEMORY | MFX_IOPATTERN_OUT_VIDEO_MEMORY;
 898   } else {
 899     thiz->param.IOPattern =
 900         MFX_IOPATTERN_IN_SYSTEM_MEMORY | MFX_IOPATTERN_OUT_SYSTEM_MEMORY;
 901   }
 902
 903   /* update input video attributes */
 904   gst_msdk_set_mfx_frame_info_from_video_info (&thiz->param.vpp.In,
 905       &thiz->sinkpad_info);
 906
 907   /* update output video attributes, only CSC and Scaling are supported for now */
 908   gst_msdk_set_mfx_frame_info_from_video_info (&thiz->param.vpp.Out,
 909       &thiz->srcpad_info);
 910
 911   /* use msdk frame rarte control if there is a mismatch in In & OUt fps  */
 912   if (GST_VIDEO_INFO_FPS_N (&thiz->srcpad_info) &&
 913       (GST_VIDEO_INFO_FPS_N (&thiz->sinkpad_info) !=
 914           GST_VIDEO_INFO_FPS_N (&thiz->srcpad_info)
 915           || GST_VIDEO_INFO_FPS_D (&thiz->sinkpad_info) !=
 916           GST_VIDEO_INFO_FPS_D (&thiz->srcpad_info))) {
 917     thiz->flags |= GST_MSDK_FLAG_FRC;
 918     /* So far this is the only algorithm which is working somewhat good */
 919     thiz->frc_algm = MFX_FRCALGM_PRESERVE_TIMESTAMP;
 920   }
 921
 922   /* work-around to avoid zero fps in msdk structure */
 923   if (!thiz->param.vpp.In.FrameRateExtN)
 924     thiz->param.vpp.In.FrameRateExtN = 30;
 925   if (!thiz->param.vpp.Out.FrameRateExtN)
 926     thiz->param.vpp.Out.FrameRateExtN = thiz->param.vpp.In.FrameRateExtN;
 927
 928   /* set vpp out picstruct as progressive if deinterlacing enabled */
 929   if (thiz->flags & GST_MSDK_FLAG_DEINTERLACE)
 930     thiz->param.vpp.Out.PicStruct = MFX_PICSTRUCT_PROGRESSIVE;
 931
 932   /* Enable the required filters */
 933   ensure_filters (thiz);
 934
 935   /* Add exteneded buffers */
 936   if (thiz->num_extra_params) {
 937     thiz->param.NumExtParam = thiz->num_extra_params;
 938     thiz->param.ExtParam = thiz->extra_params;
 939   }
 940
 941   /* validate parameters and allow the Media SDK to make adjustments */
 942   status = MFXVideoVPP_Query (session, &thiz->param, &thiz->param);
 943   if (status < MFX_ERR_NONE) {
 944     GST_ERROR_OBJECT (thiz, "Video VPP Query failed (%s)",
 945         msdk_status_to_string (status));
 946     goto no_vpp;
 947   } else if (status > MFX_ERR_NONE) {
 948     GST_WARNING_OBJECT (thiz, "Video VPP Query returned: %s",
 949         msdk_status_to_string (status));
 950   }
 951
 952   status = MFXVideoVPP_QueryIOSurf (session, &thiz->param, request);
 953   if (status < MFX_ERR_NONE) {
 954     GST_ERROR_OBJECT (thiz, "VPP Query IO surfaces failed (%s)",
 955         msdk_status_to_string (status));
 956     goto no_vpp;
 957   } else if (status > MFX_ERR_NONE) {
 958     GST_WARNING_OBJECT (thiz, "VPP Query IO surfaces returned: %s",
 959         msdk_status_to_string (status));
 960   }
 961
 962   if (thiz->use_video_memory) {
 963     /* Input surface pool pre-allocation */
 964     request[0].Type |= MFX_MEMTYPE_VIDEO_MEMORY_PROCESSOR_TARGET;
 965     if (thiz->use_sinkpad_dmabuf)
 966       request[0].Type |= MFX_MEMTYPE_EXPORT_FRAME;
 967     gst_msdk_frame_alloc (thiz->context, &(request[0]), &thiz->in_alloc_resp);
 968
 969     /* Output surface pool pre-allocation */
 970     request[1].Type |= MFX_MEMTYPE_VIDEO_MEMORY_PROCESSOR_TARGET;
 971     if (thiz->use_srcpad_dmabuf)
 972       request[1].Type |= MFX_MEMTYPE_EXPORT_FRAME;
 973     gst_msdk_frame_alloc (thiz->context, &(request[1]), &thiz->out_alloc_resp);
 974   }
 975
 976   thiz->in_num_surfaces = request[0].NumFrameSuggested;
 977   thiz->out_num_surfaces = request[1].NumFrameSuggested;
 978
 979
 980   status = MFXVideoVPP_Init (session, &thiz->param);
 981   if (status < MFX_ERR_NONE) {
 982     GST_ERROR_OBJECT (thiz, "Init failed (%s)", msdk_status_to_string (status));
 983     goto no_vpp;
 984   } else if (status > MFX_ERR_NONE) {
 985     GST_WARNING_OBJECT (thiz, "Init returned: %s",
 986         msdk_status_to_string (status));
 987   }
 988
 989   thiz->initialized = TRUE;
 990   GST_OBJECT_UNLOCK (thiz);
 991   return TRUE;
 992
 993 no_vpp:
 994   GST_OBJECT_UNLOCK (thiz);
 995   if (thiz->context)
 996     gst_object_replace ((GstObject **) & thiz->context, NULL);
 997   return FALSE;
 998 }
 999
1000 static gboolean
1001 gst_msdkvpp_set_caps (GstBaseTransform * trans, GstCaps * caps,
1002     GstCaps * out_caps)
1003 {
1004   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
1005   GstVideoInfo in_info, out_info;
1006   gboolean sinkpad_info_changed = FALSE;
1007   gboolean srcpad_info_changed = FALSE;
1008   gboolean deinterlace;
1009
1010   if (gst_caps_get_features (caps, 0) != gst_caps_get_features (out_caps, 0))
1011     thiz->need_vpp = 1;
1012
1013   gst_video_info_from_caps (&in_info, caps);
1014   gst_video_info_from_caps (&out_info, out_caps);
1015
1016   if (!gst_video_info_is_equal (&in_info, &thiz->sinkpad_info))
1017     sinkpad_info_changed = TRUE;
1018   if (!gst_video_info_is_equal (&out_info, &thiz->srcpad_info))
1019     srcpad_info_changed = TRUE;
1020
1021   thiz->sinkpad_info = in_info;
1022   thiz->srcpad_info = out_info;
1023 #ifndef _WIN32
1024   thiz->use_video_memory = TRUE;
1025 #else
1026   thiz->use_video_memory = FALSE;
1027 #endif
1028
1029   if (!sinkpad_info_changed && !srcpad_info_changed && thiz->initialized)
1030     return TRUE;
1031
1032   /* check for deinterlace requirement */
1033   deinterlace = gst_msdkvpp_is_deinterlace_enabled (thiz, &in_info);
1034   if (deinterlace)
1035     thiz->flags |= GST_MSDK_FLAG_DEINTERLACE;
1036
1037   thiz->buffer_duration = GST_VIDEO_INFO_FPS_N (&out_info) > 0 ?
1038       gst_util_uint64_scale (GST_SECOND, GST_VIDEO_INFO_FPS_D (&out_info),
1039       GST_VIDEO_INFO_FPS_N (&out_info)) : 0;
1040
1041   if (!gst_msdkvpp_initialize (thiz))
1042     return FALSE;
1043
1044   /* set passthrough according to filter operation change */
1045   gst_msdkvpp_set_passthrough (thiz);
1046
1047   /* Ensure sinkpad buffer pool */
1048   thiz->sinkpad_buffer_pool =
1049       gst_msdkvpp_create_buffer_pool (thiz, GST_PAD_SINK, caps,
1050       thiz->in_num_surfaces);
1051   if (!thiz->sinkpad_buffer_pool) {
1052     GST_ERROR_OBJECT (thiz, "Failed to ensure the sinkpad buffer pool");
1053     return FALSE;
1054   }
1055   /* Ensure a srcpad buffer pool */
1056   thiz->srcpad_buffer_pool =
1057       gst_msdkvpp_create_buffer_pool (thiz, GST_PAD_SRC, out_caps,
1058       thiz->out_num_surfaces);
1059   if (!thiz->srcpad_buffer_pool) {
1060     GST_ERROR_OBJECT (thiz, "Failed to ensure the srcpad buffer pool");
1061     return FALSE;
1062   }
1063
1064   return TRUE;
1065 }
1066
1067 static gboolean
1068 pad_can_dmabuf (GstMsdkVPP * thiz, GstPadDirection direction, GstCaps * filter)
1069 {
1070   gboolean ret = FALSE;
1071   GstCaps *caps, *out_caps;
1072   GstPad *pad;
1073   GstBaseTransform *trans = GST_BASE_TRANSFORM (thiz);
1074
1075   if (direction == GST_PAD_SRC)
1076     pad = GST_BASE_TRANSFORM_SRC_PAD (trans);
1077   else
1078     pad = GST_BASE_TRANSFORM_SINK_PAD (trans);
1079
1080   /* make a copy of filter caps since we need to alter the structure
1081    * by adding dmabuf-capsfeatures */
1082   caps = gst_caps_copy (filter);
1083   gst_caps_set_features (caps, 0,
1084       gst_caps_features_from_string (GST_CAPS_FEATURE_MEMORY_DMABUF));
1085
1086   out_caps = gst_pad_peer_query_caps (pad, caps);
1087   if (!out_caps)
1088     goto done;
1089
1090   if (gst_caps_is_any (out_caps) || gst_caps_is_empty (out_caps)
1091       || out_caps == caps)
1092     goto done;
1093
1094   if (_gst_caps_has_feature (out_caps, GST_CAPS_FEATURE_MEMORY_DMABUF))
1095     ret = TRUE;
1096 done:
1097   if (caps)
1098     gst_caps_unref (caps);
1099   if (out_caps)
1100     gst_caps_unref (out_caps);
1101   return ret;
1102 }
1103
1104 static GstCaps *
1105 gst_msdkvpp_fixate_caps (GstBaseTransform * trans,
1106     GstPadDirection direction, GstCaps * caps, GstCaps * othercaps)
1107 {
1108   GstMsdkVPP *thiz = GST_MSDKVPP (trans);
1109   GstCaps *result = NULL;
1110   gboolean *use_dmabuf;
1111
1112   if (direction == GST_PAD_SRC) {
1113     result = gst_caps_fixate (result);
1114     use_dmabuf = &thiz->use_sinkpad_dmabuf;
1115   } else {
1116     result = gst_msdkvpp_fixate_srccaps (thiz, caps, othercaps);
1117     use_dmabuf = &thiz->use_srcpad_dmabuf;
1118   }
1119
1120   GST_DEBUG_OBJECT (trans, "fixated to %" GST_PTR_FORMAT, result);
1121   gst_caps_unref (othercaps);
1122
1123   if (pad_can_dmabuf (thiz,
1124           direction == GST_PAD_SRC ? GST_PAD_SINK : GST_PAD_SRC, result)) {
1125     gst_caps_set_features (result, 0,
1126         gst_caps_features_new (GST_CAPS_FEATURE_MEMORY_DMABUF, NULL));
1127     *use_dmabuf = TRUE;
1128   }
1129
1130   return result;
1131 }
1132
1133 /* Generic code for now, requires changes in future when we
1134  * add hardware query for supported formats, Framerate control etc */
1135 static GstCaps *
1136 gst_msdkvpp_transform_caps (GstBaseTransform * trans,
1137     GstPadDirection direction, GstCaps * caps, GstCaps * filter)
1138 {
1139   GstCaps *out_caps;
1140
1141   GST_DEBUG_OBJECT (trans,
1142       "Transforming caps %" GST_PTR_FORMAT " in direction %s", caps,
1143       (direction == GST_PAD_SINK) ? "sink" : "src");
1144
1145   if (direction == GST_PAD_SRC)
1146     out_caps = gst_static_pad_template_get_caps (&gst_msdkvpp_sink_factory);
1147   else
1148     out_caps = gst_static_pad_template_get_caps (&gst_msdkvpp_src_factory);
1149
1150   if (out_caps && filter) {
1151     GstCaps *intersection;
1152
1153     intersection = gst_caps_intersect_full (out_caps, filter,
1154         GST_CAPS_INTERSECT_FIRST);
1155     gst_caps_unref (out_caps);
1156     out_caps = intersection;
1157   }
1158
1159   GST_DEBUG_OBJECT (trans, "returning caps: %" GST_PTR_FORMAT, out_caps);
1160   return out_caps;
1161 }
1162
1163 static gboolean
1164 gst_msdkvpp_start (GstBaseTransform * trans)
1165 {
1166   if (!ensure_context (trans))
1167     return FALSE;
1168   return TRUE;
1169 }
1170
1171 static gboolean
1172 gst_msdkvpp_stop (GstBaseTransform * trans)
1173 {
1174   gst_msdkvpp_close (GST_MSDKVPP (trans));
1175   return TRUE;
1176 }
1177
1178 static void
1179 gst_msdkvpp_set_property (GObject * object, guint prop_id,
1180     const GValue * value, GParamSpec * pspec)
1181 {
1182   GstMsdkVPP *thiz = GST_MSDKVPP (object);
1183
1184   switch (prop_id) {
1185     case PROP_HARDWARE:
1186       thiz->hardware = g_value_get_boolean (value);
1187       break;
1188     case PROP_ASYNC_DEPTH:
1189       thiz->async_depth = g_value_get_uint (value);
1190       break;
1191     case PROP_DENOISE:
1192       thiz->denoise_factor = g_value_get_uint (value);
1193       thiz->flags |= GST_MSDK_FLAG_DENOISE;
1194       break;
1195     case PROP_ROTATION:
1196       thiz->rotation = g_value_get_enum (value);
1197       thiz->flags |= GST_MSDK_FLAG_ROTATION;
1198       break;
1199     case PROP_DEINTERLACE_MODE:
1200       thiz->deinterlace_mode = g_value_get_enum (value);
1201       break;
1202     case PROP_DEINTERLACE_METHOD:
1203       thiz->deinterlace_method = g_value_get_enum (value);
1204       break;
1205     case PROP_HUE:
1206       thiz->hue = g_value_get_float (value);
1207       thiz->flags |= GST_MSDK_FLAG_HUE;
1208       break;
1209     case PROP_SATURATION:
1210       thiz->saturation = g_value_get_float (value);
1211       thiz->flags |= GST_MSDK_FLAG_SATURATION;
1212       break;
1213     case PROP_BRIGHTNESS:
1214       thiz->brightness = g_value_get_float (value);
1215       thiz->flags |= GST_MSDK_FLAG_BRIGHTNESS;
1216       break;
1217     case PROP_CONTRAST:
1218       thiz->contrast = g_value_get_float (value);
1219       thiz->flags |= GST_MSDK_FLAG_CONTRAST;
1220       break;
1221     case PROP_DETAIL:
1222       thiz->detail = g_value_get_uint (value);
1223       thiz->flags |= GST_MSDK_FLAG_DETAIL;
1224       break;
1225     case PROP_MIRRORING:
1226       thiz->mirroring = g_value_get_enum (value);
1227       thiz->flags |= GST_MSDK_FLAG_MIRRORING;
1228       break;
1229     case PROP_SCALING_MODE:
1230       thiz->scaling_mode = g_value_get_enum (value);
1231       thiz->flags |= GST_MSDK_FLAG_SCALING_MODE;
1232       break;
1233     case PROP_FORCE_ASPECT_RATIO:
1234       thiz->keep_aspect = g_value_get_boolean (value);
1235       break;
1236     case PROP_FRC_ALGORITHM:
1237       thiz->frc_algm = g_value_get_enum (value);
1238       break;
1239     default:
1240       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1241       break;
1242   }
1243 }
1244
1245 static void
1246 gst_msdkvpp_get_property (GObject * object, guint prop_id,
1247     GValue * value, GParamSpec * pspec)
1248 {
1249   GstMsdkVPP *thiz = GST_MSDKVPP (object);
1250
1251   switch (prop_id) {
1252     case PROP_HARDWARE:
1253       g_value_set_boolean (value, thiz->hardware);
1254       break;
1255     case PROP_ASYNC_DEPTH:
1256       g_value_set_uint (value, thiz->async_depth);
1257       break;
1258     case PROP_DENOISE:
1259       g_value_set_uint (value, thiz->denoise_factor);
1260       break;
1261     case PROP_ROTATION:
1262       g_value_set_enum (value, thiz->rotation);
1263       break;
1264     case PROP_DEINTERLACE_MODE:
1265       g_value_set_enum (value, thiz->deinterlace_mode);
1266       break;
1267     case PROP_DEINTERLACE_METHOD:
1268       g_value_set_enum (value, thiz->deinterlace_method);
1269       break;
1270     case PROP_HUE:
1271       g_value_set_float (value, thiz->hue);
1272       break;
1273     case PROP_SATURATION:
1274       g_value_set_float (value, thiz->saturation);
1275       break;
1276     case PROP_BRIGHTNESS:
1277       g_value_set_float (value, thiz->brightness);
1278       break;
1279     case PROP_CONTRAST:
1280       g_value_set_float (value, thiz->contrast);
1281       break;
1282     case PROP_DETAIL:
1283       g_value_set_uint (value, thiz->detail);
1284       break;
1285     case PROP_MIRRORING:
1286       g_value_set_enum (value, thiz->mirroring);
1287       break;
1288     case PROP_SCALING_MODE:
1289       g_value_set_enum (value, thiz->scaling_mode);
1290       break;
1291     case PROP_FORCE_ASPECT_RATIO:
1292       g_value_set_boolean (value, thiz->keep_aspect);
1293       break;
1294     case PROP_FRC_ALGORITHM:
1295       g_value_set_enum (value, thiz->frc_algm);
1296       break;
1297     default:
1298       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
1299       break;
1300   }
1301 }
1302
1303 static void
1304 gst_msdkvpp_finalize (GObject * object)
1305 {
1306   G_OBJECT_CLASS (parent_class)->finalize (object);
1307 }
1308
1309 static void
1310 gst_msdkvpp_set_context (GstElement * element, GstContext * context)
1311 {
1312   GstMsdkContext *msdk_context = NULL;
1313   GstMsdkVPP *thiz = GST_MSDKVPP (element);
1314
1315   if (gst_msdk_context_get_context (context, &msdk_context)) {
1316     gst_object_replace ((GstObject **) & thiz->context,
1317         (GstObject *) msdk_context);
1318     gst_object_unref (msdk_context);
1319   }
1320
1321   GST_ELEMENT_CLASS (parent_class)->set_context (element, context);
1322 }
1323
1324 static void
1325 gst_msdkvpp_class_init (GstMsdkVPPClass * klass)
1326 {
1327   GObjectClass *gobject_class;
1328   GstElementClass *element_class;
1329   GstBaseTransformClass *trans_class;
1330   GParamSpec *obj_properties[PROP_N] = { NULL, };
1331
1332   gobject_class = G_OBJECT_CLASS (klass);
1333   element_class = GST_ELEMENT_CLASS (klass);
1334   trans_class = GST_BASE_TRANSFORM_CLASS (klass);
1335
1336   gobject_class->set_property = gst_msdkvpp_set_property;
1337   gobject_class->get_property = gst_msdkvpp_get_property;
1338   gobject_class->finalize = gst_msdkvpp_finalize;
1339
1340   element_class->set_context = gst_msdkvpp_set_context;
1341
1342   gst_element_class_add_static_pad_template (element_class,
1343       &gst_msdkvpp_src_factory);
1344   gst_element_class_add_static_pad_template (element_class,
1345       &gst_msdkvpp_sink_factory);
1346
1347   gst_element_class_set_static_metadata (element_class,
1348       "MSDK Video Postprocessor",
1349       "Filter/Converter/Video;Filter/Converter/Video/Scaler;"
1350       "Filter/Effect/Video;Filter/Effect/Video/Deinterlace",
1351       "A MediaSDK Video Postprocessing Filter",
1352       "Sreerenj Balachandrn <sreerenj.balachandran@intel.com>");
1353
1354   trans_class->start = GST_DEBUG_FUNCPTR (gst_msdkvpp_start);
1355   trans_class->stop = GST_DEBUG_FUNCPTR (gst_msdkvpp_stop);
1356   trans_class->transform_caps = GST_DEBUG_FUNCPTR (gst_msdkvpp_transform_caps);
1357   trans_class->fixate_caps = GST_DEBUG_FUNCPTR (gst_msdkvpp_fixate_caps);
1358   trans_class->set_caps = GST_DEBUG_FUNCPTR (gst_msdkvpp_set_caps);
1359   trans_class->transform = GST_DEBUG_FUNCPTR (gst_msdkvpp_transform);
1360   trans_class->propose_allocation =
1361       GST_DEBUG_FUNCPTR (gst_msdkvpp_propose_allocation);
1362   trans_class->decide_allocation =
1363       GST_DEBUG_FUNCPTR (gst_msdkvpp_decide_allocation);
1364   trans_class->prepare_output_buffer =
1365       GST_DEBUG_FUNCPTR (gst_msdkvpp_prepare_output_buffer);
1366
1367   obj_properties[PROP_HARDWARE] =
1368       g_param_spec_boolean ("hardware", "Hardware", "Enable hardware VPP",
1369       PROP_HARDWARE_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1370
1371   obj_properties[PROP_ASYNC_DEPTH] =
1372       g_param_spec_uint ("async-depth", "Async Depth",
1373       "Depth of asynchronous pipeline",
1374       1, 1, PROP_ASYNC_DEPTH_DEFAULT,
1375       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1376
1377   obj_properties[PROP_DENOISE] =
1378       g_param_spec_uint ("denoise", "Denoising factor",
1379       "Denoising Factor",
1380       0, 100, PROP_DENOISE_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1381
1382   obj_properties[PROP_ROTATION] =
1383       g_param_spec_enum ("rotation", "Rotation",
1384       "Rotation Angle", gst_msdkvpp_rotation_get_type (),
1385       PROP_ROTATION_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1386
1387   obj_properties[PROP_DEINTERLACE_MODE] =
1388       g_param_spec_enum ("deinterlace-mode", "Deinterlace Mode",
1389       "Deinterlace mode to use", gst_msdkvpp_deinterlace_mode_get_type (),
1390       PROP_DEINTERLACE_MODE_DEFAULT,
1391       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1392
1393   obj_properties[PROP_DEINTERLACE_METHOD] =
1394       g_param_spec_enum ("deinterlace-method", "Deinterlace Method",
1395       "Deinterlace method to use", gst_msdkvpp_deinterlace_method_get_type (),
1396       PROP_DEINTERLACE_METHOD_DEFAULT,
1397       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1398
1399   obj_properties[PROP_HUE] =
1400       g_param_spec_float ("hue", "Hue",
1401       "The hue of the video",
1402       -180, 180, PROP_HUE_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1403
1404   obj_properties[PROP_SATURATION] =
1405       g_param_spec_float ("saturation", "Saturation",
1406       "The Saturation of the video",
1407       0, 10, PROP_SATURATION_DEFAULT,
1408       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1409
1410   obj_properties[PROP_BRIGHTNESS] =
1411       g_param_spec_float ("brightness", "Brightness",
1412       "The Brightness of the video",
1413       -100, 100, PROP_BRIGHTNESS_DEFAULT,
1414       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1415
1416   obj_properties[PROP_CONTRAST] =
1417       g_param_spec_float ("contrast", "Contrast",
1418       "The Contrast of the video",
1419       0, 10, PROP_CONTRAST_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1420
1421   obj_properties[PROP_DETAIL] =
1422       g_param_spec_uint ("detail", "Detail",
1423       "The factor of detail/edge enhancement filter algorithm",
1424       0, 100, PROP_DETAIL_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1425
1426   obj_properties[PROP_MIRRORING] =
1427       g_param_spec_enum ("mirroring", "Mirroring",
1428       "The Mirroring type", gst_msdkvpp_mirroring_get_type (),
1429       PROP_MIRRORING_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1430
1431   obj_properties[PROP_SCALING_MODE] =
1432       g_param_spec_enum ("scaling-mode", "Scaling Mode",
1433       "The Scaling mode to use", gst_msdkvpp_scaling_mode_get_type (),
1434       PROP_SCALING_MODE_DEFAULT, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1435
1436   obj_properties[PROP_FORCE_ASPECT_RATIO] =
1437       g_param_spec_boolean ("force-aspect-ratio", "Force Aspect Ratio",
1438       "When enabled, scaling will respect original aspect ratio",
1439       PROP_FORCE_ASPECT_RATIO_DEFAULT,
1440       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1441
1442   obj_properties[PROP_FRC_ALGORITHM] =
1443       g_param_spec_enum ("frc-algorithm", "FrameRateControl Algorithm",
1444       "The Framerate Control Alogorithm to use",
1445       gst_msdkvpp_frc_algorithm_get_type (), PROP_FRC_ALGORITHM_DEFAULT,
1446       G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS);
1447
1448   g_object_class_install_properties (gobject_class, PROP_N, obj_properties);
1449 }
1450
1451 static void
1452 gst_msdkvpp_init (GstMsdkVPP * thiz)
1453 {
1454   thiz->initialized = FALSE;
1455   thiz->hardware = PROP_HARDWARE_DEFAULT;
1456   thiz->async_depth = PROP_ASYNC_DEPTH_DEFAULT;
1457   thiz->denoise_factor = PROP_DENOISE_DEFAULT;
1458   thiz->rotation = PROP_ROTATION_DEFAULT;
1459   thiz->deinterlace_mode = PROP_DEINTERLACE_MODE_DEFAULT;
1460   thiz->deinterlace_method = PROP_DEINTERLACE_METHOD_DEFAULT;
1461   thiz->buffer_duration = GST_CLOCK_TIME_NONE;
1462   thiz->hue = PROP_HUE_DEFAULT;
1463   thiz->saturation = PROP_SATURATION_DEFAULT;
1464   thiz->brightness = PROP_BRIGHTNESS_DEFAULT;
1465   thiz->contrast = PROP_CONTRAST_DEFAULT;
1466   thiz->detail = PROP_DETAIL_DEFAULT;
1467   thiz->mirroring = PROP_MIRRORING_DEFAULT;
1468   thiz->scaling_mode = PROP_SCALING_MODE_DEFAULT;
1469   thiz->keep_aspect = PROP_FORCE_ASPECT_RATIO_DEFAULT;
1470   thiz->frc_algm = PROP_FRC_ALGORITHM_DEFAULT;
1471   gst_video_info_init (&thiz->sinkpad_info);
1472   gst_video_info_init (&thiz->srcpad_info);
1473 }