rsodec: set channel layout
[ffmpeg.git] / libavformat / rsodec.c
1 /*
2  * RSO demuxer
3  * Copyright (c) 2001 Fabrice Bellard (original AU code)
4  * Copyright (c) 2010 Rafael Carre
5  *
6  * This file is part of Libav.
7  *
8  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 #include "libavutil/channel_layout.h"
24 #include "libavutil/intreadwrite.h"
25 #include "avformat.h"
26 #include "internal.h"
27 #include "pcm.h"
28 #include "riff.h"
29 #include "rso.h"
30
31 static int rso_read_header(AVFormatContext *s)
32 {
33     AVIOContext *pb = s->pb;
34     int id, rate, bps;
35     unsigned int size;
36     enum AVCodecID codec;
37     AVStream *st;
38
39     id   = avio_rb16(pb);
40     size = avio_rb16(pb);
41     rate = avio_rb16(pb);
42     avio_rb16(pb);   /* play mode ? (0x0000 = don't loop) */
43
44     codec = ff_codec_get_id(ff_codec_rso_tags, id);
45
46     if (codec == AV_CODEC_ID_ADPCM_IMA_WAV) {
47         av_log(s, AV_LOG_ERROR, "ADPCM in RSO not implemented\n");
48         return AVERROR_PATCHWELCOME;
49     }
50
51     bps = av_get_bits_per_sample(codec);
52     if (!bps) {
53         av_log_ask_for_sample(s, "could not determine bits per sample\n");
54         return AVERROR_INVALIDDATA;
55     }
56
57     /* now we are ready: build format streams */
58     st = avformat_new_stream(s, NULL);
59     if (!st)
60         return AVERROR(ENOMEM);
61
62     st->duration            = (size * 8) / bps;
63     st->codec->codec_type   = AVMEDIA_TYPE_AUDIO;
64     st->codec->codec_tag    = id;
65     st->codec->codec_id     = codec;
66     st->codec->channels     = 1;
67     st->codec->channel_layout = AV_CH_LAYOUT_MONO;
68     st->codec->sample_rate  = rate;
69
70     avpriv_set_pts_info(st, 64, 1, rate);
71
72     return 0;
73 }
74
75 #define BLOCK_SIZE 1024 /* in samples */
76
77 static int rso_read_packet(AVFormatContext *s, AVPacket *pkt)
78 {
79     int bps = av_get_bits_per_sample(s->streams[0]->codec->codec_id);
80     int ret = av_get_packet(s->pb, pkt, BLOCK_SIZE * bps >> 3);
81
82     if (ret < 0)
83         return ret;
84
85     pkt->stream_index = 0;
86
87     /* note: we need to modify the packet size here to handle the last packet */
88     pkt->size = ret;
89
90     return 0;
91 }
92
93 AVInputFormat ff_rso_demuxer = {
94     .name           =   "rso",
95     .long_name      =   NULL_IF_CONFIG_SMALL("Lego Mindstorms RSO"),
96     .extensions     =   "rso",
97     .read_header    =   rso_read_header,
98     .read_packet    =   rso_read_packet,
99     .read_seek      =   ff_pcm_read_seek,
100     .codec_tag      =   (const AVCodecTag* const []){ff_codec_rso_tags, 0},
101 };