From patchwork Mon Oct 11 04:51:10 2021 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Soft Works X-Patchwork-Id: 31032 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6602:2084:0:0:0:0 with SMTP id a4csp3330991ioa; Sun, 10 Oct 2021 21:53:20 -0700 (PDT) X-Google-Smtp-Source: ABdhPJwHp5xjuWqxJx9YwKt9XXSJN7m3hbkmXlYJs/ySPgKWSSMx8EaCqzm8Yp3N5fjBrpQBi+Nx X-Received: by 2002:a17:906:1184:: with SMTP id n4mr23431547eja.87.1633928000317; Sun, 10 Oct 2021 21:53:20 -0700 (PDT) Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id cf7si9809316ejb.118.2021.10.10.21.53.19; Sun, 10 Oct 2021 21:53:20 -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=@hotmail.com header.s=selector1 header.b=j4cF+fNP; arc=fail (body hash mismatch); 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; dmarc=fail (p=NONE sp=NONE dis=NONE) header.from=hotmail.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 7298D68A7D4; Mon, 11 Oct 2021 07:51:23 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from NAM10-MW2-obe.outbound.protection.outlook.com (mail-mw2nam10olkn2096.outbound.protection.outlook.com [40.92.42.96]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 630C768A803 for ; Mon, 11 Oct 2021 07:51:15 +0300 (EEST) ARC-Seal: i=1; a=rsa-sha256; s=arcselector9901; d=microsoft.com; cv=none; b=lfWgWt1gCosIlrklol7UGy77nV09OJ5PH7VesLBBid5ulRiKzMmkmV1EfagE9y97K0Q/RHy/pUKKr+Cso+8NKdUkQ4MSbLuyqSsHUprJHKmVtAGn3WEg7rX7tGN5ilGn+vtzPSXnswrOSsjUTZpktIAlna2H1uHrF5uMKzfUZ15cRHeGSTwkhgbq7MpxryM58aGfnnVcbRMQ0d05vOmduehT9zYxWuh9ICPLYnqXiY3UtwQ+q9uhQ6YZ4h8JOcV231LAaeCJuhKGksHD4RPkv4feEBhE68zdx0Ea0ra3W+e4bL1W31gqPc02dyTT9lBDCX9kOSdVPaycg8bNoElbSg== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=microsoft.com; s=arcselector9901; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-AntiSpam-MessageData-ChunkCount:X-MS-Exchange-AntiSpam-MessageData-0:X-MS-Exchange-AntiSpam-MessageData-1; bh=YpQACWOEsP6ht5MnsTeXMk84q8fk3GbQlDG5qsk/78g=; b=Emt0JEd8Ja0QRmIr2++TrNiD5Ufli7X0qTz6tCHtIT93+RhTP3yF/G1K2ccAzIjj45bBVPAEK+8lp4AGs2z/29KhVTUDNBrEc969jXqFktOBwi/5vf6sh74UTxEEvDElRR2S/Hx17U+SsaVQSJmhUBP9Mtf+gJkYGdOCV5sBjBuaxi6VAcx4gmWoGsuZ2nUceN/3h8KX8ime7ex4mE9iWSusRcLxxK9DKgKrlI6/8OL7CgKntkWM+K4R76PIqbCJh2du7UPhVUIyZoKHSvEYo3sUDRogHS2nNaxu+QMwt+TDAsxVHCjWP7qa/lFea+gIAWDbPr2SgyVDrR0KG4QRCg== ARC-Authentication-Results: i=1; mx.microsoft.com 1; spf=none; dmarc=none; dkim=none; arc=none DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=hotmail.com; s=selector1; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-SenderADCheck; bh=YpQACWOEsP6ht5MnsTeXMk84q8fk3GbQlDG5qsk/78g=; b=j4cF+fNP+tWylJp11vjWE+ygcck4rotVoQmhGXKgjPFtn4THwWDO1bTBWS5Zvi4NyWfFw1wEUr+FxMr31bmMo1O3ZzdXNZRegbNrzpJ93p+UddHBOh4DGv3u1KxtWlvGveTPkUeqNaNl0MBNFNGUBkbJUHx56yB5vtt3CFUOIhKXUTMY3WRKCcRpd3vaPm/VY2EN+iLHc4VZ5oOtQ26VgWS7ZrWgK4stTqV5LxMWzBAQtzvEmRi3YS7sHF4AmmgAgPzv6KiaPZDwWOLKFBI07Z//TOlrTgmqRJlgQr15P7frASZx6fuTrJIocMUeF37uvLzX61iOzvAigZxX/cqOmg== Received: from MN2PR04MB5981.namprd04.prod.outlook.com (2603:10b6:208:da::10) by MN2PR04MB6016.namprd04.prod.outlook.com (2603:10b6:208:de::15) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.20.4587.18; Mon, 11 Oct 2021 04:51:10 +0000 Received: from MN2PR04MB5981.namprd04.prod.outlook.com ([fe80::ecfe:2528:2012:22cb]) by MN2PR04MB5981.namprd04.prod.outlook.com ([fe80::ecfe:2528:2012:22cb%5]) with mapi id 15.20.4587.026; Mon, 11 Oct 2021 04:51:10 +0000 From: Soft Works To: "ffmpeg-devel@ffmpeg.org" Thread-Topic: [PATCH v13 11/14] avfilter/stripstyles: Add stripstyles filter Thread-Index: AQHXvlucr8BIG3EqbkaxMP1xig5r5g== Date: Mon, 11 Oct 2021 04:51:10 +0000 Message-ID: References: <41c4d8049ad6a90ab6a8e29955aae8f39d1000ec.1633926637.git.softworkz@hotmail.com> <669100dbb5e7bda1dfc9e8aa8cb7eaf8c6661ec4.1633926637.git.softworkz@hotmail.com> <85114ac94da731c168cac9db60bb1a33c08832e2.1633926637.git.softworkz@hotmail.com> <7954378c42eadb165aad719398f801bb57d919bc.1633926637.git.softworkz@hotmail.com> <830dd348c9618de1f4164fda9dea181edb0f5c8f.1633926637.git.softworkz@hotmail.com> <16fd403f6d238d5d5adf39c562ff6b47fb28e8df.1633926637.git.softworkz@hotmail.com> In-Reply-To: <16fd403f6d238d5d5adf39c562ff6b47fb28e8df.1633926637.git.softworkz@hotmail.com> Accept-Language: en-US Content-Language: en-US X-MS-Has-Attach: X-MS-TNEF-Correlator: x-ms-exchange-messagesentrepresentingtype: 1 x-tmn: [9yvPKOYH68v8fEvHgtRVjD6oKZVgRPFo] x-ms-publictraffictype: Email x-ms-office365-filtering-correlation-id: 8c7367aa-f2cc-47b4-2586-08d98c72bea4 x-ms-traffictypediagnostic: MN2PR04MB6016: x-microsoft-antispam: BCL:0; x-microsoft-antispam-message-info: HONl7RGycIZhzzWwPbXhq9jFaKq5E8aj5AtQXIsZpSsdAlgeYEigP8ZKL/i9fJkcXVfKCCFwsXYGVaVNPCzKfv57fJRcgX1/nJAXyxx/2V/pYsDs8hDKyVWUsw5CX8TyHvaDhSaFBLRbO8zhfLe7DDIwYrDMxHOxasK075zEmAL8s8/OCMuSVphiXvB0mKbYVGwvuC6QocCRToQ0rDP4RGe5Lv2V+FbYa6ee76ysG9+KqH3xh+f1NPEcvsLjq47CZy9dNZcYkpW1Nk3qZbQ9iQgYxtOqhnMEFNdaDf0fnhetFb4vV0/4iSx5vgR/LOYlaB/dmn+LIOmVsVkXF8WoBa2akiOFVPdvrlPSxps5fNCJhBF3dmdPaD6ZxYADvYDBVZcPczBuE2l2Hr8GGVl6uXwx7lLwYKtKcNJhm5xaWNKESO+EwGp/B++0l6fJPZJBgN63mS+mW2sc7SK15fpbu8km7i2ijcxP7KTt04OZdzOFhnHoEvQRmkrQGH8JdnZzCm2tbbaAf1bc75DGkzYC0g== x-ms-exchange-antispam-messagedata-chunkcount: 1 x-ms-exchange-antispam-messagedata-0: riP8V0xn/Yv4R9MjmIzYiSQwWhuNkolnNJUt0OmOS/4xP7Van+YNda7dkUrO1eaS2ZfHY1CK0ICMQ102CWcOl9TPTSKjxnkiZLLEWMNXMb8GcoEWGlFrcZaGkc2vJ73oBrV/bXfjNmGGIiIU0AhsXw== x-ms-exchange-transport-forked: True MIME-Version: 1.0 X-OriginatorOrg: sct-15-20-3174-20-msonline-outlook-529c7.templateTenant X-MS-Exchange-CrossTenant-AuthAs: Internal X-MS-Exchange-CrossTenant-AuthSource: MN2PR04MB5981.namprd04.prod.outlook.com X-MS-Exchange-CrossTenant-RMS-PersistedConsumerOrg: 00000000-0000-0000-0000-000000000000 X-MS-Exchange-CrossTenant-Network-Message-Id: 8c7367aa-f2cc-47b4-2586-08d98c72bea4 X-MS-Exchange-CrossTenant-originalarrivaltime: 11 Oct 2021 04:51:10.1465 (UTC) X-MS-Exchange-CrossTenant-fromentityheader: Hosted X-MS-Exchange-CrossTenant-id: 84df9e7f-e9f6-40af-b435-aaaaaaaaaaaa X-MS-Exchange-CrossTenant-rms-persistedconsumerorg: 00000000-0000-0000-0000-000000000000 X-MS-Exchange-Transport-CrossTenantHeadersStamped: MN2PR04MB6016 Subject: [FFmpeg-devel] [PATCH v13 11/14] avfilter/stripstyles: Add stripstyles filter X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 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 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: RWoUrd7whetS - stripstyles {S -> S) Remove all inline styles from subtitle events Signed-off-by: softworkz --- doc/filters.texi | 34 +++++++ libavfilter/Makefile | 1 + libavfilter/allfilters.c | 1 + libavfilter/sf_stripstyles.c | 191 +++++++++++++++++++++++++++++++++++ 4 files changed, 227 insertions(+) create mode 100644 libavfilter/sf_stripstyles.c diff --git a/doc/filters.texi b/doc/filters.texi index 764c849fac..4ee12fdee8 100644 --- a/doc/filters.texi +++ b/doc/filters.texi @@ -25295,6 +25295,40 @@ ffmpeg -i "https://streams.videolan.org/ffmpeg/mkv_subtitles.mkv" -filter_comple @end example @end itemize +@section stripstyles + +Remove all inline styles from subtitle events. + +Inputs: +@itemize +@item 0: Subtitles[TEXT] +@end itemize + +Outputs: +@itemize +@item 0: Subtitles[TEXT] +@end itemize + +It accepts the following parameters: + +@table @option +@item remove_animated +Also remove text which is subject to animation (default: true) +Usually, animated text elements are used used in addition to static subtitle lines for creating effects, so in most cases it is safe to remove the animation content. +If subtitle text is missing, try setting this to false. + +@end table + +@subsection Examples + +@itemize +@item +Remove all styles and animations from subtitles: +@example +ffmpeg -i "https://streams.videolan.org/samples/sub/SSA/subtitle_testing_complex.mkv" -filter_complex "[0:1]stripstyles" -map 0 output.mkv +@end example +@end itemize + @section textmod diff --git a/libavfilter/Makefile b/libavfilter/Makefile index 1a659a8377..ee7da21d96 100644 --- a/libavfilter/Makefile +++ b/libavfilter/Makefile @@ -545,6 +545,7 @@ OBJS-$(CONFIG_NULLSINK_FILTER) += vsink_nullsink.o OBJS-$(CONFIG_CENSOR_FILTER) += sf_textmod.o OBJS-$(CONFIG_SHOW_SPEAKER_FILTER) += sf_textmod.o OBJS-$(CONFIG_TEXTMOD_FILTER) += sf_textmod.o +OBJS-$(CONFIG_STRIPSTYLES_FILTER) += sf_stripstyles.o # multimedia filters OBJS-$(CONFIG_ABITSCOPE_FILTER) += avf_abitscope.o diff --git a/libavfilter/allfilters.c b/libavfilter/allfilters.c index e9f1e88bd2..cc65ee892e 100644 --- a/libavfilter/allfilters.c +++ b/libavfilter/allfilters.c @@ -533,6 +533,7 @@ extern const AVFilter ff_avf_showwavespic; extern const AVFilter ff_vaf_spectrumsynth; extern const AVFilter ff_sf_censor; extern const AVFilter ff_sf_showspeaker; +extern const AVFilter ff_sf_stripstyles; extern const AVFilter ff_sf_textmod; extern const AVFilter ff_svf_graphicsub2video; extern const AVFilter ff_svf_textsub2video; diff --git a/libavfilter/sf_stripstyles.c b/libavfilter/sf_stripstyles.c new file mode 100644 index 0000000000..ed35c2afcc --- /dev/null +++ b/libavfilter/sf_stripstyles.c @@ -0,0 +1,191 @@ +/* + * Copyright (c) 2021 softworkz + * + * 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 + */ + +/** + * @file + * text subtitle filter which removes inline-styles from subtitles + */ + +#include "libavutil/opt.h" +#include "internal.h" +#include "libavutil/ass_split_internal.h" +#include "libavutil/bprint.h" + +typedef struct StripStylesContext { + const AVClass *class; + enum AVSubtitleType format; + int remove_animated; +} StripStylesContext; + +typedef struct DialogContext { + StripStylesContext* ss_ctx; + AVBPrint buffer; + int drawing_scale; + int is_animated; +} DialogContext; + +static void dialog_text_cb(void *priv, const char *text, int len) +{ + DialogContext *s = priv; + + av_log(s->ss_ctx, AV_LOG_DEBUG, "dialog_text_cb: %s\n", text); + + if (!s->drawing_scale && (!s->is_animated || !s->ss_ctx->remove_animated)) + av_bprint_append_data(&s->buffer, text, len); +} + +static void dialog_new_line_cb(void *priv, int forced) +{ + DialogContext *s = priv; + if (!s->drawing_scale && !s->is_animated) + av_bprint_append_data(&s->buffer, forced ? "\\N" : "\\n", 2); +} + +static void dialog_drawing_mode_cb(void *priv, int scale) +{ + DialogContext *s = priv; + s->drawing_scale = scale; +} + +static void dialog_animate_cb(void *priv, int t1, int t2, int accel, char *style) +{ + DialogContext *s = priv; + s->is_animated = 1; +} + +static void dialog_move_cb(void *priv, int x1, int y1, int x2, int y2, int t1, int t2) +{ + DialogContext *s = priv; + if (t1 >= 0 || t2 >= 0) + s->is_animated = 1; +} + +static const ASSCodesCallbacks dialog_callbacks = { + .text = dialog_text_cb, + .new_line = dialog_new_line_cb, + .drawing_mode = dialog_drawing_mode_cb, + .animate = dialog_animate_cb, + .move = dialog_move_cb, +}; + +static char *ass_get_line(int readorder, int layer, const char *style, + const char *speaker, const char *effect, const char *text) +{ + return av_asprintf("%d,%d,%s,%s,0,0,0,%s,%s", + readorder, layer, style ? style : "Default", + speaker ? speaker : "", effect, text); +} + +static char *process_dialog(StripStylesContext *s, const char *ass_line) +{ + DialogContext dlg_ctx = { .ss_ctx = s }; + ASSDialog *dialog = avpriv_ass_split_dialog(NULL, ass_line); + char *result = NULL; + + if (!dialog) + return NULL; + + dlg_ctx.ss_ctx = s; + + av_bprint_init(&dlg_ctx.buffer, 0, AV_BPRINT_SIZE_UNLIMITED); + + avpriv_ass_split_override_codes(&dialog_callbacks, &dlg_ctx, dialog->text); + + if (av_bprint_is_complete(&dlg_ctx.buffer) + && dlg_ctx.buffer.len > 0) + result = ass_get_line(dialog->readorder, dialog->layer, dialog->style, dialog->name, dialog->effect, dlg_ctx.buffer.str); + + av_bprint_finalize(&dlg_ctx.buffer, NULL); + avpriv_ass_free_dialog(&dialog); + + return result; +} + +static int filter_frame(AVFilterLink *inlink, AVFrame *frame) +{ + StripStylesContext *s = inlink->dst->priv; + AVFilterLink *outlink = inlink->dst->outputs[0]; + int ret; + + outlink->format = inlink->format; + + ret = av_frame_make_writable(frame); + if (ret <0 ) { + av_frame_free(&frame); + return AVERROR(ENOMEM); + } + + for (unsigned i = 0; i < frame->num_subtitle_areas; i++) { + + AVSubtitleArea *area = frame->subtitle_areas[i]; + + if (area->ass) { + char *tmp = area->ass; + area->ass = process_dialog(s, area->ass); + + if (area->ass) { + av_log(inlink->dst, AV_LOG_DEBUG, "original: %s\n", tmp); + av_log(inlink->dst, AV_LOG_DEBUG, "stripped: %s\n", area->ass); + } + else + area->ass = NULL; + + av_free(tmp); + } + } + + return ff_filter_frame(outlink, frame); +} + +#define OFFSET(x) offsetof(StripStylesContext, x) +#define FLAGS (AV_OPT_FLAG_SUBTITLE_PARAM | AV_OPT_FLAG_FILTERING_PARAM) + +static const AVOption stripstyles_options[] = { + { "remove_animated", "remove animated text (default: yes)", OFFSET(remove_animated), AV_OPT_TYPE_BOOL, {.i64 = 1 }, 0, 1, FLAGS, 0 }, + { NULL }, +}; + +AVFILTER_DEFINE_CLASS(stripstyles); + +static const AVFilterPad inputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_SUBTITLE, + .filter_frame = filter_frame, + }, +}; + +static const AVFilterPad outputs[] = { + { + .name = "default", + .type = AVMEDIA_TYPE_SUBTITLE, + }, +}; + +const AVFilter ff_sf_stripstyles = { + .name = "stripstyles", + .description = NULL_IF_CONFIG_SMALL("Strip subtitle inline styles"), + .priv_size = sizeof(StripStylesContext), + .priv_class = &stripstyles_class, + FILTER_INPUTS(inputs), + FILTER_OUTPUTS(outputs), + FILTER_SINGLE_SUBFMT(AV_SUBTITLE_FMT_ASS), +}; +