2 * This file is part of FFmpeg.
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 #include "libavutil/intreadwrite.h"
23 #define SUP_PGS_MAGIC 0x5047 /* "PG", big endian */
25 static int sup_read_header(AVFormatContext
*s
)
27 AVStream
*st
= avformat_new_stream(s
, NULL
);
29 return AVERROR(ENOMEM
);
30 st
->codec
->codec_type
= AVMEDIA_TYPE_SUBTITLE
;
31 st
->codec
->codec_id
= AV_CODEC_ID_HDMV_PGS_SUBTITLE
;
32 avpriv_set_pts_info(st
, 32, 1, 90000);
37 static int sup_read_packet(AVFormatContext
*s
, AVPacket
*pkt
)
39 int64_t pts
, dts
, pos
;
42 pos
= avio_tell(s
->pb
);
44 if (avio_rb16(s
->pb
) != SUP_PGS_MAGIC
)
45 return avio_feof(s
->pb
) ? AVERROR_EOF
: AVERROR_INVALIDDATA
;
47 pts
= avio_rb32(s
->pb
);
48 dts
= avio_rb32(s
->pb
);
50 if ((ret
= av_get_packet(s
->pb
, pkt
, 3)) < 0)
53 pkt
->stream_index
= 0;
54 pkt
->flags
|= AV_PKT_FLAG_KEY
;
57 // Many files have DTS set to 0 for all packets, so assume 0 means unset.
58 pkt
->dts
= dts
? dts
: AV_NOPTS_VALUE
;
61 // The full packet size is stored as part of the packet.
62 size_t len
= AV_RB16(pkt
->data
+ 1);
64 if ((ret
= av_append_packet(s
->pb
, pkt
, len
)) < 0)
71 static int sup_probe(AVProbeData
*p
)
73 unsigned char *buf
= p
->buf
;
74 size_t buf_size
= p
->buf_size
;
77 for (nb_packets
= 0; nb_packets
< 10; nb_packets
++) {
78 size_t full_packet_size
;
79 if (buf_size
< 10 + 3)
81 if (AV_RB16(buf
) != SUP_PGS_MAGIC
)
83 full_packet_size
= AV_RB16(buf
+ 10 + 1) + 10 + 3;
84 if (buf_size
< full_packet_size
)
86 buf
+= full_packet_size
;
87 buf_size
-= full_packet_size
;
92 return AVPROBE_SCORE_RETRY
/ 2;
94 return AVPROBE_SCORE_RETRY
;
96 return AVPROBE_SCORE_EXTENSION
;
97 return AVPROBE_SCORE_MAX
;
100 AVInputFormat ff_sup_demuxer
= {
102 .long_name
= NULL_IF_CONFIG_SMALL("raw HDMV Presentation Graphic Stream subtitles"),
104 .mime_type
= "application/x-pgs",
105 .read_probe
= sup_probe
,
106 .read_header
= sup_read_header
,
107 .read_packet
= sup_read_packet
,
108 .flags
= AVFMT_GENERIC_INDEX
,