added support for mode 6 images.
[swftools.git] / src / swfcombine.c
index 1fad28d..9dedbf8 100644 (file)
@@ -234,7 +234,7 @@ void args_callback_usage(char*name)
     printf("OR:    %s [-rXYov] --cat [-xysf] [(name1|#id1)=]slavefile1 .. [-xysf] [(nameN|#idN)=]slavefileN\n", name);
     printf("OR:    %s [-rXYomlcv] --dummy [-xys] [file]\n", name);
     printf("\n");
-    printf("-o outputfile       --output    explicitly specify output file. (otherwise, output.swf will be used\n");
+    printf("-o outputfile       --output    explicitly specify output file. (otherwise, output.swf will be used)\n");
     printf("-t                  --stack     place each slave in a seperate frame (no master movie)\n");
     printf("-T                  --stack1    place each slave in the first frame (no master movie)\n");
     printf("-m                  --merge     Don't store the slaves in Sprites/MovieClips\n");
@@ -259,14 +259,18 @@ static void makestackmaster(SWF*swf)
     int t;
     SRECT box;
     int fileversion = 1;
+    int frameRate = 256;
+    RGBA rgb;
+    rgb.r=rgb.b=rgb.g=0;
     memset(&box, 0, sizeof(box));
 
     /* scan all slaves for bounding box */
-    for(t=0;t<numslaves;t++)
+    for(t=numslaves-1;t>=0;t--)
     {
        SWF head;
        int ret;
        int fi=open(slave_filename[t],O_RDONLY);
+       TAG*tag;
        if(fi<0 || swf_ReadSWF(fi, &head)<0) {
            logf("<fatal> Couldn't open/read %s.", slave_filename[t]);
            exit(1);
@@ -276,6 +280,17 @@ static void makestackmaster(SWF*swf)
                slave_filename[t], 
                head.movieSize.xmin, head.movieSize.ymin,
                head.movieSize.xmax, head.movieSize.ymax);
+
+       tag = head.firstTag;
+       while(tag) {
+           if(tag->id == ST_SETBACKGROUNDCOLOR && tag->len>=3) {
+               rgb.r = tag->data[0];
+               rgb.g = tag->data[1];
+               rgb.b = tag->data[2];
+           }
+           tag=tag->next;
+       }
+       frameRate = head.frameRate;
        if(head.fileVersion > fileversion)
            fileversion = head.fileVersion;
        if(!t)
@@ -297,12 +312,13 @@ static void makestackmaster(SWF*swf)
     }
 
     memset(swf, 0, sizeof(SWF));
+    swf->fileVersion = fileversion;
+    swf->movieSize = box;
+    swf->frameRate = frameRate;
 
-    swf->firstTag = swf_InsertTag(tag, ST_SETBACKGROUNDCOLOR);
+    swf->firstTag = swf_InsertTag(0, ST_SETBACKGROUNDCOLOR);
     tag = swf->firstTag;
-    swf_SetU8(tag, 0);
-    swf_SetU8(tag, 0);
-    swf_SetU8(tag, 0);
+    swf_SetRGB(tag, &rgb);
     
     for(t=0;t<numslaves;t++)
     {
@@ -328,6 +344,7 @@ static void makestackmaster(SWF*swf)
        }
     }
     tag = swf_InsertTag(tag, ST_END);
+    logf("<verbose> temporary SWF created");
 }
 
 static char* slavename = 0;
@@ -343,7 +360,7 @@ static char masterbitmap[65536];
 int get_free_id(char*bitmap)
 {
     int t;
-    for(t=1;t<65537;t++)
+    for(t=1;t<65536;t++)
        if(!bitmap[t]) {
            bitmap[t] = 1;
            return t;
@@ -351,15 +368,52 @@ int get_free_id(char*bitmap)
     return -1;
 }
 
-void jpeg_assert()
+void jpeg_assert(SWF*master, SWF*slave)
 {
+    /* TODO: if there's a jpegtable found, store it
+       and handle it together with the flash file
+       headers */
+
+    /* check that master and slave don't have both
+       jpegtables (which would be fatal) */
+    int pos;
+    TAG *mpos=0, *spos=0;
+    TAG *mtag,*stag;
+    pos = 0;
+    mtag = master->firstTag;
+    stag = slave->firstTag;
+    while(mtag)
+    {
+       if(mtag->id  == ST_JPEGTABLES)
+           mpos = mtag;
+       mtag = mtag->next;
+    }
+    while(stag)
+    {
+       if(stag->id == ST_JPEGTABLES)
+           spos = stag;
+       stag = stag->next;
+    }
+    if(mpos && spos)
+    {
+       if(spos->len == mpos->len &&
+       !memcmp(spos->data, mpos->data, mpos->len))
+       {
+           // ok, both have jpegtables, but they're identical.
+           // delete one and don't throw an error
+           swf_DeleteTag(spos);
+           spos = 0;
+       }
+    }
+    if(spos && mpos) {
+       logf("<error> Master and slave have incompatible JPEGTABLES.");
+    }
 }
 
 TAG* write_sprite_defines(TAG*tag, SWF*sprite)
 {
-    int pos = 0;
     TAG*rtag = sprite->firstTag;
-    while(rtag) {
+    while(rtag && rtag->id!=ST_END) {
        if(!swf_isAllowedSpriteTag(rtag)) {
            logf("<debug> processing sprite tag %02x", tag->id);
            if(swf_isDefiningTag(rtag))
@@ -405,7 +459,7 @@ TAG* write_sprite_defines(TAG*tag, SWF*sprite)
                }
            }
        }
-       pos++;
+       rtag = rtag->next;
     }
     return tag;
 }
@@ -451,11 +505,16 @@ void write_changepos(TAG*output, TAG*tag)
                if(flags&2) {
                    swf_SetU16(output, swf_GetU16(tag)); //id
                }
+               // flags & 4
                if(flags&4) {
                    swf_GetMatrix(tag, &m);
-                   matrix_adjust(&m);
-                   swf_SetMatrix(tag, &m);
+               } else {
+                   swf_GetMatrix(0, &m);
                }
+               matrix_adjust(&m);
+               swf_SetMatrix(output, &m);
+
+               //swf_ResetReadBits(tag);
                swf_SetBlock(output, &tag->data[tag->pos], tag->len - tag->pos);
                break;
            }
@@ -463,9 +522,12 @@ void write_changepos(TAG*output, TAG*tag)
                MATRIX m;
                swf_SetU16(output, swf_GetU16(tag)); //id
                swf_SetU16(output, swf_GetU16(tag)); //depth
+               
                swf_GetMatrix(tag, &m);
                matrix_adjust(&m);
-               swf_SetMatrix(tag, &m);
+               swf_SetMatrix(output, &m);
+               
+               //swf_ResetReadBits(tag);
                swf_SetBlock(output, &tag->data[tag->pos], tag->len - tag->pos);
                break;
            }
@@ -481,7 +543,6 @@ void write_changepos(TAG*output, TAG*tag)
 
 TAG* write_sprite(TAG*tag, SWF*sprite, int spriteid, int replaceddefine)
 {
-    int pos = 0;
     TAG* definespritetag;
     TAG* rtag;
     int tmp;
@@ -491,7 +552,6 @@ TAG* write_sprite(TAG*tag, SWF*sprite, int spriteid, int replaceddefine)
     swf_SetU16(tag, sprite->frameCount);
     logf ("<notice> sprite id is %d", spriteid);
 
-    // write slave(2) (body)
     tmp = sprite->frameCount;
     logf("<debug> %d frames to go",tmp);
 
@@ -506,21 +566,21 @@ TAG* write_sprite(TAG*tag, SWF*sprite, int spriteid, int replaceddefine)
     if(config.overlay && !config.isframe) {
        tag = swf_InsertTag(tag, ST_PLACEOBJECT2);
        swf_SetU8(tag, 2); //flags: character
-       swf_SetU16(tag, 0); //depth
+       swf_SetU16(tag, 1); //depth
        swf_SetU16(tag, replaceddefine); //id
     }
 
     rtag = sprite->firstTag;
-    while(rtag)
+    while(rtag && rtag->id!=ST_END)
     {
        if (swf_isAllowedSpriteTag(rtag)) {
 
-           changedepth(rtag, +1);
            logf("<debug> [sprite main] write tag %02x (%d bytes in body)", 
                    rtag->id, rtag->len);
-
            tag = swf_InsertTag(tag, rtag->id);
-           write_changepos(rtag, tag);
+           write_changepos(tag, rtag);
+       
+           changedepth(tag, +2);
 
            if(tag->id == ST_SHOWFRAME)
            {
@@ -528,7 +588,9 @@ TAG* write_sprite(TAG*tag, SWF*sprite, int spriteid, int replaceddefine)
                logf("<debug> %d frames to go",tmp);
            }
        }
+       rtag = rtag->next;
     }
+    tag = swf_InsertTag(tag, ST_END);
     return tag;
 }
 
@@ -549,11 +611,11 @@ TAG* write_master(TAG*tag, SWF*master, SWF*slave, int spriteid, int replaceddefi
     TAG* rtag = master->firstTag;
     TAG* stag = slave->firstTag;
 
-    while(rtag)
+    while(rtag && rtag->id!=ST_END)
     {
        if(rtag->id == ST_SHOWFRAME && outputslave)
        {
-           while(stag) {
+           while(stag && stag->id!=ST_END) {
                if(stag->id == ST_SHOWFRAME) {
                    stag = stag->next;
                    sframe++;
@@ -579,9 +641,9 @@ TAG* write_master(TAG*tag, SWF*master, SWF*slave, int spriteid, int replaceddefi
            {
                if(config.overlay)
                {
-                   PUT16(rtag->data, replaceddefine);
                    tag = swf_InsertTag(tag, rtag->id);
                    swf_SetBlock(tag, rtag->data, rtag->len);
+                   swf_SetDefineID(tag, replaceddefine);
                } else {
                    /* don't write this tag */
                    logf("<verbose> replacing tag %d id %d with sprite", rtag->id
@@ -653,11 +715,11 @@ TAG* write_master(TAG*tag, SWF*master, SWF*slave, int spriteid, int replaceddefi
                swf_SetBlock(tag, rtag->data, rtag->len);
            }
        }
-       tag = tag->next;
+       rtag = rtag->next;
     }
    
     if(outputslave) 
-    while(stag)
+    while(stag && stag->id!=ST_END)
     {
            if(tag_ok_for_slave(stag->id)) {
                tag = swf_InsertTag(tag, stag->id);
@@ -674,18 +736,120 @@ TAG* write_master(TAG*tag, SWF*master, SWF*slave, int spriteid, int replaceddefi
            logf("<warning> Frame \"%s\" doesn't exist in file. No substitution will occur",
                    slavename);
     }
+    tag = swf_InsertTag(tag, ST_END);
     return tag;
 }
 
+void adjustheader(SWF*swf)
+{
+    if(config.framerate)
+       swf->frameRate = config.framerate;
+    if(config.hassizex) {
+       swf->movieSize.xmax = 
+       swf->movieSize.xmin + config.sizex;
+    }
+    if(config.hassizey) {
+       swf->movieSize.ymax = 
+       swf->movieSize.ymin + config.sizey;
+    }
+}
+
 void catcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
 {
-    printf("--cat not implemented yet. Please use the swfcombine from swftools 0.3.1 or 0.2.3.\n");
-    exit(1);
+    char* depths;
+    int t;
+    TAG*tag;
+    TAG*mtag,*stag;
+    if(config.isframe) {
+       logf("<fatal> Can't combine --cat and --frame");
+       exit(1);
+    }
+   
+    tag = master->firstTag;
+    while(tag)
+    {
+       if(swf_isDefiningTag(tag)) {
+           int defineid = swf_GetDefineID(tag);
+           logf("<debug> tagid %02x defines object %d", tag->id, defineid);
+           masterbitmap[defineid] = 1;
+       }
+       tag = tag->next;
+    }
+    
+    swf_Relocate(slave, masterbitmap);
+    jpeg_assert(master, slave);
+    
+    memcpy(newswf, master, sizeof(SWF));
+    adjustheader(newswf);
+
+    tag = newswf->firstTag = swf_InsertTag(0, ST_REFLEX); // to be removed later
+
+    depths = malloc(65536);
+    if(!depths) {
+       logf("<fatal> Couldn't allocate %d bytes of memory", 65536);
+       return;
+    }
+    memset(depths, 0, 65536);
+    mtag = master->firstTag;
+    while(mtag && mtag->id!=ST_END)
+    {
+       int num=1;
+       U16 depth;
+       logf("<debug> [master] write tag %02x (%d bytes in body)", 
+               mtag->id, mtag->len);
+       switch(mtag->id) {
+           case ST_PLACEOBJECT2:
+               num++;
+           case ST_PLACEOBJECT: {
+              depth = swf_GetDepth(mtag);
+              depths[depth] = 1;
+           }
+           break;
+           case ST_REMOVEOBJECT: {
+              depth = swf_GetDepth(mtag);
+              depths[depth] = 0;
+           }
+           break;
+           case ST_REMOVEOBJECT2: {
+              depth = swf_GetDepth(mtag);
+              depths[depth] = 0;
+           }
+           break;
+       }
+       tag = swf_InsertTag(tag, mtag->id);
+       swf_SetBlock(tag, mtag->data, mtag->len);
+
+       mtag = mtag->next;
+    }
+
+    for(t=0;t<65536;t++) 
+    if(depths[t])
+    {
+       char data[16];
+       int len;
+       tag = swf_InsertTag(tag, ST_REMOVEOBJECT2);
+       swf_SetU16(tag, t);
+    }
+    free(depths);
+
+    stag = slave->firstTag;
+    while(stag && stag->id!=ST_END)
+    {
+       logf("<debug> [slave] write tag %02x (%d bytes in body)", 
+               stag->id, stag->len);
+       tag = swf_InsertTag(tag, stag->id);
+       swf_SetBlock(tag, stag->data, stag->len);
+       stag = stag->next;
+    }
+    tag = swf_InsertTag(tag, ST_END);
+
+    tag = newswf->firstTag;
+    newswf->firstTag = newswf->firstTag->next; //remove temporary tag
+    swf_DeleteTag(tag);
 }
 
 void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
 {
-    int pos=0;
     int spriteid = -1;
     int replaceddefine = -1;
     int frame = 0;
@@ -693,22 +857,20 @@ void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
     TAG * tag = master->firstTag;
     
     // set the idtab
-    pos = 0;
-
     while(tag)
     {
        if(swf_isDefiningTag(tag)) {
            int defineid = swf_GetDefineID(tag);
-           logf("<debug> tagid %02x defines object %d", tag, defineid);
+           logf("<debug> tagid %02x defines object %d", tag->id, defineid);
            masterbitmap[defineid] = 1;
        } else if(tag->id == ST_PLACEOBJECT2) {
            char * name = swf_GetName(tag);
            int id = swf_GetPlaceID(tag);
 
            if(name)
-             logf("<verbose> tagid %02x places object %d named \"%s\"", tag, id, name);
+             logf("<verbose> tagid %02x places object %d named \"%s\"", tag->id, id, name);
            else
-             logf("<verbose> tagid %02x places object %d (no name)", tag, id);
+             logf("<verbose> tagid %02x places object %d (no name)", tag->id, id);
 
            if ((name && slavename && !strcmp(name,slavename)) || 
                (!slavename && id==slaveid)) {
@@ -729,6 +891,7 @@ void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
                logf("<notice> Slave file attached to frame %d (%s).", frame, name);
            }
        }
+       tag = tag->next;
     };
 
     if (spriteid<0 && !config.isframe) {
@@ -742,7 +905,7 @@ void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
     }
 
     swf_Relocate (slave, masterbitmap);
-    jpeg_assert();
+    jpeg_assert(slave, master);
     
     if (config.overlay)
        replaceddefine = get_free_id(masterbitmap);
@@ -750,7 +913,9 @@ void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
     // write file 
 
     memcpy(newswf, master, sizeof(SWF));
-    newswf->firstTag = tag = swf_InsertTag(0, ST_REFLEX);
+    adjustheader(newswf);
+
+    newswf->firstTag = tag = swf_InsertTag(0, ST_REFLEX); // to be removed later
 
     if (config.antistream) {
        if (config.merge) {
@@ -768,6 +933,10 @@ void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
            tag = write_master(tag, master, slave, spriteid, replaceddefine, 
                FLAGS_WRITEDEFINES|FLAGS_WRITENONDEFINES|FLAGS_WRITESPRITE);
     }
+
+    tag = newswf->firstTag;
+    newswf->firstTag = newswf->firstTag->next; //remove temporary tag
+    swf_DeleteTag(tag);
 }
 
 void combine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
@@ -968,11 +1137,14 @@ int main(int argn, char *argv[])
        }
     }
 
-    fi = open(outputname, O_RDWR|O_TRUNC|O_CREAT);
+    fi = open(outputname, O_RDWR|O_TRUNC|O_CREAT, 0777);
+
     if(config.zlib)
        swf_WriteSWC(fi, &newswf);
-    else
+    else {
+       newswf.compressed = 0;
        swf_WriteSWF(fi, &newswf);
+    }
     close(fi);
     return 0;
 }