feat Add new methods to Rectangle and Vector4D

1 files changed, 74 insertions, 6 deletions

diff --git a/src/rectangle.cpp b/src/rectangle.cpp
index 18c86a4..f678cc3 100644
--- a/src/rectangle.cpp
+++ b/src/rectangle.cpp
@@ -16,6 +16,8 @@
 
 #include "rectangle.hpp"
 
+#include <mruby/array.h>
+
 #include "vector_3d.hpp"
 #include "view_2d.hpp"
 #include "view_3d.hpp"
@@ -136,6 +138,58 @@ cg_cRectangle_set_height(mrb_state *mrb, mrb_value self)
 }
 
 static mrb_value
+cg_cRectangle_get_xy(mrb_state *mrb, mrb_value self)
+{
+  auto ptr = (std::shared_ptr<VK::Rectangle>*)DATA_PTR(self);
+
+  mrb_value array = mrb_ary_new_capa(mrb, 2);
+  mrb_ary_push(mrb, array, mrb_float_value(mrb, (*ptr)->x));
+  mrb_ary_push(mrb, array, mrb_float_value(mrb, (*ptr)->y));
+
+  return array;
+}
+
+static mrb_value
+cg_cRectangle_get_width_height(mrb_state *mrb, mrb_value self)
+{
+  auto ptr = (std::shared_ptr<VK::Rectangle>*)DATA_PTR(self);
+
+  mrb_value array = mrb_ary_new_capa(mrb, 2);
+  mrb_ary_push(mrb, array, mrb_float_value(mrb, (*ptr)->width));
+  mrb_ary_push(mrb, array, mrb_float_value(mrb, (*ptr)->height));
+
+  return array;
+}
+
+static mrb_value
+cg_cRectangle_set_xy(mrb_state *mrb, mrb_value self)
+{
+  mrb_float x, y;
+  auto ptr = (std::shared_ptr<VK::Rectangle>*)DATA_PTR(self);
+
+  mrb_get_args(mrb, "ff", &x, &y);
+
+  (*ptr)->x = x;
+  (*ptr)->y = y;
+
+  return self;
+}
+
+static mrb_value
+cg_cRectangle_set_width_height(mrb_state *mrb, mrb_value self)
+{
+  mrb_float width, height;
+  auto ptr = (std::shared_ptr<VK::Rectangle>*)DATA_PTR(self);
+
+  mrb_get_args(mrb, "ff", &width, &height);
+
+  (*ptr)->width = width;
+  (*ptr)->height = height;
+
+  return self;
+}
+
+static mrb_value
 cg_cRectangle_align_vertically(mrb_state *mrb, mrb_value self)
 {
   std::shared_ptr<VK::Rectangle> *ptr, *that;
@@ -206,19 +260,33 @@ cg_rectangle_init(mrb_state *mrb)
   mrb_define_method(
     mrb, cg_cRectangle, "x", cg_cRectangle_get_x, MRB_ARGS_NONE());
   mrb_define_method(
-    mrb, cg_cRectangle, "x=", cg_cRectangle_set_x, MRB_ARGS_REQ(1));
-  mrb_define_method(
     mrb, cg_cRectangle, "y", cg_cRectangle_get_y, MRB_ARGS_NONE());
   mrb_define_method(
-    mrb, cg_cRectangle, "y=", cg_cRectangle_set_y, MRB_ARGS_REQ(1));
-  mrb_define_method(
     mrb, cg_cRectangle, "width", cg_cRectangle_get_width, MRB_ARGS_NONE());
   mrb_define_method(
-    mrb, cg_cRectangle, "width=", cg_cRectangle_set_width, MRB_ARGS_REQ(1));
-  mrb_define_method(
     mrb, cg_cRectangle, "height", cg_cRectangle_get_height, MRB_ARGS_NONE());
+
+  mrb_define_method(
+    mrb, cg_cRectangle, "x=", cg_cRectangle_set_x, MRB_ARGS_REQ(1));
+  mrb_define_method(
+    mrb, cg_cRectangle, "y=", cg_cRectangle_set_y, MRB_ARGS_REQ(1));
+  mrb_define_method(
+    mrb, cg_cRectangle, "width=", cg_cRectangle_set_width, MRB_ARGS_REQ(1));
   mrb_define_method(
     mrb, cg_cRectangle, "height=", cg_cRectangle_set_height, MRB_ARGS_REQ(1));
+
+  mrb_define_method(
+    mrb, cg_cRectangle, "xy", cg_cRectangle_get_xy, MRB_ARGS_NONE());
+  mrb_define_method(
+    mrb, cg_cRectangle, "width_height", cg_cRectangle_get_width_height,
+    MRB_ARGS_NONE());
+
+  mrb_define_method(
+    mrb, cg_cRectangle, "set_xy", cg_cRectangle_set_xy, MRB_ARGS_REQ(2));
+  mrb_define_method(
+    mrb, cg_cRectangle, "set_width_height", cg_cRectangle_set_width_height,
+    MRB_ARGS_REQ(2));
+
   mrb_define_method(
     mrb, cg_cRectangle, "align_vertically?", cg_cRectangle_align_vertically,
     MRB_ARGS_REQ(1));