Initial commit of the MD studio demuxer, not hooked up yet
[ffmpeg.git] / libavformat / aea.c
1 /*
2  * MD STUDIO audio demuxer
3  *
4  * Copyright (c) 2009 Benjamin Larsson
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 /**
24  * @file libavformat/aea.c
25  */
26
27 #include "avformat.h"
28 #include "raw.h"
29 #include "libavutil/intreadwrite.h"
30
31 #define AT1_SU_SIZE     212
32
33 static int aea_read_probe(AVProbeData *p)
34 {
35     if (p->buf_size <= 2048+212)
36         return 0;
37
38     /* Magic is '00 08 00 00' in Little Endian*/
39     if(AV_RL32(p->buf)==0x800) {
40         int bsm_s, bsm_e, inb_s, inb_e;
41         bsm_s = p->buf[2048];
42         inb_s = p->buf[2048+1];
43         inb_e = p->buf[2048+210];
44         bsm_e = p->buf[2048+211];
45
46
47         /* Check so that the redundant bsm bytes and info bytes are valid
48          * the block size mode bytes have to be the same
49          * the info bytes have to be the same
50          * the block size mode and info byte can't be the same
51          */
52         if (bsm_s == bsm_e && inb_s == inb_e && bsm_s != inb_s)
53             return AVPROBE_SCORE_MAX/2;
54     }
55     return 0;
56 }
57
58 static int aea_read_header(AVFormatContext *s,
59                            AVFormatParameters *ap)
60 {
61     AVStream *st = av_new_stream(s, 0);
62     if (!st)
63         return AVERROR(ENOMEM);
64
65     /* Parse the amount of channels and skip to pos 2048(0x800) */
66     url_fskip(s->pb, 264);
67     st->codec->channels = get_byte(s->pb);
68     url_fskip(s->pb, 1783);
69
70
71     st->codec->codec_type     = CODEC_TYPE_AUDIO;
72     st->codec->codec_id       = CODEC_ID_ATRAC1;
73     st->codec->sample_rate    = 44100;
74     st->codec->bit_rate       = 292000;
75
76     if (st->codec->channels != 1 && st->codec->channels != 2) {
77         av_log(s,AV_LOG_ERROR,"Channels %d not supported!\n",st->codec->channels);
78         return -1;
79     }
80
81     st->codec->channel_layout = (st->codec->channels==1) ? CH_LAYOUT_MONO : CH_LAYOUT_STEREO;
82
83     st->codec->block_align = AT1_SU_SIZE * st->codec->channels;
84     return 0;
85 }
86
87 static int aea_read_packet(AVFormatContext *s, AVPacket *pkt)
88 {
89     int ret = av_get_packet(s->pb, pkt, s->streams[0]->codec->block_align);
90
91     pkt->stream_index = 0;
92     if (ret <= 0)
93         return AVERROR(EIO);
94
95     return ret;
96 }
97
98 AVInputFormat aea_demuxer = {
99     "aea",
100     NULL_IF_CONFIG_SMALL("MD STUDIO audio"),
101     0,
102     aea_read_probe,
103     aea_read_header,
104     aea_read_packet,
105     0,
106     pcm_read_seek,
107     .flags= AVFMT_GENERIC_INDEX,
108     .extensions = "aea",
109 };
110