fixed a bug
[swftools.git] / src / combine.c
1 /* combine.c 
2    Implements combine(), which merges two swfs in memory.
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 <memory.h>
14 #include "../lib/log.h"
15 #include "./flash.h"
16 #include "./reloc.h"
17 #include "./settings.h"
18
19 // TODO:
20 // * readers should be object-oriented
21
22 static char* tag_placeobject2_name (struct swf_tag* tag)
23 {
24     struct PlaceObject2 plo2;
25     placeobject2_init (&plo2, tag);
26     return plo2.name;
27 }
28
29 static u16 tag_placeobject2_character (struct swf_tag* tag)
30 {
31     struct PlaceObject2 plo2;
32     placeobject2_init (&plo2, tag);
33     return plo2.id;
34 }
35
36 static struct swffile master;
37 static struct swffile slave;
38
39 static int masterids[65536];
40
41 static int get_free_id()
42 {
43     int t;
44     for (t=1;t<65536;t++)
45     {
46         if(masterids[t] == -1)
47         {
48             masterids[t] = 1;
49             return t;
50         }
51     }
52     return -1;
53 }
54
55 void changedepth(struct swf_tag*tag, int add)
56 {
57     if(tag->id == TAGID_PLACEOBJECT)
58         (*(u16*)&tag->data[2]) += add;
59     if(tag->id == TAGID_PLACEOBJECT2)
60         (*(u16*)&tag->data[1]) += add;
61     if(tag->id == TAGID_REMOVEOBJECT)
62         (*(u16*)&tag->data[2]) += add;
63     if(tag->id == TAGID_REMOVEOBJECT2)
64         (*(u16*)&tag->data[0]) += add;
65 }
66
67 /* applies the config move and scale parameters to
68  * a matrix. (If settings would provide a rotation,
69  * this would be a matrix concatenation/multiplication
70  * routine. In this case, it's just element-wise multiplication.
71  */
72 void matrix_adjust(struct MATRIX*m)
73 {
74     if(config.scalex != 1 || config.scaley != 1)
75     {
76         if(!m->hasscale) {
77             m->hasscale = 1;
78             m->a[0][0] = config.scalex;
79             m->a[1][1] = config.scaley;
80         } else {
81             m->a[0][0] *= config.scalex;
82             m->a[1][1] *= config.scaley;
83         }
84         if(m->hasrotate) {
85             m->a[0][1] *= config.scalex;
86             m->a[1][0] *= config.scaley;
87         }
88         m->b[0] *= config.scalex;
89         m->b[1] *= config.scaley;
90     }
91 /*    printf("hasscale: %d\n",m->hasscale);
92     printf("hasrotate: %d\n", m->hasrotate);
93     printf("move: %d %d\n", m->b[0],m->b[1]);
94     printf("rot: %f %f\n",m->a[0][0],m->a[0][1]);
95     printf("     %f %f\n",m->a[1][0],m->a[1][1]);*/
96     m->b[0] += config.movex;
97     m->b[1] += config.movey;
98 }
99
100 void write_changepos(struct swf_tag*tag, struct writer_t*w)
101 {
102     if(config.movex || config.movey || config.scalex != 1 || config.scaley != 1)
103     {
104         switch(tag->id)
105         {
106             case TAGID_PLACEOBJECT: {
107                 struct PlaceObject p;
108                 placeobject_init(&p, tag);
109                 matrix_adjust(&p.matrix);
110                 placeobject_write(&p, w);
111                 break;
112             }
113             case TAGID_PLACEOBJECT2: {
114                 struct PlaceObject2 p;
115                 placeobject2_init(&p, tag);
116                 if(!p.hasmatrix) {
117                     p.hasmatrix = 1;
118                     MATRIX_init(&p.matrix);
119                 }
120                 matrix_adjust(&p.matrix);
121                 placeobject2_write(&p, w);
122                 break;
123             }
124             default:
125             writer_write(w, tag->fulldata, tag->fulllength);
126         }
127     } 
128     else 
129     {
130             writer_write(w, tag->fulldata, tag->fulllength);
131     }
132 }
133
134 uchar * combine(uchar*masterdata, int masterlength, char*slavename, uchar*slavedata, int slavelength, int*newlength)
135 {
136     char master_flash = 0;
137     char slave_flash = 0;
138
139     logf("<debug> move x (%d)", config.movex);
140     logf("<debug> move y (%d)", config.movey);
141     logf("<debug> scale x (%d)", config.scalex);
142     logf("<debug> scale y (%d)", config.scaley);
143     
144     memset(masterids, -1, sizeof(masterids));
145
146     if(masterlength < 3)
147     {
148         logf("<fatal> the master file is too small (%d bytes)", masterlength);
149         return 0;
150     }
151     if(slavelength < 3)
152     {
153         logf("<fatal> the slave file is too small (%d bytes)", slavelength);
154         return 0;
155     }
156     if(masterdata[2] == 'S' &&
157        masterdata[1] == 'W' &&
158        masterdata[0] == 'F')
159     {
160         logf("<notice> the master file is flash (swf) format\n");
161         master_flash = 1;
162     }
163     if(slavedata[2] == 'S' &&
164        slavedata[1] == 'W' &&
165        slavedata[0] == 'F')
166     {
167         logf("<notice> the slave file is flash (swf) format\n");
168         slave_flash = 1;
169     }
170
171     if(master_flash && slave_flash)
172     {
173         int length;
174         int pos=0;
175         u16 tmp;
176         u32 tmp32;
177         u32*tagidpos;
178         u8*startpos;
179         u32*headlength;
180         uchar*newdata;
181         int spriteid = -1;
182         int replaceddefine = -1;
183         struct writer_t w;
184         
185         read_swf(&master, masterdata, masterlength);
186
187         length = masterlength + slavelength*2 + 128; // this is a guess, but a good guess.
188         newdata = malloc(length);
189         writer_init(&w, newdata, length);
190
191         if(!newdata) {
192             logf("<fatal> Couldn't allocate %d bytes of memory", length);
193             return 0;
194         }
195
196         // set the idtab
197         pos = 0;
198         do {
199             int tag = master.tags[pos].id;
200             if(is_defining_tag(tag)) {
201                 int defineid = getidfromtag(&master.tags[pos]);
202                 logf("<debug> tagid %02x defines object %d", tag, defineid);
203                 masterids[defineid] = 1;
204             } else if(tag == TAGID_PLACEOBJECT2) {
205                 char * name = tag_placeobject2_name(&master.tags[pos]);
206                 int id = tag_placeobject2_character(&master.tags[pos]);
207
208                 if(name)
209                   logf("<verbose> tagid %02x places object %d named \"%s\"", tag, id, name);
210                 else
211                   logf("<verbose> tagid %02x places object %d (no name)", tag, id);
212
213                 if (name && !strcmp(name,slavename)) {
214                     if(id>=0) {
215                       spriteid = id;
216                       logf("<notice> Slave file attached to object %d.", id);
217                     }
218                 }
219             }
220         }
221         while(master.tags[pos++].id != 0);
222
223         swf_relocate (slavedata, slavelength, masterids);
224
225         read_swf(&slave, slavedata, slavelength);
226
227         // write header
228
229         writer_write(&w, "FWS",3);
230         headlength = (u32*)(writer_getpos(&w) + 1);
231         writer_write(&w, master.header.headerdata, master.header.headerlength);
232
233         // write sprite(1)
234         
235         pos = 0;
236         while(slave.tags[pos].id != 0) {
237             struct swf_tag * tag = &slave.tags[pos];
238             if(!is_sprite_tag(tag->id)) {
239                 logf("<debug> processing sprite tag %02x", slave.tags[pos].id);
240                 if(is_defining_tag(tag->id))
241                 {
242                     logf("<debug> [sprite defs] write tag %02x (%d bytes in body)", 
243                             tag->id, tag->length);
244                     writer_write(&w, tag->fulldata, tag->fulllength);
245                 }
246                 else
247                 {
248                     switch(tag->id)
249                     {case TAGID_DEFINEFONTINFO:
250                         {
251                             /* define font info is not a defining tag, in that
252                              * it doesn't define a new id, but rather extends
253                              * an existing one. It also isn't a sprite tag. 
254                              * Anyway we can't throw it out, so we just pass it
255                              * through.
256                              */
257                             break;
258                         }
259                      case TAGID_EXPORTASSETS:
260                         logf("<debug> deliberately ignoring EXPORTASSETS tag");
261                         break;
262                      case TAGID_ENABLEDEBUGGER:
263                         logf("<debug> deliberately ignoring ENABLEDEBUGGER tag");
264                         break;
265                      case TAGID_BACKGROUNDCOLOR:
266                         logf("<debug> deliberately ignoring BACKGROUNDCOLOR tag");
267                         break;
268                      case 40:
269                      case 49:
270                      case 51:
271                         logf("<notice> found tag %d. This is a Generator template, isn't it?", slave.tags[pos].id);
272                         break;
273                      default:
274                         logf("<notice> funny tag: %d is neither defining nor sprite", slave.tags[pos].id);
275                     }
276                 }
277             }
278             pos++;
279         }
280         
281         // write master (1)
282         pos = 0;
283         do {
284             if(is_defining_tag(master.tags[pos].id))
285             {
286                 logf("<debug> [master] write tag %02x (%d bytes in body)", 
287                         master.tags[pos].id, master.tags[pos].length);
288                 if( getidfromtag(&master.tags[pos]) == spriteid) 
289                 {
290                     if(config.overlay)
291                     {
292                         *(u16*)master.tags[pos].data = replaceddefine = get_free_id();
293                         writer_write(&w, master.tags[pos].fulldata, master.tags[pos].fulllength);
294                     } else {
295                         /* don't write this tag */
296                         logf("<verbose> replacing tag %d id %d with sprite", master.tags[pos].id
297                                 ,spriteid);
298                     }
299                 } else { 
300                     writer_write(&w, master.tags[pos].fulldata, master.tags[pos].fulllength);
301                 }
302             }
303         }
304         while(master.tags[pos++].id != 0);
305
306         // write slave(2) (header)
307         tmp = 0x3f + (TAGID_DEFINESPRITE << 6);
308         writer_write(&w, &tmp, 2);
309         tagidpos = (u32*)writer_getpos(&w);
310         writer_write(&w, &tmp32, 4);
311         
312         startpos = (u8*)writer_getpos(&w);
313
314         if (spriteid<0)
315         {
316             logf("<warning> Didn't find anything named %s in file. No substitutions will occur.", slavename);
317             spriteid = get_free_id();
318         }
319
320         logf ("<notice> sprite id is %d", spriteid);
321         tmp = spriteid;
322         writer_write(&w, &tmp, 2);
323         tmp = slave.header.count;
324         writer_write(&w, &tmp, 2);
325
326
327         // write slave(2) (body)
328         pos = 0;
329         tmp = slave.header.count;
330         logf("<debug> %d frames to go",tmp);
331
332         if(config.clip) {
333             tmp = 7 + (TAGID_PLACEOBJECT2 << 6);
334             writer_write(&w, &tmp, 2);
335             tmp = 2+64; //flags: character + clipaction
336             writer_write(&w, &tmp, 1);
337             tmp = 0; //depth
338             writer_write(&w, &tmp,2);
339             tmp = replaceddefine; //id
340             writer_write(&w, &tmp,2);
341             tmp = 65535; //clipdepth
342             writer_write(&w, &tmp,2);
343         }
344
345         if(config.overlay) {
346             tmp = 5 + (TAGID_PLACEOBJECT2 << 6);
347             writer_write(&w, &tmp, 2);
348             tmp = 2; //flags: character
349             writer_write(&w, &tmp, 1);
350             tmp = 0; //depth
351             writer_write(&w, &tmp,2);
352             tmp = replaceddefine; //id
353             writer_write(&w, &tmp,2);
354         }
355
356         do {
357             struct swf_tag * tag = &slave.tags[pos];
358             if (is_sprite_tag(tag->id)) {
359
360                 changedepth(tag, +1);
361                 logf("<debug> [sprite main] write tag %02x (%d bytes in body)", 
362                         slave.tags[pos].id, slave.tags[pos].length);
363                 write_changepos(tag, &w);
364
365                 if(tag->id == TAGID_SHOWFRAME)
366                 {
367                     tmp--;
368                     logf("<debug> %d frames to go",tmp);
369                 }
370             }
371         }
372         while(slave.tags[pos++].id != TAGID_END);
373
374         *tagidpos = (u8*)writer_getpos(&w) - startpos; // set length of sprite (in header)
375         logf("<verbose> sprite length is %d",*tagidpos);
376
377         // write master (2)
378         pos = 0;
379         do {
380             if(!is_defining_tag(master.tags[pos].id))
381             {
382                 logf("<debug> [master] write tag %02x (%d bytes in body)", 
383                         master.tags[pos].id, master.tags[pos].length);
384                 writer_write(&w, master.tags[pos].fulldata, master.tags[pos].fulllength);
385             }
386         }
387         while(master.tags[pos++].id != 0);
388
389         tmp32 = (u8*)writer_getpos(&w) - (u8*)newdata; //length
390         *newlength = tmp32;
391         *headlength = tmp32; // set the header to the correct length
392
393         return newdata; //length
394     }
395     
396     *newlength = 0;
397     return 0;
398 }