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