X-Git-Url: http://git.asbjorn.biz/?a=blobdiff_plain;f=src%2Fswfextract.c;h=a4a2dede7c6cd70113f73567aef750f86d6f7266;hb=8948fb16775bb8334db34e2a99ae41c28cc212c9;hp=a5f549a8d54bd7a8836d630e460edaa4bd0cf20d;hpb=e567897c0af197377b6049fa5336285b4c0b6a67;p=swftools.git

diff --git a/src/swfextract.c b/src/swfextract.c
index a5f549a..a4a2ded 100644
--- a/src/swfextract.c
+++ b/src/swfextract.c
@@ -354,8 +354,34 @@ void extractTag(SWF*swf, char*filename)
     }
     if(!extractframes && !hollow) {
 	if(!originalplaceobjects && (extractids||extractname_id>=0)) {
+            int number = 0;
+            int id = 0;
 	    int t;
-	    int s=0;
+            TAG* objtag = 0;
+            SRECT bbox;
+            memset(&bbox, 0, sizeof(SRECT));
+	    for(t=0;t<65536;t++) {
+	        if(is_in_range(t, extractids)) {
+                    id = t;
+                    number++;
+                }
+            }
+            if(number>=2)
+		printf("warning! You should use the -P when extracting multiple objects\n");
+            if(number == 1) {
+                /* if there is only one object, we will scale it.
+                   So let's figure out it's bounding box */
+                TAG*tag = swf->firstTag;
+                while(tag) {
+                    if(swf_isDefiningTag(tag) && tag->id != ST_DEFINESPRITE) {
+                        if(swf_GetDefineID(tag) == id)
+                            bbox = swf_GetDefineBBox(tag);
+                        objtag = tag;
+                    }
+                    tag = tag->next;
+                }
+            }
+
 	    if((objectbbox.xmin|objectbbox.ymin|objectbbox.xmax|objectbbox.ymax)!=0)
 		newswf.movieSize = objectbbox;
 	    if(extractname_id>=0) {
@@ -364,11 +390,22 @@ void extractTag(SWF*swf, char*filename)
 	    } else {
 		for(t=0;t<65536;t++) {
 		    if(is_in_range(t, extractids)) {
+                        MATRIX m;
 			desttag = swf_InsertTag(desttag, ST_PLACEOBJECT2);
-			swf_ObjectPlace(desttag, t, t, 0,0,0);
-			s++;
-			if(s==2)
-			    printf("warning! You should use the -P when extracting multiple objects\n");
+                        swf_GetMatrix(0, &m);
+                        if(objtag) {
+                            int width = bbox.xmax - bbox.xmin;
+                            int height = bbox.ymax - bbox.ymin;
+                            int max = width>height?width:height;
+                            m.tx = -bbox.xmin;
+                            m.ty = -bbox.ymin;
+                            if(max) {
+                                m.sx = (512*20*65536)/max;
+                                m.sy = (512*20*65536)/max;
+                            }
+                            newswf.movieSize = swf_TurnRect(newswf.movieSize, &m);
+                        }
+			swf_ObjectPlace(desttag, t, t, &m,0,0);
 		    }
 		}
 	    }
@@ -384,80 +421,111 @@ void extractTag(SWF*swf, char*filename)
     swf_FreeTags(&newswf);                       // cleanup
 }
 
+int isOfType(int t, TAG*tag)
+{
+    int show = 0;
+    if(t == 0 && (tag->id == ST_DEFINESHAPE ||
+	tag->id == ST_DEFINESHAPE2 ||
+	tag->id == ST_DEFINESHAPE3)) {
+	show = 1;
+    }
+    if(t==1 && tag->id == ST_DEFINESPRITE) {
+	show = 1;
+    }
+    if(t == 2 && (tag->id == ST_DEFINEBITS ||
+	tag->id == ST_DEFINEBITSJPEG2 ||
+	tag->id == ST_DEFINEBITSJPEG3)) {
+	show = 1;
+    }
+    if(t == 3 && (tag->id == ST_DEFINEBITSLOSSLESS ||
+	tag->id == ST_DEFINEBITSLOSSLESS2)) {
+	show = 1;
+    }
+    if(t == 4 && (tag->id == ST_DEFINESOUND)) {
+	show = 1;
+    }
+    if(t == 5 && (tag->id == ST_DEFINEFONT || tag->id == ST_DEFINEFONT2)) {
+	show = 1;
+    }
+    return show;
+}
+
 void listObjects(SWF*swf)
 {
     TAG*tag;
     char first;
     int t;
     int frame = 0;
-    char*names[] = {"Shapes","MovieClips","JPEGs","PNGs","Sounds","Fonts"};
+    char*names[] = {"Shape", "MovieClip", "JPEG", "PNG", "Sound", "Font"};
+    char*options[] = {"-i", "-i", "-j", "-p", "-s", "-F"};
+    int mp3=0;
     printf("Objects in file %s:\n",filename);
+    swf_FoldAll(swf);
     for(t=0;t<sizeof(names)/sizeof(names[0]);t++) {
+	int nr=0;
+	int lastid = -2, lastprint=-1;
+	int follow=0;
 	tag = swf->firstTag;
 	first = 1;
 	while(tag) {
-	    char show = 0;
+	    if(tag->id == ST_SOUNDSTREAMHEAD || tag->id == ST_SOUNDSTREAMHEAD2)
+		mp3 = 1;
+	    if(isOfType(t,tag))
+		nr++;
+	    tag = tag->next;
+	}
+	if(!nr)
+	    continue;
+	
+	printf(" [%s] %d %s%s: ID(s) ", options[t], nr, names[t], nr>1?"s":"");
+
+	tag = swf->firstTag;
+	while(tag) {
 	    char text[80];
-	    if(t == 0 &&
-	       (tag->id == ST_DEFINESHAPE ||
-		tag->id == ST_DEFINESHAPE2 ||
-		tag->id == ST_DEFINESHAPE3)) {
-		show = 1;
-	        sprintf(text,"%d", swf_GetDefineID(tag));
+	    char show = isOfType(t,tag);
+	    int id;
+	    if(!show) {
+		tag = tag->next;
+		continue;
 	    }
+	    id = swf_GetDefineID(tag);
 
-	    if(tag->id == ST_DEFINESPRITE) {
-		if (t == 1)  {
-		    show = 1;
-		    sprintf(text,"%d", swf_GetDefineID(tag));
+	    if(id == lastid+1) {
+		follow=1;
+	    } else {
+		if(first || !follow) {
+		    if(!first)
+			printf(", ");
+		    printf("%d", id);
+		} else {
+		    if(lastprint + 1 == lastid) 
+			printf(", %d, %d", lastid, id);
+		    else
+			printf("-%d, %d", lastid, id);
 		}
-
-		while(tag->id != ST_END)
-		    tag = tag->next;
-	    }
-
-	    if(t == 2 && (tag->id == ST_DEFINEBITS ||
-		tag->id == ST_DEFINEBITSJPEG2 ||
-		tag->id == ST_DEFINEBITSJPEG3)) {
-		show = 1;
-		sprintf(text,"%d", swf_GetDefineID(tag));
-	    }
-
-	    if(t == 3 && (tag->id == ST_DEFINEBITSLOSSLESS ||
-		tag->id == ST_DEFINEBITSLOSSLESS2)) {
-		show = 1;
-		sprintf(text,"%d", swf_GetDefineID(tag));
-	    }
-
-
-	    if(t == 4 && (tag->id == ST_DEFINESOUND)) {
-		show = 1;
-		sprintf(text,"%d", swf_GetDefineID(tag));
-	    }
-	    
-	    if(t == 5 && (tag->id == ST_DEFINEFONT || tag->id == ST_DEFINEFONT2)) {
-		show = 1;
-		sprintf(text,"%d", swf_GetDefineID(tag));
-	    }
-	    
-	    if(show) {
-		if(!first)
-		    printf(", ");
-		else
-		    printf("%s: ", names[t]);
-		printf("%s", text);
+		lastprint = id;
 		first = 0;
+		follow = 0;
 	    }
+	    lastid = id;
 	    tag=tag->next;
 	}
-	if(!first)
-	    printf("\n");
+	if(follow) {
+	    if(lastprint + 1 == lastid)
+		printf(", %d", lastid);
+	    else
+		printf("-%d", lastid);
+	}
+	printf("\n");
     }
 
     if(frame)
-	printf("Frames: 0-%d\n", frame);
+	printf(" [-f] %d Frames: ID(s) 0-%d\n", frame, frame);
     else
-	printf("Frames: 0\n");
+	printf(" [-f] 1 Frame: ID(s) 0\n");
+
+    if(mp3)
+	printf(" [-m] 1 MP3 Soundstream\n");
 }
 
 void handlefont(SWF*swf, TAG*tag)
@@ -538,7 +606,7 @@ void handlejpeg(TAG*tag)
     /* swf jpeg images have two streams, which both start with ff d8 and
        end with ff d9. The following code handles sorting the middle
        <ff d9 ff d8> bytes out, so that one stream remains */
-    if(tag->id == ST_DEFINEBITS && tag->len>2 && jpegtables) {
+    if(tag->id == ST_DEFINEBITSJPEG && tag->len>2 && jpegtables) {
 	fi = save_fopen(filename, "wb");
 	fwrite(jpegtables, 1, jpegtablessize-2, fi); //don't write end tag (ff,d8)
 	fwrite(&tag->data[2+2], tag->len-2-2, 1, fi); //don't write start tag (ff,d9)
@@ -547,13 +615,17 @@ void handlejpeg(TAG*tag)
     else if(tag->id == ST_DEFINEBITSJPEG2 && tag->len>2) {
 	int end = tag->len;
 	int pos = findjpegboundary(&tag->data[2], tag->len-2);
-	if(pos<0)
-	    return;
-	pos+=2;
-	fi = save_fopen(filename, "wb");
-	fwrite(&tag->data[2], pos-2, 1, fi);
-	fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
-	fclose(fi);
+	if(pos>=0) {
+            pos+=2;
+            fi = save_fopen(filename, "wb");
+            fwrite(&tag->data[2], pos-2, 1, fi);
+            fwrite(&tag->data[pos+4], end-(pos+4), 1, fi);
+            fclose(fi);
+        } else {
+            fi = save_fopen(filename, "wb");
+            fwrite(&tag->data[2], end-2, 1, fi);
+            fclose(fi);
+        }
     }
     else if(tag->id == ST_DEFINEBITSJPEG3 && tag->len>6) {
 	U32 end = GET32(&tag->data[2])+6;
@@ -593,7 +665,7 @@ static void make_crc32_table(void)
     crc32_table[t] = c;
   }
 }
-static void png_write_byte(FILE*fi, U8 byte)
+static inline void png_write_byte(FILE*fi, U8 byte)
 {
     fwrite(&byte,1,1,fi);
     mycrc32 = crc32_table[(mycrc32 ^ byte) & 0xff] ^ (mycrc32 >> 8);
@@ -738,8 +810,10 @@ void handlelossless(TAG*tag)
      png_write_byte(fi,8);
      if(format == 3)
      png_write_byte(fi,3); //indexed
-     else if(format == 5)
+     else if(format == 5 && alpha==0)
      png_write_byte(fi,2); //rgb
+     else if(format == 5 && alpha==1)
+     png_write_byte(fi,6); //rgba
      else return;
 
      png_write_byte(fi,0); //compression mode
@@ -749,6 +823,7 @@ void handlelossless(TAG*tag)
    
     if(format == 3) {
 	png_start_chunk(fi, "PLTE", 768);
+	 
 	 for(t=0;t<256;t++) {
 	     png_write_byte(fi,palette[t].r);
 	     png_write_byte(fi,palette[t].g);
@@ -766,12 +841,22 @@ void handlelossless(TAG*tag)
 	{
 	   data3[pos2++]=0; //filter type
 	   if(bpp==32) {
-	    // 32 bit to 24 bit "conversion"
-	    for(x=0;x<width;x++) {
-	        data3[pos2++]=data[pos+1];
-	        data3[pos2++]=data[pos+2];
-		data3[pos2++]=data[pos+3];
-		pos+=4; //ignore padding byte
+	    if(!alpha) {
+		// 32 bit to 24 bit "conversion"
+		for(x=0;x<width;x++) {
+		    data3[pos2++]=data[pos+1];
+		    data3[pos2++]=data[pos+2];
+		    data3[pos2++]=data[pos+3];
+		    pos+=4; //ignore padding byte
+		}
+	    } else {
+		for(x=0;x<width;x++) {
+		    data3[pos2++]=data[pos+1];
+		    data3[pos2++]=data[pos+2];
+		    data3[pos2++]=data[pos+3];
+		    data3[pos2++]=data[pos+0]; //alpha
+		    pos+=4;
+		}
 	    }
 	   }
 	   else {
@@ -841,25 +926,42 @@ void handledefinesound(TAG*tag)
     char buf[128];
     char*filename = buf;
     FILE*fi;
+    char*extension = 0;
+    int format;
     U16 id;
+    int rate,bits,stereo;
+    char*rates[] = {"5500","11025","22050","44100"};
     id = swf_GetU16(tag); //id
-    sprintf(buf, "sound%d.mp3", id);
+    
+    flags = swf_GetU8(tag);
+    format = flags>>4;
+    rate = (flags>>2)&3;
+    bits = flags&2?16:8;
+    stereo = flags&1;
+    
+    samples = swf_GetU32(tag);
 
+    extension = "raw";
+
+    if(format == 2) { // mp3
+	swf_GetU16(tag); //numsamples_seek
+	extension = "mp3";
+    } else if(format == 0) { // raw
+	printf("Sound is RAW, format: %s samples/sec, %d bit, %s\n", rates[rate], bits, stereo?"stereo":"mono");
+	// TODO: convert to WAV
+	extension = "raw";
+    } else if(format == 1) { // adpcm
+	printf("Sound is ADPCM, format: %s samples/sec, %d bit, %s\n", rates[rate], bits, stereo?"stereo":"mono");
+	extension = "adpcm";
+    }
+    sprintf(buf, "sound%d.%s", id, extension);
     if(numextracts==1) {
 	filename = destfilename;
-	if(!strcmp(filename,"output.swf"))
-	    filename = "output.mp3";
-    }
-    flags = swf_GetU8(tag);
-    if((flags>>4)!=2) {
-	printf("Sorry, can only extract MP3 sounds. Sound %d is ADPCM or RAW.\n", id);
-	/* not mp3 */
-	return;
+	if(!strcmp(filename,"output.swf")) {
+	    sprintf(buf, "output.%s", extension);
+	    filename = buf;
+	}
     }
-    samples = swf_GetU32(tag);
-    
-    swf_GetU16(tag); //(only for mp3) numsamples_seek
-
     fi = save_fopen(filename, "wb");
     fwrite(&tag->data[tag->pos], tag->len - tag->pos, 1, fi);
     fclose(fi);