2 * Copyright (c) 2012 Clément Bœsch
4 * This file is part of FFmpeg.
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.
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.
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
23 * SAMI subtitle demuxer
24 * @see http://msdn.microsoft.com/en-us/library/ms971327.aspx
29 #include "subtitles.h"
30 #include "libavcodec/internal.h"
31 #include "libavutil/avstring.h"
32 #include "libavutil/bprint.h"
33 #include "libavutil/intreadwrite.h"
36 FFDemuxSubtitlesQueue q
;
39 static int sami_probe(AVProbeData
*p
)
43 ff_text_init_buf(&tr
, p
->buf
, p
->buf_size
);
44 ff_text_read(&tr
, buf
, sizeof(buf
));
46 return !strncmp(buf
, "<SAMI>", 6) ? AVPROBE_SCORE_MAX
: 0;
49 static int sami_read_header(AVFormatContext
*s
)
51 SAMIContext
*sami
= s
->priv_data
;
52 AVStream
*st
= avformat_new_stream(s
, NULL
);
53 AVBPrint buf
, hdr_buf
;
55 int res
= 0, got_first_sync_point
= 0;
57 ff_text_init_avio(s
, &tr
, s
->pb
);
60 return AVERROR(ENOMEM
);
61 avpriv_set_pts_info(st
, 64, 1, 1000);
62 st
->codec
->codec_type
= AVMEDIA_TYPE_SUBTITLE
;
63 st
->codec
->codec_id
= AV_CODEC_ID_SAMI
;
65 av_bprint_init(&buf
, 0, AV_BPRINT_SIZE_UNLIMITED
);
66 av_bprint_init(&hdr_buf
, 0, AV_BPRINT_SIZE_UNLIMITED
);
68 while (!ff_text_eof(&tr
)) {
70 const int64_t pos
= ff_text_pos(&tr
) - (c
!= 0);
71 int is_sync
, n
= ff_smil_extract_next_text_chunk(&tr
, &buf
, &c
);
76 is_sync
= !av_strncasecmp(buf
.str
, "<SYNC", 5);
78 got_first_sync_point
= 1;
80 if (!got_first_sync_point
) {
81 av_bprintf(&hdr_buf
, "%s", buf
.str
);
83 sub
= ff_subtitles_queue_insert(&sami
->q
, buf
.str
, buf
.len
, !is_sync
);
85 res
= AVERROR(ENOMEM
);
89 const char *p
= ff_smil_get_attr_ptr(buf
.str
, "Start");
91 sub
->pts
= p
? strtol(p
, NULL
, 10) : 0;
95 av_bprint_clear(&buf
);
98 res
= avpriv_bprint_to_extradata(st
->codec
, &hdr_buf
);
102 ff_subtitles_queue_finalize(&sami
->q
);
105 av_bprint_finalize(&buf
, NULL
);
109 static int sami_read_packet(AVFormatContext
*s
, AVPacket
*pkt
)
111 SAMIContext
*sami
= s
->priv_data
;
112 return ff_subtitles_queue_read_packet(&sami
->q
, pkt
);
115 static int sami_read_seek(AVFormatContext
*s
, int stream_index
,
116 int64_t min_ts
, int64_t ts
, int64_t max_ts
, int flags
)
118 SAMIContext
*sami
= s
->priv_data
;
119 return ff_subtitles_queue_seek(&sami
->q
, s
, stream_index
,
120 min_ts
, ts
, max_ts
, flags
);
123 static int sami_read_close(AVFormatContext
*s
)
125 SAMIContext
*sami
= s
->priv_data
;
126 ff_subtitles_queue_clean(&sami
->q
);
130 AVInputFormat ff_sami_demuxer
= {
132 .long_name
= NULL_IF_CONFIG_SMALL("SAMI subtitle format"),
133 .priv_data_size
= sizeof(SAMIContext
),
134 .read_probe
= sami_probe
,
135 .read_header
= sami_read_header
,
136 .read_packet
= sami_read_packet
,
137 .read_seek2
= sami_read_seek
,
138 .read_close
= sami_read_close
,
139 .extensions
= "smi,sami",