mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
6204e0804b
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@66888 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
83 lines
2 KiB
C
83 lines
2 KiB
C
#include "ruby.h"
|
|
#include "rubyspec.h"
|
|
|
|
#include <string.h>
|
|
#include <stdarg.h>
|
|
|
|
#include <ruby/st.h>
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
|
|
#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
|
|
}
|
|
|
|
void Init_st_spec(void) {
|
|
VALUE cls = rb_define_class("CApiStSpecs", rb_cObject);
|
|
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);
|
|
}
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|