bb6b1731 |
/*
* 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
*/
|
4e9b46b0 |
#if defined(_MSC_VER)
#define X265_API_IMPORTS 1
#endif
|
bb6b1731 |
#include <x265.h> |
8b96e8dd |
#include <float.h> |
bb6b1731 |
#include "libavutil/internal.h"
#include "libavutil/common.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "avcodec.h"
#include "internal.h"
typedef struct libx265Context {
const AVClass *class;
x265_encoder *encoder;
x265_param *params; |
94c20de4 |
const x265_api *api; |
bb6b1731 |
|
8b96e8dd |
float crf; |
8a8902f2 |
int forced_idr; |
bb6b1731 |
char *preset;
char *tune;
char *x265_opts;
} 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;
|
94c20de4 |
ctx->api->param_free(ctx->params); |
bb6b1731 |
if (ctx->encoder) |
94c20de4 |
ctx->api->encoder_close(ctx->encoder); |
bb6b1731 |
return 0;
}
static av_cold int libx265_encode_init(AVCodecContext *avctx)
{
libx265Context *ctx = avctx->priv_data;
|
2268db2c |
ctx->api = x265_api_get(av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].depth); |
94c20de4 |
if (!ctx->api)
ctx->api = x265_api_get(0);
ctx->params = ctx->api->param_alloc(); |
bb6b1731 |
if (!ctx->params) {
av_log(avctx, AV_LOG_ERROR, "Could not allocate x265 param structure.\n");
return AVERROR(ENOMEM);
}
|
94c20de4 |
if (ctx->api->param_default_preset(ctx->params, ctx->preset, ctx->tune) < 0) { |
61090db2 |
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");
|
bb6b1731 |
return AVERROR(EINVAL);
}
ctx->params->frameNumThreads = avctx->thread_count; |
d3736471 |
ctx->params->fpsNum = avctx->time_base.den;
ctx->params->fpsDenom = avctx->time_base.num * avctx->ticks_per_frame; |
bb6b1731 |
ctx->params->sourceWidth = avctx->width;
ctx->params->sourceHeight = avctx->height; |
7c6eb0a1 |
ctx->params->bEnablePsnr = !!(avctx->flags & AV_CODEC_FLAG_PSNR); |
d3736471 |
|
ec89f35e |
if ((avctx->color_primaries <= AVCOL_PRI_BT2020 &&
avctx->color_primaries != AVCOL_PRI_UNSPECIFIED) ||
(avctx->color_trc <= AVCOL_TRC_BT2020_12 &&
avctx->color_trc != AVCOL_TRC_UNSPECIFIED) ||
(avctx->colorspace <= AVCOL_SPC_BT2020_CL &&
avctx->colorspace != AVCOL_SPC_UNSPECIFIED)) {
ctx->params->vui.bEnableVideoSignalTypePresentFlag = 1;
ctx->params->vui.bEnableColorDescriptionPresentFlag = 1;
// x265 validates the parameters internally
ctx->params->vui.colorPrimaries = avctx->color_primaries;
ctx->params->vui.transferCharacteristics = avctx->color_trc;
ctx->params->vui.matrixCoeffs = avctx->colorspace;
}
|
cdd21f1f |
if (avctx->sample_aspect_ratio.num > 0 && avctx->sample_aspect_ratio.den > 0) { |
b920db67 |
char sar[12];
int sar_num, sar_den;
|
cdd21f1f |
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); |
94c20de4 |
if (ctx->api->param_parse(ctx->params, "sar", sar) == X265_PARAM_BAD_VALUE) { |
cdd21f1f |
av_log(avctx, AV_LOG_ERROR, "Invalid SAR: %d:%d.\n", sar_num, sar_den);
return AVERROR_INVALIDDATA;
} |
8945dcbb |
} |
73ee4cf3 |
|
d102925a |
switch (avctx->pix_fmt) {
case AV_PIX_FMT_YUV420P:
case AV_PIX_FMT_YUV420P10: |
70742e59 |
case AV_PIX_FMT_YUV420P12: |
d102925a |
ctx->params->internalCsp = X265_CSP_I420;
break; |
6ee95af8 |
case AV_PIX_FMT_YUV422P:
case AV_PIX_FMT_YUV422P10: |
70742e59 |
case AV_PIX_FMT_YUV422P12: |
6ee95af8 |
ctx->params->internalCsp = X265_CSP_I422;
break; |
d11d78fa |
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; |
d102925a |
case AV_PIX_FMT_YUV444P:
case AV_PIX_FMT_YUV444P10: |
70742e59 |
case AV_PIX_FMT_YUV444P12: |
d102925a |
ctx->params->internalCsp = X265_CSP_I444;
break; |
3e4357eb |
case AV_PIX_FMT_GRAY8: |
587226ad |
case AV_PIX_FMT_GRAY10:
case AV_PIX_FMT_GRAY12: |
3e4357eb |
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; |
d102925a |
}
|
8b96e8dd |
if (ctx->crf >= 0) {
char crf[6];
snprintf(crf, sizeof(crf), "%2.2f", ctx->crf); |
94c20de4 |
if (ctx->api->param_parse(ctx->params, "crf", crf) == X265_PARAM_BAD_VALUE) { |
8b96e8dd |
av_log(avctx, AV_LOG_ERROR, "Invalid crf: %2.2f.\n", ctx->crf);
return AVERROR(EINVAL);
}
} else if (avctx->bit_rate > 0) { |
bb6b1731 |
ctx->params->rc.bitrate = avctx->bit_rate / 1000;
ctx->params->rc.rateControlMode = X265_RC_ABR;
}
|
7c6eb0a1 |
if (!(avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER)) |
1b28d9b3 |
ctx->params->bRepeatHeaders = 1;
|
bb6b1731 |
if (ctx->x265_opts) {
AVDictionary *dict = NULL;
AVDictionaryEntry *en = NULL;
if (!av_dict_parse_string(&dict, ctx->x265_opts, "=", ":", 0)) {
while ((en = av_dict_get(dict, "", en, AV_DICT_IGNORE_SUFFIX))) { |
94c20de4 |
int parse_ret = ctx->api->param_parse(ctx->params, en->key, en->value); |
bb6b1731 |
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;
}
}
av_dict_free(&dict);
}
}
|
94c20de4 |
ctx->encoder = ctx->api->encoder_open(ctx->params); |
bb6b1731 |
if (!ctx->encoder) {
av_log(avctx, AV_LOG_ERROR, "Cannot open libx265 encoder.\n");
libx265_encode_close(avctx); |
50ea9315 |
return AVERROR_INVALIDDATA; |
bb6b1731 |
}
|
7c6eb0a1 |
if (avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER) { |
b920db67 |
x265_nal *nal;
int nnal;
|
94c20de4 |
avctx->extradata_size = ctx->api->encoder_headers(ctx->encoder, &nal, &nnal); |
1b28d9b3 |
if (avctx->extradata_size <= 0) {
av_log(avctx, AV_LOG_ERROR, "Cannot encode headers.\n");
libx265_encode_close(avctx);
return AVERROR_INVALIDDATA;
} |
206d5330 |
|
059a9348 |
avctx->extradata = av_malloc(avctx->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE); |
1b28d9b3 |
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); |
dded5ed9 |
}
|
bb6b1731 |
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; |
04070dbc |
x265_picture x265pic_out = { 0 }; |
bb6b1731 |
x265_nal *nal;
uint8_t *dst;
int payload = 0;
int nnal;
int ret;
int i;
|
94c20de4 |
ctx->api->picture_init(ctx->params, &x265pic); |
955544e4 |
|
bb6b1731 |
if (pic) {
for (i = 0; i < 3; i++) {
x265pic.planes[i] = pic->data[i];
x265pic.stride[i] = pic->linesize[i];
}
|
d3736471 |
x265pic.pts = pic->pts; |
2268db2c |
x265pic.bitDepth = av_pix_fmt_desc_get(avctx->pix_fmt)->comp[0].depth; |
41e983f1 |
|
8a8902f2 |
x265pic.sliceType = pic->pict_type == AV_PICTURE_TYPE_I ?
(ctx->forced_idr ? X265_TYPE_IDR : X265_TYPE_I) : |
41e983f1 |
pic->pict_type == AV_PICTURE_TYPE_P ? X265_TYPE_P :
pic->pict_type == AV_PICTURE_TYPE_B ? X265_TYPE_B :
X265_TYPE_AUTO; |
bb6b1731 |
}
|
94c20de4 |
ret = ctx->api->encoder_encode(ctx->encoder, &nal, &nnal,
pic ? &x265pic : NULL, &x265pic_out); |
bb6b1731 |
if (ret < 0) |
6d18154f |
return AVERROR_EXTERNAL; |
bb6b1731 |
if (!nnal)
return 0;
for (i = 0; i < nnal; i++)
payload += nal[i].sizeBytes;
ret = ff_alloc_packet(pkt, 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;
|
40cf1bba |
#if FF_API_CODED_FRAME
FF_DISABLE_DEPRECATION_WARNINGS |
2ddedfd3 |
switch (x265pic_out.sliceType) {
case X265_TYPE_IDR:
case X265_TYPE_I:
avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;
break;
case X265_TYPE_P:
avctx->coded_frame->pict_type = AV_PICTURE_TYPE_P;
break;
case X265_TYPE_B:
avctx->coded_frame->pict_type = AV_PICTURE_TYPE_B;
break;
} |
40cf1bba |
FF_ENABLE_DEPRECATION_WARNINGS
#endif |
2ddedfd3 |
|
bb6b1731 |
*got_packet = 1;
return 0;
}
static const enum AVPixelFormat x265_csp_eight[] = {
AV_PIX_FMT_YUV420P, |
6ee95af8 |
AV_PIX_FMT_YUV422P, |
d102925a |
AV_PIX_FMT_YUV444P, |
d11d78fa |
AV_PIX_FMT_GBRP, |
3e4357eb |
AV_PIX_FMT_GRAY8, |
bb6b1731 |
AV_PIX_FMT_NONE
};
|
70742e59 |
static const enum AVPixelFormat x265_csp_ten[] = {
AV_PIX_FMT_YUV420P,
AV_PIX_FMT_YUV422P,
AV_PIX_FMT_YUV444P, |
d11d78fa |
AV_PIX_FMT_GBRP, |
70742e59 |
AV_PIX_FMT_YUV420P10,
AV_PIX_FMT_YUV422P10,
AV_PIX_FMT_YUV444P10, |
d11d78fa |
AV_PIX_FMT_GBRP10, |
3e4357eb |
AV_PIX_FMT_GRAY8, |
587226ad |
AV_PIX_FMT_GRAY10, |
70742e59 |
AV_PIX_FMT_NONE
};
|
bb6b1731 |
static const enum AVPixelFormat x265_csp_twelve[] = {
AV_PIX_FMT_YUV420P, |
6ee95af8 |
AV_PIX_FMT_YUV422P, |
d102925a |
AV_PIX_FMT_YUV444P, |
d11d78fa |
AV_PIX_FMT_GBRP, |
bb6b1731 |
AV_PIX_FMT_YUV420P10, |
6ee95af8 |
AV_PIX_FMT_YUV422P10, |
d102925a |
AV_PIX_FMT_YUV444P10, |
d11d78fa |
AV_PIX_FMT_GBRP10, |
70742e59 |
AV_PIX_FMT_YUV420P12,
AV_PIX_FMT_YUV422P12,
AV_PIX_FMT_YUV444P12, |
d11d78fa |
AV_PIX_FMT_GBRP12, |
3e4357eb |
AV_PIX_FMT_GRAY8, |
587226ad |
AV_PIX_FMT_GRAY10,
AV_PIX_FMT_GRAY12, |
bb6b1731 |
AV_PIX_FMT_NONE
};
static av_cold void libx265_encode_init_csp(AVCodec *codec)
{ |
70742e59 |
if (x265_api_get(12)) |
bb6b1731 |
codec->pix_fmts = x265_csp_twelve; |
70742e59 |
else if (x265_api_get(10))
codec->pix_fmts = x265_csp_ten; |
b012bd50 |
else if (x265_api_get(8))
codec->pix_fmts = x265_csp_eight; |
bb6b1731 |
}
#define OFFSET(x) offsetof(libx265Context, x)
#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
static const AVOption options[] = { |
8b96e8dd |
{ "crf", "set the x265 crf", OFFSET(crf), AV_OPT_TYPE_FLOAT, { .dbl = -1 }, -1, FLT_MAX, VE }, |
8a8902f2 |
{ "forced-idr", "if forcing keyframes, force them as IDR frames", OFFSET(forced_idr),AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VE }, |
bb6b1731 |
{ "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 },
{ "x265-params", "set the x265 configuration using a :-separated list of key=value parameters", OFFSET(x265_opts), AV_OPT_TYPE_STRING, { 0 }, 0, 0, VE },
{ NULL }
};
static const AVClass class = {
.class_name = "libx265",
.item_name = av_default_item_name,
.option = options,
.version = LIBAVUTIL_VERSION_INT,
};
|
4912b634 |
static const AVCodecDefault x265_defaults[] = {
{ "b", "0" },
{ NULL },
};
|
bb6b1731 |
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, |
4912b634 |
.defaults = x265_defaults, |
def97856 |
.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_AUTO_THREADS, |
bb6b1731 |
}; |