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