src/png2swf.c

   1 /* png2swf.c
   2
   3    PNG to SWF converter tool
   4
   5    Part of the swftools package.
   6
   7    Copyright (c) 2002,2003 Matthias Kramm <kramm@quiss.org>
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  22
  23 #include <stdio.h>
  24 #include <math.h>
  25 #include <fcntl.h>
  26 #include <zlib.h>
  27 #include "../lib/rfxswf.h"
  28 #include "../lib/args.h"
  29 #include "../lib/png.h"
  30
  31 #define MAX_INPUT_FILES 1024
  32 #define VERBOSE(x) (global.verbose>=x)
  33
  34 struct {
  35     float framerate;
  36     int max_image_width;
  37     int max_image_height;
  38     int force_width;
  39     int force_height;
  40     int nfiles;
  41     int verbose;
  42     int do_cgi;
  43     int version;
  44     char *outfile;
  45     int mkjpeg;
  46     float scale;
  47 } global;
  48
  49 struct {
  50     char *filename;
  51 } image[MAX_INPUT_FILES];
  52
  53 static int custom_move=0;
  54 static int move_x=0;
  55 static int move_y=0;
  56 static int clip_x1=0,clip_y1=0,clip_x2=0,clip_y2=0;
  57 static int custom_clip = 0;
  58
  59 TAG *MovieStart(SWF * swf, float framerate, int dx, int dy)
  60 {
  61     TAG *t;
  62     RGBA rgb;
  63
  64     memset(swf, 0x00, sizeof(SWF));
  65
  66     swf->fileVersion = global.version;
  67     swf->frameRate = (int)(256.0 * framerate);
  68     if(custom_clip) {
  69         swf->movieSize.xmin = clip_x1 * 20;
  70         swf->movieSize.ymin = clip_y1 * 20;
  71         swf->movieSize.xmax = clip_x2 * 20;
  72         swf->movieSize.ymax = clip_y2 * 20;
  73     } else {
  74         swf->movieSize.xmax = dx * 20;
  75         swf->movieSize.ymax = dy * 20;
  76     }
  77
  78     t = swf->firstTag = swf_InsertTag(NULL, ST_SETBACKGROUNDCOLOR);
  79
  80     rgb.r = rgb.g = rgb.b = rgb.a = 0x00;
  81     //rgb.g = 0xff; //<--- handy for testing alpha conversion
  82     swf_SetRGB(t, &rgb);
  83
  84     return t;
  85 }
  86
  87 int MovieFinish(SWF * swf, TAG * t, char *sname)
  88 {
  89     int f, so = fileno(stdout);
  90     t = swf_InsertTag(t, ST_END);
  91
  92     if ((!isatty(so)) && (!sname))
  93         f = so;
  94     else {
  95         if (!sname)
  96             sname = "output.swf";
  97         f = open(sname,O_WRONLY|O_CREAT|O_TRUNC|O_BINARY, 0644);
  98     }
  99
 100     if(global.do_cgi) {
 101         if FAILED(swf_WriteCGI(swf)) fprintf(stderr,"WriteCGI() failed.\n");
 102     } else {
 103         if (swf_WriteSWF(f, swf)<0)
 104             fprintf(stderr, "Unable to write output file: %s\n", sname);
 105         if (f != so)
 106             close(f);
 107     }
 108
 109     swf_FreeTags(swf);
 110     return 0;
 111 }
 112
 113 int png_read_chunk(char (*head)[4], int*destlen, U8**destdata, FILE*fi)
 114 {
 115     unsigned int len;
 116     if(destlen) *destlen=0;
 117     if(destdata) *destdata=0;
 118     if(!fread(&len, 4, 1, fi))
 119         return 0;
 120     if(!fread(head, 4, 1, fi))
 121         return 0;
 122     len = BE_32_TO_NATIVE(len);
 123     if(destlen) *destlen = len;
 124     if(destdata) {
 125         if(len)
 126             *destdata = malloc(len);
 127         else
 128             *destdata = 0;
 129         if(!fread(*destdata, len, 1, fi)) {
 130             *destdata = 0;
 131             if(destlen) *destlen=0;
 132             return 0;
 133         }
 134         fseek(fi, 4, SEEK_CUR);
 135
 136     } else {
 137         fseek(fi, len+4, SEEK_CUR);
 138     }
 139     return 1;
 140 }
 141
 142 unsigned int png_get_dword(FILE*fi)
 143 {
 144     unsigned int a;
 145     fread(&a,4,1,fi);
 146     return BE_32_TO_NATIVE(a);
 147 }
 148
 149 struct png_header
 150 {
 151     int width;
 152     int height;
 153     int bpp;
 154     int mode;
 155 };
 156
 157 int png_read_header(FILE*fi, struct png_header*header)
 158 {
 159     char id[4];
 160     int len;
 161     int ok=0;
 162     U8 head[8] = {137,80,78,71,13,10,26,10};
 163     U8 head2[8];
 164     U8*data;
 165     fread(head2,8,1,fi);
 166     if(strncmp((char*)head,(char*)head2,4))
 167         return 0;
 168
 169     while(png_read_chunk(&id, &len, &data, fi))
 170     {
 171         if(VERBOSE(2))
 172             printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
 173         if(!strncasecmp(id, "IHDR", 4)) {
 174             char a,b,c,f,i;
 175             if(len < 8) exit(1);
 176             header->width = BE_32_TO_NATIVE(*(U32*)&data[0]);
 177             header->height = BE_32_TO_NATIVE(*(U32*)&data[4]);
 178             a = data[8];      // should be 8
 179             b = data[9];      // should be 3(indexed), 2(rgb), 0(grayscale) or 6(truecolor+alpha)
 180
 181             c = data[10];     // compression mode (0)
 182             f = data[11];     // filter mode (0)
 183             i = data[12];     // interlace mode (0)
 184
 185             if(VERBOSE(2)) printf("image mode:%d\n", b);
 186             if(VERBOSE(2)) printf("bpp: %d\n", a);
 187             if(VERBOSE(2)) printf("compression: %d\n", c);
 188             if(VERBOSE(2)) printf("filter: %d\n", f);
 189             if(VERBOSE(2)) printf("interlace: %d\n", i);
 190
 191             if(b!=0 && b!=2 && b!=3 && b!=6) {
 192                 fprintf(stderr, "Image mode %d not supported!\n", b);
 193                 if(b == 4) {
 194                     fprintf(stderr, "(This is a grayscale image with alpha channel-\n");
 195                     fprintf(stderr, " try converting it into an RGB image with alpha channel)\n");
 196                 }
 197                 exit(1);
 198             }
 199             if(a!=8 && (b==2 || b==6)) {
 200                 fprintf(stderr, "Bpp %d in mode %d not supported!\n", b, a);
 201                 exit(1);
 202             }
 203             if(c!=0) {
 204                 fprintf(stderr, "Compression mode %d not supported!\n", c);
 205                 exit(1);
 206             }
 207             if(f!=0) {
 208                 fprintf(stderr, "Filter mode %d not supported!\n", f);
 209                 exit(1);
 210             }
 211             if(i!=0) {
 212                 fprintf(stderr, "Interlace mode %d not supported!\n", i);
 213                 exit(1);
 214             }
 215             if(VERBOSE(2))
 216                 printf("%dx%d %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
 217             header->bpp = a;
 218             header->mode = b;
 219             ok = 1;
 220         }
 221
 222         free(data);
 223     }
 224     return ok;
 225 }
 226
 227 typedef unsigned char byte;
 228 #define ABS(a) ((a)>0?(a):(-(a)))
 229 byte inline PaethPredictor (byte a,byte b,byte c)
 230 {
 231         // a = left, b = above, c = upper left
 232         int p = a + b - c;        // initial estimate
 233         int pa = ABS(p - a);      // distances to a, b, c
 234         int pb = ABS(p - b);
 235         int pc = ABS(p - c);
 236         // return nearest of a,b,c,
 237         // breaking ties in order a,b,c.
 238         if (pa <= pb && pa <= pc)
 239                 return a;
 240         else if (pb <= pc)
 241                 return b;
 242         else return c;
 243 }
 244
 245 void applyfilter3(int mode, U8*src, U8*old, U8*dest, int width)
 246 {
 247     int x;
 248     unsigned char lastr=0;
 249     unsigned char lastg=0;
 250     unsigned char lastb=0;
 251     unsigned char upperlastr=0;
 252     unsigned char upperlastg=0;
 253     unsigned char upperlastb=0;
 254
 255     if(mode==0) {
 256         for(x=0;x<width;x++) {
 257             dest[0] = 255;
 258             dest[1] = src[0];
 259             dest[2] = src[1];
 260             dest[3] = src[2];
 261             dest+=4;
 262             src+=3;
 263         }
 264     }
 265     else if(mode==1) {
 266         for(x=0;x<width;x++) {
 267             dest[0] = 255;
 268             dest[1] = src[0]+lastr;
 269             dest[2] = src[1]+lastg;
 270             dest[3] = src[2]+lastb;
 271             lastr = dest[1];
 272             lastg = dest[2];
 273             lastb = dest[3];
 274             dest+=4;
 275             src+=3;
 276         }
 277     }
 278     else if(mode==2) {
 279         for(x=0;x<width;x++) {
 280             dest[0] = 255;
 281             dest[1] = src[0]+old[1];
 282             dest[2] = src[1]+old[2];
 283             dest[3] = src[2]+old[3];
 284             dest+=4;
 285             old+=4;
 286             src+=3;
 287         }
 288     }
 289     else if(mode==3) {
 290         for(x=0;x<width;x++) {
 291             dest[0] = 255;
 292             dest[1] = src[0]+(old[1]+lastr)/2;
 293             dest[2] = src[1]+(old[2]+lastg)/2;
 294             dest[3] = src[2]+(old[3]+lastb)/2;
 295             lastr = dest[1];
 296             lastg = dest[2];
 297             lastb = dest[3];
 298             dest+=4;
 299             old+=4;
 300             src+=3;
 301         }
 302     }
 303     else if(mode==4) {
 304         for(x=0;x<width;x++) {
 305             dest[0] = 255;
 306             dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
 307             dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
 308             dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
 309             lastr = dest[1];
 310             lastg = dest[2];
 311             lastb = dest[3];
 312             upperlastr = old[1];
 313             upperlastg = old[2];
 314             upperlastb = old[3];
 315             dest+=4;
 316             old+=4;
 317             src+=3;
 318         }
 319     }
 320 }
 321
 322 void applyfilter4(int mode, U8*src, U8*old, U8*dest, int width)
 323 {
 324     int x;
 325     unsigned char lastr=0;
 326     unsigned char lastg=0;
 327     unsigned char lastb=0;
 328     unsigned char lasta=0; //TODO: 255?
 329     unsigned char upperlastr=0;
 330     unsigned char upperlastg=0;
 331     unsigned char upperlastb=0;
 332     unsigned char upperlasta=0; //TODO: 255?
 333
 334     if(mode==0) {
 335         for(x=0;x<width;x++) {
 336             dest[0] = src[3];
 337             dest[1] = src[0];
 338             dest[2] = src[1];
 339             dest[3] = src[2];
 340             dest+=4;
 341             src+=4;
 342         }
 343     }
 344     else if(mode==1) {
 345         for(x=0;x<width;x++) {
 346             dest[0] = src[3]+lasta;
 347             dest[1] = src[0]+lastr;
 348             dest[2] = src[1]+lastg;
 349             dest[3] = src[2]+lastb;
 350             lasta = dest[0];
 351             lastr = dest[1];
 352             lastg = dest[2];
 353             lastb = dest[3];
 354             dest+=4;
 355             src+=4;
 356         }
 357     }
 358     else if(mode==2) {
 359         for(x=0;x<width;x++) {
 360             dest[0] = src[3]+old[0];
 361             dest[1] = src[0]+old[1];
 362             dest[2] = src[1]+old[2];
 363             dest[3] = src[2]+old[3];
 364             dest+=4;
 365             old+=4;
 366             src+=4;
 367         }
 368     }
 369     else if(mode==3) {
 370         for(x=0;x<width;x++) {
 371             dest[0] = src[3]+(old[0]+lasta)/2;
 372             dest[1] = src[0]+(old[1]+lastr)/2;
 373             dest[2] = src[1]+(old[2]+lastg)/2;
 374             dest[3] = src[2]+(old[3]+lastb)/2;
 375             lasta = dest[0];
 376             lastr = dest[1];
 377             lastg = dest[2];
 378             lastb = dest[3];
 379             dest+=4;
 380             old+=4;
 381             src+=4;
 382         }
 383     }
 384     else if(mode==4) {
 385         for(x=0;x<width;x++) {
 386             dest[0] = src[3]+PaethPredictor(lasta,old[0],upperlasta);
 387             dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
 388             dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
 389             dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
 390             lasta = dest[0];
 391             lastr = dest[1];
 392             lastg = dest[2];
 393             lastb = dest[3];
 394             upperlasta = old[0];
 395             upperlastr = old[1];
 396             upperlastg = old[2];
 397             upperlastb = old[3];
 398             dest+=4;
 399             old+=4;
 400             src+=4;
 401         }
 402     }
 403
 404 }
 405
 406 void applyfilter1(int mode, U8*src, U8*old, U8*dest, int width)
 407 {
 408     int x;
 409     unsigned char last=0;
 410     unsigned char upperlast=0;
 411
 412     if(mode==0) {
 413         for(x=0;x<width;x++) {
 414             *dest = *src;
 415             dest++;
 416             src++;
 417         }
 418     }
 419     else if(mode==1) {
 420         for(x=0;x<width;x++) {
 421             *dest = *src+last;
 422             last = *dest;
 423             dest++;
 424             src++;
 425         }
 426     }
 427     else if(mode==2) {
 428         for(x=0;x<width;x++) {
 429             *dest = *src+*old;
 430             dest++;
 431             old++;
 432             src++;
 433         }
 434     }
 435     else if(mode==3) {
 436         for(x=0;x<width;x++) {
 437             *dest = *src+(*old+last)/2;
 438             last = *dest;
 439             dest++;
 440             old++;
 441             src++;
 442         }
 443     }
 444     else if(mode==4) {
 445         for(x=0;x<width;x++) {
 446             *dest = *src+PaethPredictor(last,*old,upperlast);
 447             last = *dest;
 448             upperlast = *old;
 449             dest++;
 450             old++;
 451             src++;
 452         }
 453     }
 454
 455 }
 456
 457 TAG* PNG2Image(TAG*t, U16 id, char*filename, unsigned*width, unsigned*height)
 458 {
 459     char tagid[4];
 460     U8*data;
 461     U8*imagedata;
 462     U8*zimagedata=0;
 463     unsigned long int imagedatalen;
 464     unsigned long int zimagedatalen=0;
 465     U8*palette = 0;
 466     int palettelen = 0;
 467     U8*alphapalette = 0;
 468     int alphapalettelen = 0;
 469     struct png_header header;
 470     int bypp;
 471     U8 alphacolor[3];
 472     int hasalphacolor=0;
 473     int len;
 474
 475
 476     FILE *fi;
 477     U8 *scanline;
 478
 479     if ((fi = fopen(filename, "rb")) == NULL) {
 480         if (VERBOSE(1))
 481             fprintf(stderr, "Read access failed: %s\n", filename);
 482         return t;
 483     }
 484
 485     if(!png_read_header(fi, &header))
 486         return 0;
 487
 488     *width = header.width;
 489     *height = header.height;
 490
 491     if(header.mode == 3 || header.mode == 0) bypp = 1;
 492     else
 493     if(header.mode == 2) bypp = 3;
 494     else
 495     if(header.mode == 6) bypp = 4;
 496     else
 497         return 0;
 498     imagedatalen = bypp * header.width * header.height + 65536;
 499     imagedata = malloc(imagedatalen);
 500
 501     fseek(fi,8,SEEK_SET);
 502     while(!feof(fi))
 503     {
 504         if(!png_read_chunk(&tagid, &len, &data, fi))
 505             break;
 506         if(!strncmp(tagid, "IEND", 4)) {
 507             break;
 508         }
 509         if(!strncmp(tagid, "PLTE", 4)) {
 510             palette = data;
 511             palettelen = len/3;
 512             data = 0; //don't free data
 513             if(VERBOSE(2))
 514                 printf("%d colors in palette\n", palettelen);
 515         }
 516         if(!strncmp(tagid, "tRNS", 4)) {
 517             if(header.mode == 3) {
 518                 alphapalette = data;
 519                 alphapalettelen = len;
 520                 data = 0; //don't free data
 521                 if(VERBOSE(2))
 522                     printf("found %d alpha colors\n", alphapalettelen);
 523             } else if(header.mode == 0 || header.mode == 2) {
 524                 int t;
 525                 if(header.mode == 2) { // palette or grayscale?
 526                     alphacolor[0] = data[1];
 527                     alphacolor[1] = data[3];
 528                     alphacolor[2] = data[5];
 529                 } else {
 530                     alphacolor[0] = alphacolor[1] = alphacolor[2] = data[1];
 531                 }
 532                 if(VERBOSE(2))
 533                     printf("found alpha color: %02x%02x%02x\n", alphacolor[0], alphacolor[1], alphacolor[2]);
 534                 hasalphacolor = 1;
 535             } else {
 536                 if(VERBOSE(2))
 537                     printf("ignoring tRNS %d entry (%d bytes)\n", header.mode, len);
 538             }
 539         }
 540         if(!strncmp(tagid, "IDAT", 4)) {
 541             if(!zimagedata) {
 542                 zimagedatalen = len;
 543                 zimagedata = malloc(len);
 544                 memcpy(zimagedata,data,len);
 545             } else {
 546                 zimagedata = realloc(zimagedata, zimagedatalen+len);
 547                 memcpy(&zimagedata[zimagedatalen], data, len);
 548                 zimagedatalen += len;
 549             }
 550         }
 551         if(data)
 552             free(data);
 553     }
 554
 555     if(!zimagedata || uncompress(imagedata, &imagedatalen, zimagedata, zimagedatalen) != Z_OK) {
 556         fprintf(stderr, "Couldn't uncompress IDAT chunk (%lu bytes) in %s!\n", imagedatalen, filename);
 557         if(zimagedata)
 558             free(zimagedata);
 559         return 0;
 560     }
 561     free(zimagedata);
 562
 563     if(alphapalette || hasalphacolor)
 564         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS2);
 565     else
 566         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
 567
 568     swf_SetU16(t, id);          // id
 569     if(header.mode == 2 || header.mode == 6) {
 570         U8*data2 = malloc(header.width*header.height*4);
 571         int i,s=0;
 572         int x,y;
 573         int pos=0;
 574         int opaque=0;
 575         int transparent=0;
 576         int semitransparent=0;
 577         /* in case for mode 2, the following also performs 24->32 bit conversion */
 578         unsigned char* firstline = malloc(header.width*4);
 579
 580         for(y=0;y<header.height;y++) {
 581             int mode = imagedata[pos++]; //filter mode
 582             U8*src;
 583             U8*dest;
 584             U8*old;
 585             dest = &data2[(y*header.width)*4];
 586
 587             if(header.bpp == 8)
 588             {
 589                 /* one byte per pixel */
 590                 src = &imagedata[pos];
 591                 pos+=header.width*(header.mode==6?4:3);
 592             } else {
 593                 /* not implemented yet */
 594                 exit(1);
 595             }
 596
 597             if(!y) {
 598                 old = firstline;
 599                 memset(old, 0, header.width*4); //TODO: fill alpha with 255?
 600             } else {
 601                 old = &data2[(y-1)*header.width*4];
 602             }
 603             if(header.mode==6) {
 604                 applyfilter4(mode, src, old, dest, header.width);
 605             } else if(header.mode==2) {
 606                 applyfilter3(mode, src, old, dest, header.width);
 607                 /* replace alpha color */
 608                 if(hasalphacolor) {
 609                     int x;
 610                     for(x=0;x<header.width;x++) {
 611                         if(dest[x*4+1] == alphacolor[0] &&
 612                            dest[x*4+2] == alphacolor[1] &&
 613                            dest[x*4+3] == alphacolor[2]) {
 614                             *(U32*)&dest[x*4] = 0;
 615                         }
 616                     }
 617                 }
 618             }
 619         }
 620         free(firstline);
 621
 622         /* the image is now compressed and stored in data. Now let's take
 623            a look at the alpha values */
 624         if(header.mode == 6) {
 625             for(y=0;y<header.height;y++) {
 626                 U8*l = &data2[(y*header.width)*4];
 627                 for(x=0;x<header.width;x++) {
 628                     U8 a = l[x*4];
 629                     U8 b = l[x*4+1];
 630                     U8 g = l[x*4+2];
 631                     U8 r = l[x*4+3];
 632                     if(a==255) transparent++;
 633                     else {
 634                         if(a==0) opaque++;
 635                         else semitransparent++;
 636                         l[x*4+3]=(int)r*a/255;
 637                         l[x*4+2]=(int)g*a/255;
 638                         l[x*4+1]=(int)b*a/255;
 639                     }
 640                 }
 641             }
 642             if(semitransparent || opaque) {
 643                 t->id = ST_DEFINEBITSLOSSLESS2;
 644             }
 645         }
 646         swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
 647         free(data2);
 648     } else if(header.mode == 0 || header.mode == 3) {
 649         RGBA*rgba;
 650         int swf_width = BYTES_PER_SCANLINE(header.width);
 651         U8*data2 = malloc(swf_width*header.height);
 652         U8*tmpline = malloc(header.width);
 653         int i,x,y;
 654         int pos=0;
 655         if(header.mode == 3) { // palette or grayscale?
 656             rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
 657             if(!palette) {
 658                 fprintf(stderr, "Error: No palette found!\n");
 659                 exit(1);
 660             }
 661             /* 24->32 bit conversion */
 662             for(i=0;i<palettelen;i++) {
 663                 rgba[i].r = palette[i*3+0];
 664                 rgba[i].g = palette[i*3+1];
 665                 rgba[i].b = palette[i*3+2];
 666                 if(alphapalette && i<alphapalettelen) {
 667                     rgba[i].a = alphapalette[i];
 668                     rgba[i].r = ((int)rgba[i].r*rgba[i].a)/255;
 669                     rgba[i].g = ((int)rgba[i].g*rgba[i].a)/255;
 670                     rgba[i].b = ((int)rgba[i].b*rgba[i].a)/255;
 671                 } else {
 672                     rgba[i].a = 255;
 673                 }
 674                 if(hasalphacolor) {
 675                     if(rgba[i].r == alphacolor[0] &&
 676                        rgba[i].g == alphacolor[1] &&
 677                        rgba[i].b == alphacolor[2]) {
 678                         rgba[i].r = 0;
 679                         rgba[i].g = 0;
 680                         rgba[i].b = 0;
 681                         rgba[i].a = 0;
 682                     }
 683                 }
 684             }
 685         } else {
 686             int mult = (0x1ff>>header.bpp);
 687             palettelen = 1<<header.bpp;
 688             rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
 689             for(i=0;i<palettelen;i++) {
 690                 rgba[i].r = i*mult;
 691                 rgba[i].g = i*mult;
 692                 rgba[i].b = i*mult;
 693                 rgba[i].a = 255;
 694                 if(hasalphacolor) {
 695                     if(rgba[i].r == alphacolor[0]) {
 696                         rgba[i].r = 0;
 697                         rgba[i].g = 0;
 698                         rgba[i].b = 0;
 699                         rgba[i].a = 0;
 700                     }
 701                 }
 702             }
 703         }
 704
 705         for(y=0;y<header.height;y++) {
 706             int mode = imagedata[pos++]; //filter mode
 707             U8*old;
 708             U8*dest = &data2[y*swf_width];
 709             U8*src;
 710             src = &imagedata[pos];
 711             if(header.bpp == 8) {
 712                 pos+=header.width;
 713             } else {
 714                 int x,s=0;
 715                 int bitpos = 0;
 716                 U32 v = (1<<header.bpp)-1;
 717                 for(x=0;x<header.width;x++) {
 718                     U32 r = src[s/8]<<8 |
 719                             src[s/8+1];
 720                     int t;
 721                     tmpline[x] = (r>>(16-header.bpp-(s&7)))&v;
 722                     s+=header.bpp;
 723                 }
 724                 src = tmpline;
 725                 pos+=(header.width*header.bpp+7)/8;
 726             }
 727
 728             if(!y) {
 729                 memset(data2,0,swf_width);
 730                 old = &data2[y*swf_width];
 731             } else {
 732                 old = &data2[(y-1)*swf_width];
 733             }
 734             applyfilter1(mode, src, old, dest, header.width);
 735         }
 736         swf_SetLosslessBitsIndexed(t, header.width, header.height, data2, rgba, palettelen);
 737         free(tmpline);
 738         free(rgba);
 739         free(data2);
 740     }
 741     fclose(fi);
 742     return t;
 743 }
 744
 745 TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
 746 {
 747     SHAPE *s;
 748     SRECT r;
 749     MATRIX m;
 750     int fs;
 751
 752     unsigned width=0, height=0;
 753
 754 #ifndef HAVE_JPEGLIB
 755     if(global.mkjpeg) {
 756         global.mkjpeg = 0;
 757         msg("<warning> No jpeg support compiled in");
 758     }
 759 #endif
 760     if(global.mkjpeg) {
 761 #ifdef HAVE_JPEGLIB
 762         RGBA*data = 0;
 763         getPNG(sname, &width, &height, (unsigned char**)&data);
 764         if(!data)
 765             exit(1);
 766         if(swf_ImageHasAlpha(data, width, height)) {
 767             t = swf_InsertTag(t, ST_DEFINEBITSJPEG3);
 768             swf_SetU16(t, id);
 769             swf_SetJPEGBits3(t, width,height,data,global.mkjpeg);
 770         } else {
 771             t = swf_InsertTag(t, ST_DEFINEBITSJPEG2);
 772             swf_SetU16(t, id);
 773             swf_SetJPEGBits2(t, width,height,data,global.mkjpeg);
 774         }
 775 #endif
 776     } else if(1) {
 777         RGBA*data = 0;
 778         getPNG(sname, &width, &height, (unsigned char**)&data);
 779         if(!data)
 780             exit(1);
 781         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
 782         swf_SetU16(t, id);
 783         swf_SetLosslessImage(t, data,width,height);
 784     } else {
 785         /* old code: transform PNG encoding 1:1 to SWF */
 786         t = PNG2Image(t, id, sname, &width, &height);
 787     }
 788
 789     t = swf_InsertTag(t, ST_DEFINESHAPE3);
 790
 791     swf_ShapeNew(&s);
 792     swf_GetMatrix(NULL, &m);
 793     m.sx = (int)(20 * 0x10000);
 794     m.sy = (int)(20 * 0x10000);
 795     m.tx = 0;
 796     m.ty = 0;
 797     fs = swf_ShapeAddBitmapFillStyle(s, &m, id, 1);
 798
 799     swf_SetU16(t, id + 1);      // id
 800
 801     r.xmin = r.ymin = 0;
 802     r.xmax = width * 20;
 803     r.ymax = height * 20;
 804     swf_SetRect(t, &r);
 805
 806     swf_SetShapeHeader(t, s);
 807
 808     swf_ShapeSetAll(t, s, 0, 0, 0, fs, 0);
 809     swf_ShapeSetLine(t, s, r.xmax, 0);
 810     swf_ShapeSetLine(t, s, 0, r.ymax);
 811     swf_ShapeSetLine(t, s, -r.xmax, 0);
 812     swf_ShapeSetLine(t, s, 0, -r.ymax);
 813
 814     swf_ShapeSetEnd(t);
 815
 816     t = swf_InsertTag(t, ST_REMOVEOBJECT2);
 817     swf_SetU16(t, 50);          // depth
 818
 819     t = swf_InsertTag(t, ST_PLACEOBJECT2);
 820
 821     swf_GetMatrix(NULL, &m);
 822     m.sx = (int)(0x10000 * global.scale);
 823     m.sy = (int)(0x10000 * global.scale);
 824
 825     if(custom_move) {
 826         m.tx = move_x*20;
 827         m.ty = move_y*20;
 828     } else {
 829         m.tx = (swf->movieSize.xmax - (int) (width * global.scale * 20)) / 2;
 830         m.ty = (swf->movieSize.ymax - (int) (height * global.scale * 20)) / 2;
 831     }
 832     swf_ObjectPlace(t, id + 1, 50, &m, NULL, NULL);
 833
 834     t = swf_InsertTag(t, ST_SHOWFRAME);
 835
 836     return t;
 837 }
 838
 839
 840 int CheckInputFile(char *fname, char **realname)
 841 {
 842     FILE *fi;
 843     char *s = malloc(strlen(fname) + 5);
 844     struct png_header head;
 845
 846     if (!s)
 847         exit(2);
 848     (*realname) = s;
 849     strcpy(s, fname);
 850
 851     // Check whether file exists (with typical extensions)
 852
 853     if ((fi = fopen(s, "rb")) == NULL) {
 854         sprintf(s, "%s.png", fname);
 855         if ((fi = fopen(s, "rb")) == NULL) {
 856             sprintf(s, "%s.PNG", fname);
 857             if ((fi = fopen(s, "rb")) == NULL) {
 858                 sprintf(s, "%s.Png", fname);
 859                 if ((fi = fopen(s, "rb")) == NULL) {
 860                     fprintf(stderr, "Couldn't open %s!\n", fname);
 861                     return -1;
 862                 }
 863             }
 864         }
 865     }
 866
 867     if(!png_read_header(fi, &head)) {
 868         fprintf(stderr, "%s is not a PNG file!\n", fname);
 869         return -1;
 870     }
 871
 872     if (global.max_image_width < head.width)
 873         global.max_image_width = head.width;
 874     if (global.max_image_height < head.height)
 875         global.max_image_height = head.height;
 876
 877     fclose(fi);
 878
 879     return 0;
 880 }
 881
 882 int args_callback_option(char *arg, char *val)
 883 {
 884     int res = 0;
 885     if (arg[1])
 886         res = -1;
 887     else
 888         switch (arg[0]) {
 889         case 'r':
 890             if (val)
 891                 global.framerate = atof(val);
 892             /* removed framerate>0 restriction in order to make
 893                Flash Communication Server compatible SWFs */
 894             if ((global.framerate < 0) ||(global.framerate >= 256.0)) {
 895                 if (VERBOSE(1))
 896                     fprintf(stderr,
 897                             "Error: You must specify a valid framerate between 1/256 and 255.\n");
 898                 exit(1);
 899             }
 900             res = 1;
 901             break;
 902
 903         case 'o':
 904             if (val)
 905                 global.outfile = val;
 906             res = 1;
 907             break;
 908
 909         case 's':
 910             global.scale = atof(val)/100;
 911             res = 1;
 912             break;
 913
 914         case 'z':
 915             if(global.version<6)
 916                 global.version = 6;
 917             res = 0;
 918             break;
 919
 920         case 'j':
 921             global.mkjpeg = atoi(val);
 922             res = 1;
 923             break;
 924
 925         case 'T':
 926             global.version = atoi(val);
 927             res = 1;
 928             break;
 929
 930         case 'C':
 931             global.do_cgi = 1;
 932             break;
 933
 934         case 'v':
 935             global.verbose++;
 936             res = 0;
 937             break;
 938
 939         case 'q':
 940             global.verbose--;
 941             if(global.verbose<0)
 942                 global.verbose = 0;
 943             res = 0;
 944             break;
 945
 946         case 'X':
 947             if (val)
 948                 global.force_width = atoi(val);
 949             res = 1;
 950             break;
 951
 952         case 'Y':
 953             if (val)
 954                 global.force_height = atoi(val);
 955             res = 1;
 956             break;
 957
 958         case 'V':
 959             printf("png2swf - part of %s %s\n", PACKAGE, VERSION);
 960             exit(0);
 961
 962         case 'c': {
 963             char*s = strdup(val);
 964             char*x1 = strtok(s, ":");
 965             char*y1 = strtok(0, ":");
 966             char*x2 = strtok(0, ":");
 967             char*y2 = strtok(0, ":");
 968             if(!(x1 && y1 && x2 && y2)) {
 969                 fprintf(stderr, "-m option requires four arguments, <x1>:<y1>:<x2>:<y2>\n");
 970                 exit(1);
 971             }
 972             custom_clip = 1;
 973             clip_x1 = atoi(x1);
 974             clip_y1 = atoi(y1);
 975             clip_x2 = atoi(x2);
 976             clip_y2 = atoi(y2);
 977             free(s);
 978
 979             res = 1;
 980             break;
 981         }
 982
 983         case 'm': {
 984             char*s = strdup(val);
 985             char*c = strchr(s, ':');
 986             if(!c) {
 987                 fprintf(stderr, "-m option requires two arguments, <x>:<y>\n");
 988                 exit(1);
 989             }
 990             *c = 0;
 991             custom_move = 1;
 992             move_x = atoi(val);
 993             move_y = atoi(c+1);
 994             free(s);
 995
 996             res = 1;
 997             break;
 998         }
 999
1000         default:
1001             res = -1;
1002             break;
1003         }
1004
1005     if (res < 0) {
1006         if (VERBOSE(1))
1007             fprintf(stderr, "Unknown option: -%s\n", arg);
1008         exit(1);
1009         return 0;
1010     }
1011     return res;
1012 }
1013
1014 static struct options_t options[] = {
1015 {"r", "rate"},
1016 {"o", "output"},
1017 {"j", "jpeg"},
1018 {"z", "zlib"},
1019 {"T", "flashversion"},
1020 {"X", "pixel"},
1021 {"Y", "pixel"},
1022 {"v", "verbose"},
1023 {"q", "quiet"},
1024 {"C", "cgi"},
1025 {"V", "version"},
1026 {"s", "scale"},
1027 {0,0}
1028 };
1029
1030 int args_callback_longoption(char *name, char *val)
1031 {
1032     return args_long2shortoption(options, name, val);
1033 }
1034
1035 int args_callback_command(char *arg, char *next)        // actually used as filename
1036 {
1037     char *s;
1038     if (CheckInputFile(arg, &s) < 0) {
1039         if (VERBOSE(1))
1040             fprintf(stderr, "Error opening input file: %s\n", arg);
1041         free(s);
1042     } else {
1043         image[global.nfiles].filename = s;
1044         global.nfiles++;
1045         if (global.nfiles >= MAX_INPUT_FILES) {
1046             if (VERBOSE(1))
1047                 fprintf(stderr, "Error: Too many input files.\n");
1048             exit(1);
1049         }
1050     }
1051     return 0;
1052 }
1053
1054 void args_callback_usage(char *name)
1055 {
1056     printf("\n");
1057     printf("Usage: %s [-X width] [-Y height] [-o file.swf] [-r rate] file1.png [file2.png...]\n", name);
1058     printf("\n");
1059     printf("-r , --rate <framerate>        Set movie framerate (frames per second)\n");
1060     printf("-o , --output <filename>       Set name for SWF output file.\n");
1061     printf("-j , --jpeg <quality>          Generate a lossy jpeg bitmap inside the SWF, with a given quality (1-100)\n");
1062     printf("-z , --zlib <zlib>             Enable Flash 6 (MX) Zlib Compression\n");
1063     printf("-T , --flashversion            Set the flash version to generate\n");
1064     printf("-X , --pixel <width>           Force movie width to <width> (default: autodetect)\n");
1065     printf("-Y , --pixel <height>          Force movie height to <height> (default: autodetect)\n");
1066     printf("-v , --verbose <level>         Set verbose level (0=quiet, 1=default, 2=debug)\n");
1067     printf("-q , --quiet                   Omit normal log messages, only log errors\n");
1068     printf("-C , --cgi                     For use as CGI- prepend http header, write to stdout\n");
1069     printf("-V , --version                 Print version information and exit\n");
1070     printf("-s , --scale <percent>         Scale image to <percent>%% size.\n");
1071     printf("\n");
1072 }
1073
1074 int main(int argc, char **argv)
1075 {
1076     SWF swf;
1077     TAG *t;
1078
1079     memset(&global, 0x00, sizeof(global));
1080
1081     global.framerate = 1.0;
1082     global.verbose = 1;
1083     global.version = 8;
1084     global.scale = 1.0;
1085
1086     processargs(argc, argv);
1087
1088     if(global.nfiles<=0) {
1089         fprintf(stderr, "No png files found in arguments\n");
1090         return 1;
1091     }
1092
1093     if (VERBOSE(2))
1094         fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
1095
1096     t = MovieStart(&swf, global.framerate,
1097                    global.force_width ? global.force_width : (int)(global.max_image_width*global.scale),
1098                    global.force_height ? global.force_height : (int)(global.max_image_height*global.scale));
1099
1100     {
1101         int i;
1102         for (i = 0; i < global.nfiles; i++) {
1103             if (VERBOSE(3))
1104                 fprintf(stderr, "[%03i] %s\n", i,
1105                         image[i].filename);
1106             t = MovieAddFrame(&swf, t, image[i].filename, (i * 2) + 1);
1107             free(image[i].filename);
1108         }
1109     }
1110
1111     MovieFinish(&swf, t, global.outfile);
1112
1113     return 0;
1114 }