1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-26 19:01:44 +02:00
FFmpeg/libavcodec/libxeve.c
Niklas Haas 703288cec6 avcodec/internal: add FFCodec.color_ranges
I went through all codecs and put them into five basic categories:

1. JPEG range only
2. MPEG range only
3. Explicitly tagged
4. Broken (codec supports both but encoder ignores tags)
5. N/A (headerless or pseudo-formats)

Filters in category 5 remain untouched. The rest gain an explicit
assignment of their supported color ranges, with codecs in category
4 being set to MPEG-only for safety.

It might be considered redundant to distinguish between 0 (category 5)
and MPEG+JPEG (category 3), but in doing so we effectively communicate
that we can guarantee that these tags will be encoded, which is distinct
from the situation where there are some codecs that simply don't have
tagging or implied semantics (e.g. rawvideo).

A full list of codecs follows:

JPEG range only:
 - amv
 - roqvideo

MPEG range only:
 - asv1, asv2
 - avui
 - cfhd
 - cljr
 - dnxhd
 - dvvideo
 - ffv1
 - flv
 - h261, h263, h263p
 - {h263,vp8}_v4l2m2m
 - huffyuv, ffvhuff
 - jpeg2000
 - libopenjpeg
 - libtheora
 - libwebp, libwebp_anim
 - libx262
 - libxavs, libxavs2
 - libxvid
 - mpeg1video, mpeg2video
 - mpeg2_qsv
 - mpeg2_vaapi
 - mpeg4, msmpeg4, msmpeg4v2, wmv1, wmv2
 - mpeg4_omx
 - prores, prores_aw, prores_ks
 - rv10, rv20
 - snow
 - speedhq
 - svq1
 - tiff
 - utvideo

Explicitly tagged (MPEG/JPEG):
 - {av1,h264,hevc}_nvenc
 - {av1,h264,hevc}_vaapi
 - {av1,h264,hevc,vp8,vp9,mpeg4}_mediacodec
 - {av1,h264,hevc,vp9}_qsv
 - h264_amf
 - {h264,hevc,prores}_videotoolbox
 - libaom-av1
 - libkvazaar
 - libopenh264
 - librav1e
 - libsvtav1
 - libvpx, libvpx-vp9
 - libx264
 - libx265
 - ljpeg
 - mjpeg
 - vc2

Broken (encoder ignores tags):
 - {av1,hevc}_amf
 - {h264,hevc,mpeg4}_v4l2m2m
 - h264_omx
 - libxeve
 - magicyuv
 - {vp8,vp9,mjpeg}_vaapi

N/A:
 - ayuv, yuv4, y41p, v308, v210, v410, v408 (headerless)
 - pgmyuv (headerless)
 - rawvideo, bitpacked (headerless)
 - vnull, wrapped_avframe (pseudocodecs)
2024-09-08 13:58:11 +02:00

620 lines
23 KiB
C

/*
* libxeve encoder
* EVC (MPEG-5 Essential Video Coding) encoding using XEVE MPEG-5 EVC encoder library
*
* Copyright (C) 2021 Dawid Kozinski <d.kozinski@samsung.com>
*
* 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
*/
#include <float.h>
#include <stdlib.h>
#include <xeve.h>
#include "libavutil/internal.h"
#include "libavutil/common.h"
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/pixfmt.h"
#include "libavutil/time.h"
#include "libavutil/cpu.h"
#include "libavutil/avstring.h"
#include "avcodec.h"
#include "internal.h"
#include "packet_internal.h"
#include "codec_internal.h"
#include "profiles.h"
#include "encode.h"
#define MAX_BS_BUF (16*1024*1024)
/**
* Error codes
*/
#define XEVE_PARAM_BAD_NAME -100
#define XEVE_PARAM_BAD_VALUE -200
/**
* Encoder states
*
* STATE_ENCODING - the encoder receives and processes input frames
* STATE_BUMPING - there are no more input frames, however the encoder still processes previously received data
*/
typedef enum State {
STATE_ENCODING,
STATE_BUMPING,
} State;
/**
* The structure stores all the states associated with the instance of Xeve MPEG-5 EVC encoder
*/
typedef struct XeveContext {
const AVClass *class;
XEVE id; // XEVE instance identifier
XEVE_CDSC cdsc; // coding parameters i.e profile, width & height of input frame, num of therads, frame rate ...
XEVE_BITB bitb; // bitstream buffer (output)
XEVE_STAT stat; // encoding status (output)
XEVE_IMGB imgb; // image buffer (input)
State state; // encoder state (skipping, encoding, bumping)
int profile_id; // encoder profile (main, baseline)
int preset_id; // preset of xeve ( fast, medium, slow, placebo)
int tune_id; // tune of xeve (psnr, zerolatency)
// variables for rate control modes
int rc_mode; // Rate control mode [ 0(CQP) / 1(ABR) / 2(CRF) ]
int qp; // quantization parameter (QP) [0,51]
int crf; // constant rate factor (CRF) [10,49]
int hash; // embed picture signature (HASH) for conformance checking in decoding
int sei_info; // embed Supplemental enhancement information while encoding
int color_format; // input data color format: currently only XEVE_CF_YCBCR420 is supported
AVDictionary *xeve_params;
} XeveContext;
/**
* Convert FFmpeg pixel format (AVPixelFormat) to XEVE pre-defined color format
*
* @param[in] av_pix_fmt pixel format (@see https://ffmpeg.org/doxygen/trunk/pixfmt_8h.html#a9a8e335cf3be472042bc9f0cf80cd4c5)
* @param[out] xeve_col_fmt XEVE pre-defined color format (@see xeve.h)
*
* @return 0 on success, negative value on failure
*/
static int libxeve_color_fmt(enum AVPixelFormat av_pix_fmt, int *xeve_col_fmt)
{
switch (av_pix_fmt) {
case AV_PIX_FMT_YUV420P:
*xeve_col_fmt = XEVE_CF_YCBCR420;
break;
case AV_PIX_FMT_YUV420P10:
*xeve_col_fmt = XEVE_CF_YCBCR420;
break;
default:
*xeve_col_fmt = XEVE_CF_UNKNOWN;
return AVERROR_INVALIDDATA;
}
return 0;
}
/**
* Convert FFmpeg pixel format (AVPixelFormat) into XEVE pre-defined color space
*
* @param[in] px_fmt pixel format (@see https://ffmpeg.org/doxygen/trunk/pixfmt_8h.html#a9a8e335cf3be472042bc9f0cf80cd4c5)
*
* @return XEVE pre-defined color space (@see xeve.h) on success, XEVE_CF_UNKNOWN on failure
*/
static int libxeve_color_space(enum AVPixelFormat av_pix_fmt)
{
/* color space of input image */
int cs = XEVE_CF_UNKNOWN;
switch (av_pix_fmt) {
case AV_PIX_FMT_YUV420P:
cs = XEVE_CS_YCBCR420;
break;
case AV_PIX_FMT_YUV420P10:
#if AV_HAVE_BIGENDIAN
cs = XEVE_CS_SET(XEVE_CF_YCBCR420, 10, 1);
#else
cs = XEVE_CS_YCBCR420_10LE;
#endif
break;
default:
cs = XEVE_CF_UNKNOWN;
break;
}
return cs;
}
/**
* The function returns a pointer to the object of the XEVE_CDSC type.
* XEVE_CDSC contains all encoder parameters that should be initialized before the encoder is used.
*
* The field values of the XEVE_CDSC structure are populated based on:
* - the corresponding field values of the AvCodecConetxt structure,
* - the xeve encoder specific option values,
* (the full list of options available for xeve encoder is displayed after executing the command ./ffmpeg --help encoder = libxeve)
*
* The order of processing input data and populating the XEVE_CDSC structure
* 1) first, the fields of the AVCodecContext structure corresponding to the provided input options are processed,
* (i.e -pix_fmt yuv420p -s:v 1920x1080 -r 30 -profile:v 0)
* 2) then xeve-specific options added as AVOption to the xeve AVCodec implementation
* (i.e -preset 0)
*
* Keep in mind that, there are options that can be set in different ways.
* In this case, please follow the above-mentioned order of processing.
* The most recent assignments overwrite the previous values.
*
* @param[in] avctx codec context (AVCodecContext)
* @param[out] cdsc contains all Xeve MPEG-5 EVC encoder encoder parameters that should be initialized before the encoder is use
*
* @return 0 on success, negative error code on failure
*/
static int get_conf(AVCodecContext *avctx, XEVE_CDSC *cdsc)
{
XeveContext *xectx = NULL;
int ret;
xectx = avctx->priv_data;
/* initialize xeve_param struct with default values */
ret = xeve_param_default(&cdsc->param);
if (XEVE_FAILED(ret)) {
av_log(avctx, AV_LOG_ERROR, "Cannot set_default parameter\n");
return AVERROR_EXTERNAL;
}
/* read options from AVCodecContext */
if (avctx->width > 0)
cdsc->param.w = avctx->width;
if (avctx->height > 0)
cdsc->param.h = avctx->height;
if (avctx->framerate.num > 0) {
// fps can be float number, but xeve API doesn't support it
cdsc->param.fps.num = avctx->framerate.num;
cdsc->param.fps.den = avctx->framerate.den;
}
// GOP size (key-frame interval, I-picture period)
cdsc->param.keyint = avctx->gop_size; // 0: only one I-frame at the first time; 1: every frame is coded in I-frame
if (avctx->max_b_frames == 0 || avctx->max_b_frames == 1 || avctx->max_b_frames == 3 ||
avctx->max_b_frames == 7 || avctx->max_b_frames == 15) // number of b-frames
cdsc->param.bframes = avctx->max_b_frames;
else {
av_log(avctx, AV_LOG_ERROR, "Incorrect value for maximum number of B frames: (%d) \n"
"Acceptable values for bf option (maximum number of B frames) are 0,1,3,7 or 15\n", avctx->max_b_frames);
return AVERROR_INVALIDDATA;
}
cdsc->param.level_idc = avctx->level;
if (avctx->rc_buffer_size) // VBV buf size
cdsc->param.vbv_bufsize = (int)(avctx->rc_buffer_size / 1000);
cdsc->param.rc_type = xectx->rc_mode;
if (xectx->rc_mode == XEVE_RC_CQP)
cdsc->param.qp = xectx->qp;
else if (xectx->rc_mode == XEVE_RC_ABR) {
if (avctx->bit_rate / 1000 > INT_MAX || avctx->rc_max_rate / 1000 > INT_MAX) {
av_log(avctx, AV_LOG_ERROR, "Not supported bitrate bit_rate and rc_max_rate > %d000\n", INT_MAX);
return AVERROR_INVALIDDATA;
}
cdsc->param.bitrate = (int)(avctx->bit_rate / 1000);
} else if (xectx->rc_mode == XEVE_RC_CRF)
cdsc->param.crf = xectx->crf;
else {
av_log(avctx, AV_LOG_ERROR, "Not supported rate control type: %d\n", xectx->rc_mode);
return AVERROR_INVALIDDATA;
}
if (avctx->thread_count <= 0) {
int cpu_count = av_cpu_count();
cdsc->param.threads = (cpu_count < XEVE_MAX_THREADS) ? cpu_count : XEVE_MAX_THREADS;
} else if (avctx->thread_count > XEVE_MAX_THREADS)
cdsc->param.threads = XEVE_MAX_THREADS;
else
cdsc->param.threads = avctx->thread_count;
libxeve_color_fmt(avctx->pix_fmt, &xectx->color_format);
cdsc->param.cs = XEVE_CS_SET(xectx->color_format, cdsc->param.codec_bit_depth, AV_HAVE_BIGENDIAN);
cdsc->max_bs_buf_size = MAX_BS_BUF;
ret = xeve_param_ppt(&cdsc->param, xectx->profile_id, xectx->preset_id, xectx->tune_id);
if (XEVE_FAILED(ret)) {
av_log(avctx, AV_LOG_ERROR, "Cannot set profile(%d), preset(%d), tune(%d)\n", xectx->profile_id, xectx->preset_id, xectx->tune_id);
return AVERROR_EXTERNAL;
}
return 0;
}
/**
* Set XEVE_CFG_SET_USE_PIC_SIGNATURE for encoder
*
* @param[in] logger context
* @param[in] id XEVE encodec instance identifier
* @param[in] ctx the structure stores all the states associated with the instance of Xeve MPEG-5 EVC encoder
*
* @return 0 on success, negative error code on failure
*/
static int set_extra_config(AVCodecContext *avctx, XEVE id, XeveContext *ctx)
{
int ret, size;
size = 4;
// embed SEI messages identifying encoder parameters and command line arguments
// - 0: off\n"
// - 1: emit sei info"
//
// SEI - Supplemental enhancement information contains information
// that is not necessary to decode the samples of coded pictures from VCL NAL units.
// Some SEI message information is required to check bitstream conformance
// and for output timing decoder conformance.
// @see ISO_IEC_23094-1_2020 7.4.3.5
// @see ISO_IEC_23094-1_2020 Annex D
ret = xeve_config(id, XEVE_CFG_SET_SEI_CMD, &ctx->sei_info, &size); // sei_cmd_info
if (XEVE_FAILED(ret)) {
av_log(avctx, AV_LOG_ERROR, "Failed to set config for sei command info messages\n");
return AVERROR_EXTERNAL;
}
ret = xeve_config(id, XEVE_CFG_SET_USE_PIC_SIGNATURE, &ctx->hash, &size);
if (XEVE_FAILED(ret)) {
av_log(avctx, AV_LOG_ERROR, "Failed to set config for picture signature\n");
return AVERROR_EXTERNAL;
}
return 0;
}
/**
* @brief Switch encoder to bumping mode
*
* @param id XEVE encodec instance identifier
* @return 0 on success, negative error code on failure
*/
static int setup_bumping(XEVE id)
{
int val = 1;
int size = sizeof(int);
if (XEVE_FAILED(xeve_config(id, XEVE_CFG_SET_FORCE_OUT, (void *)(&val), &size)))
return AVERROR_EXTERNAL;
return 0;
}
/**
* @brief Initialize eXtra-fast Essential Video Encoder codec
* Create an encoder instance and allocate all the needed resources
*
* @param avctx codec context
* @return 0 on success, negative error code on failure
*/
static av_cold int libxeve_init(AVCodecContext *avctx)
{
XeveContext *xectx = avctx->priv_data;
unsigned char *bs_buf = NULL;
int i;
int shift_h = 0;
int shift_v = 0;
int width_chroma = 0;
int height_chroma = 0;
XEVE_IMGB *imgb = NULL;
int ret = 0;
XEVE_CDSC *cdsc = &(xectx->cdsc);
/* allocate bitstream buffer */
bs_buf = av_malloc(MAX_BS_BUF);
if (bs_buf == NULL) {
av_log(avctx, AV_LOG_ERROR, "Cannot allocate bitstream buffer\n");
return AVERROR(ENOMEM);
}
xectx->bitb.addr = bs_buf;
xectx->bitb.bsize = MAX_BS_BUF;
/* read configurations and set values for created descriptor (XEVE_CDSC) */
if ((ret = get_conf(avctx, cdsc)) != 0) {
av_log(avctx, AV_LOG_ERROR, "Cannot get configuration\n");
return AVERROR(EINVAL);
}
if ((ret = xeve_param_check(&cdsc->param)) != 0) {
av_log(avctx, AV_LOG_ERROR, "Invalid configuration\n");
return AVERROR(EINVAL);
}
{
const AVDictionaryEntry *en = NULL;
while (en = av_dict_iterate(xectx->xeve_params, en)) {
if ((ret = xeve_param_parse(&cdsc->param, en->key, en->value)) < 0) {
av_log(avctx, AV_LOG_WARNING,
"Error parsing option '%s = %s'.\n",
en->key, en->value);
}
}
}
/* create encoder */
xectx->id = xeve_create(cdsc, NULL);
if (xectx->id == NULL) {
av_log(avctx, AV_LOG_ERROR, "Cannot create XEVE encoder\n");
return AVERROR_EXTERNAL;
}
if ((ret = set_extra_config(avctx, xectx->id, xectx)) != 0) {
av_log(avctx, AV_LOG_ERROR, "Cannot set extra configuration\n");
return AVERROR(EINVAL);
}
if ((ret = av_pix_fmt_get_chroma_sub_sample(avctx->pix_fmt, &shift_h, &shift_v)) != 0) {
av_log(avctx, AV_LOG_ERROR, "Failed to get chroma shift\n");
return AVERROR(EINVAL);
}
// Chroma subsampling
//
// YUV format explanation
// shift_h == 1 && shift_v == 1 : YUV420
// shift_h == 1 && shift_v == 0 : YUV422
// shift_h == 0 && shift_v == 0 : YUV444
//
width_chroma = AV_CEIL_RSHIFT(avctx->width, shift_h);
height_chroma = AV_CEIL_RSHIFT(avctx->height, shift_v);
/* set default values for input image buffer */
imgb = &xectx->imgb;
imgb->cs = libxeve_color_space(avctx->pix_fmt);
imgb->np = 3; /* only for yuv420p, yuv420ple */
for (i = 0; i < imgb->np; i++)
imgb->x[i] = imgb->y[i] = 0;
imgb->w[0] = imgb->aw[0] = avctx->width; // width luma
imgb->w[1] = imgb->w[2] = imgb->aw[1] = imgb->aw[2] = width_chroma;
imgb->h[0] = imgb->ah[0] = avctx->height; // height luma
imgb->h[1] = imgb->h[2] = imgb->ah[1] = imgb->ah[2] = height_chroma;
xectx->state = STATE_ENCODING;
return 0;
}
/**
* Encode raw data frame into EVC packet
*
* @param[in] avctx codec context
* @param[out] avpkt output AVPacket containing encoded data
* @param[in] frame AVFrame containing the raw data to be encoded
* @param[out] got_packet encoder sets to 0 or 1 to indicate that a
* non-empty packet was returned in pkt
*
* @return 0 on success, negative error code on failure
*/
static int libxeve_encode(AVCodecContext *avctx, AVPacket *avpkt,
const AVFrame *frame, int *got_packet)
{
XeveContext *xectx = avctx->priv_data;
int ret = -1;
// No more input frames are available but encoder still can have some data in its internal buffer to process
// and some frames to dump.
if (xectx->state == STATE_ENCODING && frame == NULL) {
if (setup_bumping(xectx->id) == 0)
xectx->state = STATE_BUMPING; // Entering bumping process
else {
av_log(avctx, AV_LOG_ERROR, "Failed to setup bumping\n");
return 0;
}
}
if (xectx->state == STATE_ENCODING) {
int i;
XEVE_IMGB *imgb = NULL;
imgb = &xectx->imgb;
for (i = 0; i < imgb->np; i++) {
imgb->a[i] = frame->data[i];
imgb->s[i] = frame->linesize[i];
}
imgb->ts[XEVE_TS_PTS] = frame->pts;
/* push image to encoder */
ret = xeve_push(xectx->id, imgb);
if (XEVE_FAILED(ret)) {
av_log(avctx, AV_LOG_ERROR, "xeve_push() failed\n");
return AVERROR_EXTERNAL;
}
}
if (xectx->state == STATE_ENCODING || xectx->state == STATE_BUMPING) {
/* encoding */
ret = xeve_encode(xectx->id, &(xectx->bitb), &(xectx->stat));
if (XEVE_FAILED(ret)) {
av_log(avctx, AV_LOG_ERROR, "xeve_encode() failed\n");
return AVERROR_EXTERNAL;
}
/* store bitstream */
if (ret == XEVE_OK_OUT_NOT_AVAILABLE) { // Return OK but picture is not available yet
*got_packet = 0;
return 0;
} else if (ret == XEVE_OK) {
int av_pic_type;
if (xectx->stat.write > 0) {
ret = ff_get_encode_buffer(avctx, avpkt, xectx->stat.write, 0);
if (ret < 0)
return ret;
memcpy(avpkt->data, xectx->bitb.addr, xectx->stat.write);
avpkt->time_base.num = xectx->cdsc.param.fps.den;
avpkt->time_base.den = xectx->cdsc.param.fps.num;
avpkt->pts = xectx->bitb.ts[XEVE_TS_PTS];
avpkt->dts = xectx->bitb.ts[XEVE_TS_DTS];
switch(xectx->stat.stype) {
case XEVE_ST_I:
av_pic_type = AV_PICTURE_TYPE_I;
avpkt->flags |= AV_PKT_FLAG_KEY;
break;
case XEVE_ST_P:
av_pic_type = AV_PICTURE_TYPE_P;
break;
case XEVE_ST_B:
av_pic_type = AV_PICTURE_TYPE_B;
break;
case XEVE_ST_UNKNOWN:
av_log(avctx, AV_LOG_ERROR, "Unknown slice type\n");
return AVERROR_INVALIDDATA;
}
ff_side_data_set_encoder_stats(avpkt, xectx->stat.qp * FF_QP2LAMBDA, NULL, 0, av_pic_type);
*got_packet = 1;
}
} else if (ret == XEVE_OK_NO_MORE_FRM) {
// Return OK but no more frames
return 0;
} else {
av_log(avctx, AV_LOG_ERROR, "Invalid return value: %d\n", ret);
return AVERROR_EXTERNAL;
}
} else {
av_log(avctx, AV_LOG_ERROR, "Udefined encoder state\n");
return AVERROR_INVALIDDATA;
}
return 0;
}
/**
* Destroy the encoder and release all the allocated resources
*
* @param avctx codec context
* @return 0 on success, negative error code on failure
*/
static av_cold int libxeve_close(AVCodecContext *avctx)
{
XeveContext *xectx = avctx->priv_data;
if (xectx->id) {
xeve_delete(xectx->id);
xectx->id = NULL;
}
av_free(xectx->bitb.addr); /* release bitstream buffer */
return 0;
}
#define OFFSET(x) offsetof(XeveContext, x)
#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM
static const enum AVPixelFormat supported_pixel_formats[] = {
AV_PIX_FMT_YUV420P,
AV_PIX_FMT_YUV420P10,
AV_PIX_FMT_NONE
};
// Consider using following options (./ffmpeg --help encoder=libxeve)
//
static const AVOption libxeve_options[] = {
{ "preset", "Encoding preset for setting encoding speed", OFFSET(preset_id), AV_OPT_TYPE_INT, { .i64 = XEVE_PRESET_MEDIUM }, XEVE_PRESET_DEFAULT, XEVE_PRESET_PLACEBO, VE, .unit = "preset" },
{ "default", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PRESET_DEFAULT }, INT_MIN, INT_MAX, VE, .unit = "preset" },
{ "fast", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PRESET_FAST }, INT_MIN, INT_MAX, VE, .unit = "preset" },
{ "medium", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PRESET_MEDIUM }, INT_MIN, INT_MAX, VE, .unit = "preset" },
{ "slow", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PRESET_SLOW }, INT_MIN, INT_MAX, VE, .unit = "preset" },
{ "placebo", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PRESET_PLACEBO }, INT_MIN, INT_MAX, VE, .unit = "preset" },
{ "tune", "Tuning parameter for special purpose operation", OFFSET(tune_id), AV_OPT_TYPE_INT, { .i64 = XEVE_TUNE_NONE }, XEVE_TUNE_NONE, XEVE_TUNE_PSNR, VE, .unit = "tune"},
{ "none", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_TUNE_NONE }, INT_MIN, INT_MAX, VE, .unit = "tune" },
{ "zerolatency", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_TUNE_ZEROLATENCY }, INT_MIN, INT_MAX, VE, .unit = "tune" },
{ "psnr", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_TUNE_PSNR }, INT_MIN, INT_MAX, VE, .unit = "tune" },
{ "profile", "Encoding profile", OFFSET(profile_id), AV_OPT_TYPE_INT, { .i64 = XEVE_PROFILE_BASELINE }, XEVE_PROFILE_BASELINE, XEVE_PROFILE_MAIN, VE, .unit = "profile" },
{ "baseline", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PROFILE_BASELINE }, INT_MIN, INT_MAX, VE, .unit = "profile" },
{ "main", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_PROFILE_MAIN }, INT_MIN, INT_MAX, VE, .unit = "profile" },
{ "rc_mode", "Rate control mode", OFFSET(rc_mode), AV_OPT_TYPE_INT, { .i64 = XEVE_RC_CQP }, XEVE_RC_CQP, XEVE_RC_CRF, VE, .unit = "rc_mode" },
{ "CQP", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_RC_CQP }, INT_MIN, INT_MAX, VE, .unit = "rc_mode" },
{ "ABR", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_RC_ABR }, INT_MIN, INT_MAX, VE, .unit = "rc_mode" },
{ "CRF", NULL, 0, AV_OPT_TYPE_CONST, { .i64 = XEVE_RC_CRF }, INT_MIN, INT_MAX, VE, .unit = "rc_mode" },
{ "qp", "Quantization parameter value for CQP rate control mode", OFFSET(qp), AV_OPT_TYPE_INT, { .i64 = 32 }, 0, 51, VE },
{ "crf", "Constant rate factor value for CRF rate control mode", OFFSET(crf), AV_OPT_TYPE_INT, { .i64 = 32 }, 10, 49, VE },
{ "hash", "Embed picture signature (HASH) for conformance checking in decoding", OFFSET(hash), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
{ "sei_info", "Embed SEI messages identifying encoder parameters and command line arguments", OFFSET(sei_info), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, VE },
{ "xeve-params", "Override the xeve configuration using a :-separated list of key=value parameters", OFFSET(xeve_params), AV_OPT_TYPE_DICT, { 0 }, 0, 0, VE },
{ NULL }
};
static const AVClass libxeve_class = {
.class_name = "libxeve",
.item_name = av_default_item_name,
.option = libxeve_options,
.version = LIBAVUTIL_VERSION_INT,
};
/**
* libavcodec generic global options, which can be set on all the encoders and decoders
* @see https://www.ffmpeg.org/ffmpeg-codecs.html#Codec-Options
*/
static const FFCodecDefault libxeve_defaults[] = {
{ "b", "0" }, // bitrate in terms of kilo-bits per second
{ "g", "0" }, // gop_size (key-frame interval 0: only one I-frame at the first time; 1: every frame is coded in I-frame)
{ "bf", "15"}, // maximum number of B frames (0: no B-frames, 1,3,7,15)
{ "threads", "0"}, // number of threads to be used (0: automatically select the number of threads to set)
{ NULL },
};
const FFCodec ff_libxeve_encoder = {
.p.name = "libxeve",
.p.long_name = NULL_IF_CONFIG_SMALL("libxeve MPEG-5 EVC"),
.p.type = AVMEDIA_TYPE_VIDEO,
.p.id = AV_CODEC_ID_EVC,
.init = libxeve_init,
FF_CODEC_ENCODE_CB(libxeve_encode),
.close = libxeve_close,
.priv_data_size = sizeof(XeveContext),
.p.priv_class = &libxeve_class,
.defaults = libxeve_defaults,
.p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_OTHER_THREADS | AV_CODEC_CAP_DR1,
.p.profiles = NULL_IF_CONFIG_SMALL(ff_evc_profiles),
.p.wrapper_name = "libxeve",
.p.pix_fmts = supported_pixel_formats,
.color_ranges = AVCOL_RANGE_MPEG, /* FIXME: implement tagging */
.caps_internal = FF_CODEC_CAP_INIT_CLEANUP | FF_CODEC_CAP_NOT_INIT_THREADSAFE,
};