#include "libavutil/opt.h"
#include "bsf.h"
+#include "bsf_internal.h"
#include "cbs.h"
#include "cbs_vp9.h"
int color_space;
int color_range;
- int color_range_rgb_warned;
+ int color_warnings;
} VP9MetadataContext;
-static int vp9_metadata_filter(AVBSFContext *bsf, AVPacket *out)
+static int vp9_metadata_filter(AVBSFContext *bsf, AVPacket *pkt)
{
VP9MetadataContext *ctx = bsf->priv_data;
- AVPacket *in = NULL;
CodedBitstreamFragment *frag = &ctx->fragment;
int err, i;
- err = ff_bsf_get_packet(bsf, &in);
+ err = ff_bsf_get_packet_ref(bsf, pkt);
if (err < 0)
return err;
- err = ff_cbs_read_packet(ctx->cbc, frag, in);
+ err = ff_cbs_read_packet(ctx->cbc, frag, pkt);
if (err < 0) {
av_log(bsf, AV_LOG_ERROR, "Failed to read packet.\n");
goto fail;
for (i = 0; i < frag->nb_units; i++) {
VP9RawFrame *frame = frag->units[i].content;
VP9RawFrameHeader *header = &frame->header;
+ int profile = (header->profile_high_bit << 1) + header->profile_low_bit;
+
+ if (header->frame_type == VP9_KEY_FRAME ||
+ header->intra_only && profile > 0) {
+ if (ctx->color_space >= 0) {
+ if (!(profile & 1) && ctx->color_space == VP9_CS_RGB) {
+ if (!(ctx->color_warnings & 2)) {
+ av_log(bsf, AV_LOG_WARNING, "Warning: RGB "
+ "incompatible with profiles 0 and 2.\n");
+ ctx->color_warnings |= 2;
+ }
+ } else
+ header->color_space = ctx->color_space;
+ }
- if (ctx->color_space >= 0) {
- header->color_space = ctx->color_space;
- }
- if (ctx->color_range >= 0) {
- if (ctx->color_range == 0 &&
- header->color_space == VP9_CS_RGB &&
- !ctx->color_range_rgb_warned) {
- av_log(bsf, AV_LOG_WARNING, "Warning: color_range cannot "
- "be set to limited in RGB streams.\n");
- ctx->color_range_rgb_warned = 1;
- } else {
+ if (ctx->color_range >= 0)
header->color_range = ctx->color_range;
+ if (header->color_space == VP9_CS_RGB) {
+ if (!(ctx->color_warnings & 1) && !header->color_range) {
+ av_log(bsf, AV_LOG_WARNING, "Warning: Color space RGB "
+ "implicitly sets color range to PC range.\n");
+ ctx->color_warnings |= 1;
+ }
+ header->color_range = 1;
}
+ } else if (!(ctx->color_warnings & 4) && header->intra_only && !profile &&
+ ctx->color_space >= 0 && ctx->color_space != VP9_CS_BT_601) {
+ av_log(bsf, AV_LOG_WARNING, "Warning: Intra-only frames in "
+ "profile 0 are automatically BT.601.\n");
+ ctx->color_warnings |= 4;
}
}
- err = ff_cbs_write_packet(ctx->cbc, out, frag);
+ err = ff_cbs_write_packet(ctx->cbc, pkt, frag);
if (err < 0) {
av_log(bsf, AV_LOG_ERROR, "Failed to write packet.\n");
goto fail;
}
- err = av_packet_copy_props(out, in);
- if (err < 0)
- goto fail;
-
err = 0;
fail:
- ff_cbs_fragment_reset(ctx->cbc, frag);
+ ff_cbs_fragment_reset(frag);
if (err < 0)
- av_packet_unref(out);
- av_packet_free(&in);
+ av_packet_unref(pkt);
return err;
}
{
VP9MetadataContext *ctx = bsf->priv_data;
- ff_cbs_fragment_free(ctx->cbc, &ctx->fragment);
+ ff_cbs_fragment_free(&ctx->fragment);
ff_cbs_close(&ctx->cbc);
}