Libav
|
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 FFmpeg. 00006 * 00007 * FFmpeg 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 * FFmpeg 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 FFmpeg; 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(ByteIOContext *pb) { 00081 uint8_t size, byte; 00082 int i; 00083 uint32_t word; 00084 00085 size = get_byte(pb); 00086 00087 word = 0; 00088 for (i = 0; i < size; i++) { 00089 byte = get_byte(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 ByteIOContext *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 (inHeader) { 00113 int inSubheader; 00114 uint8_t byte; 00115 byte = get_byte(pb); 00116 00117 switch (byte) { 00118 case 0xFD: 00119 av_log (s, AV_LOG_DEBUG, "entered audio subheader\n"); 00120 inSubheader = 1; 00121 while (inSubheader) { 00122 uint8_t subbyte; 00123 subbyte = get_byte(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 ByteIOContext *pb = s->pb; 00219 int compression_type; 00220 00221 ea->sample_rate = ea->big_endian ? get_be32(pb) : get_le32(pb); 00222 ea->bytes = get_byte(pb); /* 1=8-bit, 2=16-bit */ 00223 ea->num_channels = get_byte(pb); 00224 compression_type = get_byte(pb); 00225 url_fskip(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 ByteIOContext *pb = s->pb; 00251 00252 ea->sample_rate = get_le32(pb); 00253 ea->bytes = get_le32(pb); /* 1=8-bit, 2=16-bit */ 00254 ea->num_channels = get_le32(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 ByteIOContext *pb = s->pb; 00264 url_fskip(pb, 4); 00265 ea->width = get_le16(pb); 00266 ea->height = get_le16(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 ByteIOContext *pb = s->pb; 00276 00277 url_fskip(pb, 16); 00278 ea->time_base.den = get_le32(pb); 00279 ea->time_base.num = get_le32(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 ByteIOContext *pb = s->pb; 00293 int i; 00294 00295 for (i=0; i<5 && (!ea->audio_codec || !ea->video_codec); i++) { 00296 unsigned int startpos = url_ftell(pb); 00297 int err = 0; 00298 00299 blockid = get_le32(pb); 00300 size = get_le32(pb); 00301 if (i == 0) 00302 ea->big_endian = size > 0x000FFFFF; 00303 if (ea->big_endian) 00304 size = bswap_32(size); 00305 00306 switch (blockid) { 00307 case ISNh_TAG: 00308 if (get_le32(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 = get_le32(pb); 00318 if (blockid == GSTR_TAG) { 00319 url_fskip(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 url_fseek(pb, startpos + size, SEEK_SET); 00373 } 00374 00375 url_fseek(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 /* initialize the audio decoder stream */ 00427 st = av_new_stream(s, 0); 00428 if (!st) 00429 return AVERROR(ENOMEM); 00430 av_set_pts_info(st, 33, 1, ea->sample_rate); 00431 st->codec->codec_type = AVMEDIA_TYPE_AUDIO; 00432 st->codec->codec_id = ea->audio_codec; 00433 st->codec->codec_tag = 0; /* no tag */ 00434 st->codec->channels = ea->num_channels; 00435 st->codec->sample_rate = ea->sample_rate; 00436 st->codec->bits_per_coded_sample = ea->bytes * 8; 00437 st->codec->bit_rate = st->codec->channels * st->codec->sample_rate * 00438 st->codec->bits_per_coded_sample / 4; 00439 st->codec->block_align = st->codec->channels*st->codec->bits_per_coded_sample; 00440 ea->audio_stream_index = st->index; 00441 ea->audio_frame_counter = 0; 00442 } 00443 00444 return 1; 00445 } 00446 00447 static int ea_read_packet(AVFormatContext *s, 00448 AVPacket *pkt) 00449 { 00450 EaDemuxContext *ea = s->priv_data; 00451 ByteIOContext *pb = s->pb; 00452 int ret = 0; 00453 int packet_read = 0; 00454 unsigned int chunk_type, chunk_size; 00455 int key = 0; 00456 int av_uninit(num_samples); 00457 00458 while (!packet_read) { 00459 chunk_type = get_le32(pb); 00460 chunk_size = ea->big_endian ? get_be32(pb) : get_le32(pb); 00461 if (chunk_size <= 8) 00462 return AVERROR_INVALIDDATA; 00463 chunk_size -= 8; 00464 00465 switch (chunk_type) { 00466 /* audio data */ 00467 case ISNh_TAG: 00468 /* header chunk also contains data; skip over the header portion*/ 00469 if (chunk_size < 32) 00470 return AVERROR_INVALIDDATA; 00471 url_fskip(pb, 32); 00472 chunk_size -= 32; 00473 case ISNd_TAG: 00474 case SCDl_TAG: 00475 case SNDC_TAG: 00476 case SDEN_TAG: 00477 if (!ea->audio_codec) { 00478 url_fskip(pb, chunk_size); 00479 break; 00480 } else if (ea->audio_codec == CODEC_ID_PCM_S16LE_PLANAR || 00481 ea->audio_codec == CODEC_ID_MP3) { 00482 num_samples = get_le32(pb); 00483 url_fskip(pb, 8); 00484 chunk_size -= 12; 00485 } 00486 ret = av_get_packet(pb, pkt, chunk_size); 00487 if (ret < 0) 00488 return ret; 00489 pkt->stream_index = ea->audio_stream_index; 00490 pkt->pts = 90000; 00491 pkt->pts *= ea->audio_frame_counter; 00492 pkt->pts /= ea->sample_rate; 00493 00494 switch (ea->audio_codec) { 00495 case CODEC_ID_ADPCM_EA: 00496 /* 2 samples/byte, 1 or 2 samples per frame depending 00497 * on stereo; chunk also has 12-byte header */ 00498 ea->audio_frame_counter += ((chunk_size - 12) * 2) / 00499 ea->num_channels; 00500 break; 00501 case CODEC_ID_PCM_S16LE_PLANAR: 00502 case CODEC_ID_MP3: 00503 ea->audio_frame_counter += num_samples; 00504 break; 00505 default: 00506 ea->audio_frame_counter += chunk_size / 00507 (ea->bytes * ea->num_channels); 00508 } 00509 00510 packet_read = 1; 00511 break; 00512 00513 /* ending tag */ 00514 case 0: 00515 case ISNe_TAG: 00516 case SCEl_TAG: 00517 case SEND_TAG: 00518 case SEEN_TAG: 00519 ret = AVERROR(EIO); 00520 packet_read = 1; 00521 break; 00522 00523 case MVIh_TAG: 00524 case kVGT_TAG: 00525 case pQGT_TAG: 00526 case TGQs_TAG: 00527 case MADk_TAG: 00528 key = AV_PKT_FLAG_KEY; 00529 case MVIf_TAG: 00530 case fVGT_TAG: 00531 case MADm_TAG: 00532 case MADe_TAG: 00533 url_fseek(pb, -8, SEEK_CUR); // include chunk preamble 00534 chunk_size += 8; 00535 goto get_video_packet; 00536 00537 case mTCD_TAG: 00538 url_fseek(pb, 8, SEEK_CUR); // skip ea dct header 00539 chunk_size -= 8; 00540 goto get_video_packet; 00541 00542 case MV0K_TAG: 00543 case MPCh_TAG: 00544 case pIQT_TAG: 00545 key = AV_PKT_FLAG_KEY; 00546 case MV0F_TAG: 00547 get_video_packet: 00548 ret = av_get_packet(pb, pkt, chunk_size); 00549 if (ret < 0) 00550 return ret; 00551 pkt->stream_index = ea->video_stream_index; 00552 pkt->flags |= key; 00553 packet_read = 1; 00554 break; 00555 00556 default: 00557 url_fseek(pb, chunk_size, SEEK_CUR); 00558 break; 00559 } 00560 } 00561 00562 return ret; 00563 } 00564 00565 AVInputFormat ea_demuxer = { 00566 "ea", 00567 NULL_IF_CONFIG_SMALL("Electronic Arts Multimedia Format"), 00568 sizeof(EaDemuxContext), 00569 ea_probe, 00570 ea_read_header, 00571 ea_read_packet, 00572 };