lib/ruby/gfx.c

   1 #include <ruby.h>
   2 #include "../gfxdevice.h"
   3 #include "../gfxsource.h"
   4 #include "../devices/pdf.h"
   5 #include "../readers/swf.h"
   6 #include "../readers/image.h"
   7 #include "../pdf/pdf.h"
   8 #include "../mem.h"
   9 #include "../types.h"
  10
  11 #define RUBY_GFX_VERSION  "0.9.0"
  12
  13 static VALUE GFX;
  14 static VALUE Bitmap, Document, DocumentPage, PDFClass, SWFClass, ImageClass, Device;
  15 static ID id_doc;
  16
  17 typedef struct doc_internal {
  18     gfxsource_t*driver; // filled by alloc
  19     gfxdocument_t*doc;
  20     gfxfontlist_t*list;
  21 } doc_internal_t;
  22
  23 typedef struct page_internal {
  24     gfxpage_t*page;
  25 } page_internal_t;
  26
  27 typedef struct image_internal {
  28     gfximage_t*image;
  29 } image_internal_t;
  30
  31 static gfxsource_t* pdfdriver = 0;
  32 static gfxsource_t* imagedriver = 0;
  33 static gfxsource_t* swfdriver = 0;
  34
  35 #define Get_Doc(doc,cls) doc_internal_t*doc=0;Data_Get_Struct(cls, doc_internal_t, doc);
  36 #define Get_Page(page,cls) page_internal_t*page=0;Data_Get_Struct(cls, page_internal_t, page);
  37
  38 static VALUE doc_allocate(VALUE cls, gfxsource_t*driver);
  39 static VALUE page_allocate(VALUE cls);
  40
  41 // ------------------------ documents ---------------------------------------
  42
  43 static VALUE doc_initialize(VALUE cls, VALUE _filename)
  44 {
  45     Check_Type(_filename, T_STRING);
  46     Get_Doc(doc,cls);
  47     const char*filename = StringValuePtr(_filename);
  48     doc->doc = pdfdriver->open(pdfdriver, filename);
  49     return cls;
  50 }
  51
  52 static VALUE doc_num_pages(VALUE cls)
  53 {
  54     Get_Doc(doc,cls)
  55     return INT2FIX(doc->doc->num_pages);
  56 }
  57
  58
  59 static VALUE doc_get_page(VALUE cls, VALUE _nr)
  60 {
  61     Check_Type(_nr, T_FIXNUM);
  62     int nr = FIX2INT(_nr);
  63     Get_Doc(doc,cls);
  64
  65     VALUE v = page_allocate(DocumentPage);
  66     Get_Page(page,v)
  67     page->page = doc->doc->getpage(doc->doc, nr);
  68     if(!page->page) {
  69         rb_raise(rb_eArgError, "No page %d in document", nr);
  70         return;
  71     }
  72     return v;
  73 }
  74
  75 static VALUE doc_each_page(VALUE cls)
  76 {
  77     Get_Doc(doc,cls);
  78     int t;
  79     for(t=1;t<=doc->doc->num_pages;t++) {
  80         VALUE v = page_allocate(DocumentPage);
  81         Get_Page(page,v)
  82         page->page = doc->doc->getpage(doc->doc, t);
  83         rb_yield(v);
  84     }
  85     return cls;
  86 }
  87
  88 static void doc_mark(doc_internal_t*doc)
  89 {
  90     // e.g. rb_gc_mark(z->page);
  91 }
  92
  93 static void doc_free(doc_internal_t*doc)
  94 {
  95     doc->doc->destroy(doc->doc);
  96     free(doc);
  97 }
  98
  99 static VALUE doc_allocate(VALUE cls, gfxsource_t*driver)
 100 {
 101     doc_internal_t*doc = 0;
 102     VALUE v = Data_Make_Struct(cls, doc_internal_t, 0, doc_free, doc);
 103     memset(doc, 0, sizeof(doc_internal_t));
 104     doc->driver = driver;
 105     return v;
 106 }
 107
 108 static VALUE pdf_allocate(VALUE cls) {return doc_allocate(cls, pdfdriver);}
 109 static VALUE swf_allocate(VALUE cls) {return doc_allocate(cls, swfdriver);}
 110 static VALUE imgdrv_allocate(VALUE cls) {return doc_allocate(cls, imagedriver);}
 111
 112 // ------------------------ doc pages ---------------------------------------
 113
 114 static void page_free(page_internal_t*page)
 115 {
 116     if(!page) return;
 117     if(page->page) {
 118         page->page->destroy(page->page);
 119         page->page = 0;
 120     }
 121     free(page);
 122 }
 123 static VALUE page_allocate(VALUE cls)
 124 {
 125     page_internal_t*page = 0;
 126     VALUE v = Data_Make_Struct(cls, page_internal_t, 0, page_free, page);
 127     memset(page, 0, sizeof(page_internal_t));
 128     return v;
 129 }
 130 static VALUE page_nr(VALUE cls)
 131 {
 132     Get_Page(page,cls)
 133     return INT2FIX(page->page->nr);
 134 }
 135 static VALUE page_width(VALUE cls)
 136 {
 137     Get_Page(page,cls)
 138     return INT2FIX(page->page->width);
 139 }
 140 static VALUE page_height(VALUE cls)
 141 {
 142     Get_Page(page,cls)
 143     return INT2FIX(page->page->height);
 144 }
 145
 146 // ------------------------ image -------------------------------------------
 147
 148 #define Get_Image(image,cls) image_internal_t*image=0;Data_Get_Struct(cls, image_internal_t, image);
 149
 150 static void image_free(image_internal_t*image)
 151 {
 152     free(image);
 153 }
 154 static VALUE image_allocate(VALUE cls)
 155 {
 156     image_internal_t*image = 0;
 157     VALUE v = Data_Make_Struct(cls, image_internal_t, 0, image_free, image);
 158     memset(image, 0, sizeof(image_internal_t));
 159     return v;
 160 }
 161 static VALUE image_width(VALUE cls)
 162 {
 163     Get_Image(image,cls)
 164     return INT2FIX(image->image->width);
 165 }
 166 static VALUE image_height(VALUE cls)
 167 {
 168     Get_Image(image,cls)
 169     return INT2FIX(image->image->height);
 170 }
 171 static VALUE image_save_jpeg(VALUE cls, VALUE _filename, VALUE quality)
 172 {
 173     Get_Image(image,cls)
 174     Check_Type(_filename, T_STRING);
 175     Check_Type(quality, T_FIXNUM);
 176     const char*filename = StringValuePtr(_filename);
 177     int l = strlen(filename);
 178     char jpeg = 0;
 179     if(l>=4 && !strcmp(&filename[l-4], ".jpg")) jpeg = 1;
 180     if(l>=5 && !strcmp(&filename[l-5], ".jpeg")) jpeg = 1;
 181     if(l>=4 && !strcmp(&filename[l-4], ".JPG")) jpeg = 1;
 182     jpeg_save(image->image->data, image->image->width, image->image->height, FIX2INT(quality), filename);
 183     return cls;
 184 }
 185 VALUE convert_image(gfximage_t*_image)
 186 {
 187     VALUE v = image_allocate(Bitmap);
 188     Get_Image(image,v)
 189     image->image = _image;
 190     return v;
 191 }
 192 void invalidate_image(VALUE v)
 193 {
 194     Get_Image(image,v)
 195     image->image = 0;
 196 }
 197
 198 // ------------------------ gfx device --------------------------------------
 199
 200 static ID id_setparameter = 0;
 201 static ID id_startpage = 0;
 202 static ID id_startclip = 0;
 203 static ID id_endclip = 0;
 204 static ID id_stroke = 0;
 205 static ID id_fill = 0;
 206 static ID id_fillbitmap = 0;
 207 static ID id_fillgradient = 0;
 208 static ID id_addfont = 0;
 209 static ID id_drawchar = 0;
 210 static ID id_drawlink = 0;
 211 static ID id_endpage = 0;
 212 static ID id_geterror = 0;
 213 static ID id_finish = 0;
 214 static ID id_butt = 0;
 215 static ID id_round = 0;
 216 static ID id_square = 0;
 217 static ID id_bevel = 0;
 218 static ID id_miter = 0;
 219 static ID id_move = 0;
 220 static ID id_line = 0;
 221 static ID id_spline = 0;
 222 static ID id_radial = 0;
 223 static ID id_linear = 0;
 224
 225 static VALUE noop(int argc, VALUE *argv, VALUE obj) {return obj;}
 226
 227 #define forward(v,id,args...) rb_respond_to((v), (id))?rb_funcall((v), (id), args):0
 228
 229 VALUE convert_line(gfxline_t*line)
 230 {
 231     int len = 0;
 232     gfxline_t*l = line;
 233     while(l) {l=l->next;len++;}
 234
 235     VALUE*a = malloc(sizeof(VALUE)*len);
 236     int pos = 0;
 237     l = line;
 238     while(l) {
 239         if(l->type == gfx_moveTo) {
 240             a[pos] = rb_ary_new3(3, ID2SYM(id_move), rb_float_new(l->x), rb_float_new(l->y));
 241         } else if(l->type == gfx_lineTo) {
 242             a[pos] = rb_ary_new3(3, ID2SYM(id_line), rb_float_new(l->x), rb_float_new(l->y));
 243         } else {
 244             a[pos] = rb_ary_new3(5, ID2SYM(id_spline), rb_float_new(l->x), rb_float_new(l->y), rb_float_new(l->sx), rb_float_new(l->sy));
 245         }
 246         pos++;
 247         l=l->next;
 248     }
 249     return rb_ary_new4(len, a);
 250 }
 251 VALUE convert_color(gfxcolor_t*color)
 252 {
 253     return rb_ary_new3(4, INT2FIX(color->a), INT2FIX(color->r), INT2FIX(color->g), INT2FIX(color->b));
 254 }
 255 VALUE convert_matrix(gfxmatrix_t*matrix)
 256 {
 257     return rb_ary_new3(3,
 258             rb_ary_new3(2, rb_float_new(matrix->m00), rb_float_new(matrix->m01)),
 259             rb_ary_new3(2, rb_float_new(matrix->m10), rb_float_new(matrix->m11)),
 260             rb_ary_new3(2, rb_float_new(matrix->tx), rb_float_new(matrix->ty)));
 261 }
 262 int rb_setparameter(gfxdevice_t*dev, const char*key, const char*value)
 263 {
 264     VALUE v = (VALUE)(ptroff_t)dev->internal;
 265     VALUE ret = forward(v,id_setparameter,2,rb_tainted_str_new2(key),rb_tainted_str_new2(value));
 266     return 0;
 267 }
 268 void rb_startpage(gfxdevice_t*dev, int width, int height)
 269 {
 270     VALUE v = (VALUE)(ptroff_t)dev->internal;
 271     VALUE ret = forward(v,id_startpage,2,INT2FIX(width),INT2FIX(height));
 272 }
 273 void rb_startclip(gfxdevice_t*dev, gfxline_t*line)
 274 {
 275     VALUE v = (VALUE)(ptroff_t)dev->internal;
 276     VALUE ret = forward(v,id_startclip,1,convert_line(line));
 277 }
 278 void rb_endclip(gfxdevice_t*dev)
 279 {
 280     VALUE v = (VALUE)(ptroff_t)dev->internal;
 281     VALUE ret = forward(v,id_endclip,0);
 282 }
 283 void rb_stroke(gfxdevice_t*dev, gfxline_t*line, gfxcoord_t width, gfxcolor_t*color, gfx_capType cap_style, gfx_joinType joint_style, gfxcoord_t miterLimit)
 284 {
 285     VALUE v = (VALUE)(ptroff_t)dev->internal;
 286
 287     ID cap = 0;
 288     if(cap_style == gfx_capButt) cap = id_butt;
 289     else if(cap_style == gfx_capRound) cap = id_round;
 290     else if(cap_style == gfx_capSquare) cap = id_square;
 291
 292     ID joint = 0;
 293     if(joint_style == gfx_joinRound) joint = id_round;
 294     else if(joint_style == gfx_joinMiter) joint = id_miter;
 295     else if(joint_style == gfx_joinBevel) joint = id_bevel;
 296
 297     forward(v, id_stroke, 6, convert_line(line), rb_float_new(width), convert_color(color), ID2SYM(cap), ID2SYM(joint), rb_float_new(miterLimit));
 298 }
 299 void rb_fill(gfxdevice_t*dev, gfxline_t*line, gfxcolor_t*color)
 300 {
 301     VALUE v = (VALUE)(ptroff_t)dev->internal;
 302     forward(v, id_fill, 2, convert_line(line), convert_color(color));
 303 }
 304 void rb_fillbitmap(gfxdevice_t*dev, gfxline_t*line, gfximage_t*img, gfxmatrix_t*matrix, gfxcxform_t*cxform)
 305 {
 306     VALUE v = (VALUE)(ptroff_t)dev->internal;
 307     VALUE image = convert_image(img);
 308     forward(v, id_fillbitmap, 4, convert_line(line), image, convert_matrix(matrix), Qnil);
 309     invalidate_image(image);
 310 }
 311 void rb_fillgradient(gfxdevice_t*dev, gfxline_t*line, gfxgradient_t*gradient, gfxgradienttype_t type, gfxmatrix_t*matrix)
 312 {
 313     VALUE v = (VALUE)(ptroff_t)dev->internal;
 314     ID typeid = (type == gfxgradient_linear)? id_linear : id_radial;
 315     forward(v, id_fillgradient, 4, convert_line(line), convert_gradient(gradient), ID2SYM(typeid), convert_matrix(matrix));
 316 }
 317 void rb_addfont(gfxdevice_t*dev, gfxfont_t*font)
 318 {
 319     VALUE v = (VALUE)(ptroff_t)dev->internal;
 320     //forward(v, id_addfont, 1, convert_font(font));
 321 }
 322 void rb_drawchar(gfxdevice_t*dev, gfxfont_t*font, int glyphnr, gfxcolor_t*color, gfxmatrix_t*matrix)
 323 {
 324     VALUE v = (VALUE)(ptroff_t)dev->internal;
 325     //forward(v, id_drawchar, 1, convert_font(font));
 326 }
 327 void rb_drawlink(gfxdevice_t*dev, gfxline_t*line, const char*action)
 328 {
 329     VALUE v = (VALUE)(ptroff_t)dev->internal;
 330     forward(v, id_drawlink, convert_line(line), rb_tainted_str_new2(action));
 331 }
 332 void rb_endpage(gfxdevice_t*dev)
 333 {
 334     VALUE v = (VALUE)(ptroff_t)dev->internal;
 335     forward(v, id_endpage, 0);
 336 }
 337 gfxresult_t* rb_finish(gfxdevice_t*dev)
 338 {
 339     VALUE v = (VALUE)(ptroff_t)dev->internal;
 340     VALUE ret = forward(v, id_endpage, 0);
 341     gfxresult_t*r = (gfxresult_t*)rfx_calloc(sizeof(gfxresult_t));
 342     r->internal = (void*)(ptroff_t)ret;
 343     return r;
 344 }
 345
 346 static VALUE page_render(VALUE cls, VALUE device)
 347 {
 348     Check_Type(device, T_OBJECT);
 349     Get_Page(page,cls)
 350
 351     gfxdevice_t dev;
 352     dev.internal = (void*)(ptroff_t)device;
 353     dev.setparameter = rb_setparameter;
 354     dev.startpage = rb_startpage;
 355     dev.startclip = rb_startclip;
 356     dev.endclip = rb_endclip;
 357     dev.stroke = rb_stroke;
 358     dev.fill = rb_fill;
 359     dev.fillbitmap = rb_fillbitmap;
 360     dev.fillgradient = rb_fillgradient;
 361     dev.addfont = rb_addfont;
 362     dev.drawchar = rb_drawchar;
 363     dev.drawlink = rb_drawlink;
 364     dev.endpage = rb_endpage;
 365     dev.finish = rb_finish;
 366
 367     page->page->render(page->page, &dev);
 368
 369     return cls;
 370 }
 371
 372
 373 // --------------------------------------------------------------------------
 374
 375 void Init_gfx()
 376 {
 377     initLog(0,0,0,0,0,2);
 378     pdfdriver = gfxsource_pdf_create();
 379     swfdriver = gfxsource_swf_create();
 380     imagedriver = gfxsource_image_create();
 381
 382     GFX = rb_define_module("GFX");
 383
 384     DocumentPage = rb_define_class_under(GFX, "DocumentPage", rb_cObject);
 385     rb_define_alloc_func(DocumentPage, page_allocate);
 386     rb_define_method(DocumentPage, "width", page_width, 0);
 387     rb_define_method(DocumentPage, "height", page_height, 0);
 388     rb_define_method(DocumentPage, "nr", page_nr, 0);
 389     rb_define_method(DocumentPage, "render", page_render, 1);
 390
 391     Document = rb_define_class_under(GFX, "Document", rb_cObject);
 392     rb_define_method(Document, "initialize", doc_initialize, 1);
 393     rb_define_method(Document, "page", doc_get_page, 1);
 394     rb_define_method(Document, "each_page", doc_each_page, 0);
 395
 396     Bitmap = rb_define_class_under(GFX, "Bitmap", rb_cObject);
 397     rb_define_alloc_func(Bitmap, image_allocate);
 398     rb_define_method(Bitmap, "save_jpeg", image_save_jpeg, 2);
 399
 400     Device = rb_define_class_under(GFX, "Device", rb_cObject);
 401     rb_define_method(Device, "startpage", noop, -1);
 402     rb_define_method(Device, "endpage", noop, -1);
 403     rb_define_method(Device, "startclip", noop, -1);
 404     rb_define_method(Device, "endclip", noop, -1);
 405     rb_define_method(Device, "stroke", noop, -1);
 406     rb_define_method(Device, "fill", noop, -1);
 407     rb_define_method(Device, "fillbitmap", noop, -1);
 408     rb_define_method(Device, "fillgradient", noop, -1);
 409     rb_define_method(Device, "addfont", noop, -1);
 410     rb_define_method(Device, "drawchar", noop, -1);
 411     rb_define_method(Device, "drawlink", noop, -1);
 412     rb_define_method(Device, "endpage", noop, -1);
 413
 414     PDFClass = rb_define_class_under(GFX, "PDF", Document);
 415     rb_define_alloc_func(PDFClass, pdf_allocate);
 416
 417     SWFClass = rb_define_class_under(GFX, "SWF", Document);
 418     rb_define_alloc_func(SWFClass, swf_allocate);
 419
 420     ImageClass = rb_define_class_under(GFX, "ImageRead", Document);
 421     rb_define_alloc_func(ImageClass, imgdrv_allocate);
 422
 423     id_setparameter = rb_intern("setparameter");
 424     id_startpage = rb_intern("startpage") ;
 425     id_startclip = rb_intern("startclip") ;
 426     id_endclip = rb_intern("endclip") ;
 427     id_stroke = rb_intern("stroke") ;
 428     id_fill = rb_intern("fill") ;
 429     id_fillbitmap = rb_intern("fillbitmap") ;
 430     id_fillgradient = rb_intern("fillgradient") ;
 431     id_addfont = rb_intern("addfont") ;
 432     id_drawchar = rb_intern("drawchar") ;
 433     id_drawlink = rb_intern("drawlink") ;
 434     id_endpage = rb_intern("endpage") ;
 435     id_geterror = rb_intern("geterror") ;
 436     id_finish = rb_intern("finish") ;
 437     id_butt = rb_intern("butt");
 438     id_round = rb_intern("round");
 439     id_square = rb_intern("square");
 440     id_miter = rb_intern("miter");
 441     id_bevel = rb_intern("bevel");
 442     id_move = rb_intern("move");
 443     id_line = rb_intern("line");
 444     id_spline = rb_intern("spline");
 445     id_radial = rb_intern("radial");
 446     id_linear = rb_intern("linear");
 447 }
 448