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 CodecID video_codec;
00067 AVRational time_base;
00068 int width, height;
00069 int video_stream_index;
00070
00071 enum CodecID audio_codec;
00072 int audio_stream_index;
00073 int audio_frame_counter;
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 (!pb->eof_reached && 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 (!pb->eof_reached && 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 = CODEC_ID_PCM_S16LE; break;
00179 case 7: ea->audio_codec = CODEC_ID_ADPCM_EA; break;
00180 case -1:
00181 switch (revision) {
00182 case 1: ea->audio_codec = CODEC_ID_ADPCM_EA_R1; break;
00183 case 2: ea->audio_codec = CODEC_ID_ADPCM_EA_R2; break;
00184 case 3: ea->audio_codec = CODEC_ID_ADPCM_EA_R3; break;
00185 case -1: break;
00186 default:
00187 av_log(s, AV_LOG_ERROR, "unsupported stream type; revision=%i\n", revision);
00188 return 0;
00189 }
00190 switch (revision2) {
00191 case 8: ea->audio_codec = CODEC_ID_PCM_S16LE_PLANAR; break;
00192 case 10: ea->audio_codec = CODEC_ID_ADPCM_EA_R2; break;
00193 case 16: ea->audio_codec = CODEC_ID_MP3; break;
00194 case -1: break;
00195 default:
00196 ea->audio_codec = CODEC_ID_NONE;
00197 av_log(s, AV_LOG_ERROR, "unsupported stream type; revision2=%i\n", revision2);
00198 return 0;
00199 }
00200 break;
00201 default:
00202 av_log(s, AV_LOG_ERROR, "unsupported stream type; compression_type=%i\n", compression_type);
00203 return 0;
00204 }
00205
00206 if (ea->sample_rate == -1)
00207 ea->sample_rate = revision==3 ? 48000 : 22050;
00208
00209 return 1;
00210 }
00211
00212
00213
00214
00215
00216 static int process_audio_header_eacs(AVFormatContext *s)
00217 {
00218 EaDemuxContext *ea = s->priv_data;
00219 AVIOContext *pb = s->pb;
00220 int compression_type;
00221
00222 ea->sample_rate = ea->big_endian ? avio_rb32(pb) : avio_rl32(pb);
00223 ea->bytes = avio_r8(pb);
00224 ea->num_channels = avio_r8(pb);
00225 compression_type = avio_r8(pb);
00226 avio_skip(pb, 13);
00227
00228 switch (compression_type) {
00229 case 0:
00230 switch (ea->bytes) {
00231 case 1: ea->audio_codec = CODEC_ID_PCM_S8; break;
00232 case 2: ea->audio_codec = CODEC_ID_PCM_S16LE; break;
00233 }
00234 break;
00235 case 1: ea->audio_codec = CODEC_ID_PCM_MULAW; ea->bytes = 1; break;
00236 case 2: ea->audio_codec = CODEC_ID_ADPCM_IMA_EA_EACS; break;
00237 default:
00238 av_log (s, AV_LOG_ERROR, "unsupported stream type; audio compression_type=%i\n", compression_type);
00239 }
00240
00241 return 1;
00242 }
00243
00244
00245
00246
00247
00248 static int process_audio_header_sead(AVFormatContext *s)
00249 {
00250 EaDemuxContext *ea = s->priv_data;
00251 AVIOContext *pb = s->pb;
00252
00253 ea->sample_rate = avio_rl32(pb);
00254 ea->bytes = avio_rl32(pb);
00255 ea->num_channels = avio_rl32(pb);
00256 ea->audio_codec = CODEC_ID_ADPCM_IMA_EA_SEAD;
00257
00258 return 1;
00259 }
00260
00261 static int process_video_header_mdec(AVFormatContext *s)
00262 {
00263 EaDemuxContext *ea = s->priv_data;
00264 AVIOContext *pb = s->pb;
00265 avio_skip(pb, 4);
00266 ea->width = avio_rl16(pb);
00267 ea->height = avio_rl16(pb);
00268 ea->time_base = (AVRational){1,15};
00269 ea->video_codec = CODEC_ID_MDEC;
00270 return 1;
00271 }
00272
00273 static int process_video_header_vp6(AVFormatContext *s)
00274 {
00275 EaDemuxContext *ea = s->priv_data;
00276 AVIOContext *pb = s->pb;
00277
00278 avio_skip(pb, 16);
00279 ea->time_base.den = avio_rl32(pb);
00280 ea->time_base.num = avio_rl32(pb);
00281 ea->video_codec = CODEC_ID_VP6;
00282
00283 return 1;
00284 }
00285
00286
00287
00288
00289
00290 static int process_ea_header(AVFormatContext *s) {
00291 uint32_t blockid, size = 0;
00292 EaDemuxContext *ea = s->priv_data;
00293 AVIOContext *pb = s->pb;
00294 int i;
00295
00296 for (i=0; i<5 && (!ea->audio_codec || !ea->video_codec); i++) {
00297 unsigned int startpos = avio_tell(pb);
00298 int err = 0;
00299
00300 blockid = avio_rl32(pb);
00301 size = avio_rl32(pb);
00302 if (i == 0)
00303 ea->big_endian = size > 0x000FFFFF;
00304 if (ea->big_endian)
00305 size = av_bswap32(size);
00306
00307 switch (blockid) {
00308 case ISNh_TAG:
00309 if (avio_rl32(pb) != EACS_TAG) {
00310 av_log (s, AV_LOG_ERROR, "unknown 1SNh headerid\n");
00311 return 0;
00312 }
00313 err = process_audio_header_eacs(s);
00314 break;
00315
00316 case SCHl_TAG :
00317 case SHEN_TAG :
00318 blockid = avio_rl32(pb);
00319 if (blockid == GSTR_TAG) {
00320 avio_skip(pb, 4);
00321 } else if ((blockid & 0xFFFF)!=PT00_TAG) {
00322 av_log (s, AV_LOG_ERROR, "unknown SCHl headerid\n");
00323 return 0;
00324 }
00325 err = process_audio_header_elements(s);
00326 break;
00327
00328 case SEAD_TAG:
00329 err = process_audio_header_sead(s);
00330 break;
00331
00332 case MVIh_TAG :
00333 ea->video_codec = CODEC_ID_CMV;
00334 ea->time_base = (AVRational){0,0};
00335 break;
00336
00337 case kVGT_TAG:
00338 ea->video_codec = CODEC_ID_TGV;
00339 ea->time_base = (AVRational){0,0};
00340 break;
00341
00342 case mTCD_TAG :
00343 err = process_video_header_mdec(s);
00344 break;
00345
00346 case MPCh_TAG:
00347 ea->video_codec = CODEC_ID_MPEG2VIDEO;
00348 break;
00349
00350 case pQGT_TAG:
00351 case TGQs_TAG:
00352 ea->video_codec = CODEC_ID_TGQ;
00353 break;
00354
00355 case pIQT_TAG:
00356 ea->video_codec = CODEC_ID_TQI;
00357 break;
00358
00359 case MADk_TAG :
00360 ea->video_codec = CODEC_ID_MAD;
00361 break;
00362
00363 case MVhd_TAG :
00364 err = process_video_header_vp6(s);
00365 break;
00366 }
00367
00368 if (err < 0) {
00369 av_log(s, AV_LOG_ERROR, "error parsing header: %i\n", err);
00370 return err;
00371 }
00372
00373 avio_seek(pb, startpos + size, SEEK_SET);
00374 }
00375
00376 avio_seek(pb, 0, SEEK_SET);
00377
00378 return 1;
00379 }
00380
00381
00382 static int ea_probe(AVProbeData *p)
00383 {
00384 switch (AV_RL32(&p->buf[0])) {
00385 case ISNh_TAG:
00386 case SCHl_TAG:
00387 case SEAD_TAG:
00388 case SHEN_TAG:
00389 case kVGT_TAG:
00390 case MADk_TAG:
00391 case MPCh_TAG:
00392 case MVhd_TAG:
00393 case MVIh_TAG:
00394 break;
00395 default:
00396 return 0;
00397 }
00398 if (AV_RL32(&p->buf[4]) > 0xfffff && AV_RB32(&p->buf[4]) > 0xfffff)
00399 return 0;
00400 return AVPROBE_SCORE_MAX;
00401 }
00402
00403 static int ea_read_header(AVFormatContext *s,
00404 AVFormatParameters *ap)
00405 {
00406 EaDemuxContext *ea = s->priv_data;
00407 AVStream *st;
00408
00409 if (!process_ea_header(s))
00410 return AVERROR(EIO);
00411
00412 if (ea->video_codec) {
00413
00414 st = avformat_new_stream(s, NULL);
00415 if (!st)
00416 return AVERROR(ENOMEM);
00417 ea->video_stream_index = st->index;
00418 st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
00419 st->codec->codec_id = ea->video_codec;
00420 st->codec->codec_tag = 0;
00421 st->codec->time_base = ea->time_base;
00422 st->codec->width = ea->width;
00423 st->codec->height = ea->height;
00424 }
00425
00426 if (ea->audio_codec) {
00427 if (ea->num_channels <= 0) {
00428 av_log(s, AV_LOG_WARNING, "Unsupported number of channels: %d\n", ea->num_channels);
00429 ea->audio_codec = 0;
00430 return 1;
00431 }
00432 if (ea->sample_rate <= 0) {
00433 av_log(s, AV_LOG_ERROR, "Unsupported sample rate: %d\n", ea->sample_rate);
00434 ea->audio_codec = 0;
00435 return 1;
00436 }
00437 if (ea->bytes <= 0) {
00438 av_log(s, AV_LOG_ERROR, "Invalid number of bytes per sample: %d\n", ea->bytes);
00439 ea->audio_codec = CODEC_ID_NONE;
00440 return 1;
00441 }
00442
00443
00444 st = avformat_new_stream(s, NULL);
00445 if (!st)
00446 return AVERROR(ENOMEM);
00447 avpriv_set_pts_info(st, 33, 1, ea->sample_rate);
00448 st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
00449 st->codec->codec_id = ea->audio_codec;
00450 st->codec->codec_tag = 0;
00451 st->codec->channels = ea->num_channels;
00452 st->codec->sample_rate = ea->sample_rate;
00453 st->codec->bits_per_coded_sample = ea->bytes * 8;
00454 st->codec->bit_rate = st->codec->channels * st->codec->sample_rate *
00455 st->codec->bits_per_coded_sample / 4;
00456 st->codec->block_align = st->codec->channels*st->codec->bits_per_coded_sample;
00457 ea->audio_stream_index = st->index;
00458 ea->audio_frame_counter = 0;
00459 }
00460
00461 return 1;
00462 }
00463
00464 static int ea_read_packet(AVFormatContext *s,
00465 AVPacket *pkt)
00466 {
00467 EaDemuxContext *ea = s->priv_data;
00468 AVIOContext *pb = s->pb;
00469 int ret = 0;
00470 int packet_read = 0;
00471 unsigned int chunk_type, chunk_size;
00472 int key = 0;
00473 int av_uninit(num_samples);
00474
00475 while (!packet_read) {
00476 chunk_type = avio_rl32(pb);
00477 chunk_size = (ea->big_endian ? avio_rb32(pb) : avio_rl32(pb)) - 8;
00478
00479 switch (chunk_type) {
00480
00481 case ISNh_TAG:
00482
00483 avio_skip(pb, 32);
00484 chunk_size -= 32;
00485 case ISNd_TAG:
00486 case SCDl_TAG:
00487 case SNDC_TAG:
00488 case SDEN_TAG:
00489 if (!ea->audio_codec) {
00490 avio_skip(pb, chunk_size);
00491 break;
00492 } else if (ea->audio_codec == CODEC_ID_PCM_S16LE_PLANAR ||
00493 ea->audio_codec == CODEC_ID_MP3) {
00494 num_samples = avio_rl32(pb);
00495 avio_skip(pb, 8);
00496 chunk_size -= 12;
00497 }
00498 ret = av_get_packet(pb, pkt, chunk_size);
00499 if (ret < 0)
00500 return ret;
00501 pkt->stream_index = ea->audio_stream_index;
00502 pkt->pts = 90000;
00503 pkt->pts *= ea->audio_frame_counter;
00504 pkt->pts /= ea->sample_rate;
00505
00506 switch (ea->audio_codec) {
00507 case CODEC_ID_ADPCM_EA:
00508
00509
00510 ea->audio_frame_counter += ((chunk_size - 12) * 2) /
00511 ea->num_channels;
00512 break;
00513 case CODEC_ID_PCM_S16LE_PLANAR:
00514 case CODEC_ID_MP3:
00515 ea->audio_frame_counter += num_samples;
00516 break;
00517 default:
00518 ea->audio_frame_counter += chunk_size /
00519 (ea->bytes * ea->num_channels);
00520 }
00521
00522 packet_read = 1;
00523 break;
00524
00525
00526 case 0:
00527 case ISNe_TAG:
00528 case SCEl_TAG:
00529 case SEND_TAG:
00530 case SEEN_TAG:
00531 ret = AVERROR(EIO);
00532 packet_read = 1;
00533 break;
00534
00535 case MVIh_TAG:
00536 case kVGT_TAG:
00537 case pQGT_TAG:
00538 case TGQs_TAG:
00539 case MADk_TAG:
00540 key = AV_PKT_FLAG_KEY;
00541 case MVIf_TAG:
00542 case fVGT_TAG:
00543 case MADm_TAG:
00544 case MADe_TAG:
00545 avio_seek(pb, -8, SEEK_CUR);
00546 chunk_size += 8;
00547 goto get_video_packet;
00548
00549 case mTCD_TAG:
00550 avio_skip(pb, 8);
00551 chunk_size -= 8;
00552 goto get_video_packet;
00553
00554 case MV0K_TAG:
00555 case MPCh_TAG:
00556 case pIQT_TAG:
00557 key = AV_PKT_FLAG_KEY;
00558 case MV0F_TAG:
00559 get_video_packet:
00560 ret = av_get_packet(pb, pkt, chunk_size);
00561 if (ret < 0)
00562 return ret;
00563 pkt->stream_index = ea->video_stream_index;
00564 pkt->flags |= key;
00565 packet_read = 1;
00566 break;
00567
00568 default:
00569 avio_skip(pb, chunk_size);
00570 break;
00571 }
00572 }
00573
00574 return ret;
00575 }
00576
00577 AVInputFormat ff_ea_demuxer = {
00578 .name = "ea",
00579 .long_name = NULL_IF_CONFIG_SMALL("Electronic Arts Multimedia Format"),
00580 .priv_data_size = sizeof(EaDemuxContext),
00581 .read_probe = ea_probe,
00582 .read_header = ea_read_header,
00583 .read_packet = ea_read_packet,
00584 };