2015-03-13 09:20:31 +00:00
/*
* Intel MediaSDK QSV based H .264 enccoder
*
* copyright ( c ) 2013 Yukinori Yamazoe
*
2015-03-28 00:33:27 +00:00
* This file is part of FFmpeg .
2015-03-13 09:20:31 +00:00
*
2015-03-28 00:33:27 +00:00
* FFmpeg is free software ; you can redistribute it and / or
2015-03-13 09:20:31 +00:00
* 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 .
*
2015-03-28 00:33:27 +00:00
* FFmpeg is distributed in the hope that it will be useful ,
2015-03-13 09:20:31 +00:00
* 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
2015-03-28 00:33:27 +00:00
* License along with FFmpeg ; if not , write to the Free Software
2015-03-13 09:20:31 +00:00
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
*/
# include <stdint.h>
# include <sys/types.h>
# include <mfx/mfxvideo.h>
2016-05-18 08:22:24 +00:00
# include "libavutil/common.h"
2015-03-13 09:20:31 +00:00
# include "libavutil/opt.h"
# include "avcodec.h"
# include "internal.h"
# include "qsv.h"
# include "qsv_internal.h"
# include "qsvenc.h"
typedef struct QSVH264EncContext {
AVClass * class ;
QSVEncContext qsv ;
} QSVH264EncContext ;
2015-10-27 17:08:45 +00:00
static int qsv_h264_set_encode_ctrl ( AVCodecContext * avctx ,
const AVFrame * frame , mfxEncodeCtrl * enc_ctrl )
{
2016-06-19 15:04:03 +00:00
QSVH264EncContext * qh264 = avctx - > priv_data ;
QSVEncContext * q = & qh264 - > qsv ;
2015-10-27 17:08:45 +00:00
if ( q - > a53_cc & & frame ) {
2016-06-04 18:11:52 +00:00
mfxPayload * payload ;
mfxU8 * sei_data ;
size_t sei_size ;
int res ;
res = ff_alloc_a53_sei ( frame , sizeof ( mfxPayload ) + 2 , ( void * * ) & payload , & sei_size ) ;
2016-10-05 13:34:25 +00:00
if ( res < 0 | | ! payload )
2016-06-04 18:11:52 +00:00
return res ;
sei_data = ( mfxU8 * ) ( payload + 1 ) ;
// SEI header
sei_data [ 0 ] = 4 ;
sei_data [ 1 ] = ( mfxU8 ) sei_size ; // size of SEI data
// SEI data filled in by ff_alloc_a53_sei
payload - > BufSize = sei_size + 2 ;
payload - > NumBit = payload - > BufSize * 8 ;
payload - > Type = 4 ;
payload - > Data = sei_data ;
enc_ctrl - > NumExtParam = 0 ;
enc_ctrl - > NumPayload = 1 ;
enc_ctrl - > Payload [ 0 ] = payload ;
2015-10-27 17:08:45 +00:00
}
return 0 ;
}
2015-03-13 09:20:31 +00:00
static av_cold int qsv_enc_init ( AVCodecContext * avctx )
{
QSVH264EncContext * q = avctx - > priv_data ;
2015-10-27 17:08:45 +00:00
q - > qsv . set_encode_ctrl_cb = qsv_h264_set_encode_ctrl ;
2015-03-13 09:20:31 +00:00
return ff_qsv_enc_init ( avctx , & q - > qsv ) ;
}
static int qsv_enc_frame ( AVCodecContext * avctx , AVPacket * pkt ,
const AVFrame * frame , int * got_packet )
{
QSVH264EncContext * q = avctx - > priv_data ;
return ff_qsv_encode ( avctx , & q - > qsv , pkt , frame , got_packet ) ;
}
static av_cold int qsv_enc_close ( AVCodecContext * avctx )
{
QSVH264EncContext * q = avctx - > priv_data ;
return ff_qsv_enc_close ( avctx , & q - > qsv ) ;
}
# define OFFSET(x) offsetof(QSVH264EncContext, x)
# define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
static const AVOption options [ ] = {
2015-11-07 19:50:41 +00:00
QSV_COMMON_OPTS
2017-11-27 16:19:57 +00:00
{ " cavlc " , " Enable CAVLC " , OFFSET ( qsv . cavlc ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , 1 , VE } ,
2015-03-13 09:20:31 +00:00
{ " idr_interval " , " Distance (in I-frames) between IDR frames " , OFFSET ( qsv . idr_interval ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , INT_MAX , VE } ,
2015-08-19 08:17:40 +00:00
{ " pic_timing_sei " , " Insert picture timing SEI with pic_struct_syntax element " , OFFSET ( qsv . pic_timing_sei ) , AV_OPT_TYPE_INT , { . i64 = 1 } , 0 , 1 , VE } ,
2015-11-08 06:58:49 +00:00
{ " single_sei_nal_unit " , " Put all the SEI messages into one NALU " , OFFSET ( qsv . single_sei_nal_unit ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , 1 , VE } ,
{ " max_dec_frame_buffering " , " Maximum number of frames buffered in the DPB " , OFFSET ( qsv . max_dec_frame_buffering ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , UINT16_MAX , VE } ,
2015-03-13 09:20:31 +00:00
2015-11-29 14:18:08 +00:00
# if QSV_HAVE_LA
2018-03-01 05:20:19 +00:00
{ " look_ahead " , " Use VBR algorithm with look ahead " , OFFSET ( qsv . look_ahead ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , 1 , VE } ,
2015-08-21 08:17:35 +00:00
{ " look_ahead_depth " , " Depth of look ahead in number frames " , OFFSET ( qsv . look_ahead_depth ) , AV_OPT_TYPE_INT , { . i64 = 0 } , 0 , 100 , VE } ,
# endif
2016-10-21 17:57:12 +00:00
# if QSV_HAVE_LA_DS
2017-12-14 22:04:37 +00:00
{ " look_ahead_downsampling " , " Downscaling factor for the frames saved for the lookahead analysis " , OFFSET ( qsv . look_ahead_downsampling ) ,
AV_OPT_TYPE_INT , { . i64 = MFX_LOOKAHEAD_DS_UNKNOWN } , MFX_LOOKAHEAD_DS_UNKNOWN , MFX_LOOKAHEAD_DS_2x , VE , " look_ahead_downsampling " } ,
2015-08-21 08:17:35 +00:00
{ " unknown " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_LOOKAHEAD_DS_UNKNOWN } , INT_MIN , INT_MAX , VE , " look_ahead_downsampling " } ,
2017-12-14 22:04:37 +00:00
{ " auto " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_LOOKAHEAD_DS_UNKNOWN } , INT_MIN , INT_MAX , VE , " look_ahead_downsampling " } ,
2015-08-21 08:17:35 +00:00
{ " off " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_LOOKAHEAD_DS_OFF } , INT_MIN , INT_MAX , VE , " look_ahead_downsampling " } ,
{ " 2x " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_LOOKAHEAD_DS_2x } , INT_MIN , INT_MAX , VE , " look_ahead_downsampling " } ,
2017-12-14 22:04:37 +00:00
{ " 4x " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_LOOKAHEAD_DS_4x } , INT_MIN , INT_MAX , VE , " look_ahead_downsampling " } ,
2015-08-21 08:17:35 +00:00
# endif
2015-11-08 06:58:49 +00:00
{ " int_ref_type " , " Intra refresh type " , OFFSET ( qsv . int_ref_type ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , UINT16_MAX , VE , " int_ref_type " } ,
{ " none " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = 0 } , . flags = VE , " int_ref_type " } ,
{ " vertical " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = 1 } , . flags = VE , " int_ref_type " } ,
{ " int_ref_cycle_size " , " Number of frames in the intra refresh cycle " , OFFSET ( qsv . int_ref_cycle_size ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , UINT16_MAX , VE } ,
{ " int_ref_qp_delta " , " QP difference for the refresh MBs " , OFFSET ( qsv . int_ref_qp_delta ) , AV_OPT_TYPE_INT , { . i64 = INT16_MIN } , INT16_MIN , INT16_MAX , VE } ,
{ " recovery_point_sei " , " Insert recovery point SEI messages " , OFFSET ( qsv . recovery_point_sei ) , AV_OPT_TYPE_INT , { . i64 = - 1 } , - 1 , 1 , VE } ,
{ " trellis " , " Trellis quantization " , OFFSET ( qsv . trellis ) , AV_OPT_TYPE_FLAGS , { . i64 = 0 } , 0 , UINT_MAX , VE , " trellis " } ,
{ " off " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_TRELLIS_OFF } , . flags = VE , " trellis " } ,
{ " I " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_TRELLIS_I } , . flags = VE , " trellis " } ,
{ " P " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_TRELLIS_P } , . flags = VE , " trellis " } ,
{ " B " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_TRELLIS_B } , . flags = VE , " trellis " } ,
2015-03-13 09:20:31 +00:00
{ " profile " , NULL , OFFSET ( qsv . profile ) , AV_OPT_TYPE_INT , { . i64 = MFX_PROFILE_UNKNOWN } , 0 , INT_MAX , VE , " profile " } ,
{ " unknown " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_PROFILE_UNKNOWN } , INT_MIN , INT_MAX , VE , " profile " } ,
{ " baseline " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_PROFILE_AVC_BASELINE } , INT_MIN , INT_MAX , VE , " profile " } ,
{ " main " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_PROFILE_AVC_MAIN } , INT_MIN , INT_MAX , VE , " profile " } ,
{ " high " , NULL , 0 , AV_OPT_TYPE_CONST , { . i64 = MFX_PROFILE_AVC_HIGH } , INT_MIN , INT_MAX , VE , " profile " } ,
2016-09-09 15:56:46 +00:00
{ " a53cc " , " Use A53 Closed Captions (if available) " , OFFSET ( qsv . a53_cc ) , AV_OPT_TYPE_INT , { . i64 = 1 } , 0 , 1 , VE } ,
2015-03-13 09:20:31 +00:00
{ NULL } ,
} ;
static const AVClass class = {
. class_name = " h264_qsv encoder " ,
. item_name = av_default_item_name ,
. option = options ,
. version = LIBAVUTIL_VERSION_INT ,
} ;
static const AVCodecDefault qsv_enc_defaults [ ] = {
{ " b " , " 1M " } ,
{ " refs " , " 0 " } ,
// same as the x264 default
{ " g " , " 250 " } ,
{ " bf " , " 3 " } ,
2017-11-27 16:19:57 +00:00
# if FF_API_CODER_TYPE
{ " coder " , " -1 " } ,
# endif
2015-03-13 09:20:31 +00:00
{ " flags " , " +cgop " } ,
2015-09-09 02:18:26 +00:00
# if FF_API_PRIVATE_OPT
2015-11-08 06:58:49 +00:00
{ " b_strategy " , " -1 " } ,
2015-09-09 02:18:26 +00:00
# endif
2015-03-13 09:20:31 +00:00
{ NULL } ,
} ;
AVCodec ff_h264_qsv_encoder = {
. name = " h264_qsv " ,
. long_name = NULL_IF_CONFIG_SMALL ( " H.264 / AVC / MPEG-4 AVC / MPEG-4 part 10 (Intel Quick Sync Video acceleration) " ) ,
. priv_data_size = sizeof ( QSVH264EncContext ) ,
. type = AVMEDIA_TYPE_VIDEO ,
. id = AV_CODEC_ID_H264 ,
. init = qsv_enc_init ,
. encode2 = qsv_enc_frame ,
. close = qsv_enc_close ,
2017-12-11 15:18:44 +00:00
. capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HYBRID ,
2015-03-13 09:20:31 +00:00
. pix_fmts = ( const enum AVPixelFormat [ ] ) { AV_PIX_FMT_NV12 ,
2016-06-22 17:57:28 +00:00
AV_PIX_FMT_P010 ,
2015-03-13 09:20:31 +00:00
AV_PIX_FMT_QSV ,
AV_PIX_FMT_NONE } ,
. priv_class = & class ,
. defaults = qsv_enc_defaults ,
2015-08-05 11:42:08 +00:00
. caps_internal = FF_CODEC_CAP_INIT_CLEANUP ,
2017-12-11 15:18:44 +00:00
. wrapper_name = " qsv " ,
2015-03-13 09:20:31 +00:00
} ;