2015-06-16 16:22:11 +00:00
/*
* Intel MediaSDK QSV based MPEG2 decoder
*
* copyright ( c ) 2015 Anton Khirnov
*
* This file is part of Libav .
*
* Libav is free software ; you can redistribute it and / or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
* version 2.1 of the License , or ( at your option ) any later version .
*
* Libav is distributed in the hope that it will be useful ,
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
*
* You should have received a copy of the GNU Lesser General Public
* License along with Libav ; if not , write to the Free Software
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
*/
# include <stdint.h>
# include <string.h>
# include <mfx/mfxvideo.h>
# include "libavutil/common.h"
# include "libavutil/fifo.h"
# include "libavutil/opt.h"
# include "avcodec.h"
# include "internal.h"
# include "qsv_internal.h"
# include "qsvdec.h"
# include "qsv.h"
typedef struct QSVMPEG2Context {
AVClass * class ;
QSVContext qsv ;
AVFifoBuffer * packet_fifo ;
AVPacket input_ref ;
} QSVMPEG2Context ;
static void qsv_clear_buffers ( QSVMPEG2Context * s )
{
AVPacket pkt ;
while ( av_fifo_size ( s - > packet_fifo ) > = sizeof ( pkt ) ) {
av_fifo_generic_read ( s - > packet_fifo , & pkt , sizeof ( pkt ) , NULL ) ;
av_packet_unref ( & pkt ) ;
}
av_packet_unref ( & s - > input_ref ) ;
}
static av_cold int qsv_decode_close ( AVCodecContext * avctx )
{
QSVMPEG2Context * s = avctx - > priv_data ;
ff_qsv_decode_close ( & s - > qsv ) ;
qsv_clear_buffers ( s ) ;
av_fifo_free ( s - > packet_fifo ) ;
return 0 ;
}
static av_cold int qsv_decode_init ( AVCodecContext * avctx )
{
QSVMPEG2Context * s = avctx - > priv_data ;
int ret ;
s - > packet_fifo = av_fifo_alloc ( sizeof ( AVPacket ) ) ;
if ( ! s - > packet_fifo ) {
ret = AVERROR ( ENOMEM ) ;
goto fail ;
}
s - > qsv . iopattern = MFX_IOPATTERN_OUT_SYSTEM_MEMORY ;
return 0 ;
fail :
qsv_decode_close ( avctx ) ;
return ret ;
}
static int qsv_decode_frame ( AVCodecContext * avctx , void * data ,
int * got_frame , AVPacket * avpkt )
{
QSVMPEG2Context * s = avctx - > priv_data ;
AVFrame * frame = data ;
int ret ;
/* buffer the input packet */
if ( avpkt - > size ) {
AVPacket input_ref = { 0 } ;
if ( av_fifo_space ( s - > packet_fifo ) < sizeof ( input_ref ) ) {
ret = av_fifo_realloc2 ( s - > packet_fifo ,
av_fifo_size ( s - > packet_fifo ) + sizeof ( input_ref ) ) ;
if ( ret < 0 )
return ret ;
}
ret = av_packet_ref ( & input_ref , avpkt ) ;
if ( ret < 0 )
return ret ;
av_fifo_generic_write ( s - > packet_fifo , & input_ref , sizeof ( input_ref ) , NULL ) ;
}
/* process buffered data */
while ( ! * got_frame ) {
if ( s - > input_ref . size < = 0 ) {
/* no more data */
if ( av_fifo_size ( s - > packet_fifo ) < sizeof ( AVPacket ) )
return avpkt - > size ? avpkt - > size : ff_qsv_process_data ( avctx , & s - > qsv , frame , got_frame , avpkt ) ;
av_packet_unref ( & s - > input_ref ) ;
av_fifo_generic_read ( s - > packet_fifo , & s - > input_ref , sizeof ( s - > input_ref ) , NULL ) ;
}
ret = ff_qsv_process_data ( avctx , & s - > qsv , frame , got_frame , & s - > input_ref ) ;
if ( ret < 0 )
return ret ;
s - > input_ref . size - = ret ;
s - > input_ref . data + = ret ;
}
return avpkt - > size ;
}
static void qsv_decode_flush ( AVCodecContext * avctx )
{
QSVMPEG2Context * s = avctx - > priv_data ;
qsv_clear_buffers ( s ) ;
ff_qsv_decode_flush ( avctx , & s - > qsv ) ;
}
AVHWAccel ff_mpeg2_qsv_hwaccel = {
. name = " mpeg2_qsv " ,
. type = AVMEDIA_TYPE_VIDEO ,
. id = AV_CODEC_ID_MPEG2VIDEO ,
. pix_fmt = AV_PIX_FMT_QSV ,
} ;
# define OFFSET(x) offsetof(QSVMPEG2Context, x)
# define VD AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
static const AVOption options [ ] = {
{ " async_depth " , " Internal parallelization depth, the higher the value the higher the latency. " , OFFSET ( qsv . async_depth ) , AV_OPT_TYPE_INT , { . i64 = ASYNC_DEPTH_DEFAULT } , 0 , INT_MAX , VD } ,
{ NULL } ,
} ;
static const AVClass class = {
. class_name = " mpeg2_qsv " ,
. item_name = av_default_item_name ,
. option = options ,
. version = LIBAVUTIL_VERSION_INT ,
} ;
AVCodec ff_mpeg2_qsv_decoder = {
. name = " mpeg2_qsv " ,
. long_name = NULL_IF_CONFIG_SMALL ( " MPEG-2 video (Intel Quick Sync Video acceleration) " ) ,
. priv_data_size = sizeof ( QSVMPEG2Context ) ,
. type = AVMEDIA_TYPE_VIDEO ,
. id = AV_CODEC_ID_MPEG2VIDEO ,
. init = qsv_decode_init ,
. decode = qsv_decode_frame ,
. flush = qsv_decode_flush ,
. close = qsv_decode_close ,
2015-10-14 13:08:49 +00:00
. capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_DR1 ,
2015-06-16 16:22:11 +00:00
. priv_class = & class ,
2015-07-21 07:45:05 +00:00
. pix_fmts = ( const enum AVPixelFormat [ ] ) { AV_PIX_FMT_NV12 ,
AV_PIX_FMT_QSV ,
AV_PIX_FMT_NONE } ,
2015-06-16 16:22:11 +00:00
} ;