2017-05-04 22:01:51 +00:00
|
|
|
/*
|
|
|
|
* 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 <string.h>
|
|
|
|
|
|
|
|
#include "config.h"
|
|
|
|
|
|
|
|
#include "libavutil/avassert.h"
|
2018-02-11 00:51:33 +00:00
|
|
|
#include "libavutil/buffer.h"
|
2017-05-04 22:01:51 +00:00
|
|
|
#include "libavutil/common.h"
|
2020-11-15 21:55:38 +00:00
|
|
|
#include "libavutil/opt.h"
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2021-03-02 09:55:16 +00:00
|
|
|
#include "avcodec.h"
|
2017-05-04 22:01:51 +00:00
|
|
|
#include "cbs.h"
|
|
|
|
#include "cbs_internal.h"
|
|
|
|
|
|
|
|
|
2020-12-27 16:36:59 +00:00
|
|
|
static const CodedBitstreamType *const cbs_type_table[] = {
|
2018-09-09 22:08:11 +00:00
|
|
|
#if CONFIG_CBS_AV1
|
|
|
|
&ff_cbs_type_av1,
|
|
|
|
#endif
|
2017-05-14 15:18:25 +00:00
|
|
|
#if CONFIG_CBS_H264
|
|
|
|
&ff_cbs_type_h264,
|
|
|
|
#endif
|
2017-05-14 15:32:00 +00:00
|
|
|
#if CONFIG_CBS_H265
|
|
|
|
&ff_cbs_type_h265,
|
|
|
|
#endif
|
2023-03-21 15:01:15 +00:00
|
|
|
#if CONFIG_CBS_H266
|
|
|
|
&ff_cbs_type_h266,
|
|
|
|
#endif
|
2018-09-18 22:30:53 +00:00
|
|
|
#if CONFIG_CBS_JPEG
|
|
|
|
&ff_cbs_type_jpeg,
|
|
|
|
#endif
|
2017-05-04 22:03:03 +00:00
|
|
|
#if CONFIG_CBS_MPEG2
|
|
|
|
&ff_cbs_type_mpeg2,
|
|
|
|
#endif
|
2017-06-22 21:42:51 +00:00
|
|
|
#if CONFIG_CBS_VP9
|
|
|
|
&ff_cbs_type_vp9,
|
|
|
|
#endif
|
2017-05-04 22:01:51 +00:00
|
|
|
};
|
|
|
|
|
2018-03-11 17:27:15 +00:00
|
|
|
const enum AVCodecID ff_cbs_all_codec_ids[] = {
|
2018-09-09 22:08:11 +00:00
|
|
|
#if CONFIG_CBS_AV1
|
|
|
|
AV_CODEC_ID_AV1,
|
|
|
|
#endif
|
2018-03-11 17:27:15 +00:00
|
|
|
#if CONFIG_CBS_H264
|
|
|
|
AV_CODEC_ID_H264,
|
|
|
|
#endif
|
|
|
|
#if CONFIG_CBS_H265
|
|
|
|
AV_CODEC_ID_H265,
|
|
|
|
#endif
|
2023-03-21 15:01:15 +00:00
|
|
|
#if CONFIG_CBS_H266
|
|
|
|
AV_CODEC_ID_H266,
|
|
|
|
#endif
|
2018-09-18 22:30:53 +00:00
|
|
|
#if CONFIG_CBS_JPEG
|
|
|
|
AV_CODEC_ID_MJPEG,
|
|
|
|
#endif
|
2018-03-11 17:27:15 +00:00
|
|
|
#if CONFIG_CBS_MPEG2
|
|
|
|
AV_CODEC_ID_MPEG2VIDEO,
|
2017-06-22 21:42:51 +00:00
|
|
|
#endif
|
|
|
|
#if CONFIG_CBS_VP9
|
|
|
|
AV_CODEC_ID_VP9,
|
2018-03-11 17:27:15 +00:00
|
|
|
#endif
|
|
|
|
AV_CODEC_ID_NONE
|
|
|
|
};
|
|
|
|
|
2022-07-01 11:00:07 +00:00
|
|
|
av_cold int ff_cbs_init(CodedBitstreamContext **ctx_ptr,
|
|
|
|
enum AVCodecID codec_id, void *log_ctx)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
|
|
|
CodedBitstreamContext *ctx;
|
|
|
|
const CodedBitstreamType *type;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
type = NULL;
|
|
|
|
for (i = 0; i < FF_ARRAY_ELEMS(cbs_type_table); i++) {
|
|
|
|
if (cbs_type_table[i]->codec_id == codec_id) {
|
|
|
|
type = cbs_type_table[i];
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if (!type)
|
|
|
|
return AVERROR(EINVAL);
|
|
|
|
|
|
|
|
ctx = av_mallocz(sizeof(*ctx));
|
|
|
|
if (!ctx)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
|
|
|
ctx->log_ctx = log_ctx;
|
2021-03-09 19:16:04 +00:00
|
|
|
ctx->codec = type; /* Must be before any error */
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2019-11-17 07:34:35 +00:00
|
|
|
if (type->priv_data_size) {
|
|
|
|
ctx->priv_data = av_mallocz(ctx->codec->priv_data_size);
|
|
|
|
if (!ctx->priv_data) {
|
|
|
|
av_freep(&ctx);
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
}
|
2020-11-15 21:55:38 +00:00
|
|
|
if (type->priv_class) {
|
|
|
|
*(const AVClass **)ctx->priv_data = type->priv_class;
|
|
|
|
av_opt_set_defaults(ctx->priv_data);
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
ctx->decompose_unit_types = NULL;
|
|
|
|
|
|
|
|
ctx->trace_enable = 0;
|
|
|
|
ctx->trace_level = AV_LOG_TRACE;
|
|
|
|
|
|
|
|
*ctx_ptr = ctx;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-07-01 11:00:07 +00:00
|
|
|
av_cold void ff_cbs_flush(CodedBitstreamContext *ctx)
|
2020-09-24 20:57:42 +00:00
|
|
|
{
|
2021-03-09 19:16:04 +00:00
|
|
|
if (ctx->codec->flush)
|
2020-09-24 20:57:42 +00:00
|
|
|
ctx->codec->flush(ctx);
|
|
|
|
}
|
|
|
|
|
2022-07-01 11:00:07 +00:00
|
|
|
av_cold void ff_cbs_close(CodedBitstreamContext **ctx_ptr)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
|
|
|
CodedBitstreamContext *ctx = *ctx_ptr;
|
|
|
|
|
|
|
|
if (!ctx)
|
|
|
|
return;
|
|
|
|
|
2021-03-09 19:16:04 +00:00
|
|
|
if (ctx->codec->close)
|
2017-05-04 22:01:51 +00:00
|
|
|
ctx->codec->close(ctx);
|
|
|
|
|
2019-11-17 07:34:35 +00:00
|
|
|
av_freep(&ctx->write_buffer);
|
2020-11-15 21:55:38 +00:00
|
|
|
|
|
|
|
if (ctx->codec->priv_class && ctx->priv_data)
|
|
|
|
av_opt_free(ctx->priv_data);
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
av_freep(&ctx->priv_data);
|
|
|
|
av_freep(ctx_ptr);
|
|
|
|
}
|
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
static void cbs_unit_uninit(CodedBitstreamUnit *unit)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
2018-02-11 00:51:33 +00:00
|
|
|
av_buffer_unref(&unit->content_ref);
|
|
|
|
unit->content = NULL;
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
av_buffer_unref(&unit->data_ref);
|
|
|
|
unit->data = NULL;
|
|
|
|
unit->data_size = 0;
|
2017-05-04 22:01:51 +00:00
|
|
|
unit->data_bit_padding = 0;
|
|
|
|
}
|
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
void ff_cbs_fragment_reset(CodedBitstreamFragment *frag)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
|
|
|
|
for (i = 0; i < frag->nb_units; i++)
|
2020-07-05 22:34:25 +00:00
|
|
|
cbs_unit_uninit(&frag->units[i]);
|
2017-05-04 22:01:51 +00:00
|
|
|
frag->nb_units = 0;
|
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
av_buffer_unref(&frag->data_ref);
|
|
|
|
frag->data = NULL;
|
2017-05-04 22:01:51 +00:00
|
|
|
frag->data_size = 0;
|
|
|
|
frag->data_bit_padding = 0;
|
|
|
|
}
|
|
|
|
|
2022-07-01 11:00:07 +00:00
|
|
|
av_cold void ff_cbs_fragment_free(CodedBitstreamFragment *frag)
|
2019-02-11 22:47:43 +00:00
|
|
|
{
|
2020-07-05 22:34:25 +00:00
|
|
|
ff_cbs_fragment_reset(frag);
|
2019-02-11 22:47:43 +00:00
|
|
|
|
|
|
|
av_freep(&frag->units);
|
|
|
|
frag->nb_units_allocated = 0;
|
|
|
|
}
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
static int cbs_read_fragment_content(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag)
|
|
|
|
{
|
|
|
|
int err, i, j;
|
|
|
|
|
|
|
|
for (i = 0; i < frag->nb_units; i++) {
|
2018-04-30 23:18:16 +00:00
|
|
|
CodedBitstreamUnit *unit = &frag->units[i];
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
if (ctx->decompose_unit_types) {
|
|
|
|
for (j = 0; j < ctx->nb_decompose_unit_types; j++) {
|
2018-04-30 23:18:16 +00:00
|
|
|
if (ctx->decompose_unit_types[j] == unit->type)
|
2017-05-04 22:01:51 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
if (j >= ctx->nb_decompose_unit_types)
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2018-04-30 23:18:16 +00:00
|
|
|
av_buffer_unref(&unit->content_ref);
|
|
|
|
unit->content = NULL;
|
|
|
|
|
|
|
|
av_assert0(unit->data && unit->data_ref);
|
2018-02-11 00:51:33 +00:00
|
|
|
|
2018-04-30 23:18:16 +00:00
|
|
|
err = ctx->codec->read_unit(ctx, unit);
|
2017-05-04 22:01:51 +00:00
|
|
|
if (err == AVERROR(ENOSYS)) {
|
2017-12-11 00:28:40 +00:00
|
|
|
av_log(ctx->log_ctx, AV_LOG_VERBOSE,
|
2017-05-04 22:01:51 +00:00
|
|
|
"Decomposition unimplemented for unit %d "
|
2018-04-30 23:18:16 +00:00
|
|
|
"(type %"PRIu32").\n", i, unit->type);
|
2020-11-15 21:55:39 +00:00
|
|
|
} else if (err == AVERROR(EAGAIN)) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_VERBOSE,
|
|
|
|
"Skipping decomposition of unit %d "
|
|
|
|
"(type %"PRIu32").\n", i, unit->type);
|
|
|
|
av_buffer_unref(&unit->content_ref);
|
|
|
|
unit->content = NULL;
|
2017-05-04 22:01:51 +00:00
|
|
|
} else if (err < 0) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to read unit %d "
|
2018-04-30 23:18:16 +00:00
|
|
|
"(type %"PRIu32").\n", i, unit->type);
|
2017-05-04 22:01:51 +00:00
|
|
|
return err;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
static int cbs_fill_fragment_data(CodedBitstreamFragment *frag,
|
2018-02-11 00:51:33 +00:00
|
|
|
const uint8_t *data, size_t size)
|
|
|
|
{
|
|
|
|
av_assert0(!frag->data && !frag->data_ref);
|
|
|
|
|
|
|
|
frag->data_ref =
|
|
|
|
av_buffer_alloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
|
|
|
|
if (!frag->data_ref)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
|
|
|
frag->data = frag->data_ref->data;
|
|
|
|
frag->data_size = size;
|
|
|
|
|
|
|
|
memcpy(frag->data, data, size);
|
|
|
|
memset(frag->data + size, 0,
|
|
|
|
AV_INPUT_BUFFER_PADDING_SIZE);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
static int cbs_read_data(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
AVBufferRef *buf,
|
|
|
|
const uint8_t *data, size_t size,
|
|
|
|
int header)
|
2018-04-25 22:27:04 +00:00
|
|
|
{
|
|
|
|
int err;
|
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
if (buf) {
|
|
|
|
frag->data_ref = av_buffer_ref(buf);
|
2018-02-11 00:51:33 +00:00
|
|
|
if (!frag->data_ref)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
frag->data = (uint8_t *)data;
|
|
|
|
frag->data_size = size;
|
2018-02-11 00:51:33 +00:00
|
|
|
|
|
|
|
} else {
|
2020-11-23 17:53:57 +00:00
|
|
|
err = cbs_fill_fragment_data(frag, data, size);
|
2018-02-11 00:51:33 +00:00
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
err = ctx->codec->split_fragment(ctx, frag, header);
|
2017-05-04 22:01:51 +00:00
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
|
|
|
|
return cbs_read_fragment_content(ctx, frag);
|
|
|
|
}
|
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
int ff_cbs_read_extradata(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
const AVCodecParameters *par)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
2020-11-23 17:53:57 +00:00
|
|
|
return cbs_read_data(ctx, frag, NULL,
|
|
|
|
par->extradata,
|
|
|
|
par->extradata_size, 1);
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
int ff_cbs_read_extradata_from_codec(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
const AVCodecContext *avctx)
|
|
|
|
{
|
|
|
|
return cbs_read_data(ctx, frag, NULL,
|
|
|
|
avctx->extradata,
|
|
|
|
avctx->extradata_size, 1);
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
int ff_cbs_read_packet(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
const AVPacket *pkt)
|
|
|
|
{
|
|
|
|
return cbs_read_data(ctx, frag, pkt->buf,
|
|
|
|
pkt->data, pkt->size, 0);
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2021-08-05 14:59:55 +00:00
|
|
|
int ff_cbs_read_packet_side_data(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
const AVPacket *pkt)
|
|
|
|
{
|
|
|
|
size_t side_data_size;
|
|
|
|
const uint8_t *side_data =
|
|
|
|
av_packet_get_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA,
|
|
|
|
&side_data_size);
|
|
|
|
|
|
|
|
return cbs_read_data(ctx, frag, NULL,
|
|
|
|
side_data, side_data_size, 1);
|
|
|
|
}
|
|
|
|
|
2020-11-23 17:53:57 +00:00
|
|
|
int ff_cbs_read(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
const uint8_t *data, size_t size)
|
|
|
|
{
|
|
|
|
return cbs_read_data(ctx, frag, NULL,
|
|
|
|
data, size, 0);
|
2017-05-04 22:01:51 +00:00
|
|
|
}
|
|
|
|
|
2021-09-06 10:37:26 +00:00
|
|
|
/**
|
|
|
|
* Allocate a new internal data buffer of the given size in the unit.
|
|
|
|
*
|
|
|
|
* The data buffer will have input padding.
|
|
|
|
*/
|
|
|
|
static int cbs_alloc_unit_data(CodedBitstreamUnit *unit,
|
|
|
|
size_t size)
|
|
|
|
{
|
|
|
|
av_assert0(!unit->data && !unit->data_ref);
|
|
|
|
|
|
|
|
unit->data_ref = av_buffer_alloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
|
|
|
|
if (!unit->data_ref)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
|
|
|
unit->data = unit->data_ref->data;
|
|
|
|
unit->data_size = size;
|
|
|
|
|
|
|
|
memset(unit->data + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2019-11-17 07:34:35 +00:00
|
|
|
static int cbs_write_unit_data(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamUnit *unit)
|
|
|
|
{
|
|
|
|
PutBitContext pbc;
|
|
|
|
int ret;
|
|
|
|
|
|
|
|
if (!ctx->write_buffer) {
|
|
|
|
// Initial write buffer size is 1MB.
|
|
|
|
ctx->write_buffer_size = 1024 * 1024;
|
|
|
|
|
|
|
|
reallocate_and_try_again:
|
|
|
|
ret = av_reallocp(&ctx->write_buffer, ctx->write_buffer_size);
|
|
|
|
if (ret < 0) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "Unable to allocate a "
|
|
|
|
"sufficiently large write buffer (last attempt "
|
|
|
|
"%"SIZE_SPECIFIER" bytes).\n", ctx->write_buffer_size);
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
init_put_bits(&pbc, ctx->write_buffer, ctx->write_buffer_size);
|
|
|
|
|
|
|
|
ret = ctx->codec->write_unit(ctx, unit, &pbc);
|
|
|
|
if (ret < 0) {
|
|
|
|
if (ret == AVERROR(ENOSPC)) {
|
|
|
|
// Overflow.
|
2019-11-17 07:34:36 +00:00
|
|
|
if (ctx->write_buffer_size == INT_MAX / 8)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
ctx->write_buffer_size = FFMIN(2 * ctx->write_buffer_size, INT_MAX / 8);
|
2019-11-17 07:34:35 +00:00
|
|
|
goto reallocate_and_try_again;
|
|
|
|
}
|
|
|
|
// Write failed for some other reason.
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Overflow but we didn't notice.
|
|
|
|
av_assert0(put_bits_count(&pbc) <= 8 * ctx->write_buffer_size);
|
|
|
|
|
|
|
|
if (put_bits_count(&pbc) % 8)
|
|
|
|
unit->data_bit_padding = 8 - put_bits_count(&pbc) % 8;
|
|
|
|
else
|
|
|
|
unit->data_bit_padding = 0;
|
|
|
|
|
|
|
|
flush_put_bits(&pbc);
|
|
|
|
|
2021-09-06 10:37:26 +00:00
|
|
|
ret = cbs_alloc_unit_data(unit, put_bytes_output(&pbc));
|
2019-11-17 07:34:35 +00:00
|
|
|
if (ret < 0)
|
|
|
|
return ret;
|
|
|
|
|
|
|
|
memcpy(unit->data, ctx->write_buffer, unit->data_size);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
int ff_cbs_write_fragment_data(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag)
|
|
|
|
{
|
|
|
|
int err, i;
|
|
|
|
|
|
|
|
for (i = 0; i < frag->nb_units; i++) {
|
2018-02-11 00:51:33 +00:00
|
|
|
CodedBitstreamUnit *unit = &frag->units[i];
|
|
|
|
|
|
|
|
if (!unit->content)
|
2017-05-04 22:01:51 +00:00
|
|
|
continue;
|
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
av_buffer_unref(&unit->data_ref);
|
|
|
|
unit->data = NULL;
|
|
|
|
|
2019-11-17 07:34:35 +00:00
|
|
|
err = cbs_write_unit_data(ctx, unit);
|
2017-05-04 22:01:51 +00:00
|
|
|
if (err < 0) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to write unit %d "
|
2018-02-11 00:51:33 +00:00
|
|
|
"(type %"PRIu32").\n", i, unit->type);
|
2017-05-04 22:01:51 +00:00
|
|
|
return err;
|
|
|
|
}
|
2018-04-30 23:18:16 +00:00
|
|
|
av_assert0(unit->data && unit->data_ref);
|
2017-05-04 22:01:51 +00:00
|
|
|
}
|
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
av_buffer_unref(&frag->data_ref);
|
|
|
|
frag->data = NULL;
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
err = ctx->codec->assemble_fragment(ctx, frag);
|
|
|
|
if (err < 0) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "Failed to assemble fragment.\n");
|
|
|
|
return err;
|
|
|
|
}
|
2018-04-30 23:18:16 +00:00
|
|
|
av_assert0(frag->data && frag->data_ref);
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int ff_cbs_write_extradata(CodedBitstreamContext *ctx,
|
|
|
|
AVCodecParameters *par,
|
|
|
|
CodedBitstreamFragment *frag)
|
|
|
|
{
|
|
|
|
int err;
|
|
|
|
|
|
|
|
err = ff_cbs_write_fragment_data(ctx, frag);
|
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
|
|
|
|
av_freep(&par->extradata);
|
2022-09-20 12:37:38 +00:00
|
|
|
par->extradata_size = 0;
|
|
|
|
|
|
|
|
if (!frag->data_size)
|
|
|
|
return 0;
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
par->extradata = av_malloc(frag->data_size +
|
|
|
|
AV_INPUT_BUFFER_PADDING_SIZE);
|
|
|
|
if (!par->extradata)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
|
|
|
memcpy(par->extradata, frag->data, frag->data_size);
|
|
|
|
memset(par->extradata + frag->data_size, 0,
|
|
|
|
AV_INPUT_BUFFER_PADDING_SIZE);
|
|
|
|
par->extradata_size = frag->data_size;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int ff_cbs_write_packet(CodedBitstreamContext *ctx,
|
|
|
|
AVPacket *pkt,
|
|
|
|
CodedBitstreamFragment *frag)
|
|
|
|
{
|
2018-03-05 14:04:59 +00:00
|
|
|
AVBufferRef *buf;
|
2017-05-04 22:01:51 +00:00
|
|
|
int err;
|
|
|
|
|
|
|
|
err = ff_cbs_write_fragment_data(ctx, frag);
|
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
|
2018-03-05 14:04:59 +00:00
|
|
|
buf = av_buffer_ref(frag->data_ref);
|
|
|
|
if (!buf)
|
|
|
|
return AVERROR(ENOMEM);
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2019-06-17 03:42:06 +00:00
|
|
|
av_buffer_unref(&pkt->buf);
|
|
|
|
|
2018-03-05 14:04:59 +00:00
|
|
|
pkt->buf = buf;
|
|
|
|
pkt->data = frag->data;
|
2017-05-04 22:01:51 +00:00
|
|
|
pkt->size = frag->data_size;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void ff_cbs_trace_header(CodedBitstreamContext *ctx,
|
|
|
|
const char *name)
|
|
|
|
{
|
|
|
|
if (!ctx->trace_enable)
|
|
|
|
return;
|
|
|
|
|
|
|
|
av_log(ctx->log_ctx, ctx->trace_level, "%s\n", name);
|
|
|
|
}
|
|
|
|
|
|
|
|
void ff_cbs_trace_syntax_element(CodedBitstreamContext *ctx, int position,
|
2018-04-30 21:35:30 +00:00
|
|
|
const char *str, const int *subscripts,
|
|
|
|
const char *bits, int64_t value)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
2018-04-30 21:35:30 +00:00
|
|
|
char name[256];
|
2017-05-04 22:01:51 +00:00
|
|
|
size_t name_len, bits_len;
|
2018-04-30 21:35:30 +00:00
|
|
|
int pad, subs, i, j, k, n;
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
if (!ctx->trace_enable)
|
|
|
|
return;
|
|
|
|
|
|
|
|
av_assert0(value >= INT_MIN && value <= UINT32_MAX);
|
|
|
|
|
2018-04-30 21:35:30 +00:00
|
|
|
subs = subscripts ? subscripts[0] : 0;
|
|
|
|
n = 0;
|
|
|
|
for (i = j = 0; str[i];) {
|
|
|
|
if (str[i] == '[') {
|
|
|
|
if (n < subs) {
|
|
|
|
++n;
|
|
|
|
k = snprintf(name + j, sizeof(name) - j, "[%d", subscripts[n]);
|
|
|
|
av_assert0(k > 0 && j + k < sizeof(name));
|
|
|
|
j += k;
|
|
|
|
for (++i; str[i] && str[i] != ']'; i++);
|
|
|
|
av_assert0(str[i] == ']');
|
|
|
|
} else {
|
|
|
|
while (str[i] && str[i] != ']')
|
|
|
|
name[j++] = str[i++];
|
|
|
|
av_assert0(str[i] == ']');
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
av_assert0(j + 1 < sizeof(name));
|
|
|
|
name[j++] = str[i++];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
av_assert0(j + 1 < sizeof(name));
|
|
|
|
name[j] = 0;
|
|
|
|
av_assert0(n == subs);
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
name_len = strlen(name);
|
|
|
|
bits_len = strlen(bits);
|
|
|
|
|
|
|
|
if (name_len + bits_len > 60)
|
|
|
|
pad = bits_len + 2;
|
|
|
|
else
|
|
|
|
pad = 61 - name_len;
|
|
|
|
|
|
|
|
av_log(ctx->log_ctx, ctx->trace_level, "%-10d %s%*s = %"PRId64"\n",
|
|
|
|
position, name, pad, bits, value);
|
|
|
|
}
|
|
|
|
|
|
|
|
int ff_cbs_read_unsigned(CodedBitstreamContext *ctx, GetBitContext *gbc,
|
2018-04-30 21:35:30 +00:00
|
|
|
int width, const char *name,
|
|
|
|
const int *subscripts, uint32_t *write_to,
|
2017-05-04 22:01:51 +00:00
|
|
|
uint32_t range_min, uint32_t range_max)
|
|
|
|
{
|
|
|
|
uint32_t value;
|
|
|
|
int position;
|
|
|
|
|
|
|
|
av_assert0(width > 0 && width <= 32);
|
|
|
|
|
|
|
|
if (get_bits_left(gbc) < width) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid value at "
|
|
|
|
"%s: bitstream ended.\n", name);
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ctx->trace_enable)
|
|
|
|
position = get_bits_count(gbc);
|
|
|
|
|
|
|
|
value = get_bits_long(gbc, width);
|
|
|
|
|
|
|
|
if (ctx->trace_enable) {
|
|
|
|
char bits[33];
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < width; i++)
|
|
|
|
bits[i] = value >> (width - i - 1) & 1 ? '1' : '0';
|
|
|
|
bits[i] = 0;
|
|
|
|
|
2018-04-30 21:35:30 +00:00
|
|
|
ff_cbs_trace_syntax_element(ctx, position, name, subscripts,
|
|
|
|
bits, value);
|
2017-05-04 22:01:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (value < range_min || value > range_max) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
|
|
|
|
"%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
|
|
|
|
name, value, range_min, range_max);
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
|
|
|
*write_to = value;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int ff_cbs_write_unsigned(CodedBitstreamContext *ctx, PutBitContext *pbc,
|
2018-04-30 21:35:30 +00:00
|
|
|
int width, const char *name,
|
|
|
|
const int *subscripts, uint32_t value,
|
2017-05-04 22:01:51 +00:00
|
|
|
uint32_t range_min, uint32_t range_max)
|
|
|
|
{
|
|
|
|
av_assert0(width > 0 && width <= 32);
|
|
|
|
|
|
|
|
if (value < range_min || value > range_max) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
|
|
|
|
"%"PRIu32", but must be in [%"PRIu32",%"PRIu32"].\n",
|
|
|
|
name, value, range_min, range_max);
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (put_bits_left(pbc) < width)
|
|
|
|
return AVERROR(ENOSPC);
|
|
|
|
|
|
|
|
if (ctx->trace_enable) {
|
|
|
|
char bits[33];
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < width; i++)
|
|
|
|
bits[i] = value >> (width - i - 1) & 1 ? '1' : '0';
|
|
|
|
bits[i] = 0;
|
|
|
|
|
2018-04-30 21:35:30 +00:00
|
|
|
ff_cbs_trace_syntax_element(ctx, put_bits_count(pbc),
|
|
|
|
name, subscripts, bits, value);
|
2017-05-04 22:01:51 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (width < 32)
|
|
|
|
put_bits(pbc, width, value);
|
|
|
|
else
|
|
|
|
put_bits32(pbc, value);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2019-04-15 20:46:53 +00:00
|
|
|
int ff_cbs_read_signed(CodedBitstreamContext *ctx, GetBitContext *gbc,
|
|
|
|
int width, const char *name,
|
|
|
|
const int *subscripts, int32_t *write_to,
|
|
|
|
int32_t range_min, int32_t range_max)
|
|
|
|
{
|
|
|
|
int32_t value;
|
|
|
|
int position;
|
|
|
|
|
|
|
|
av_assert0(width > 0 && width <= 32);
|
|
|
|
|
|
|
|
if (get_bits_left(gbc) < width) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid value at "
|
|
|
|
"%s: bitstream ended.\n", name);
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ctx->trace_enable)
|
|
|
|
position = get_bits_count(gbc);
|
|
|
|
|
|
|
|
value = get_sbits_long(gbc, width);
|
|
|
|
|
|
|
|
if (ctx->trace_enable) {
|
|
|
|
char bits[33];
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < width; i++)
|
|
|
|
bits[i] = value & (1U << (width - i - 1)) ? '1' : '0';
|
|
|
|
bits[i] = 0;
|
|
|
|
|
|
|
|
ff_cbs_trace_syntax_element(ctx, position, name, subscripts,
|
|
|
|
bits, value);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (value < range_min || value > range_max) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
|
|
|
|
"%"PRId32", but must be in [%"PRId32",%"PRId32"].\n",
|
|
|
|
name, value, range_min, range_max);
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
|
|
|
*write_to = value;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
int ff_cbs_write_signed(CodedBitstreamContext *ctx, PutBitContext *pbc,
|
|
|
|
int width, const char *name,
|
|
|
|
const int *subscripts, int32_t value,
|
|
|
|
int32_t range_min, int32_t range_max)
|
|
|
|
{
|
|
|
|
av_assert0(width > 0 && width <= 32);
|
|
|
|
|
|
|
|
if (value < range_min || value > range_max) {
|
|
|
|
av_log(ctx->log_ctx, AV_LOG_ERROR, "%s out of range: "
|
|
|
|
"%"PRId32", but must be in [%"PRId32",%"PRId32"].\n",
|
|
|
|
name, value, range_min, range_max);
|
|
|
|
return AVERROR_INVALIDDATA;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (put_bits_left(pbc) < width)
|
|
|
|
return AVERROR(ENOSPC);
|
|
|
|
|
|
|
|
if (ctx->trace_enable) {
|
|
|
|
char bits[33];
|
|
|
|
int i;
|
|
|
|
for (i = 0; i < width; i++)
|
|
|
|
bits[i] = value & (1U << (width - i - 1)) ? '1' : '0';
|
|
|
|
bits[i] = 0;
|
|
|
|
|
|
|
|
ff_cbs_trace_syntax_element(ctx, put_bits_count(pbc),
|
|
|
|
name, subscripts, bits, value);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (width < 32)
|
|
|
|
put_sbits(pbc, width, value);
|
|
|
|
else
|
|
|
|
put_bits32(pbc, value);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
static int cbs_insert_unit(CodedBitstreamFragment *frag,
|
2017-05-04 22:01:51 +00:00
|
|
|
int position)
|
|
|
|
{
|
|
|
|
CodedBitstreamUnit *units;
|
|
|
|
|
2019-02-11 22:47:43 +00:00
|
|
|
if (frag->nb_units < frag->nb_units_allocated) {
|
|
|
|
units = frag->units;
|
|
|
|
|
|
|
|
if (position < frag->nb_units)
|
|
|
|
memmove(units + position + 1, units + position,
|
|
|
|
(frag->nb_units - position) * sizeof(*units));
|
|
|
|
} else {
|
2020-04-10 20:05:07 +00:00
|
|
|
units = av_malloc_array(frag->nb_units*2 + 1, sizeof(*units));
|
2019-02-11 22:47:43 +00:00
|
|
|
if (!units)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
2020-04-10 20:05:07 +00:00
|
|
|
frag->nb_units_allocated = 2*frag->nb_units_allocated + 1;
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2019-02-11 22:47:43 +00:00
|
|
|
if (position > 0)
|
|
|
|
memcpy(units, frag->units, position * sizeof(*units));
|
|
|
|
|
|
|
|
if (position < frag->nb_units)
|
|
|
|
memcpy(units + position + 1, frag->units + position,
|
|
|
|
(frag->nb_units - position) * sizeof(*units));
|
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
memset(units + position, 0, sizeof(*units));
|
|
|
|
|
2019-02-11 22:47:43 +00:00
|
|
|
if (units != frag->units) {
|
|
|
|
av_free(frag->units);
|
|
|
|
frag->units = units;
|
|
|
|
}
|
|
|
|
|
2017-05-04 22:01:51 +00:00
|
|
|
++frag->nb_units;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
int ff_cbs_insert_unit_content(CodedBitstreamFragment *frag,
|
2017-05-04 22:01:51 +00:00
|
|
|
int position,
|
|
|
|
CodedBitstreamUnitType type,
|
2018-02-11 00:51:33 +00:00
|
|
|
void *content,
|
|
|
|
AVBufferRef *content_buf)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
2018-02-11 00:51:33 +00:00
|
|
|
CodedBitstreamUnit *unit;
|
|
|
|
AVBufferRef *content_ref;
|
2017-05-04 22:01:51 +00:00
|
|
|
int err;
|
|
|
|
|
|
|
|
if (position == -1)
|
|
|
|
position = frag->nb_units;
|
|
|
|
av_assert0(position >= 0 && position <= frag->nb_units);
|
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
if (content_buf) {
|
|
|
|
content_ref = av_buffer_ref(content_buf);
|
|
|
|
if (!content_ref)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
} else {
|
|
|
|
content_ref = NULL;
|
|
|
|
}
|
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
err = cbs_insert_unit(frag, position);
|
2018-02-11 00:51:33 +00:00
|
|
|
if (err < 0) {
|
|
|
|
av_buffer_unref(&content_ref);
|
2017-05-04 22:01:51 +00:00
|
|
|
return err;
|
2018-02-11 00:51:33 +00:00
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
unit = &frag->units[position];
|
|
|
|
unit->type = type;
|
|
|
|
unit->content = content;
|
|
|
|
unit->content_ref = content_ref;
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-02-04 12:58:36 +00:00
|
|
|
static int cbs_insert_unit_data(CodedBitstreamFragment *frag,
|
|
|
|
CodedBitstreamUnitType type,
|
|
|
|
uint8_t *data, size_t data_size,
|
|
|
|
AVBufferRef *data_buf,
|
|
|
|
int position)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
2018-02-11 00:51:33 +00:00
|
|
|
CodedBitstreamUnit *unit;
|
|
|
|
AVBufferRef *data_ref;
|
2017-05-04 22:01:51 +00:00
|
|
|
int err;
|
|
|
|
|
|
|
|
av_assert0(position >= 0 && position <= frag->nb_units);
|
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
if (data_buf)
|
|
|
|
data_ref = av_buffer_ref(data_buf);
|
|
|
|
else
|
|
|
|
data_ref = av_buffer_create(data, data_size, NULL, NULL, 0);
|
2019-11-18 07:47:58 +00:00
|
|
|
if (!data_ref) {
|
|
|
|
if (!data_buf)
|
|
|
|
av_free(data);
|
2018-02-11 00:51:33 +00:00
|
|
|
return AVERROR(ENOMEM);
|
2019-11-18 07:47:58 +00:00
|
|
|
}
|
2018-02-11 00:51:33 +00:00
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
err = cbs_insert_unit(frag, position);
|
2018-02-11 00:51:33 +00:00
|
|
|
if (err < 0) {
|
|
|
|
av_buffer_unref(&data_ref);
|
2017-05-04 22:01:51 +00:00
|
|
|
return err;
|
2018-02-11 00:51:33 +00:00
|
|
|
}
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2018-02-11 00:51:33 +00:00
|
|
|
unit = &frag->units[position];
|
|
|
|
unit->type = type;
|
|
|
|
unit->data = data;
|
|
|
|
unit->data_size = data_size;
|
|
|
|
unit->data_ref = data_ref;
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-02-04 12:58:36 +00:00
|
|
|
int ff_cbs_append_unit_data(CodedBitstreamFragment *frag,
|
|
|
|
CodedBitstreamUnitType type,
|
|
|
|
uint8_t *data, size_t data_size,
|
|
|
|
AVBufferRef *data_buf)
|
|
|
|
{
|
|
|
|
return cbs_insert_unit_data(frag, type,
|
|
|
|
data, data_size, data_buf,
|
|
|
|
frag->nb_units);
|
|
|
|
}
|
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
void ff_cbs_delete_unit(CodedBitstreamFragment *frag,
|
2019-07-07 23:14:01 +00:00
|
|
|
int position)
|
2017-05-04 22:01:51 +00:00
|
|
|
{
|
2019-07-07 23:14:01 +00:00
|
|
|
av_assert0(0 <= position && position < frag->nb_units
|
|
|
|
&& "Unit to be deleted not in fragment.");
|
2017-05-04 22:01:51 +00:00
|
|
|
|
2020-07-05 22:34:25 +00:00
|
|
|
cbs_unit_uninit(&frag->units[position]);
|
2017-05-04 22:01:51 +00:00
|
|
|
|
|
|
|
--frag->nb_units;
|
|
|
|
|
2019-02-11 22:47:43 +00:00
|
|
|
if (frag->nb_units > 0)
|
2017-05-04 22:01:51 +00:00
|
|
|
memmove(frag->units + position,
|
|
|
|
frag->units + position + 1,
|
|
|
|
(frag->nb_units - position) * sizeof(*frag->units));
|
|
|
|
}
|
2020-07-27 16:32:18 +00:00
|
|
|
|
|
|
|
static void cbs_default_free_unit_content(void *opaque, uint8_t *data)
|
|
|
|
{
|
|
|
|
const CodedBitstreamUnitTypeDescriptor *desc = opaque;
|
2022-10-19 14:35:47 +00:00
|
|
|
|
|
|
|
for (int i = 0; i < desc->type.ref.nb_offsets; i++) {
|
|
|
|
void **ptr = (void**)(data + desc->type.ref.offsets[i]);
|
|
|
|
av_buffer_unref((AVBufferRef**)(ptr + 1));
|
2020-07-27 16:32:18 +00:00
|
|
|
}
|
|
|
|
av_free(data);
|
|
|
|
}
|
|
|
|
|
|
|
|
static const CodedBitstreamUnitTypeDescriptor
|
|
|
|
*cbs_find_unit_type_desc(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamUnit *unit)
|
|
|
|
{
|
|
|
|
const CodedBitstreamUnitTypeDescriptor *desc;
|
|
|
|
int i, j;
|
|
|
|
|
|
|
|
if (!ctx->codec->unit_types)
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
for (i = 0;; i++) {
|
|
|
|
desc = &ctx->codec->unit_types[i];
|
|
|
|
if (desc->nb_unit_types == 0)
|
|
|
|
break;
|
|
|
|
if (desc->nb_unit_types == CBS_UNIT_TYPE_RANGE) {
|
2022-08-04 09:02:14 +00:00
|
|
|
if (unit->type >= desc->unit_type.range.start &&
|
|
|
|
unit->type <= desc->unit_type.range.end)
|
2020-07-27 16:32:18 +00:00
|
|
|
return desc;
|
|
|
|
} else {
|
|
|
|
for (j = 0; j < desc->nb_unit_types; j++) {
|
2022-08-04 09:02:14 +00:00
|
|
|
if (desc->unit_type.list[j] == unit->type)
|
2020-07-27 16:32:18 +00:00
|
|
|
return desc;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2022-08-04 09:31:18 +00:00
|
|
|
int ff_cbs_alloc_unit_content(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamUnit *unit)
|
2020-07-27 16:32:18 +00:00
|
|
|
{
|
|
|
|
const CodedBitstreamUnitTypeDescriptor *desc;
|
|
|
|
|
|
|
|
av_assert0(!unit->content && !unit->content_ref);
|
|
|
|
|
|
|
|
desc = cbs_find_unit_type_desc(ctx, unit);
|
|
|
|
if (!desc)
|
|
|
|
return AVERROR(ENOSYS);
|
|
|
|
|
|
|
|
unit->content = av_mallocz(desc->content_size);
|
|
|
|
if (!unit->content)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
|
|
|
unit->content_ref =
|
|
|
|
av_buffer_create(unit->content, desc->content_size,
|
2022-08-04 09:02:14 +00:00
|
|
|
desc->content_type == CBS_CONTENT_TYPE_COMPLEX
|
|
|
|
? desc->type.complex.content_free
|
2020-07-27 16:32:18 +00:00
|
|
|
: cbs_default_free_unit_content,
|
|
|
|
(void*)desc, 0);
|
|
|
|
if (!unit->content_ref) {
|
|
|
|
av_freep(&unit->content);
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
2020-07-27 16:32:21 +00:00
|
|
|
|
2022-08-04 04:52:55 +00:00
|
|
|
static int cbs_clone_internal_refs_unit_content(AVBufferRef **clone_ref,
|
|
|
|
const CodedBitstreamUnit *unit,
|
|
|
|
const CodedBitstreamUnitTypeDescriptor *desc)
|
2020-07-27 16:32:21 +00:00
|
|
|
{
|
2022-08-04 05:19:29 +00:00
|
|
|
const uint8_t *src;
|
|
|
|
uint8_t *copy;
|
2020-07-27 16:32:21 +00:00
|
|
|
int err, i;
|
|
|
|
|
|
|
|
av_assert0(unit->content);
|
|
|
|
src = unit->content;
|
|
|
|
|
|
|
|
copy = av_memdup(src, desc->content_size);
|
|
|
|
if (!copy)
|
|
|
|
return AVERROR(ENOMEM);
|
|
|
|
|
2022-08-04 09:02:14 +00:00
|
|
|
for (i = 0; i < desc->type.ref.nb_offsets; i++) {
|
|
|
|
const uint8_t *const *src_ptr = (const uint8_t* const*)(src + desc->type.ref.offsets[i]);
|
2022-08-04 05:19:29 +00:00
|
|
|
const AVBufferRef *src_buf = *(AVBufferRef**)(src_ptr + 1);
|
2022-08-04 09:02:14 +00:00
|
|
|
uint8_t **copy_ptr = (uint8_t**)(copy + desc->type.ref.offsets[i]);
|
2022-08-04 05:19:29 +00:00
|
|
|
AVBufferRef **copy_buf = (AVBufferRef**)(copy_ptr + 1);
|
2020-07-27 16:32:21 +00:00
|
|
|
|
|
|
|
if (!*src_ptr) {
|
2022-08-04 05:19:29 +00:00
|
|
|
av_assert0(!src_buf);
|
2020-07-27 16:32:21 +00:00
|
|
|
continue;
|
|
|
|
}
|
2022-08-04 05:19:29 +00:00
|
|
|
if (!src_buf) {
|
2020-07-27 16:32:21 +00:00
|
|
|
// We can't handle a non-refcounted pointer here - we don't
|
|
|
|
// have enough information to handle whatever structure lies
|
|
|
|
// at the other end of it.
|
|
|
|
err = AVERROR(EINVAL);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
2022-08-04 05:19:29 +00:00
|
|
|
*copy_buf = av_buffer_ref(src_buf);
|
2020-07-27 16:32:21 +00:00
|
|
|
if (!*copy_buf) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
*clone_ref = av_buffer_create(copy, desc->content_size,
|
|
|
|
cbs_default_free_unit_content,
|
|
|
|
(void*)desc, 0);
|
|
|
|
if (!*clone_ref) {
|
|
|
|
err = AVERROR(ENOMEM);
|
|
|
|
goto fail;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
fail:
|
|
|
|
for (--i; i >= 0; i--)
|
2022-08-04 09:02:14 +00:00
|
|
|
av_buffer_unref((AVBufferRef**)(copy + desc->type.ref.offsets[i]));
|
2020-07-27 16:32:21 +00:00
|
|
|
av_freep(©);
|
|
|
|
*clone_ref = NULL;
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
2022-08-04 04:52:55 +00:00
|
|
|
/*
|
|
|
|
* On success, unit->content and unit->content_ref are updated with
|
|
|
|
* the new content; unit is untouched on failure.
|
|
|
|
* Any old content_ref is simply overwritten and not freed.
|
|
|
|
*/
|
|
|
|
static int cbs_clone_unit_content(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamUnit *unit)
|
2020-07-27 16:32:21 +00:00
|
|
|
{
|
|
|
|
const CodedBitstreamUnitTypeDescriptor *desc;
|
|
|
|
AVBufferRef *ref;
|
|
|
|
int err;
|
|
|
|
|
|
|
|
desc = cbs_find_unit_type_desc(ctx, unit);
|
|
|
|
if (!desc)
|
|
|
|
return AVERROR(ENOSYS);
|
|
|
|
|
|
|
|
switch (desc->content_type) {
|
|
|
|
case CBS_CONTENT_TYPE_INTERNAL_REFS:
|
2022-08-04 04:52:55 +00:00
|
|
|
err = cbs_clone_internal_refs_unit_content(&ref, unit, desc);
|
2020-07-27 16:32:21 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
case CBS_CONTENT_TYPE_COMPLEX:
|
2022-08-04 09:02:14 +00:00
|
|
|
if (!desc->type.complex.content_clone)
|
2020-07-27 16:32:21 +00:00
|
|
|
return AVERROR_PATCHWELCOME;
|
2022-08-04 09:02:14 +00:00
|
|
|
err = desc->type.complex.content_clone(&ref, unit);
|
2020-07-27 16:32:21 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
av_assert0(0 && "Invalid content type.");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (err < 0)
|
|
|
|
return err;
|
|
|
|
|
|
|
|
unit->content_ref = ref;
|
|
|
|
unit->content = ref->data;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2022-08-04 04:52:55 +00:00
|
|
|
int ff_cbs_make_unit_refcounted(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamUnit *unit)
|
|
|
|
{
|
|
|
|
av_assert0(unit->content);
|
|
|
|
if (unit->content_ref)
|
|
|
|
return 0;
|
|
|
|
return cbs_clone_unit_content(ctx, unit);
|
|
|
|
}
|
|
|
|
|
2020-07-27 16:32:21 +00:00
|
|
|
int ff_cbs_make_unit_writable(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamUnit *unit)
|
|
|
|
{
|
2022-08-04 04:52:55 +00:00
|
|
|
AVBufferRef *ref = unit->content_ref;
|
2020-07-27 16:32:21 +00:00
|
|
|
int err;
|
|
|
|
|
2022-08-04 04:52:55 +00:00
|
|
|
av_assert0(unit->content);
|
|
|
|
if (ref && av_buffer_is_writable(ref))
|
2020-07-27 16:32:21 +00:00
|
|
|
return 0;
|
|
|
|
|
2022-08-04 04:52:55 +00:00
|
|
|
err = cbs_clone_unit_content(ctx, unit);
|
2020-07-27 16:32:21 +00:00
|
|
|
if (err < 0)
|
|
|
|
return err;
|
2022-08-04 04:52:55 +00:00
|
|
|
av_buffer_unref(&ref);
|
2020-07-27 16:32:21 +00:00
|
|
|
return 0;
|
|
|
|
}
|
2023-05-17 21:23:38 +00:00
|
|
|
|
|
|
|
void ff_cbs_discard_units(CodedBitstreamContext *ctx,
|
|
|
|
CodedBitstreamFragment *frag,
|
|
|
|
enum AVDiscard skip,
|
|
|
|
int flags)
|
|
|
|
{
|
|
|
|
if (!ctx->codec->discarded_unit)
|
|
|
|
return;
|
|
|
|
|
|
|
|
for (int i = frag->nb_units - 1; i >= 0; i--) {
|
|
|
|
if (ctx->codec->discarded_unit(ctx, &frag->units[i], skip)) {
|
|
|
|
// discard all units
|
|
|
|
if (!(flags & DISCARD_FLAG_KEEP_NON_VCL)) {
|
|
|
|
ff_cbs_fragment_free(frag);
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ff_cbs_delete_unit(frag, i);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|