summaryrefslogtreecommitdiffhomepage
path: root/mrbgems/mruby-numeric-ext
diff options
context:
space:
mode:
Diffstat (limited to 'mrbgems/mruby-numeric-ext')
-rw-r--r--mrbgems/mruby-numeric-ext/mrbgem.rake5
-rw-r--r--mrbgems/mruby-numeric-ext/mrblib/numeric_ext.rb25
-rw-r--r--mrbgems/mruby-numeric-ext/src/numeric_ext.c102
-rw-r--r--mrbgems/mruby-numeric-ext/test/numeric.rb28
4 files changed, 160 insertions, 0 deletions
diff --git a/mrbgems/mruby-numeric-ext/mrbgem.rake b/mrbgems/mruby-numeric-ext/mrbgem.rake
new file mode 100644
index 000000000..6db7e589e
--- /dev/null
+++ b/mrbgems/mruby-numeric-ext/mrbgem.rake
@@ -0,0 +1,5 @@
+MRuby::Gem::Specification.new('mruby-numeric-ext') do |spec|
+ spec.license = 'MIT'
+ spec.author = 'mruby developers'
+ spec.summary = 'Numeric class extension'
+end
diff --git a/mrbgems/mruby-numeric-ext/mrblib/numeric_ext.rb b/mrbgems/mruby-numeric-ext/mrblib/numeric_ext.rb
new file mode 100644
index 000000000..f250538fe
--- /dev/null
+++ b/mrbgems/mruby-numeric-ext/mrblib/numeric_ext.rb
@@ -0,0 +1,25 @@
+module Integral
+ def div(other)
+ self.divmod(other)[0]
+ end
+
+ def zero?
+ self == 0
+ end
+
+ def nonzero?
+ if self == 0
+ nil
+ else
+ self
+ end
+ end
+
+ def positive?
+ self > 0
+ end
+
+ def negative?
+ self < 0
+ end
+end
diff --git a/mrbgems/mruby-numeric-ext/src/numeric_ext.c b/mrbgems/mruby-numeric-ext/src/numeric_ext.c
new file mode 100644
index 000000000..1d6a07769
--- /dev/null
+++ b/mrbgems/mruby-numeric-ext/src/numeric_ext.c
@@ -0,0 +1,102 @@
+#include <limits.h>
+#include <mruby.h>
+
+static inline mrb_int
+to_int(mrb_value x)
+{
+ double f;
+
+ if (mrb_fixnum_p(x)) return mrb_fixnum(x);
+ f = mrb_float(x);
+ return (mrb_int)f;
+}
+
+/*
+ * Document-method: Integer#chr
+ * call-seq:
+ * int.chr -> string
+ *
+ * Returns a string containing the character represented by the +int+'s value
+ * according to +encoding+.
+ *
+ * 65.chr #=> "A"
+ * 230.chr #=> "\xE6"
+ */
+static mrb_value
+mrb_int_chr(mrb_state *mrb, mrb_value x)
+{
+ mrb_int chr;
+ char c;
+
+ chr = to_int(x);
+ if (chr >= (1 << CHAR_BIT)) {
+ mrb_raisef(mrb, E_RANGE_ERROR, "%S out of char range", x);
+ }
+ c = (char)chr;
+
+ return mrb_str_new(mrb, &c, 1);
+}
+
+/*
+ * call-seq:
+ * int.allbits?(mask) -> true or false
+ *
+ * Returns +true+ if all bits of <code>+int+ & +mask+</code> are 1.
+ */
+static mrb_value
+mrb_int_allbits(mrb_state *mrb, mrb_value self)
+{
+ mrb_int n, m;
+
+ n = to_int(self);
+ mrb_get_args(mrb, "i", &m);
+ return mrb_bool_value((n & m) == m);
+}
+
+/*
+ * call-seq:
+ * int.anybits?(mask) -> true or false
+ *
+ * Returns +true+ if any bits of <code>+int+ & +mask+</code> are 1.
+ */
+static mrb_value
+mrb_int_anybits(mrb_state *mrb, mrb_value self)
+{
+ mrb_int n, m;
+
+ n = to_int(self);
+ mrb_get_args(mrb, "i", &m);
+ return mrb_bool_value((n & m) != 0);
+}
+
+/*
+ * call-seq:
+ * int.nobits?(mask) -> true or false
+ *
+ * Returns +true+ if no bits of <code>+int+ & +mask+</code> are 1.
+ */
+static mrb_value
+mrb_int_nobits(mrb_state *mrb, mrb_value self)
+{
+ mrb_int n, m;
+
+ n = to_int(self);
+ mrb_get_args(mrb, "i", &m);
+ return mrb_bool_value((n & m) == 0);
+}
+
+void
+mrb_mruby_numeric_ext_gem_init(mrb_state* mrb)
+{
+ struct RClass *i = mrb_module_get(mrb, "Integral");
+
+ mrb_define_method(mrb, i, "chr", mrb_int_chr, MRB_ARGS_NONE());
+ mrb_define_method(mrb, i, "allbits?", mrb_int_allbits, MRB_ARGS_REQ(1));
+ mrb_define_method(mrb, i, "anybits?", mrb_int_anybits, MRB_ARGS_REQ(1));
+ mrb_define_method(mrb, i, "nobits?", mrb_int_nobits, MRB_ARGS_REQ(1));
+}
+
+void
+mrb_mruby_numeric_ext_gem_final(mrb_state* mrb)
+{
+}
diff --git a/mrbgems/mruby-numeric-ext/test/numeric.rb b/mrbgems/mruby-numeric-ext/test/numeric.rb
new file mode 100644
index 000000000..6ea0c14e7
--- /dev/null
+++ b/mrbgems/mruby-numeric-ext/test/numeric.rb
@@ -0,0 +1,28 @@
+##
+# Numeric(Ext) Test
+
+assert('Integer#chr') do
+ assert_equal("A", 65.chr)
+ assert_equal("B", 0x42.chr)
+
+ # multibyte encoding (not support yet)
+ assert_raise(RangeError) { 256.chr }
+end
+
+assert('Integer#div') do
+ assert_equal 52, 365.div(7)
+end
+
+assert('Float#div') do
+ assert_float 52, 365.2425.div(7)
+end if class_defined?("Float")
+
+assert('Integer#zero?') do
+ assert_equal true, 0.zero?
+ assert_equal false, 1.zero?
+end
+
+assert('Integer#nonzero?') do
+ assert_equal nil, 0.nonzero?
+ assert_equal 1000, 1000.nonzero?
+end