6f352af96c24988928598fbca15991c04ffae9f3
[libav.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 "rso.h"
29
30 static int rso_read_header(AVFormatContext *s)
31 {
32 AVIOContext *pb = s->pb;
33 int id, rate, bps;
34 unsigned int size;
35 enum AVCodecID codec;
36 AVStream *st;
37
38 id = avio_rb16(pb);
39 size = avio_rb16(pb);
40 rate = avio_rb16(pb);
41 avio_rb16(pb); /* play mode ? (0x0000 = don't loop) */
42
43 codec = ff_codec_get_id(ff_codec_rso_tags, id);
44
45 if (codec == AV_CODEC_ID_ADPCM_IMA_WAV) {
46 avpriv_report_missing_feature(s, "ADPCM in RSO");
47 return AVERROR_PATCHWELCOME;
48 }
49
50 bps = av_get_bits_per_sample(codec);
51 if (!bps) {
52 avpriv_request_sample(s, "Unknown bits per sample");
53 return AVERROR_PATCHWELCOME;
54 }
55
56 /* now we are ready: build format streams */
57 st = avformat_new_stream(s, NULL);
58 if (!st)
59 return AVERROR(ENOMEM);
60
61 st->duration = (size * 8) / bps;
62 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
63 st->codecpar->codec_tag = id;
64 st->codecpar->codec_id = codec;
65 st->codecpar->channels = 1;
66 st->codecpar->channel_layout = AV_CH_LAYOUT_MONO;
67 st->codecpar->sample_rate = rate;
68
69 avpriv_set_pts_info(st, 64, 1, rate);
70
71 return 0;
72 }
73
74 #define BLOCK_SIZE 1024 /* in samples */
75
76 static int rso_read_packet(AVFormatContext *s, AVPacket *pkt)
77 {
78 int bps = av_get_bits_per_sample(s->streams[0]->codecpar->codec_id);
79 int ret = av_get_packet(s->pb, pkt, BLOCK_SIZE * bps >> 3);
80
81 if (ret < 0)
82 return ret;
83
84 pkt->stream_index = 0;
85
86 /* note: we need to modify the packet size here to handle the last packet */
87 pkt->size = ret;
88
89 return 0;
90 }
91
92 AVInputFormat ff_rso_demuxer = {
93 .name = "rso",
94 .long_name = NULL_IF_CONFIG_SMALL("Lego Mindstorms RSO"),
95 .extensions = "rso",
96 .read_header = rso_read_header,
97 .read_packet = rso_read_packet,
98 .read_seek = ff_pcm_read_seek,
99 .codec_tag = (const AVCodecTag* const []){ff_codec_rso_tags, 0},
100 };