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