yading@11: /* yading@11: * iLBC storage file format yading@11: * Copyright (c) 2012 Martin Storsjo yading@11: * yading@11: * This file is part of FFmpeg. yading@11: * yading@11: * FFmpeg is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * FFmpeg is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with FFmpeg; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: yading@11: #include "avformat.h" yading@11: #include "internal.h" yading@11: yading@11: static const char mode20_header[] = "#!iLBC20\n"; yading@11: static const char mode30_header[] = "#!iLBC30\n"; yading@11: yading@11: static int ilbc_write_header(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: AVCodecContext *enc; yading@11: yading@11: if (s->nb_streams != 1) { yading@11: av_log(s, AV_LOG_ERROR, "Unsupported number of streams\n"); yading@11: return AVERROR(EINVAL); yading@11: } yading@11: enc = s->streams[0]->codec; yading@11: yading@11: if (enc->codec_id != AV_CODEC_ID_ILBC) { yading@11: av_log(s, AV_LOG_ERROR, "Unsupported codec\n"); yading@11: return AVERROR(EINVAL); yading@11: } yading@11: yading@11: if (enc->block_align == 50) { yading@11: avio_write(pb, mode30_header, sizeof(mode30_header) - 1); yading@11: } else if (enc->block_align == 38) { yading@11: avio_write(pb, mode20_header, sizeof(mode20_header) - 1); yading@11: } else { yading@11: av_log(s, AV_LOG_ERROR, "Unsupported mode\n"); yading@11: return AVERROR(EINVAL); yading@11: } yading@11: avio_flush(pb); yading@11: return 0; yading@11: } yading@11: yading@11: static int ilbc_write_packet(AVFormatContext *s, AVPacket *pkt) yading@11: { yading@11: avio_write(s->pb, pkt->data, pkt->size); yading@11: return 0; yading@11: } yading@11: yading@11: static int ilbc_probe(AVProbeData *p) yading@11: { yading@11: // Only check for "#!iLBC" which matches both formats yading@11: if (!memcmp(p->buf, mode20_header, 6)) yading@11: return AVPROBE_SCORE_MAX; yading@11: else yading@11: return 0; yading@11: } yading@11: yading@11: static int ilbc_read_header(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: AVStream *st; yading@11: uint8_t header[9]; yading@11: yading@11: avio_read(pb, header, 9); yading@11: yading@11: st = avformat_new_stream(s, NULL); yading@11: if (!st) yading@11: return AVERROR(ENOMEM); yading@11: st->codec->codec_id = AV_CODEC_ID_ILBC; yading@11: st->codec->sample_rate = 8000; yading@11: st->codec->channels = 1; yading@11: st->codec->codec_type = AVMEDIA_TYPE_AUDIO; yading@11: st->start_time = 0; yading@11: avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate); yading@11: if (!memcmp(header, mode20_header, sizeof(mode20_header) - 1)) { yading@11: st->codec->block_align = 38; yading@11: st->codec->bit_rate = 15200; yading@11: } else if (!memcmp(header, mode30_header, sizeof(mode30_header) - 1)) { yading@11: st->codec->block_align = 50; yading@11: st->codec->bit_rate = 13333; yading@11: } else { yading@11: av_log(s, AV_LOG_ERROR, "Unrecognized iLBC file header\n"); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: static int ilbc_read_packet(AVFormatContext *s, yading@11: AVPacket *pkt) yading@11: { yading@11: AVCodecContext *enc = s->streams[0]->codec; yading@11: int ret; yading@11: yading@11: if ((ret = av_new_packet(pkt, enc->block_align)) < 0) yading@11: return ret; yading@11: yading@11: pkt->stream_index = 0; yading@11: pkt->pos = avio_tell(s->pb); yading@11: pkt->duration = enc->block_align == 38 ? 160 : 240; yading@11: if ((ret = avio_read(s->pb, pkt->data, enc->block_align)) != enc->block_align) { yading@11: av_free_packet(pkt); yading@11: return ret < 0 ? ret : AVERROR(EIO); yading@11: } yading@11: yading@11: return 0; yading@11: } yading@11: yading@11: AVInputFormat ff_ilbc_demuxer = { yading@11: .name = "ilbc", yading@11: .long_name = NULL_IF_CONFIG_SMALL("iLBC storage"), yading@11: .read_probe = ilbc_probe, yading@11: .read_header = ilbc_read_header, yading@11: .read_packet = ilbc_read_packet, yading@11: .flags = AVFMT_GENERIC_INDEX, yading@11: }; yading@11: yading@11: AVOutputFormat ff_ilbc_muxer = { yading@11: .name = "ilbc", yading@11: .long_name = NULL_IF_CONFIG_SMALL("iLBC storage"), yading@11: .mime_type = "audio/iLBC", yading@11: .extensions = "lbc", yading@11: .audio_codec = AV_CODEC_ID_ILBC, yading@11: .write_header = ilbc_write_header, yading@11: .write_packet = ilbc_write_packet, yading@11: .flags = AVFMT_NOTIMESTAMPS, yading@11: };