LoadSo/array.c at master · wanabe/LoadSo · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#include "load_so.h"

#define FL_UNSET_EMBED(ary) FL_UNSET((ary), RARRAY_EMBED_FLAG|RARRAY_EMBED_LEN_MASK)
#define ARY_SET_CAPA(ary, n) do { \
    RARRAY(ary)->as.heap.aux.capa = (n); \
} while (0)
#define ARY_SET_PTR(ary, p) do { \
    RARRAY(ary)->as.heap.ptr = (p); \
} while (0)
#define ARY_SET_EMBED_LEN(ary, n) do { \
    long tmp_n = (n); \
    RBASIC(ary)->flags &= ~RARRAY_EMBED_LEN_MASK; \
    RBASIC(ary)->flags |= (tmp_n) << RARRAY_EMBED_LEN_SHIFT; \
} while (0)
#define ARY_SET_HEAP_LEN(ary, n) do { \
    RARRAY(ary)->as.heap.len = (n); \
} while (0)

static VALUE (*rb_ary_push_m)(int, VALUE*, VALUE);
static VALUE (*rb_ary_join_m)(int, VALUE*, VALUE);
static VALUE (*rb_ary_clear_)(VALUE);
static VALUE (*rb_ary_concat_)(VALUE, VALUE);
static VALUE (*rb_ary_unshift_m)(int, VALUE*, VALUE);
static VALUE (*rb_ary_delete_at_m)(VALUE, VALUE);
static VALUE (*rb_ary_aset)(int, VALUE*, VALUE);
static VALUE (*rb_ary_delete_)(VALUE, VALUE);


VALUE rb_ary_new() {
  return rb_class_new_instance(0, NULL, rb_cArray);
}

VALUE rb_ary_new_with_len(long len) {
  VALUE ary = INT2FIX(len);
  VALUE block = push_block();
  ary = rb_class_new_instance(1, &ary, rb_cArray);
  pop_block(block);
  return ary;
}

VALUE rb_ary_new3(long n, ...) {
  va_list ar;
  VALUE ary;
  long i;
  ary = rb_ary_new_with_len(n);

  va_start(ar, n);
  for (i=0; i<n; i++) {
    RARRAY_PTR(ary)[i] = va_arg(ar, VALUE);
  }
  va_end(ar);

  return ary;
}

VALUE rb_ary_new4(long n, const VALUE *elts) {
  VALUE ary;

  ary = rb_ary_new_with_len(n);
  if (n > 0 && elts) {
    memcpy(RARRAY_PTR(ary), elts, sizeof(VALUE) * n);
  }
  return ary;
}

VALUE rb_ary_entry(VALUE ary, long offset) {
  if(offset < 0) {
    offset += RARRAY_LEN(ary);
  }
  if(offset < 0 || offset >= RARRAY_LEN(ary)) {
    return Qnil;
  }

  return RARRAY_PTR(ary)[offset];
}

VALUE rb_ary_push(VALUE ary, VALUE item) {
  return rb_ary_push_m(1, &item, ary);
}

VALUE rb_ary_join(VALUE ary, VALUE sep) {
  rb_ary_join_m(1, &sep, ary);
  return rb_ary_join_m(1, &sep, ary);
}

VALUE rb_ary_clear(VALUE ary) {
  return rb_ary_clear_(ary);
}

VALUE rb_ary_concat(VALUE x, VALUE y) {
  return rb_ary_concat_(x, y);
}

VALUE rb_ary_unshift(VALUE ary, VALUE item) {
  return rb_ary_unshift_m(1, &item, ary);
}

VALUE rb_ary_delete_at(VALUE ary, long pos) {
  return rb_ary_delete_at_m(ary, INT2FIX(pos));
}

void rb_ary_store(VALUE ary, long idx, VALUE val) {
  VALUE argv[2] = {INT2FIX(idx), val};
  rb_ary_aset(2, argv, ary);
}

VALUE rb_ary_delete(VALUE ary, VALUE item) {
  return rb_ary_delete_(ary, item);
}

void Init_Array() {
  rb_ary_push_m = get_instance_method(rb_cArray, "push");
  rb_ary_join_m = get_instance_method(rb_cArray, "join");
  rb_ary_clear_ = get_instance_method(rb_cArray, "clear");
  rb_ary_concat_ = get_instance_method(rb_cArray, "concat");
  rb_ary_unshift_m = get_instance_method(rb_cArray, "unshift");
  rb_ary_delete_at_m = get_instance_method(rb_cArray, "delete_at");
  rb_ary_aset = get_instance_method(rb_cArray, "[]=");
  rb_ary_delete_ = get_instance_method(rb_cArray, "delete");
}