ast: Rename ast.c -> astdec.c
[ffmpeg.git] / libavformat / astdec.c
1 /*
2  * AST demuxer
3  * Copyright (c) 2012 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/channel_layout.h"
23 #include "libavutil/intreadwrite.h"
24 #include "avformat.h"
25 #include "internal.h"
26
27 static int ast_probe(AVProbeData *p)
28 {
29     if (AV_RL32(p->buf) == MKTAG('S','T','R','M') &&
30         AV_RB16(p->buf + 10) &&
31         AV_RB16(p->buf + 12) &&
32         AV_RB32(p->buf + 16))
33         return AVPROBE_SCORE_MAX / 3 * 2;
34     return 0;
35 }
36
37 static int ast_read_header(AVFormatContext *s)
38 {
39     int codec, depth;
40     AVStream *st;
41
42     st = avformat_new_stream(s, NULL);
43     if (!st)
44         return AVERROR(ENOMEM);
45
46     avio_skip(s->pb, 8);
47     codec = avio_rb16(s->pb);
48     switch (codec) {
49     case 0:
50         st->codec->codec_id = AV_CODEC_ID_ADPCM_AFC;
51         break;
52     case 1:
53         st->codec->codec_id = AV_CODEC_ID_PCM_S16BE_PLANAR;
54         break;
55     default:
56         av_log(s, AV_LOG_ERROR, "unsupported codec %d\n", codec);
57     }
58
59     depth = avio_rb16(s->pb);
60     if (depth != 16) {
61         av_log_ask_for_sample(s, "unsupported depth %d\n", depth);
62         return AVERROR_INVALIDDATA;
63     }
64
65     st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
66     st->codec->channels = avio_rb16(s->pb);
67     if (!st->codec->channels)
68         return AVERROR_INVALIDDATA;
69
70     if (st->codec->channels == 2)
71         st->codec->channel_layout = AV_CH_LAYOUT_STEREO;
72     else if (st->codec->channels == 4)
73         st->codec->channel_layout = AV_CH_LAYOUT_4POINT0;
74
75     avio_skip(s->pb, 2);
76     st->codec->sample_rate = avio_rb32(s->pb);
77     if (st->codec->sample_rate <= 0)
78         return AVERROR_INVALIDDATA;
79     st->start_time         = 0;
80     st->duration           = avio_rb32(s->pb);
81     avio_skip(s->pb, 40);
82     avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate);
83
84     return 0;
85 }
86
87 static int ast_read_packet(AVFormatContext *s, AVPacket *pkt)
88 {
89     uint32_t type, size;
90     int64_t pos;
91     int ret;
92
93     if (url_feof(s->pb))
94         return AVERROR_EOF;
95
96     pos  = avio_tell(s->pb);
97     type = avio_rl32(s->pb);
98     size = avio_rb32(s->pb);
99     if (size > INT_MAX / s->streams[0]->codec->channels)
100         return AVERROR_INVALIDDATA;
101
102     size *= s->streams[0]->codec->channels;
103     if ((ret = avio_skip(s->pb, 24)) < 0) // padding
104         return ret;
105
106     if (type == MKTAG('B','L','C','K')) {
107         ret = av_get_packet(s->pb, pkt, size);
108         pkt->stream_index = 0;
109         pkt->pos = pos;
110     } else {
111         av_log(s, AV_LOG_ERROR, "unknown chunk %x\n", type);
112         avio_skip(s->pb, size);
113         ret = AVERROR_INVALIDDATA;
114     }
115
116     return ret;
117 }
118
119 AVInputFormat ff_ast_demuxer = {
120     .name           = "ast",
121     .long_name      = NULL_IF_CONFIG_SMALL("AST (Audio Stream)"),
122     .read_probe     = ast_probe,
123     .read_header    = ast_read_header,
124     .read_packet    = ast_read_packet,
125     .extensions     = "ast",
126     .flags          = AVFMT_GENERIC_INDEX,
127 };