Imported Debian version 2.5.2~trusty
[deb_ffmpeg.git] / ffmpeg / libavformat / sdr2.c
... / ...
CommitLineData
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
26static int sdr2_probe(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
36static 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->codec->codec_type = AVMEDIA_TYPE_VIDEO;
51 st->codec->width = avio_rl32(s->pb);
52 st->codec->height = avio_rl32(s->pb);
53 st->codec->codec_id = AV_CODEC_ID_H264;
54 st->need_parsing = AVSTREAM_PARSE_FULL;
55
56 ast->codec->codec_type = AVMEDIA_TYPE_AUDIO;
57 ast->codec->channels = 1;
58 ast->codec->sample_rate = 8000;
59 ast->codec->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
67static 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
73static 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 (av_new_packet(pkt, next - 52 + 24) < 0)
94 return AVERROR(ENOMEM);
95 memcpy(pkt->data, header, 24);
96 ret = avio_read(s->pb, pkt->data + 24, next - 52);
97 if (ret < 0) {
98 av_free_packet(pkt);
99 return ret;
100 }
101 av_shrink_packet(pkt, ret + 24);
102 } else {
103 ret = av_get_packet(s->pb, pkt, next - 52);
104 }
105 pkt->stream_index = !!is_video;
106 pkt->pos = pos;
107 if (flags & (1 << 12))
108 pkt->flags |= AV_PKT_FLAG_KEY;
109
110 return ret;
111}
112
113AVInputFormat ff_sdr2_demuxer = {
114 .name = "sdr2",
115 .long_name = NULL_IF_CONFIG_SMALL("SDR2"),
116 .read_probe = sdr2_probe,
117 .read_header = sdr2_read_header,
118 .read_packet = sdr2_read_packet,
119 .extensions = "sdr2",
120 .flags = AVFMT_GENERIC_INDEX,
121};