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