mirror of https://github.com/FFmpeg/FFmpeg.git
parent
98fed176cf
commit
4f5440adc2
7 changed files with 134 additions and 1 deletions
@ -0,0 +1,128 @@ |
||||
/*
|
||||
* NIST Sphere demuxer |
||||
* Copyright (c) 2012 Paul B Mahol |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "libavutil/avstring.h" |
||||
#include "libavutil/intreadwrite.h" |
||||
#include "avformat.h" |
||||
#include "internal.h" |
||||
#include "pcm.h" |
||||
|
||||
static int nist_probe(AVProbeData *p) |
||||
{ |
||||
if (AV_RL64(p->buf) == AV_RL64("NIST_1A\x0a")) |
||||
return AVPROBE_SCORE_MAX; |
||||
return 0; |
||||
} |
||||
|
||||
static int nist_read_header(AVFormatContext *s) |
||||
{ |
||||
char buffer[32], coding[32] = "pcm", format[32] = "01"; |
||||
int bps = 0, be = 0; |
||||
int32_t header_size; |
||||
AVStream *st; |
||||
|
||||
st = avformat_new_stream(s, NULL); |
||||
if (!st) |
||||
return AVERROR(ENOMEM); |
||||
|
||||
st->codec->codec_type = AVMEDIA_TYPE_AUDIO; |
||||
|
||||
ff_get_line(s->pb, buffer, sizeof(buffer)); |
||||
ff_get_line(s->pb, buffer, sizeof(buffer)); |
||||
sscanf(buffer, "%"SCNd32, &header_size); |
||||
if (header_size <= 0) |
||||
return AVERROR_INVALIDDATA; |
||||
|
||||
while (!url_feof(s->pb)) { |
||||
ff_get_line(s->pb, buffer, sizeof(buffer)); |
||||
|
||||
if (avio_tell(s->pb) >= header_size) |
||||
return AVERROR_INVALIDDATA; |
||||
|
||||
if (!memcmp(buffer, "end_head", 8)) { |
||||
if (!st->codec->bits_per_coded_sample) |
||||
st->codec->bits_per_coded_sample = bps << 3; |
||||
|
||||
if (!av_strcasecmp(coding, "pcm")) { |
||||
st->codec->codec_id = ff_get_pcm_codec_id(st->codec->bits_per_coded_sample, |
||||
0, be, 0xFFFF); |
||||
} else if (!av_strcasecmp(coding, "alaw")) { |
||||
st->codec->codec_id = AV_CODEC_ID_PCM_ALAW; |
||||
} else if (!av_strcasecmp(coding, "ulaw") || |
||||
!av_strcasecmp(coding, "mu-law")) { |
||||
st->codec->codec_id = AV_CODEC_ID_PCM_MULAW; |
||||
} else { |
||||
av_log_ask_for_sample(s, "unsupported coding: %s\n", coding); |
||||
} |
||||
|
||||
avpriv_set_pts_info(st, 64, 1, st->codec->sample_rate); |
||||
|
||||
st->codec->block_align = st->codec->bits_per_coded_sample * st->codec->channels / 8; |
||||
|
||||
if (avio_tell(s->pb) > header_size) |
||||
return AVERROR_INVALIDDATA; |
||||
|
||||
avio_skip(s->pb, header_size - avio_tell(s->pb)); |
||||
|
||||
return 0; |
||||
} else if (!memcmp(buffer, "channel_count", 13)) { |
||||
sscanf(buffer, "%*s %*s %"SCNd32, &st->codec->channels); |
||||
} else if (!memcmp(buffer, "sample_byte_format", 18)) { |
||||
sscanf(buffer, "%*s %*s %31s", format); |
||||
|
||||
if (!av_strcasecmp(format, "01")) { |
||||
be = 0; |
||||
} else if (!av_strcasecmp(format, "10")) { |
||||
be = 1; |
||||
} else if (av_strcasecmp(format, "1")) { |
||||
av_log_ask_for_sample(s, "unsupported sample byte format: %s\n", format); |
||||
return AVERROR_PATCHWELCOME; |
||||
} |
||||
} else if (!memcmp(buffer, "sample_coding", 13)) { |
||||
sscanf(buffer, "%*s %*s %31s", coding); |
||||
} else if (!memcmp(buffer, "sample_count", 12)) { |
||||
sscanf(buffer, "%*s %*s %"SCNd64, &st->duration); |
||||
} else if (!memcmp(buffer, "sample_n_bytes", 14)) { |
||||
sscanf(buffer, "%*s %*s %"SCNd32, &bps); |
||||
} else if (!memcmp(buffer, "sample_rate", 11)) { |
||||
sscanf(buffer, "%*s %*s %"SCNd32, &st->codec->sample_rate); |
||||
} else if (!memcmp(buffer, "sample_sig_bits", 15)) { |
||||
sscanf(buffer, "%*s %*s %"SCNd32, &st->codec->bits_per_coded_sample); |
||||
} else { |
||||
char key[32], value[32]; |
||||
sscanf(buffer, "%31s %*s %31s", key, value); |
||||
av_dict_set(&s->metadata, key, value, AV_DICT_APPEND); |
||||
} |
||||
} |
||||
|
||||
return AVERROR_EOF; |
||||
} |
||||
|
||||
AVInputFormat ff_nistsphere_demuxer = { |
||||
.name = "nistsphere", |
||||
.long_name = NULL_IF_CONFIG_SMALL("NIST SPeech HEader REsources"), |
||||
.read_probe = nist_probe, |
||||
.read_header = nist_read_header, |
||||
.read_packet = ff_pcm_read_packet, |
||||
.read_seek = ff_pcm_read_seek, |
||||
.extensions = "nist,sph", |
||||
.flags = AVFMT_GENERIC_INDEX, |
||||
}; |
Loading…
Reference in new issue