src/swfcombine.c

   1 /* swfcombine.c
   2    main routine for swfcombine(1), a tool for merging .swf-files.
   3
   4    Part of the swftools package.
   5
   6    Copyright (c) 2001,2002,2003 Matthias Kramm <kramm@quiss.org>
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 2 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program; if not, write to the Free Software
  20    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  21
  22 #include <stdlib.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25 #include "../lib/rfxswf.h"
  26 #include "../lib/args.h"
  27 #include "../lib/log.h"
  28 #include "../config.h"
  29
  30 struct config_t
  31 {
  32    char overlay;
  33    char alloctest;
  34    char clip;
  35    char stack;
  36    char stack1;
  37    char antistream;
  38    char dummy;
  39    char zlib;
  40    char cat;
  41    char merge;
  42    char isframe;
  43    int loglevel;
  44    int sizex;
  45    char hassizex;
  46    int sizey;
  47    char hassizey;
  48    int framerate;
  49    int movex;
  50    int movey;
  51    float scalex;
  52    float scaley;
  53    int mastermovex;
  54    int mastermovey;
  55    float masterscalex;
  56    float masterscaley;
  57 };
  58 struct config_t config;
  59
  60 char * master_filename = 0;
  61 char * master_name = 0;
  62 char * slave_filename[128];
  63 char * slave_name[128];
  64 int slave_movex[128];
  65 int slave_movey[128];
  66 float slave_scalex[128];
  67 float slave_scaley[128];
  68 char slave_isframe[128];
  69 int numslaves = 0;
  70
  71 char * outputname = "output.swf";
  72
  73 int args_callback_option(char*name,char*val) {
  74     if(!strcmp(name,"c"))
  75     {
  76         config.clip = 1;
  77         return 0;
  78     }
  79     else if(!strcmp(name,"l"))
  80     {
  81         config.overlay = 1;
  82         return 0;
  83     }
  84     else if (!strcmp(name, "o"))
  85     {
  86         outputname = val;
  87         return 1;
  88     }
  89     else if (!strcmp(name, "v"))
  90     {
  91         config.loglevel ++;
  92         return 0;
  93     }
  94     else if (!strcmp(name, "a"))
  95     {
  96         config.cat = 1;
  97         return 0;
  98     }
  99     else if (!strcmp(name, "A"))
 100     {
 101         config.alloctest = 1;
 102         return 0;
 103     }
 104     else if (!strcmp(name, "x"))
 105     {
 106         float x = atof(val);
 107         config.movex = (int)(x*20+0.5);
 108         return 1;
 109     }
 110     else if (!strcmp(name, "y"))
 111     {
 112         float y = atof(val);
 113         config.movey = (int)(y*20+0.5);
 114         return 1;
 115     }
 116     else if (!strcmp(name, "m"))
 117     {
 118         config.merge = 1;
 119         return 0;
 120     }
 121     else if (!strcmp(name, "f"))
 122     {
 123         config.isframe = 1;
 124         return 0;
 125     }
 126     else if (!strcmp(name, "d"))
 127     {
 128         config.dummy = 1;
 129         return 0;
 130     }
 131     else if (!strcmp(name, "z"))
 132     {
 133         config.zlib = 1;
 134         return 0;
 135     }
 136     else if (!strcmp(name, "r"))
 137     {
 138
 139         float rate = atof(val);
 140         if ((rate < 0) ||(rate >= 256.0)) {
 141             fprintf(stderr, "Error: You must specify a valid framerate between 1/256 and 255.\n");
 142             exit(1);
 143         }
 144         config.framerate = (int)(rate*256);
 145         return 1;
 146     }
 147     else if (!strcmp(name, "X"))
 148     {
 149         config.sizex = atoi(val)*20;
 150         config.hassizex = 1;
 151         return 1;
 152     }
 153     else if (!strcmp(name, "Y"))
 154     {
 155         config.sizey = atoi(val)*20;
 156         config.hassizey = 1;
 157         return 1;
 158     }
 159     else if (!strcmp(name, "s"))
 160     {
 161         config.scalex = config.scaley = atoi(val)/100.0;
 162         return 1;
 163     }
 164     else if (!strcmp(name, "t") || !strcmp(name, "T"))
 165     {
 166         if(master_filename) {
 167             fprintf(stderr, "error with arguments. Try --help.\n");
 168             exit(1);
 169         }
 170         config.stack = 1;
 171         if(!strcmp(name,"T"))
 172             config.stack1 = 1;
 173         master_filename = "__none__";
 174         return 0;
 175     }
 176     else if (!strcmp(name, "V"))
 177     {
 178         printf("swfcombine - part of %s %s\n", PACKAGE, VERSION);
 179         exit(0);
 180     }
 181     else
 182     {
 183         fprintf(stderr, "Unknown option: -%s\n", name);
 184         exit(1);
 185     }
 186 }
 187
 188 static struct options_t options[] = {
 189 {"o", "output"},
 190 {"t", "stack"},
 191 {"T", "stack1"},
 192 {"m", "merge"},
 193 {"a", "cat"},
 194 {"l", "overlay"},
 195 {"c", "clip"},
 196 {"v", "verbose"},
 197 {"d", "dummy"},
 198 {"f", "frame"},
 199 {"x", "movex"},
 200 {"y", "movey"},
 201 {"s", "scale"},
 202 {"r", "rate"},
 203 {"X", "width"},
 204 {"Y", "height"},
 205 {"z", "zlib"},
 206 {0,0}
 207 };
 208
 209 int args_callback_longoption(char*name,char*val) {
 210     return args_long2shortoption(options, name, val);
 211 }
 212
 213 int args_callback_command(char*name, char*val) {
 214     char*myname = strdup(name);
 215     char*filename;
 216     filename = strchr(myname, '=');
 217     if(filename) {
 218         *filename = 0;
 219         filename++;
 220     } else {
 221         // argument has no explicit name field. guess one from the file name
 222         char*path = strrchr(myname, '/');
 223         char*ext = strrchr(myname, '.');
 224         if(!path) path = myname;
 225         else path ++;
 226         if(ext) *ext = 0;
 227         myname = path;
 228         filename = name;
 229     }
 230
 231     if(!master_filename) {
 232         master_filename = filename;
 233         master_name = myname;
 234         config.mastermovex = config.movex;
 235         config.mastermovey = config.movey;
 236         config.masterscalex = config.scalex;
 237         config.masterscaley = config.scaley;
 238         config.movex = config.movey = 0;
 239         config.scalex = config.scaley = 1.0;
 240     } else {
 241         msg("<verbose> slave entity %s (named \"%s\")\n", filename, myname);
 242
 243         slave_filename[numslaves] = filename;
 244         slave_name[numslaves] = myname;
 245         slave_movex[numslaves] = config.movex;
 246         slave_movey[numslaves] = config.movey;
 247         slave_scalex[numslaves] = config.scalex;
 248         slave_scaley[numslaves] = config.scaley;
 249         slave_isframe[numslaves] = config.isframe;
 250         config.isframe = 0;
 251         config.movex = config.movey = 0;
 252         config.scalex = config.scaley = 1.0;
 253         numslaves ++;
 254     }
 255     return 0;
 256 }
 257
 258 void args_callback_usage(char *name)
 259 {
 260     printf("\n");
 261     printf("Usage: %s [-rXYomlcv] [-f] masterfile [-xysf] [(name1|#id1)=]slavefile1 .. [-xysf] [(nameN|#idN)=]slavefileN\n", name);
 262     printf("OR:    %s [-rXYomv] --stack[1] [-xysf] [(name1|#id1)=]slavefile1 .. [-xysf] [(nameN|#idN)=]slavefileN\n", name);
 263     printf("OR:    %s [-rXYov] --cat [-xysf] [(name1|#id1)=]slavefile1 .. [-xysf] [(nameN|#idN)=]slavefileN\n", name);
 264     printf("OR:    %s [-rXYomlcv] --dummy [-xys] [file]\n", name);
 265     printf("\n");
 266     printf("-o , --output <outputfile>      explicitly specify output file. (otherwise, output.swf will be used)\n");
 267     printf("-t , --stack                   place each slave in a seperate frame (no master movie)\n");
 268     printf("-T , --stack1                  place each slave in the first frame (no master movie)\n");
 269     printf("-m , --merge                   Don't store the slaves in Sprites/MovieClips\n");
 270     printf("-a , --cat                     concatenate all slave files (no master movie)\n");
 271     printf("-l , --overlay                 Don't remove any master objects, only overlay new objects\n");
 272     printf("-c , --clip                    Clip the slave objects by the corresponding master objects\n");
 273     printf("-v , --verbose                 Be verbose. Use more than one -v for greater effect \n");
 274     printf("-d , --dummy                   Don't require slave objects (for changing movie attributes)\n");
 275     printf("-f , --frame                   The following identifier is a frame or framelabel, not an id or objectname\n");
 276     printf("-x , --movex <xpos>            x Adjust position of slave by <xpos> pixels\n");
 277     printf("-y , --movey <ypos>            y Adjust position of slave by <ypos> pixels\n");
 278     printf("-s , --scale <scale>           Adjust size of slave by <scale> percent (e.g. 100% = original size)\n");
 279     printf("-r , --rate <fps>              Set movie framerate to <fps> (frames/sec)\n");
 280     printf("-X , --width <width>           Force movie bbox width to <width> (default: use master width (not with -t))\n");
 281     printf("-Y , --height <height>          Force movie bbox height to <height> (default: use master height (not with -t))\n");
 282     printf("-z , --zlib <zlib>             Enable Flash 6 (MX) Zlib Compression\n");
 283     printf("\n");
 284 }
 285
 286 static void makestackmaster(SWF*swf)
 287 {
 288     TAG*tag;
 289     int t;
 290     SRECT box;
 291     int fileversion = config.zlib?6:3;
 292     int frameRate = 256;
 293     RGBA rgb;
 294     rgb.r=rgb.b=rgb.g=0;
 295     memset(&box, 0, sizeof(box));
 296
 297     /* scan all slaves for bounding box */
 298     for(t=numslaves-1;t>=0;t--)
 299     {
 300         SWF head;
 301         int ret;
 302         int fi=open(slave_filename[t],O_RDONLY|O_BINARY);
 303         TAG*tag;
 304         if(fi<0 || swf_ReadSWF(fi, &head)<0) {
 305             msg("<fatal> Couldn't open/read %s.", slave_filename[t]);
 306             exit(1);
 307         }
 308         close(fi);
 309         swf_RemoveJPEGTables(&head);
 310         msg("<verbose> File %s has bounding box %d:%d:%d:%d\n",
 311                 slave_filename[t],
 312                 head.movieSize.xmin, head.movieSize.ymin,
 313                 head.movieSize.xmax, head.movieSize.ymax);
 314
 315         tag = head.firstTag;
 316         while(tag) {
 317             if(tag->id == ST_SETBACKGROUNDCOLOR && tag->len>=3) {
 318                 rgb.r = tag->data[0];
 319                 rgb.g = tag->data[1];
 320                 rgb.b = tag->data[2];
 321             }
 322             tag=tag->next;
 323         }
 324         frameRate = head.frameRate;
 325         if(head.fileVersion > fileversion)
 326             fileversion = head.fileVersion;
 327         if(!t)
 328             box = head.movieSize;
 329         else {
 330             if(head.movieSize.xmin < box.xmin)
 331                 box.xmin = head.movieSize.xmin;
 332             if(head.movieSize.ymin < box.ymin)
 333                 box.ymin = head.movieSize.ymin;
 334             if(head.movieSize.xmax > box.xmax)
 335                 box.xmax = head.movieSize.xmax;
 336             if(head.movieSize.ymax > box.ymax)
 337                 box.ymax = head.movieSize.ymax;
 338         }
 339         msg("<verbose> New master bounding box is %d:%d:%d:%d\n",
 340                 box.xmin, box.ymin,
 341                 box.xmax, box.ymax);
 342         swf_FreeTags(&head);
 343     }
 344
 345     memset(swf, 0, sizeof(SWF));
 346     swf->fileVersion = fileversion;
 347     swf->movieSize = box;
 348     swf->frameRate = frameRate;
 349
 350     swf->firstTag = swf_InsertTag(0, ST_SETBACKGROUNDCOLOR);
 351     tag = swf->firstTag;
 352     swf_SetRGB(tag, &rgb);
 353
 354     for(t=0;t<numslaves;t++)
 355     {
 356         char buf[128];
 357         sprintf(buf, "Frame%02d", t);
 358         slave_name[t] = strdup(buf);
 359
 360         tag = swf_InsertTag(tag, ST_DEFINESPRITE);
 361         swf_SetU16(tag, t+1);
 362         swf_SetU16(tag, 0);
 363         tag = swf_InsertTag(tag, ST_END);
 364         tag = swf_InsertTag(tag, ST_PLACEOBJECT2);
 365         swf_ObjectPlace(tag, t+1, 1+t,0,0, slave_name[t]);
 366
 367         if(!config.stack1 || t == numslaves-1) {
 368             tag = swf_InsertTag(tag, ST_SHOWFRAME);
 369         }
 370         if(!config.stack)
 371         if(t!=numslaves-1)
 372         {
 373             tag = swf_InsertTag(tag, ST_REMOVEOBJECT2);
 374             swf_SetU16(tag, 1+t);
 375         }
 376     }
 377     tag = swf_InsertTag(tag, ST_END);
 378     msg("<verbose> temporary SWF created");
 379 }
 380
 381 static char* slavename = 0;
 382 static int slaveid = -1;
 383 static int slaveframe = -1;
 384 static char masterbitmap[65536];
 385 static char depthbitmap[65536];
 386
 387 #define FLAGS_WRITEDEFINES 1
 388 #define FLAGS_WRITENONDEFINES 2
 389 #define FLAGS_WRITESPRITE 4
 390 #define FLAGS_WRITESLAVE 8
 391
 392 int get_free_id(char*bitmap)
 393 {
 394     int t;
 395     for(t=1;t<65536;t++)
 396         if(!bitmap[t]) {
 397             bitmap[t] = 1;
 398             return t;
 399         }
 400     return -1;
 401 }
 402
 403 void jpeg_assert(SWF*master, SWF*slave)
 404 {
 405     /* TODO: if there's a jpegtable found, store it
 406        and handle it together with the flash file
 407        headers */
 408
 409     /* check that master and slave don't have both
 410        jpegtables (which would be fatal) */
 411     int pos;
 412     TAG *mpos=0, *spos=0;
 413     TAG *mtag,*stag;
 414     pos = 0;
 415     mtag = master->firstTag;
 416     stag = slave->firstTag;
 417     while(mtag)
 418     {
 419         if(mtag->id  == ST_JPEGTABLES)
 420             mpos = mtag;
 421         mtag = mtag->next;
 422     }
 423     while(stag)
 424     {
 425         if(stag->id == ST_JPEGTABLES)
 426             spos = stag;
 427         stag = stag->next;
 428     }
 429     if(mpos && spos)
 430     {
 431         if(spos->len == mpos->len &&
 432         !memcmp(spos->data, mpos->data, mpos->len))
 433         {
 434             // ok, both have jpegtables, but they're identical.
 435             // delete one and don't throw an error
 436             swf_DeleteTag(spos);
 437             spos = 0;
 438         }
 439     }
 440     if(spos && mpos) {
 441         msg("<error> Master and slave have incompatible JPEGTABLES.");
 442     }
 443 }
 444
 445 TAG* write_sprite_defines(TAG*tag, SWF*sprite)
 446 {
 447     TAG*rtag = sprite->firstTag;
 448     while(rtag && rtag->id!=ST_END) {
 449         if(!swf_isAllowedSpriteTag(rtag)) {
 450             msg("<debug> processing sprite tag %02x", tag->id);
 451             if(swf_isDefiningTag(rtag))
 452             {
 453                 msg("<debug> [sprite defs] write tag %02x (%d bytes in body)",
 454                         tag->id, tag->len);
 455                 tag = swf_InsertTag(tag, rtag->id);
 456                 swf_SetBlock(tag, rtag->data, rtag->len);
 457             }
 458             else if(swf_isPseudoDefiningTag(rtag))
 459             {
 460                 msg("<debug> [sprite defs] write tag %02x (%d bytes in body)",
 461                         tag->id, tag->len);
 462                 tag = swf_InsertTag(tag, rtag->id);
 463                 swf_SetBlock(tag, rtag->data, rtag->len);
 464             }
 465             else {
 466                 switch(rtag->id)
 467                 {
 468                     case ST_JPEGTABLES:
 469                            /* if we get here, jpeg_assert has already run,
 470                               ensuring this is the only one of it's kind,
 471                               so we may safely write it out */
 472                            tag = swf_InsertTag(tag, rtag->id);
 473                            swf_SetBlock(tag, rtag->data, rtag->len);
 474                        break;
 475                     case ST_EXPORTASSETS:
 476                        msg("<debug> deliberately ignoring EXPORTASSETS tag");
 477                        break;
 478                     case ST_ENABLEDEBUGGER:
 479                        msg("<debug> deliberately ignoring ENABLEDEBUGGER tag");
 480                        break;
 481                     case ST_SETBACKGROUNDCOLOR:
 482                        msg("<debug> deliberately ignoring BACKGROUNDCOLOR tag");
 483                        break;
 484                     case ST_SHOWFRAME:
 485                        msg("<debug> deliberately ignoring SHOWFRAME tag");
 486                        break;
 487                     case ST_REFLEX:
 488                        msg("<debug> deliberately ignoring REFLEX tag");
 489                        break;
 490                     case 40:
 491                     case 49:
 492                     case 51:
 493                        msg("<notice> found tag %d. This is a Generator template, isn't it?", rtag->id);
 494                        break;
 495                     default:
 496                        msg("<notice> funny tag: %d is neither defining nor sprite", rtag->id);
 497                 }
 498             }
 499         }
 500         rtag = rtag->next;
 501     }
 502     return tag;
 503 }
 504
 505 void changedepth(TAG*tag, int add)
 506 {
 507     if(tag->id == ST_PLACEOBJECT)
 508         PUT16(&tag->data[2],GET16(&tag->data[2])+add);
 509     if(tag->id == ST_PLACEOBJECT2)
 510         PUT16(&tag->data[1],GET16(&tag->data[1])+add);
 511     if(tag->id == ST_REMOVEOBJECT)
 512         PUT16(&tag->data[2],GET16(&tag->data[2])+add);
 513     if(tag->id == ST_REMOVEOBJECT2)
 514         PUT16(&tag->data[0],GET16(&tag->data[0])+add);
 515     if(tag->id == ST_PLACEOBJECT2) {
 516         SWFPLACEOBJECT obj;
 517         U8 flags;
 518         swf_SetTagPos(tag, 0);
 519         flags = swf_GetU8(tag);
 520         if(flags&2) swf_GetU16(tag); //id
 521         if(flags&4) swf_GetMatrix(tag, 0);
 522         if(flags&8) swf_GetCXForm(tag, 0,1);
 523         if(flags&16) swf_GetU16(tag); //ratio
 524         if(flags&64) {
 525             swf_ResetReadBits(tag);
 526             printf("%d->%d\n", GET16(&tag->data[tag->pos]),
 527                                GET16(&tag->data[tag->pos])+add);
 528             PUT16(&tag->data[tag->pos],GET16(&tag->data[tag->pos])+add);
 529         }
 530         msg("<warning> Depth relocation not fully working yet with clipdepths", tag->id);
 531     }
 532 }
 533
 534 void matrix_adjust(MATRIX*m, int movex, int movey, float scalex, float scaley, int scalepos)
 535 {
 536     m->sx = (int)(m->sx*scalex);
 537     m->sy = (int)(m->sy*scaley);
 538     m->r1 = (int)(m->r1*scalex);
 539     m->r0 = (int)(m->r0*scaley);
 540     if(scalepos) {
 541         m->tx *= scalex;
 542         m->ty *= scaley;
 543     }
 544     m->tx += movex;
 545     m->ty += movey;
 546 }
 547
 548 void write_changepos(TAG*output, TAG*tag, int movex, int movey, float scalex, float scaley, int scalepos)
 549 {
 550     if(movex || movey || scalex != 1.0 || scaley != 1.0)
 551     {
 552         switch(tag->id)
 553         {
 554             case ST_PLACEOBJECT2: {
 555                 MATRIX m;
 556                 U8 flags;
 557                 swf_GetMatrix(0, &m);
 558                 tag->pos = 0;
 559                 tag->readBit = 0;
 560
 561                 flags = swf_GetU8(tag);
 562                 swf_SetU8(output, flags|4);
 563                 swf_SetU16(output, swf_GetU16(tag)); //depth
 564                 //flags&1: move
 565                 if(flags&2) {
 566                     swf_SetU16(output, swf_GetU16(tag)); //id
 567                 }
 568                 // flags & 4
 569                 if(flags&4) {
 570                     swf_GetMatrix(tag, &m);
 571                 } else {
 572                     swf_GetMatrix(0, &m);
 573                 }
 574                 matrix_adjust(&m, movex, movey, scalex, scaley, scalepos);
 575                 swf_SetMatrix(output, &m);
 576
 577                 if (tag->readBit)  { tag->pos++; tag->readBit = 0; } //swf_ResetReadBits(tag);
 578
 579                 swf_SetBlock(output, &tag->data[tag->pos], tag->len - tag->pos);
 580                 break;
 581             }
 582             case ST_PLACEOBJECT: {
 583                 MATRIX m;
 584                 swf_SetU16(output, swf_GetU16(tag)); //id
 585                 swf_SetU16(output, swf_GetU16(tag)); //depth
 586
 587                 swf_GetMatrix(tag, &m);
 588                 matrix_adjust(&m, movex, movey, scalex, scaley, scalepos);
 589                 swf_SetMatrix(output, &m);
 590
 591                 if (tag->readBit)  { tag->pos++; tag->readBit = 0; } //swf_ResetReadBits(tag);
 592
 593                 swf_SetBlock(output, &tag->data[tag->pos], tag->len - tag->pos);
 594                 break;
 595             }
 596             default:
 597             swf_SetBlock(output, tag->data, tag->len);
 598         }
 599     }
 600     else
 601     {
 602             swf_SetBlock(output, tag->data, tag->len);
 603     }
 604 }
 605
 606 TAG* write_sprite(TAG*tag, SWF*sprite, int spriteid, int replaceddefine)
 607 {
 608     TAG* definespritetag;
 609     TAG* rtag;
 610     int tmp;
 611
 612     definespritetag = tag = swf_InsertTag(tag, ST_DEFINESPRITE);
 613     swf_SetU16(tag, spriteid);
 614     swf_SetU16(tag, sprite->frameCount);
 615     msg ("<notice> sprite id is %d", spriteid);
 616
 617     tmp = sprite->frameCount;
 618     msg("<debug> %d frames to go",tmp);
 619
 620     if(config.clip) {
 621         tag = swf_InsertTag(tag, ST_PLACEOBJECT2);
 622         swf_SetU8(tag, 2+64); //flags: character+clipdepth
 623         swf_SetU16(tag, 0); //depth
 624         swf_SetU16(tag, replaceddefine); //id
 625         swf_SetU16(tag, 65535); //clipdepth
 626     }
 627
 628     if(config.overlay && !config.isframe) {
 629         tag = swf_InsertTag(tag, ST_PLACEOBJECT2);
 630         swf_SetU8(tag, 2); //flags: character
 631         swf_SetU16(tag, 1); //depth
 632         swf_SetU16(tag, replaceddefine); //id
 633     }
 634
 635     rtag = sprite->firstTag;
 636     while(rtag && rtag->id!=ST_END)
 637     {
 638         if (swf_isAllowedSpriteTag(rtag)) {
 639
 640             msg("<debug> [sprite main] write tag %02x (%d bytes in body)",
 641                     rtag->id, rtag->len);
 642             tag = swf_InsertTag(tag, rtag->id);
 643             write_changepos(tag, rtag, config.movex, config.movey, config.scalex, config.scaley, 0);
 644
 645             if(config.clip || (config.overlay && !config.isframe))
 646                 changedepth(tag, +2);
 647
 648             if(tag->id == ST_SHOWFRAME)
 649             {
 650                 tmp--;
 651                 msg("<debug> %d frames to go",tmp);
 652             }
 653         }
 654         rtag = rtag->next;
 655     }
 656     tag = swf_InsertTag(tag, ST_END);
 657     return tag;
 658 }
 659
 660 static char tag_ok_for_slave(int id)
 661 {
 662     if(id == ST_SETBACKGROUNDCOLOR)
 663         return 0;
 664     return 1;
 665 }
 666
 667 TAG* write_master(TAG*tag, SWF*master, SWF*slave, int spriteid, int replaceddefine, int flags)
 668 {
 669     int outputslave = 0;
 670     int frame = 1;
 671     int sframe = 0;
 672     int slavewritten = 0;
 673     int deletedepth = -1;
 674
 675     TAG* rtag = master->firstTag;
 676     TAG* stag = slave->firstTag;
 677
 678     while(rtag && rtag->id!=ST_END)
 679     {
 680         if(rtag->id == ST_SHOWFRAME && outputslave)
 681         {
 682             while(stag && stag->id!=ST_END) {
 683                 if(stag->id == ST_SHOWFRAME) {
 684                     stag = stag->next;
 685                     sframe++;
 686                     break;
 687                 }
 688                 if(tag_ok_for_slave(stag->id)) {
 689                     tag = swf_InsertTag(tag, stag->id);
 690                     write_changepos(tag, stag, config.movex, config.movey, config.scalex, config.scaley, 0);
 691                 }
 692                 stag = stag->next;
 693             }
 694         }
 695         if(rtag->id == ST_SHOWFRAME)
 696         {
 697             frame ++;
 698             tag = swf_InsertTag(tag, ST_SHOWFRAME);
 699             if(deletedepth>=0) {
 700                 tag = swf_InsertTag(tag, ST_REMOVEOBJECT2);
 701                 swf_SetU16(tag, deletedepth);
 702                 deletedepth=-1;
 703             }
 704             rtag = rtag->next;
 705             continue;
 706         }
 707
 708         if(swf_isDefiningTag(rtag) && (flags&FLAGS_WRITEDEFINES))
 709         {
 710             msg("<debug> [master] write tag %02x (%d bytes in body)",
 711                     rtag->id, rtag->len);
 712             if(swf_GetDefineID(rtag) == spriteid && !config.isframe)
 713             {
 714                 if(config.overlay)
 715                 {
 716                     tag = swf_InsertTag(tag, rtag->id);
 717                     swf_SetBlock(tag, rtag->data, rtag->len);
 718                     swf_SetDefineID(tag, replaceddefine);
 719                 } else {
 720                     /* don't write this tag */
 721                     msg("<verbose> replacing tag %d ID %d with sprite", rtag->id ,spriteid);
 722                 }
 723
 724                 if(flags&FLAGS_WRITESPRITE)
 725                 {
 726                     msg("<debug> writing sprite defines");
 727                     tag = write_sprite_defines(tag, slave);
 728                     msg("<debug> writing sprite");
 729                     tag = write_sprite(tag, slave, spriteid, replaceddefine);
 730                 }
 731                 if(flags&FLAGS_WRITESLAVE)
 732                 {
 733                     msg("<debug> writing slave");
 734                     outputslave = 1;
 735                 }
 736             } else {
 737                 tag = swf_InsertTag(tag, rtag->id);
 738                 swf_SetBlock(tag, rtag->data, rtag->len);
 739             }
 740         }
 741         if(frame == slaveframe) /* only happens with config.isframe: put slave at specific frame */
 742         {
 743             if(flags&FLAGS_WRITESLAVE) {
 744                 outputslave = 1;
 745                 slavewritten = 1;
 746             }
 747             if((flags&FLAGS_WRITESPRITE) && !slavewritten)
 748             {
 749                 int id = get_free_id(masterbitmap);
 750                 int depth = 65535;
 751                 deletedepth = 65535;
 752                 if(config.clip) {
 753                     msg("<fatal> Can't combine --clip and --frame");
 754                 }
 755
 756                 tag = write_sprite_defines(tag, slave);
 757                 tag = write_sprite(tag, slave, id, -1);
 758
 759                 tag = swf_InsertTag(tag, ST_PLACEOBJECT2);
 760                     swf_SetU8(tag, 2); //flags: id
 761                     swf_SetU16(tag, depth);
 762                     swf_SetU16(tag, id);
 763
 764                 slavewritten = 1;
 765             }
 766         }
 767         if(!swf_isDefiningTag(rtag) && (flags&FLAGS_WRITENONDEFINES))
 768         {
 769             int dontwrite = 0;
 770             switch(rtag->id) {
 771                 case ST_PLACEOBJECT:
 772                 case ST_PLACEOBJECT2:
 773                     if(frame == slaveframe && !config.overlay)
 774                         dontwrite = 1;
 775                 case ST_REMOVEOBJECT:
 776                     /* place/removetags for the object we replaced
 777                        should be discarded, too, as the object to insert
 778                        isn't a sprite
 779                      */
 780                     if(spriteid>=0 && swf_GetPlaceID(rtag) == spriteid &&
 781                             !config.isframe && config.merge)
 782                         dontwrite = 1;
 783                 break;
 784                 case ST_REMOVEOBJECT2:
 785                 break;
 786             }
 787             if(!dontwrite) {
 788                 msg("<debug> [master] write tag %02x (%d bytes in body)",
 789                         rtag->id, rtag->len);
 790                 tag = swf_InsertTag(tag, rtag->id);
 791                 write_changepos(tag, rtag, config.mastermovex, config.mastermovey, config.masterscalex, config.masterscaley, 1);
 792
 793             }
 794         }
 795         rtag = rtag->next;
 796     }
 797
 798     if(outputslave)
 799     while(stag && stag->id!=ST_END)
 800     {
 801             if(tag_ok_for_slave(stag->id)) {
 802                 msg("<debug> [slave] write tag %02x (%d bytes in body), %.2f %.2f", rtag->id, rtag->len, config.movex /20.0, config.movey /20.0);
 803                 tag = swf_InsertTag(tag, stag->id);
 804                 write_changepos(tag, stag, config.movex, config.movey, config.scalex, config.scaley, 0);
 805             }
 806             stag = stag->next;
 807     }
 808     if(!slavewritten && config.isframe && (flags&(FLAGS_WRITESLAVE|FLAGS_WRITESPRITE)))
 809     {
 810         if(slaveframe>=0)
 811             msg("<warning> Frame %d doesn't exist in file. No substitution will occur",
 812                     slaveframe);
 813         else
 814             msg("<warning> Frame \"%s\" doesn't exist in file. No substitution will occur",
 815                     slavename);
 816     }
 817     tag = swf_InsertTag(tag, ST_END);
 818     return tag;
 819 }
 820
 821 void adjustheader(SWF*swf)
 822 {
 823     if(config.framerate)
 824         swf->frameRate = config.framerate;
 825     if(config.hassizex) {
 826         swf->movieSize.xmax =
 827         swf->movieSize.xmin + config.sizex;
 828     }
 829     if(config.hassizey) {
 830         swf->movieSize.ymax =
 831         swf->movieSize.ymin + config.sizey;
 832     }
 833 }
 834
 835 void catcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
 836 {
 837     char* depths;
 838     int t;
 839     TAG*tag;
 840     TAG*mtag,*stag;
 841     if(config.isframe) {
 842         msg("<fatal> Can't combine --cat and --frame");
 843         exit(1);
 844     }
 845
 846     tag = master->firstTag;
 847     while(tag)
 848     {
 849         if(swf_isDefiningTag(tag)) {
 850             int defineid = swf_GetDefineID(tag);
 851             msg("<debug> tagid %02x defines object %d", tag->id, defineid);
 852             masterbitmap[defineid] = 1;
 853         }
 854         tag = tag->next;
 855     }
 856
 857     swf_Relocate(slave, masterbitmap);
 858     jpeg_assert(master, slave);
 859
 860     memcpy(newswf, master, sizeof(SWF));
 861     adjustheader(newswf);
 862
 863     tag = newswf->firstTag = swf_InsertTag(0, ST_REFLEX); // to be removed later
 864
 865     depths = malloc(65536);
 866     if(!depths) {
 867         msg("<fatal> Couldn't allocate %d bytes of memory", 65536);
 868         return;
 869     }
 870     memset(depths, 0, 65536);
 871     mtag = master->firstTag;
 872     while(mtag && mtag->id!=ST_END)
 873     {
 874         int num=1;
 875         U16 depth;
 876         msg("<debug> [master] write tag %02x (%d bytes in body)",
 877                 mtag->id, mtag->len);
 878         switch(mtag->id) {
 879             case ST_PLACEOBJECT2:
 880                 num++;
 881             case ST_PLACEOBJECT: {
 882                depth = swf_GetDepth(mtag);
 883                depths[depth] = 1;
 884             }
 885             break;
 886             case ST_REMOVEOBJECT: {
 887                depth = swf_GetDepth(mtag);
 888                depths[depth] = 0;
 889             }
 890             break;
 891             case ST_REMOVEOBJECT2: {
 892                depth = swf_GetDepth(mtag);
 893                depths[depth] = 0;
 894             }
 895             break;
 896         }
 897         tag = swf_InsertTag(tag, mtag->id);
 898         swf_SetBlock(tag, mtag->data, mtag->len);
 899
 900         mtag = mtag->next;
 901     }
 902
 903     for(t=0;t<65536;t++)
 904     if(depths[t])
 905     {
 906         char data[16];
 907         int len;
 908         tag = swf_InsertTag(tag, ST_REMOVEOBJECT2);
 909         swf_SetU16(tag, t);
 910     }
 911     free(depths);
 912
 913     stag = slave->firstTag;
 914     while(stag && stag->id!=ST_END)
 915     {
 916         msg("<debug> [slave] write tag %02x (%d bytes in body)",
 917                 stag->id, stag->len);
 918         tag = swf_InsertTag(tag, stag->id);
 919         swf_SetBlock(tag, stag->data, stag->len);
 920         stag = stag->next;
 921     }
 922     tag = swf_InsertTag(tag, ST_END);
 923
 924     tag = newswf->firstTag;
 925     newswf->firstTag = newswf->firstTag->next; //remove temporary tag
 926     swf_DeleteTag(tag);
 927 }
 928
 929 void normalcombine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
 930 {
 931     int spriteid = -1;
 932     int replaceddefine = -1;
 933     int frame = 0;
 934     char*framelabel;
 935     TAG * tag = master->firstTag;
 936
 937     memset(depthbitmap, 0, sizeof(depthbitmap));
 938
 939     // set the idtab
 940     while(tag)
 941     {
 942         int depth = swf_GetDepth(tag);
 943         if(depth>=0) {
 944             depthbitmap[depth] = 1;
 945         }
 946         if(swf_isDefiningTag(tag)) {
 947             int defineid = swf_GetDefineID(tag);
 948             msg("<debug> tagid %02x defines object %d", tag->id, defineid);
 949             masterbitmap[defineid] = 1;
 950
 951             if (!slavename && defineid==slaveid) {
 952                 if(defineid>=0) {
 953                   spriteid = defineid;
 954                   msg("<notice> Slave file attached to object %d.", defineid);
 955                 }
 956             }
 957         } else if(tag->id == ST_PLACEOBJECT2) {
 958             char * name = swf_GetName(tag);
 959             int id = swf_GetPlaceID(tag);
 960
 961             {
 962                 SWFPLACEOBJECT obj;
 963                 swf_GetPlaceObject(tag, &obj);
 964                 swf_PlaceObjectFree(&obj);
 965                 if(obj.clipdepth) {
 966                     depthbitmap[obj.clipdepth] = 1;
 967                 }
 968             }
 969
 970             if(name)
 971               msg("<verbose> tagid %02x places object %d named \"%s\"", tag->id, id, name);
 972             else
 973               msg("<verbose> tagid %02x places object %d (no name)", tag->id, id);
 974
 975             if (name && slavename && !strcmp(name,slavename)) {
 976                 if(id>=0) {
 977                   spriteid = id;
 978                   msg("<notice> Slave file attached to named object %s (%d).", name, id);
 979                 }
 980             }
 981         } else if(tag->id == ST_SHOWFRAME) {
 982             if(slaveframe>=0 && frame==slaveframe) {
 983                 msg("<notice> Slave file attached to frame %d.", frame);
 984             }
 985             frame++;
 986         } else if(tag->id == ST_FRAMELABEL) {
 987             char * name = tag->data;
 988             if(name && slavename && config.isframe && !strcmp(name, slavename)) {
 989                 slaveframe = frame;
 990                 msg("<notice> Slave file attached to frame %d (%s).", frame, name);
 991             }
 992         }
 993         tag = tag->next;
 994     };
 995
 996     if (spriteid<0 && !config.isframe) {
 997         if(slavename) {
 998             if(strcmp(slavename,"!!dummy!!"))
 999                 msg("<warning> Didn't find anything named %s in file. No substitutions will occur.", slavename);
1000         }
1001         else
1002             msg("<warning> Didn't find id %d in file. No substitutions will occur.", slaveid);
1003         spriteid = get_free_id(masterbitmap);
1004     }
1005
1006     swf_Relocate (slave, masterbitmap);
1007     if(config.merge)
1008         swf_RelocateDepth (slave, depthbitmap);
1009     jpeg_assert(slave, master);
1010
1011     if (config.overlay)
1012         replaceddefine = get_free_id(masterbitmap);
1013
1014     // write file
1015
1016     memcpy(newswf, master, sizeof(SWF));
1017     adjustheader(newswf);
1018
1019     newswf->firstTag = tag = swf_InsertTag(0, ST_REFLEX); // to be removed later
1020
1021     if (config.antistream) {
1022         if (config.merge) {
1023             msg("<fatal> Can't combine --antistream and --merge");
1024         }
1025         tag = write_sprite_defines(tag, slave);
1026         tag = write_sprite(tag, slave, spriteid, replaceddefine);
1027         tag = write_master(tag, master, slave, spriteid, replaceddefine, FLAGS_WRITEDEFINES);
1028         tag = write_master(tag, master, slave, spriteid, replaceddefine, FLAGS_WRITENONDEFINES);
1029     } else {
1030         if (config.merge)
1031             tag = write_master(tag, master, slave, spriteid, replaceddefine,
1032                 FLAGS_WRITEDEFINES|FLAGS_WRITENONDEFINES|   FLAGS_WRITESLAVE    );
1033         else
1034             tag = write_master(tag, master, slave, spriteid, replaceddefine,
1035                 FLAGS_WRITEDEFINES|FLAGS_WRITENONDEFINES|   FLAGS_WRITESPRITE   );
1036     }
1037
1038     tag = newswf->firstTag;
1039     newswf->firstTag = newswf->firstTag->next; //remove temporary tag
1040     swf_DeleteTag(tag);
1041 }
1042
1043 void combine(SWF*master, char*slave_name, SWF*slave, SWF*newswf)
1044 {
1045     slavename = slave_name;
1046     slaveid = -1;
1047     slaveframe = -1;
1048
1049     if(!master->fileVersion && slave)
1050         master->fileVersion = slave->fileVersion;
1051
1052     swf_FoldAll(master);
1053     swf_FoldAll(slave);
1054
1055     if(slavename[0] == '#')
1056     {
1057         slaveid = atoi(&slavename[1]);
1058         slavename = 0;
1059     }
1060
1061     if(config.isframe)
1062     {
1063         if(slavename && slavename[0]!='#') {
1064             int tmp;
1065             int len;
1066             sscanf(slavename, "%d%n", &tmp, &len);
1067             if(len == strlen(slavename)) {
1068             /* if the name the slave should replace
1069                consists only of digits and the -f
1070                option is given, it probably is not
1071                a frame name but a frame number.
1072              */
1073                 slaveid = tmp;
1074                 slavename = 0;
1075             }
1076         }
1077
1078         if(slaveid>=0) {
1079             slaveframe = slaveid;
1080             slaveid = -1;
1081         } else {
1082         /* if id wasn't given as either #number or number,
1083            the name is a frame label. BTW: The user wouldn't necessarily have
1084            needed to supply the -f option in this case */
1085         }
1086     }
1087
1088     msg("<debug> move x (%d)", config.movex);
1089     msg("<debug> move y (%d)", config.movey);
1090     msg("<debug> scale x (%f)", config.scalex);
1091     msg("<debug> scale y (%f)", config.scaley);
1092     msg("<debug> master move x (%d)", config.mastermovex);
1093     msg("<debug> master move y (%d)", config.mastermovey);
1094     msg("<debug> master scale x (%f)", config.masterscalex);
1095     msg("<debug> master scale y (%f)", config.masterscaley);
1096     msg("<debug> is frame (%d)", config.isframe);
1097
1098     memset(masterbitmap, 0, sizeof(masterbitmap));
1099
1100     if(config.cat)
1101         return catcombine(master, slave_name, slave, newswf);
1102     else
1103         return normalcombine(master, slave_name, slave, newswf);
1104 }
1105
1106 int main(int argn, char *argv[])
1107 {
1108     int fi;
1109     SWF master;
1110     SWF slave;
1111     SWF newswf;
1112     int t;
1113
1114     config.overlay = 0;
1115     config.antistream = 0;
1116     config.alloctest = 0;
1117     config.cat = 0;
1118     config.merge = 0;
1119     config.clip = 0;
1120     config.loglevel = 2;
1121     config.movex = 0;
1122     config.movey = 0;
1123     config.scalex = 1.0;
1124     config.scaley = 1.0;
1125     config.sizex = 0;
1126     config.sizey = 0;
1127     config.masterscalex = 1.0;
1128     config.masterscaley = 1.0;
1129     config.mastermovex = 0;
1130     config.mastermovey = 0;
1131     config.hassizex = 0;
1132     config.hassizey = 0;
1133     config.framerate = 0;
1134     config.stack = 0;
1135     config.stack1 = 0;
1136     config.dummy = 0;
1137     config.zlib = 0;
1138
1139     processargs(argn, argv);
1140     initLog(0,-1,0,0,-1,config.loglevel);
1141
1142     if(config.merge && config.cat) {
1143         msg("<error> Can't combine --cat and --merge");
1144         exit(1);
1145     }
1146
1147     if(config.stack && config.cat) {
1148         msg("<error> Can't combine --cat and --stack");
1149         exit(1);
1150     }
1151
1152     if(config.stack) {
1153         if(config.overlay) {
1154             msg("<error> Can't combine -l and -t");
1155             exit(1);
1156         }
1157         if(config.clip) {
1158             msg("<error> Can't combine -c and -t");
1159             exit(1);
1160         }
1161         msg("<verbose> (stacking) %d files found\n", numslaves);
1162
1163         makestackmaster(&master);
1164     }
1165     else {
1166         int ret;
1167         msg("<verbose> master entity %s (named \"%s\")\n", master_filename, master_name);
1168         fi = open(master_filename, O_RDONLY|O_BINARY);
1169         if(fi<0) {
1170             msg("<fatal> Failed to open %s\n", master_filename);
1171             exit(1);
1172         }
1173         ret = swf_ReadSWF(fi, &master);
1174         if(ret<0) {
1175             msg("<fatal> Failed to read from %s\n", master_filename);
1176             exit(1);
1177         }
1178         swf_RemoveJPEGTables(&master);
1179         msg("<debug> Read %d bytes from masterfile\n", ret);
1180         close(fi);
1181     }
1182
1183     for(t=0;t<numslaves;t++) {
1184             msg("<verbose> slave entity(%d) %s (%s \"%s\")\n", t+1, slave_filename[t],
1185                     slave_isframe[t]?"frame":"object", slave_name[t]);
1186     }
1187
1188     if(config.dummy)
1189     {
1190         if(numslaves)
1191         {
1192             msg("<error> --dummy (-d) implies there are zero slave objects. You supplied %d.", numslaves);
1193             exit(1);
1194         }
1195         numslaves = 1;
1196         slave_filename[0] = "!!dummy!!";
1197         slave_name[0] = "!!dummy!!";
1198         slave_isframe[0] = 0;
1199     }
1200
1201     if (config.alloctest)
1202     {
1203         char*bitmap = malloc(sizeof(char)*65536);
1204         memset(bitmap, 0, 65536*sizeof(char));
1205         memset(bitmap, 1, 101*sizeof(char));
1206         swf_Relocate(&master, bitmap);
1207         newswf = master;
1208         free(bitmap);
1209 //      makestackmaster(&newswf);
1210     }
1211     else
1212     {
1213         if (!numslaves)
1214         {
1215             if(config.cat)
1216                 msg("<error> You must have at least two objects.");
1217             else
1218                 msg("<error> You must have at least one slave entity.");
1219             return 0;
1220         }
1221         for(t = 0; t < numslaves; t++)
1222         {
1223             config.movex = slave_movex[t];
1224             config.movey = slave_movey[t];
1225             config.scalex = slave_scalex[t];
1226             config.scaley = slave_scaley[t];
1227             config.isframe = slave_isframe[t];
1228
1229             msg("<notice> Combine [%s]%s and [%s]%s", master_name, master_filename,
1230                     slave_name[t], slave_filename[t]);
1231             if(!config.dummy)
1232             {
1233                 int ret;
1234                 fi = open(slave_filename[t], O_RDONLY|O_BINARY);
1235                 if(!fi) {
1236                     msg("<fatal> Failed to open %s\n", slave_filename[t]);
1237                     exit(1);
1238                 }
1239                 ret = swf_ReadSWF(fi, &slave);
1240                 if(ret<0) {
1241                     msg("<fatal> Failed to read from %s\n", slave_filename[t]);
1242                     exit(1);
1243                 }
1244                 msg("<debug> Read %d bytes from slavefile\n", ret);
1245                 close(fi);
1246                 swf_RemoveJPEGTables(&slave);
1247             }
1248             else
1249             {
1250                 memset(&slave, 0, sizeof(slave));
1251                 slave.firstTag = swf_InsertTag(0, ST_END);
1252                 slave.frameRate = 0;
1253                 slave.fileVersion = 0;
1254                 slave.frameCount = 0;
1255             }
1256
1257             combine(&master, slave_name[t], &slave, &newswf);
1258             master = newswf;
1259         }
1260         if(config.dummy && !config.hassizex && !config.hassizey && !config.mastermovex && !config.mastermovey) {
1261             newswf.movieSize.xmin = newswf.movieSize.xmin*config.masterscalex;
1262             newswf.movieSize.ymin = newswf.movieSize.ymin*config.masterscaley;
1263             newswf.movieSize.xmax = newswf.movieSize.xmax*config.masterscalex;
1264             newswf.movieSize.ymax = newswf.movieSize.ymax*config.masterscaley;
1265         }
1266     }
1267
1268     if(!newswf.fileVersion)
1269         newswf.fileVersion = 4;
1270
1271     fi = open(outputname, O_BINARY|O_RDWR|O_TRUNC|O_CREAT, 0777);
1272
1273     if(config.zlib) {
1274         if(newswf.fileVersion < 6)
1275             newswf.fileVersion = 6;
1276         swf_WriteSWC(fi, &newswf);
1277     } else {
1278         newswf.compressed = 0;
1279         swf_WriteSWF(fi, &newswf);
1280     }
1281     close(fi);
1282     return 0;
1283 }
1284