lib/ruby/gfx.c

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