1/* 2 * Copyright (c) 2012 Cl��ment B��sch 3 * 4 * This file is part of FFmpeg. 5 * 6 * FFmpeg is free software; you can redistribute it and/or 7 * modify it under the terms of the GNU Lesser General Public 8 * License as published by the Free Software Foundation; either 9 * version 2.1 of the License, or (at your option) any later version. 10 * 11 * FFmpeg is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14 * Lesser General Public License for more details. 15 * 16 * You should have received a copy of the GNU Lesser General Public 17 * License along with FFmpeg; if not, write to the Free Software 18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 19 */ 20 21/** 22 * @file 23 * SubViewer subtitle decoder 24 * @see https://en.wikipedia.org/wiki/SubViewer 25 */ 26 27#include "avcodec.h" 28#include "ass.h" 29#include "libavutil/bprint.h" 30 31static int subviewer_event_to_ass(AVBPrint *buf, const char *p) 32{ 33 while (*p) { 34 if (!strncmp(p, "[br]", 4)) { 35 av_bprintf(buf, "\\N"); 36 p += 4; 37 } else { 38 if (p[0] == '\n' && p[1]) 39 av_bprintf(buf, "\\N"); 40 else if (*p != '\n' && *p != '\r') 41 av_bprint_chars(buf, *p, 1); 42 p++; 43 } 44 } 45 46 av_bprintf(buf, "\r\n"); 47 return 0; 48} 49 50static int subviewer_decode_frame(AVCodecContext *avctx, 51 void *data, int *got_sub_ptr, AVPacket *avpkt) 52{ 53 char c; 54 AVSubtitle *sub = data; 55 const char *ptr = avpkt->data; 56 AVBPrint buf; 57 58 /* To be removed later */ 59 if (ptr && sscanf(ptr, "%*u:%*u:%*u.%*u,%*u:%*u:%*u.%*u%c", &c) == 1) { 60 av_log(avctx, AV_LOG_ERROR, "AVPacket is not clean (contains timing " 61 "information). You need to upgrade your libavformat or " 62 "sanitize your packet.\n"); 63 return AVERROR_INVALIDDATA; 64 } 65 66 av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED); 67 // note: no need to rescale pts & duration since they are in the same 68 // timebase as ASS (1/100) 69 if (ptr && avpkt->size > 0 && !subviewer_event_to_ass(&buf, ptr)) 70 ff_ass_add_rect(sub, buf.str, avpkt->pts, avpkt->duration, 0); 71 *got_sub_ptr = sub->num_rects > 0; 72 av_bprint_finalize(&buf, NULL); 73 return avpkt->size; 74} 75 76AVCodec ff_subviewer_decoder = { 77 .name = "subviewer", 78 .long_name = NULL_IF_CONFIG_SMALL("SubViewer subtitle"), 79 .type = AVMEDIA_TYPE_SUBTITLE, 80 .id = AV_CODEC_ID_SUBVIEWER, 81 .decode = subviewer_decode_frame, 82 .init = ff_ass_subtitle_header_default, 83}; 84