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     printf("id: %.4s len: %d\n", head, len);
  91     if(destlen) *destlen = len;
  92     if(destdata) {
  93         if(len)
  94             *destdata = malloc(len);
  95         else
  96             *destdata = 0;
  97         if(!fread(*destdata, len, 1, fi)) {
  98             *destdata = 0;
  99             if(destlen) *destlen=0;
 100             return 0;
 101         }
 102         fseek(fi, 4, SEEK_CUR);
 103
 104     } else {
 105         fseek(fi, len+4, SEEK_CUR);
 106     }
 107     return 1;
 108 }
 109
 110 unsigned int png_get_dword(FILE*fi)
 111 {
 112     unsigned int a;
 113     fread(&a,4,1,fi);
 114     return REVERSESWAP32(a);
 115 }
 116
 117 struct png_header
 118 {
 119     int width;
 120     int height;
 121     int bpp;
 122     int mode;
 123 };
 124
 125 int png_read_header(FILE*fi, struct png_header*header)
 126 {
 127     char id[4];
 128     int len;
 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         printf("%c%c%c%c %d\n", id[0],id[1],id[2],id[3],len);
 139         if(!strncasecmp(id, "IHDR", 4)) {
 140             char a,b,c,f,i;
 141             if(len < 8) exit(1);
 142             header->width = REVERSESWAP32(*(U32*)&data[0]);
 143             header->height = REVERSESWAP32(*(U32*)&data[4]);
 144             a = data[8];      // should be 8
 145             b = data[9];      // should be 3(indexed) or 2(rgb)
 146
 147             c = data[10];     // compression mode (0)
 148             f = data[11];     // filter mode (0)
 149             i = data[12];     // interlace mode (0)
 150             printf("%08xx%08x %d %d %d %d %d\n",header->width, header->height, a,b,c,f,i);
 151             header->bpp = a;
 152             header->mode = b;
 153             return 1;
 154         } else {
 155             fseek(fi, len, SEEK_CUR);
 156         }
 157         free(data);
 158     }
 159     return 0;
 160 }
 161
 162 TAG *MovieAddFrame(SWF * swf, TAG * t, char *sname, int id)
 163 {
 164     SHAPE *s;
 165     SRECT r;
 166     MATRIX m;
 167     int fs;
 168
 169     char tagid[4];
 170     int len;
 171     U8*data;
 172     U8*imagedata;
 173     unsigned long int imagedatalen;
 174     U8*palette;
 175     int palettelen;
 176     struct png_header header;
 177     int bypp;
 178
 179     FILE *fi;
 180     U8 *scanline;
 181
 182     if ((fi = fopen(sname, "rb")) == NULL) {
 183         if (VERBOSE(1))
 184             fprintf(stderr, "Read access failed: %s\n", sname);
 185         return t;
 186     }
 187
 188     if(!png_read_header(fi, &header))
 189         return 0;
 190
 191     if(header.mode == 3) bypp = 1;
 192     else
 193     if(header.mode == 2) bypp = 3;
 194     else
 195         return 0;
 196     imagedatalen = bypp * header.width * header.height + 65536;
 197     imagedata = malloc(imagedatalen);
 198
 199     fseek(fi,8,SEEK_SET);
 200     while(!feof(fi))
 201     {
 202         if(!png_read_chunk(&tagid, &len, &data, fi))
 203             break;
 204         if(!strncmp(tagid, "IEND", 4)) {
 205             break;
 206         }
 207         if(!strncmp(tagid, "PLTE", 4)) {
 208             palette = data;
 209             palettelen = len/3;
 210             data = 0;
 211             printf("%d palette\n", len);
 212         }
 213         if(!strncmp(tagid, "IDAT", 4)) {
 214             if(uncompress(imagedata, &imagedatalen, data, len) != Z_OK) {
 215                 fprintf(stderr, "Couldn't uncompress %s!\n", sname);
 216                 return 0;
 217             }
 218             printf("IDAT %d -> %d\n", len, imagedatalen);
 219         }
 220         if(data)
 221             free(data);
 222     }
 223
 224     t = swf_InsertTag(t, ST_DEFINEBITSLOSSLESS);
 225     swf_SetU16(t, id);          // id
 226     if(header.mode == 2) {
 227         U8*data2 = malloc(header.width*header.height*4);
 228         int i,s=0;
 229         int x,y;
 230         int pos=0;
 231         /* 24->32 bit conversion */
 232         for(y=0;y<header.height;y++) {
 233             pos++; // filter mode
 234             for(x=0;x<header.width;x++) {
 235                 U8*src  = &imagedata[pos];
 236                 U8*dest = &data2[(y*header.width+x)*4];
 237                 dest[0] = 255;
 238                 dest[1] = src[0];
 239                 dest[2] = src[1];
 240                 dest[3] = src[2];
 241                 pos+=3;
 242             }
 243         }
 244         swf_SetLosslessBits(t, header.width, header.height, data2, BMF_32BIT);
 245         free(data2);
 246     }
 247     else {
 248         RGBA*rgba = (RGBA*)malloc(palettelen*sizeof(RGBA));
 249         U8*data2 = malloc((header.width+4)*header.height);
 250         int i,x,y;
 251         int pos=0;
 252         /* 24->32 bit conversion */
 253         for(i=0;i<palettelen;i++) {
 254             rgba[i].r = palette[i*3+0];
 255             rgba[i].g = palette[i*3+1];
 256             rgba[i].b = palette[i*3+2];
 257             rgba[i].a = 255;
 258         }
 259         for(y=0;y<header.height;y++) {
 260             pos++; //filter mode
 261             for(x=0;x<header.width;x++) {
 262                 data2[y*header.width+x] =
 263                     imagedata[pos++];
 264             }
 265         }
 266         swf_SetLosslessBitsIndexed(t, header.width, header.height, data2, rgba, palettelen);
 267         free(rgba);
 268         free(data2);
 269     }
 270
 271     t = swf_InsertTag(t, ST_DEFINESHAPE);
 272
 273     swf_ShapeNew(&s);
 274     swf_GetMatrix(NULL, &m);
 275     m.sx = 20 * 0x10000;
 276     m.sy = 20 * 0x10000;
 277     fs = swf_ShapeAddBitmapFillStyle(s, &m, id, 0);
 278
 279     swf_SetU16(t, id + 1);      // id
 280
 281     r.xmin = r.ymin = 0;
 282     r.xmax = header.width * 20;
 283     r.ymax = header.height * 20;
 284     swf_SetRect(t, &r);
 285
 286     swf_SetShapeHeader(t, s);
 287
 288     swf_ShapeSetAll(t, s, 0, 0, 0, fs, 0);
 289     swf_ShapeSetLine(t, s, r.xmax, 0);
 290     swf_ShapeSetLine(t, s, 0, r.ymax);
 291     swf_ShapeSetLine(t, s, -r.xmax, 0);
 292     swf_ShapeSetLine(t, s, 0, -r.ymax);
 293
 294     swf_ShapeSetEnd(t);
 295
 296     t = swf_InsertTag(t, ST_REMOVEOBJECT2);
 297     swf_SetU16(t, 1);           // depth
 298
 299     t = swf_InsertTag(t, ST_PLACEOBJECT2);
 300
 301     swf_GetMatrix(NULL, &m);
 302     m.tx = (swf->movieSize.xmax - (int) header.width * 20) / 2;
 303     m.ty = (swf->movieSize.ymax - (int) header.height * 20) / 2;
 304     swf_ObjectPlace(t, id + 1, 1, &m, NULL, NULL);
 305
 306     t = swf_InsertTag(t, ST_SHOWFRAME);
 307
 308     fclose(fi);
 309
 310     return t;
 311 }
 312
 313
 314 int CheckInputFile(char *fname, char **realname)
 315 {
 316     FILE *fi;
 317     char *s = malloc(strlen(fname) + 5);
 318     struct png_header head;
 319
 320     if (!s)
 321         exit(2);
 322     (*realname) = s;
 323     strcpy(s, fname);
 324
 325     // Check whether file exists (with typical extensions)
 326
 327     if ((fi = fopen(s, "rb")) == NULL) {
 328         sprintf(s, "%s.png", fname);
 329         if ((fi = fopen(s, "rb")) == NULL) {
 330             sprintf(s, "%s.PNG", fname);
 331             if ((fi = fopen(s, "rb")) == NULL) {
 332                 sprintf(s, "%s.Png", fname);
 333                 if ((fi = fopen(s, "rb")) == NULL)
 334                     fprintf(stderr, "Couldn't open %s!\n", fname);
 335                     return -1;
 336             }
 337         }
 338     }
 339
 340     if(!png_read_header(fi, &head)) {
 341         fprintf(stderr, "%s is not a PNG file!\n", fname);
 342         return -1;
 343     }
 344     if(head.mode!=2 && head.mode!=3) {
 345         fprintf(stderr, "%s has unsupported mode %d\n", head.mode);
 346         return -1;
 347     }
 348     if(head.bpp!=8) {
 349         fprintf(stderr, "%s has unsupported bpp %d\n", head.bpp);
 350         return -1;
 351     }
 352
 353     if (global.max_image_width < head.width)
 354         global.max_image_width = head.width;
 355     if (global.max_image_height < head.height)
 356         global.max_image_height = head.height;
 357
 358     fclose(fi);
 359
 360     return 0;
 361 }
 362
 363 int args_callback_option(char *arg, char *val)
 364 {
 365     int res = 0;
 366     if (arg[1])
 367         res = -1;
 368     else
 369         switch (arg[0]) {
 370         case 'r':
 371             if (val)
 372                 global.framerate = atoi(val);
 373             if ((global.framerate < 1) ||(global.framerate > 5000)) {
 374                 if (VERBOSE(1))
 375                     fprintf(stderr,
 376                             "Error: You must specify a valid framerate between 1 and 10000.\n");
 377                 exit(1);
 378             }
 379             res = 1;
 380             break;
 381
 382         case 'o':
 383             if (val)
 384                 global.outfile = val;
 385             res = 1;
 386             break;
 387
 388         case 'v':
 389             if (val)
 390                 global.verbose = atoi(val);
 391             res = 1;
 392             break;
 393
 394         case 'X':
 395             if (val)
 396                 global.force_width = atoi(val);
 397             res = 1;
 398             break;
 399
 400         case 'Y':
 401             if (val)
 402                 global.force_height = atoi(val);
 403             res = 1;
 404             break;
 405
 406         case 'V':
 407             printf("png2swf - part of %s %s\n", PACKAGE, VERSION);
 408             exit(0);
 409
 410         default:
 411             res = -1;
 412             break;
 413         }
 414
 415     if (res < 0) {
 416         if (VERBOSE(1))
 417             fprintf(stderr, "Unknown option: -%s\n", arg);
 418         exit(1);
 419         return 0;
 420     }
 421     return res;
 422 }
 423
 424 struct options_t options[] =
 425 {
 426 {"o", "output"},
 427 {"r", "rate"},
 428 {"v", "verbose"},
 429 {"X", "width"},
 430 {"Y", "height"},
 431 {"V", "version"},
 432 };
 433
 434 int args_callback_longoption(char *name, char *val)
 435 {
 436     return args_long2shortoption(options, name, val);
 437 }
 438
 439 int args_callback_command(char *arg, char *next)        // actually used as filename
 440 {
 441     char *s;
 442     if (CheckInputFile(arg, &s) < 0) {
 443         if (VERBOSE(1))
 444             fprintf(stderr, "Error opening input file: %s\n", arg);
 445         free(s);
 446     } else {
 447         image[global.nfiles].filename = s;
 448         global.nfiles++;
 449         if (global.nfiles >= MAX_INPUT_FILES) {
 450             if (VERBOSE(1))
 451                 fprintf(stderr, "Error: Too many input files.\n");
 452             exit(1);
 453         }
 454     }
 455     return 0;
 456 }
 457
 458 void args_callback_usage(char *name)
 459 {
 460     printf("Usage: %s  [-options [value]] imagefiles[.png] [...]\n", name);
 461     printf("-r framerate          (rate) Set movie framerate (100/sec)\n");
 462     printf("-o outputfile         (output) Set name for SWF output file\n");
 463     printf("-X pixel              (width) Force movie width to pixel (default: autodetect)\n");
 464     printf("-Y pixel              (height) Force movie height to pixel (default: autodetect)\n");
 465     printf("-v level              (verbose) Set verbose level (0=quiet, 1=default, 2=debug)\n");
 466     printf("-V                    (version) Print version information and exit\n");
 467     printf("The following options can be set independently for each image: -q -s\n");
 468 }
 469
 470
 471 int main(int argc, char **argv)
 472 {
 473     SWF swf;
 474     TAG *t;
 475
 476     memset(&global, 0x00, sizeof(global));
 477
 478     global.framerate = 100;
 479     global.verbose = 1;
 480
 481     processargs(argc, argv);
 482
 483     if (VERBOSE(2))
 484         fprintf(stderr, "Processing %i file(s)...\n", global.nfiles);
 485
 486     t = MovieStart(&swf, global.framerate,
 487                    global.force_width ? global.force_width : global.
 488                    max_image_width,
 489                    global.force_height ? global.force_height : global.
 490                    max_image_height);
 491
 492     {
 493         int i;
 494         for (i = 0; i < global.nfiles; i++) {
 495             if (VERBOSE(3))
 496                 fprintf(stderr, "[%03i] %s\n", i,
 497                         image[i].filename);
 498             t = MovieAddFrame(&swf, t, image[i].filename, (i * 2) + 1);
 499             free(image[i].filename);
 500         }
 501     }
 502
 503     MovieFinish(&swf, t, global.outfile);
 504
 505     return 0;
 506 }