yading@11: /*
yading@11:  * Sony OpenMG (OMA) muxer
yading@11:  *
yading@11:  * Copyright (c) 2011 Michael Karcher
yading@11:  *
yading@11:  * This file is part of Libav.
yading@11:  *
yading@11:  * Libav 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:  * Libav 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 Libav; 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 "avio_internal.h"
yading@11: #include "id3v2.h"
yading@11: #include "internal.h"
yading@11: #include "oma.h"
yading@11: #include "rawenc.h"
yading@11: 
yading@11: static av_cold int oma_write_header(AVFormatContext *s)
yading@11: {
yading@11:     int i;
yading@11:     AVCodecContext *format;
yading@11:     int srate_index;
yading@11:     int isjointstereo;
yading@11: 
yading@11:     format = s->streams[0]->codec;
yading@11:     /* check for support of the format first */
yading@11: 
yading@11:     for (srate_index = 0; ; srate_index++) {
yading@11:         if (ff_oma_srate_tab[srate_index] == 0) {
yading@11:             av_log(s, AV_LOG_ERROR, "Sample rate %d not supported in OpenMG audio\n",
yading@11:                    format->sample_rate);
yading@11:             return AVERROR(EINVAL);
yading@11:         }
yading@11: 
yading@11:         if (ff_oma_srate_tab[srate_index] * 100 == format->sample_rate)
yading@11:             break;
yading@11:     }
yading@11: 
yading@11:     /* Metadata; OpenMG does not support ID3v2.4 */
yading@11:     ff_id3v2_write_simple(s, 3, ID3v2_EA3_MAGIC);
yading@11: 
yading@11:     ffio_wfourcc(s->pb, "EA3\0");
yading@11:     avio_w8(s->pb, EA3_HEADER_SIZE >> 7);
yading@11:     avio_w8(s->pb, EA3_HEADER_SIZE & 0x7F);
yading@11:     avio_wl16(s->pb, 0xFFFF);       /* not encrypted */
yading@11:     for (i = 0; i < 6; i++)
yading@11:         avio_wl32(s->pb, 0);        /* Padding + DRM id */
yading@11: 
yading@11:     switch(format->codec_tag) {
yading@11:     case OMA_CODECID_ATRAC3:
yading@11:         if (format->channels != 2) {
yading@11:             av_log(s, AV_LOG_ERROR, "ATRAC3 in OMA is only supported with 2 channels\n");
yading@11:             return AVERROR(EINVAL);
yading@11:         }
yading@11:         if (format->extradata_size == 14) /* WAV format extradata */
yading@11:             isjointstereo = format->extradata[6] != 0;
yading@11:         else if(format->extradata_size == 10) /* RM format extradata */
yading@11:             isjointstereo = format->extradata[8] == 0x12;
yading@11:         else {
yading@11:             av_log(s, AV_LOG_ERROR, "ATRAC3: Unsupported extradata size\n");
yading@11:             return AVERROR(EINVAL);
yading@11:         }
yading@11:         avio_wb32(s->pb, (OMA_CODECID_ATRAC3 << 24) |
yading@11:                          (isjointstereo << 17) |
yading@11:                          (srate_index << 13) |
yading@11:                          (format->block_align/8));
yading@11:         break;
yading@11:     case OMA_CODECID_ATRAC3P:
yading@11:         avio_wb32(s->pb, (OMA_CODECID_ATRAC3P << 24) |
yading@11:                          (srate_index << 13) |
yading@11:                          (format->channels << 10) |
yading@11:                          (format->block_align/8 - 1));
yading@11:         break;
yading@11:     default:
yading@11:         av_log(s, AV_LOG_ERROR, "unsupported codec tag %d for write\n",
yading@11:                format->codec_tag);
yading@11:         return AVERROR(EINVAL);
yading@11:     }
yading@11:     for (i = 0; i < (EA3_HEADER_SIZE - 36)/4; i++)
yading@11:         avio_wl32(s->pb, 0);        /* Padding */
yading@11: 
yading@11:     return 0;
yading@11: }
yading@11: 
yading@11: AVOutputFormat ff_oma_muxer = {
yading@11:     .name              = "oma",
yading@11:     .long_name         = NULL_IF_CONFIG_SMALL("Sony OpenMG audio"),
yading@11:     .mime_type         = "audio/x-oma",
yading@11:     .extensions        = "oma",
yading@11:     .audio_codec       = AV_CODEC_ID_ATRAC3,
yading@11:     .write_header      = oma_write_header,
yading@11:     .write_packet      = ff_raw_write_packet,
yading@11:     .codec_tag         = (const AVCodecTag* const []){ff_oma_codec_tags, 0},
yading@11: };