Mercurial > libavformat.hg
view raw.c @ 6439:4053f191ae61 libavformat
move dnxhd demuxer to its own file
author | aurel |
---|---|
date | Sun, 29 Aug 2010 21:51:47 +0000 |
parents | 2facedc9f9d7 |
children | 8beba113f242 |
line wrap: on
line source
/* * RAW muxer and demuxer * Copyright (c) 2001 Fabrice Bellard * Copyright (c) 2005 Alex Beregszaszi * * 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/crc.h" #include "libavcodec/ac3_parser.h" #include "libavcodec/get_bits.h" #include "libavcodec/bytestream.h" #include "avformat.h" #include "raw.h" #include "id3v2.h" #include "id3v1.h" /* simple formats */ #if CONFIG_NULL_MUXER static int null_write_packet(struct AVFormatContext *s, AVPacket *pkt) { return 0; } #endif #if CONFIG_MUXERS int ff_raw_write_packet(AVFormatContext *s, AVPacket *pkt) { put_buffer(s->pb, pkt->data, pkt->size); put_flush_packet(s->pb); return 0; } #endif #if CONFIG_DEMUXERS /* raw input */ static int raw_read_header(AVFormatContext *s, AVFormatParameters *ap) { AVStream *st; enum CodecID id; st = av_new_stream(s, 0); if (!st) return AVERROR(ENOMEM); id = s->iformat->value; if (id == CODEC_ID_RAWVIDEO) { st->codec->codec_type = AVMEDIA_TYPE_VIDEO; } else { st->codec->codec_type = AVMEDIA_TYPE_AUDIO; } st->codec->codec_id = id; switch(st->codec->codec_type) { case AVMEDIA_TYPE_AUDIO: st->codec->sample_rate = ap->sample_rate; if(ap->channels) st->codec->channels = ap->channels; else st->codec->channels = 1; st->codec->bits_per_coded_sample = av_get_bits_per_sample(st->codec->codec_id); assert(st->codec->bits_per_coded_sample > 0); st->codec->block_align = st->codec->bits_per_coded_sample*st->codec->channels/8; av_set_pts_info(st, 64, 1, st->codec->sample_rate); break; case AVMEDIA_TYPE_VIDEO: if(ap->time_base.num) av_set_pts_info(st, 64, ap->time_base.num, ap->time_base.den); else av_set_pts_info(st, 64, 1, 25); st->codec->width = ap->width; st->codec->height = ap->height; st->codec->pix_fmt = ap->pix_fmt; if(st->codec->pix_fmt == PIX_FMT_NONE) st->codec->pix_fmt= PIX_FMT_YUV420P; break; default: return -1; } return 0; } #define RAW_PACKET_SIZE 1024 #define RAW_SAMPLES 1024 static int raw_read_packet(AVFormatContext *s, AVPacket *pkt) { int ret, size, bps; // AVStream *st = s->streams[0]; size= RAW_SAMPLES*s->streams[0]->codec->block_align; ret= av_get_packet(s->pb, pkt, size); pkt->stream_index = 0; if (ret < 0) return ret; bps= av_get_bits_per_sample(s->streams[0]->codec->codec_id); assert(bps); // if false there IS a bug elsewhere (NOT in this function) pkt->dts= pkt->pts= pkt->pos*8 / (bps * s->streams[0]->codec->channels); return ret; } int ff_raw_read_partial_packet(AVFormatContext *s, AVPacket *pkt) { int ret, size; size = RAW_PACKET_SIZE; if (av_new_packet(pkt, size) < 0) return AVERROR(ENOMEM); pkt->pos= url_ftell(s->pb); pkt->stream_index = 0; ret = get_partial_buffer(s->pb, pkt->data, size); if (ret < 0) { av_free_packet(pkt); return ret; } pkt->size = ret; return ret; } #endif #if CONFIG_RAWVIDEO_DEMUXER static int rawvideo_read_packet(AVFormatContext *s, AVPacket *pkt) { int packet_size, ret, width, height; AVStream *st = s->streams[0]; width = st->codec->width; height = st->codec->height; packet_size = avpicture_get_size(st->codec->pix_fmt, width, height); if (packet_size < 0) return -1; ret= av_get_packet(s->pb, pkt, packet_size); pkt->pts= pkt->dts= pkt->pos / packet_size; pkt->stream_index = 0; if (ret < 0) return ret; return 0; } #endif #if CONFIG_DEMUXERS int pcm_read_seek(AVFormatContext *s, int stream_index, int64_t timestamp, int flags) { AVStream *st; int block_align, byte_rate; int64_t pos, ret; st = s->streams[0]; block_align = st->codec->block_align ? st->codec->block_align : (av_get_bits_per_sample(st->codec->codec_id) * st->codec->channels) >> 3; byte_rate = st->codec->bit_rate ? st->codec->bit_rate >> 3 : block_align * st->codec->sample_rate; if (block_align <= 0 || byte_rate <= 0) return -1; if (timestamp < 0) timestamp = 0; /* compute the position by aligning it to block_align */ pos = av_rescale_rnd(timestamp * byte_rate, st->time_base.num, st->time_base.den * (int64_t)block_align, (flags & AVSEEK_FLAG_BACKWARD) ? AV_ROUND_DOWN : AV_ROUND_UP); pos *= block_align; /* recompute exact position */ st->cur_dts = av_rescale(pos, st->time_base.den, byte_rate * (int64_t)st->time_base.num); if ((ret = url_fseek(s->pb, pos + s->data_offset, SEEK_SET)) < 0) return ret; return 0; } int ff_raw_audio_read_header(AVFormatContext *s, AVFormatParameters *ap) { AVStream *st = av_new_stream(s, 0); if (!st) return AVERROR(ENOMEM); st->codec->codec_type = AVMEDIA_TYPE_AUDIO; st->codec->codec_id = s->iformat->value; st->need_parsing = AVSTREAM_PARSE_FULL; /* the parameters will be extracted from the compressed bitstream */ return 0; } /* MPEG-1/H.263 input */ int ff_raw_video_read_header(AVFormatContext *s, AVFormatParameters *ap) { AVStream *st; st = av_new_stream(s, 0); if (!st) return AVERROR(ENOMEM); st->codec->codec_type = AVMEDIA_TYPE_VIDEO; st->codec->codec_id = s->iformat->value; st->need_parsing = AVSTREAM_PARSE_FULL; /* for MJPEG, specify frame rate */ /* for MPEG-4 specify it, too (most MPEG-4 streams do not have the fixed_vop_rate set ...)*/ if (ap->time_base.num) { st->codec->time_base= ap->time_base; } else if ( st->codec->codec_id == CODEC_ID_MJPEG || st->codec->codec_id == CODEC_ID_MPEG4 || st->codec->codec_id == CODEC_ID_DIRAC || st->codec->codec_id == CODEC_ID_DNXHD || st->codec->codec_id == CODEC_ID_VC1 || st->codec->codec_id == CODEC_ID_H264) { st->codec->time_base= (AVRational){1,25}; } av_set_pts_info(st, 64, 1, 1200000); return 0; } #endif #if CONFIG_AC3_DEMUXER || CONFIG_EAC3_DEMUXER static int ac3_eac3_probe(AVProbeData *p, enum CodecID expected_codec_id) { int max_frames, first_frames = 0, frames; uint8_t *buf, *buf2, *end; AC3HeaderInfo hdr; GetBitContext gbc; enum CodecID codec_id = CODEC_ID_AC3; max_frames = 0; buf = p->buf; end = buf + p->buf_size; for(; buf < end; buf++) { buf2 = buf; for(frames = 0; buf2 < end; frames++) { init_get_bits(&gbc, buf2, 54); if(ff_ac3_parse_header(&gbc, &hdr) < 0) break; if(buf2 + hdr.frame_size > end || av_crc(av_crc_get_table(AV_CRC_16_ANSI), 0, buf2 + 2, hdr.frame_size - 2)) break; if (hdr.bitstream_id > 10) codec_id = CODEC_ID_EAC3; buf2 += hdr.frame_size; } max_frames = FFMAX(max_frames, frames); if(buf == p->buf) first_frames = frames; } if(codec_id != expected_codec_id) return 0; // keep this in sync with mp3 probe, both need to avoid // issues with MPEG-files! if (first_frames>=4) return AVPROBE_SCORE_MAX/2+1; else if(max_frames>500)return AVPROBE_SCORE_MAX/2; else if(max_frames>=4) return AVPROBE_SCORE_MAX/4; else if(max_frames>=1) return 1; else return 0; } #endif #if CONFIG_AC3_DEMUXER static int ac3_probe(AVProbeData *p) { return ac3_eac3_probe(p, CODEC_ID_AC3); } #endif #if CONFIG_EAC3_DEMUXER static int eac3_probe(AVProbeData *p) { return ac3_eac3_probe(p, CODEC_ID_EAC3); } #endif /* Note: Do not forget to add new entries to the Makefile as well. */ #if CONFIG_AC3_DEMUXER AVInputFormat ac3_demuxer = { "ac3", NULL_IF_CONFIG_SMALL("raw AC-3"), 0, ac3_probe, ff_raw_audio_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "ac3", .value = CODEC_ID_AC3, }; #endif #if CONFIG_AC3_MUXER AVOutputFormat ac3_muxer = { "ac3", NULL_IF_CONFIG_SMALL("raw AC-3"), "audio/x-ac3", "ac3", 0, CODEC_ID_AC3, CODEC_ID_NONE, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_DIRAC_MUXER AVOutputFormat dirac_muxer = { "dirac", NULL_IF_CONFIG_SMALL("raw Dirac"), NULL, "drc", 0, CODEC_ID_NONE, CODEC_ID_DIRAC, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_DNXHD_MUXER AVOutputFormat dnxhd_muxer = { "dnxhd", NULL_IF_CONFIG_SMALL("raw DNxHD (SMPTE VC-3)"), NULL, "dnxhd", 0, CODEC_ID_NONE, CODEC_ID_DNXHD, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_DTS_MUXER AVOutputFormat dts_muxer = { "dts", NULL_IF_CONFIG_SMALL("raw DTS"), "audio/x-dca", "dts", 0, CODEC_ID_DTS, CODEC_ID_NONE, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_EAC3_DEMUXER AVInputFormat eac3_demuxer = { "eac3", NULL_IF_CONFIG_SMALL("raw E-AC-3"), 0, eac3_probe, ff_raw_audio_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "eac3", .value = CODEC_ID_EAC3, }; #endif #if CONFIG_EAC3_MUXER AVOutputFormat eac3_muxer = { "eac3", NULL_IF_CONFIG_SMALL("raw E-AC-3"), "audio/x-eac3", "eac3", 0, CODEC_ID_EAC3, CODEC_ID_NONE, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_GSM_DEMUXER AVInputFormat gsm_demuxer = { "gsm", NULL_IF_CONFIG_SMALL("raw GSM"), 0, NULL, ff_raw_audio_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "gsm", .value = CODEC_ID_GSM, }; #endif #if CONFIG_H261_MUXER AVOutputFormat h261_muxer = { "h261", NULL_IF_CONFIG_SMALL("raw H.261"), "video/x-h261", "h261", 0, CODEC_ID_NONE, CODEC_ID_H261, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_H263_MUXER AVOutputFormat h263_muxer = { "h263", NULL_IF_CONFIG_SMALL("raw H.263"), "video/x-h263", "h263", 0, CODEC_ID_NONE, CODEC_ID_H263, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_H264_MUXER AVOutputFormat h264_muxer = { "h264", NULL_IF_CONFIG_SMALL("raw H.264 video format"), NULL, "h264", 0, CODEC_ID_NONE, CODEC_ID_H264, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_CAVSVIDEO_MUXER AVOutputFormat cavsvideo_muxer = { "cavsvideo", NULL_IF_CONFIG_SMALL("raw Chinese AVS video"), NULL, "cavs", 0, CODEC_ID_NONE, CODEC_ID_CAVS, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_M4V_MUXER AVOutputFormat m4v_muxer = { "m4v", NULL_IF_CONFIG_SMALL("raw MPEG-4 video format"), NULL, "m4v", 0, CODEC_ID_NONE, CODEC_ID_MPEG4, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_MJPEG_DEMUXER AVInputFormat mjpeg_demuxer = { "mjpeg", NULL_IF_CONFIG_SMALL("raw MJPEG video"), 0, NULL, ff_raw_video_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "mjpg,mjpeg", .value = CODEC_ID_MJPEG, }; #endif #if CONFIG_MJPEG_MUXER AVOutputFormat mjpeg_muxer = { "mjpeg", NULL_IF_CONFIG_SMALL("raw MJPEG video"), "video/x-mjpeg", "mjpg,mjpeg", 0, CODEC_ID_NONE, CODEC_ID_MJPEG, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_MLP_DEMUXER AVInputFormat mlp_demuxer = { "mlp", NULL_IF_CONFIG_SMALL("raw MLP"), 0, NULL, ff_raw_audio_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "mlp", .value = CODEC_ID_MLP, }; #endif #if CONFIG_MLP_MUXER AVOutputFormat mlp_muxer = { "mlp", NULL_IF_CONFIG_SMALL("raw MLP"), NULL, "mlp", 0, CODEC_ID_MLP, CODEC_ID_NONE, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_SRT_MUXER AVOutputFormat srt_muxer = { .name = "srt", .long_name = NULL_IF_CONFIG_SMALL("SubRip subtitle format"), .mime_type = "application/x-subrip", .extensions = "srt", .write_packet = ff_raw_write_packet, .flags = AVFMT_NOTIMESTAMPS, .subtitle_codec = CODEC_ID_SRT, }; #endif #if CONFIG_TRUEHD_DEMUXER AVInputFormat truehd_demuxer = { "truehd", NULL_IF_CONFIG_SMALL("raw TrueHD"), 0, NULL, ff_raw_audio_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "thd", .value = CODEC_ID_TRUEHD, }; #endif #if CONFIG_TRUEHD_MUXER AVOutputFormat truehd_muxer = { "truehd", NULL_IF_CONFIG_SMALL("raw TrueHD"), NULL, "thd", 0, CODEC_ID_TRUEHD, CODEC_ID_NONE, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_MPEG1VIDEO_MUXER AVOutputFormat mpeg1video_muxer = { "mpeg1video", NULL_IF_CONFIG_SMALL("raw MPEG-1 video"), "video/x-mpeg", "mpg,mpeg,m1v", 0, CODEC_ID_NONE, CODEC_ID_MPEG1VIDEO, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_MPEG2VIDEO_MUXER AVOutputFormat mpeg2video_muxer = { "mpeg2video", NULL_IF_CONFIG_SMALL("raw MPEG-2 video"), NULL, "m2v", 0, CODEC_ID_NONE, CODEC_ID_MPEG2VIDEO, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_NULL_MUXER AVOutputFormat null_muxer = { "null", NULL_IF_CONFIG_SMALL("raw null video format"), NULL, NULL, 0, AV_NE(CODEC_ID_PCM_S16BE, CODEC_ID_PCM_S16LE), CODEC_ID_RAWVIDEO, NULL, null_write_packet, .flags = AVFMT_NOFILE | AVFMT_RAWPICTURE | AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_RAWVIDEO_DEMUXER AVInputFormat rawvideo_demuxer = { "rawvideo", NULL_IF_CONFIG_SMALL("raw video format"), 0, NULL, raw_read_header, rawvideo_read_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "yuv,cif,qcif,rgb", .value = CODEC_ID_RAWVIDEO, }; #endif #if CONFIG_RAWVIDEO_MUXER AVOutputFormat rawvideo_muxer = { "rawvideo", NULL_IF_CONFIG_SMALL("raw video format"), NULL, "yuv,rgb", 0, CODEC_ID_NONE, CODEC_ID_RAWVIDEO, NULL, ff_raw_write_packet, .flags= AVFMT_NOTIMESTAMPS, }; #endif #if CONFIG_SHORTEN_DEMUXER AVInputFormat shorten_demuxer = { "shn", NULL_IF_CONFIG_SMALL("raw Shorten"), 0, NULL, ff_raw_audio_read_header, ff_raw_read_partial_packet, .flags= AVFMT_GENERIC_INDEX, .extensions = "shn", .value = CODEC_ID_SHORTEN, }; #endif #if CONFIG_VC1_DEMUXER AVInputFormat vc1_demuxer = { "vc1", NULL_IF_CONFIG_SMALL("raw VC-1"), 0, NULL /* vc1_probe */, ff_raw_video_read_header, ff_raw_read_partial_packet, .extensions = "vc1", .value = CODEC_ID_VC1, }; #endif /* PCM formats */ #define PCMINPUTDEF(name, long_name, ext, codec) \ AVInputFormat pcm_ ## name ## _demuxer = {\ #name,\ NULL_IF_CONFIG_SMALL(long_name),\ 0,\ NULL,\ raw_read_header,\ raw_read_packet,\ NULL,\ pcm_read_seek,\ .flags= AVFMT_GENERIC_INDEX,\ .extensions = ext,\ .value = codec,\ }; #define PCMOUTPUTDEF(name, long_name, ext, codec) \ AVOutputFormat pcm_ ## name ## _muxer = {\ #name,\ NULL_IF_CONFIG_SMALL(long_name),\ NULL,\ ext,\ 0,\ codec,\ CODEC_ID_NONE,\ NULL,\ ff_raw_write_packet,\ .flags= AVFMT_NOTIMESTAMPS,\ }; #if !CONFIG_MUXERS && CONFIG_DEMUXERS #define PCMDEF(name, long_name, ext, codec) \ PCMINPUTDEF(name, long_name, ext, codec) #elif CONFIG_MUXERS && !CONFIG_DEMUXERS #define PCMDEF(name, long_name, ext, codec) \ PCMOUTPUTDEF(name, long_name, ext, codec) #elif CONFIG_MUXERS && CONFIG_DEMUXERS #define PCMDEF(name, long_name, ext, codec) \ PCMINPUTDEF(name, long_name, ext, codec)\ PCMOUTPUTDEF(name, long_name, ext, codec) #else #define PCMDEF(name, long_name, ext, codec) #endif #if HAVE_BIGENDIAN #define BE_DEF(s) s #define LE_DEF(s) NULL #else #define BE_DEF(s) NULL #define LE_DEF(s) s #endif PCMDEF(f64be, "PCM 64 bit floating-point big-endian format", NULL, CODEC_ID_PCM_F64BE) PCMDEF(f64le, "PCM 64 bit floating-point little-endian format", NULL, CODEC_ID_PCM_F64LE) PCMDEF(f32be, "PCM 32 bit floating-point big-endian format", NULL, CODEC_ID_PCM_F32BE) PCMDEF(f32le, "PCM 32 bit floating-point little-endian format", NULL, CODEC_ID_PCM_F32LE) PCMDEF(s32be, "PCM signed 32 bit big-endian format", NULL, CODEC_ID_PCM_S32BE) PCMDEF(s32le, "PCM signed 32 bit little-endian format", NULL, CODEC_ID_PCM_S32LE) PCMDEF(s24be, "PCM signed 24 bit big-endian format", NULL, CODEC_ID_PCM_S24BE) PCMDEF(s24le, "PCM signed 24 bit little-endian format", NULL, CODEC_ID_PCM_S24LE) PCMDEF(s16be, "PCM signed 16 bit big-endian format", BE_DEF("sw"), CODEC_ID_PCM_S16BE) PCMDEF(s16le, "PCM signed 16 bit little-endian format", LE_DEF("sw"), CODEC_ID_PCM_S16LE) PCMDEF(s8, "PCM signed 8 bit format", "sb", CODEC_ID_PCM_S8) PCMDEF(u32be, "PCM unsigned 32 bit big-endian format", NULL, CODEC_ID_PCM_U32BE) PCMDEF(u32le, "PCM unsigned 32 bit little-endian format", NULL, CODEC_ID_PCM_U32LE) PCMDEF(u24be, "PCM unsigned 24 bit big-endian format", NULL, CODEC_ID_PCM_U24BE) PCMDEF(u24le, "PCM unsigned 24 bit little-endian format", NULL, CODEC_ID_PCM_U24LE) PCMDEF(u16be, "PCM unsigned 16 bit big-endian format", BE_DEF("uw"), CODEC_ID_PCM_U16BE) PCMDEF(u16le, "PCM unsigned 16 bit little-endian format", LE_DEF("uw"), CODEC_ID_PCM_U16LE) PCMDEF(u8, "PCM unsigned 8 bit format", "ub", CODEC_ID_PCM_U8) PCMDEF(alaw, "PCM A-law format", "al", CODEC_ID_PCM_ALAW) PCMDEF(mulaw, "PCM mu-law format", "ul", CODEC_ID_PCM_MULAW)