yading@11: /* yading@11: * American Laser Games MM Format Demuxer yading@11: * Copyright (c) 2006 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: * American Laser Games MM Format Demuxer yading@11: * by Peter Ross (pross@xvid.org) yading@11: * yading@11: * The MM format was used by IBM-PC ports of ALG's "arcade shooter" games, yading@11: * including Mad Dog McCree and Crime Patrol. yading@11: * yading@11: * Technical details here: yading@11: * http://wiki.multimedia.cx/index.php?title=American_Laser_Games_MM yading@11: */ yading@11: yading@11: #include "libavutil/channel_layout.h" yading@11: #include "libavutil/intreadwrite.h" yading@11: #include "avformat.h" yading@11: #include "internal.h" yading@11: yading@11: #define MM_PREAMBLE_SIZE 6 yading@11: yading@11: #define MM_TYPE_HEADER 0x0 yading@11: #define MM_TYPE_INTER 0x5 yading@11: #define MM_TYPE_INTRA 0x8 yading@11: #define MM_TYPE_INTRA_HH 0xc yading@11: #define MM_TYPE_INTER_HH 0xd yading@11: #define MM_TYPE_INTRA_HHV 0xe yading@11: #define MM_TYPE_INTER_HHV 0xf yading@11: #define MM_TYPE_AUDIO 0x15 yading@11: #define MM_TYPE_PALETTE 0x31 yading@11: yading@11: #define MM_HEADER_LEN_V 0x16 /* video only */ yading@11: #define MM_HEADER_LEN_AV 0x18 /* video + audio */ yading@11: yading@11: #define MM_PALETTE_COUNT 128 yading@11: #define MM_PALETTE_SIZE (MM_PALETTE_COUNT*3) yading@11: yading@11: typedef struct { yading@11: unsigned int audio_pts, video_pts; yading@11: } MmDemuxContext; yading@11: yading@11: static int probe(AVProbeData *p) yading@11: { yading@11: int len, type, fps, w, h; yading@11: if (p->buf_size < MM_HEADER_LEN_AV + MM_PREAMBLE_SIZE) yading@11: return 0; yading@11: /* the first chunk is always the header */ yading@11: if (AV_RL16(&p->buf[0]) != MM_TYPE_HEADER) yading@11: return 0; yading@11: len = AV_RL32(&p->buf[2]); yading@11: if (len != MM_HEADER_LEN_V && len != MM_HEADER_LEN_AV) yading@11: return 0; yading@11: fps = AV_RL16(&p->buf[8]); yading@11: w = AV_RL16(&p->buf[12]); yading@11: h = AV_RL16(&p->buf[14]); yading@11: if (!fps || fps > 60 || !w || w > 2048 || !h || h > 2048) yading@11: return 0; yading@11: type = AV_RL16(&p->buf[len]); yading@11: if (!type || type > 0x31) yading@11: return 0; yading@11: yading@11: /* only return half certainty since this check is a bit sketchy */ yading@11: return AVPROBE_SCORE_MAX / 2; yading@11: } yading@11: yading@11: static int read_header(AVFormatContext *s) yading@11: { yading@11: MmDemuxContext *mm = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: AVStream *st; yading@11: yading@11: unsigned int type, length; yading@11: unsigned int frame_rate, width, height; yading@11: yading@11: type = avio_rl16(pb); yading@11: length = avio_rl32(pb); yading@11: yading@11: if (type != MM_TYPE_HEADER) yading@11: return AVERROR_INVALIDDATA; yading@11: yading@11: /* read header */ yading@11: avio_rl16(pb); /* total number of chunks */ yading@11: frame_rate = avio_rl16(pb); yading@11: avio_rl16(pb); /* ibm-pc video bios mode */ yading@11: width = avio_rl16(pb); yading@11: height = avio_rl16(pb); yading@11: avio_skip(pb, length - 10); /* unknown data */ 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_MMVIDEO; yading@11: st->codec->codec_tag = 0; /* no fourcc */ yading@11: st->codec->width = width; yading@11: st->codec->height = height; yading@11: avpriv_set_pts_info(st, 64, 1, frame_rate); yading@11: yading@11: /* audio stream */ yading@11: if (length == MM_HEADER_LEN_AV) { 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_AUDIO; yading@11: st->codec->codec_tag = 0; /* no fourcc */ yading@11: st->codec->codec_id = AV_CODEC_ID_PCM_U8; yading@11: st->codec->channels = 1; yading@11: st->codec->channel_layout = AV_CH_LAYOUT_MONO; yading@11: st->codec->sample_rate = 8000; yading@11: avpriv_set_pts_info(st, 64, 1, 8000); /* 8000 hz */ yading@11: } yading@11: yading@11: mm->audio_pts = 0; yading@11: mm->video_pts = 0; yading@11: return 0; yading@11: } yading@11: yading@11: static int read_packet(AVFormatContext *s, yading@11: AVPacket *pkt) yading@11: { yading@11: MmDemuxContext *mm = s->priv_data; yading@11: AVIOContext *pb = s->pb; yading@11: unsigned char preamble[MM_PREAMBLE_SIZE]; yading@11: unsigned int type, length; yading@11: yading@11: while(1) { yading@11: yading@11: if (avio_read(pb, preamble, MM_PREAMBLE_SIZE) != MM_PREAMBLE_SIZE) { yading@11: return AVERROR(EIO); yading@11: } yading@11: yading@11: type = AV_RL16(&preamble[0]); yading@11: length = AV_RL16(&preamble[2]); yading@11: yading@11: switch(type) { yading@11: case MM_TYPE_PALETTE : yading@11: case MM_TYPE_INTER : yading@11: case MM_TYPE_INTRA : yading@11: case MM_TYPE_INTRA_HH : yading@11: case MM_TYPE_INTER_HH : yading@11: case MM_TYPE_INTRA_HHV : yading@11: case MM_TYPE_INTER_HHV : yading@11: /* output preamble + data */ yading@11: if (av_new_packet(pkt, length + MM_PREAMBLE_SIZE)) yading@11: return AVERROR(ENOMEM); yading@11: memcpy(pkt->data, preamble, MM_PREAMBLE_SIZE); yading@11: if (avio_read(pb, pkt->data + MM_PREAMBLE_SIZE, length) != length) yading@11: return AVERROR(EIO); yading@11: pkt->size = length + MM_PREAMBLE_SIZE; yading@11: pkt->stream_index = 0; yading@11: pkt->pts = mm->video_pts; yading@11: if (type!=MM_TYPE_PALETTE) yading@11: mm->video_pts++; yading@11: return 0; yading@11: yading@11: case MM_TYPE_AUDIO : yading@11: if (av_get_packet(s->pb, pkt, length)<0) yading@11: return AVERROR(ENOMEM); yading@11: pkt->stream_index = 1; yading@11: pkt->pts = mm->audio_pts++; yading@11: return 0; yading@11: yading@11: default : yading@11: av_log(s, AV_LOG_INFO, "unknown chunk type 0x%x\n", type); yading@11: avio_skip(pb, length); yading@11: } yading@11: } yading@11: } yading@11: yading@11: AVInputFormat ff_mm_demuxer = { yading@11: .name = "mm", yading@11: .long_name = NULL_IF_CONFIG_SMALL("American Laser Games MM"), yading@11: .priv_data_size = sizeof(MmDemuxContext), yading@11: .read_probe = probe, yading@11: .read_header = read_header, yading@11: .read_packet = read_packet, yading@11: };