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 Matthias Kramm <kramm@quiss.org>
   8
   9    This file is distributed under the GPL, see file COPYING for details
  10
  11 */
  12
  13 #include <stdio.h>
  14 #include <math.h>
  15 #include <fcntl.h>
  16 #include <zlib.h>
  17 #include "../lib/rfxswf.h"
  18 #include "../lib/args.h"
  19
  20 #define MAX_INPUT_FILES 1024
  21 #define VERBOSE(x) (global.verbose>=x)
  22
  23 struct {
  24     int framerate;
  25     int max_image_width;
  26     int max_image_height;
  27     int force_width;
  28     int force_height;
  29     int nfiles;
  30     int verbose;
  31     char *outfile;
  32 } global;
  33
  34 struct {
  35     char *filename;
  36 } image[MAX_INPUT_FILES];
  37
  38 TAG *MovieStart(SWF * swf, int framerate, int dx, int dy)
  39 {
  40     TAG *t;
  41     RGBA rgb;
  42
  43     memset(swf, 0x00, sizeof(SWF));
  44
  45     swf->fileVersion = 4;
  46     swf->frameRate = (25600 / framerate);
  47     swf->movieSize.xmax = dx * 20;
  48     swf->movieSize.ymax = dy * 20;
  49
  50     t = swf->firstTag = swf_InsertTag(NULL, ST_SETBACKGROUNDCOLOR);
  51
  52     rgb.r = rgb.g = rgb.b = rgb.a = 0x00;
  53     swf_SetRGB(t, &rgb);
  54
  55     return t;
  56 }
  57
  58 int MovieFinish(SWF * swf, TAG * t, char *sname)
  59 {
  60     int handle, so = fileno(stdout);
  61     t = swf_InsertTag(t, ST_END);
  62
  63     if ((!isatty(so)) && (!sname))
  64         handle = so;
  65     else {
  66         if (!sname)
  67             sname = "output.swf";
  68         handle = open(sname, O_RDWR | O_CREAT | O_TRUNC, 0666);
  69     }
  70     if FAILED
  71         (swf_WriteSWF(handle, swf)) if (VERBOSE(1))
  72             fprintf(stderr, "Unable to write output file: %s\n", sname);
  73     if (handle != so)
  74         close(handle);
  75
  76     swf_FreeTags(swf);
  77     return 0;
  78 }
  79
  80 int png_read_chunk(char (*head)[4], int*destlen, U8**destdata, FILE*fi)
  81 {
  82     unsigned int len;
  83     if(destlen) *destlen=0;
  84     if(destdata) *destdata=0;
  85     if(!fread(&len, 4, 1, fi))
  86         return 0;
  87     if(!fread(head, 4, 1, fi))
  88         return 0;
  89     len = REVERSESWAP32(len);
  90     if(destlen) *destlen = len;
  91     if(destdata) {
  92         if(len)
  93             *destdata = malloc(len);
  94         else
  95             *destdata = 0;
  96         if(!fread(*destdata, len, 1, fi)) {
  97             *destdata = 0;
  98             if(destlen) *destlen=0;
  99             return 0;
 100         }
 101         fseek(fi, 4, SEEK_CUR);
 102
 103     } else {
 104         fseek(fi, len+4, SEEK_CUR);
 105     }
 106     return 1;
 107 }
 108
 109 unsigned int png_get_dword(FILE*fi)
 110 {
 111     unsigned int a;
 112     fread(&a,4,1,fi);
 113     return REVERSESWAP32(a);
 114 }
 115
 116 struct png_header
 117 {
 118     int width;
 119     int height;
 120     int bpp;
 121     int mode;
 122 };
 123
 124 int png_read_header(FILE*fi, struct png_header*header)
 125 {
 126     char id[4];
 127     int len;
 128     int ok=0;
 129     U8 head[8] = {137,80,78,71,13,10,26,10};
 130     U8 head2[8];
 131     U8*data;
 132     fread(head2,8,1,fi);
 133     if(strncmp(head,head2,4))
 134         return 0;
 135
 136     while(png_read_chunk(&id, &len, &data, fi))
 137     {
 138         if(VERBOSE(2))
 139             printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
 140         if(!strncasecmp(id, "IHDR", 4)) {
 141             char a,b,c,f,i;
 142             if(len < 8) exit(1);
 143             header->width = REVERSESWAP32(*(U32*)&data[0]);
 144             header->height = REVERSESWAP32(*(U32*)&data[4]);
 145             a = data[8];      // should be 8
 146             b = data[9];      // should be 3(indexed) or 2(rgb)
 147
 148             c = data[10];     // compression mode (0)
 149             f = data[11];     // filter mode (0)
 150             i = data[12];     // interlace mode (0)
 151
 152             if(b!=2 && b!=3) {
 153                 fprintf(stderr, "Image mode %d not supported!\n", b);
 154                 exit(1);
 155             }
 156             if(a!=8 && b==2) {
 157                 fprintf(stderr, "Bpp %d in mode %d not supported!\n", a);
 158                 exit(1);
 159             }
 160             if(c!=0) {
 161                 fprintf(stderr, "Compression mode %d not supported!\n", c);
 162                 exit(1);
 163             }
 164             if(f!=0) {
 165                 fprintf(stderr, "Filter mode %d not supported!\n", f);
 166                 exit(1);
 167             }
 168             if(i!=0) {
 169                 fprintf(stderr, "Interlace mode %d not supported!\n", i);
 170                 exit(1);
 171             }
 172             if(VERBOSE(2))
 173                 printf("%dx%d %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
 174             header->bpp = a;
 175             header->mode = b;
 176             ok = 1;
 177         }
 178
 179         free(data);
 180     }
 181     return ok;
 182 }
 183
 184 typedef unsigned char byte;
 185 #define ABS(a) ((a)>0?(a):(-(a)))
 186 byte inline PaethPredictor (byte a,byte b,byte c)
 187 {
 188         // a = left, b = above, c = upper left
 189         int p = a + b - c;        // initial estimate
 190         int pa = ABS(p - a);      // distances to a, b, c
 191         int pb = ABS(p - b);
 192         int pc = ABS(p - c);
 193         // return nearest of a,b,c,
 194         // breaking ties in order a,b,c.
 195         if (pa <= pb && pa <= pc)
 196                 return a;
 197         else if (pb <= pc)
 198                 return b;
 199         else return c;
 200 }
 201
 202 void applyfilter(int mode, U8*src, U8*old, U8*dest, int width)
 203 {
 204     int x;
 205     unsigned char lastr=0;
 206     unsigned char lastg=0;
 207     unsigned char lastb=0;
 208     unsigned char upperlastr=0;
 209     unsigned char upperlastg=0;
 210     unsigned char upperlastb=0;
 211
 212     if(mode==0) {
 213         for(x=0;x<width;x++) {
 214             dest[0] = 255;
 215             dest[1] = src[0];
 216             dest[2] = src[1];
 217             dest[3] = src[2];
 218             dest+=4;
 219             src+=3;
 220         }
 221     }
 222     else if(mode==1) {
 223         for(x=0;x<width;x++) {
 224             dest[0] = 255;
 225             dest[1] = src[0]+lastr;
 226             dest[2] = src[1]+lastg;
 227             dest[3] = src[2]+lastb;
 228             lastr = dest[1];
 229             lastg = dest[2];
 230             lastb = dest[3];
 231             dest+=4;
 232             src+=3;
 233         }
 234     }
 235     else if(mode==2) {
 236         for(x=0;x<width;x++) {
 237             dest[0] = 255;
 238             dest[1] = src[0]+old[1];
 239             dest[2] = src[1]+old[2];
 240             dest[3] = src[2]+old[3];
 241             dest+=4;
 242             old+=4;
 243             src+=3;
 244         }
 245     }
 246     else if(mode==3) {
 247         for(x=0;x<width;x++) {
 248             dest[0] = 255;
 249             dest[1] = src[0]+(old[1]+lastr)/2;
 250             dest[2] = src[1]+(old[2]+lastg)/2;
 251             dest[3] = src[2]+(old[3]+lastb)/2;
 252             lastr = dest[1];
 253             lastg = dest[2];
 254             lastb = dest[3];
 255             dest+=4;
 256             old+=4;
 257             src+=3;
 258         }
 259     }
 260     else if(mode==4) {
 261         for(x=0;x<width;x++) {
 262             dest[0] = 255;
 263             dest[1] = src[0]+PaethPredictor(lastr,old[1],upperlastr);
 264             dest[2] = src[1]+PaethPredictor(lastg,old[2],upperlastg);
 265             dest[3] = src[2]+PaethPredictor(lastb,old[3],upperlastb);
 266             lastr = dest[1];
 267             lastg = dest[2];
 268             lastb = dest[3];
 269             upperlastr = old[1];
 270             upperlastg = old[2];
 271             upperlastb = old[3];
 272             dest+=4;
 273             old+=4;
 274             src+=3;
 275         }
 276     }
 277
 278 }
 279
 280 void applyfilter1(int mode, U8*src, U8*old, U8*dest, int width)
 281 {
 282     int x;
 283     unsigned char last=0;
 284     unsigned char upperlast=0;
 285
 286     if(mode==0) {
 287         for(x=0;x<width;x++) {
 288             *dest = *src;
 289             dest++;
 290             src++;
 291         }
 292     }
 293     else if(mode==1) {
 294         for(x=0;x<width;x++) {
 295             *dest = *src+last;
 296             last = *dest;
 297             dest++;
 298             src++;
 299         }
 300     }
 301     else if(mode==2) {
 302         for(x=0;x<width;x++) {
 303             *dest = *src+*old;
 304             dest++;
 305             old++;
 306             src++;
 307         }
 308     }
 309     else if(mode==3) {
 310         for(x=0;x<width;x++) {
 311             *dest = *src+(*old+last)/2;
 312             dest++;
 313             old++;
 314             src++;
 315         }
 316     }
 317     else if(mode==4) {
 318         for(x=0;x<width;x++) {
 319             *dest = *src+PaethPredictor(last,*old,upperlast);
 320             last = *dest;
 321             upperlast = *old;
 322             dest++;
 323             old++;
 324             src++;
 325         }
 326     }
 327
 328 }
 329
 330 TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
 331 {
 332     SHAPE *s;
 333     SRECT r;
 334     MATRIX m;
 335     int fs;
 336
 337     char tagid[4];
 338     int len;
 339     U8*data;
 340     U8*imagedata;
 341     U8*zimagedata=0;
 342     unsigned long int imagedatalen;
 343     unsigned long int zimagedatalen=0;
 344     U8*palette = 0;
 345     int palettelen = 0;
 346     U8*alphapalette = 0;
 347     int alphapalettelen = 0;
 348     struct png_header header;
 349     int bypp;
 350
 351     FILE *fi;
 352     U8 *scanline;
 353
 354     if ((fi = fopen(sname, "rb")) == NULL) {
 355         if (VERBOSE(1))
 356             fprintf(stderr, "Read access failed: %s\n", sname);
 357         return t;
 358     }
 359
 360     if(!png_read_header(fi, &header))
 361         return 0;
 362
 363     if(header.mode == 3) bypp = 1;
 364     else
 365     if(header.mode == 2) bypp = 3;
 366     else
 367         return 0;
 368     imagedatalen = bypp * header.width * header.height + 65536;
 369     imagedata = malloc(imagedatalen);
 370
 371     fseek(fi,8,SEEK_SET);
 372     while(!feof(fi))
 373     {
 374         if(!png_read_chunk(&tagid, &len, &data, fi))
 375             break;
 376         if(!strncmp(tagid, "IEND", 4)) {
 377             break;
 378         }
 379         if(!strncmp(tagid, "PLTE", 4)) {
 380             palette = data;
 381             palettelen = len/3;
 382             data = 0; //don't free data
 383             if(VERBOSE(2))
 384                 printf("%d colors in palette\n", palettelen);
 385         }
 386         if(!strncmp(tagid, "tRNS", 4)) {
 387             if(header.mode == 3) {
 388                 alphapalette = data;
 389                 alphapalettelen = len;
 390                 data = 0; //don't free data
 391                 if(VERBOSE(2))
 392                     printf("found %d alpha colors\n", alphapalettelen);
 393             }
 394         }
 395         if(!strncmp(tagid, "IDAT", 4)) {
 396             if(!zimagedata) {
 397                 zimagedatalen = len;
 398                 zimagedata = malloc(len);
 399                 memcpy(zimagedata,data,len);
 400             } else {
 401                 zimagedata = realloc(zimagedata, zimagedatalen+len);
 402                 memcpy(&zimagedata[zimagedatalen], data, len);
 403                 zimagedatalen += len;
 404             }
 405         }
 406         if(data)
 407             free(data);
 408     }
 409
 410     if(!zimagedata || uncompress(imagedata, &imagedatalen, zimagedata, zimagedatalen) != Z_OK) {
 411         fprintf(stderr, "Couldn't uncompress %s!\n", sname);
 412         if(zimagedata)
 413             free(zimagedata);
 414         return 0;
 415     }
 416     free(zimagedata);
 417
 418     if(alphapalette)
 419         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS2);
 420     else
 421         t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
 422
 423     swf_SetU16(t, id);          // id
 424     if(header.mode == 2) {
 425         U8*data2 = malloc(header.width*header.height*4);
 426         int i,s=0;
 427         int x,y;
 428         int pos=0;
 429         /* 24->32 bit conversion */
 430         for(y=0;y<header.height;y++) {
 431             int mode = imagedata[pos++]; //filter mode
 432             U8*src;
 433             U8*dest;
 434             U8*old;
 435             dest = &data2[(y*header.width)*4];
 436
 437             if(header.bpp == 8)
 438             {
 439                 /* one byte per pixel */
 440                 src = &imagedata[pos];
 441                 pos+=header.width*3;
 442             } else {
 443                 /* not implemented yet */
 444                 exit(1);
 445             }
 446
 447             if(!y) {
 448                 memset(data2,0,header.width*4);
 449                 old = &data2[y*header.width*4];
 450             } else {
 451                 old = &data2[(y-1)*header.width*4];
 452             }
 453             applyfilter(mode, src, old, dest, header.width);
 454         }
 455         swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
 456         free(data2);
 457     }
 458     else {
 459         RGBA*rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
 460         int swf_width = BYTES_PER_SCANLINE(header.width);
 461         U8*data2 = malloc(swf_width*header.height);
 462         U8*tmpline = malloc(header.width);
 463         int i,x,y;
 464         int pos=0;
 465         if(!palette) {
 466             fprintf(stderr, "Error: No palette found!\n");
 467             exit(1);
 468         }
 469         /* 24->32 bit conversion */
 470         for(i=0;i<palettelen;i++) {
 471             rgba[i].r = palette[i*3+0];
 472             rgba[i].g = palette[i*3+1];
 473             rgba[i].b = palette[i*3+2];
 474             if(alphapalette && i<alphapalettelen) {
 475                 rgba[i].a = alphapalette[i];
 476                 if(alphapalette[i] == 0) {
 477                     /* if the color is fully transparent, it doesn't matter
 478                        what it's rgb values are. furthermore, all Flash
 479                        players up to Flash 5 can't deal with anything beyond
 480                        one transparent color with value (00,00,00,00). */
 481                     rgba[i].r = rgba[i].g = rgba[i].b = 0;
 482                 }
 483             } else {
 484                 rgba[i].a = 255;
 485             }
 486         }
 487
 488         for(y=0;y<header.height;y++) {
 489             int mode = imagedata[pos++]; //filter mode
 490             U8*old;
 491             U8*dest = &data2[y*swf_width];
 492             U8*src;
 493             src = &imagedata[pos];
 494             if(header.bpp == 8) {
 495                 pos+=header.width;
 496             } else {
 497                 int x,s=0;
 498                 int bitpos = 0;
 499                 U32 v = (1<<header.bpp)-1;
 500                 for(x=0;x<header.width;x++) {
 501                     U32 r = src[s/8]<<8 |
 502                             src[s/8+1];
 503                     int t;
 504                     tmpline[x] = (r>>(16-header.bpp-(s&7)))&v;
 505                     s+=header.bpp;
 506                 }
 507                 src = tmpline;
 508                 pos+=(header.width*header.bpp+7)/8;
 509             }
 510
 511             if(!y) {
 512                 memset(data2,0,swf_width);
 513                 old = &data2[y*swf_width];
 514             } else {
 515                 old = &data2[(y-1)*swf_width];
 516             }
 517             applyfilter1(mode, src, old, dest, header.width);
 518         }
 519         swf_SetLosslessBitsIndexed(t, header.width, header.height, data2, rgba, palettelen);
 520         free(tmpline);
 521         free(rgba);
 522         free(data2);
 523     }
 524
 525     t = swf_InsertTag(t, ST_DEFINESHAPE);
 526
 527     swf_ShapeNew(&s);
 528     swf_GetMatrix(NULL, &m);
 529     m.sx = 20 * 0x10000;
 530     m.sy = 20 * 0x10000;
 531     fs = swf_ShapeAddBitmapFillStyle(s, &m, id, 0);
 532
 533     swf_SetU16(t, id + 1);      // id
 534
 535     r.xmin = r.ymin = 0;
 536     r.xmax = header.width * 20;
 537     r.ymax = header.height * 20;
 538     swf_SetRect(t, &r);
 539
 540     swf_SetShapeHeader(t, s);
 541
 542     swf_ShapeSetAll(t, s, 0, 0, 0, fs, 0);
 543     swf_ShapeSetLine(t, s, r.xmax, 0);
 544     swf_ShapeSetLine(t, s, 0, r.ymax);
 545     swf_ShapeSetLine(t, s, -r.xmax, 0);
 546     swf_ShapeSetLine(t, s, 0, -r.ymax);
 547
 548     swf_ShapeSetEnd(t);
 549
 550     t = swf_InsertTag(t, ST_REMOVEOBJECT2);
 551     swf_SetU16(t, 1);           // depth
 552
 553     t = swf_InsertTag(t, ST_PLACEOBJECT2);
 554
 555     swf_GetMatrix(NULL, &m);
 556     m.tx = (swf->movieSize.xmax - (int) header.width * 20) / 2;
 557     m.ty = (swf->movieSize.ymax - (int) header.height * 20) / 2;
 558     swf_ObjectPlace(t, id + 1, 1, &m, NULL, NULL);
 559
 560     t = swf_InsertTag(t, ST_SHOWFRAME);
 561
 562     fclose(fi);
 563
 564     return t;
 565 }
 566
 567
 568 int CheckInputFile(char *fname, char **realname)
 569 {
 570     FILE *fi;
 571     char *s = malloc(strlen(fname) + 5);
 572     struct png_header head;
 573
 574     if (!s)
 575         exit(2);
 576     (*realname) = s;
 577     strcpy(s, fname);
 578
 579     // Check whether file exists (with typical extensions)
 580
 581     if ((fi = fopen(s, "rb")) == NULL) {
 582         sprintf(s, "%s.png", fname);
 583         if ((fi = fopen(s, "rb")) == NULL) {
 584             sprintf(s, "%s.PNG", fname);
 585             if ((fi = fopen(s, "rb")) == NULL) {
 586                 sprintf(s, "%s.Png", fname);
 587                 if ((fi = fopen(s, "rb")) == NULL)
 588                     fprintf(stderr, "Couldn't open %s!\n", fname);
 589                     return -1;
 590             }
 591         }
 592     }
 593
 594     if(!png_read_header(fi, &head)) {
 595         fprintf(stderr, "%s is not a PNG file!\n", fname);
 596         return -1;
 597     }
 598
 599     if (global.max_image_width < head.width)
 600         global.max_image_width = head.width;
 601     if (global.max_image_height < head.height)
 602         global.max_image_height = head.height;
 603
 604     fclose(fi);
 605
 606     return 0;
 607 }
 608
 609 int args_callback_option(char *arg, char *val)
 610 {
 611     int res = 0;
 612     if (arg[1])
 613         res = -1;
 614     else
 615         switch (arg[0]) {
 616         case 'r':
 617             if (val)
 618                 global.framerate = atoi(val);
 619             if ((global.framerate < 1) ||(global.framerate > 5000)) {
 620                 if (VERBOSE(1))
 621                     fprintf(stderr,
 622                             "Error: You must specify a valid framerate between 1 and 10000.\n");
 623                 exit(1);
 624             }
 625             res = 1;
 626             break;
 627
 628         case 'o':
 629             if (val)
 630                 global.outfile = val;
 631             res = 1;
 632             break;
 633
 634         case 'v':
 635             if (val)
 636                 global.verbose = atoi(val);
 637             res = 1;
 638             break;
 639
 640         case 'X':
 641             if (val)
 642                 global.force_width = atoi(val);
 643             res = 1;
 644             break;
 645
 646         case 'Y':
 647             if (val)
 648                 global.force_height = atoi(val);
 649             res = 1;
 650             break;
 651
 652         case 'V':
 653             printf("png2swf - part of %s %s\n", PACKAGE, VERSION);
 654             exit(0);
 655
 656         default:
 657             res = -1;
 658             break;
 659         }
 660
 661     if (res < 0) {
 662         if (VERBOSE(1))
 663             fprintf(stderr, "Unknown option: -%s\n", arg);
 664         exit(1);
 665         return 0;
 666     }
 667     return res;
 668 }
 669
 670 struct options_t options[] =
 671 {
 672 {"o", "output"},
 673 {"r", "rate"},
 674 {"v", "verbose"},
 675 {"X", "width"},
 676 {"Y", "height"},
 677 {"V", "version"},
 678 };
 679
 680 int args_callback_longoption(char *name, char *val)
 681 {
 682     return args_long2shortoption(options, name, val);
 683 }
 684
 685 int args_callback_command(char *arg, char *next)        // actually used as filename
 686 {
 687     char *s;
 688     if (CheckInputFile(arg, &s) < 0) {
 689         if (VERBOSE(1))
 690             fprintf(stderr, "Error opening input file: %s\n", arg);
 691         free(s);
 692     } else {
 693         image[global.nfiles].filename = s;
 694         global.nfiles++;
 695         if (global.nfiles >= MAX_INPUT_FILES) {
 696             if (VERBOSE(1))
 697                 fprintf(stderr, "Error: Too many input files.\n");
 698             exit(1);
 699         }
 700     }
 701     return 0;
 702 }
 703
 704 void args_callback_usage(char *name)
 705 {
 706     printf("Usage: %s  [-options [value]] imagefiles[.png] [...]\n", name);
 707     printf("-r framerate          (rate) Set movie framerate (100/sec)\n");
 708     printf("-o outputfile         (output) Set name for SWF output file\n");
 709     printf("-X pixel              (width) Force movie width to pixel (default: autodetect)\n");
 710     printf("-Y pixel              (height) Force movie height to pixel (default: autodetect)\n");
 711     printf("-v level              (verbose) Set verbose level (0=quiet, 1=default, 2=debug)\n");
 712     printf("-V                    (version) Print version information and exit\n");
 713 }
 714
 715
 716 int main(int argc, char **argv)
 717 {
 718     SWF swf;
 719     TAG *t;
 720
 721     memset(&global, 0x00, sizeof(global));
 722
 723     global.framerate = 100;
 724     global.verbose = 1;
 725
 726     processargs(argc, argv);
 727
 728     if(global.nfiles<=0)
 729         return 1;
 730
 731     if (VERBOSE(2))
 732         fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
 733
 734     t = MovieStart(&swf, global.framerate,
 735                    global.force_width ? global.force_width : global.
 736                    max_image_width,
 737                    global.force_height ? global.force_height : global.
 738                    max_image_height);
 739
 740     {
 741         int i;
 742         for (i = 0; i < global.nfiles; i++) {
 743             if (VERBOSE(3))
 744                 fprintf(stderr, "[%03i] %s\n", i,
 745                         image[i].filename);
 746             t = MovieAddFrame(&swf, t, image[i].filename, (i * 2) + 1);
 747             free(image[i].filename);
 748         }
 749     }
 750
 751     MovieFinish(&swf, t, global.outfile);
 752
 753     return 0;
 754 }