FFmpeg
mpl2dec.c
Go to the documentation of this file.
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 
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 
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;
73 
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 
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,
91  .flush = ff_ass_decoder_flush,
92  .priv_data_size = sizeof(FFASSDecoderContext),
93 };
AVSubtitle
Definition: avcodec.h:2694
AVCodec
AVCodec.
Definition: codec.h:190
AVMEDIA_TYPE_SUBTITLE
@ AVMEDIA_TYPE_SUBTITLE
Definition: avutil.h:204
AV_BPRINT_SIZE_UNLIMITED
#define AV_BPRINT_SIZE_UNLIMITED
ff_ass_subtitle_header_default
int ff_ass_subtitle_header_default(AVCodecContext *avctx)
Generate a suitable AVCodecContext.subtitle_header for SUBTITLE_ASS with default style.
Definition: ass.c:97
av_bprint_finalize
int av_bprint_finalize(AVBPrint *buf, char **ret_str)
Finalize a print buffer.
Definition: bprint.c:235
av_bprint_init
void av_bprint_init(AVBPrint *buf, unsigned size_init, unsigned size_max)
Definition: bprint.c:69
AVSubtitle::num_rects
unsigned num_rects
Definition: avcodec.h:2698
ff_ass_add_rect
int ff_ass_add_rect(AVSubtitle *sub, const char *dialog, int readorder, int layer, const char *style, const char *speaker)
Add an ASS dialog to a subtitle.
Definition: ass.c:118
AVPacket::data
uint8_t * data
Definition: packet.h:355
data
const char data[16]
Definition: mxf.c:91
AV_CODEC_ID_MPL2
@ AV_CODEC_ID_MPL2
Definition: codec_id.h:528
ass.h
s
#define s(width, name)
Definition: cbs_vp9.c:257
NULL
#define NULL
Definition: coverity.c:32
AVPacket::size
int size
Definition: packet.h:356
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:188
ff_mpl2_decoder
AVCodec ff_mpl2_decoder
Definition: mpl2dec.c:84
mpl2_decode_frame
static int mpl2_decode_frame(AVCodecContext *avctx, void *data, int *got_sub_ptr, AVPacket *avpkt)
Definition: mpl2dec.c:65
ff_ass_decoder_flush
void ff_ass_decoder_flush(AVCodecContext *avctx)
Helper to flush a text subtitles decoder making use of the FFASSDecoderContext.
Definition: ass.c:141
bprint.h
AVCodec::name
const char * name
Name of the codec implementation.
Definition: codec.h:197
avcodec.h
ret
ret
Definition: filter_design.txt:187
av_bprintf
void av_bprintf(AVBPrint *buf, const char *fmt,...)
Definition: bprint.c:94
AVCodecContext
main external API structure.
Definition: avcodec.h:526
mpl2_event_to_ass
static int mpl2_event_to_ass(AVBPrint *buf, const char *p)
Definition: mpl2dec.c:32
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:553
AVPacket
This structure stores compressed data.
Definition: packet.h:332
FFASSDecoderContext
Definition: ass.h:46
av_bprint_chars
void av_bprint_chars(AVBPrint *buf, char c, unsigned n)
Append char c n times to a print buffer.
Definition: bprint.c:140