From patchwork Mon Apr 30 23:26:04 2018 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Mark Thompson X-Patchwork-Id: 8703 Delivered-To: ffmpegpatchwork@gmail.com Received: by 2002:a02:155:0:0:0:0:0 with SMTP id c82-v6csp4000744jad; Mon, 30 Apr 2018 16:26:37 -0700 (PDT) X-Google-Smtp-Source: AB8JxZrFevoEUrLQQ9U/txlbGOEA33pTma8V7JXltsnQgvhJLQs6MfTruNBFXUwKOUfRjWWgBv2G X-Received: by 2002:adf:b595:: with SMTP id c21-v6mr10719979wre.233.1525130797749; Mon, 30 Apr 2018 16:26:37 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1525130797; cv=none; d=google.com; s=arc-20160816; b=GyCrNXk7nTcKBakGzM2LfRADiKOkbVbD2inuSDFDApvBTZYpws22+sULG/rj+TFgQA D7c+/5havUlSrwmvZNFgFgYtyWgGVk2OaZsodtVXvOg7cWMGPge+XF3SCm+68urSDynn krw25a9qA7zUlELQx63MYTaI43mrfIkGu5eMGj+/jXKCF11Mb+nsupsMHeShQH2GJn4D 1zELCv7DX1uVErE6bir5HPN2ol2krPGky+4s7xl3j4eah8elJYNC0x6gQQV8QWEw7bWx jfWeNarixz3mRfJ4VGROGdxjDIbR1dfaRHLnd3VNrpEUTH+7HAX+XgOPMapprMR3ujoX mSyQ== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:mime-version:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:references:in-reply-to:message-id:date :to:from:dkim-signature:delivered-to:arc-authentication-results; bh=Sd8dloc0ZefdLguG9lgsagl3+z1F/qTEolnYapsQbnU=; b=y+Zos/5wyjqatIofa7/LBte59UR0st132HjRuLTcQp5G0S9kuOAZUqxUfKU/oXhrUH Jiqoj4O+hkqU46j++F2VSBV+kSVEO/T4rOrZJvfALA691sLvKLZlBvRcIqvMFAjF7ox0 4krDakQxqHLiXqzwiG79aROFffur+a8abhgVHfPraOrcy+575xZfLyYy4qpGUA9GYA2+ OiOzPLVt+YX/B4ZJHmGh1HFXqky4x+ZPa9BJrpEYvSxwlqQdq0lLw4vFOdlywtL9tK2C inQzEXSv6yPijA3qFixonOkS8Ag87awB2aZfQ/Fqzp1e0wboeMSps5WleRNB6gKi8Iss AKnQ== ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@jkqxz-net.20150623.gappssmtp.com header.s=20150623 header.b=Q4RpnOgG; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id 89-v6si7135128wrl.416.2018.04.30.16.26.37; Mon, 30 Apr 2018 16:26:37 -0700 (PDT) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@jkqxz-net.20150623.gappssmtp.com header.s=20150623 header.b=Q4RpnOgG; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 3B67A68A3A6; Tue, 1 May 2018 02:25:50 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-wm0-f46.google.com (mail-wm0-f46.google.com [74.125.82.46]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 36FD068A2D4 for ; Tue, 1 May 2018 02:25:43 +0300 (EEST) Received: by mail-wm0-f46.google.com with SMTP id j5so16816192wme.5 for ; Mon, 30 Apr 2018 16:26:15 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=jkqxz-net.20150623.gappssmtp.com; s=20150623; h=from:to:subject:date:message-id:in-reply-to:references; bh=+gd0qtaWtEuf1bWrO8220EhiAiueIXdMagZKoodtf+g=; b=Q4RpnOgGOyK8hEY6WMvKQeTNr5hYmy+gArGtsJ2yfCjZI+YTq6Qy41bFS9K8qg3DWq K4gU0XjKr5DEPw90cCNORDoagT5gFpTecymLISx2kH1RWu9LRrPDTLxs60P+FYUEsBW2 zPcJHAQJvlUfONzgHCuPyOmXkfixvr+q+SLqIdfX6Ah7JWW4CzGrVpNK/Ezq7CGHvAED L6SPuxDvBzL1vZm8PUnWWWcsYH4jZEcHT3sd6VdNf+KUlN0SRywhu/VCi0QUi4wOnXyg 8FLO/h4u00cC0GRjLf+81v1F1Lzbnlg+LSzS2Df6qxpLUpZ3pH/O3tdpCsuhGWIi8WqA oB9w== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20161025; h=x-gm-message-state:from:to:subject:date:message-id:in-reply-to :references; bh=+gd0qtaWtEuf1bWrO8220EhiAiueIXdMagZKoodtf+g=; b=LXOvxgndhhpUczXQUArPIWZw9X0K3mkiVA+cl/knAFO8sj/tFxU8Le5uUZBj6aXHZ5 +HJOzdC1kEAcWOmK53FKQi6K1I1ACAIcWXJLMZpU2KG10oUeCpRrNRTciheK4h+XoxQR tRc4rJ+/clpkLVZ830fv894BpQxEg7Ivau1L0iC3Hl8PMfr/5eJ262B33DZ+hSZOPce5 VcjJGpea7JMFaHBLS402murgP50RaUVNldDUHSP7QgO3Gn64hDQqfk4g2clcP7oNXssK fkQ6ATVmz6f/TD7FVOkuEdb+wPS+nVXS89w7Xqjy+bb8GT26pc2isyM9Tr4GdI/bsPof 4HQA== X-Gm-Message-State: ALQs6tBYbqrUTr/ffvLoKzVVmofjMHScoiaact9vDL/n0+GIf9BhcAS9 OdfCU8ZWYOMidm7A8kuX0COrMYct X-Received: by 10.28.55.194 with SMTP id e185mr8325204wma.20.1525130774660; Mon, 30 Apr 2018 16:26:14 -0700 (PDT) Received: from rywe.jkqxz.net (cpc91242-cmbg18-2-0-cust650.5-4.cable.virginm.net. [82.8.130.139]) by smtp.gmail.com with ESMTPSA id f5-v6sm8777229wrh.35.2018.04.30.16.26.13 for (version=TLS1_2 cipher=ECDHE-RSA-AES128-GCM-SHA256 bits=128/128); Mon, 30 Apr 2018 16:26:13 -0700 (PDT) From: Mark Thompson To: ffmpeg-devel@ffmpeg.org Date: Tue, 1 May 2018 00:26:04 +0100 Message-Id: <20180430232605.4846-5-sw@jkqxz.net> X-Mailer: git-send-email 2.16.3 In-Reply-To: <20180430232605.4846-1-sw@jkqxz.net> References: <20180430232605.4846-1-sw@jkqxz.net> Subject: [FFmpeg-devel] [PATCH 5/6] lavc: Add VP9 metadata bitstream filter X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.20 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Can adjust the colour information. --- configure | 1 + doc/bitstream_filters.texi | 26 +++++++ libavcodec/Makefile | 1 + libavcodec/bitstream_filters.c | 1 + libavcodec/vp9_metadata_bsf.c | 162 +++++++++++++++++++++++++++++++++++++++++ 5 files changed, 191 insertions(+) create mode 100644 libavcodec/vp9_metadata_bsf.c diff --git a/configure b/configure index 57748e1424..7f199c634d 100755 --- a/configure +++ b/configure @@ -2989,6 +2989,7 @@ hevc_metadata_bsf_select="cbs_h265" mjpeg2jpeg_bsf_select="jpegtables" mpeg2_metadata_bsf_select="cbs_mpeg2" trace_headers_bsf_select="cbs" +vp9_metadata_bsf_select="cbs_vp9" # external libraries aac_at_decoder_deps="audiotoolbox" diff --git a/doc/bitstream_filters.texi b/doc/bitstream_filters.texi index 7f98cb714f..7d7e97503a 100644 --- a/doc/bitstream_filters.texi +++ b/doc/bitstream_filters.texi @@ -507,6 +507,32 @@ This can be useful for debugging low-level stream issues. Supports H.264, H.265, MPEG-2 and VP9. +@section vp9_metadata + +Modify metadata embedded in a VP9 stream. + +@table @option +@item color_space +Set the color space value in the frame header. +@table @samp +@item unknown +@item bt601 +@item bt709 +@item smpte170 +@item smpte240 +@item bt2020 +@item rgb +@end table + +@item color_range +Set the color range value in the frame header. Note that this cannot +be set in RGB streams. +@table @samp +@item tv +@item pc +@end table +@end table + @section vp9_superframe Merge VP9 invisible (alt-ref) frames back into VP9 superframes. This diff --git a/libavcodec/Makefile b/libavcodec/Makefile index 15e8c87385..663bdce4e4 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -1066,6 +1066,7 @@ OBJS-$(CONFIG_NULL_BSF) += null_bsf.o OBJS-$(CONFIG_REMOVE_EXTRADATA_BSF) += remove_extradata_bsf.o OBJS-$(CONFIG_TEXT2MOVSUB_BSF) += movsub_bsf.o OBJS-$(CONFIG_TRACE_HEADERS_BSF) += trace_headers_bsf.o +OBJS-$(CONFIG_VP9_METADATA_BSF) += vp9_metadata_bsf.o OBJS-$(CONFIG_VP9_RAW_REORDER_BSF) += vp9_raw_reorder_bsf.o OBJS-$(CONFIG_VP9_SUPERFRAME_BSF) += vp9_superframe_bsf.o OBJS-$(CONFIG_VP9_SUPERFRAME_SPLIT_BSF) += vp9_superframe_split_bsf.o diff --git a/libavcodec/bitstream_filters.c b/libavcodec/bitstream_filters.c index 18b698a85f..c21373621c 100644 --- a/libavcodec/bitstream_filters.c +++ b/libavcodec/bitstream_filters.c @@ -49,6 +49,7 @@ extern const AVBitStreamFilter ff_null_bsf; extern const AVBitStreamFilter ff_remove_extradata_bsf; extern const AVBitStreamFilter ff_text2movsub_bsf; extern const AVBitStreamFilter ff_trace_headers_bsf; +extern const AVBitStreamFilter ff_vp9_metadata_bsf; extern const AVBitStreamFilter ff_vp9_raw_reorder_bsf; extern const AVBitStreamFilter ff_vp9_superframe_bsf; extern const AVBitStreamFilter ff_vp9_superframe_split_bsf; diff --git a/libavcodec/vp9_metadata_bsf.c b/libavcodec/vp9_metadata_bsf.c new file mode 100644 index 0000000000..be010edc3f --- /dev/null +++ b/libavcodec/vp9_metadata_bsf.c @@ -0,0 +1,162 @@ +/* + * 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 "libavutil/avstring.h" +#include "libavutil/common.h" +#include "libavutil/opt.h" + +#include "bsf.h" +#include "cbs.h" +#include "cbs_vp9.h" + +typedef struct VP9MetadataContext { + const AVClass *class; + + CodedBitstreamContext *cbc; + CodedBitstreamFragment fragment; + + int color_space; + int color_range; + + int color_range_rgb_warned; +} VP9MetadataContext; + + +static int vp9_metadata_filter(AVBSFContext *bsf, AVPacket *out) +{ + VP9MetadataContext *ctx = bsf->priv_data; + AVPacket *in = NULL; + CodedBitstreamFragment *frag = &ctx->fragment; + int err, i; + + err = ff_bsf_get_packet(bsf, &in); + if (err < 0) + return err; + + err = ff_cbs_read_packet(ctx->cbc, frag, in); + 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; + + 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 { + header->color_range = ctx->color_range; + } + } + } + + err = ff_cbs_write_packet(ctx->cbc, out, 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_uninit(ctx->cbc, frag); + + if (err < 0) + av_packet_unref(out); + av_packet_free(&in); + + return err; +} + +static int vp9_metadata_init(AVBSFContext *bsf) +{ + VP9MetadataContext *ctx = bsf->priv_data; + + return ff_cbs_init(&ctx->cbc, AV_CODEC_ID_VP9, bsf); +} + +static void vp9_metadata_close(AVBSFContext *bsf) +{ + VP9MetadataContext *ctx = bsf->priv_data; + ff_cbs_close(&ctx->cbc); +} + +#define OFFSET(x) offsetof(VP9MetadataContext, x) +#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM|AV_OPT_FLAG_BSF_PARAM) +static const AVOption vp9_metadata_options[] = { + { "color_space", "Set colour space (section 7.2.2)", + OFFSET(color_space), AV_OPT_TYPE_INT, + { .i64 = -1 }, -1, VP9_CS_RGB, FLAGS, "cs" }, + { "unknown", "Unknown/unspecified", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_UNKNOWN }, .flags = FLAGS, .unit = "cs" }, + { "bt601", "ITU-R BT.601-7", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_BT_601 }, .flags = FLAGS, .unit = "cs" }, + { "bt709", "ITU-R BT.709-6", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_BT_709 }, .flags = FLAGS, .unit = "cs" }, + { "smpte170", "SMPTE-170", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_SMPTE_170 }, .flags = FLAGS, .unit = "cs" }, + { "smpte240", "SMPTE-240", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_SMPTE_240 }, .flags = FLAGS, .unit = "cs" }, + { "bt2020", "ITU-R BT.2020-2", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_BT_2020 }, .flags = FLAGS, .unit = "cs" }, + { "rgb", "sRGB / IEC 61966-2-1", 0, AV_OPT_TYPE_CONST, + { .i64 = VP9_CS_RGB }, .flags = FLAGS, .unit = "cs" }, + + { "color_range", "Set colour range (section 7.2.2)", + OFFSET(color_range), AV_OPT_TYPE_INT, + { .i64 = -1 }, -1, 1, FLAGS, "cr" }, + { "tv", "TV (limited) range", 0, AV_OPT_TYPE_CONST, + { .i64 = 0 }, .flags = FLAGS, .unit = "cr" }, + { "pc", "PC (full) range", 0, AV_OPT_TYPE_CONST, + { .i64 = 1 }, .flags = FLAGS, .unit = "cr" }, + + { NULL } +}; + +static const AVClass vp9_metadata_class = { + .class_name = "vp9_metadata_bsf", + .item_name = av_default_item_name, + .option = vp9_metadata_options, + .version = LIBAVUTIL_VERSION_INT, +}; + +static const enum AVCodecID vp9_metadata_codec_ids[] = { + AV_CODEC_ID_VP9, AV_CODEC_ID_NONE, +}; + +const AVBitStreamFilter ff_vp9_metadata_bsf = { + .name = "vp9_metadata", + .priv_data_size = sizeof(VP9MetadataContext), + .priv_class = &vp9_metadata_class, + .init = &vp9_metadata_init, + .close = &vp9_metadata_close, + .filter = &vp9_metadata_filter, + .codec_ids = vp9_metadata_codec_ids, +};