3 * Copyright (c) 2005 Vidar Madsen
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
22 #include "libavutil/channel_layout.h"
24 #include "avio_internal.h"
31 int64_t atrpos
, atsqpos
, awapos
;
36 static const int mmf_rates
[] = { 4000, 8000, 11025, 22050, 44100 };
38 static int mmf_rate(int code
)
40 if ((code
< 0) || (code
> 4))
42 return mmf_rates
[code
];
46 static int mmf_rate_code(int rate
)
49 for (i
= 0; i
< 5; i
++)
50 if (mmf_rates
[i
] == rate
)
55 /* Copy of end_tag() from avienc.c, but for big-endian chunk size */
56 static void end_tag_be(AVIOContext
*pb
, int64_t start
)
61 avio_seek(pb
, start
- 4, SEEK_SET
);
62 avio_wb32(pb
, (uint32_t)(pos
- start
));
63 avio_seek(pb
, pos
, SEEK_SET
);
66 static int mmf_write_header(AVFormatContext
*s
)
68 MMFContext
*mmf
= s
->priv_data
;
69 AVIOContext
*pb
= s
->pb
;
72 const char *version
= s
->flags
& AVFMT_FLAG_BITEXACT
?
74 "VN:"LIBAVFORMAT_IDENT
",";
76 rate
= mmf_rate_code(s
->streams
[0]->codec
->sample_rate
);
78 av_log(s
, AV_LOG_ERROR
, "Unsupported sample rate %d, supported are 4000, 8000, 11025, 22050 and 44100\n",
79 s
->streams
[0]->codec
->sample_rate
);
80 return AVERROR(EINVAL
);
83 mmf
->stereo
= s
->streams
[0]->codec
->channels
> 1;
85 s
->strict_std_compliance
> FF_COMPLIANCE_EXPERIMENTAL
) {
86 av_log(s
, AV_LOG_ERROR
, "Yamaha SMAF stereo is experimental, "
87 "add '-strict %d' if you want to use it.\n",
88 FF_COMPLIANCE_EXPERIMENTAL
);
89 return AVERROR(EINVAL
);
92 ffio_wfourcc(pb
, "MMMD");
94 pos
= ff_start_tag(pb
, "CNTI");
95 avio_w8(pb
, 0); /* class */
96 avio_w8(pb
, 1); /* type */
97 avio_w8(pb
, 1); /* code type */
98 avio_w8(pb
, 0); /* status */
99 avio_w8(pb
, 0); /* counts */
102 pos
= ff_start_tag(pb
, "OPDA");
103 avio_write(pb
, version
, strlen(version
)); /* metadata ("ST:songtitle,VN:version,...") */
106 avio_write(pb
, "ATR\x00", 4);
108 mmf
->atrpos
= avio_tell(pb
);
109 avio_w8(pb
, 0); /* format type */
110 avio_w8(pb
, 0); /* sequence type */
111 avio_w8(pb
, (mmf
->stereo
<< 7) | (1 << 4) | rate
); /* (channel << 7) | (format << 4) | rate */
112 avio_w8(pb
, 0); /* wave base bit */
113 avio_w8(pb
, 2); /* time base d */
114 avio_w8(pb
, 2); /* time base g */
116 ffio_wfourcc(pb
, "Atsq");
118 mmf
->atsqpos
= avio_tell(pb
);
119 /* Will be filled on close */
120 avio_write(pb
, "\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00", 16);
122 mmf
->awapos
= ff_start_tag(pb
, "Awa\x01");
124 avpriv_set_pts_info(s
->streams
[0], 64, 1, s
->streams
[0]->codec
->sample_rate
);
131 /* Write a variable-length symbol */
132 static void put_varlength(AVIOContext
*pb
, int val
)
138 avio_w8(pb
, 0x80 | val
>> 7);
139 avio_w8(pb
, 0x7f & val
);
143 static int mmf_write_trailer(AVFormatContext
*s
)
145 AVIOContext
*pb
= s
->pb
;
146 MMFContext
*mmf
= s
->priv_data
;
150 if (s
->pb
->seekable
) {
151 /* Fill in length fields */
152 end_tag_be(pb
, mmf
->awapos
);
153 end_tag_be(pb
, mmf
->atrpos
);
157 size
= pos
- mmf
->awapos
;
159 /* Fill Atsq chunk */
160 avio_seek(pb
, mmf
->atsqpos
, SEEK_SET
);
163 avio_w8(pb
, 0); /* start time */
164 avio_w8(pb
, (mmf
->stereo
<< 6) | 1); /* (channel << 6) | wavenum */
165 gatetime
= size
* 500 / s
->streams
[0]->codec
->sample_rate
;
166 put_varlength(pb
, gatetime
); /* duration */
169 put_varlength(pb
, gatetime
); /* start time */
170 avio_write(pb
, "\xff\x00", 2); /* nop */
172 /* "end of sequence" */
173 avio_write(pb
, "\x00\x00\x00\x00", 4);
175 avio_seek(pb
, pos
, SEEK_SET
);
181 #endif /* CONFIG_MMF_MUXER */
183 static int mmf_probe(AVProbeData
*p
)
185 /* check file header */
186 if (p
->buf
[0] == 'M' && p
->buf
[1] == 'M' &&
187 p
->buf
[2] == 'M' && p
->buf
[3] == 'D' &&
188 p
->buf
[8] == 'C' && p
->buf
[9] == 'N' &&
189 p
->buf
[10] == 'T' && p
->buf
[11] == 'I')
190 return AVPROBE_SCORE_MAX
;
196 static int mmf_read_header(AVFormatContext
*s
)
198 MMFContext
*mmf
= s
->priv_data
;
200 AVIOContext
*pb
= s
->pb
;
206 if (tag
!= MKTAG('M', 'M', 'M', 'D'))
207 return AVERROR_INVALIDDATA
;
208 avio_skip(pb
, 4); /* file_size */
210 /* Skip some unused chunks that may or may not be present */
211 for (;; avio_skip(pb
, size
)) {
213 size
= avio_rb32(pb
);
214 if (tag
== MKTAG('C', 'N', 'T', 'I'))
216 if (tag
== MKTAG('O', 'P', 'D', 'A'))
221 /* Tag = "ATRx", where "x" = track number */
222 if ((tag
& 0xffffff) == MKTAG('M', 'T', 'R', 0)) {
223 av_log(s
, AV_LOG_ERROR
, "MIDI like format found, unsupported\n");
224 return AVERROR_PATCHWELCOME
;
226 if ((tag
& 0xffffff) != MKTAG('A', 'T', 'R', 0)) {
227 av_log(s
, AV_LOG_ERROR
, "Unsupported SMAF chunk %08x\n", tag
);
228 return AVERROR_PATCHWELCOME
;
231 avio_r8(pb
); /* format type */
232 avio_r8(pb
); /* sequence type */
233 params
= avio_r8(pb
); /* (channel << 7) | (format << 4) | rate */
234 rate
= mmf_rate(params
& 0x0f);
236 av_log(s
, AV_LOG_ERROR
, "Invalid sample rate\n");
237 return AVERROR_INVALIDDATA
;
239 avio_r8(pb
); /* wave base bit */
240 avio_r8(pb
); /* time base d */
241 avio_r8(pb
); /* time base g */
243 /* Skip some unused chunks that may or may not be present */
244 for (;; avio_skip(pb
, size
)) {
246 size
= avio_rb32(pb
);
247 if (tag
== MKTAG('A', 't', 's', 'q'))
249 if (tag
== MKTAG('A', 's', 'p', 'I'))
254 /* Make sure it's followed by an Awa chunk, aka wave data */
255 if ((tag
& 0xffffff) != MKTAG('A', 'w', 'a', 0)) {
256 av_log(s
, AV_LOG_ERROR
, "Unexpected SMAF chunk %08x\n", tag
);
257 return AVERROR_INVALIDDATA
;
259 mmf
->data_end
= avio_tell(pb
) + size
;
261 st
= avformat_new_stream(s
, NULL
);
263 return AVERROR(ENOMEM
);
265 st
->codec
->codec_type
= AVMEDIA_TYPE_AUDIO
;
266 st
->codec
->codec_id
= AV_CODEC_ID_ADPCM_YAMAHA
;
267 st
->codec
->sample_rate
= rate
;
268 st
->codec
->channels
= (params
>> 7) + 1;
269 st
->codec
->channel_layout
= params
>> 7 ? AV_CH_LAYOUT_STEREO
: AV_CH_LAYOUT_MONO
;
270 st
->codec
->bits_per_coded_sample
= 4;
271 st
->codec
->bit_rate
= st
->codec
->sample_rate
*
272 st
->codec
->bits_per_coded_sample
;
274 avpriv_set_pts_info(st
, 64, 1, st
->codec
->sample_rate
);
279 #define MAX_SIZE 4096
281 static int mmf_read_packet(AVFormatContext
*s
, AVPacket
*pkt
)
283 MMFContext
*mmf
= s
->priv_data
;
287 left
= mmf
->data_end
- avio_tell(s
->pb
);
288 size
= FFMIN(left
, MAX_SIZE
);
289 if (avio_feof(s
->pb
) || size
<= 0)
292 ret
= av_get_packet(s
->pb
, pkt
, size
);
296 pkt
->stream_index
= 0;
301 #if CONFIG_MMF_DEMUXER
302 AVInputFormat ff_mmf_demuxer
= {
304 .long_name
= NULL_IF_CONFIG_SMALL("Yamaha SMAF"),
305 .priv_data_size
= sizeof(MMFContext
),
306 .read_probe
= mmf_probe
,
307 .read_header
= mmf_read_header
,
308 .read_packet
= mmf_read_packet
,
309 .flags
= AVFMT_GENERIC_INDEX
,
314 AVOutputFormat ff_mmf_muxer
= {
316 .long_name
= NULL_IF_CONFIG_SMALL("Yamaha SMAF"),
317 .mime_type
= "application/vnd.smaf",
319 .priv_data_size
= sizeof(MMFContext
),
320 .audio_codec
= AV_CODEC_ID_ADPCM_YAMAHA
,
321 .video_codec
= AV_CODEC_ID_NONE
,
322 .write_header
= mmf_write_header
,
323 .write_packet
= ff_raw_write_packet
,
324 .write_trailer
= mmf_write_trailer
,