644 行
17 KiB

  1. /*
  2. * libcucul Ruby bindings
  3. * Copyright (c) 2007 Pascal Terjan <pterjan@linuxfr.org>
  4. *
  5. * This library is free software. It comes without any warranty, to
  6. * the extent permitted by applicable law. You can redistribute it
  7. * and/or modify it under the terms of the Do What The Fuck You Want
  8. * To Public License, Version 2, as published by Sam Hocevar. See
  9. * http://sam.zoy.org/wtfpl/COPYING for more details.
  10. */
  11. #include <ruby.h>
  12. #include <cucul.h>
  13. #include <errno.h>
  14. #define _SELF (DATA_PTR(self))
  15. #define simple_func(x) \
  16. static VALUE x (VALUE self) \
  17. { \
  18. if( cucul_##x (_SELF) <0) \
  19. rb_raise(rb_eRuntimeError, strerror(errno)); \
  20. \
  21. return self; \
  22. }
  23. #define get_int(x) \
  24. static VALUE get_##x (VALUE self) \
  25. { \
  26. return INT2NUM(cucul_get_##x (_SELF)); \
  27. }
  28. static void canvas_free(void * p)
  29. {
  30. cucul_free_canvas((cucul_canvas_t *)p);
  31. }
  32. static VALUE canvas_alloc(VALUE klass)
  33. {
  34. cucul_canvas_t *canvas;
  35. VALUE obj;
  36. canvas = cucul_create_canvas(0, 0);
  37. obj = Data_Wrap_Struct(klass, 0, canvas_free, canvas);
  38. return obj;
  39. }
  40. static VALUE canvas_initialize(VALUE self, VALUE width, VALUE height)
  41. {
  42. unsigned int w, h;
  43. w = NUM2INT(width);
  44. h = NUM2INT(height);
  45. cucul_set_canvas_size(_SELF, w, h);
  46. return self;
  47. }
  48. get_int(canvas_height)
  49. get_int(canvas_width)
  50. static VALUE set_canvas_width(VALUE self, VALUE width)
  51. {
  52. cucul_set_canvas_size(_SELF, NUM2INT(width), cucul_get_canvas_height(_SELF));
  53. return width;
  54. }
  55. static VALUE set_canvas_width2(VALUE self, VALUE width)
  56. {
  57. set_canvas_width(self, width);
  58. return self;
  59. }
  60. static VALUE set_canvas_height(VALUE self, VALUE height)
  61. {
  62. cucul_set_canvas_size(_SELF, cucul_get_canvas_width(_SELF), NUM2INT(height));
  63. return height;
  64. }
  65. static VALUE set_canvas_height2(VALUE self, VALUE height)
  66. {
  67. set_canvas_height(self, height);
  68. return self;
  69. }
  70. static VALUE set_canvas_size(VALUE self, VALUE height, VALUE width)
  71. {
  72. cucul_set_canvas_size(_SELF, NUM2INT(width), NUM2INT(height));
  73. return self;
  74. }
  75. /****/
  76. static VALUE gotoxy(VALUE self, VALUE x, VALUE y)
  77. {
  78. if( cucul_gotoxy(_SELF, NUM2INT(x), NUM2INT(y)) <0) {
  79. rb_raise(rb_eRuntimeError, strerror(errno));
  80. }
  81. return self;
  82. }
  83. get_int(cursor_x)
  84. get_int(cursor_y)
  85. simple_func(clear_canvas)
  86. static VALUE put_char(VALUE self, VALUE x, VALUE y, VALUE ch)
  87. {
  88. cucul_put_char(_SELF, NUM2INT(x), NUM2INT(y), NUM2ULONG(ch));
  89. return self;
  90. }
  91. static VALUE get_char(VALUE self, VALUE x, VALUE y)
  92. {
  93. unsigned long int ch;
  94. ch = cucul_get_char(_SELF, NUM2INT(x), NUM2INT(y));
  95. return INT2NUM(ch);
  96. }
  97. static VALUE put_str(VALUE self, VALUE x, VALUE y, VALUE str)
  98. {
  99. cucul_put_str(_SELF, NUM2INT(x), NUM2INT(y), StringValuePtr(str));
  100. return self;
  101. }
  102. static VALUE get_attr(VALUE self, VALUE x, VALUE y)
  103. {
  104. unsigned long int ch;
  105. ch = cucul_get_attr(_SELF, NUM2INT(x), NUM2INT(y));
  106. return INT2NUM(ch);
  107. }
  108. static VALUE set_attr(VALUE self, VALUE attr)
  109. {
  110. if(cucul_set_attr(_SELF, NUM2ULONG(attr)) <0)
  111. rb_raise(rb_eRuntimeError, strerror(errno));
  112. return self;
  113. }
  114. static VALUE set_attr2(VALUE self, VALUE attr)
  115. {
  116. set_attr(self, attr);
  117. return self;
  118. }
  119. static VALUE put_attr(VALUE self, VALUE x, VALUE y, VALUE attr)
  120. {
  121. if(cucul_put_attr(_SELF, NUM2INT(x), NUM2INT(y), NUM2ULONG(attr)) <0)
  122. rb_raise(rb_eRuntimeError, strerror(errno));
  123. return self;
  124. }
  125. static VALUE set_color_ansi(VALUE self, VALUE fg, VALUE bg)
  126. {
  127. if(cucul_set_color_ansi(_SELF, NUM2INT(fg), NUM2INT(bg)) <0)
  128. rb_raise(rb_eRuntimeError, strerror(errno));
  129. return self;
  130. }
  131. static VALUE set_color_argb(VALUE self, VALUE fg, VALUE bg)
  132. {
  133. if(cucul_set_color_argb(_SELF, NUM2UINT(fg), NUM2UINT(bg)) <0) {
  134. rb_raise(rb_eRuntimeError, strerror(errno));
  135. }
  136. return self;
  137. }
  138. static VALUE cprintf(int argc, VALUE* argv, VALUE self)
  139. {
  140. int x, y;
  141. VALUE rx, ry, format, rest, string;
  142. rb_scan_args(argc, argv, "3*", &rx, &ry, &format, &rest);
  143. x = NUM2INT(rx);
  144. y = NUM2INT(ry);
  145. string = rb_funcall2(rb_mKernel, rb_intern("sprintf"), argc-2, argv+2);
  146. cucul_put_str(_SELF, x, y, StringValuePtr(string));
  147. return self;
  148. }
  149. get_int(canvas_handle_x)
  150. get_int(canvas_handle_y)
  151. static VALUE set_canvas_handle(VALUE self, VALUE x, VALUE y)
  152. {
  153. cucul_set_canvas_handle(_SELF, NUM2INT(x), NUM2INT(y));
  154. return self;
  155. }
  156. static VALUE blit(int argc, VALUE* argv, VALUE self) {
  157. VALUE x, y, src, mask;
  158. cucul_canvas_t *csrc, *cmask;
  159. rb_scan_args(argc, argv, "31", &x, &y, &src, &mask);
  160. Check_Type(x, T_FIXNUM);
  161. Check_Type(y, T_FIXNUM);
  162. Check_Type(src, TYPE(self));
  163. Data_Get_Struct(src, cucul_canvas_t, csrc);
  164. if(!NIL_P(mask))
  165. {
  166. Check_Type(mask, TYPE(self));
  167. Data_Get_Struct(mask, cucul_canvas_t, cmask);
  168. }
  169. else
  170. cmask = NULL;
  171. if(cucul_blit(_SELF, NUM2INT(x), NUM2INT(y), csrc, cmask)<0)
  172. rb_raise(rb_eRuntimeError, strerror(errno));
  173. return self;
  174. }
  175. static VALUE set_canvas_boundaries(VALUE self, VALUE x, VALUE y, VALUE w, VALUE h)
  176. {
  177. if(cucul_set_canvas_boundaries(_SELF, NUM2INT(x), NUM2INT(y), NUM2UINT(w), NUM2UINT(h))<0)
  178. {
  179. rb_raise(rb_eRuntimeError, strerror(errno));
  180. }
  181. return self;
  182. }
  183. /****/
  184. simple_func(invert)
  185. simple_func(flip)
  186. simple_func(flop)
  187. simple_func(rotate_180)
  188. simple_func(rotate_left)
  189. simple_func(rotate_right)
  190. simple_func(stretch_left)
  191. simple_func(stretch_right)
  192. /****/
  193. static VALUE draw_line(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2, VALUE ch)
  194. {
  195. cucul_draw_line(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2),NUM2ULONG(ch));
  196. return self;
  197. }
  198. static VALUE draw_polyline(VALUE self, VALUE points, VALUE ch)
  199. {
  200. int i, n;
  201. int *ax, *ay;
  202. int error = 0;
  203. VALUE v, x, y;
  204. n = RARRAY(points)->len;
  205. ax = (int*)malloc(n*sizeof(int));
  206. if(!ax)
  207. rb_raise(rb_eNoMemError,"Out of memory");
  208. ay = (int*)malloc(n*sizeof(int));
  209. if(!ay)
  210. {
  211. free(ax);
  212. rb_raise(rb_eNoMemError,"Out of memory");
  213. }
  214. for(i=0; i<n; i++)
  215. {
  216. v = rb_ary_entry(points, i);
  217. if((TYPE(v) == T_ARRAY) && (RARRAY(v)->len == 2))
  218. {
  219. x = rb_ary_entry(v,0);
  220. y = rb_ary_entry(v,1);
  221. if(rb_obj_is_kind_of(x, rb_cInteger) &&
  222. rb_obj_is_kind_of(y, rb_cInteger))
  223. {
  224. ax[i] = NUM2INT(x);
  225. ay[i] = NUM2INT(y);
  226. } else
  227. error = 1;
  228. }
  229. else
  230. error = 1;
  231. }
  232. if(error)
  233. {
  234. free(ax);
  235. free(ay);
  236. rb_raise(rb_eArgError, "Invalid list of points");
  237. }
  238. n--;
  239. cucul_draw_polyline(_SELF, ax, ay, n, NUM2ULONG(ch));
  240. free(ax);
  241. free(ay);
  242. return self;
  243. }
  244. static VALUE draw_thin_line(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2)
  245. {
  246. cucul_draw_thin_line(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2));
  247. return self;
  248. }
  249. static VALUE draw_thin_polyline(VALUE self, VALUE points)
  250. {
  251. int i, n;
  252. int *ax, *ay;
  253. int error = 0;
  254. VALUE v, x, y;
  255. n = RARRAY(points)->len;
  256. ax = (int*)malloc(n*sizeof(int));
  257. if(!ax)
  258. rb_raise(rb_eNoMemError,"Out of memory");
  259. ay = (int*)malloc(n*sizeof(int));
  260. if(!ay)
  261. {
  262. free(ax);
  263. rb_raise(rb_eNoMemError,"Out of memory");
  264. }
  265. for(i=0; i<n; i++)
  266. {
  267. v = rb_ary_entry(points, i);
  268. if((TYPE(v) == T_ARRAY) && (RARRAY(v)->len == 2))
  269. {
  270. x = rb_ary_entry(v,0);
  271. y = rb_ary_entry(v,1);
  272. if(rb_obj_is_kind_of(x, rb_cInteger) &&
  273. rb_obj_is_kind_of(y, rb_cInteger))
  274. {
  275. ax[i] = NUM2INT(x);
  276. ay[i] = NUM2INT(y);
  277. } else
  278. error = 1;
  279. }
  280. else
  281. error = 1;
  282. }
  283. if(error)
  284. {
  285. free(ax);
  286. free(ay);
  287. rb_raise(rb_eArgError, "Invalid list of points");
  288. }
  289. n--;
  290. cucul_draw_thin_polyline(_SELF, ax, ay, n);
  291. free(ax);
  292. free(ay);
  293. return self;
  294. }
  295. static VALUE draw_circle(VALUE self, VALUE x, VALUE y, VALUE r, VALUE ch)
  296. {
  297. cucul_draw_circle(_SELF, NUM2INT(x), NUM2INT(y), NUM2INT(r), NUM2ULONG(ch));
  298. return self;
  299. }
  300. static VALUE draw_ellipse(VALUE self, VALUE x, VALUE y, VALUE a, VALUE b, VALUE ch)
  301. {
  302. cucul_draw_ellipse(_SELF, NUM2INT(x), NUM2INT(y), NUM2INT(a), NUM2INT(b), NUM2ULONG(ch));
  303. return self;
  304. }
  305. static VALUE draw_thin_ellipse(VALUE self, VALUE x, VALUE y, VALUE a, VALUE b)
  306. {
  307. cucul_draw_thin_ellipse(_SELF, NUM2INT(x), NUM2INT(y), NUM2INT(a), NUM2INT(b));
  308. return self;
  309. }
  310. static VALUE fill_ellipse(VALUE self, VALUE x, VALUE y, VALUE a, VALUE b, VALUE ch)
  311. {
  312. cucul_fill_ellipse(_SELF, NUM2INT(x), NUM2INT(y), NUM2INT(a), NUM2INT(b), NUM2ULONG(ch));
  313. return self;
  314. }
  315. static VALUE draw_box(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2, VALUE ch)
  316. {
  317. cucul_draw_box(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2), NUM2ULONG(ch));
  318. return self;
  319. }
  320. static VALUE draw_thin_box(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2)
  321. {
  322. cucul_draw_thin_box(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2));
  323. return self;
  324. }
  325. static VALUE draw_cp437_box(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2)
  326. {
  327. cucul_draw_cp437_box(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2));
  328. return self;
  329. }
  330. static VALUE fill_box(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2, VALUE ch)
  331. {
  332. cucul_fill_box(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2), NUM2ULONG(ch));
  333. return self;
  334. }
  335. static VALUE draw_triangle(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2, VALUE x3, VALUE y3, VALUE ch)
  336. {
  337. cucul_draw_triangle(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2), NUM2INT(x3), NUM2INT(y3), NUM2ULONG(ch));
  338. return self;
  339. }
  340. static VALUE draw_thin_triangle(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2, VALUE x3, VALUE y3)
  341. {
  342. cucul_draw_thin_triangle(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2), NUM2INT(x3), NUM2INT(y3));
  343. return self;
  344. }
  345. static VALUE fill_triangle(VALUE self, VALUE x1, VALUE y1, VALUE x2, VALUE y2, VALUE x3, VALUE y3, VALUE ch)
  346. {
  347. cucul_fill_triangle(_SELF, NUM2INT(x1), NUM2INT(y1), NUM2INT(x2), NUM2INT(y2), NUM2INT(x3), NUM2INT(y3), NUM2ULONG(ch));
  348. return self;
  349. }
  350. /****/
  351. get_int(frame_count)
  352. static VALUE set_frame(VALUE self, VALUE id)
  353. {
  354. if(cucul_set_frame(_SELF, NUM2INT(id))<0)
  355. rb_raise(rb_eArgError, strerror(errno));
  356. return self;
  357. }
  358. static VALUE set_frame2(VALUE self, VALUE id)
  359. {
  360. set_frame(self, id);
  361. return self;
  362. }
  363. static VALUE get_frame_name(VALUE self)
  364. {
  365. return rb_str_new2(cucul_get_frame_name(_SELF));
  366. }
  367. static VALUE set_frame_name(VALUE self, VALUE name)
  368. {
  369. if(cucul_set_frame_name(_SELF, StringValuePtr(name))<0)
  370. rb_raise(rb_eRuntimeError, strerror(errno));
  371. return self;
  372. }
  373. static VALUE set_frame_name2(VALUE self, VALUE name)
  374. {
  375. set_frame_name(self, name);
  376. return self;
  377. }
  378. static VALUE create_frame(VALUE self, VALUE id)
  379. {
  380. if(cucul_create_frame(_SELF, NUM2INT(id))<0) {
  381. rb_raise(rb_eRuntimeError, strerror(errno));
  382. }
  383. return self;
  384. }
  385. static VALUE free_frame(VALUE self, VALUE id)
  386. {
  387. if(cucul_free_frame(_SELF, NUM2INT(id))<0) {
  388. rb_raise(rb_eArgError, strerror(errno));
  389. }
  390. return self;
  391. }
  392. /****/
  393. static VALUE import_memory(VALUE self, VALUE data, VALUE format)
  394. {
  395. long int bytes;
  396. bytes = cucul_import_memory (_SELF, StringValuePtr(data), RSTRING(StringValue(data))->len, StringValuePtr(format));
  397. if(bytes <= 0)
  398. rb_raise(rb_eRuntimeError, strerror(errno));
  399. return self;
  400. }
  401. static VALUE import_file(VALUE self, VALUE filename, VALUE format)
  402. {
  403. long int bytes;
  404. bytes = cucul_import_file (_SELF, StringValuePtr(filename), StringValuePtr(format));
  405. if(bytes <= 0)
  406. rb_raise(rb_eRuntimeError, strerror(errno));
  407. return self;
  408. }
  409. static VALUE export_memory(VALUE self, VALUE format)
  410. {
  411. unsigned long int bytes;
  412. void *result;
  413. VALUE ret;
  414. result = cucul_export_memory (_SELF, StringValuePtr(format), &bytes);
  415. ret = rb_str_new(result, bytes);
  416. free(result);
  417. return ret;
  418. }
  419. static VALUE export_list()
  420. {
  421. VALUE ary, ary2;
  422. char const* const* list;
  423. list = cucul_get_export_list();
  424. ary = rb_ary_new();
  425. while (*list != NULL)
  426. {
  427. ary2 = rb_ary_new();
  428. rb_ary_push(ary2, rb_str_new2(*list));
  429. list++;
  430. rb_ary_push(ary2, rb_str_new2(*list));
  431. list++;
  432. rb_ary_push(ary, ary2);
  433. }
  434. return ary;
  435. }
  436. static VALUE import_list()
  437. {
  438. VALUE ary, ary2;
  439. char const* const* list;
  440. list = cucul_get_import_list();
  441. ary = rb_ary_new();
  442. while (*list != NULL)
  443. {
  444. ary2 = rb_ary_new();
  445. rb_ary_push(ary2, rb_str_new2(*list));
  446. list++;
  447. rb_ary_push(ary2, rb_str_new2(*list));
  448. list++;
  449. rb_ary_push(ary, ary2);
  450. }
  451. return ary;
  452. }
  453. /****/
  454. void Init_cucul_canvas(VALUE mCucul)
  455. {
  456. VALUE cCanvas = rb_define_class_under(mCucul, "Canvas", rb_cObject);
  457. rb_define_alloc_func(cCanvas, canvas_alloc);
  458. rb_define_method(cCanvas, "initialize", canvas_initialize, 2);
  459. rb_define_method(cCanvas, "width", get_canvas_width, 0);
  460. rb_define_method(cCanvas, "width=", set_canvas_width, 1);
  461. rb_define_method(cCanvas, "set_width", set_canvas_width2, 1);
  462. rb_define_method(cCanvas, "height", get_canvas_height, 0);
  463. rb_define_method(cCanvas, "height=", set_canvas_height, 0);
  464. rb_define_method(cCanvas, "set_height", set_canvas_height2, 0);
  465. rb_define_method(cCanvas, "set_size", set_canvas_size, 2);
  466. rb_define_method(cCanvas, "gotoxy", gotoxy, 2);
  467. rb_define_method(cCanvas, "cursor_x", get_cursor_x, 0);
  468. rb_define_method(cCanvas, "cursor_y", get_cursor_y, 0);
  469. rb_define_method(cCanvas, "handle_x", get_canvas_handle_x, 0);
  470. rb_define_method(cCanvas, "handle_y", get_canvas_handle_y, 0);
  471. rb_define_method(cCanvas, "set_handle", set_canvas_handle, 2);
  472. rb_define_method(cCanvas, "blit", blit, -1);
  473. rb_define_method(cCanvas, "set_boundaries", set_canvas_boundaries, 4);
  474. rb_define_method(cCanvas, "clear", clear_canvas, 0);
  475. rb_define_method(cCanvas, "put_char", put_char, 3);
  476. rb_define_method(cCanvas, "get_char", get_char, 2);
  477. rb_define_method(cCanvas, "put_str", put_str, 3);
  478. rb_define_method(cCanvas, "printf", cprintf, -1);
  479. rb_define_method(cCanvas, "get_attr", get_attr, 3);
  480. rb_define_method(cCanvas, "attr=", set_attr, 1);
  481. rb_define_method(cCanvas, "set_attr", set_attr2, 1);
  482. rb_define_method(cCanvas, "put_attr", put_attr, 3);
  483. rb_define_method(cCanvas, "set_color_ansi", set_color_ansi, 2);
  484. rb_define_method(cCanvas, "set_color_argb", set_color_argb, 2);
  485. rb_define_method(cCanvas, "invert", invert, 0);
  486. rb_define_method(cCanvas, "flip", flip, 0);
  487. rb_define_method(cCanvas, "flop", flop, 0);
  488. rb_define_method(cCanvas, "rotate_180", rotate_180, 0);
  489. rb_define_method(cCanvas, "rotate_left", rotate_left, 0);
  490. rb_define_method(cCanvas, "rotate_right", rotate_right, 0);
  491. rb_define_method(cCanvas, "stretch_left", stretch_left, 0);
  492. rb_define_method(cCanvas, "stretch_right", stretch_right, 0);
  493. rb_define_method(cCanvas, "draw_line", draw_line, 5);
  494. rb_define_method(cCanvas, "draw_polyline", draw_polyline, 2);
  495. rb_define_method(cCanvas, "draw_thin_line", draw_thin_line, 4);
  496. rb_define_method(cCanvas, "draw_thin_polyline", draw_thin_polyline, 1);
  497. rb_define_method(cCanvas, "draw_circle", draw_circle, 4);
  498. rb_define_method(cCanvas, "draw_ellipse", draw_ellipse, 5);
  499. rb_define_method(cCanvas, "draw_thin_ellipse", draw_thin_ellipse, 4);
  500. rb_define_method(cCanvas, "fill_ellipse", fill_ellipse, 5);
  501. rb_define_method(cCanvas, "draw_box", draw_box, 5);
  502. rb_define_method(cCanvas, "draw_thin_box", draw_thin_box, 4);
  503. rb_define_method(cCanvas, "draw_cp437_box", draw_cp437_box, 4);
  504. rb_define_method(cCanvas, "fill_box", fill_box, 5);
  505. rb_define_method(cCanvas, "draw_triangle", draw_triangle, 7);
  506. rb_define_method(cCanvas, "draw_thin_triangle", draw_thin_triangle, 6);
  507. rb_define_method(cCanvas, "fill_triangle", fill_triangle, 7);
  508. rb_define_method(cCanvas, "frame_count", get_frame_count, 0);
  509. rb_define_method(cCanvas, "frame=", set_frame, 1);
  510. rb_define_method(cCanvas, "set_frame", set_frame2, 1);
  511. rb_define_method(cCanvas, "frame_name", get_frame_name, 0);
  512. rb_define_method(cCanvas, "frame_name=", set_frame_name, 1);
  513. rb_define_method(cCanvas, "set_frame_name", set_frame_name2, 1);
  514. rb_define_method(cCanvas, "create_frame", create_frame, 1);
  515. rb_define_method(cCanvas, "free_frame", free_frame, 1);
  516. rb_define_method(cCanvas, "import_memory", import_memory, 2);
  517. rb_define_method(cCanvas, "import_file", import_file, 2);
  518. rb_define_method(cCanvas, "export_memory", export_memory, 1);
  519. rb_define_singleton_method(cCanvas, "export_list", export_list, 0);
  520. rb_define_singleton_method(cCanvas, "import_list", import_list, 0);
  521. }