Commit: 27b6a2155f0c87345d880ea239c3c0f709445ed4 - mruby_tad (git)

TADや、その他BTRON基本データ型を扱うmrbgemの開発用

Commit MetaInfo

• Révision: 27b6a2155f0c87345d880ea239c3c0f709445ed4 (tree)
• l'heure: 2017-06-23 17:46:36
• Auteur: ornse01 <ornse01@user...>
• Commiter: ornse01

Message de Log

rouphly coding.

Change Summary

• modified: src/tad.c

Modification

--- a/src/tad.c
+++ b/src/tad.c
@@ -1,8 +1,192 @@
 #include "mruby.h"
 
+const static struct mrb_data_type mrb_PNT_type = { "PNT", mrb_free };
+
+LOCAL mrb_value
+mrb_PNT_init(mrb_state *mrb, mrb_value self)
+{
+	PNT *p, arg = {0, 0};
+	mrb_int x = 0, y = 0, argc;
+
+	argc = mrb_get_args(mrb, "|ii", &x, &y);
+	if (argc >= 1) {
+		if (x < -32768 || 32767 < x) {
+			mrb_raise(mrb, E_ARGUMENT_ERROR, "Too large value.");
+		}
+		arg.x = x;
+	}
+	if (argc >= 2) {
+		if (y < -32768 || 32767 < y) {
+			mrb_raise(mrb, E_ARGUMENT_ERROR, "Too large value.");
+		}
+		arg.y = y;
+	}
+
+	p = (PNT *)mrb_malloc(mrb, sizeof(PNT));
+	p->x = arg.x;
+	p->y = arg.y;
+
+	DATA_TYPE(self) = &mrb_PNT_type;
+	DATA_PTR(self) = h;
+
+	return self;
+}
+
+LOCAL mrb_value
+mrb_PNT_get_x(mrb_state *mrb, mrb_value self)
+{
+	PNT *p;
+
+	p = DATA_PTR(self);
+	return mrb_fixnum_value(p->x);
+}
+
+LOCAL mrb_value
+mrb_PNT_get_y(mrb_state *mrb, mrb_value self)
+{
+	PNT *p;
+
+	p = DATA_PTR(self);
+	return mrb_fixnum_value(p->y);
+}
+
+LOCAL mrb_value
+mrb_PNT_set_x(mrb_state *mrb, mrb_value self)
+{
+	PNT *p;
+	mrb_int num;
+
+	mrb_get_args(mrb, "i", &num);
+	p = DATA_PTR(self);
+	p->x = num;
+	return mrb_fixnum_value(num);
+}
+
+LOCAL mrb_value
+mrb_PNT_set_y(mrb_state *mrb, mrb_value self)
+{
+	PNT *p;
+	mrb_int num;
+
+	mrb_get_args(mrb, "i", &num);
+	p = DATA_PTR(self);
+	p->y = num;
+	return mrb_fixnum_value(num);
+}
+
+LOCAL mrb_value
+mrb_PNT_init_copy(mrb_state *mrb, mrb_value copy)
+{
+	mrb_value src;
+
+	mrb_get_args(mrb, "o", &src);
+	if (mrb_obj_equal(mrb, copy, src)) {
+		return copy;
+	}
+	if (!mrb_obj_is_instance_of(mrb, src, mrb_obj_class(mrb, copy))) {
+		mrb_raise(mrb, E_TYPE_ERROR, "wrong argument class");
+	}
+	if (!DATA_PTR(copy)) {
+		DATA_PTR(copy) = (struct hoge *)mrb_malloc(mrb, sizeof(struct hoge));
+		DATA_TYPE(copy) = &mrb_PNT_type;
+	}
+	*(PNT*)DATA_PTR(copy) = *(PNT*)DATA_PTR(src);
+
+	return copy;
+}
+
+LOCAL mrb_value
+mrb_PNT_equal(mrb_state *mrb, mrb_value self)
+{
+	mrb_value src;
+	PNT *self_real, *other_real;
+
+	mrb_get_args(mrb, "o", &other);
+	if (mrb_obj_equal(mrb, self, other)) {
+		return mrb_bool_value(FALSE);
+	}
+	if (!mrb_obj_is_instance_of(mrb, other, mrb_obj_class(mrb, copy))) {
+		return mrb_bool_value(FALSE);
+	}
+	if (!DATA_PTR(other)) {
+		return mrb_bool_value(FALSE);
+	}
+
+	self_real = DATA_PTR(self);
+	other_real = DATA_PTR(other);
+	if ((self_real->x == other_real->x) && (self_real->y == other_real.y)) {
+		return mrb_bool_value(TRUE);
+	}
+	return mrb_bool_value(FALSE);
+}
+
+LOCAL mrb_value
+mrb_PNT_plus(mrb_state *mrb, mrb_value self)
+{
+	struct RClass *cls;
+	mrb_value src, result;
+	PNT *self_real, *other_real;
+
+	cls = mrb_obj_class(mrb, self);
+	result = mrb_obj_value(cls);
+
+	mrb_get_args(mrb, "o", &other);
+	if (!mrb_obj_is_instance_of(mrb, other, cls)) {
+		mrb_raise(mrb, E_TYPE_ERROR, "wrong argument class");
+	}
+	if (!DATA_PTR(other)) {
+		mrb_raise(mrb, E_TYPE_ERROR, "wrong argument value");
+	}
+
+	self_real = DATA_PTR(self);
+	other_real = DATA_PTR(other);
+
+	return mrb_funcall(mrb, result, "new", 2,
+					   mrb_fixnum_value(self_real->x + other_real->x),
+					   mrb_fixnum_value(self_real->y + other_real->y));
+}
+
+LOCAL mrb_value
+mrb_PNT_minus(mrb_state *mrb, mrb_value self)
+{
+	struct RClass *cls;
+	mrb_value src, result;
+	PNT *self_real, *other_real;
+
+	cls = mrb_obj_class(mrb, self);
+	result = mrb_obj_value(cls);
+
+	mrb_get_args(mrb, "o", &other);
+	if (!mrb_obj_is_instance_of(mrb, other, cls)) {
+		mrb_raise(mrb, E_TYPE_ERROR, "wrong argument class");
+	}
+	if (!DATA_PTR(other)) {
+		mrb_raise(mrb, E_TYPE_ERROR, "wrong argument value");
+	}
+
+	self_real = DATA_PTR(self);
+	other_real = DATA_PTR(other);
+
+	return mrb_funcall(mrb, result, "new", 2,
+					   mrb_fixnum_value(self_real->x - other_real->x),
+					   mrb_fixnum_value(self_real->y - other_real->y));
+}
+
 void
 mrb_mruby_tad_gem_init(mrb_state* mrb)
 {
+	struct RClass *cls;
+
+	cls = mrb_define_class(mrb, "PNT", mrb->object_class);
+	MRB_SET_INSTANCE_TT(cls, MRB_TT_DATA);
+	mrb_define_method(mrb, cls, "initialize", mrb_PNT_init, MRB_ARGS_OPT(2));
+	mrb_define_method(mrb, cls, "x", mrb_PNT_get_x, MRB_ARGS_NONE());
+	mrb_define_method(mrb, cls, "y", mrb_PNT_get_y, MRB_ARGS_NONE());
+	mrb_define_method(mrb, cls, "put", mrb_PNT_set_x, MRB_ARGS_REQ(1));
+	mrb_define_method(mrb, cls, "initialize_copy", mrb_PNT_init_copy, MRB_ARGS_REQ(1));
+	mrb_define_method(mrb, cls, "==", mrb_PNT_equal, MRB_ARGS_REQ(1));
+	mrb_define_method(mrb, cls, "+", mrb_PNT_plus, MRB_ARGS_REQ(1));
+	mrb_define_method(mrb, cls, "-", mrb_PNT_minus, MRB_ARGS_REQ(1));
 }
 
 void