From 993d033617eaeb582e0e5f0525fa51e1c83893da Mon Sep 17 00:00:00 2001
From: Frederico Linhares <fred@linhares.blue>
Date: Thu, 15 Dec 2022 15:38:13 -0300
Subject: feat Add new methods to Rectangle and Vector4D

---
 src/rectangle.cpp | 80 ++++++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 74 insertions(+), 6 deletions(-)

(limited to 'src/rectangle.cpp')

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"
@@ -135,6 +137,58 @@ cg_cRectangle_set_height(mrb_state *mrb, mrb_value self)
   return mrb_float_value(mrb, (*ptr)->height);
 }
 
+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)
 {
@@ -205,20 +259,34 @@ cg_rectangle_init(mrb_state *mrb)
     MRB_ARGS_REQ(4));
   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));
-- 
cgit v1.2.3