diff options
Diffstat (limited to 'ext')
-rw-r--r-- | ext/-test-/arith_seq/extract/extconf.rb | 2 | ||||
-rw-r--r-- | ext/-test-/arith_seq/extract/extract.c | 27 |
2 files changed, 29 insertions, 0 deletions
diff --git a/ext/-test-/arith_seq/extract/extconf.rb b/ext/-test-/arith_seq/extract/extconf.rb new file mode 100644 index 0000000000..9c368bf50e --- /dev/null +++ b/ext/-test-/arith_seq/extract/extconf.rb @@ -0,0 +1,2 @@ +# frozen_string_literal: false +create_makefile("-test-/arith_seq/extract") diff --git a/ext/-test-/arith_seq/extract/extract.c b/ext/-test-/arith_seq/extract/extract.c new file mode 100644 index 0000000000..602afa7ca1 --- /dev/null +++ b/ext/-test-/arith_seq/extract/extract.c @@ -0,0 +1,27 @@ +#include "ruby/ruby.h" + +static VALUE +arith_seq_s_extract(VALUE mod, VALUE obj) +{ + rb_arithmetic_sequence_components_t x; + VALUE ret; + int f, r; + + r = rb_arithmetic_sequence_extract(obj, &x); + + ret = rb_ary_new2(5); + rb_ary_store(ret, 0, r ? x.begin : Qnil); + rb_ary_store(ret, 1, r ? x.end : Qnil); + rb_ary_store(ret, 2, r ? x.step : Qnil); + rb_ary_store(ret, 3, r ? INT2FIX(x.exclude_end) : Qnil); + rb_ary_store(ret, 4, INT2FIX(r)); + + return ret; +} + +void +Init_extract(void) +{ + VALUE cArithSeq = rb_path2class("Enumerator::ArithmeticSequence"); + rb_define_singleton_method(cArithSeq, "__extract__", arith_seq_s_extract, 1); +} |