1cabdff1aSopenharmony_ci/* 2cabdff1aSopenharmony_ci * Copyright (c) 2010 David Conrad 3cabdff1aSopenharmony_ci * 4cabdff1aSopenharmony_ci * This file is part of FFmpeg. 5cabdff1aSopenharmony_ci * 6cabdff1aSopenharmony_ci * FFmpeg is free software; you can redistribute it and/or 7cabdff1aSopenharmony_ci * modify it under the terms of the GNU Lesser General Public 8cabdff1aSopenharmony_ci * License as published by the Free Software Foundation; either 9cabdff1aSopenharmony_ci * version 2.1 of the License, or (at your option) any later version. 10cabdff1aSopenharmony_ci * 11cabdff1aSopenharmony_ci * FFmpeg is distributed in the hope that it will be useful, 12cabdff1aSopenharmony_ci * but WITHOUT ANY WARRANTY; without even the implied warranty of 13cabdff1aSopenharmony_ci * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 14cabdff1aSopenharmony_ci * Lesser General Public License for more details. 15cabdff1aSopenharmony_ci * 16cabdff1aSopenharmony_ci * You should have received a copy of the GNU Lesser General Public 17cabdff1aSopenharmony_ci * License along with FFmpeg; if not, write to the Free Software 18cabdff1aSopenharmony_ci * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 19cabdff1aSopenharmony_ci */ 20cabdff1aSopenharmony_ci 21cabdff1aSopenharmony_ci#include "avformat.h" 22cabdff1aSopenharmony_ci#include "internal.h" 23cabdff1aSopenharmony_ci#include "riff.h" 24cabdff1aSopenharmony_ci#include "libavutil/intreadwrite.h" 25cabdff1aSopenharmony_ci 26cabdff1aSopenharmony_cistatic int probe(const AVProbeData *p) 27cabdff1aSopenharmony_ci{ 28cabdff1aSopenharmony_ci if (AV_RL32(p->buf) == MKTAG('D','K','I','F') 29cabdff1aSopenharmony_ci && !AV_RL16(p->buf+4) && AV_RL16(p->buf+6) == 32) 30cabdff1aSopenharmony_ci return AVPROBE_SCORE_MAX-2; 31cabdff1aSopenharmony_ci 32cabdff1aSopenharmony_ci return 0; 33cabdff1aSopenharmony_ci} 34cabdff1aSopenharmony_ci 35cabdff1aSopenharmony_cistatic int read_header(AVFormatContext *s) 36cabdff1aSopenharmony_ci{ 37cabdff1aSopenharmony_ci AVStream *st; 38cabdff1aSopenharmony_ci AVRational time_base; 39cabdff1aSopenharmony_ci 40cabdff1aSopenharmony_ci avio_rl32(s->pb); // DKIF 41cabdff1aSopenharmony_ci avio_rl16(s->pb); // version 42cabdff1aSopenharmony_ci avio_rl16(s->pb); // header size 43cabdff1aSopenharmony_ci 44cabdff1aSopenharmony_ci st = avformat_new_stream(s, NULL); 45cabdff1aSopenharmony_ci if (!st) 46cabdff1aSopenharmony_ci return AVERROR(ENOMEM); 47cabdff1aSopenharmony_ci 48cabdff1aSopenharmony_ci 49cabdff1aSopenharmony_ci st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; 50cabdff1aSopenharmony_ci st->codecpar->codec_tag = avio_rl32(s->pb); 51cabdff1aSopenharmony_ci st->codecpar->codec_id = ff_codec_get_id(ff_codec_bmp_tags, st->codecpar->codec_tag); 52cabdff1aSopenharmony_ci st->codecpar->width = avio_rl16(s->pb); 53cabdff1aSopenharmony_ci st->codecpar->height = avio_rl16(s->pb); 54cabdff1aSopenharmony_ci time_base.den = avio_rl32(s->pb); 55cabdff1aSopenharmony_ci time_base.num = avio_rl32(s->pb); 56cabdff1aSopenharmony_ci st->duration = avio_rl32(s->pb); 57cabdff1aSopenharmony_ci avio_skip(s->pb, 4); // unused 58cabdff1aSopenharmony_ci 59cabdff1aSopenharmony_ci ffstream(st)->need_parsing = AVSTREAM_PARSE_HEADERS; 60cabdff1aSopenharmony_ci 61cabdff1aSopenharmony_ci if (!time_base.den || !time_base.num) { 62cabdff1aSopenharmony_ci av_log(s, AV_LOG_ERROR, "Invalid frame rate\n"); 63cabdff1aSopenharmony_ci return AVERROR_INVALIDDATA; 64cabdff1aSopenharmony_ci } 65cabdff1aSopenharmony_ci 66cabdff1aSopenharmony_ci avpriv_set_pts_info(st, 64, time_base.num, time_base.den); 67cabdff1aSopenharmony_ci 68cabdff1aSopenharmony_ci return 0; 69cabdff1aSopenharmony_ci} 70cabdff1aSopenharmony_ci 71cabdff1aSopenharmony_cistatic int read_packet(AVFormatContext *s, AVPacket *pkt) 72cabdff1aSopenharmony_ci{ 73cabdff1aSopenharmony_ci int ret, size = avio_rl32(s->pb); 74cabdff1aSopenharmony_ci int64_t pts = avio_rl64(s->pb); 75cabdff1aSopenharmony_ci 76cabdff1aSopenharmony_ci ret = av_get_packet(s->pb, pkt, size); 77cabdff1aSopenharmony_ci pkt->stream_index = 0; 78cabdff1aSopenharmony_ci pkt->pts = pts; 79cabdff1aSopenharmony_ci pkt->pos -= 12; 80cabdff1aSopenharmony_ci 81cabdff1aSopenharmony_ci return ret; 82cabdff1aSopenharmony_ci} 83cabdff1aSopenharmony_ci 84cabdff1aSopenharmony_ciconst AVInputFormat ff_ivf_demuxer = { 85cabdff1aSopenharmony_ci .name = "ivf", 86cabdff1aSopenharmony_ci .long_name = NULL_IF_CONFIG_SMALL("On2 IVF"), 87cabdff1aSopenharmony_ci .read_probe = probe, 88cabdff1aSopenharmony_ci .read_header = read_header, 89cabdff1aSopenharmony_ci .read_packet = read_packet, 90cabdff1aSopenharmony_ci .flags = AVFMT_GENERIC_INDEX, 91cabdff1aSopenharmony_ci .codec_tag = (const AVCodecTag* const []){ ff_codec_bmp_tags, 0 }, 92cabdff1aSopenharmony_ci}; 93