libavformat/omaenc.c

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