src/swfrender.c

   1 #include "../config.h"
   2 #include <stdio.h>
   3 #include <stdarg.h>
   4 #include <assert.h>
   5 #include <unistd.h>
   6 #include <fcntl.h>
   7 #include "../lib/rfxswf.h"
   8 #include "../lib/png.h"
   9 #include "../lib/args.h"
  10 #include "../lib/gfxsource.h"
  11 #include "../lib/readers/swf.h"
  12 #include "../lib/devices/render.h"
  13 #include "../lib/devices/rescale.h"
  14
  15 static struct options_t options[] = {
  16 {"h", "help"},
  17 {"o", "output"},
  18 {"p", "pages"},
  19 {"l", "legacy"},
  20 {"V", "version"},
  21 {"X", "width"},
  22 {"Y", "height"},
  23 {0,0}
  24 };
  25
  26 static int ng = 1;
  27 static char*filename = 0;
  28 static char*outputname = "output.png";
  29 static int quantize = 0;
  30 static char*pagerange = 0;
  31
  32 static int width = 0;
  33 static int height = 0;
  34
  35 typedef struct _parameter {
  36     const char*name;
  37     const char*value;
  38     struct _parameter*next;
  39 } parameter_t;
  40
  41 parameter_t*params = 0;
  42
  43 int args_callback_option(char*name,char*val)
  44 {
  45     if(!strcmp(name, "V")) {
  46         printf("swfrender - part of %s %s\n", PACKAGE, VERSION);
  47         exit(0);
  48     } else if(!strcmp(name, "o")) {
  49         outputname = strdup(val);
  50         return 1;
  51     } else if(!strcmp(name, "l")) {
  52         ng = 0;
  53         return 0;
  54     } else if(!strcmp(name, "q")) {
  55         quantize = 1;
  56         return 0;
  57     } else if(!strcmp(name, "p")) {
  58         pagerange = val;
  59         return 1;
  60     } else if(!strcmp(name, "s")) {
  61         char*s = strdup(val);
  62         char*c = strchr(s, '=');
  63         parameter_t*p = malloc(sizeof(parameter_t));
  64         p->next = params;
  65         params = p;
  66         if(c && *c && c[1])  {
  67             *c = 0;
  68             c++;
  69             p->name = s;
  70             p->value = s;
  71         } else {
  72             p->name = s;
  73             p->value = "1";
  74         }
  75         return 1;
  76     } else if(!strcmp(name, "X")) {
  77         width = atoi(val);
  78         return 1;
  79     } else if(!strcmp(name, "Y")) {
  80         height = atoi(val);
  81         return 1;
  82     } else {
  83         printf("Unknown option: -%s\n", name);
  84         exit(1);
  85     }
  86
  87     return 0;
  88 }
  89 int args_callback_longoption(char*name,char*val)
  90 {
  91     return args_long2shortoption(options, name, val);
  92 }
  93 void args_callback_usage(char *name)
  94 {
  95     printf("\n");
  96     printf("Usage: %s file.swf [-o output.png]\n", name);
  97     printf("\n");
  98     printf("-h , --help                    Print short help message and exit\n");
  99     printf("-l , --legacy                  Use old rendering framework\n");
 100     printf("-o , --output                  Output file (default: output.png)\n");
 101     printf("\n");
 102 }
 103 int args_callback_command(char*name,char*val)
 104 {
 105     if(filename) {
 106         fprintf(stderr, "Only one file allowed. You supplied at least two. (%s and %s)\n",
 107                  filename, name);
 108     }
 109     filename = name;
 110     return 0;
 111 }
 112
 113
 114
 115 int main(int argn, char*argv[])
 116 {
 117     SWF swf;
 118     int fi;
 119
 120     processargs(argn, argv);
 121
 122     if(!filename) {
 123         fprintf(stderr, "You must supply a filename.\n");
 124         return 1;
 125     }
 126
 127     if(!ng) {
 128         fi = open(filename, O_RDONLY|O_BINARY);
 129         if (fi<=0) {
 130             fprintf(stderr,"Couldn't open %s\n", filename);
 131             perror(filename);
 132             exit(1);
 133         }
 134         if(swf_ReadSWF(fi,&swf)<0) {
 135             fprintf(stderr,"%s is not a valid SWF file or contains errors.\n",argv[1]);
 136             close(fi);
 137         }
 138         assert(swf.movieSize.xmax > swf.movieSize.xmin && swf.movieSize.ymax > swf.movieSize.ymin);
 139         RENDERBUF buf;
 140         swf_Render_Init(&buf, 0,0, (swf.movieSize.xmax - swf.movieSize.xmin) / 20,
 141                                    (swf.movieSize.ymax - swf.movieSize.ymin) / 20, 2, 1);
 142         swf_RenderSWF(&buf, &swf);
 143         RGBA* img = swf_Render(&buf);
 144         if(quantize)
 145             png_write_palette_based_2(outputname, (unsigned char*)img, buf.width, buf.height);
 146         else
 147             png_write(outputname, (unsigned char*)img, buf.width, buf.height);
 148         swf_Render_Delete(&buf);
 149     } else {
 150         parameter_t*p;
 151
 152         gfxsource_t*src = gfxsource_swf_create();
 153         for(p=params;p;p=p->next) {
 154             src->setparameter(src, p->name, p->value);
 155         }
 156         gfxdocument_t*doc = src->open(src, filename);
 157         for(p=params;p;p=p->next) {
 158             doc->setparameter(doc, p->name, p->value);
 159         }
 160         if(!doc) {
 161             fprintf(stderr,"Couldn't open %s\n", filename);
 162             exit(1);
 163         }
 164         gfxdevice_t dev2,*dev=&dev2;
 165         gfxdevice_render_init(dev);
 166         dev->setparameter(dev, "antialise", "4");
 167         if(quantize) {
 168             dev->setparameter(dev, "palette", "1");
 169         }
 170         if(width || height) {
 171             dev = gfxdevice_rescale_new(dev, width, height, 0);
 172         }
 173         for(p=params;p;p=p->next) {
 174             dev->setparameter(dev, p->name, p->value);
 175         }
 176
 177         int t;
 178         for(t=1;t<=doc->num_pages;t++) {
 179             if(!is_in_range(t, pagerange))
 180                 continue;
 181             gfxpage_t* page = doc->getpage(doc, t);
 182             dev->startpage(dev, page->width, page->height);
 183             page->render(page, dev);
 184             dev->endpage(dev);
 185             page->destroy(page);
 186             break;
 187         }
 188         gfxresult_t*result = dev->finish(dev);
 189         if(result) {
 190             if(result->save(result, outputname) < 0) {
 191                 exit(1);
 192             }
 193             result->destroy(result);
 194         }
 195         doc->destroy(doc);
 196     }
 197     return 0;
 198 }
 199