2 * QCP format (.qcp) demuxer
3 * Copyright (c) 2009 Kenan Gillet
5 * This file is part of FFmpeg.
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.
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.
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
24 * QCP format (.qcp) demuxer
25 * @author Kenan Gillet
26 * @see RFC 3625: "The QCP File Format and Media Types for Speech Data"
27 * http://tools.ietf.org/html/rfc3625
30 #include "libavutil/channel_layout.h"
31 #include "libavutil/intreadwrite.h"
36 uint32_t data_size
; ///< size of data chunk
38 #define QCP_MAX_MODE 4
39 int16_t rates_per_mode
[QCP_MAX_MODE
+1]; ///< contains the packet size corresponding
40 ///< to each mode, -1 if no size.
44 * Last 15 out of 16 bytes of QCELP-13K GUID, as stored in the file;
45 * the first byte of the GUID can be either 0x41 or 0x42.
47 static const uint8_t guid_qcelp_13k_part
[15] = {
48 0x6d, 0x7f, 0x5e, 0x15, 0xb1, 0xd0, 0x11, 0xba,
49 0x91, 0x00, 0x80, 0x5f, 0xb4, 0xb9, 0x7e
53 * EVRC GUID as stored in the file
55 static const uint8_t guid_evrc
[16] = {
56 0x8d, 0xd4, 0x89, 0xe6, 0x76, 0x90, 0xb5, 0x46,
57 0x91, 0xef, 0x73, 0x6a, 0x51, 0x00, 0xce, 0xb4
61 * SMV GUID as stored in the file
63 static const uint8_t guid_smv
[16] = {
64 0x75, 0x2b, 0x7c, 0x8d, 0x97, 0xa7, 0x49, 0xed,
65 0x98, 0x5e, 0xd5, 0x3c, 0x8c, 0xc7, 0x5f, 0x84
69 * @param guid contains at least 16 bytes
70 * @return 1 if the guid is a qcelp_13k guid, 0 otherwise
72 static int is_qcelp_13k_guid(const uint8_t *guid
) {
73 return (guid
[0] == 0x41 || guid
[0] == 0x42)
74 && !memcmp(guid
+1, guid_qcelp_13k_part
, sizeof(guid_qcelp_13k_part
));
77 static int qcp_probe(AVProbeData
*pd
)
79 if (AV_RL32(pd
->buf
) == AV_RL32("RIFF") &&
80 AV_RL64(pd
->buf
+8) == AV_RL64("QLCMfmt "))
81 return AVPROBE_SCORE_MAX
;
85 static int qcp_read_header(AVFormatContext
*s
)
87 AVIOContext
*pb
= s
->pb
;
88 QCPContext
*c
= s
->priv_data
;
89 AVStream
*st
= avformat_new_stream(s
, NULL
);
94 return AVERROR(ENOMEM
);
96 avio_rb32(pb
); // "RIFF"
97 avio_skip(pb
, 4 + 8 + 4 + 1 + 1); // filesize + "QLCMfmt " + chunk-size + major-version + minor-version
99 st
->codec
->codec_type
= AVMEDIA_TYPE_AUDIO
;
100 st
->codec
->channels
= 1;
101 st
->codec
->channel_layout
= AV_CH_LAYOUT_MONO
;
102 avio_read(pb
, buf
, 16);
103 if (is_qcelp_13k_guid(buf
)) {
104 st
->codec
->codec_id
= AV_CODEC_ID_QCELP
;
105 } else if (!memcmp(buf
, guid_evrc
, 16)) {
106 st
->codec
->codec_id
= AV_CODEC_ID_EVRC
;
107 } else if (!memcmp(buf
, guid_smv
, 16)) {
108 st
->codec
->codec_id
= AV_CODEC_ID_SMV
;
110 av_log(s
, AV_LOG_ERROR
, "Unknown codec GUID "FF_PRI_GUID
".\n",
112 return AVERROR_INVALIDDATA
;
114 avio_skip(pb
, 2 + 80); // codec-version + codec-name
115 st
->codec
->bit_rate
= avio_rl16(pb
);
117 s
->packet_size
= avio_rl16(pb
);
118 avio_skip(pb
, 2); // block-size
119 st
->codec
->sample_rate
= avio_rl16(pb
);
120 avio_skip(pb
, 2); // sample-size
122 memset(c
->rates_per_mode
, -1, sizeof(c
->rates_per_mode
));
123 nb_rates
= avio_rl32(pb
);
124 nb_rates
= FFMIN(nb_rates
, 8);
125 for (i
=0; i
<nb_rates
; i
++) {
126 int size
= avio_r8(pb
);
127 int mode
= avio_r8(pb
);
128 if (mode
> QCP_MAX_MODE
) {
129 av_log(s
, AV_LOG_WARNING
, "Unknown entry %d=>%d in rate-map-table.\n ", mode
, size
);
131 c
->rates_per_mode
[mode
] = size
;
133 avio_skip(pb
, 16 - 2*nb_rates
+ 20); // empty entries of rate-map-table + reserved
138 static int qcp_read_packet(AVFormatContext
*s
, AVPacket
*pkt
)
140 AVIOContext
*pb
= s
->pb
;
141 QCPContext
*c
= s
->priv_data
;
142 unsigned int chunk_size
, tag
;
144 while(!avio_feof(pb
)) {
146 int pkt_size
, ret
, mode
= avio_r8(pb
);
148 if (s
->packet_size
) {
149 pkt_size
= s
->packet_size
- 1;
150 } else if (mode
> QCP_MAX_MODE
|| (pkt_size
= c
->rates_per_mode
[mode
]) < 0) {
155 if (c
->data_size
<= pkt_size
) {
156 av_log(s
, AV_LOG_WARNING
, "Data chunk is too small.\n");
157 pkt_size
= c
->data_size
- 1;
160 if ((ret
= av_get_packet(pb
, pkt
, pkt_size
)) >= 0) {
162 av_log(s
, AV_LOG_ERROR
, "Packet size is too small.\n");
164 c
->data_size
-= pkt_size
+ 1;
169 if (avio_tell(pb
) & 1 && avio_r8(pb
))
170 av_log(s
, AV_LOG_WARNING
, "Padding should be 0.\n");
173 chunk_size
= avio_rl32(pb
);
175 case MKTAG('v', 'r', 'a', 't'):
176 if (avio_rl32(pb
)) // var-rate-flag
178 avio_skip(pb
, 4); // size-in-packets
180 case MKTAG('d', 'a', 't', 'a'):
181 c
->data_size
= chunk_size
;
185 avio_skip(pb
, chunk_size
);
191 AVInputFormat ff_qcp_demuxer
= {
193 .long_name
= NULL_IF_CONFIG_SMALL("QCP"),
194 .priv_data_size
= sizeof(QCPContext
),
195 .read_probe
= qcp_probe
,
196 .read_header
= qcp_read_header
,
197 .read_packet
= qcp_read_packet
,