yading@11: /* yading@11: * Deluxe Paint Animation demuxer yading@11: * Copyright (c) 2009 Peter Ross yading@11: * yading@11: * This file is part of FFmpeg. yading@11: * yading@11: * FFmpeg is free software; you can redistribute it and/or yading@11: * modify it under the terms of the GNU Lesser General Public yading@11: * License as published by the Free Software Foundation; either yading@11: * version 2.1 of the License, or (at your option) any later version. yading@11: * yading@11: * FFmpeg is distributed in the hope that it will be useful, yading@11: * but WITHOUT ANY WARRANTY; without even the implied warranty of yading@11: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU yading@11: * Lesser General Public License for more details. yading@11: * yading@11: * You should have received a copy of the GNU Lesser General Public yading@11: * License along with FFmpeg; if not, write to the Free Software yading@11: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA yading@11: */ yading@11: yading@11: /** yading@11: * @file yading@11: * Deluxe Paint Animation demuxer yading@11: */ yading@11: yading@11: #include "libavutil/intreadwrite.h" yading@11: #include "avformat.h" yading@11: #include "internal.h" yading@11: yading@11: typedef struct { yading@11: int base_record; yading@11: unsigned int nb_records; yading@11: int size; yading@11: } Page; yading@11: yading@11: typedef struct { yading@11: unsigned int nb_pages; /**< total pages in file */ yading@11: unsigned int nb_records; /**< total records in file */ yading@11: int page_table_offset; yading@11: #define MAX_PAGES 256 /**< Deluxe Paint hardcoded value */ yading@11: Page pt[MAX_PAGES]; /**< page table */ yading@11: int page; /**< current page (or AVERROR_xxx code) */ yading@11: int record; /**< current record (with in page) */ yading@11: } AnmDemuxContext; yading@11: yading@11: #define LPF_TAG MKTAG('L','P','F',' ') yading@11: #define ANIM_TAG MKTAG('A','N','I','M') yading@11: yading@11: static int probe(AVProbeData *p) yading@11: { yading@11: /* verify tags and video dimensions */ yading@11: if (AV_RL32(&p->buf[0]) == LPF_TAG && yading@11: AV_RL32(&p->buf[16]) == ANIM_TAG && yading@11: AV_RL16(&p->buf[20]) && AV_RL16(&p->buf[22])) yading@11: return AVPROBE_SCORE_MAX; yading@11: return 0; yading@11: } yading@11: yading@11: /** yading@11: * @return page containing the requested record or AVERROR_XXX yading@11: */ yading@11: static int find_record(const AnmDemuxContext *anm, int record) yading@11: { yading@11: int i; yading@11: yading@11: if (record >= anm->nb_records) yading@11: return AVERROR_EOF; yading@11: yading@11: for (i = 0; i < MAX_PAGES; i++) { yading@11: const Page *p = &anm->pt[i]; yading@11: if (p->nb_records > 0 && record >= p->base_record && record < p->base_record + p->nb_records) yading@11: return i; yading@11: } yading@11: yading@11: return AVERROR_INVALIDDATA; yading@11: } yading@11: yading@11: static int read_header(AVFormatContext *s) yading@11: { yading@11: AnmDemuxContext *anm = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: AVStream *st; yading@11: int i, ret; yading@11: yading@11: avio_skip(pb, 4); /* magic number */ yading@11: if (avio_rl16(pb) != MAX_PAGES) { yading@11: avpriv_request_sample(s, "max_pages != " AV_STRINGIFY(MAX_PAGES)); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: yading@11: anm->nb_pages = avio_rl16(pb); yading@11: anm->nb_records = avio_rl32(pb); yading@11: avio_skip(pb, 2); /* max records per page */ yading@11: anm->page_table_offset = avio_rl16(pb); yading@11: if (avio_rl32(pb) != ANIM_TAG) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: /* video stream */ yading@11: st = avformat_new_stream(s, NULL); yading@11: if (!st) yading@11: return AVERROR(ENOMEM); yading@11: st->codec->codec_type = AVMEDIA_TYPE_VIDEO; yading@11: st->codec->codec_id = AV_CODEC_ID_ANM; yading@11: st->codec->codec_tag = 0; /* no fourcc */ yading@11: st->codec->width = avio_rl16(pb); yading@11: st->codec->height = avio_rl16(pb); yading@11: if (avio_r8(pb) != 0) yading@11: goto invalid; yading@11: avio_skip(pb, 1); /* frame rate multiplier info */ yading@11: yading@11: /* ignore last delta record (used for looping) */ yading@11: if (avio_r8(pb)) /* has_last_delta */ yading@11: anm->nb_records = FFMAX(anm->nb_records - 1, 0); yading@11: yading@11: avio_skip(pb, 1); /* last_delta_valid */ yading@11: yading@11: if (avio_r8(pb) != 0) yading@11: goto invalid; yading@11: yading@11: if (avio_r8(pb) != 1) yading@11: goto invalid; yading@11: yading@11: avio_skip(pb, 1); /* other recs per frame */ yading@11: yading@11: if (avio_r8(pb) != 1) yading@11: goto invalid; yading@11: yading@11: avio_skip(pb, 32); /* record_types */ yading@11: st->nb_frames = avio_rl32(pb); yading@11: avpriv_set_pts_info(st, 64, 1, avio_rl16(pb)); yading@11: avio_skip(pb, 58); yading@11: yading@11: /* color cycling and palette data */ yading@11: st->codec->extradata_size = 16*8 + 4*256; yading@11: st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); yading@11: if (!st->codec->extradata) { yading@11: return AVERROR(ENOMEM); yading@11: } yading@11: ret = avio_read(pb, st->codec->extradata, st->codec->extradata_size); yading@11: if (ret < 0) yading@11: return ret; yading@11: yading@11: /* read page table */ yading@11: ret = avio_seek(pb, anm->page_table_offset, SEEK_SET); yading@11: if (ret < 0) yading@11: return ret; yading@11: yading@11: for (i = 0; i < MAX_PAGES; i++) { yading@11: Page *p = &anm->pt[i]; yading@11: p->base_record = avio_rl16(pb); yading@11: p->nb_records = avio_rl16(pb); yading@11: p->size = avio_rl16(pb); yading@11: } yading@11: yading@11: /* find page of first frame */ yading@11: anm->page = find_record(anm, 0); yading@11: if (anm->page < 0) { yading@11: return anm->page; yading@11: } yading@11: yading@11: anm->record = -1; yading@11: return 0; yading@11: yading@11: invalid: yading@11: avpriv_request_sample(s, "Invalid header element"); yading@11: return AVERROR_PATCHWELCOME; yading@11: } yading@11: yading@11: static int read_packet(AVFormatContext *s, yading@11: AVPacket *pkt) yading@11: { yading@11: AnmDemuxContext *anm = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: Page *p; yading@11: int tmp, record_size; yading@11: yading@11: if (url_feof(s->pb)) yading@11: return AVERROR(EIO); yading@11: yading@11: if (anm->page < 0) yading@11: return anm->page; yading@11: yading@11: repeat: yading@11: p = &anm->pt[anm->page]; yading@11: yading@11: /* parse page header */ yading@11: if (anm->record < 0) { yading@11: avio_seek(pb, anm->page_table_offset + MAX_PAGES*6 + (anm->page<<16), SEEK_SET); yading@11: avio_skip(pb, 8 + 2*p->nb_records); yading@11: anm->record = 0; yading@11: } yading@11: yading@11: /* if we have fetched all records in this page, then find the yading@11: next page and repeat */ yading@11: if (anm->record >= p->nb_records) { yading@11: anm->page = find_record(anm, p->base_record + p->nb_records); yading@11: if (anm->page < 0) yading@11: return anm->page; yading@11: anm->record = -1; yading@11: goto repeat; yading@11: } yading@11: yading@11: /* fetch record size */ yading@11: tmp = avio_tell(pb); yading@11: avio_seek(pb, anm->page_table_offset + MAX_PAGES*6 + (anm->page<<16) + yading@11: 8 + anm->record * 2, SEEK_SET); yading@11: record_size = avio_rl16(pb); yading@11: avio_seek(pb, tmp, SEEK_SET); yading@11: yading@11: /* fetch record */ yading@11: pkt->size = av_get_packet(s->pb, pkt, record_size); yading@11: if (pkt->size < 0) yading@11: return pkt->size; yading@11: if (p->base_record + anm->record == 0) yading@11: pkt->flags |= AV_PKT_FLAG_KEY; yading@11: yading@11: anm->record++; yading@11: return 0; yading@11: } yading@11: yading@11: AVInputFormat ff_anm_demuxer = { yading@11: .name = "anm", yading@11: .long_name = NULL_IF_CONFIG_SMALL("Deluxe Paint Animation"), yading@11: .priv_data_size = sizeof(AnmDemuxContext), yading@11: .read_probe = probe, yading@11: .read_header = read_header, yading@11: .read_packet = read_packet, yading@11: };