sys/msdk/msdk.c - mtk-gst-plugins-bad - Git at Google

 /* GStreamer Intel MSDK plugin
  * Copyright (c) 2016, Oblong Industries, Inc.
  * All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are met:
  *
  * 1. Redistributions of source code must retain the above copyright notice,
  *    this list of conditions and the following disclaimer.
  *
  * 2. Redistributions in binary form must reproduce the above copyright notice,
  *    this list of conditions and the following disclaimer in the documentation
  *    and/or other materials provided with the distribution.
  *
  * 3. Neither the name of the copyright holder nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
  * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
  * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */

 #include "msdk.h"
 #include "gstmsdkvideomemory.h"
 #include "gstmsdksystemmemory.h"

 GST_DEBUG_CATEGORY_EXTERN (gst_msdk_debug);
 #define GST_CAT_DEFAULT gst_msdk_debug

 #define INVALID_INDEX         ((guint) -1)
 #define GST_MSDK_ALIGNMENT_PADDING(num) (32 - ((num) & 31))

 struct map
 {
   GstVideoFormat format;
   mfxU16 mfx_chroma_format;
   mfxU32 mfx_fourcc;
 };

 #define GST_VIDEO_INFO_TO_MFX_MAP(FORMAT, CHROMA, FOURCC) \
     { GST_VIDEO_FORMAT_##FORMAT, MFX_CHROMAFORMAT_##CHROMA, MFX_FOURCC_##FOURCC }

 static const struct map gst_msdk_video_format_to_mfx_map[] = {
   GST_VIDEO_INFO_TO_MFX_MAP (NV12, YUV420, NV12),
   GST_VIDEO_INFO_TO_MFX_MAP (YV12, YUV420, YV12),
   GST_VIDEO_INFO_TO_MFX_MAP (I420, YUV420, YV12),
   GST_VIDEO_INFO_TO_MFX_MAP (YUY2, YUV422, YUY2),
   GST_VIDEO_INFO_TO_MFX_MAP (UYVY, YUV422, UYVY),
   GST_VIDEO_INFO_TO_MFX_MAP (BGRA, YUV444, RGB4),
   {0, 0, 0}
 };

 const gchar *
 msdk_status_to_string (mfxStatus status)
 {
   switch (status) {
       /* no error */
     case MFX_ERR_NONE:
       return "no error";
       /* reserved for unexpected errors */
     case MFX_ERR_UNKNOWN:
       return "unknown error";
       /* error codes <0 */
     case MFX_ERR_NULL_PTR:
       return "null pointer";
     case MFX_ERR_UNSUPPORTED:
       return "undeveloped feature";
     case MFX_ERR_MEMORY_ALLOC:
       return "failed to allocate memory";
     case MFX_ERR_NOT_ENOUGH_BUFFER:
       return "insufficient buffer at input/output";
     case MFX_ERR_INVALID_HANDLE:
       return "invalid handle";
     case MFX_ERR_LOCK_MEMORY:
       return "failed to lock the memory block";
     case MFX_ERR_NOT_INITIALIZED:
       return "member function called before initialization";
     case MFX_ERR_NOT_FOUND:
       return "the specified object is not found";
     case MFX_ERR_MORE_DATA:
       return "expect more data at input";
     case MFX_ERR_MORE_SURFACE:
       return "expect more surface at output";
     case MFX_ERR_ABORTED:
       return "operation aborted";
     case MFX_ERR_DEVICE_LOST:
       return "lose the HW acceleration device";
     case MFX_ERR_INCOMPATIBLE_VIDEO_PARAM:
       return "incompatible video parameters";
     case MFX_ERR_INVALID_VIDEO_PARAM:
       return "invalid video parameters";
     case MFX_ERR_UNDEFINED_BEHAVIOR:
       return "undefined behavior";
     case MFX_ERR_DEVICE_FAILED:
       return "device operation failure";
     case MFX_ERR_MORE_BITSTREAM:
       return "expect more bitstream buffers at output";
     case MFX_ERR_INCOMPATIBLE_AUDIO_PARAM:
       return "incompatible audio parameters";
     case MFX_ERR_INVALID_AUDIO_PARAM:
       return "invalid audio parameters";
       /* warnings >0 */
     case MFX_WRN_IN_EXECUTION:
       return "the previous asynchronous operation is in execution";
     case MFX_WRN_DEVICE_BUSY:
       return "the HW acceleration device is busy";
     case MFX_WRN_VIDEO_PARAM_CHANGED:
       return "the video parameters are changed during decoding";
     case MFX_WRN_PARTIAL_ACCELERATION:
       return "SW is used";
     case MFX_WRN_INCOMPATIBLE_VIDEO_PARAM:
       return "incompatible video parameters";
     case MFX_WRN_VALUE_NOT_CHANGED:
       return "the value is saturated based on its valid range";
     case MFX_WRN_OUT_OF_RANGE:
       return "the value is out of valid range";
     case MFX_WRN_FILTER_SKIPPED:
       return "one of requested filters has been skipped";
     case MFX_WRN_INCOMPATIBLE_AUDIO_PARAM:
       return "incompatible audio parameters";
     default:
       break;
   }
   return "undefiend error";
 }

 void
 msdk_close_session (mfxSession session)
 {
   mfxStatus status;

   if (!session)
     return;

   status = MFXClose (session);
   if (status != MFX_ERR_NONE)
     GST_ERROR ("Close failed (%s)", msdk_status_to_string (status));
 }

 mfxSession
 msdk_open_session (mfxIMPL impl)
 {
   mfxSession session = NULL;
   mfxVersion version = { {1, 1}
   };
   mfxIMPL implementation;
   mfxStatus status;

   static const gchar *implementation_names[] = {
     "AUTO", "SOFTWARE", "HARDWARE", "AUTO_ANY", "HARDWARE_ANY", "HARDWARE2",
     "HARDWARE3", "HARDWARE4", "RUNTIME"
   };

   status = MFXInit (impl, &version, &session);
   if (status != MFX_ERR_NONE) {
     GST_ERROR ("Intel Media SDK not available (%s)",
         msdk_status_to_string (status));
     goto failed;
   }

   MFXQueryIMPL (session, &implementation);
   if (status != MFX_ERR_NONE) {
     GST_ERROR ("Query implementation failed (%s)",
         msdk_status_to_string (status));
     goto failed;
   }

   MFXQueryVersion (session, &version);
   if (status != MFX_ERR_NONE) {
     GST_ERROR ("Query version failed (%s)", msdk_status_to_string (status));
     goto failed;
   }

   GST_INFO ("MSDK implementation: 0x%04x (%s)", implementation,
       implementation_names[MFX_IMPL_BASETYPE (implementation)]);
   GST_INFO ("MSDK version: %d.%d", version.Major, version.Minor);

   return session;

 failed:
   msdk_close_session (session);
   return NULL;
 }

 gboolean
 msdk_is_available (void)
 {
   mfxSession session = msdk_open_session (MFX_IMPL_AUTO_ANY);
   if (!session) {
     return FALSE;
   }

   msdk_close_session (session);
   return TRUE;
 }

 void
 gst_msdk_set_video_alignment (GstVideoInfo * info,
     GstVideoAlignment * alignment)
 {
   guint i, width, height;

   width = GST_VIDEO_INFO_WIDTH (info);
   height = GST_VIDEO_INFO_HEIGHT (info);

   gst_video_alignment_reset (alignment);
   for (i = 0; i < GST_VIDEO_INFO_N_PLANES (info); i++)
     alignment->stride_align[i] = 31;    /* 32-byte alignment */

   if (width & 31)
     alignment->padding_right = GST_MSDK_ALIGNMENT_PADDING (width);
   if (height & 31)
     alignment->padding_bottom = GST_MSDK_ALIGNMENT_PADDING (height);
 }

 static const struct map *
 _map_lookup_format (GstVideoFormat format)
 {
   const struct map *m = gst_msdk_video_format_to_mfx_map;

   for (; m->format != 0; m++) {
     if (m->format == format)
       return m;
   }
   return NULL;
 }

 gint
 gst_msdk_get_mfx_chroma_from_format (GstVideoFormat format)
 {
   const struct map *const m = _map_lookup_format (format);

   return m ? m->mfx_chroma_format : -1;
 }

 gint
 gst_msdk_get_mfx_fourcc_from_format (GstVideoFormat format)
 {
   const struct map *const m = _map_lookup_format (format);

   return m ? m->mfx_fourcc : -1;
 }

 void
 gst_msdk_set_mfx_frame_info_from_video_info (mfxFrameInfo * mfx_info,
     GstVideoInfo * info)
 {
   g_return_if_fail (info && mfx_info);

   mfx_info->Width = GST_ROUND_UP_32 (GST_VIDEO_INFO_WIDTH (info));
   mfx_info->Height = GST_ROUND_UP_32 (GST_VIDEO_INFO_HEIGHT (info));
   mfx_info->CropW = GST_VIDEO_INFO_WIDTH (info);
   mfx_info->CropH = GST_VIDEO_INFO_HEIGHT (info);
   mfx_info->FrameRateExtN = GST_VIDEO_INFO_FPS_N (info);
   mfx_info->FrameRateExtD = GST_VIDEO_INFO_FPS_D (info);
   mfx_info->AspectRatioW = GST_VIDEO_INFO_PAR_N (info);
   mfx_info->AspectRatioH = GST_VIDEO_INFO_PAR_D (info);
   mfx_info->PicStruct = MFX_PICSTRUCT_PROGRESSIVE;      /* this is by default */
   mfx_info->FourCC =
       gst_msdk_get_mfx_fourcc_from_format (GST_VIDEO_INFO_FORMAT (info));
   mfx_info->ChromaFormat =
       gst_msdk_get_mfx_chroma_from_format (GST_VIDEO_INFO_FORMAT (info));

   return;
 }

 gboolean
 gst_msdk_is_msdk_buffer (GstBuffer * buf)
 {
   GstAllocator *allocator;
   GstMemory *mem = gst_buffer_peek_memory (buf, 0);

   allocator = GST_MEMORY_CAST (mem)->allocator;

   if (allocator && (GST_IS_MSDK_VIDEO_ALLOCATOR (allocator) ||
           GST_IS_MSDK_SYSTEM_ALLOCATOR (allocator)))
     return TRUE;
   else
     return FALSE;
 }

 mfxFrameSurface1 *
 gst_msdk_get_surface_from_buffer (GstBuffer * buf)
 {
   GstAllocator *allocator;
   GstMemory *mem = gst_buffer_peek_memory (buf, 0);

   allocator = GST_MEMORY_CAST (mem)->allocator;

   if (GST_IS_MSDK_VIDEO_ALLOCATOR (allocator))
     return GST_MSDK_VIDEO_MEMORY_CAST (mem)->surface;
   else
     return GST_MSDK_SYSTEM_MEMORY_CAST (mem)->surface;
 }
	/* GStreamer Intel MSDK plugin
	* Copyright (c) 2016, Oblong Industries, Inc.
	* All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions are met:
	*
	* 1. Redistributions of source code must retain the above copyright notice,
	* this list of conditions and the following disclaimer.
	*
	* 2. Redistributions in binary form must reproduce the above copyright notice,
	* this list of conditions and the following disclaimer in the documentation
	* and/or other materials provided with the distribution.
	*
	* 3. Neither the name of the copyright holder nor the names of its contributors
	* may be used to endorse or promote products derived from this software
	* without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
	* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
	* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
	* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
	* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
	* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
	* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
	* WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
	* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
	* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
	*/

	#include "msdk.h"
	#include "gstmsdkvideomemory.h"
	#include "gstmsdksystemmemory.h"

	GST_DEBUG_CATEGORY_EXTERN (gst_msdk_debug);
	#define GST_CAT_DEFAULT gst_msdk_debug

	#define INVALID_INDEX ((guint) -1)
	#define GST_MSDK_ALIGNMENT_PADDING(num) (32 - ((num) & 31))

	struct map
	{
	GstVideoFormat format;
	mfxU16 mfx_chroma_format;
	mfxU32 mfx_fourcc;
	};

	#define GST_VIDEO_INFO_TO_MFX_MAP(FORMAT, CHROMA, FOURCC) \
	{ GST_VIDEO_FORMAT_##FORMAT, MFX_CHROMAFORMAT_##CHROMA, MFX_FOURCC_##FOURCC }

	static const struct map gst_msdk_video_format_to_mfx_map[] = {
	GST_VIDEO_INFO_TO_MFX_MAP (NV12, YUV420, NV12),
	GST_VIDEO_INFO_TO_MFX_MAP (YV12, YUV420, YV12),
	GST_VIDEO_INFO_TO_MFX_MAP (I420, YUV420, YV12),
	GST_VIDEO_INFO_TO_MFX_MAP (YUY2, YUV422, YUY2),
	GST_VIDEO_INFO_TO_MFX_MAP (UYVY, YUV422, UYVY),
	GST_VIDEO_INFO_TO_MFX_MAP (BGRA, YUV444, RGB4),
	{0, 0, 0}
	};

	const gchar *
	msdk_status_to_string (mfxStatus status)
	{
	switch (status) {
	/* no error */
	case MFX_ERR_NONE:
	return "no error";
	/* reserved for unexpected errors */
	case MFX_ERR_UNKNOWN:
	return "unknown error";
	/* error codes <0 */
	case MFX_ERR_NULL_PTR:
	return "null pointer";
	case MFX_ERR_UNSUPPORTED:
	return "undeveloped feature";
	case MFX_ERR_MEMORY_ALLOC:
	return "failed to allocate memory";
	case MFX_ERR_NOT_ENOUGH_BUFFER:
	return "insufficient buffer at input/output";
	case MFX_ERR_INVALID_HANDLE:
	return "invalid handle";
	case MFX_ERR_LOCK_MEMORY:
	return "failed to lock the memory block";
	case MFX_ERR_NOT_INITIALIZED:
	return "member function called before initialization";
	case MFX_ERR_NOT_FOUND:
	return "the specified object is not found";
	case MFX_ERR_MORE_DATA:
	return "expect more data at input";
	case MFX_ERR_MORE_SURFACE:
	return "expect more surface at output";
	case MFX_ERR_ABORTED:
	return "operation aborted";
	case MFX_ERR_DEVICE_LOST:
	return "lose the HW acceleration device";
	case MFX_ERR_INCOMPATIBLE_VIDEO_PARAM:
	return "incompatible video parameters";
	case MFX_ERR_INVALID_VIDEO_PARAM:
	return "invalid video parameters";
	case MFX_ERR_UNDEFINED_BEHAVIOR:
	return "undefined behavior";
	case MFX_ERR_DEVICE_FAILED:
	return "device operation failure";
	case MFX_ERR_MORE_BITSTREAM:
	return "expect more bitstream buffers at output";
	case MFX_ERR_INCOMPATIBLE_AUDIO_PARAM:
	return "incompatible audio parameters";
	case MFX_ERR_INVALID_AUDIO_PARAM:
	return "invalid audio parameters";
	/* warnings >0 */
	case MFX_WRN_IN_EXECUTION:
	return "the previous asynchronous operation is in execution";
	case MFX_WRN_DEVICE_BUSY:
	return "the HW acceleration device is busy";
	case MFX_WRN_VIDEO_PARAM_CHANGED:
	return "the video parameters are changed during decoding";
	case MFX_WRN_PARTIAL_ACCELERATION:
	return "SW is used";
	case MFX_WRN_INCOMPATIBLE_VIDEO_PARAM:
	return "incompatible video parameters";
	case MFX_WRN_VALUE_NOT_CHANGED:
	return "the value is saturated based on its valid range";
	case MFX_WRN_OUT_OF_RANGE:
	return "the value is out of valid range";
	case MFX_WRN_FILTER_SKIPPED:
	return "one of requested filters has been skipped";
	case MFX_WRN_INCOMPATIBLE_AUDIO_PARAM:
	return "incompatible audio parameters";
	default:
	break;
	}
	return "undefiend error";
	}

	void
	msdk_close_session (mfxSession session)
	{
	mfxStatus status;

	if (!session)
	return;

	status = MFXClose (session);
	if (status != MFX_ERR_NONE)
	GST_ERROR ("Close failed (%s)", msdk_status_to_string (status));
	}

	mfxSession
	msdk_open_session (mfxIMPL impl)
	{
	mfxSession session = NULL;
	mfxVersion version = { {1, 1}
	};
	mfxIMPL implementation;
	mfxStatus status;

	static const gchar *implementation_names[] = {
	"AUTO", "SOFTWARE", "HARDWARE", "AUTO_ANY", "HARDWARE_ANY", "HARDWARE2",
	"HARDWARE3", "HARDWARE4", "RUNTIME"
	};

	status = MFXInit (impl, &version, &session);
	if (status != MFX_ERR_NONE) {
	GST_ERROR ("Intel Media SDK not available (%s)",
	msdk_status_to_string (status));
	goto failed;
	}

	MFXQueryIMPL (session, &implementation);
	if (status != MFX_ERR_NONE) {
	GST_ERROR ("Query implementation failed (%s)",
	msdk_status_to_string (status));
	goto failed;
	}

	MFXQueryVersion (session, &version);
	if (status != MFX_ERR_NONE) {
	GST_ERROR ("Query version failed (%s)", msdk_status_to_string (status));
	goto failed;
	}

	GST_INFO ("MSDK implementation: 0x%04x (%s)", implementation,
	implementation_names[MFX_IMPL_BASETYPE (implementation)]);
	GST_INFO ("MSDK version: %d.%d", version.Major, version.Minor);

	return session;

	failed:
	msdk_close_session (session);
	return NULL;
	}

	gboolean
	msdk_is_available (void)
	{
	mfxSession session = msdk_open_session (MFX_IMPL_AUTO_ANY);
	if (!session) {
	return FALSE;
	}

	msdk_close_session (session);
	return TRUE;
	}

	void
	gst_msdk_set_video_alignment (GstVideoInfo * info,
	GstVideoAlignment * alignment)
	{
	guint i, width, height;

	width = GST_VIDEO_INFO_WIDTH (info);
	height = GST_VIDEO_INFO_HEIGHT (info);

	gst_video_alignment_reset (alignment);
	for (i = 0; i < GST_VIDEO_INFO_N_PLANES (info); i++)
	alignment->stride_align[i] = 31; /* 32-byte alignment */

	if (width & 31)
	alignment->padding_right = GST_MSDK_ALIGNMENT_PADDING (width);
	if (height & 31)
	alignment->padding_bottom = GST_MSDK_ALIGNMENT_PADDING (height);
	}

	static const struct map *
	_map_lookup_format (GstVideoFormat format)
	{
	const struct map *m = gst_msdk_video_format_to_mfx_map;

	for (; m->format != 0; m++) {
	if (m->format == format)
	return m;
	}
	return NULL;
	}

	gint
	gst_msdk_get_mfx_chroma_from_format (GstVideoFormat format)
	{
	const struct map *const m = _map_lookup_format (format);

	return m ? m->mfx_chroma_format : -1;
	}

	gint
	gst_msdk_get_mfx_fourcc_from_format (GstVideoFormat format)
	{
	const struct map *const m = _map_lookup_format (format);

	return m ? m->mfx_fourcc : -1;
	}

	void
	gst_msdk_set_mfx_frame_info_from_video_info (mfxFrameInfo * mfx_info,
	GstVideoInfo * info)
	{
	g_return_if_fail (info && mfx_info);

	mfx_info->Width = GST_ROUND_UP_32 (GST_VIDEO_INFO_WIDTH (info));
	mfx_info->Height = GST_ROUND_UP_32 (GST_VIDEO_INFO_HEIGHT (info));
	mfx_info->CropW = GST_VIDEO_INFO_WIDTH (info);
	mfx_info->CropH = GST_VIDEO_INFO_HEIGHT (info);
	mfx_info->FrameRateExtN = GST_VIDEO_INFO_FPS_N (info);
	mfx_info->FrameRateExtD = GST_VIDEO_INFO_FPS_D (info);
	mfx_info->AspectRatioW = GST_VIDEO_INFO_PAR_N (info);
	mfx_info->AspectRatioH = GST_VIDEO_INFO_PAR_D (info);
	mfx_info->PicStruct = MFX_PICSTRUCT_PROGRESSIVE; /* this is by default */
	mfx_info->FourCC =
	gst_msdk_get_mfx_fourcc_from_format (GST_VIDEO_INFO_FORMAT (info));
	mfx_info->ChromaFormat =
	gst_msdk_get_mfx_chroma_from_format (GST_VIDEO_INFO_FORMAT (info));

	return;
	}

	gboolean
	gst_msdk_is_msdk_buffer (GstBuffer * buf)
	{
	GstAllocator *allocator;
	GstMemory *mem = gst_buffer_peek_memory (buf, 0);

	allocator = GST_MEMORY_CAST (mem)->allocator;

	if (allocator && (GST_IS_MSDK_VIDEO_ALLOCATOR (allocator) \|\|
	GST_IS_MSDK_SYSTEM_ALLOCATOR (allocator)))
	return TRUE;
	else
	return FALSE;
	}

	mfxFrameSurface1 *
	gst_msdk_get_surface_from_buffer (GstBuffer * buf)
	{
	GstAllocator *allocator;
	GstMemory *mem = gst_buffer_peek_memory (buf, 0);

	allocator = GST_MEMORY_CAST (mem)->allocator;

	if (GST_IS_MSDK_VIDEO_ALLOCATOR (allocator))
	return GST_MSDK_VIDEO_MEMORY_CAST (mem)->surface;
	else
	return GST_MSDK_SYSTEM_MEMORY_CAST (mem)->surface;
	}