1/* 2 * Sony OpenMG (OMA) muxer 3 * 4 * Copyright (c) 2011 Michael Karcher 5 * 6 * This file is part of FFmpeg. 7 * 8 * FFmpeg 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 * FFmpeg 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 FFmpeg; if not, write to the Free Software 20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 21 */ 22 23#include "avformat.h" 24#include "avio_internal.h" 25#include "id3v2.h" 26#include "internal.h" 27#include "oma.h" 28#include "rawenc.h" 29 30static av_cold int oma_write_header(AVFormatContext *s) 31{ 32 AVCodecParameters *par; 33 int srate_index; 34 int isjointstereo; 35 36 par = s->streams[0]->codecpar; 37 /* check for support of the format first */ 38 39 for (srate_index = 0; ; srate_index++) { 40 if (ff_oma_srate_tab[srate_index] == 0) { 41 av_log(s, AV_LOG_ERROR, "Sample rate %d not supported in OpenMG audio\n", 42 par->sample_rate); 43 return AVERROR(EINVAL); 44 } 45 46 if (ff_oma_srate_tab[srate_index] * 100 == par->sample_rate) 47 break; 48 } 49 50 /* Metadata; OpenMG does not support ID3v2.4 */ 51 ff_id3v2_write_simple(s, 3, ID3v2_EA3_MAGIC); 52 53 ffio_wfourcc(s->pb, "EA3\0"); 54 avio_w8(s->pb, EA3_HEADER_SIZE >> 7); 55 avio_w8(s->pb, EA3_HEADER_SIZE & 0x7F); 56 avio_wl16(s->pb, 0xFFFF); /* not encrypted */ 57 ffio_fill(s->pb, 0, 6 * 4); /* Padding + DRM id */ 58 59 switch (par->codec_tag) { 60 case OMA_CODECID_ATRAC3: 61 if (par->ch_layout.nb_channels != 2) { 62 av_log(s, AV_LOG_ERROR, "ATRAC3 in OMA is only supported with 2 channels\n"); 63 return AVERROR(EINVAL); 64 } 65 if (par->extradata_size == 14) /* WAV format extradata */ 66 isjointstereo = par->extradata[6] != 0; 67 else if(par->extradata_size == 10) /* RM format extradata */ 68 isjointstereo = par->extradata[8] == 0x12; 69 else { 70 av_log(s, AV_LOG_ERROR, "ATRAC3: Unsupported extradata size\n"); 71 return AVERROR(EINVAL); 72 } 73 avio_wb32(s->pb, (OMA_CODECID_ATRAC3 << 24) | 74 (isjointstereo << 17) | 75 (srate_index << 13) | 76 (par->block_align/8)); 77 break; 78 case OMA_CODECID_ATRAC3P: 79 avio_wb32(s->pb, (OMA_CODECID_ATRAC3P << 24) | 80 (srate_index << 13) | 81 (par->ch_layout.nb_channels << 10) | 82 (par->block_align/8 - 1)); 83 break; 84 default: 85 av_log(s, AV_LOG_ERROR, "unsupported codec tag %s for write\n", 86 av_fourcc2str(par->codec_tag)); 87 return AVERROR(EINVAL); 88 } 89 ffio_fill(s->pb, 0, EA3_HEADER_SIZE - 36); /* Padding */ 90 91 return 0; 92} 93 94const AVOutputFormat ff_oma_muxer = { 95 .name = "oma", 96 .long_name = NULL_IF_CONFIG_SMALL("Sony OpenMG audio"), 97 .mime_type = "audio/x-oma", 98 .extensions = "oma", 99 .audio_codec = AV_CODEC_ID_ATRAC3, 100 .write_header = oma_write_header, 101 .write_packet = ff_raw_write_packet, 102 .codec_tag = ff_oma_codec_tags_list, 103 .flags = AVFMT_NOTIMESTAMPS, 104}; 105