summaryrefslogtreecommitdiff
path: root/spec/ruby/optional/capi/ext/enumerator_spec.c
blob: d9e34afd9692a11ed0cf35edb76cd2f436c1857d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include "ruby.h"
#include "rubyspec.h"

#ifdef __cplusplus
extern "C" {
#endif

#ifdef HAVE_RB_ENUMERATORIZE
VALUE enumerator_spec_rb_enumeratorize(int argc, VALUE *argv, VALUE self) {
  VALUE obj, meth, args;
  rb_scan_args(argc, argv, "2*", &obj, &meth, &args);
  return rb_enumeratorize(obj, meth, (int)RARRAY_LEN(args), RARRAY_PTR(args));
}
#endif

#ifdef HAVE_RB_ENUMERATORIZE_WITH_SIZE
VALUE enumerator_spec_size_fn(VALUE obj, VALUE args, VALUE anEnum) {
  return INT2NUM(7);
}

VALUE enumerator_spec_rb_enumeratorize_with_size(int argc, VALUE *argv, VALUE self) {
  VALUE obj, meth, args;
  rb_scan_args(argc, argv, "2*", &obj, &meth, &args);
  return rb_enumeratorize_with_size(obj, meth, (int)RARRAY_LEN(args), RARRAY_PTR(args), enumerator_spec_size_fn);
}
#endif

void Init_enumerator_spec(void) {
  VALUE cls;
  cls = rb_define_class("CApiEnumeratorSpecs", rb_cObject);

#ifdef HAVE_RB_ENUMERATORIZE
  rb_define_method(cls, "rb_enumeratorize", enumerator_spec_rb_enumeratorize, -1);
#endif
#ifdef HAVE_RB_ENUMERATORIZE_WITH_SIZE
  rb_define_method(cls, "rb_enumeratorize_with_size", enumerator_spec_rb_enumeratorize_with_size, -1);
#endif
}

#ifdef __cplusplus
}
#endif