00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00034 #include "avformat.h"
00035
00036 #define MM_PREAMBLE_SIZE 6
00037
00038 #define MM_TYPE_HEADER 0x0
00039 #define MM_TYPE_INTER 0x5
00040 #define MM_TYPE_INTRA 0x8
00041 #define MM_TYPE_INTRA_HH 0xc
00042 #define MM_TYPE_INTER_HH 0xd
00043 #define MM_TYPE_INTRA_HHV 0xe
00044 #define MM_TYPE_INTER_HHV 0xf
00045 #define MM_TYPE_AUDIO 0x15
00046 #define MM_TYPE_PALETTE 0x31
00047
00048 #define MM_HEADER_LEN_V 0x16
00049 #define MM_HEADER_LEN_AV 0x18
00050
00051 #define MM_PALETTE_COUNT 128
00052 #define MM_PALETTE_SIZE (MM_PALETTE_COUNT*3)
00053
00054 typedef struct {
00055 AVPaletteControl palette_control;
00056 unsigned int audio_pts, video_pts;
00057 } MmDemuxContext;
00058
00059 static int mm_probe(AVProbeData *p)
00060 {
00061
00062 if (AV_RL16(&p->buf[0]) != MM_TYPE_HEADER)
00063 return 0;
00064 if (AV_RL32(&p->buf[2]) != MM_HEADER_LEN_V && AV_RL32(&p->buf[2]) != MM_HEADER_LEN_AV)
00065 return 0;
00066
00067
00068 return AVPROBE_SCORE_MAX / 2;
00069 }
00070
00071 static int mm_read_header(AVFormatContext *s,
00072 AVFormatParameters *ap)
00073 {
00074 MmDemuxContext *mm = s->priv_data;
00075 ByteIOContext *pb = s->pb;
00076 AVStream *st;
00077
00078 unsigned int type, length;
00079 unsigned int frame_rate, width, height;
00080
00081 type = get_le16(pb);
00082 length = get_le32(pb);
00083
00084 if (type != MM_TYPE_HEADER)
00085 return AVERROR_INVALIDDATA;
00086
00087
00088 get_le16(pb);
00089 frame_rate = get_le16(pb);
00090 get_le16(pb);
00091 width = get_le16(pb);
00092 height = get_le16(pb);
00093 url_fseek(pb, length - 10, SEEK_CUR);
00094
00095
00096 st = av_new_stream(s, 0);
00097 if (!st)
00098 return AVERROR(ENOMEM);
00099 st->codec->codec_type = CODEC_TYPE_VIDEO;
00100 st->codec->codec_id = CODEC_ID_MMVIDEO;
00101 st->codec->codec_tag = 0;
00102 st->codec->width = width;
00103 st->codec->height = height;
00104 st->codec->palctrl = &mm->palette_control;
00105 av_set_pts_info(st, 64, 1, frame_rate);
00106
00107
00108 if (length == MM_HEADER_LEN_AV) {
00109 st = av_new_stream(s, 0);
00110 if (!st)
00111 return AVERROR(ENOMEM);
00112 st->codec->codec_type = CODEC_TYPE_AUDIO;
00113 st->codec->codec_tag = 0;
00114 st->codec->codec_id = CODEC_ID_PCM_U8;
00115 st->codec->channels = 1;
00116 st->codec->sample_rate = 8000;
00117 av_set_pts_info(st, 64, 1, 8000);
00118 }
00119
00120 mm->palette_control.palette_changed = 0;
00121 mm->audio_pts = 0;
00122 mm->video_pts = 0;
00123 return 0;
00124 }
00125
00126 static int mm_read_packet(AVFormatContext *s,
00127 AVPacket *pkt)
00128 {
00129 MmDemuxContext *mm = s->priv_data;
00130 ByteIOContext *pb = s->pb;
00131 unsigned char preamble[MM_PREAMBLE_SIZE];
00132 unsigned char pal[MM_PALETTE_SIZE];
00133 unsigned int type, length;
00134 int i;
00135
00136 while(1) {
00137
00138 if (get_buffer(pb, preamble, MM_PREAMBLE_SIZE) != MM_PREAMBLE_SIZE) {
00139 return AVERROR(EIO);
00140 }
00141
00142 type = AV_RL16(&preamble[0]);
00143 length = AV_RL16(&preamble[2]);
00144
00145 switch(type) {
00146 case MM_TYPE_PALETTE :
00147 url_fseek(pb, 4, SEEK_CUR);
00148 if (get_buffer(pb, pal, MM_PALETTE_SIZE) != MM_PALETTE_SIZE)
00149 return AVERROR(EIO);
00150 url_fseek(pb, length - (4 + MM_PALETTE_SIZE), SEEK_CUR);
00151
00152 for (i=0; i<MM_PALETTE_COUNT; i++) {
00153 int r = pal[i*3 + 0];
00154 int g = pal[i*3 + 1];
00155 int b = pal[i*3 + 2];
00156 mm->palette_control.palette[i] = (r << 16) | (g << 8) | (b);
00157
00158 mm->palette_control.palette[i+128] = (r << 18) | (g << 10) | (b<<2);
00159 }
00160 mm->palette_control.palette_changed = 1;
00161 break;
00162
00163 case MM_TYPE_INTER :
00164 case MM_TYPE_INTRA :
00165 case MM_TYPE_INTRA_HH :
00166 case MM_TYPE_INTER_HH :
00167 case MM_TYPE_INTRA_HHV :
00168 case MM_TYPE_INTER_HHV :
00169
00170 if (av_new_packet(pkt, length + MM_PREAMBLE_SIZE))
00171 return AVERROR(ENOMEM);
00172 memcpy(pkt->data, preamble, MM_PREAMBLE_SIZE);
00173 if (get_buffer(pb, pkt->data + MM_PREAMBLE_SIZE, length) != length)
00174 return AVERROR(EIO);
00175 pkt->size = length + MM_PREAMBLE_SIZE;
00176 pkt->stream_index = 0;
00177 pkt->pts = mm->video_pts++;
00178 return 0;
00179
00180 case MM_TYPE_AUDIO :
00181 if (av_get_packet(s->pb, pkt, length)<0)
00182 return AVERROR(ENOMEM);
00183 pkt->size = length;
00184 pkt->stream_index = 1;
00185 pkt->pts = mm->audio_pts++;
00186 return 0;
00187
00188 default :
00189 av_log(NULL, AV_LOG_INFO, "mm: unknown chunk type 0x%x\n", type);
00190 url_fseek(pb, length, SEEK_CUR);
00191 }
00192 }
00193
00194 return 0;
00195 }
00196
00197 static int mm_read_close(AVFormatContext *s)
00198 {
00199 return 0;
00200 }
00201
00202 AVInputFormat mm_demuxer = {
00203 "mm",
00204 "American Laser Games MM format",
00205 sizeof(MmDemuxContext),
00206 mm_probe,
00207 mm_read_header,
00208 mm_read_packet,
00209 mm_read_close,
00210 };