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 * MPL2 subtitles decoder
24 *
25 * @see http://web.archive.org/web/20090328040233/http://napisy.ussbrowarek.org/mpl2-eng.html
26 */
27
28 #include "avcodec.h"
29 #include "ass.h"
30 #include "libavutil/bprint.h"
31
mpl2_event_to_ass(AVBPrint * buf,const char * p)32 static int mpl2_event_to_ass(AVBPrint *buf, const char *p)
33 {
34 if (*p == ' ')
35 p++;
36
37 while (*p) {
38 int got_style = 0;
39
40 while (*p && strchr("/\\_", *p)) {
41 if (*p == '/') av_bprintf(buf, "{\\i1}");
42 else if (*p == '\\') av_bprintf(buf, "{\\b1}");
43 else if (*p == '_') av_bprintf(buf, "{\\u1}");
44 got_style = 1;
45 p++;
46 }
47
48 while (*p && *p != '|') {
49 if (*p != '\r' && *p != '\n')
50 av_bprint_chars(buf, *p, 1);
51 p++;
52 }
53
54 if (*p == '|') {
55 if (got_style)
56 av_bprintf(buf, "{\\r}");
57 av_bprintf(buf, "\\N");
58 p++;
59 }
60 }
61
62 return 0;
63 }
64
mpl2_decode_frame(AVCodecContext * avctx,void * data,int * got_sub_ptr,AVPacket * avpkt)65 static int mpl2_decode_frame(AVCodecContext *avctx, void *data,
66 int *got_sub_ptr, AVPacket *avpkt)
67 {
68 int ret = 0;
69 AVBPrint buf;
70 AVSubtitle *sub = data;
71 const char *ptr = avpkt->data;
72 FFASSDecoderContext *s = avctx->priv_data;
73
74 av_bprint_init(&buf, 0, AV_BPRINT_SIZE_UNLIMITED);
75 if (ptr && avpkt->size > 0 && *ptr && !mpl2_event_to_ass(&buf, ptr))
76 ret = ff_ass_add_rect(sub, buf.str, s->readorder++, 0, NULL, NULL);
77 av_bprint_finalize(&buf, NULL);
78 if (ret < 0)
79 return ret;
80 *got_sub_ptr = sub->num_rects > 0;
81 return avpkt->size;
82 }
83
84 AVCodec ff_mpl2_decoder = {
85 .name = "mpl2",
86 .long_name = NULL_IF_CONFIG_SMALL("MPL2 subtitle"),
87 .type = AVMEDIA_TYPE_SUBTITLE,
88 .id = AV_CODEC_ID_MPL2,
89 .decode = mpl2_decode_frame,
90 .init = ff_ass_subtitle_header_default,
91 .flush = ff_ass_decoder_flush,
92 .priv_data_size = sizeof(FFASSDecoderContext),
93 };
94