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