git.sesse.net Git - ffmpeg/blob - libavformat/pcmdec.c

   1 /*
   2  * RAW PCM demuxers
   3  * Copyright (c) 2002 Fabrice Bellard
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * FFmpeg is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with FFmpeg; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20  */
  21
  22 #include "libavutil/avstring.h"
  23 #include "avformat.h"
  24 #include "internal.h"
  25 #include "pcm.h"
  26 #include "libavutil/log.h"
  27 #include "libavutil/opt.h"
  28 #include "libavutil/avassert.h"
  29
  30 typedef struct PCMAudioDemuxerContext {
  31     AVClass *class;
  32     int sample_rate;
  33     int channels;
  34 } PCMAudioDemuxerContext;
  35
  36 static int pcm_read_header(AVFormatContext *s)
  37 {
  38     PCMAudioDemuxerContext *s1 = s->priv_data;
  39     AVCodecParameters *par;
  40     AVStream *st;
  41     uint8_t *mime_type = NULL;
  42
  43     st = avformat_new_stream(s, NULL);
  44     if (!st)
  45         return AVERROR(ENOMEM);
  46     par = st->codecpar;
  47
  48     par->codec_type  = AVMEDIA_TYPE_AUDIO;
  49     par->codec_id    = s->iformat->raw_codec_id;
  50     par->sample_rate = s1->sample_rate;
  51     par->channels    = s1->channels;
  52
  53     av_opt_get(s->pb, "mime_type", AV_OPT_SEARCH_CHILDREN, &mime_type);
  54     if (mime_type && s->iformat->mime_type) {
  55         int rate = 0, channels = 0, little_endian = 0;
  56         const char *options;
  57         if (av_stristart(mime_type, s->iformat->mime_type, &options)) { /* audio/L16 */
  58             while (options = strchr(options, ';')) {
  59                 options++;
  60                 if (!rate)
  61                     sscanf(options, " rate=%d",     &rate);
  62                 if (!channels)
  63                     sscanf(options, " channels=%d", &channels);
  64                 if (!little_endian) {
  65                      char val[sizeof("little-endian")];
  66                      if (sscanf(options, " endianness=%13s", val) == 1) {
  67                          little_endian = strcmp(val, "little-endian") == 0;
  68                      }
  69                 }
  70             }
  71             if (rate <= 0) {
  72                 av_log(s, AV_LOG_ERROR,
  73                        "Invalid sample_rate found in mime_type \"%s\"\n",
  74                        mime_type);
  75                 av_freep(&mime_type);
  76                 return AVERROR_INVALIDDATA;
  77             }
  78             par->sample_rate = rate;
  79             if (channels > 0)
  80                 par->channels = channels;
  81             if (little_endian)
  82                 par->codec_id = AV_CODEC_ID_PCM_S16LE;
  83         }
  84     }
  85     av_freep(&mime_type);
  86
  87     par->bits_per_coded_sample = av_get_bits_per_sample(par->codec_id);
  88
  89     av_assert0(par->bits_per_coded_sample > 0);
  90
  91     par->block_align = par->bits_per_coded_sample * par->channels / 8;
  92
  93     avpriv_set_pts_info(st, 64, 1, par->sample_rate);
  94     return 0;
  95 }
  96
  97 static const AVOption pcm_options[] = {
  98     { "sample_rate", "", offsetof(PCMAudioDemuxerContext, sample_rate), AV_OPT_TYPE_INT, {.i64 = 44100}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
  99     { "channels",    "", offsetof(PCMAudioDemuxerContext, channels),    AV_OPT_TYPE_INT, {.i64 = 1}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
 100     { NULL },
 101 };
 102
 103 #define PCMDEF_0(name_, long_name_, ext, codec, ...)
 104 #define PCMDEF_1(name_, long_name_, ext, codec, ...)        \
 105 static const AVClass name_ ## _demuxer_class = {            \
 106     .class_name = #name_ " demuxer",                        \
 107     .item_name  = av_default_item_name,                     \
 108     .option     = pcm_options,                              \
 109     .version    = LIBAVUTIL_VERSION_INT,                    \
 110 };                                                          \
 111 const AVInputFormat ff_pcm_ ## name_ ## _demuxer = {        \
 112     .name           = #name_,                               \
 113     .long_name      = NULL_IF_CONFIG_SMALL(long_name_),     \
 114     .priv_data_size = sizeof(PCMAudioDemuxerContext),       \
 115     .read_header    = pcm_read_header,                      \
 116     .read_packet    = ff_pcm_read_packet,                   \
 117     .read_seek      = ff_pcm_read_seek,                     \
 118     .flags          = AVFMT_GENERIC_INDEX,                  \
 119     .extensions     = ext,                                  \
 120     .raw_codec_id   = codec,                                \
 121     .priv_class     = &name_ ## _demuxer_class,             \
 122     __VA_ARGS__                                             \
 123 };
 124 #define PCMDEF_2(name, long_name, ext, codec, enabled, ...) \
 125     PCMDEF_ ## enabled(name, long_name, ext, codec, __VA_ARGS__)
 126 #define PCMDEF_3(name, long_name, ext, codec, config, ...)  \
 127     PCMDEF_2(name, long_name, ext, codec, config,   __VA_ARGS__)
 128 #define PCMDEF_EXT(name, long_name, ext, uppercase, ...)    \
 129     PCMDEF_3(name, long_name, ext, AV_CODEC_ID_PCM_ ## uppercase, \
 130              CONFIG_PCM_ ## uppercase ## _DEMUXER,  __VA_ARGS__)
 131 #define PCMDEF(name, long_name, ext, uppercase)             \
 132     PCMDEF_EXT(name, long_name, ext, uppercase, )
 133
 134 PCMDEF(f64be, "PCM 64-bit floating-point big-endian",           NULL, F64BE)
 135 PCMDEF(f64le, "PCM 64-bit floating-point little-endian",        NULL, F64LE)
 136 PCMDEF(f32be, "PCM 32-bit floating-point big-endian",           NULL, F32BE)
 137 PCMDEF(f32le, "PCM 32-bit floating-point little-endian",        NULL, F32LE)
 138 PCMDEF(s32be, "PCM signed 32-bit big-endian",                   NULL, S32BE)
 139 PCMDEF(s32le, "PCM signed 32-bit little-endian",                NULL, S32LE)
 140 PCMDEF(s24be, "PCM signed 24-bit big-endian",                   NULL, S24BE)
 141 PCMDEF(s24le, "PCM signed 24-bit little-endian",                NULL, S24LE)
 142 PCMDEF_EXT(s16be, "PCM signed 16-bit big-endian",
 143            AV_NE("sw", NULL), S16BE, .mime_type = "audio/L16")
 144 PCMDEF(s16le, "PCM signed 16-bit little-endian",   AV_NE(NULL, "sw"), S16LE)
 145 PCMDEF(s8,    "PCM signed 8-bit",                               "sb",    S8)
 146 PCMDEF(u32be, "PCM unsigned 32-bit big-endian",                 NULL, U32BE)
 147 PCMDEF(u32le, "PCM unsigned 32-bit little-endian",              NULL, U32LE)
 148 PCMDEF(u24be, "PCM unsigned 24-bit big-endian",                 NULL, U24BE)
 149 PCMDEF(u24le, "PCM unsigned 24-bit little-endian",              NULL, U24LE)
 150 PCMDEF(u16be, "PCM unsigned 16-bit big-endian",    AV_NE("uw", NULL), U16BE)
 151 PCMDEF(u16le, "PCM unsigned 16-bit little-endian", AV_NE(NULL, "uw"), U16LE)
 152 PCMDEF(u8,    "PCM unsigned 8-bit",                             "ub",    U8)
 153 PCMDEF(alaw,  "PCM A-law",                                      "al",  ALAW)
 154 PCMDEF(mulaw, "PCM mu-law",                                     "ul", MULAW)
 155 PCMDEF(vidc,  "PCM Archimedes VIDC",                            NULL,  VIDC)
 156
 157 #if CONFIG_SLN_DEMUXER
 158 static const AVOption sln_options[] = {
 159     { "sample_rate", "", offsetof(PCMAudioDemuxerContext, sample_rate), AV_OPT_TYPE_INT, {.i64 = 8000}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
 160     { "channels",    "", offsetof(PCMAudioDemuxerContext, channels),    AV_OPT_TYPE_INT, {.i64 = 1}, 0, INT_MAX, AV_OPT_FLAG_DECODING_PARAM },
 161     { NULL },
 162 };
 163
 164 static const AVClass sln_demuxer_class = {
 165     .class_name = "sln demuxer",
 166     .item_name  = av_default_item_name,
 167     .option     = sln_options,
 168     .version    = LIBAVUTIL_VERSION_INT,
 169 };
 170
 171 const AVInputFormat ff_sln_demuxer = {
 172     .name           = "sln",
 173     .long_name      = NULL_IF_CONFIG_SMALL("Asterisk raw pcm"),
 174     .priv_data_size = sizeof(PCMAudioDemuxerContext),
 175     .read_header    = pcm_read_header,
 176     .read_packet    = ff_pcm_read_packet,
 177     .read_seek      = ff_pcm_read_seek,
 178     .flags          = AVFMT_GENERIC_INDEX,
 179     .extensions     = "sln",
 180     .raw_codec_id   = AV_CODEC_ID_PCM_S16LE,
 181     .priv_class     = &sln_demuxer_class,
 182 };
 183 #endif