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
|
#include "ruby.h"
#include "rubyspec.h"
#include <string.h>
#include <stdarg.h>
#ifdef HAVE_RB_ST
#include <ruby/st.h>
#endif
#ifdef __cplusplus
extern "C" {
#endif
#ifdef HAVE_RB_ST
#if SIZEOF_LONG == SIZEOF_VOIDP
# define ST2NUM(x) ULONG2NUM(x)
#else
# define ST2NUM(x) ULL2NUM(x)
#endif
VALUE st_spec_st_init_numtable(VALUE self) {
st_table *tbl = st_init_numtable();
st_index_t entries = tbl->num_entries;
st_free_table(tbl);
return ST2NUM(entries);
}
VALUE st_spec_st_init_numtable_with_size(VALUE self) {
st_table *tbl = st_init_numtable_with_size(128);
st_index_t entries = tbl->num_entries;
st_free_table(tbl);
return ST2NUM(entries);
}
VALUE st_spec_st_insert(VALUE self) {
st_index_t entries;
st_table *tbl = st_init_numtable_with_size(128);
st_insert(tbl, 1, 1);
entries = tbl->num_entries;
st_free_table(tbl);
return ST2NUM(entries);
}
static int sum(st_data_t key, st_data_t value, st_data_t arg) {
*(int*)arg += (int)value;
return ST_CONTINUE;
}
VALUE st_spec_st_foreach(VALUE self) {
int total = 0;
st_table *tbl = st_init_numtable_with_size(128);
st_insert(tbl, 1, 3);
st_insert(tbl, 2, 4);
st_foreach(tbl, sum, (st_data_t)&total);
st_free_table(tbl);
return INT2FIX(total);
}
VALUE st_spec_st_lookup(VALUE self) {
st_data_t result = (st_data_t)0;
st_table *tbl = st_init_numtable_with_size(128);
st_insert(tbl, 7, 42);
st_insert(tbl, 2, 4);
st_lookup(tbl, (st_data_t)7, &result);
st_free_table(tbl);
#if SIZEOF_LONG == SIZEOF_VOIDP
return ULONG2NUM(result);
#else
return ULL2NUM(result);
#endif
}
#endif
void Init_st_spec(void) {
VALUE cls;
cls = rb_define_class("CApiStSpecs", rb_cObject);
#ifdef HAVE_RB_ST
rb_define_method(cls, "st_init_numtable", st_spec_st_init_numtable, 0);
rb_define_method(cls, "st_init_numtable_with_size", st_spec_st_init_numtable_with_size, 0);
rb_define_method(cls, "st_insert", st_spec_st_insert, 0);
rb_define_method(cls, "st_foreach", st_spec_st_foreach, 0);
rb_define_method(cls, "st_lookup", st_spec_st_lookup, 0);
#endif
}
#ifdef __cplusplus
}
#endif
|