1 /*
2 * SDR2 demuxer
3 * Copyright (c) 2014 Paul B Mahol
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include "libavutil/intreadwrite.h"
23 #include "avformat.h"
24 #include "internal.h"
25
sdr2_probe(const AVProbeData * p)26 static int sdr2_probe(const AVProbeData *p)
27 {
28 if (AV_RL32(p->buf) != MKTAG('S', 'R', 'A', 1))
29 return 0;
30
31 return AVPROBE_SCORE_EXTENSION;
32 }
33
34 #define FIRST 0xA8
35
sdr2_read_header(AVFormatContext * s)36 static int sdr2_read_header(AVFormatContext *s)
37 {
38 AVStream *st, *ast;
39
40 ast = avformat_new_stream(s, 0);
41 if (!ast)
42 return AVERROR(ENOMEM);
43
44 st = avformat_new_stream(s, 0);
45 if (!st)
46 return AVERROR(ENOMEM);
47
48 avio_skip(s->pb, 20);
49 avpriv_set_pts_info(st, 64, 1, avio_rl32(s->pb));
50 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
51 st->codecpar->width = avio_rl32(s->pb);
52 st->codecpar->height = avio_rl32(s->pb);
53 st->codecpar->codec_id = AV_CODEC_ID_H264;
54 ffstream(st)->need_parsing = AVSTREAM_PARSE_FULL;
55
56 ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
57 ast->codecpar->ch_layout.nb_channels = 1;
58 ast->codecpar->sample_rate = 8000;
59 ast->codecpar->codec_id = AV_CODEC_ID_PCM_S16LE;
60 avpriv_set_pts_info(ast, 64, 1, 8000);
61
62 avio_seek(s->pb, FIRST, SEEK_SET);
63
64 return 0;
65 }
66
67 static const uint8_t header[24] = {
68 0x00, 0x00, 0x00, 0x01, 0x67, 0x42, 0x00, 0x1e,
69 0xa6, 0x80, 0xb0, 0x7e, 0x40, 0x00, 0x00, 0x00,
70 0x01, 0x68, 0xce, 0x38, 0x80, 0x00, 0x00, 0x00
71 };
72
sdr2_read_packet(AVFormatContext * s,AVPacket * pkt)73 static int sdr2_read_packet(AVFormatContext *s, AVPacket *pkt)
74 {
75 int64_t pos;
76 unsigned next;
77 int flags, ret = 0, is_video;
78
79 pos = avio_tell(s->pb);
80
81 flags = avio_rl32(s->pb);
82 avio_skip(s->pb, 4);
83
84 next = avio_rl32(s->pb);
85 if (next <= 52)
86 return AVERROR_INVALIDDATA;
87
88 avio_skip(s->pb, 6);
89 is_video = avio_rl32(s->pb);
90 avio_skip(s->pb, 30);
91
92 if (pos == FIRST) {
93 if ((ret = av_new_packet(pkt, next - 52 + 24)) < 0)
94 return ret;
95 memcpy(pkt->data, header, 24);
96 ret = avio_read(s->pb, pkt->data + 24, next - 52);
97 if (ret < 0) {
98 return ret;
99 }
100 av_shrink_packet(pkt, ret + 24);
101 } else {
102 ret = av_get_packet(s->pb, pkt, next - 52);
103 }
104 pkt->stream_index = !!is_video;
105 pkt->pos = pos;
106 if (flags & (1 << 12))
107 pkt->flags |= AV_PKT_FLAG_KEY;
108
109 return ret;
110 }
111
112 const AVInputFormat ff_sdr2_demuxer = {
113 .name = "sdr2",
114 .long_name = NULL_IF_CONFIG_SMALL("SDR2"),
115 .read_probe = sdr2_probe,
116 .read_header = sdr2_read_header,
117 .read_packet = sdr2_read_packet,
118 .extensions = "sdr2",
119 .flags = AVFMT_GENERIC_INDEX,
120 };
121