00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00028 #include "libavutil/intreadwrite.h"
00029 #include "avformat.h"
00030 #include "internal.h"
00031
00032 #define SCHl_TAG MKTAG('S', 'C', 'H', 'l')
00033 #define SEAD_TAG MKTAG('S', 'E', 'A', 'D')
00034 #define SNDC_TAG MKTAG('S', 'N', 'D', 'C')
00035 #define SEND_TAG MKTAG('S', 'E', 'N', 'D')
00036 #define SHEN_TAG MKTAG('S', 'H', 'E', 'N')
00037 #define SDEN_TAG MKTAG('S', 'D', 'E', 'N')
00038 #define SEEN_TAG MKTAG('S', 'E', 'E', 'N')
00039 #define ISNh_TAG MKTAG('1', 'S', 'N', 'h')
00040 #define EACS_TAG MKTAG('E', 'A', 'C', 'S')
00041 #define ISNd_TAG MKTAG('1', 'S', 'N', 'd')
00042 #define ISNe_TAG MKTAG('1', 'S', 'N', 'e')
00043 #define PT00_TAG MKTAG('P', 'T', 0x0, 0x0)
00044 #define GSTR_TAG MKTAG('G', 'S', 'T', 'R')
00045 #define SCDl_TAG MKTAG('S', 'C', 'D', 'l')
00046 #define SCEl_TAG MKTAG('S', 'C', 'E', 'l')
00047 #define kVGT_TAG MKTAG('k', 'V', 'G', 'T')
00048 #define fVGT_TAG MKTAG('f', 'V', 'G', 'T')
00049 #define mTCD_TAG MKTAG('m', 'T', 'C', 'D')
00050 #define MADk_TAG MKTAG('M', 'A', 'D', 'k')
00051 #define MADm_TAG MKTAG('M', 'A', 'D', 'm')
00052 #define MADe_TAG MKTAG('M', 'A', 'D', 'e')
00053 #define MPCh_TAG MKTAG('M', 'P', 'C', 'h')
00054 #define TGQs_TAG MKTAG('T', 'G', 'Q', 's')
00055 #define pQGT_TAG MKTAG('p', 'Q', 'G', 'T')
00056 #define pIQT_TAG MKTAG('p', 'I', 'Q', 'T')
00057 #define MVhd_TAG MKTAG('M', 'V', 'h', 'd')
00058 #define MV0K_TAG MKTAG('M', 'V', '0', 'K')
00059 #define MV0F_TAG MKTAG('M', 'V', '0', 'F')
00060 #define MVIh_TAG MKTAG('M', 'V', 'I', 'h')
00061 #define MVIf_TAG MKTAG('M', 'V', 'I', 'f')
00062
00063 typedef struct EaDemuxContext {
00064 int big_endian;
00065
00066 enum AVCodecID video_codec;
00067 AVRational time_base;
00068 int width, height;
00069 int nb_frames;
00070 int video_stream_index;
00071
00072 enum AVCodecID audio_codec;
00073 int audio_stream_index;
00074
00075 int bytes;
00076 int sample_rate;
00077 int num_channels;
00078 int num_samples;
00079 } EaDemuxContext;
00080
00081 static uint32_t read_arbitary(AVIOContext *pb) {
00082 uint8_t size, byte;
00083 int i;
00084 uint32_t word;
00085
00086 size = avio_r8(pb);
00087
00088 word = 0;
00089 for (i = 0; i < size; i++) {
00090 byte = avio_r8(pb);
00091 word <<= 8;
00092 word |= byte;
00093 }
00094
00095 return word;
00096 }
00097
00098
00099
00100
00101
00102 static int process_audio_header_elements(AVFormatContext *s)
00103 {
00104 int inHeader = 1;
00105 EaDemuxContext *ea = s->priv_data;
00106 AVIOContext *pb = s->pb;
00107 int compression_type = -1, revision = -1, revision2 = -1;
00108
00109 ea->bytes = 2;
00110 ea->sample_rate = -1;
00111 ea->num_channels = 1;
00112
00113 while (!url_feof(pb) && inHeader) {
00114 int inSubheader;
00115 uint8_t byte;
00116 byte = avio_r8(pb);
00117
00118 switch (byte) {
00119 case 0xFD:
00120 av_log (s, AV_LOG_DEBUG, "entered audio subheader\n");
00121 inSubheader = 1;
00122 while (!url_feof(pb) && inSubheader) {
00123 uint8_t subbyte;
00124 subbyte = avio_r8(pb);
00125
00126 switch (subbyte) {
00127 case 0x80:
00128 revision = read_arbitary(pb);
00129 av_log (s, AV_LOG_DEBUG, "revision (element 0x80) set to 0x%08x\n", revision);
00130 break;
00131 case 0x82:
00132 ea->num_channels = read_arbitary(pb);
00133 av_log (s, AV_LOG_DEBUG, "num_channels (element 0x82) set to 0x%08x\n", ea->num_channels);
00134 break;
00135 case 0x83:
00136 compression_type = read_arbitary(pb);
00137 av_log (s, AV_LOG_DEBUG, "compression_type (element 0x83) set to 0x%08x\n", compression_type);
00138 break;
00139 case 0x84:
00140 ea->sample_rate = read_arbitary(pb);
00141 av_log (s, AV_LOG_DEBUG, "sample_rate (element 0x84) set to %i\n", ea->sample_rate);
00142 break;
00143 case 0x85:
00144 ea->num_samples = read_arbitary(pb);
00145 av_log (s, AV_LOG_DEBUG, "num_samples (element 0x85) set to 0x%08x\n", ea->num_samples);
00146 break;
00147 case 0x8A:
00148 av_log (s, AV_LOG_DEBUG, "element 0x%02x set to 0x%08x\n", subbyte, read_arbitary(pb));
00149 av_log (s, AV_LOG_DEBUG, "exited audio subheader\n");
00150 inSubheader = 0;
00151 break;
00152 case 0xA0:
00153 revision2 = read_arbitary(pb);
00154 av_log (s, AV_LOG_DEBUG, "revision2 (element 0xA0) set to 0x%08x\n", revision2);
00155 break;
00156 case 0xFF:
00157 av_log (s, AV_LOG_DEBUG, "end of header block reached (within audio subheader)\n");
00158 inSubheader = 0;
00159 inHeader = 0;
00160 break;
00161 default:
00162 av_log (s, AV_LOG_DEBUG, "element 0x%02x set to 0x%08x\n", subbyte, read_arbitary(pb));
00163 break;
00164 }
00165 }
00166 break;
00167 case 0xFF:
00168 av_log (s, AV_LOG_DEBUG, "end of header block reached\n");
00169 inHeader = 0;
00170 break;
00171 default:
00172 av_log (s, AV_LOG_DEBUG, "header element 0x%02x set to 0x%08x\n", byte, read_arbitary(pb));
00173 break;
00174 }
00175 }
00176
00177 switch (compression_type) {
00178 case 0: ea->audio_codec = AV_CODEC_ID_PCM_S16LE; break;
00179 case 7: ea->audio_codec = AV_CODEC_ID_ADPCM_EA; break;
00180 case -1:
00181 switch (revision) {
00182 case 1: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R1; break;
00183 case 2: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R2; break;
00184 case 3: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R3; break;
00185 case -1: break;
00186 default:
00187 av_log_ask_for_sample(s, "unsupported stream type; revision=%i\n", revision);
00188 return 0;
00189 }
00190 switch (revision2) {
00191 case 8: ea->audio_codec = AV_CODEC_ID_PCM_S16LE_PLANAR; break;
00192 case 10:
00193 switch (revision) {
00194 case -1:
00195 case 2: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R1; break;
00196 case 3: ea->audio_codec = AV_CODEC_ID_ADPCM_EA_R2; break;
00197 default:
00198 av_log_ask_for_sample(s, "unsupported stream type; revision=%i, revision2=%i\n", revision, revision2);
00199 return 0;
00200 }
00201 break;
00202 case 16: ea->audio_codec = AV_CODEC_ID_MP3; break;
00203 case -1: break;
00204 default:
00205 ea->audio_codec = AV_CODEC_ID_NONE;
00206 av_log_ask_for_sample(s, "unsupported stream type; revision2=%i\n", revision2);
00207 return 0;
00208 }
00209 break;
00210 default:
00211 av_log_ask_for_sample(s, "unsupported stream type; compression_type=%i\n", compression_type);
00212 return 0;
00213 }
00214
00215 if (ea->sample_rate == -1)
00216 ea->sample_rate = revision==3 ? 48000 : 22050;
00217
00218 return 1;
00219 }
00220
00221
00222
00223
00224
00225 static int process_audio_header_eacs(AVFormatContext *s)
00226 {
00227 EaDemuxContext *ea = s->priv_data;
00228 AVIOContext *pb = s->pb;
00229 int compression_type;
00230
00231 ea->sample_rate = ea->big_endian ? avio_rb32(pb) : avio_rl32(pb);
00232 ea->bytes = avio_r8(pb);
00233 ea->num_channels = avio_r8(pb);
00234 compression_type = avio_r8(pb);
00235 avio_skip(pb, 13);
00236
00237 switch (compression_type) {
00238 case 0:
00239 switch (ea->bytes) {
00240 case 1: ea->audio_codec = AV_CODEC_ID_PCM_S8; break;
00241 case 2: ea->audio_codec = AV_CODEC_ID_PCM_S16LE; break;
00242 }
00243 break;
00244 case 1: ea->audio_codec = AV_CODEC_ID_PCM_MULAW; ea->bytes = 1; break;
00245 case 2: ea->audio_codec = AV_CODEC_ID_ADPCM_IMA_EA_EACS; break;
00246 default:
00247 av_log_ask_for_sample(s, "unsupported stream type; audio compression_type=%i\n", compression_type);
00248 }
00249
00250 return 1;
00251 }
00252
00253
00254
00255
00256
00257 static int process_audio_header_sead(AVFormatContext *s)
00258 {
00259 EaDemuxContext *ea = s->priv_data;
00260 AVIOContext *pb = s->pb;
00261
00262 ea->sample_rate = avio_rl32(pb);
00263 ea->bytes = avio_rl32(pb);
00264 ea->num_channels = avio_rl32(pb);
00265 ea->audio_codec = AV_CODEC_ID_ADPCM_IMA_EA_SEAD;
00266
00267 return 1;
00268 }
00269
00270 static int process_video_header_mdec(AVFormatContext *s)
00271 {
00272 EaDemuxContext *ea = s->priv_data;
00273 AVIOContext *pb = s->pb;
00274 avio_skip(pb, 4);
00275 ea->width = avio_rl16(pb);
00276 ea->height = avio_rl16(pb);
00277 ea->time_base = (AVRational){1,15};
00278 ea->video_codec = AV_CODEC_ID_MDEC;
00279 return 1;
00280 }
00281
00282 static int process_video_header_vp6(AVFormatContext *s)
00283 {
00284 EaDemuxContext *ea = s->priv_data;
00285 AVIOContext *pb = s->pb;
00286
00287 avio_skip(pb, 8);
00288 ea->nb_frames = avio_rl32(pb);
00289 avio_skip(pb, 4);
00290 ea->time_base.den = avio_rl32(pb);
00291 ea->time_base.num = avio_rl32(pb);
00292 ea->video_codec = AV_CODEC_ID_VP6;
00293
00294 return 1;
00295 }
00296
00297 static int process_video_header_cmv(AVFormatContext *s)
00298 {
00299 EaDemuxContext *ea = s->priv_data;
00300 int fps;
00301
00302 avio_skip(s->pb, 10);
00303 fps = avio_rl16(s->pb);
00304 if (fps)
00305 ea->time_base = (AVRational){1, fps};
00306 ea->video_codec = AV_CODEC_ID_CMV;
00307
00308 return 0;
00309 }
00310
00311
00312
00313
00314
00315 static int process_ea_header(AVFormatContext *s) {
00316 uint32_t blockid, size = 0;
00317 EaDemuxContext *ea = s->priv_data;
00318 AVIOContext *pb = s->pb;
00319 int i;
00320
00321 for (i=0; i<5 && (!ea->audio_codec || !ea->video_codec); i++) {
00322 unsigned int startpos = avio_tell(pb);
00323 int err = 0;
00324
00325 blockid = avio_rl32(pb);
00326 size = avio_rl32(pb);
00327 if (i == 0)
00328 ea->big_endian = size > 0x000FFFFF;
00329 if (ea->big_endian)
00330 size = av_bswap32(size);
00331
00332 switch (blockid) {
00333 case ISNh_TAG:
00334 if (avio_rl32(pb) != EACS_TAG) {
00335 av_log_ask_for_sample(s, "unknown 1SNh headerid\n");
00336 return 0;
00337 }
00338 err = process_audio_header_eacs(s);
00339 break;
00340
00341 case SCHl_TAG :
00342 case SHEN_TAG :
00343 blockid = avio_rl32(pb);
00344 if (blockid == GSTR_TAG) {
00345 avio_skip(pb, 4);
00346 } else if ((blockid & 0xFFFF)!=PT00_TAG) {
00347 av_log_ask_for_sample(s, "unknown SCHl headerid\n");
00348 return 0;
00349 }
00350 err = process_audio_header_elements(s);
00351 break;
00352
00353 case SEAD_TAG:
00354 err = process_audio_header_sead(s);
00355 break;
00356
00357 case MVIh_TAG :
00358 err = process_video_header_cmv(s);
00359 break;
00360
00361 case kVGT_TAG:
00362 ea->video_codec = AV_CODEC_ID_TGV;
00363 break;
00364
00365 case mTCD_TAG :
00366 err = process_video_header_mdec(s);
00367 break;
00368
00369 case MPCh_TAG:
00370 ea->video_codec = AV_CODEC_ID_MPEG2VIDEO;
00371 break;
00372
00373 case pQGT_TAG:
00374 case TGQs_TAG:
00375 ea->video_codec = AV_CODEC_ID_TGQ;
00376 break;
00377
00378 case pIQT_TAG:
00379 ea->video_codec = AV_CODEC_ID_TQI;
00380 break;
00381
00382 case MADk_TAG :
00383 ea->video_codec = AV_CODEC_ID_MAD;
00384 break;
00385
00386 case MVhd_TAG :
00387 err = process_video_header_vp6(s);
00388 break;
00389 }
00390
00391 if (err < 0) {
00392 av_log(s, AV_LOG_ERROR, "error parsing header: %i\n", err);
00393 return err;
00394 }
00395
00396 avio_seek(pb, startpos + size, SEEK_SET);
00397 }
00398
00399 avio_seek(pb, 0, SEEK_SET);
00400
00401 return 1;
00402 }
00403
00404
00405 static int ea_probe(AVProbeData *p)
00406 {
00407 switch (AV_RL32(&p->buf[0])) {
00408 case ISNh_TAG:
00409 case SCHl_TAG:
00410 case SEAD_TAG:
00411 case SHEN_TAG:
00412 case kVGT_TAG:
00413 case MADk_TAG:
00414 case MPCh_TAG:
00415 case MVhd_TAG:
00416 case MVIh_TAG:
00417 break;
00418 default:
00419 return 0;
00420 }
00421 if (AV_RL32(&p->buf[4]) > 0xfffff && AV_RB32(&p->buf[4]) > 0xfffff)
00422 return 0;
00423 return AVPROBE_SCORE_MAX;
00424 }
00425
00426 static int ea_read_header(AVFormatContext *s)
00427 {
00428 EaDemuxContext *ea = s->priv_data;
00429 AVStream *st;
00430
00431 if (process_ea_header(s)<=0)
00432 return AVERROR(EIO);
00433
00434 if (ea->video_codec) {
00435
00436 st = avformat_new_stream(s, NULL);
00437 if (!st)
00438 return AVERROR(ENOMEM);
00439 ea->video_stream_index = st->index;
00440 st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
00441 st->codec->codec_id = ea->video_codec;
00442
00443 if (st->codec->codec_id == AV_CODEC_ID_MPEG2VIDEO)
00444 st->need_parsing = AVSTREAM_PARSE_HEADERS;
00445 st->codec->codec_tag = 0;
00446 st->codec->width = ea->width;
00447 st->codec->height = ea->height;
00448 st->duration = st->nb_frames = ea->nb_frames;
00449 if (ea->time_base.num)
00450 avpriv_set_pts_info(st, 64, ea->time_base.num, ea->time_base.den);
00451 st->r_frame_rate =
00452 st->avg_frame_rate = av_inv_q(ea->time_base);
00453 }
00454
00455 if (ea->audio_codec) {
00456 if (ea->num_channels <= 0) {
00457 av_log(s, AV_LOG_WARNING, "Unsupported number of channels: %d\n", ea->num_channels);
00458 ea->audio_codec = 0;
00459 return 1;
00460 }
00461 if (ea->sample_rate <= 0) {
00462 av_log(s, AV_LOG_ERROR, "Unsupported sample rate: %d\n", ea->sample_rate);
00463 ea->audio_codec = 0;
00464 return 1;
00465 }
00466 if (ea->bytes <= 0) {
00467 av_log(s, AV_LOG_ERROR, "Invalid number of bytes per sample: %d\n", ea->bytes);
00468 ea->audio_codec = AV_CODEC_ID_NONE;
00469 return 1;
00470 }
00471
00472
00473 st = avformat_new_stream(s, NULL);
00474 if (!st)
00475 return AVERROR(ENOMEM);
00476 avpriv_set_pts_info(st, 33, 1, ea->sample_rate);
00477 st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
00478 st->codec->codec_id = ea->audio_codec;
00479 st->codec->codec_tag = 0;
00480 st->codec->channels = ea->num_channels;
00481 st->codec->sample_rate = ea->sample_rate;
00482 st->codec->bits_per_coded_sample = ea->bytes * 8;
00483 st->codec->bit_rate = st->codec->channels * st->codec->sample_rate *
00484 st->codec->bits_per_coded_sample / 4;
00485 st->codec->block_align = st->codec->channels*st->codec->bits_per_coded_sample;
00486 ea->audio_stream_index = st->index;
00487 st->start_time = 0;
00488 }
00489
00490 return 1;
00491 }
00492
00493 static int ea_read_packet(AVFormatContext *s,
00494 AVPacket *pkt)
00495 {
00496 EaDemuxContext *ea = s->priv_data;
00497 AVIOContext *pb = s->pb;
00498 int ret = 0;
00499 int packet_read = 0;
00500 int partial_packet = 0;
00501 unsigned int chunk_type, chunk_size;
00502 int key = 0;
00503 int av_uninit(num_samples);
00504
00505 while (!packet_read || partial_packet) {
00506 chunk_type = avio_rl32(pb);
00507 chunk_size = ea->big_endian ? avio_rb32(pb) : avio_rl32(pb);
00508 if (chunk_size <= 8)
00509 return AVERROR_INVALIDDATA;
00510 chunk_size -= 8;
00511
00512 switch (chunk_type) {
00513
00514 case ISNh_TAG:
00515
00516 if (chunk_size < 32)
00517 return AVERROR_INVALIDDATA;
00518 avio_skip(pb, 32);
00519 chunk_size -= 32;
00520 case ISNd_TAG:
00521 case SCDl_TAG:
00522 case SNDC_TAG:
00523 case SDEN_TAG:
00524 if (!ea->audio_codec) {
00525 avio_skip(pb, chunk_size);
00526 break;
00527 } else if (ea->audio_codec == AV_CODEC_ID_PCM_S16LE_PLANAR ||
00528 ea->audio_codec == AV_CODEC_ID_MP3) {
00529 num_samples = avio_rl32(pb);
00530 avio_skip(pb, 8);
00531 chunk_size -= 12;
00532 }
00533 if (partial_packet) {
00534 av_log_ask_for_sample(s, "video header followed by audio packet not supported.\n");
00535 av_free_packet(pkt);
00536 partial_packet = 0;
00537 }
00538 ret = av_get_packet(pb, pkt, chunk_size);
00539 if (ret < 0)
00540 return ret;
00541 pkt->stream_index = ea->audio_stream_index;
00542
00543 switch (ea->audio_codec) {
00544 case AV_CODEC_ID_ADPCM_EA:
00545 case AV_CODEC_ID_ADPCM_EA_R1:
00546 case AV_CODEC_ID_ADPCM_EA_R2:
00547 case AV_CODEC_ID_ADPCM_IMA_EA_EACS:
00548 if (pkt->size >= 4)
00549 pkt->duration = AV_RL32(pkt->data);
00550 break;
00551 case AV_CODEC_ID_ADPCM_EA_R3:
00552 if (pkt->size >= 4)
00553 pkt->duration = AV_RB32(pkt->data);
00554 break;
00555 case AV_CODEC_ID_ADPCM_IMA_EA_SEAD:
00556 pkt->duration = ret * 2 / ea->num_channels;
00557 break;
00558 case AV_CODEC_ID_PCM_S16LE_PLANAR:
00559 case AV_CODEC_ID_MP3:
00560 pkt->duration = num_samples;
00561 break;
00562 default:
00563 pkt->duration = chunk_size / (ea->bytes * ea->num_channels);
00564 }
00565
00566 packet_read = 1;
00567 break;
00568
00569
00570 case 0:
00571 case ISNe_TAG:
00572 case SCEl_TAG:
00573 case SEND_TAG:
00574 case SEEN_TAG:
00575 ret = AVERROR(EIO);
00576 packet_read = 1;
00577 break;
00578
00579 case MVIh_TAG:
00580 case kVGT_TAG:
00581 case pQGT_TAG:
00582 case TGQs_TAG:
00583 case MADk_TAG:
00584 key = AV_PKT_FLAG_KEY;
00585 case MVIf_TAG:
00586 case fVGT_TAG:
00587 case MADm_TAG:
00588 case MADe_TAG:
00589 avio_seek(pb, -8, SEEK_CUR);
00590 chunk_size += 8;
00591 goto get_video_packet;
00592
00593 case mTCD_TAG:
00594 avio_skip(pb, 8);
00595 chunk_size -= 8;
00596 goto get_video_packet;
00597
00598 case MV0K_TAG:
00599 case MPCh_TAG:
00600 case pIQT_TAG:
00601 key = AV_PKT_FLAG_KEY;
00602 case MV0F_TAG:
00603 get_video_packet:
00604 if (partial_packet) {
00605 ret = av_append_packet(pb, pkt, chunk_size);
00606 } else
00607 ret = av_get_packet(pb, pkt, chunk_size);
00608 if (ret < 0) {
00609 packet_read = 1;
00610 break;
00611 }
00612 partial_packet = chunk_type == MVIh_TAG;
00613 pkt->stream_index = ea->video_stream_index;
00614 pkt->flags |= key;
00615 packet_read = 1;
00616 break;
00617
00618 default:
00619 avio_skip(pb, chunk_size);
00620 break;
00621 }
00622 }
00623
00624 if (ret < 0 && partial_packet)
00625 av_free_packet(pkt);
00626 return ret;
00627 }
00628
00629 AVInputFormat ff_ea_demuxer = {
00630 .name = "ea",
00631 .long_name = NULL_IF_CONFIG_SMALL("Electronic Arts Multimedia"),
00632 .priv_data_size = sizeof(EaDemuxContext),
00633 .read_probe = ea_probe,
00634 .read_header = ea_read_header,
00635 .read_packet = ea_read_packet,
00636 };