]> git.sesse.net Git - ffmpeg/blobdiff - libavformat/electronicarts.c
set bytes per sample in the context
[ffmpeg] / libavformat / electronicarts.c
index 9a9e770918b688100b093e819b449eac766dd056..02099545f404058a2f8383b2c5af694afa54787d 100644 (file)
 
 #define SCHl_TAG MKTAG('S', 'C', 'H', 'l')
 #define PT00_TAG MKTAG('P', 'T', 0x0, 0x0)
+#define GSTR_TAG MKTAG('G', 'S', 'T', 'R')
 #define SCDl_TAG MKTAG('S', 'C', 'D', 'l')
 #define SCEl_TAG MKTAG('S', 'C', 'E', 'l')
-
-#define EA_SAMPLE_RATE 22050
-#define EA_BITS_PER_SAMPLE 16
-#define EA_PREAMBLE_SIZE 8
+#define MVhd_TAG MKTAG('M', 'V', 'h', 'd')
+#define MV0K_TAG MKTAG('M', 'V', '0', 'K')
+#define MV0F_TAG MKTAG('M', 'V', '0', 'F')
 
 typedef struct EaDemuxContext {
-    int width;
-    int height;
+    int big_endian;
+
+    int video_codec;
+    AVRational time_base;
     int video_stream_index;
-    int track_count;
 
+    int audio_codec;
     int audio_stream_index;
     int audio_frame_counter;
 
     int64_t audio_pts;
-    int64_t video_pts;
-    int video_pts_inc;
-    float fps;
 
+    int bytes;
+    int sample_rate;
     int num_channels;
     int num_samples;
-    int compression_type;
 } EaDemuxContext;
 
 static uint32_t read_arbitary(ByteIOContext *pb) {
@@ -72,28 +72,24 @@ static uint32_t read_arbitary(ByteIOContext *pb) {
 }
 
 /*
- * Process WVE file header
- * Returns 1 if the WVE file is valid and successfully opened, 0 otherwise
+ * Process PT/GSTR sound header
+ * return 1 if success, 0 if invalid format, otherwise AVERROR_xxx
  */
-static int process_ea_header(AVFormatContext *s) {
-    int inHeader;
-    uint32_t blockid, size;
+static int process_audio_header_elements(AVFormatContext *s)
+{
+    int inHeader = 1;
     EaDemuxContext *ea = s->priv_data;
     ByteIOContext *pb = &s->pb;
+    int compression_type = -1, revision = -1;
 
-    if (get_le32(pb) != SCHl_TAG)
-        return 0;
-    size = get_le32(pb);
-    if (get_le32(pb) != PT00_TAG) {
-        av_log (s, AV_LOG_ERROR, "PT header missing\n");
-        return 0;
-    }
+    ea->bytes = 2;
+    ea->sample_rate = -1;
+    ea->num_channels = 1;
 
-    inHeader = 1;
     while (inHeader) {
         int inSubheader;
         uint8_t byte;
-        byte = get_byte(pb) & 0xFF;
+        byte = get_byte(pb);
 
         switch (byte) {
         case 0xFD:
@@ -101,16 +97,24 @@ static int process_ea_header(AVFormatContext *s) {
             inSubheader = 1;
             while (inSubheader) {
                 uint8_t subbyte;
-                subbyte = get_byte(pb) & 0xFF;
+                subbyte = get_byte(pb);
 
                 switch (subbyte) {
+                case 0x80:
+                    revision = read_arbitary(pb);
+                    av_log (s, AV_LOG_INFO, "revision (element 0x80) set to 0x%08x\n", revision);
+                    break;
                 case 0x82:
                     ea->num_channels = read_arbitary(pb);
                     av_log (s, AV_LOG_INFO, "num_channels (element 0x82) set to 0x%08x\n", ea->num_channels);
                     break;
                 case 0x83:
-                    ea->compression_type = read_arbitary(pb);
-                    av_log (s, AV_LOG_INFO, "compression_type (element 0x83) set to 0x%08x\n", ea->compression_type);
+                    compression_type = read_arbitary(pb);
+                    av_log (s, AV_LOG_INFO, "compression_type (element 0x83) set to 0x%08x\n", compression_type);
+                    break;
+                case 0x84:
+                    ea->sample_rate = read_arbitary(pb);
+                    av_log (s, AV_LOG_INFO, "sample_rate (element 0x84) set to %i\n", ea->sample_rate);
                     break;
                 case 0x85:
                     ea->num_samples = read_arbitary(pb);
@@ -121,6 +125,11 @@ static int process_ea_header(AVFormatContext *s) {
                     av_log (s, AV_LOG_INFO, "exited audio subheader\n");
                     inSubheader = 0;
                     break;
+                case 0xFF:
+                    av_log (s, AV_LOG_INFO, "end of header block reached (within audio subheader)\n");
+                    inSubheader = 0;
+                    inHeader = 0;
+                    break;
                 default:
                     av_log (s, AV_LOG_INFO, "element 0x%02x set to 0x%08x\n", subbyte, read_arbitary(pb));
                     break;
@@ -137,13 +146,80 @@ static int process_ea_header(AVFormatContext *s) {
         }
     }
 
-    if ((ea->num_channels != 2) || (ea->compression_type != 7)) {
-        av_log (s, AV_LOG_ERROR, "unsupported stream type\n");
+    switch (compression_type) {
+    case  0: ea->audio_codec = CODEC_ID_PCM_S16LE; break;
+    case  7: ea->audio_codec = CODEC_ID_ADPCM_EA; break;
+    default:
+        av_log(s, AV_LOG_ERROR, "unsupported stream type; compression_type=%i\n", compression_type);
         return 0;
     }
 
-    /* skip to the start of the data */
-    url_fseek(pb, size, SEEK_SET);
+    if (ea->sample_rate == -1)
+        ea->sample_rate = revision==3 ? 48000 : 22050;
+
+    return 1;
+}
+
+static int process_video_header_vp6(AVFormatContext *s)
+{
+    EaDemuxContext *ea = s->priv_data;
+    ByteIOContext *pb = &s->pb;
+
+    url_fskip(pb, 16);
+    ea->time_base.den = get_le32(pb);
+    ea->time_base.num = get_le32(pb);
+    ea->video_codec = CODEC_ID_VP6;
+
+    return 1;
+}
+
+/*
+ * Process EA file header
+ * Returns 1 if the EA file is valid and successfully opened, 0 otherwise
+ */
+static int process_ea_header(AVFormatContext *s) {
+    uint32_t blockid, size = 0;
+    EaDemuxContext *ea = s->priv_data;
+    ByteIOContext *pb = &s->pb;
+    int i;
+
+    for (i=0; i<5 && (!ea->audio_codec || !ea->video_codec); i++) {
+        unsigned int startpos = url_ftell(pb);
+        int err = 0;
+
+        blockid = get_le32(pb);
+        size = get_le32(pb);
+        if (i == 0)
+            ea->big_endian = size > 0x000FFFFF;
+        if (ea->big_endian)
+            size = bswap_32(size);
+
+        switch (blockid) {
+            case SCHl_TAG :
+                blockid = get_le32(pb);
+                if (blockid == GSTR_TAG) {
+                    url_fskip(pb, 4);
+                } else if (blockid != PT00_TAG) {
+                    av_log (s, AV_LOG_ERROR, "unknown SCHl headerid\n");
+                    return 0;
+                }
+                err = process_audio_header_elements(s);
+                break;
+
+            case MVhd_TAG :
+                err = process_video_header_vp6(s);
+                break;
+        }
+
+        if (err < 0) {
+            av_log(s, AV_LOG_ERROR, "error parsing header: %i\n", err);
+            return err;
+        }
+
+        url_fseek(pb, startpos + size, SEEK_SET);
+    }
+
+    url_fseek(pb, 0, SEEK_SET);
 
     return 1;
 }
@@ -151,10 +227,13 @@ static int process_ea_header(AVFormatContext *s) {
 
 static int ea_probe(AVProbeData *p)
 {
-    if (AV_RL32(&p->buf[0]) != SCHl_TAG)
-        return 0;
+    uint32_t tag;
+
+    tag = AV_RL32(&p->buf[0]);
+    if (tag == SCHl_TAG || tag == MVhd_TAG)
+        return AVPROBE_SCORE_MAX;
 
-    return AVPROBE_SCORE_MAX;
+    return 0;
 }
 
 static int ea_read_header(AVFormatContext *s,
@@ -166,29 +245,29 @@ static int ea_read_header(AVFormatContext *s,
     if (!process_ea_header(s))
         return AVERROR(EIO);
 
-#if 0
-    /* initialize the video decoder stream */
-    st = av_new_stream(s, 0);
-    if (!st)
-        return AVERROR(ENOMEM);
-    av_set_pts_info(st, 33, 1, 90000);
-    ea->video_stream_index = st->index;
-    st->codec->codec_type = CODEC_TYPE_VIDEO;
-    st->codec->codec_id = CODEC_ID_EA_MJPEG;
-    st->codec->codec_tag = 0;  /* no fourcc */
-#endif
+    if (ea->time_base.num && ea->time_base.den) {
+        /* initialize the video decoder stream */
+        st = av_new_stream(s, 0);
+        if (!st)
+            return AVERROR(ENOMEM);
+        ea->video_stream_index = st->index;
+        st->codec->codec_type = CODEC_TYPE_VIDEO;
+        st->codec->codec_id = ea->video_codec;
+        st->codec->codec_tag = 0;  /* no fourcc */
+        st->codec->time_base = ea->time_base;
+    }
 
     /* initialize the audio decoder stream */
     st = av_new_stream(s, 0);
     if (!st)
         return AVERROR(ENOMEM);
-    av_set_pts_info(st, 33, 1, EA_SAMPLE_RATE);
+    av_set_pts_info(st, 33, 1, ea->sample_rate);
     st->codec->codec_type = CODEC_TYPE_AUDIO;
-    st->codec->codec_id = CODEC_ID_ADPCM_EA;
+    st->codec->codec_id = ea->audio_codec;
     st->codec->codec_tag = 0;  /* no tag */
     st->codec->channels = ea->num_channels;
-    st->codec->sample_rate = EA_SAMPLE_RATE;
-    st->codec->bits_per_sample = EA_BITS_PER_SAMPLE;
+    st->codec->sample_rate = ea->sample_rate;
+    st->codec->bits_per_sample = ea->bytes * 8;
     st->codec->bit_rate = st->codec->channels * st->codec->sample_rate *
         st->codec->bits_per_sample / 4;
     st->codec->block_align = st->codec->channels * st->codec->bits_per_sample;
@@ -206,15 +285,12 @@ static int ea_read_packet(AVFormatContext *s,
     ByteIOContext *pb = &s->pb;
     int ret = 0;
     int packet_read = 0;
-    unsigned char preamble[EA_PREAMBLE_SIZE];
     unsigned int chunk_type, chunk_size;
+    int key = 0;
 
     while (!packet_read) {
-
-        if (get_buffer(pb, preamble, EA_PREAMBLE_SIZE) != EA_PREAMBLE_SIZE)
-            return AVERROR(EIO);
-        chunk_type = AV_RL32(&preamble[0]);
-        chunk_size = AV_RL32(&preamble[4]) - EA_PREAMBLE_SIZE;
+        chunk_type = get_le32(pb);
+        chunk_size = (ea->big_endian ? get_be32(pb) : get_le32(pb)) - 8;
 
         switch (chunk_type) {
         /* audio data */
@@ -226,7 +302,7 @@ static int ea_read_packet(AVFormatContext *s,
                     pkt->stream_index = ea->audio_stream_index;
                     pkt->pts = 90000;
                     pkt->pts *= ea->audio_frame_counter;
-                    pkt->pts /= EA_SAMPLE_RATE;
+                    pkt->pts /= ea->sample_rate;
 
                     /* 2 samples/byte, 1 or 2 samples per frame depending
                      * on stereo; chunk also has 12-byte header */
@@ -243,14 +319,23 @@ static int ea_read_packet(AVFormatContext *s,
             packet_read = 1;
             break;
 
+        case MV0K_TAG:
+            key = PKT_FLAG_KEY;
+        case MV0F_TAG:
+            ret = av_get_packet(pb, pkt, chunk_size);
+            if (ret != chunk_size)
+                ret = AVERROR_IO;
+            else {
+                pkt->stream_index = ea->video_stream_index;
+                pkt->flags |= key;
+            }
+            packet_read = 1;
+            break;
+
         default:
             url_fseek(pb, chunk_size, SEEK_CUR);
             break;
         }
-
-        /* ending packet */
-        if (chunk_type == SCEl_TAG) {
-        }
     }
 
     return ret;