yading@11: /* yading@11: * Core Audio Format muxer yading@11: * Copyright (c) 2011 Carl Eugen Hoyos 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 "caf.h" yading@11: #include "isom.h" yading@11: #include "avio_internal.h" yading@11: #include "libavutil/intfloat.h" yading@11: #include "libavutil/dict.h" yading@11: yading@11: typedef struct { yading@11: int64_t data; yading@11: uint8_t *pkt_sizes; yading@11: int size_buffer_size; yading@11: int size_entries_used; yading@11: int packets; yading@11: } CAFContext; yading@11: yading@11: static uint32_t codec_flags(enum AVCodecID codec_id) { yading@11: switch (codec_id) { yading@11: case AV_CODEC_ID_PCM_F32BE: yading@11: case AV_CODEC_ID_PCM_F64BE: yading@11: return 1; //< kCAFLinearPCMFormatFlagIsFloat yading@11: case AV_CODEC_ID_PCM_S16LE: yading@11: case AV_CODEC_ID_PCM_S24LE: yading@11: case AV_CODEC_ID_PCM_S32LE: yading@11: return 2; //< kCAFLinearPCMFormatFlagIsLittleEndian yading@11: case AV_CODEC_ID_PCM_F32LE: yading@11: case AV_CODEC_ID_PCM_F64LE: yading@11: return 3; //< kCAFLinearPCMFormatFlagIsFloat | kCAFLinearPCMFormatFlagIsLittleEndian yading@11: default: yading@11: return 0; yading@11: } yading@11: } yading@11: yading@11: static uint32_t samples_per_packet(enum AVCodecID codec_id, int channels) { yading@11: switch (codec_id) { yading@11: case AV_CODEC_ID_PCM_S8: yading@11: case AV_CODEC_ID_PCM_S16LE: yading@11: case AV_CODEC_ID_PCM_S16BE: yading@11: case AV_CODEC_ID_PCM_S24LE: yading@11: case AV_CODEC_ID_PCM_S24BE: yading@11: case AV_CODEC_ID_PCM_S32LE: yading@11: case AV_CODEC_ID_PCM_S32BE: yading@11: case AV_CODEC_ID_PCM_F32LE: yading@11: case AV_CODEC_ID_PCM_F32BE: yading@11: case AV_CODEC_ID_PCM_F64LE: yading@11: case AV_CODEC_ID_PCM_F64BE: yading@11: case AV_CODEC_ID_PCM_ALAW: yading@11: case AV_CODEC_ID_PCM_MULAW: yading@11: return 1; yading@11: case AV_CODEC_ID_MACE3: yading@11: case AV_CODEC_ID_MACE6: yading@11: return 6; yading@11: case AV_CODEC_ID_ADPCM_IMA_QT: yading@11: return 64; yading@11: case AV_CODEC_ID_AMR_NB: yading@11: case AV_CODEC_ID_GSM: yading@11: case AV_CODEC_ID_ILBC: yading@11: case AV_CODEC_ID_QCELP: yading@11: return 160; yading@11: case AV_CODEC_ID_GSM_MS: yading@11: return 320; yading@11: case AV_CODEC_ID_MP1: yading@11: return 384; yading@11: case AV_CODEC_ID_MP2: yading@11: case AV_CODEC_ID_MP3: yading@11: return 1152; yading@11: case AV_CODEC_ID_AC3: yading@11: return 1536; yading@11: case AV_CODEC_ID_ALAC: yading@11: case AV_CODEC_ID_QDM2: yading@11: return 4096; yading@11: case AV_CODEC_ID_ADPCM_IMA_WAV: yading@11: return (1024 - 4 * channels) * 8 / (4 * channels) + 1; yading@11: case AV_CODEC_ID_ADPCM_MS: yading@11: return (1024 - 7 * channels) * 2 / channels + 2; yading@11: default: yading@11: return 0; yading@11: } yading@11: } yading@11: yading@11: static int caf_write_header(AVFormatContext *s) yading@11: { yading@11: AVIOContext *pb = s->pb; yading@11: AVCodecContext *enc = s->streams[0]->codec; yading@11: CAFContext *caf = s->priv_data; yading@11: AVDictionaryEntry *t = NULL; yading@11: unsigned int codec_tag = ff_codec_get_tag(ff_codec_caf_tags, enc->codec_id); yading@11: int64_t chunk_size = 0; yading@11: yading@11: switch (enc->codec_id) { yading@11: case AV_CODEC_ID_AAC: yading@11: case AV_CODEC_ID_AC3: yading@11: av_log(s, AV_LOG_ERROR, "muxing codec currently unsupported\n"); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: yading@11: switch (enc->codec_id) { yading@11: case AV_CODEC_ID_PCM_S8: yading@11: case AV_CODEC_ID_PCM_S16LE: yading@11: case AV_CODEC_ID_PCM_S16BE: yading@11: case AV_CODEC_ID_PCM_S24LE: yading@11: case AV_CODEC_ID_PCM_S24BE: yading@11: case AV_CODEC_ID_PCM_S32LE: yading@11: case AV_CODEC_ID_PCM_S32BE: yading@11: case AV_CODEC_ID_PCM_F32LE: yading@11: case AV_CODEC_ID_PCM_F32BE: yading@11: case AV_CODEC_ID_PCM_F64LE: yading@11: case AV_CODEC_ID_PCM_F64BE: yading@11: case AV_CODEC_ID_PCM_ALAW: yading@11: case AV_CODEC_ID_PCM_MULAW: yading@11: codec_tag = MKTAG('l','p','c','m'); yading@11: } yading@11: yading@11: if (!codec_tag) { yading@11: av_log(s, AV_LOG_ERROR, "unsupported codec\n"); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: yading@11: if (!enc->block_align && !pb->seekable) { yading@11: av_log(s, AV_LOG_ERROR, "Muxing variable packet size not supported on non seekable output\n"); yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: yading@11: ffio_wfourcc(pb, "caff"); //< mFileType yading@11: avio_wb16(pb, 1); //< mFileVersion yading@11: avio_wb16(pb, 0); //< mFileFlags yading@11: yading@11: ffio_wfourcc(pb, "desc"); //< Audio Description chunk yading@11: avio_wb64(pb, 32); //< mChunkSize yading@11: avio_wb64(pb, av_double2int(enc->sample_rate)); //< mSampleRate yading@11: avio_wl32(pb, codec_tag); //< mFormatID yading@11: avio_wb32(pb, codec_flags(enc->codec_id)); //< mFormatFlags yading@11: avio_wb32(pb, enc->block_align); //< mBytesPerPacket yading@11: avio_wb32(pb, samples_per_packet(enc->codec_id, enc->channels)); //< mFramesPerPacket yading@11: avio_wb32(pb, enc->channels); //< mChannelsPerFrame yading@11: avio_wb32(pb, av_get_bits_per_sample(enc->codec_id)); //< mBitsPerChannel yading@11: yading@11: if (enc->channel_layout) { yading@11: ffio_wfourcc(pb, "chan"); yading@11: avio_wb64(pb, 12); yading@11: ff_mov_write_chan(pb, enc->channel_layout); yading@11: } yading@11: yading@11: if (enc->codec_id == AV_CODEC_ID_ALAC) { yading@11: ffio_wfourcc(pb, "kuki"); yading@11: avio_wb64(pb, 12 + enc->extradata_size); yading@11: avio_write(pb, "\0\0\0\14frmaalac", 12); yading@11: avio_write(pb, enc->extradata, enc->extradata_size); yading@11: } else if (enc->codec_id == AV_CODEC_ID_AMR_NB) { yading@11: ffio_wfourcc(pb, "kuki"); yading@11: avio_wb64(pb, 29); yading@11: avio_write(pb, "\0\0\0\14frmasamr", 12); yading@11: avio_wb32(pb, 0x11); /* size */ yading@11: avio_write(pb, "samrFFMP", 8); yading@11: avio_w8(pb, 0); /* decoder version */ yading@11: yading@11: avio_wb16(pb, 0x81FF); /* Mode set (all modes for AMR_NB) */ yading@11: avio_w8(pb, 0x00); /* Mode change period (no restriction) */ yading@11: avio_w8(pb, 0x01); /* Frames per sample */ yading@11: } else if (enc->codec_id == AV_CODEC_ID_QDM2) { yading@11: ffio_wfourcc(pb, "kuki"); yading@11: avio_wb64(pb, enc->extradata_size); yading@11: avio_write(pb, enc->extradata, enc->extradata_size); yading@11: } yading@11: yading@11: if (av_dict_count(s->metadata)) { yading@11: ffio_wfourcc(pb, "info"); //< Information chunk yading@11: while ((t = av_dict_get(s->metadata, "", t, AV_DICT_IGNORE_SUFFIX))) { yading@11: chunk_size += strlen(t->key) + strlen(t->value) + 2; yading@11: } yading@11: avio_wb64(pb, chunk_size + 4); yading@11: avio_wb32(pb, av_dict_count(s->metadata)); yading@11: t = NULL; yading@11: while ((t = av_dict_get(s->metadata, "", t, AV_DICT_IGNORE_SUFFIX))) { yading@11: avio_put_str(pb, t->key); yading@11: avio_put_str(pb, t->value); yading@11: } yading@11: } yading@11: yading@11: ffio_wfourcc(pb, "data"); //< Audio Data chunk yading@11: caf->data = avio_tell(pb); yading@11: avio_wb64(pb, -1); //< mChunkSize yading@11: avio_wb32(pb, 0); //< mEditCount yading@11: yading@11: avio_flush(pb); yading@11: return 0; yading@11: } yading@11: yading@11: static int caf_write_packet(AVFormatContext *s, AVPacket *pkt) yading@11: { yading@11: CAFContext *caf = s->priv_data; yading@11: yading@11: avio_write(s->pb, pkt->data, pkt->size); yading@11: if (!s->streams[0]->codec->block_align) { yading@11: void *pkt_sizes = caf->pkt_sizes; yading@11: int i, alloc_size = caf->size_entries_used + 5; yading@11: if (alloc_size < 0) { yading@11: caf->pkt_sizes = NULL; yading@11: } else { yading@11: caf->pkt_sizes = av_fast_realloc(caf->pkt_sizes, yading@11: &caf->size_buffer_size, yading@11: alloc_size); yading@11: } yading@11: if (!caf->pkt_sizes) { yading@11: av_free(pkt_sizes); yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: for (i = 4; i > 0; i--) { yading@11: unsigned top = pkt->size >> i * 7; yading@11: if (top) yading@11: caf->pkt_sizes[caf->size_entries_used++] = 128 | top; yading@11: } yading@11: caf->pkt_sizes[caf->size_entries_used++] = pkt->size & 127; yading@11: caf->packets++; yading@11: } yading@11: return 0; yading@11: } yading@11: yading@11: static int caf_write_trailer(AVFormatContext *s) yading@11: { yading@11: CAFContext *caf = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: AVCodecContext *enc = s->streams[0]->codec; yading@11: yading@11: if (pb->seekable) { yading@11: int64_t file_size = avio_tell(pb); yading@11: yading@11: avio_seek(pb, caf->data, SEEK_SET); yading@11: avio_wb64(pb, file_size - caf->data - 8); yading@11: avio_seek(pb, file_size, SEEK_SET); yading@11: if (!enc->block_align) { yading@11: ffio_wfourcc(pb, "pakt"); yading@11: avio_wb64(pb, caf->size_entries_used + 24); yading@11: avio_wb64(pb, caf->packets); ///< mNumberPackets yading@11: avio_wb64(pb, caf->packets * samples_per_packet(enc->codec_id, enc->channels)); ///< mNumberValidFrames yading@11: avio_wb32(pb, 0); ///< mPrimingFrames yading@11: avio_wb32(pb, 0); ///< mRemainderFrames yading@11: avio_write(pb, caf->pkt_sizes, caf->size_entries_used); yading@11: caf->size_buffer_size = 0; yading@11: } yading@11: avio_flush(pb); yading@11: } yading@11: av_freep(&caf->pkt_sizes); yading@11: return 0; yading@11: } yading@11: yading@11: AVOutputFormat ff_caf_muxer = { yading@11: .name = "caf", yading@11: .long_name = NULL_IF_CONFIG_SMALL("Apple CAF (Core Audio Format)"), yading@11: .mime_type = "audio/x-caf", yading@11: .extensions = "caf", yading@11: .priv_data_size = sizeof(CAFContext), yading@11: .audio_codec = AV_CODEC_ID_PCM_S16BE, yading@11: .video_codec = AV_CODEC_ID_NONE, yading@11: .write_header = caf_write_header, yading@11: .write_packet = caf_write_packet, yading@11: .write_trailer = caf_write_trailer, yading@11: .codec_tag = (const AVCodecTag* const []){ff_codec_caf_tags, 0}, yading@11: };