Libav 0.7.1
|
00001 /* Electronic Arts Multimedia File Demuxer 00002 * Copyright (c) 2004 The ffmpeg Project 00003 * Copyright (c) 2006-2008 Peter Ross 00004 * 00005 * This file is part of Libav. 00006 * 00007 * Libav is free software; you can redistribute it and/or 00008 * modify it under the terms of the GNU Lesser General Public 00009 * License as published by the Free Software Foundation; either 00010 * version 2.1 of the License, or (at your option) any later version. 00011 * 00012 * Libav is distributed in the hope that it will be useful, 00013 * but WITHOUT ANY WARRANTY; without even the implied warranty of 00014 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 00015 * Lesser General Public License for more details. 00016 * 00017 * You should have received a copy of the GNU Lesser General Public 00018 * License along with Libav; if not, write to the Free Software 00019 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 00020 */ 00021 00028 #include "libavutil/intreadwrite.h" 00029 #include "avformat.h" 00030 00031 #define SCHl_TAG MKTAG('S', 'C', 'H', 'l') 00032 #define SEAD_TAG MKTAG('S', 'E', 'A', 'D') /* Sxxx header */ 00033 #define SNDC_TAG MKTAG('S', 'N', 'D', 'C') /* Sxxx data */ 00034 #define SEND_TAG MKTAG('S', 'E', 'N', 'D') /* Sxxx end */ 00035 #define SHEN_TAG MKTAG('S', 'H', 'E', 'N') /* SxEN header */ 00036 #define SDEN_TAG MKTAG('S', 'D', 'E', 'N') /* SxEN data */ 00037 #define SEEN_TAG MKTAG('S', 'E', 'E', 'N') /* SxEN end */ 00038 #define ISNh_TAG MKTAG('1', 'S', 'N', 'h') /* 1SNx header */ 00039 #define EACS_TAG MKTAG('E', 'A', 'C', 'S') 00040 #define ISNd_TAG MKTAG('1', 'S', 'N', 'd') /* 1SNx data */ 00041 #define ISNe_TAG MKTAG('1', 'S', 'N', 'e') /* 1SNx end */ 00042 #define PT00_TAG MKTAG('P', 'T', 0x0, 0x0) 00043 #define GSTR_TAG MKTAG('G', 'S', 'T', 'R') 00044 #define SCDl_TAG MKTAG('S', 'C', 'D', 'l') 00045 #define SCEl_TAG MKTAG('S', 'C', 'E', 'l') 00046 #define kVGT_TAG MKTAG('k', 'V', 'G', 'T') /* TGV i-frame */ 00047 #define fVGT_TAG MKTAG('f', 'V', 'G', 'T') /* TGV p-frame */ 00048 #define mTCD_TAG MKTAG('m', 'T', 'C', 'D') /* MDEC */ 00049 #define MADk_TAG MKTAG('M', 'A', 'D', 'k') /* MAD i-frame */ 00050 #define MADm_TAG MKTAG('M', 'A', 'D', 'm') /* MAD p-frame */ 00051 #define MADe_TAG MKTAG('M', 'A', 'D', 'e') /* MAD lqp-frame */ 00052 #define MPCh_TAG MKTAG('M', 'P', 'C', 'h') /* MPEG2 */ 00053 #define TGQs_TAG MKTAG('T', 'G', 'Q', 's') /* TGQ i-frame (appears in .TGQ files) */ 00054 #define pQGT_TAG MKTAG('p', 'Q', 'G', 'T') /* TGQ i-frame (appears in .UV files) */ 00055 #define pIQT_TAG MKTAG('p', 'I', 'Q', 'T') /* TQI/UV2 i-frame (.UV2/.WVE) */ 00056 #define MVhd_TAG MKTAG('M', 'V', 'h', 'd') 00057 #define MV0K_TAG MKTAG('M', 'V', '0', 'K') 00058 #define MV0F_TAG MKTAG('M', 'V', '0', 'F') 00059 #define MVIh_TAG MKTAG('M', 'V', 'I', 'h') /* CMV header */ 00060 #define MVIf_TAG MKTAG('M', 'V', 'I', 'f') /* CMV i-frame */ 00061 00062 typedef struct EaDemuxContext { 00063 int big_endian; 00064 00065 enum CodecID video_codec; 00066 AVRational time_base; 00067 int width, height; 00068 int video_stream_index; 00069 00070 enum CodecID audio_codec; 00071 int audio_stream_index; 00072 int audio_frame_counter; 00073 00074 int bytes; 00075 int sample_rate; 00076 int num_channels; 00077 int num_samples; 00078 } EaDemuxContext; 00079 00080 static uint32_t read_arbitary(AVIOContext *pb) { 00081 uint8_t size, byte; 00082 int i; 00083 uint32_t word; 00084 00085 size = avio_r8(pb); 00086 00087 word = 0; 00088 for (i = 0; i < size; i++) { 00089 byte = avio_r8(pb); 00090 word <<= 8; 00091 word |= byte; 00092 } 00093 00094 return word; 00095 } 00096 00097 /* 00098 * Process PT/GSTR sound header 00099 * return 1 if success, 0 if invalid format, otherwise AVERROR_xxx 00100 */ 00101 static int process_audio_header_elements(AVFormatContext *s) 00102 { 00103 int inHeader = 1; 00104 EaDemuxContext *ea = s->priv_data; 00105 AVIOContext *pb = s->pb; 00106 int compression_type = -1, revision = -1, revision2 = -1; 00107 00108 ea->bytes = 2; 00109 ea->sample_rate = -1; 00110 ea->num_channels = 1; 00111 00112 while (!pb->eof_reached && inHeader) { 00113 int inSubheader; 00114 uint8_t byte; 00115 byte = avio_r8(pb); 00116 00117 switch (byte) { 00118 case 0xFD: 00119 av_log (s, AV_LOG_DEBUG, "entered audio subheader\n"); 00120 inSubheader = 1; 00121 while (!pb->eof_reached && inSubheader) { 00122 uint8_t subbyte; 00123 subbyte = avio_r8(pb); 00124 00125 switch (subbyte) { 00126 case 0x80: 00127 revision = read_arbitary(pb); 00128 av_log (s, AV_LOG_DEBUG, "revision (element 0x80) set to 0x%08x\n", revision); 00129 break; 00130 case 0x82: 00131 ea->num_channels = read_arbitary(pb); 00132 av_log (s, AV_LOG_DEBUG, "num_channels (element 0x82) set to 0x%08x\n", ea->num_channels); 00133 break; 00134 case 0x83: 00135 compression_type = read_arbitary(pb); 00136 av_log (s, AV_LOG_DEBUG, "compression_type (element 0x83) set to 0x%08x\n", compression_type); 00137 break; 00138 case 0x84: 00139 ea->sample_rate = read_arbitary(pb); 00140 av_log (s, AV_LOG_DEBUG, "sample_rate (element 0x84) set to %i\n", ea->sample_rate); 00141 break; 00142 case 0x85: 00143 ea->num_samples = read_arbitary(pb); 00144 av_log (s, AV_LOG_DEBUG, "num_samples (element 0x85) set to 0x%08x\n", ea->num_samples); 00145 break; 00146 case 0x8A: 00147 av_log (s, AV_LOG_DEBUG, "element 0x%02x set to 0x%08x\n", subbyte, read_arbitary(pb)); 00148 av_log (s, AV_LOG_DEBUG, "exited audio subheader\n"); 00149 inSubheader = 0; 00150 break; 00151 case 0xA0: 00152 revision2 = read_arbitary(pb); 00153 av_log (s, AV_LOG_DEBUG, "revision2 (element 0xA0) set to 0x%08x\n", revision2); 00154 break; 00155 case 0xFF: 00156 av_log (s, AV_LOG_DEBUG, "end of header block reached (within audio subheader)\n"); 00157 inSubheader = 0; 00158 inHeader = 0; 00159 break; 00160 default: 00161 av_log (s, AV_LOG_DEBUG, "element 0x%02x set to 0x%08x\n", subbyte, read_arbitary(pb)); 00162 break; 00163 } 00164 } 00165 break; 00166 case 0xFF: 00167 av_log (s, AV_LOG_DEBUG, "end of header block reached\n"); 00168 inHeader = 0; 00169 break; 00170 default: 00171 av_log (s, AV_LOG_DEBUG, "header element 0x%02x set to 0x%08x\n", byte, read_arbitary(pb)); 00172 break; 00173 } 00174 } 00175 00176 switch (compression_type) { 00177 case 0: ea->audio_codec = CODEC_ID_PCM_S16LE; break; 00178 case 7: ea->audio_codec = CODEC_ID_ADPCM_EA; break; 00179 case -1: 00180 switch (revision) { 00181 case 1: ea->audio_codec = CODEC_ID_ADPCM_EA_R1; break; 00182 case 2: ea->audio_codec = CODEC_ID_ADPCM_EA_R2; break; 00183 case 3: ea->audio_codec = CODEC_ID_ADPCM_EA_R3; break; 00184 case -1: break; 00185 default: 00186 av_log(s, AV_LOG_ERROR, "unsupported stream type; revision=%i\n", revision); 00187 return 0; 00188 } 00189 switch (revision2) { 00190 case 8: ea->audio_codec = CODEC_ID_PCM_S16LE_PLANAR; break; 00191 case 10: ea->audio_codec = CODEC_ID_ADPCM_EA_R2; break; 00192 case 16: ea->audio_codec = CODEC_ID_MP3; break; 00193 case -1: break; 00194 default: 00195 ea->audio_codec = CODEC_ID_NONE; 00196 av_log(s, AV_LOG_ERROR, "unsupported stream type; revision2=%i\n", revision2); 00197 return 0; 00198 } 00199 break; 00200 default: 00201 av_log(s, AV_LOG_ERROR, "unsupported stream type; compression_type=%i\n", compression_type); 00202 return 0; 00203 } 00204 00205 if (ea->sample_rate == -1) 00206 ea->sample_rate = revision==3 ? 48000 : 22050; 00207 00208 return 1; 00209 } 00210 00211 /* 00212 * Process EACS sound header 00213 * return 1 if success, 0 if invalid format, otherwise AVERROR_xxx 00214 */ 00215 static int process_audio_header_eacs(AVFormatContext *s) 00216 { 00217 EaDemuxContext *ea = s->priv_data; 00218 AVIOContext *pb = s->pb; 00219 int compression_type; 00220 00221 ea->sample_rate = ea->big_endian ? avio_rb32(pb) : avio_rl32(pb); 00222 ea->bytes = avio_r8(pb); /* 1=8-bit, 2=16-bit */ 00223 ea->num_channels = avio_r8(pb); 00224 compression_type = avio_r8(pb); 00225 avio_skip(pb, 13); 00226 00227 switch (compression_type) { 00228 case 0: 00229 switch (ea->bytes) { 00230 case 1: ea->audio_codec = CODEC_ID_PCM_S8; break; 00231 case 2: ea->audio_codec = CODEC_ID_PCM_S16LE; break; 00232 } 00233 break; 00234 case 1: ea->audio_codec = CODEC_ID_PCM_MULAW; ea->bytes = 1; break; 00235 case 2: ea->audio_codec = CODEC_ID_ADPCM_IMA_EA_EACS; break; 00236 default: 00237 av_log (s, AV_LOG_ERROR, "unsupported stream type; audio compression_type=%i\n", compression_type); 00238 } 00239 00240 return 1; 00241 } 00242 00243 /* 00244 * Process SEAD sound header 00245 * return 1 if success, 0 if invalid format, otherwise AVERROR_xxx 00246 */ 00247 static int process_audio_header_sead(AVFormatContext *s) 00248 { 00249 EaDemuxContext *ea = s->priv_data; 00250 AVIOContext *pb = s->pb; 00251 00252 ea->sample_rate = avio_rl32(pb); 00253 ea->bytes = avio_rl32(pb); /* 1=8-bit, 2=16-bit */ 00254 ea->num_channels = avio_rl32(pb); 00255 ea->audio_codec = CODEC_ID_ADPCM_IMA_EA_SEAD; 00256 00257 return 1; 00258 } 00259 00260 static int process_video_header_mdec(AVFormatContext *s) 00261 { 00262 EaDemuxContext *ea = s->priv_data; 00263 AVIOContext *pb = s->pb; 00264 avio_skip(pb, 4); 00265 ea->width = avio_rl16(pb); 00266 ea->height = avio_rl16(pb); 00267 ea->time_base = (AVRational){1,15}; 00268 ea->video_codec = CODEC_ID_MDEC; 00269 return 1; 00270 } 00271 00272 static int process_video_header_vp6(AVFormatContext *s) 00273 { 00274 EaDemuxContext *ea = s->priv_data; 00275 AVIOContext *pb = s->pb; 00276 00277 avio_skip(pb, 16); 00278 ea->time_base.den = avio_rl32(pb); 00279 ea->time_base.num = avio_rl32(pb); 00280 ea->video_codec = CODEC_ID_VP6; 00281 00282 return 1; 00283 } 00284 00285 /* 00286 * Process EA file header 00287 * Returns 1 if the EA file is valid and successfully opened, 0 otherwise 00288 */ 00289 static int process_ea_header(AVFormatContext *s) { 00290 uint32_t blockid, size = 0; 00291 EaDemuxContext *ea = s->priv_data; 00292 AVIOContext *pb = s->pb; 00293 int i; 00294 00295 for (i=0; i<5 && (!ea->audio_codec || !ea->video_codec); i++) { 00296 unsigned int startpos = avio_tell(pb); 00297 int err = 0; 00298 00299 blockid = avio_rl32(pb); 00300 size = avio_rl32(pb); 00301 if (i == 0) 00302 ea->big_endian = size > 0x000FFFFF; 00303 if (ea->big_endian) 00304 size = av_bswap32(size); 00305 00306 switch (blockid) { 00307 case ISNh_TAG: 00308 if (avio_rl32(pb) != EACS_TAG) { 00309 av_log (s, AV_LOG_ERROR, "unknown 1SNh headerid\n"); 00310 return 0; 00311 } 00312 err = process_audio_header_eacs(s); 00313 break; 00314 00315 case SCHl_TAG : 00316 case SHEN_TAG : 00317 blockid = avio_rl32(pb); 00318 if (blockid == GSTR_TAG) { 00319 avio_skip(pb, 4); 00320 } else if ((blockid & 0xFFFF)!=PT00_TAG) { 00321 av_log (s, AV_LOG_ERROR, "unknown SCHl headerid\n"); 00322 return 0; 00323 } 00324 err = process_audio_header_elements(s); 00325 break; 00326 00327 case SEAD_TAG: 00328 err = process_audio_header_sead(s); 00329 break; 00330 00331 case MVIh_TAG : 00332 ea->video_codec = CODEC_ID_CMV; 00333 ea->time_base = (AVRational){0,0}; 00334 break; 00335 00336 case kVGT_TAG: 00337 ea->video_codec = CODEC_ID_TGV; 00338 ea->time_base = (AVRational){0,0}; 00339 break; 00340 00341 case mTCD_TAG : 00342 err = process_video_header_mdec(s); 00343 break; 00344 00345 case MPCh_TAG: 00346 ea->video_codec = CODEC_ID_MPEG2VIDEO; 00347 break; 00348 00349 case pQGT_TAG: 00350 case TGQs_TAG: 00351 ea->video_codec = CODEC_ID_TGQ; 00352 break; 00353 00354 case pIQT_TAG: 00355 ea->video_codec = CODEC_ID_TQI; 00356 break; 00357 00358 case MADk_TAG : 00359 ea->video_codec = CODEC_ID_MAD; 00360 break; 00361 00362 case MVhd_TAG : 00363 err = process_video_header_vp6(s); 00364 break; 00365 } 00366 00367 if (err < 0) { 00368 av_log(s, AV_LOG_ERROR, "error parsing header: %i\n", err); 00369 return err; 00370 } 00371 00372 avio_seek(pb, startpos + size, SEEK_SET); 00373 } 00374 00375 avio_seek(pb, 0, SEEK_SET); 00376 00377 return 1; 00378 } 00379 00380 00381 static int ea_probe(AVProbeData *p) 00382 { 00383 switch (AV_RL32(&p->buf[0])) { 00384 case ISNh_TAG: 00385 case SCHl_TAG: 00386 case SEAD_TAG: 00387 case SHEN_TAG: 00388 case kVGT_TAG: 00389 case MADk_TAG: 00390 case MPCh_TAG: 00391 case MVhd_TAG: 00392 case MVIh_TAG: 00393 break; 00394 default: 00395 return 0; 00396 } 00397 if (AV_RL32(&p->buf[4]) > 0xfffff && AV_RB32(&p->buf[4]) > 0xfffff) 00398 return 0; 00399 return AVPROBE_SCORE_MAX; 00400 } 00401 00402 static int ea_read_header(AVFormatContext *s, 00403 AVFormatParameters *ap) 00404 { 00405 EaDemuxContext *ea = s->priv_data; 00406 AVStream *st; 00407 00408 if (!process_ea_header(s)) 00409 return AVERROR(EIO); 00410 00411 if (ea->video_codec) { 00412 /* initialize the video decoder stream */ 00413 st = av_new_stream(s, 0); 00414 if (!st) 00415 return AVERROR(ENOMEM); 00416 ea->video_stream_index = st->index; 00417 st->codec->codec_type = AVMEDIA_TYPE_VIDEO; 00418 st->codec->codec_id = ea->video_codec; 00419 st->codec->codec_tag = 0; /* no fourcc */ 00420 st->codec->time_base = ea->time_base; 00421 st->codec->width = ea->width; 00422 st->codec->height = ea->height; 00423 } 00424 00425 if (ea->audio_codec) { 00426 if (ea->num_channels <= 0) { 00427 av_log(s, AV_LOG_WARNING, "Unsupported number of channels: %d\n", ea->num_channels); 00428 ea->audio_codec = 0; 00429 return 1; 00430 } 00431 if (ea->sample_rate <= 0) { 00432 av_log(s, AV_LOG_ERROR, "Unsupported sample rate: %d\n", ea->sample_rate); 00433 ea->audio_codec = 0; 00434 return 1; 00435 } 00436 00437 /* initialize the audio decoder stream */ 00438 st = av_new_stream(s, 0); 00439 if (!st) 00440 return AVERROR(ENOMEM); 00441 av_set_pts_info(st, 33, 1, ea->sample_rate); 00442 st->codec->codec_type = AVMEDIA_TYPE_AUDIO; 00443 st->codec->codec_id = ea->audio_codec; 00444 st->codec->codec_tag = 0; /* no tag */ 00445 st->codec->channels = ea->num_channels; 00446 st->codec->sample_rate = ea->sample_rate; 00447 st->codec->bits_per_coded_sample = ea->bytes * 8; 00448 st->codec->bit_rate = st->codec->channels * st->codec->sample_rate * 00449 st->codec->bits_per_coded_sample / 4; 00450 st->codec->block_align = st->codec->channels*st->codec->bits_per_coded_sample; 00451 ea->audio_stream_index = st->index; 00452 ea->audio_frame_counter = 0; 00453 } 00454 00455 return 1; 00456 } 00457 00458 static int ea_read_packet(AVFormatContext *s, 00459 AVPacket *pkt) 00460 { 00461 EaDemuxContext *ea = s->priv_data; 00462 AVIOContext *pb = s->pb; 00463 int ret = 0; 00464 int packet_read = 0; 00465 unsigned int chunk_type, chunk_size; 00466 int key = 0; 00467 int av_uninit(num_samples); 00468 00469 while (!packet_read) { 00470 chunk_type = avio_rl32(pb); 00471 chunk_size = (ea->big_endian ? avio_rb32(pb) : avio_rl32(pb)) - 8; 00472 00473 switch (chunk_type) { 00474 /* audio data */ 00475 case ISNh_TAG: 00476 /* header chunk also contains data; skip over the header portion*/ 00477 avio_skip(pb, 32); 00478 chunk_size -= 32; 00479 case ISNd_TAG: 00480 case SCDl_TAG: 00481 case SNDC_TAG: 00482 case SDEN_TAG: 00483 if (!ea->audio_codec) { 00484 avio_skip(pb, chunk_size); 00485 break; 00486 } else if (ea->audio_codec == CODEC_ID_PCM_S16LE_PLANAR || 00487 ea->audio_codec == CODEC_ID_MP3) { 00488 num_samples = avio_rl32(pb); 00489 avio_skip(pb, 8); 00490 chunk_size -= 12; 00491 } 00492 ret = av_get_packet(pb, pkt, chunk_size); 00493 if (ret < 0) 00494 return ret; 00495 pkt->stream_index = ea->audio_stream_index; 00496 pkt->pts = 90000; 00497 pkt->pts *= ea->audio_frame_counter; 00498 pkt->pts /= ea->sample_rate; 00499 00500 switch (ea->audio_codec) { 00501 case CODEC_ID_ADPCM_EA: 00502 /* 2 samples/byte, 1 or 2 samples per frame depending 00503 * on stereo; chunk also has 12-byte header */ 00504 ea->audio_frame_counter += ((chunk_size - 12) * 2) / 00505 ea->num_channels; 00506 break; 00507 case CODEC_ID_PCM_S16LE_PLANAR: 00508 case CODEC_ID_MP3: 00509 ea->audio_frame_counter += num_samples; 00510 break; 00511 default: 00512 ea->audio_frame_counter += chunk_size / 00513 (ea->bytes * ea->num_channels); 00514 } 00515 00516 packet_read = 1; 00517 break; 00518 00519 /* ending tag */ 00520 case 0: 00521 case ISNe_TAG: 00522 case SCEl_TAG: 00523 case SEND_TAG: 00524 case SEEN_TAG: 00525 ret = AVERROR(EIO); 00526 packet_read = 1; 00527 break; 00528 00529 case MVIh_TAG: 00530 case kVGT_TAG: 00531 case pQGT_TAG: 00532 case TGQs_TAG: 00533 case MADk_TAG: 00534 key = AV_PKT_FLAG_KEY; 00535 case MVIf_TAG: 00536 case fVGT_TAG: 00537 case MADm_TAG: 00538 case MADe_TAG: 00539 avio_seek(pb, -8, SEEK_CUR); // include chunk preamble 00540 chunk_size += 8; 00541 goto get_video_packet; 00542 00543 case mTCD_TAG: 00544 avio_skip(pb, 8); // skip ea dct header 00545 chunk_size -= 8; 00546 goto get_video_packet; 00547 00548 case MV0K_TAG: 00549 case MPCh_TAG: 00550 case pIQT_TAG: 00551 key = AV_PKT_FLAG_KEY; 00552 case MV0F_TAG: 00553 get_video_packet: 00554 ret = av_get_packet(pb, pkt, chunk_size); 00555 if (ret < 0) 00556 return ret; 00557 pkt->stream_index = ea->video_stream_index; 00558 pkt->flags |= key; 00559 packet_read = 1; 00560 break; 00561 00562 default: 00563 avio_skip(pb, chunk_size); 00564 break; 00565 } 00566 } 00567 00568 return ret; 00569 } 00570 00571 AVInputFormat ff_ea_demuxer = { 00572 "ea", 00573 NULL_IF_CONFIG_SMALL("Electronic Arts Multimedia Format"), 00574 sizeof(EaDemuxContext), 00575 ea_probe, 00576 ea_read_header, 00577 ea_read_packet, 00578 };