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"
12 #define RUBY_GFX_VERSION "0.9.0"
15 static VALUE Font, Glyph, Bitmap, Document, DocumentPage, PDFClass, SWFClass, ImageClass, Device;
18 typedef struct doc_internal {
20 gfxsource_t*driver; // filled by alloc
22 gfxfontlist_t*fontlist;
25 typedef struct page_internal {
30 typedef struct image_internal {
34 typedef struct glyph_internal {
39 typedef struct font_internal {
44 static gfxsource_t* pdfdriver = 0;
45 static gfxsource_t* imagedriver = 0;
46 static gfxsource_t* swfdriver = 0;
48 #define Get_Doc(doc,cls) doc_internal_t*doc=0;Data_Get_Struct(cls, doc_internal_t, doc);
49 #define Get_Page(page,cls) page_internal_t*page=0;Data_Get_Struct(cls, page_internal_t, page);
51 static VALUE doc_allocate(VALUE cls, gfxsource_t*driver);
52 static VALUE page_allocate(VALUE cls);
54 // ------------------------ documents ---------------------------------------
56 static VALUE doc_initialize(VALUE cls, VALUE _filename)
58 Check_Type(_filename, T_STRING);
60 const char*filename = StringValuePtr(_filename);
61 doc->fontlist = gfxfontlist_create();
62 doc->doc = pdfdriver->open(pdfdriver, filename);
66 static VALUE doc_num_pages(VALUE cls)
69 return INT2FIX(doc->doc->num_pages);
72 static VALUE doc_get_page(VALUE cls, VALUE _nr)
74 Check_Type(_nr, T_FIXNUM);
75 int nr = FIX2INT(_nr);
78 VALUE v = page_allocate(DocumentPage);
80 page->page = doc->doc->getpage(doc->doc, nr);
83 rb_raise(rb_eArgError, "No page %d in document", nr);
89 static VALUE doc_each_page(VALUE cls)
93 for(t=1;t<=doc->doc->num_pages;t++) {
94 VALUE v = page_allocate(DocumentPage);
96 page->page = doc->doc->getpage(doc->doc, t);
103 static void doc_mark(doc_internal_t*doc)
105 gfxfontlist_t*l = doc->fontlist;
108 rb_gc_mark((VALUE)l->user);
113 static void doc_free(doc_internal_t*doc)
115 gfxfontlist_free(doc->fontlist, 0);
116 doc->doc->destroy(doc->doc);
120 static VALUE doc_allocate(VALUE cls, gfxsource_t*driver)
122 doc_internal_t*doc = 0;
123 VALUE v = Data_Make_Struct(cls, doc_internal_t, doc_mark, doc_free, doc);
125 memset(doc, 0, sizeof(doc_internal_t));
126 doc->driver = driver;
130 static VALUE pdf_allocate(VALUE cls) {return doc_allocate(cls, pdfdriver);}
131 static VALUE swf_allocate(VALUE cls) {return doc_allocate(cls, swfdriver);}
132 static VALUE imgdrv_allocate(VALUE cls) {return doc_allocate(cls, imagedriver);}
134 // ------------------------ doc pages ---------------------------------------
136 static void page_free(page_internal_t*page)
140 page->page->destroy(page->page);
145 static void page_mark(page_internal_t*page)
147 rb_gc_mark(page->doc->self);
149 static VALUE page_allocate(VALUE cls)
151 page_internal_t*page = 0;
152 VALUE v = Data_Make_Struct(cls, page_internal_t, page_mark, page_free, page);
153 memset(page, 0, sizeof(page_internal_t));
156 static VALUE page_nr(VALUE cls)
159 return INT2FIX(page->page->nr);
161 static VALUE page_width(VALUE cls)
164 return INT2FIX(page->page->width);
166 static VALUE page_height(VALUE cls)
169 return INT2FIX(page->page->height);
172 // ------------------------ image -------------------------------------------
174 #define Get_Image(image,cls) image_internal_t*image=0;Data_Get_Struct(cls, image_internal_t, image);
176 static void image_free(image_internal_t*image)
180 static VALUE image_allocate(VALUE cls)
182 image_internal_t*image = 0;
183 VALUE v = Data_Make_Struct(cls, image_internal_t, 0, image_free, image);
184 memset(image, 0, sizeof(image_internal_t));
187 static VALUE image_width(VALUE cls)
190 return INT2FIX(image->image->width);
192 static VALUE image_height(VALUE cls)
195 return INT2FIX(image->image->height);
197 static VALUE image_save_jpeg(VALUE cls, VALUE _filename, VALUE quality)
200 Check_Type(_filename, T_STRING);
201 Check_Type(quality, T_FIXNUM);
202 const char*filename = StringValuePtr(_filename);
203 int l = strlen(filename);
205 if(l>=4 && !strcmp(&filename[l-4], ".jpg")) jpeg = 1;
206 if(l>=5 && !strcmp(&filename[l-5], ".jpeg")) jpeg = 1;
207 if(l>=4 && !strcmp(&filename[l-4], ".JPG")) jpeg = 1;
208 jpeg_save(image->image->data, image->image->width, image->image->height, FIX2INT(quality), filename);
211 VALUE convert_image(gfximage_t*_image)
213 VALUE v = image_allocate(Bitmap);
215 image->image = _image;
218 void invalidate_image(VALUE v)
224 // ------------------------ glyphs ------------------------------------------
226 static VALUE convert_line(gfxline_t*line);
228 #define Get_Glyph(glyph,cls) glyph_internal_t*glyph=0;Data_Get_Struct(cls, glyph_internal_t, glyph);
230 static void glyph_free(glyph_internal_t*glyph)
235 static VALUE glyph_allocate(VALUE cls)
237 glyph_internal_t*glyph = 0;
238 VALUE v = Data_Make_Struct(cls, glyph_internal_t, 0, glyph_free, glyph);
239 memset(glyph, 0, sizeof(glyph_internal_t));
243 static VALUE glyph_polygon(VALUE cls)
245 Get_Glyph(glyph,cls);
246 return convert_line(glyph->font->glyphs[glyph->nr].line);
249 static VALUE glyph_advance(VALUE cls)
251 Get_Glyph(glyph,cls);
252 return rb_float_new(glyph->font->glyphs[glyph->nr].advance);
255 static VALUE glyph_unicode(VALUE cls)
257 Get_Glyph(glyph,cls);
258 return INT2FIX(glyph->font->glyphs[glyph->nr].unicode);
261 // ------------------------ font --------------------------------------------
263 #define Get_Font(font,cls) font_internal_t*font=0;Data_Get_Struct(cls, font_internal_t, font);
265 static void font_mark(font_internal_t*font)
267 rb_gc_mark(font->glyph_array);
270 static void font_free(font_internal_t*font)
275 static VALUE font_allocate(VALUE cls)
277 font_internal_t*font = 0;
278 VALUE v = Data_Make_Struct(cls, font_internal_t, font_mark, font_free, font);
279 memset(font, 0, sizeof(font_internal_t));
283 static VALUE font_name(VALUE cls)
286 return rb_tainted_str_new2(font->font->id);
289 static VALUE font_glyphs(VALUE cls)
292 return font->glyph_array;
295 // ------------------------ gfx device --------------------------------------
297 typedef struct device_internal {
302 static ID id_setparameter = 0;
303 static ID id_startpage = 0;
304 static ID id_startclip = 0;
305 static ID id_endclip = 0;
306 static ID id_stroke = 0;
307 static ID id_fill = 0;
308 static ID id_fillbitmap = 0;
309 static ID id_fillgradient = 0;
310 static ID id_addfont = 0;
311 static ID id_drawchar = 0;
312 static ID id_drawlink = 0;
313 static ID id_endpage = 0;
314 static ID id_geterror = 0;
315 static ID id_finish = 0;
316 static ID id_butt = 0;
317 static ID id_round = 0;
318 static ID id_square = 0;
319 static ID id_bevel = 0;
320 static ID id_miter = 0;
321 static ID id_move = 0;
322 static ID id_line = 0;
323 static ID id_spline = 0;
324 static ID id_radial = 0;
325 static ID id_linear = 0;
327 static VALUE noop(int argc, VALUE *argv, VALUE obj) {return obj;}
329 #define forward(v,id,args...) rb_respond_to((v), (id))?rb_funcall((v), (id), args):0
331 VALUE convert_line(gfxline_t*line)
335 while(l) {l=l->next;len++;}
337 VALUE array = rb_ary_new2(len);
338 rb_gc_register_address(&array);
344 if(l->type == gfx_moveTo) {
345 e = rb_ary_new3(3, ID2SYM(id_move), Qfalse, Qfalse);
346 rb_ary_store(array, pos, e);
347 rb_ary_store(e, 1, rb_float_new(l->x));
348 rb_ary_store(e, 2, rb_float_new(l->y));
349 } else if(l->type == gfx_lineTo) {
350 e = rb_ary_new3(3, ID2SYM(id_line), Qfalse, Qfalse);
351 rb_ary_store(array, pos, e);
352 rb_ary_store(e, 1, rb_float_new(l->x));
353 rb_ary_store(e, 2, rb_float_new(l->y));
355 e = rb_ary_new3(5, ID2SYM(id_spline), Qfalse, Qfalse, Qfalse, Qfalse);
356 rb_ary_store(array, pos, e);
357 rb_ary_store(e, 1, rb_float_new(l->x));
358 rb_ary_store(e, 2, rb_float_new(l->y));
359 rb_ary_store(e, 3, rb_float_new(l->sx));
360 rb_ary_store(e, 4, rb_float_new(l->sy));
365 rb_gc_unregister_address(&array);
368 VALUE convert_color(gfxcolor_t*color)
370 return rb_ary_new3(4, INT2FIX(color->a), INT2FIX(color->r), INT2FIX(color->g), INT2FIX(color->b));
372 VALUE convert_matrix(gfxmatrix_t*matrix)
374 VALUE array = rb_ary_new2(3);
375 rb_gc_register_address(&array);
376 VALUE a = rb_ary_new2(2);
377 rb_ary_store(array, 0, a);
378 rb_ary_store(a, 0, rb_float_new(matrix->m00));
379 rb_ary_store(a, 1, rb_float_new(matrix->m01));
381 rb_ary_store(array, 1, a);
382 rb_ary_store(a, 0, rb_float_new(matrix->m10));
383 rb_ary_store(a, 1, rb_float_new(matrix->m11));
385 rb_ary_store(array, 2, a);
386 rb_ary_store(a, 0, rb_float_new(matrix->tx));
387 rb_ary_store(a, 1, rb_float_new(matrix->ty));
388 rb_gc_unregister_address(&array);
391 VALUE convert_font(device_internal_t*i, gfxfont_t*font)
393 VALUE v = (VALUE)gfxfontlist_getuserdata(i->doc->fontlist, font->id);
396 VALUE*a = (VALUE*)malloc(sizeof(VALUE)*font->num_glyphs);
398 for(t=0;t<font->num_glyphs;t++) {
399 a[t] = glyph_allocate(Glyph);
404 VALUE v2 = font_allocate(Font);
408 f->glyph_array = rb_ary_new4(font->num_glyphs, a);
410 i->doc->fontlist = gfxfontlist_addfont2(i->doc->fontlist, font, (void*)v2);
414 device_internal_t*i = (device_internal_t*)dev->internal; \
416 int rb_setparameter(gfxdevice_t*dev, const char*key, const char*value)
419 VALUE ret = forward(v,id_setparameter,2,rb_tainted_str_new2(key),rb_tainted_str_new2(value));
422 void rb_startpage(gfxdevice_t*dev, int width, int height)
425 VALUE ret = forward(v,id_startpage,2,INT2FIX(width),INT2FIX(height));
427 void rb_startclip(gfxdevice_t*dev, gfxline_t*line)
430 VALUE ret = forward(v,id_startclip,1,convert_line(line));
432 void rb_endclip(gfxdevice_t*dev)
435 VALUE ret = forward(v,id_endclip,0);
437 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)
442 if(cap_style == gfx_capButt) cap = id_butt;
443 else if(cap_style == gfx_capRound) cap = id_round;
444 else if(cap_style == gfx_capSquare) cap = id_square;
447 if(joint_style == gfx_joinRound) joint = id_round;
448 else if(joint_style == gfx_joinMiter) joint = id_miter;
449 else if(joint_style == gfx_joinBevel) joint = id_bevel;
451 VALUE v_line = convert_line(line);
452 rb_gc_register_address(&v_line);
453 VALUE v_width = rb_float_new(width);
454 rb_gc_register_address(&v_width);
455 VALUE v_color = convert_color(color);
456 rb_gc_register_address(&v_color);
457 VALUE v_miter = rb_float_new(miterLimit);
458 rb_gc_register_address(&v_miter);
460 forward(v, id_stroke, 6, v_line, v_width, v_color, ID2SYM(cap), ID2SYM(joint), v_miter);
462 rb_gc_unregister_address(&v_miter);
463 rb_gc_unregister_address(&v_color);
464 rb_gc_unregister_address(&v_width);
465 rb_gc_unregister_address(&v_line);
467 void rb_fill(gfxdevice_t*dev, gfxline_t*line, gfxcolor_t*color)
471 VALUE v_line = convert_line(line);
472 rb_gc_register_address(&v_line);
473 VALUE v_color = convert_color(color);
474 rb_gc_register_address(&v_color);
476 forward(v, id_fill, 2, v_line, v_color);
478 rb_gc_unregister_address(&v_color);
479 rb_gc_unregister_address(&v_line);
481 void rb_fillbitmap(gfxdevice_t*dev, gfxline_t*line, gfximage_t*img, gfxmatrix_t*matrix, gfxcxform_t*cxform)
484 VALUE v_image = convert_image(img);
485 rb_gc_register_address(&v_image);
486 VALUE v_line = convert_line(line);
487 rb_gc_register_address(&v_line);
488 VALUE v_matrix = convert_matrix(matrix);
489 rb_gc_register_address(&v_matrix);
490 forward(v, id_fillbitmap, 4, v_line, v_image, v_matrix, Qnil);
491 rb_gc_unregister_address(&v_matrix);
492 rb_gc_unregister_address(&v_line);
493 rb_gc_unregister_address(&v_image);
494 invalidate_image(v_image);
496 void rb_fillgradient(gfxdevice_t*dev, gfxline_t*line, gfxgradient_t*gradient, gfxgradienttype_t type, gfxmatrix_t*matrix)
499 ID typeid = (type == gfxgradient_linear)? id_linear : id_radial;
501 VALUE v_line = convert_line(line);
502 rb_gc_register_address(&v_line);
503 VALUE v_matrix = convert_matrix(matrix);
504 rb_gc_register_address(&v_matrix);
505 VALUE v_gradient = convert_gradient(gradient);
506 rb_gc_register_address(&v_gradient);
507 forward(v, id_fillgradient, 4, v_line, v_gradient, ID2SYM(typeid), convert_matrix(matrix));
508 rb_gc_unregister_address(&v_gradient);
509 rb_gc_unregister_address(&v_matrix);
510 rb_gc_unregister_address(&v_line);
512 void rb_addfont(gfxdevice_t*dev, gfxfont_t*font)
515 forward(v, id_addfont, 1, convert_font(i, font));
517 void rb_drawchar(gfxdevice_t*dev, gfxfont_t*font, int glyphnr, gfxcolor_t*color, gfxmatrix_t*matrix)
520 forward(v, id_drawchar, 4, convert_font(i, font), INT2FIX(glyphnr), convert_color(color), convert_matrix(matrix));
522 void rb_drawlink(gfxdevice_t*dev, gfxline_t*line, const char*action)
525 forward(v, id_drawlink, convert_line(line), rb_tainted_str_new2(action));
527 void rb_endpage(gfxdevice_t*dev)
530 forward(v, id_endpage, 0);
532 gfxresult_t* rb_finish(gfxdevice_t*dev)
535 VALUE ret = forward(v, id_endpage, 0);
536 gfxresult_t*r = (gfxresult_t*)rfx_calloc(sizeof(gfxresult_t));
537 r->internal = (void*)(ptroff_t)ret;
541 static VALUE page_render(VALUE cls, VALUE device)
543 Check_Type(device, T_OBJECT);
552 dev.setparameter = rb_setparameter;
553 dev.startpage = rb_startpage;
554 dev.startclip = rb_startclip;
555 dev.endclip = rb_endclip;
556 dev.stroke = rb_stroke;
558 dev.fillbitmap = rb_fillbitmap;
559 dev.fillgradient = rb_fillgradient;
560 dev.addfont = rb_addfont;
561 dev.drawchar = rb_drawchar;
562 dev.drawlink = rb_drawlink;
563 dev.endpage = rb_endpage;
564 dev.finish = rb_finish;
566 page->page->render(page->page, &dev);
572 // --------------------------------------------------------------------------
576 initLog(0,0,0,0,0,2);
577 pdfdriver = gfxsource_pdf_create();
578 swfdriver = gfxsource_swf_create();
579 imagedriver = gfxsource_image_create();
581 GFX = rb_define_module("GFX");
583 DocumentPage = rb_define_class_under(GFX, "DocumentPage", rb_cObject);
584 rb_define_method(DocumentPage, "width", page_width, 0);
585 rb_define_method(DocumentPage, "height", page_height, 0);
586 rb_define_method(DocumentPage, "nr", page_nr, 0);
587 rb_define_method(DocumentPage, "render", page_render, 1);
589 Document = rb_define_class_under(GFX, "Document", rb_cObject);
590 rb_define_method(Document, "initialize", doc_initialize, 1);
591 rb_define_method(Document, "page", doc_get_page, 1);
592 rb_define_method(Document, "each_page", doc_each_page, 0);
594 Bitmap = rb_define_class_under(GFX, "Bitmap", rb_cObject);
595 rb_define_method(Bitmap, "save_jpeg", image_save_jpeg, 2);
597 Glyph = rb_define_class_under(GFX, "Glyph", rb_cObject);
598 rb_define_method(Glyph, "polygon", glyph_polygon, 0);
599 rb_define_method(Glyph, "unicode", glyph_unicode, 0);
600 rb_define_method(Glyph, "advance", glyph_advance, 0);
602 Font = rb_define_class_under(GFX, "Font", rb_cObject);
603 rb_define_method(Font, "name", font_name, 0);
604 rb_define_method(Font, "glyphs", font_glyphs, 0);
606 Device = rb_define_class_under(GFX, "Device", rb_cObject);
607 rb_define_method(Device, "startpage", noop, -1);
608 rb_define_method(Device, "endpage", noop, -1);
609 rb_define_method(Device, "startclip", noop, -1);
610 rb_define_method(Device, "endclip", noop, -1);
611 rb_define_method(Device, "stroke", noop, -1);
612 rb_define_method(Device, "fill", noop, -1);
613 rb_define_method(Device, "fillbitmap", noop, -1);
614 rb_define_method(Device, "fillgradient", noop, -1);
615 rb_define_method(Device, "addfont", noop, -1);
616 rb_define_method(Device, "drawchar", noop, -1);
617 rb_define_method(Device, "drawlink", noop, -1);
618 rb_define_method(Device, "endpage", noop, -1);
620 PDFClass = rb_define_class_under(GFX, "PDF", Document);
621 rb_define_alloc_func(PDFClass, pdf_allocate);
623 SWFClass = rb_define_class_under(GFX, "SWF", Document);
624 rb_define_alloc_func(SWFClass, swf_allocate);
626 ImageClass = rb_define_class_under(GFX, "ImageRead", Document);
627 rb_define_alloc_func(ImageClass, imgdrv_allocate);
629 id_setparameter = rb_intern("setparameter");
630 id_startpage = rb_intern("startpage") ;
631 id_startclip = rb_intern("startclip") ;
632 id_endclip = rb_intern("endclip") ;
633 id_stroke = rb_intern("stroke") ;
634 id_fill = rb_intern("fill") ;
635 id_fillbitmap = rb_intern("fillbitmap") ;
636 id_fillgradient = rb_intern("fillgradient") ;
637 id_addfont = rb_intern("addfont") ;
638 id_drawchar = rb_intern("drawchar") ;
639 id_drawlink = rb_intern("drawlink") ;
640 id_endpage = rb_intern("endpage") ;
641 id_geterror = rb_intern("geterror") ;
642 id_finish = rb_intern("finish") ;
643 id_butt = rb_intern("butt");
644 id_round = rb_intern("round");
645 id_square = rb_intern("square");
646 id_miter = rb_intern("miter");
647 id_bevel = rb_intern("bevel");
648 id_move = rb_intern("move");
649 id_line = rb_intern("line");
650 id_spline = rb_intern("spline");
651 id_radial = rb_intern("radial");
652 id_linear = rb_intern("linear");