mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
706 lines
24 KiB
706 lines
24 KiB
/* |
|
* libx265 encoder |
|
* |
|
* Copyright (c) 2013-2014 Derek Buitenhuis |
|
* |
|
* This file is part of FFmpeg. |
|
* |
|
* FFmpeg 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. |
|
* |
|
* FFmpeg 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 FFmpeg; if not, write to the Free Software |
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
*/ |
|
|
|
#if defined(_MSC_VER) |
|
#define X265_API_IMPORTS 1 |
|
#endif |
|
|
|
#include <x265.h> |
|
#include <float.h> |
|
|
|
#include "libavutil/internal.h" |
|
#include "libavutil/common.h" |
|
#include "libavutil/opt.h" |
|
#include "libavutil/pixdesc.h" |
|
#include "avcodec.h" |
|
#include "internal.h" |
|
#include "packet_internal.h" |
|
|
|
typedef struct libx265Context { |
|
const AVClass *class; |
|
|
|
x265_encoder *encoder; |
|
x265_param *params; |
|
const x265_api *api; |
|
|
|
float crf; |
|
int cqp; |
|
int forced_idr; |
|
char *preset; |
|
char *tune; |
|
char *profile; |
|
AVDictionary *x265_opts; |
|
|
|
/** |
|
* If the encoder does not support ROI then warn the first time we |
|
* encounter a frame with ROI side data. |
|
*/ |
|
int roi_warned; |
|
} libx265Context; |
|
|
|
static int is_keyframe(NalUnitType naltype) |
|
{ |
|
switch (naltype) { |
|
case NAL_UNIT_CODED_SLICE_BLA_W_LP: |
|
case NAL_UNIT_CODED_SLICE_BLA_W_RADL: |
|
case NAL_UNIT_CODED_SLICE_BLA_N_LP: |
|
case NAL_UNIT_CODED_SLICE_IDR_W_RADL: |
|
case NAL_UNIT_CODED_SLICE_IDR_N_LP: |
|
case NAL_UNIT_CODED_SLICE_CRA: |
|
return 1; |
|
default: |
|
return 0; |
|
} |
|
} |
|
|
|
static av_cold int libx265_encode_close(AVCodecContext *avctx) |
|
{ |
|
libx265Context *ctx = avctx->priv_data; |
|
|
|
ctx->api->param_free(ctx->params); |
|
|
|
if (ctx->encoder) |
|
ctx->api->encoder_close(ctx->encoder); |
|
|
|
return 0; |
|
} |
|
|
|
static av_cold int libx265_param_parse_float(AVCodecContext *avctx, |
|
const char *key, float value) |
|
{ |
|
libx265Context *ctx = avctx->priv_data; |
|
char buf[256]; |
|
|
|
snprintf(buf, sizeof(buf), "%2.2f", value); |
|
if (ctx->api->param_parse(ctx->params, key, buf) == X265_PARAM_BAD_VALUE) { |
|
av_log(avctx, AV_LOG_ERROR, "Invalid value %2.2f for param \"%s\".\n", value, key); |
|
return AVERROR(EINVAL); |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static av_cold int libx265_param_parse_int(AVCodecContext *avctx, |
|
const char *key, int value) |
|
{ |
|
libx265Context *ctx = avctx->priv_data; |
|
char buf[256]; |
|
|
|
snprintf(buf, sizeof(buf), "%d", value); |
|
if (ctx->api->param_parse(ctx->params, key, buf) == X265_PARAM_BAD_VALUE) { |
|
av_log(avctx, AV_LOG_ERROR, "Invalid value %d for param \"%s\".\n", value, key); |
|
return AVERROR(EINVAL); |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static av_cold int libx265_encode_init(AVCodecContext *avctx) |
|
{ |
|
libx265Context *ctx = avctx->priv_data; |
|
AVCPBProperties *cpb_props = NULL; |
|
int ret; |
|
|
|
ctx->api = x265_api_get(av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].depth); |
|
if (!ctx->api) |
|
ctx->api = x265_api_get(0); |
|
|
|
ctx->params = ctx->api->param_alloc(); |
|
if (!ctx->params) { |
|
av_log(avctx, AV_LOG_ERROR, "Could not allocate x265 param structure.\n"); |
|
return AVERROR(ENOMEM); |
|
} |
|
|
|
if (ctx->api->param_default_preset(ctx->params, ctx->preset, ctx->tune) < 0) { |
|
int i; |
|
|
|
av_log(avctx, AV_LOG_ERROR, "Error setting preset/tune %s/%s.\n", ctx->preset, ctx->tune); |
|
av_log(avctx, AV_LOG_INFO, "Possible presets:"); |
|
for (i = 0; x265_preset_names[i]; i++) |
|
av_log(avctx, AV_LOG_INFO, " %s", x265_preset_names[i]); |
|
|
|
av_log(avctx, AV_LOG_INFO, "\n"); |
|
av_log(avctx, AV_LOG_INFO, "Possible tunes:"); |
|
for (i = 0; x265_tune_names[i]; i++) |
|
av_log(avctx, AV_LOG_INFO, " %s", x265_tune_names[i]); |
|
|
|
av_log(avctx, AV_LOG_INFO, "\n"); |
|
|
|
return AVERROR(EINVAL); |
|
} |
|
|
|
ctx->params->frameNumThreads = avctx->thread_count; |
|
if (avctx->framerate.num > 0 && avctx->framerate.den > 0) { |
|
ctx->params->fpsNum = avctx->framerate.num; |
|
ctx->params->fpsDenom = avctx->framerate.den; |
|
} else { |
|
ctx->params->fpsNum = avctx->time_base.den; |
|
ctx->params->fpsDenom = avctx->time_base.num * avctx->ticks_per_frame; |
|
} |
|
ctx->params->sourceWidth = avctx->width; |
|
ctx->params->sourceHeight = avctx->height; |
|
ctx->params->bEnablePsnr = !!(avctx->flags & AV_CODEC_FLAG_PSNR); |
|
ctx->params->bOpenGOP = !(avctx->flags & AV_CODEC_FLAG_CLOSED_GOP); |
|
|
|
/* Tune the CTU size based on input resolution. */ |
|
if (ctx->params->sourceWidth < 64 || ctx->params->sourceHeight < 64) |
|
ctx->params->maxCUSize = 32; |
|
if (ctx->params->sourceWidth < 32 || ctx->params->sourceHeight < 32) |
|
ctx->params->maxCUSize = 16; |
|
if (ctx->params->sourceWidth < 16 || ctx->params->sourceHeight < 16) { |
|
av_log(avctx, AV_LOG_ERROR, "Image size is too small (%dx%d).\n", |
|
ctx->params->sourceWidth, ctx->params->sourceHeight); |
|
return AVERROR(EINVAL); |
|
} |
|
|
|
|
|
ctx->params->vui.bEnableVideoSignalTypePresentFlag = 1; |
|
|
|
ctx->params->vui.bEnableVideoFullRangeFlag = avctx->pix_fmt == AV_PIX_FMT_YUVJ420P || |
|
avctx->pix_fmt == AV_PIX_FMT_YUVJ422P || |
|
avctx->pix_fmt == AV_PIX_FMT_YUVJ444P || |
|
avctx->color_range == AVCOL_RANGE_JPEG; |
|
|
|
if ((avctx->color_primaries <= AVCOL_PRI_SMPTE432 && |
|
avctx->color_primaries != AVCOL_PRI_UNSPECIFIED) || |
|
(avctx->color_trc <= AVCOL_TRC_ARIB_STD_B67 && |
|
avctx->color_trc != AVCOL_TRC_UNSPECIFIED) || |
|
(avctx->colorspace <= AVCOL_SPC_ICTCP && |
|
avctx->colorspace != AVCOL_SPC_UNSPECIFIED)) { |
|
|
|
ctx->params->vui.bEnableColorDescriptionPresentFlag = 1; |
|
|
|
// x265 validates the parameters internally |
|
ctx->params->vui.colorPrimaries = avctx->color_primaries; |
|
ctx->params->vui.transferCharacteristics = avctx->color_trc; |
|
#if X265_BUILD >= 159 |
|
if (avctx->color_trc == AVCOL_TRC_ARIB_STD_B67) |
|
ctx->params->preferredTransferCharacteristics = ctx->params->vui.transferCharacteristics; |
|
#endif |
|
ctx->params->vui.matrixCoeffs = avctx->colorspace; |
|
} |
|
|
|
if (avctx->sample_aspect_ratio.num > 0 && avctx->sample_aspect_ratio.den > 0) { |
|
char sar[12]; |
|
int sar_num, sar_den; |
|
|
|
av_reduce(&sar_num, &sar_den, |
|
avctx->sample_aspect_ratio.num, |
|
avctx->sample_aspect_ratio.den, 65535); |
|
snprintf(sar, sizeof(sar), "%d:%d", sar_num, sar_den); |
|
if (ctx->api->param_parse(ctx->params, "sar", sar) == X265_PARAM_BAD_VALUE) { |
|
av_log(avctx, AV_LOG_ERROR, "Invalid SAR: %d:%d.\n", sar_num, sar_den); |
|
return AVERROR_INVALIDDATA; |
|
} |
|
} |
|
|
|
switch (avctx->pix_fmt) { |
|
case AV_PIX_FMT_YUV420P: |
|
case AV_PIX_FMT_YUV420P10: |
|
case AV_PIX_FMT_YUV420P12: |
|
ctx->params->internalCsp = X265_CSP_I420; |
|
break; |
|
case AV_PIX_FMT_YUV422P: |
|
case AV_PIX_FMT_YUV422P10: |
|
case AV_PIX_FMT_YUV422P12: |
|
ctx->params->internalCsp = X265_CSP_I422; |
|
break; |
|
case AV_PIX_FMT_GBRP: |
|
case AV_PIX_FMT_GBRP10: |
|
case AV_PIX_FMT_GBRP12: |
|
ctx->params->vui.matrixCoeffs = AVCOL_SPC_RGB; |
|
ctx->params->vui.bEnableVideoSignalTypePresentFlag = 1; |
|
ctx->params->vui.bEnableColorDescriptionPresentFlag = 1; |
|
case AV_PIX_FMT_YUV444P: |
|
case AV_PIX_FMT_YUV444P10: |
|
case AV_PIX_FMT_YUV444P12: |
|
ctx->params->internalCsp = X265_CSP_I444; |
|
break; |
|
case AV_PIX_FMT_GRAY8: |
|
case AV_PIX_FMT_GRAY10: |
|
case AV_PIX_FMT_GRAY12: |
|
if (ctx->api->api_build_number < 85) { |
|
av_log(avctx, AV_LOG_ERROR, |
|
"libx265 version is %d, must be at least 85 for gray encoding.\n", |
|
ctx->api->api_build_number); |
|
return AVERROR_INVALIDDATA; |
|
} |
|
ctx->params->internalCsp = X265_CSP_I400; |
|
break; |
|
} |
|
|
|
if (ctx->crf >= 0) { |
|
char crf[6]; |
|
|
|
snprintf(crf, sizeof(crf), "%2.2f", ctx->crf); |
|
if (ctx->api->param_parse(ctx->params, "crf", crf) == X265_PARAM_BAD_VALUE) { |
|
av_log(avctx, AV_LOG_ERROR, "Invalid crf: %2.2f.\n", ctx->crf); |
|
return AVERROR(EINVAL); |
|
} |
|
} else if (avctx->bit_rate > 0) { |
|
ctx->params->rc.bitrate = avctx->bit_rate / 1000; |
|
ctx->params->rc.rateControlMode = X265_RC_ABR; |
|
} else if (ctx->cqp >= 0) { |
|
ret = libx265_param_parse_int(avctx, "qp", ctx->cqp); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
|
|
#if X265_BUILD >= 89 |
|
if (avctx->qmin >= 0) { |
|
ret = libx265_param_parse_int(avctx, "qpmin", avctx->qmin); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->qmax >= 0) { |
|
ret = libx265_param_parse_int(avctx, "qpmax", avctx->qmax); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
#endif |
|
if (avctx->max_qdiff >= 0) { |
|
ret = libx265_param_parse_int(avctx, "qpstep", avctx->max_qdiff); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->qblur >= 0) { |
|
ret = libx265_param_parse_float(avctx, "qblur", avctx->qblur); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->qcompress >= 0) { |
|
ret = libx265_param_parse_float(avctx, "qcomp", avctx->qcompress); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->i_quant_factor >= 0) { |
|
ret = libx265_param_parse_float(avctx, "ipratio", avctx->i_quant_factor); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->b_quant_factor >= 0) { |
|
ret = libx265_param_parse_float(avctx, "pbratio", avctx->b_quant_factor); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
|
|
ctx->params->rc.vbvBufferSize = avctx->rc_buffer_size / 1000; |
|
ctx->params->rc.vbvMaxBitrate = avctx->rc_max_rate / 1000; |
|
|
|
cpb_props = ff_add_cpb_side_data(avctx); |
|
if (!cpb_props) |
|
return AVERROR(ENOMEM); |
|
cpb_props->buffer_size = ctx->params->rc.vbvBufferSize * 1000; |
|
cpb_props->max_bitrate = ctx->params->rc.vbvMaxBitrate * 1000LL; |
|
cpb_props->avg_bitrate = ctx->params->rc.bitrate * 1000LL; |
|
|
|
if (!(avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER)) |
|
ctx->params->bRepeatHeaders = 1; |
|
|
|
if (avctx->gop_size >= 0) { |
|
ret = libx265_param_parse_int(avctx, "keyint", avctx->gop_size); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->keyint_min > 0) { |
|
ret = libx265_param_parse_int(avctx, "min-keyint", avctx->keyint_min); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->max_b_frames >= 0) { |
|
ret = libx265_param_parse_int(avctx, "bframes", avctx->max_b_frames); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
if (avctx->refs >= 0) { |
|
ret = libx265_param_parse_int(avctx, "ref", avctx->refs); |
|
if (ret < 0) |
|
return ret; |
|
} |
|
|
|
{ |
|
AVDictionaryEntry *en = NULL; |
|
while ((en = av_dict_get(ctx->x265_opts, "", en, AV_DICT_IGNORE_SUFFIX))) { |
|
int parse_ret = ctx->api->param_parse(ctx->params, en->key, en->value); |
|
|
|
switch (parse_ret) { |
|
case X265_PARAM_BAD_NAME: |
|
av_log(avctx, AV_LOG_WARNING, |
|
"Unknown option: %s.\n", en->key); |
|
break; |
|
case X265_PARAM_BAD_VALUE: |
|
av_log(avctx, AV_LOG_WARNING, |
|
"Invalid value for %s: %s.\n", en->key, en->value); |
|
break; |
|
default: |
|
break; |
|
} |
|
} |
|
} |
|
|
|
if (ctx->params->rc.vbvBufferSize && avctx->rc_initial_buffer_occupancy > 1000 && |
|
ctx->params->rc.vbvBufferInit == 0.9) { |
|
ctx->params->rc.vbvBufferInit = (float)avctx->rc_initial_buffer_occupancy / 1000; |
|
} |
|
|
|
if (ctx->profile) { |
|
if (ctx->api->param_apply_profile(ctx->params, ctx->profile) < 0) { |
|
int i; |
|
av_log(avctx, AV_LOG_ERROR, "Invalid or incompatible profile set: %s.\n", ctx->profile); |
|
av_log(avctx, AV_LOG_INFO, "Possible profiles:"); |
|
for (i = 0; x265_profile_names[i]; i++) |
|
av_log(avctx, AV_LOG_INFO, " %s", x265_profile_names[i]); |
|
av_log(avctx, AV_LOG_INFO, "\n"); |
|
return AVERROR(EINVAL); |
|
} |
|
} |
|
|
|
ctx->encoder = ctx->api->encoder_open(ctx->params); |
|
if (!ctx->encoder) { |
|
av_log(avctx, AV_LOG_ERROR, "Cannot open libx265 encoder.\n"); |
|
libx265_encode_close(avctx); |
|
return AVERROR_INVALIDDATA; |
|
} |
|
|
|
if (avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER) { |
|
x265_nal *nal; |
|
int nnal; |
|
|
|
avctx->extradata_size = ctx->api->encoder_headers(ctx->encoder, &nal, &nnal); |
|
if (avctx->extradata_size <= 0) { |
|
av_log(avctx, AV_LOG_ERROR, "Cannot encode headers.\n"); |
|
libx265_encode_close(avctx); |
|
return AVERROR_INVALIDDATA; |
|
} |
|
|
|
avctx->extradata = av_malloc(avctx->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE); |
|
if (!avctx->extradata) { |
|
av_log(avctx, AV_LOG_ERROR, |
|
"Cannot allocate HEVC header of size %d.\n", avctx->extradata_size); |
|
libx265_encode_close(avctx); |
|
return AVERROR(ENOMEM); |
|
} |
|
|
|
memcpy(avctx->extradata, nal[0].payload, avctx->extradata_size); |
|
memset(avctx->extradata + avctx->extradata_size, 0, AV_INPUT_BUFFER_PADDING_SIZE); |
|
} |
|
|
|
return 0; |
|
} |
|
|
|
static av_cold int libx265_encode_set_roi(libx265Context *ctx, const AVFrame *frame, x265_picture* pic) |
|
{ |
|
AVFrameSideData *sd = av_frame_get_side_data(frame, AV_FRAME_DATA_REGIONS_OF_INTEREST); |
|
if (sd) { |
|
if (ctx->params->rc.aqMode == X265_AQ_NONE) { |
|
if (!ctx->roi_warned) { |
|
ctx->roi_warned = 1; |
|
av_log(ctx, AV_LOG_WARNING, "Adaptive quantization must be enabled to use ROI encoding, skipping ROI.\n"); |
|
} |
|
} else { |
|
/* 8x8 block when qg-size is 8, 16*16 block otherwise. */ |
|
int mb_size = (ctx->params->rc.qgSize == 8) ? 8 : 16; |
|
int mbx = (frame->width + mb_size - 1) / mb_size; |
|
int mby = (frame->height + mb_size - 1) / mb_size; |
|
int qp_range = 51 + 6 * (pic->bitDepth - 8); |
|
int nb_rois; |
|
const AVRegionOfInterest *roi; |
|
uint32_t roi_size; |
|
float *qoffsets; /* will be freed after encode is called. */ |
|
|
|
roi = (const AVRegionOfInterest*)sd->data; |
|
roi_size = roi->self_size; |
|
if (!roi_size || sd->size % roi_size != 0) { |
|
av_log(ctx, AV_LOG_ERROR, "Invalid AVRegionOfInterest.self_size.\n"); |
|
return AVERROR(EINVAL); |
|
} |
|
nb_rois = sd->size / roi_size; |
|
|
|
qoffsets = av_mallocz_array(mbx * mby, sizeof(*qoffsets)); |
|
if (!qoffsets) |
|
return AVERROR(ENOMEM); |
|
|
|
// This list must be iterated in reverse because the first |
|
// region in the list applies when regions overlap. |
|
for (int i = nb_rois - 1; i >= 0; i--) { |
|
int startx, endx, starty, endy; |
|
float qoffset; |
|
|
|
roi = (const AVRegionOfInterest*)(sd->data + roi_size * i); |
|
|
|
starty = FFMIN(mby, roi->top / mb_size); |
|
endy = FFMIN(mby, (roi->bottom + mb_size - 1)/ mb_size); |
|
startx = FFMIN(mbx, roi->left / mb_size); |
|
endx = FFMIN(mbx, (roi->right + mb_size - 1)/ mb_size); |
|
|
|
if (roi->qoffset.den == 0) { |
|
av_free(qoffsets); |
|
av_log(ctx, AV_LOG_ERROR, "AVRegionOfInterest.qoffset.den must not be zero.\n"); |
|
return AVERROR(EINVAL); |
|
} |
|
qoffset = roi->qoffset.num * 1.0f / roi->qoffset.den; |
|
qoffset = av_clipf(qoffset * qp_range, -qp_range, +qp_range); |
|
|
|
for (int y = starty; y < endy; y++) |
|
for (int x = startx; x < endx; x++) |
|
qoffsets[x + y*mbx] = qoffset; |
|
} |
|
|
|
pic->quantOffsets = qoffsets; |
|
} |
|
} |
|
return 0; |
|
} |
|
|
|
static int libx265_encode_frame(AVCodecContext *avctx, AVPacket *pkt, |
|
const AVFrame *pic, int *got_packet) |
|
{ |
|
libx265Context *ctx = avctx->priv_data; |
|
x265_picture x265pic; |
|
x265_picture x265pic_out = { 0 }; |
|
x265_nal *nal; |
|
uint8_t *dst; |
|
int pict_type; |
|
int payload = 0; |
|
int nnal; |
|
int ret; |
|
int i; |
|
|
|
ctx->api->picture_init(ctx->params, &x265pic); |
|
|
|
if (pic) { |
|
for (i = 0; i < 3; i++) { |
|
x265pic.planes[i] = pic->data[i]; |
|
x265pic.stride[i] = pic->linesize[i]; |
|
} |
|
|
|
x265pic.pts = pic->pts; |
|
x265pic.bitDepth = av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].depth; |
|
|
|
x265pic.sliceType = pic->pict_type == AV_PICTURE_TYPE_I ? |
|
(ctx->forced_idr ? X265_TYPE_IDR : X265_TYPE_I) : |
|
pic->pict_type == AV_PICTURE_TYPE_P ? X265_TYPE_P : |
|
pic->pict_type == AV_PICTURE_TYPE_B ? X265_TYPE_B : |
|
X265_TYPE_AUTO; |
|
|
|
ret = libx265_encode_set_roi(ctx, pic, &x265pic); |
|
if (ret < 0) |
|
return ret; |
|
|
|
if (pic->reordered_opaque) { |
|
x265pic.userData = av_malloc(sizeof(pic->reordered_opaque)); |
|
if (!x265pic.userData) { |
|
av_freep(&x265pic.quantOffsets); |
|
return AVERROR(ENOMEM); |
|
} |
|
|
|
memcpy(x265pic.userData, &pic->reordered_opaque, sizeof(pic->reordered_opaque)); |
|
} |
|
} |
|
|
|
ret = ctx->api->encoder_encode(ctx->encoder, &nal, &nnal, |
|
pic ? &x265pic : NULL, &x265pic_out); |
|
|
|
av_freep(&x265pic.quantOffsets); |
|
|
|
if (ret < 0) |
|
return AVERROR_EXTERNAL; |
|
|
|
if (!nnal) |
|
return 0; |
|
|
|
for (i = 0; i < nnal; i++) |
|
payload += nal[i].sizeBytes; |
|
|
|
ret = ff_alloc_packet2(avctx, pkt, payload, payload); |
|
if (ret < 0) { |
|
av_log(avctx, AV_LOG_ERROR, "Error getting output packet.\n"); |
|
return ret; |
|
} |
|
dst = pkt->data; |
|
|
|
for (i = 0; i < nnal; i++) { |
|
memcpy(dst, nal[i].payload, nal[i].sizeBytes); |
|
dst += nal[i].sizeBytes; |
|
|
|
if (is_keyframe(nal[i].type)) |
|
pkt->flags |= AV_PKT_FLAG_KEY; |
|
} |
|
|
|
pkt->pts = x265pic_out.pts; |
|
pkt->dts = x265pic_out.dts; |
|
|
|
switch (x265pic_out.sliceType) { |
|
case X265_TYPE_IDR: |
|
case X265_TYPE_I: |
|
pict_type = AV_PICTURE_TYPE_I; |
|
break; |
|
case X265_TYPE_P: |
|
pict_type = AV_PICTURE_TYPE_P; |
|
break; |
|
case X265_TYPE_B: |
|
case X265_TYPE_BREF: |
|
pict_type = AV_PICTURE_TYPE_B; |
|
break; |
|
default: |
|
av_log(avctx, AV_LOG_ERROR, "Unknown picture type encountered.\n"); |
|
return AVERROR_EXTERNAL; |
|
} |
|
|
|
#if FF_API_CODED_FRAME |
|
FF_DISABLE_DEPRECATION_WARNINGS |
|
avctx->coded_frame->pict_type = pict_type; |
|
FF_ENABLE_DEPRECATION_WARNINGS |
|
#endif |
|
|
|
#if X265_BUILD >= 130 |
|
if (x265pic_out.sliceType == X265_TYPE_B) |
|
#else |
|
if (x265pic_out.frameData.sliceType == 'b') |
|
#endif |
|
pkt->flags |= AV_PKT_FLAG_DISPOSABLE; |
|
|
|
ff_side_data_set_encoder_stats(pkt, x265pic_out.frameData.qp * FF_QP2LAMBDA, NULL, 0, pict_type); |
|
|
|
if (x265pic_out.userData) { |
|
memcpy(&avctx->reordered_opaque, x265pic_out.userData, sizeof(avctx->reordered_opaque)); |
|
av_freep(&x265pic_out.userData); |
|
} else |
|
avctx->reordered_opaque = 0; |
|
|
|
*got_packet = 1; |
|
return 0; |
|
} |
|
|
|
static const enum AVPixelFormat x265_csp_eight[] = { |
|
AV_PIX_FMT_YUV420P, |
|
AV_PIX_FMT_YUVJ420P, |
|
AV_PIX_FMT_YUV422P, |
|
AV_PIX_FMT_YUVJ422P, |
|
AV_PIX_FMT_YUV444P, |
|
AV_PIX_FMT_YUVJ444P, |
|
AV_PIX_FMT_GBRP, |
|
AV_PIX_FMT_GRAY8, |
|
AV_PIX_FMT_NONE |
|
}; |
|
|
|
static const enum AVPixelFormat x265_csp_ten[] = { |
|
AV_PIX_FMT_YUV420P, |
|
AV_PIX_FMT_YUVJ420P, |
|
AV_PIX_FMT_YUV422P, |
|
AV_PIX_FMT_YUVJ422P, |
|
AV_PIX_FMT_YUV444P, |
|
AV_PIX_FMT_YUVJ444P, |
|
AV_PIX_FMT_GBRP, |
|
AV_PIX_FMT_YUV420P10, |
|
AV_PIX_FMT_YUV422P10, |
|
AV_PIX_FMT_YUV444P10, |
|
AV_PIX_FMT_GBRP10, |
|
AV_PIX_FMT_GRAY8, |
|
AV_PIX_FMT_GRAY10, |
|
AV_PIX_FMT_NONE |
|
}; |
|
|
|
static const enum AVPixelFormat x265_csp_twelve[] = { |
|
AV_PIX_FMT_YUV420P, |
|
AV_PIX_FMT_YUVJ420P, |
|
AV_PIX_FMT_YUV422P, |
|
AV_PIX_FMT_YUVJ422P, |
|
AV_PIX_FMT_YUV444P, |
|
AV_PIX_FMT_YUVJ444P, |
|
AV_PIX_FMT_GBRP, |
|
AV_PIX_FMT_YUV420P10, |
|
AV_PIX_FMT_YUV422P10, |
|
AV_PIX_FMT_YUV444P10, |
|
AV_PIX_FMT_GBRP10, |
|
AV_PIX_FMT_YUV420P12, |
|
AV_PIX_FMT_YUV422P12, |
|
AV_PIX_FMT_YUV444P12, |
|
AV_PIX_FMT_GBRP12, |
|
AV_PIX_FMT_GRAY8, |
|
AV_PIX_FMT_GRAY10, |
|
AV_PIX_FMT_GRAY12, |
|
AV_PIX_FMT_NONE |
|
}; |
|
|
|
static av_cold void libx265_encode_init_csp(AVCodec *codec) |
|
{ |
|
if (x265_api_get(12)) |
|
codec->pix_fmts = x265_csp_twelve; |
|
else if (x265_api_get(10)) |
|
codec->pix_fmts = x265_csp_ten; |
|
else if (x265_api_get(8)) |
|
codec->pix_fmts = x265_csp_eight; |
|
} |
|
|
|
#define OFFSET(x) offsetof(libx265Context, x) |
|
#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM |
|
static const AVOption options[] = { |
|
{ "crf", "set the x265 crf", OFFSET(crf), AV_OPT_TYPE_FLOAT, { .dbl = -1 }, -1, FLT_MAX, VE }, |
|
{ "qp", "set the x265 qp", OFFSET(cqp), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, VE }, |
|
{ "forced-idr", "if forcing keyframes, force them as IDR frames", OFFSET(forced_idr),AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VE }, |
|
{ "preset", "set the x265 preset", OFFSET(preset), AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE }, |
|
{ "tune", "set the x265 tune parameter", OFFSET(tune), AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE }, |
|
{ "profile", "set the x265 profile", OFFSET(profile), AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE }, |
|
{ "x265-params", "set the x265 configuration using a :-separated list of key=value parameters", OFFSET(x265_opts), AV_OPT_TYPE_DICT, { 0 }, 0, 0, VE }, |
|
{ NULL } |
|
}; |
|
|
|
static const AVClass class = { |
|
.class_name = "libx265", |
|
.item_name = av_default_item_name, |
|
.option = options, |
|
.version = LIBAVUTIL_VERSION_INT, |
|
}; |
|
|
|
static const AVCodecDefault x265_defaults[] = { |
|
{ "b", "0" }, |
|
{ "bf", "-1" }, |
|
{ "g", "-1" }, |
|
{ "keyint_min", "-1" }, |
|
{ "refs", "-1" }, |
|
{ "qmin", "-1" }, |
|
{ "qmax", "-1" }, |
|
{ "qdiff", "-1" }, |
|
{ "qblur", "-1" }, |
|
{ "qcomp", "-1" }, |
|
{ "i_qfactor", "-1" }, |
|
{ "b_qfactor", "-1" }, |
|
{ NULL }, |
|
}; |
|
|
|
AVCodec ff_libx265_encoder = { |
|
.name = "libx265", |
|
.long_name = NULL_IF_CONFIG_SMALL("libx265 H.265 / HEVC"), |
|
.type = AVMEDIA_TYPE_VIDEO, |
|
.id = AV_CODEC_ID_HEVC, |
|
.init = libx265_encode_init, |
|
.init_static_data = libx265_encode_init_csp, |
|
.encode2 = libx265_encode_frame, |
|
.close = libx265_encode_close, |
|
.priv_data_size = sizeof(libx265Context), |
|
.priv_class = &class, |
|
.defaults = x265_defaults, |
|
.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AUTO_THREADS | |
|
AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE, |
|
.wrapper_name = "libx265", |
|
};
|
|
|