X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavformat%2Fnuv.c;h=492f68f493024bb4634dcee462c67200f47f2331;hb=79fdbfdb3e50f3f906903e027714ee04c1a00e89;hp=2d6e93a79cdea4123afc21fca6ca71549343facb;hpb=ac2b2226ac2c5e151bcf0760948a775329ef21b3;p=ffmpeg diff --git a/libavformat/nuv.c b/libavformat/nuv.c index 2d6e93a79cd..492f68f4930 100644 --- a/libavformat/nuv.c +++ b/libavformat/nuv.c @@ -2,42 +2,53 @@ * NuppelVideo demuxer. * Copyright (c) 2006 Reimar Doeffinger * - * This file is part of FFmpeg. + * This file is part of Libav. * - * FFmpeg is free software; you can redistribute it and/or + * Libav 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, + * Libav 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 + * License along with Libav; if not, write to the Free Software * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ +#include "libavutil/channel_layout.h" +#include "libavutil/imgutils.h" #include "libavutil/intreadwrite.h" +#include "libavutil/intfloat.h" #include "avformat.h" +#include "internal.h" #include "riff.h" -typedef struct { +static const AVCodecTag nuv_audio_tags[] = { + { AV_CODEC_ID_PCM_S16LE, MKTAG('R', 'A', 'W', 'A') }, + { AV_CODEC_ID_MP3, MKTAG('L', 'A', 'M', 'E') }, + { AV_CODEC_ID_NONE, 0 }, +}; + +typedef struct NUVContext { int v_id; int a_id; int rtjpg_video; } NUVContext; typedef enum { - NUV_VIDEO = 'V', + NUV_VIDEO = 'V', NUV_EXTRADATA = 'D', - NUV_AUDIO = 'A', - NUV_SEEKP = 'R', - NUV_MYTHEXT = 'X' + NUV_AUDIO = 'A', + NUV_SEEKP = 'R', + NUV_MYTHEXT = 'X' } nuv_frametype; -static int nuv_probe(AVProbeData *p) { +static int nuv_probe(AVProbeData *p) +{ if (!memcmp(p->buf, "NuppelVideo", 12)) return AVPROBE_SCORE_MAX; if (!memcmp(p->buf, "MythTVVideo", 12)) @@ -45,219 +56,273 @@ static int nuv_probe(AVProbeData *p) { return 0; } -//! little macro to sanitize packet size +/// little macro to sanitize packet size #define PKTSIZE(s) (s & 0xffffff) /** - * \brief read until we found all data needed for decoding - * \param vst video stream of which to change parameters - * \param ast video stream of which to change parameters - * \param myth set if this is a MythTVVideo format file - * \return 1 if all required codec data was found + * @brief read until we found all data needed for decoding + * @param vst video stream of which to change parameters + * @param ast video stream of which to change parameters + * @param myth set if this is a MythTVVideo format file + * @return 0 or AVERROR code */ -static int get_codec_data(ByteIOContext *pb, AVStream *vst, - AVStream *ast, int myth) { +static int get_codec_data(AVIOContext *pb, AVStream *vst, + AVStream *ast, int myth) +{ nuv_frametype frametype; + if (!vst && !myth) return 1; // no codec data needed - while (!url_feof(pb)) { + while (!pb->eof_reached) { int size, subtype; - frametype = get_byte(pb); + + frametype = avio_r8(pb); switch (frametype) { - case NUV_EXTRADATA: - subtype = get_byte(pb); - url_fskip(pb, 6); - size = PKTSIZE(get_le32(pb)); - if (vst && subtype == 'R') { - vst->codec->extradata_size = size; - vst->codec->extradata = av_malloc(size); - get_buffer(pb, vst->codec->extradata, size); - size = 0; - if (!myth) - return 1; + case NUV_EXTRADATA: + subtype = avio_r8(pb); + avio_skip(pb, 6); + size = PKTSIZE(avio_rl32(pb)); + if (vst && subtype == 'R') { + if (vst->codecpar->extradata) { + av_freep(&vst->codecpar->extradata); + vst->codecpar->extradata_size = 0; } + vst->codecpar->extradata = av_malloc(size); + if (!vst->codecpar->extradata) + return AVERROR(ENOMEM); + vst->codecpar->extradata_size = size; + avio_read(pb, vst->codecpar->extradata, size); + size = 0; + if (!myth) + return 0; + } + break; + case NUV_MYTHEXT: + avio_skip(pb, 7); + size = PKTSIZE(avio_rl32(pb)); + if (size != 128 * 4) break; - case NUV_MYTHEXT: - url_fskip(pb, 7); - size = PKTSIZE(get_le32(pb)); - if (size != 128 * 4) - break; - get_le32(pb); // version - if (vst) { - vst->codec->codec_tag = get_le32(pb); - vst->codec->codec_id = - codec_get_id(codec_bmp_tags, vst->codec->codec_tag); - if (vst->codec->codec_tag == MKTAG('R', 'J', 'P', 'G')) - vst->codec->codec_id = CODEC_ID_NUV; - } else - url_fskip(pb, 4); - - if (ast) { - ast->codec->codec_tag = get_le32(pb); - ast->codec->sample_rate = get_le32(pb); - ast->codec->bits_per_coded_sample = get_le32(pb); - ast->codec->channels = get_le32(pb); - ast->codec->codec_id = - wav_codec_get_id(ast->codec->codec_tag, - ast->codec->bits_per_coded_sample); - ast->need_parsing = AVSTREAM_PARSE_FULL; - } else - url_fskip(pb, 4 * 4); - - size -= 6 * 4; - url_fskip(pb, size); - return 1; - case NUV_SEEKP: - size = 11; - break; - default: - url_fskip(pb, 7); - size = PKTSIZE(get_le32(pb)); - break; + avio_rl32(pb); // version + if (vst) { + vst->codecpar->codec_tag = avio_rl32(pb); + vst->codecpar->codec_id = + ff_codec_get_id(ff_codec_bmp_tags, vst->codecpar->codec_tag); + if (vst->codecpar->codec_tag == MKTAG('R', 'J', 'P', 'G')) + vst->codecpar->codec_id = AV_CODEC_ID_NUV; + } else + avio_skip(pb, 4); + + if (ast) { + int id; + + ast->codecpar->codec_tag = avio_rl32(pb); + ast->codecpar->sample_rate = avio_rl32(pb); + ast->codecpar->bits_per_coded_sample = avio_rl32(pb); + ast->codecpar->channels = avio_rl32(pb); + ast->codecpar->channel_layout = 0; + + id = ff_wav_codec_get_id(ast->codecpar->codec_tag, + ast->codecpar->bits_per_coded_sample); + if (id == AV_CODEC_ID_NONE) { + id = ff_codec_get_id(nuv_audio_tags, ast->codecpar->codec_tag); + if (id == AV_CODEC_ID_PCM_S16LE) + id = ff_get_pcm_codec_id(ast->codecpar->bits_per_coded_sample, + 0, 0, ~1); + } + ast->codecpar->codec_id = id; + + ast->need_parsing = AVSTREAM_PARSE_FULL; + } else + avio_skip(pb, 4 * 4); + + size -= 6 * 4; + avio_skip(pb, size); + return 0; + case NUV_SEEKP: + size = 11; + break; + default: + avio_skip(pb, 7); + size = PKTSIZE(avio_rl32(pb)); + break; } - url_fskip(pb, size); + avio_skip(pb, size); } + return 0; } -static int nuv_header(AVFormatContext *s, AVFormatParameters *ap) { +static int nuv_header(AVFormatContext *s) +{ NUVContext *ctx = s->priv_data; - ByteIOContext *pb = s->pb; + AVIOContext *pb = s->pb; char id_string[12]; double aspect, fps; - int is_mythtv, width, height, v_packs, a_packs; - int stream_nr = 0; + int is_mythtv, width, height, v_packs, a_packs, ret; AVStream *vst = NULL, *ast = NULL; - get_buffer(pb, id_string, 12); + + avio_read(pb, id_string, 12); is_mythtv = !memcmp(id_string, "MythTVVideo", 12); - url_fskip(pb, 5); // version string - url_fskip(pb, 3); // padding - width = get_le32(pb); - height = get_le32(pb); - get_le32(pb); // unused, "desiredwidth" - get_le32(pb); // unused, "desiredheight" - get_byte(pb); // 'P' == progressive, 'I' == interlaced - url_fskip(pb, 3); // padding - aspect = av_int2dbl(get_le64(pb)); + avio_skip(pb, 5); // version string + avio_skip(pb, 3); // padding + width = avio_rl32(pb); + height = avio_rl32(pb); + avio_rl32(pb); // unused, "desiredwidth" + avio_rl32(pb); // unused, "desiredheight" + avio_r8(pb); // 'P' == progressive, 'I' == interlaced + avio_skip(pb, 3); // padding + aspect = av_int2double(avio_rl64(pb)); if (aspect > 0.9999 && aspect < 1.0001) aspect = 4.0 / 3.0; - fps = av_int2dbl(get_le64(pb)); + fps = av_int2double(avio_rl64(pb)); + if (fps < 0.0f) { + if (s->error_recognition & AV_EF_EXPLODE) { + av_log(s, AV_LOG_ERROR, "Invalid frame rate %f\n", fps); + return AVERROR_INVALIDDATA; + } else { + av_log(s, AV_LOG_WARNING, "Invalid frame rate %f, setting to 0.\n", fps); + fps = 0.0f; + } + } // number of packets per stream type, -1 means unknown, e.g. streaming - v_packs = get_le32(pb); - a_packs = get_le32(pb); - get_le32(pb); // text + v_packs = avio_rl32(pb); + a_packs = avio_rl32(pb); + avio_rl32(pb); // text - get_le32(pb); // keyframe distance (?) + avio_rl32(pb); // keyframe distance (?) if (v_packs) { - ctx->v_id = stream_nr++; - vst = av_new_stream(s, ctx->v_id); + vst = avformat_new_stream(s, NULL); if (!vst) return AVERROR(ENOMEM); - vst->codec->codec_type = CODEC_TYPE_VIDEO; - vst->codec->codec_id = CODEC_ID_NUV; - vst->codec->width = width; - vst->codec->height = height; - vst->codec->bits_per_coded_sample = 10; - vst->sample_aspect_ratio = av_d2q(aspect * height / width, 10000); - vst->r_frame_rate = av_d2q(fps, 60000); - av_set_pts_info(vst, 32, 1, 1000); + ctx->v_id = vst->index; + + ret = av_image_check_size(width, height, 0, s); + if (ret < 0) + return ret; + + vst->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; + vst->codecpar->codec_id = AV_CODEC_ID_NUV; + vst->codecpar->width = width; + vst->codecpar->height = height; + vst->codecpar->bits_per_coded_sample = 10; + vst->sample_aspect_ratio = av_d2q(aspect * height / width, + 10000); + vst->avg_frame_rate = av_d2q(fps, 60000); + avpriv_set_pts_info(vst, 32, 1, 1000); } else ctx->v_id = -1; if (a_packs) { - ctx->a_id = stream_nr++; - ast = av_new_stream(s, ctx->a_id); + ast = avformat_new_stream(s, NULL); if (!ast) return AVERROR(ENOMEM); - ast->codec->codec_type = CODEC_TYPE_AUDIO; - ast->codec->codec_id = CODEC_ID_PCM_S16LE; - ast->codec->channels = 2; - ast->codec->sample_rate = 44100; - ast->codec->bit_rate = 2 * 2 * 44100 * 8; - ast->codec->block_align = 2 * 2; - ast->codec->bits_per_coded_sample = 16; - av_set_pts_info(ast, 32, 1, 1000); + ctx->a_id = ast->index; + + ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; + ast->codecpar->codec_id = AV_CODEC_ID_PCM_S16LE; + ast->codecpar->channels = 2; + ast->codecpar->channel_layout = AV_CH_LAYOUT_STEREO; + ast->codecpar->sample_rate = 44100; + ast->codecpar->bit_rate = 2 * 2 * 44100 * 8; + ast->codecpar->block_align = 2 * 2; + ast->codecpar->bits_per_coded_sample = 16; + avpriv_set_pts_info(ast, 32, 1, 1000); } else ctx->a_id = -1; - get_codec_data(pb, vst, ast, is_mythtv); - ctx->rtjpg_video = vst && vst->codec->codec_id == CODEC_ID_NUV; + if ((ret = get_codec_data(pb, vst, ast, is_mythtv)) < 0) + return ret; + + ctx->rtjpg_video = vst && vst->codecpar->codec_id == AV_CODEC_ID_NUV; + return 0; } #define HDRSIZE 12 -static int nuv_packet(AVFormatContext *s, AVPacket *pkt) { +static int nuv_packet(AVFormatContext *s, AVPacket *pkt) +{ NUVContext *ctx = s->priv_data; - ByteIOContext *pb = s->pb; + AVIOContext *pb = s->pb; uint8_t hdr[HDRSIZE]; nuv_frametype frametype; int ret, size; - while (!url_feof(pb)) { + + while (!pb->eof_reached) { int copyhdrsize = ctx->rtjpg_video ? HDRSIZE : 0; - uint64_t pos = url_ftell(pb); - ret = get_buffer(pb, hdr, HDRSIZE); + uint64_t pos = avio_tell(pb); + + ret = avio_read(pb, hdr, HDRSIZE); if (ret < HDRSIZE) return ret < 0 ? ret : AVERROR(EIO); + frametype = hdr[0]; - size = PKTSIZE(AV_RL32(&hdr[8])); + size = PKTSIZE(AV_RL32(&hdr[8])); + switch (frametype) { - case NUV_EXTRADATA: - if (!ctx->rtjpg_video) { - url_fskip(pb, size); - break; - } - case NUV_VIDEO: - if (ctx->v_id < 0) { - av_log(s, AV_LOG_ERROR, "Video packet in file without video stream!\n"); - url_fskip(pb, size); - break; - } - ret = av_new_packet(pkt, copyhdrsize + size); - if (ret < 0) - return ret; - // HACK: we have no idea if it is a keyframe, - // but if we mark none seeking will not work at all. - pkt->flags |= PKT_FLAG_KEY; - pkt->pos = pos; - pkt->pts = AV_RL32(&hdr[4]); - pkt->stream_index = ctx->v_id; - memcpy(pkt->data, hdr, copyhdrsize); - ret = get_buffer(pb, pkt->data + copyhdrsize, size); + case NUV_EXTRADATA: + if (!ctx->rtjpg_video) { + avio_skip(pb, size); + break; + } + case NUV_VIDEO: + if (ctx->v_id < 0) { + av_log(s, AV_LOG_ERROR, "Video packet in file without video stream!\n"); + avio_skip(pb, size); + break; + } + ret = av_new_packet(pkt, copyhdrsize + size); + if (ret < 0) return ret; - case NUV_AUDIO: - if (ctx->a_id < 0) { - av_log(s, AV_LOG_ERROR, "Audio packet in file without audio stream!\n"); - url_fskip(pb, size); - break; - } - ret = av_get_packet(pb, pkt, size); - pkt->flags |= PKT_FLAG_KEY; - pkt->pos = pos; - pkt->pts = AV_RL32(&hdr[4]); - pkt->stream_index = ctx->a_id; + // HACK: we have no idea if it is a keyframe, + // but if we mark none seeking will not work at all. + pkt->flags |= AV_PKT_FLAG_KEY; + pkt->pos = pos; + pkt->pts = AV_RL32(&hdr[4]); + pkt->stream_index = ctx->v_id; + memcpy(pkt->data, hdr, copyhdrsize); + ret = avio_read(pb, pkt->data + copyhdrsize, size); + if (ret < 0) { + av_packet_unref(pkt); return ret; - case NUV_SEEKP: - // contains no data, size value is invalid - break; - default: - url_fskip(pb, size); + } + if (ret < size) + av_shrink_packet(pkt, copyhdrsize + ret); + return 0; + case NUV_AUDIO: + if (ctx->a_id < 0) { + av_log(s, AV_LOG_ERROR, "Audio packet in file without audio stream!\n"); + avio_skip(pb, size); break; + } + ret = av_get_packet(pb, pkt, size); + pkt->flags |= AV_PKT_FLAG_KEY; + pkt->pos = pos; + pkt->pts = AV_RL32(&hdr[4]); + pkt->stream_index = ctx->a_id; + if (ret < 0) + return ret; + return 0; + case NUV_SEEKP: + // contains no data, size value is invalid + break; + default: + avio_skip(pb, size); + break; } } + return AVERROR(EIO); } -AVInputFormat nuv_demuxer = { - "nuv", - NULL_IF_CONFIG_SMALL("NuppelVideo format"), - sizeof(NUVContext), - nuv_probe, - nuv_header, - nuv_packet, - NULL, - NULL, - .flags = AVFMT_GENERIC_INDEX, +AVInputFormat ff_nuv_demuxer = { + .name = "nuv", + .long_name = NULL_IF_CONFIG_SMALL("NuppelVideo"), + .priv_data_size = sizeof(NUVContext), + .read_probe = nuv_probe, + .read_header = nuv_header, + .read_packet = nuv_packet, + .flags = AVFMT_GENERIC_INDEX, };