diff options
Diffstat (limited to 'ext/-test-/st/numhash')
| -rw-r--r-- | ext/-test-/st/numhash/depend | 163 | ||||
| -rw-r--r-- | ext/-test-/st/numhash/extconf.rb | 2 | ||||
| -rw-r--r-- | ext/-test-/st/numhash/numhash.c | 137 |
3 files changed, 302 insertions, 0 deletions
diff --git a/ext/-test-/st/numhash/depend b/ext/-test-/st/numhash/depend new file mode 100644 index 0000000000..18320d55f5 --- /dev/null +++ b/ext/-test-/st/numhash/depend @@ -0,0 +1,163 @@ +# AUTOGENERATED DEPENDENCIES START +numhash.o: $(RUBY_EXTCONF_H) +numhash.o: $(arch_hdrdir)/ruby/config.h +numhash.o: $(hdrdir)/ruby.h +numhash.o: $(hdrdir)/ruby/assert.h +numhash.o: $(hdrdir)/ruby/backward.h +numhash.o: $(hdrdir)/ruby/backward/2/assume.h +numhash.o: $(hdrdir)/ruby/backward/2/attributes.h +numhash.o: $(hdrdir)/ruby/backward/2/bool.h +numhash.o: $(hdrdir)/ruby/backward/2/inttypes.h +numhash.o: $(hdrdir)/ruby/backward/2/limits.h +numhash.o: $(hdrdir)/ruby/backward/2/long_long.h +numhash.o: $(hdrdir)/ruby/backward/2/stdalign.h +numhash.o: $(hdrdir)/ruby/backward/2/stdarg.h +numhash.o: $(hdrdir)/ruby/defines.h +numhash.o: $(hdrdir)/ruby/intern.h +numhash.o: $(hdrdir)/ruby/internal/abi.h +numhash.o: $(hdrdir)/ruby/internal/anyargs.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/char.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/double.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/fixnum.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/gid_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/int.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/intptr_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/long.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/long_long.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/mode_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/off_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/pid_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/short.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/size_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/st_data_t.h +numhash.o: $(hdrdir)/ruby/internal/arithmetic/uid_t.h +numhash.o: $(hdrdir)/ruby/internal/assume.h +numhash.o: $(hdrdir)/ruby/internal/attr/alloc_size.h +numhash.o: $(hdrdir)/ruby/internal/attr/artificial.h +numhash.o: $(hdrdir)/ruby/internal/attr/cold.h +numhash.o: $(hdrdir)/ruby/internal/attr/const.h +numhash.o: $(hdrdir)/ruby/internal/attr/constexpr.h +numhash.o: $(hdrdir)/ruby/internal/attr/deprecated.h +numhash.o: $(hdrdir)/ruby/internal/attr/diagnose_if.h +numhash.o: $(hdrdir)/ruby/internal/attr/enum_extensibility.h +numhash.o: $(hdrdir)/ruby/internal/attr/error.h +numhash.o: $(hdrdir)/ruby/internal/attr/flag_enum.h +numhash.o: $(hdrdir)/ruby/internal/attr/forceinline.h +numhash.o: $(hdrdir)/ruby/internal/attr/format.h +numhash.o: $(hdrdir)/ruby/internal/attr/maybe_unused.h +numhash.o: $(hdrdir)/ruby/internal/attr/noalias.h +numhash.o: $(hdrdir)/ruby/internal/attr/nodiscard.h +numhash.o: $(hdrdir)/ruby/internal/attr/noexcept.h +numhash.o: $(hdrdir)/ruby/internal/attr/noinline.h +numhash.o: $(hdrdir)/ruby/internal/attr/nonnull.h +numhash.o: $(hdrdir)/ruby/internal/attr/noreturn.h +numhash.o: $(hdrdir)/ruby/internal/attr/packed_struct.h +numhash.o: $(hdrdir)/ruby/internal/attr/pure.h +numhash.o: $(hdrdir)/ruby/internal/attr/restrict.h +numhash.o: $(hdrdir)/ruby/internal/attr/returns_nonnull.h +numhash.o: $(hdrdir)/ruby/internal/attr/warning.h +numhash.o: $(hdrdir)/ruby/internal/attr/weakref.h +numhash.o: $(hdrdir)/ruby/internal/cast.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is/apple.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is/clang.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is/gcc.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is/intel.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is/msvc.h +numhash.o: $(hdrdir)/ruby/internal/compiler_is/sunpro.h +numhash.o: $(hdrdir)/ruby/internal/compiler_since.h +numhash.o: $(hdrdir)/ruby/internal/config.h +numhash.o: $(hdrdir)/ruby/internal/constant_p.h +numhash.o: $(hdrdir)/ruby/internal/core.h +numhash.o: $(hdrdir)/ruby/internal/core/rarray.h +numhash.o: $(hdrdir)/ruby/internal/core/rbasic.h +numhash.o: $(hdrdir)/ruby/internal/core/rbignum.h +numhash.o: $(hdrdir)/ruby/internal/core/rclass.h +numhash.o: $(hdrdir)/ruby/internal/core/rdata.h +numhash.o: $(hdrdir)/ruby/internal/core/rfile.h +numhash.o: $(hdrdir)/ruby/internal/core/rhash.h +numhash.o: $(hdrdir)/ruby/internal/core/robject.h +numhash.o: $(hdrdir)/ruby/internal/core/rregexp.h +numhash.o: $(hdrdir)/ruby/internal/core/rstring.h +numhash.o: $(hdrdir)/ruby/internal/core/rstruct.h +numhash.o: $(hdrdir)/ruby/internal/core/rtypeddata.h +numhash.o: $(hdrdir)/ruby/internal/ctype.h +numhash.o: $(hdrdir)/ruby/internal/dllexport.h +numhash.o: $(hdrdir)/ruby/internal/dosish.h +numhash.o: $(hdrdir)/ruby/internal/error.h +numhash.o: $(hdrdir)/ruby/internal/eval.h +numhash.o: $(hdrdir)/ruby/internal/event.h +numhash.o: $(hdrdir)/ruby/internal/fl_type.h +numhash.o: $(hdrdir)/ruby/internal/gc.h +numhash.o: $(hdrdir)/ruby/internal/glob.h +numhash.o: $(hdrdir)/ruby/internal/globals.h +numhash.o: $(hdrdir)/ruby/internal/has/attribute.h +numhash.o: $(hdrdir)/ruby/internal/has/builtin.h +numhash.o: $(hdrdir)/ruby/internal/has/c_attribute.h +numhash.o: $(hdrdir)/ruby/internal/has/cpp_attribute.h +numhash.o: $(hdrdir)/ruby/internal/has/declspec_attribute.h +numhash.o: $(hdrdir)/ruby/internal/has/extension.h +numhash.o: $(hdrdir)/ruby/internal/has/feature.h +numhash.o: $(hdrdir)/ruby/internal/has/warning.h +numhash.o: $(hdrdir)/ruby/internal/intern/array.h +numhash.o: $(hdrdir)/ruby/internal/intern/bignum.h +numhash.o: $(hdrdir)/ruby/internal/intern/class.h +numhash.o: $(hdrdir)/ruby/internal/intern/compar.h +numhash.o: $(hdrdir)/ruby/internal/intern/complex.h +numhash.o: $(hdrdir)/ruby/internal/intern/cont.h +numhash.o: $(hdrdir)/ruby/internal/intern/dir.h +numhash.o: $(hdrdir)/ruby/internal/intern/enum.h +numhash.o: $(hdrdir)/ruby/internal/intern/enumerator.h +numhash.o: $(hdrdir)/ruby/internal/intern/error.h +numhash.o: $(hdrdir)/ruby/internal/intern/eval.h +numhash.o: $(hdrdir)/ruby/internal/intern/file.h +numhash.o: $(hdrdir)/ruby/internal/intern/hash.h +numhash.o: $(hdrdir)/ruby/internal/intern/io.h +numhash.o: $(hdrdir)/ruby/internal/intern/load.h +numhash.o: $(hdrdir)/ruby/internal/intern/marshal.h +numhash.o: $(hdrdir)/ruby/internal/intern/numeric.h +numhash.o: $(hdrdir)/ruby/internal/intern/object.h +numhash.o: $(hdrdir)/ruby/internal/intern/parse.h +numhash.o: $(hdrdir)/ruby/internal/intern/proc.h +numhash.o: $(hdrdir)/ruby/internal/intern/process.h +numhash.o: $(hdrdir)/ruby/internal/intern/random.h +numhash.o: $(hdrdir)/ruby/internal/intern/range.h +numhash.o: $(hdrdir)/ruby/internal/intern/rational.h +numhash.o: $(hdrdir)/ruby/internal/intern/re.h +numhash.o: $(hdrdir)/ruby/internal/intern/ruby.h +numhash.o: $(hdrdir)/ruby/internal/intern/select.h +numhash.o: $(hdrdir)/ruby/internal/intern/select/largesize.h +numhash.o: $(hdrdir)/ruby/internal/intern/set.h +numhash.o: $(hdrdir)/ruby/internal/intern/signal.h +numhash.o: $(hdrdir)/ruby/internal/intern/sprintf.h +numhash.o: $(hdrdir)/ruby/internal/intern/string.h +numhash.o: $(hdrdir)/ruby/internal/intern/struct.h +numhash.o: $(hdrdir)/ruby/internal/intern/thread.h +numhash.o: $(hdrdir)/ruby/internal/intern/time.h +numhash.o: $(hdrdir)/ruby/internal/intern/variable.h +numhash.o: $(hdrdir)/ruby/internal/intern/vm.h +numhash.o: $(hdrdir)/ruby/internal/interpreter.h +numhash.o: $(hdrdir)/ruby/internal/iterator.h +numhash.o: $(hdrdir)/ruby/internal/memory.h +numhash.o: $(hdrdir)/ruby/internal/method.h +numhash.o: $(hdrdir)/ruby/internal/module.h +numhash.o: $(hdrdir)/ruby/internal/newobj.h +numhash.o: $(hdrdir)/ruby/internal/scan_args.h +numhash.o: $(hdrdir)/ruby/internal/special_consts.h +numhash.o: $(hdrdir)/ruby/internal/static_assert.h +numhash.o: $(hdrdir)/ruby/internal/stdalign.h +numhash.o: $(hdrdir)/ruby/internal/stdbool.h +numhash.o: $(hdrdir)/ruby/internal/stdckdint.h +numhash.o: $(hdrdir)/ruby/internal/symbol.h +numhash.o: $(hdrdir)/ruby/internal/value.h +numhash.o: $(hdrdir)/ruby/internal/value_type.h +numhash.o: $(hdrdir)/ruby/internal/variable.h +numhash.o: $(hdrdir)/ruby/internal/warning_push.h +numhash.o: $(hdrdir)/ruby/internal/xmalloc.h +numhash.o: $(hdrdir)/ruby/missing.h +numhash.o: $(hdrdir)/ruby/ruby.h +numhash.o: $(hdrdir)/ruby/st.h +numhash.o: $(hdrdir)/ruby/subst.h +numhash.o: numhash.c +# AUTOGENERATED DEPENDENCIES END diff --git a/ext/-test-/st/numhash/extconf.rb b/ext/-test-/st/numhash/extconf.rb new file mode 100644 index 0000000000..27d28a0a77 --- /dev/null +++ b/ext/-test-/st/numhash/extconf.rb @@ -0,0 +1,2 @@ +# frozen_string_literal: false +create_makefile("-test-/st/numhash") diff --git a/ext/-test-/st/numhash/numhash.c b/ext/-test-/st/numhash/numhash.c new file mode 100644 index 0000000000..7e8d5d9fe2 --- /dev/null +++ b/ext/-test-/st/numhash/numhash.c @@ -0,0 +1,137 @@ +#include <ruby.h> +#include <ruby/st.h> + +static void +numhash_free(void *ptr) +{ + if (ptr) st_free_table(ptr); +} + +static size_t +numhash_memsize(const void *ptr) +{ + return st_memsize(ptr); +} + +static const rb_data_type_t numhash_type = { + "numhash", + {0, numhash_free, numhash_memsize,}, + 0, 0, + RUBY_TYPED_FREE_IMMEDIATELY|RUBY_TYPED_WB_PROTECTED, +}; + +static VALUE +numhash_alloc(VALUE klass) +{ + return TypedData_Wrap_Struct(klass, &numhash_type, 0); +} + +static VALUE +numhash_init(VALUE self) +{ + st_table *tbl = (st_table *)Check_TypedStruct(self, &numhash_type); + if (tbl) st_free_table(tbl); + DATA_PTR(self) = st_init_numtable(); + return self; +} + +static VALUE +numhash_aref(VALUE self, VALUE key) +{ + st_data_t data; + st_table *tbl = (st_table *)Check_TypedStruct(self, &numhash_type); + if (!SPECIAL_CONST_P(key)) rb_raise(rb_eArgError, "not a special const"); + if (st_lookup(tbl, (st_data_t)key, &data)) + return (VALUE)data; + return Qnil; +} + +static VALUE +numhash_aset(VALUE self, VALUE key, VALUE data) +{ + st_table *tbl = (st_table *)Check_TypedStruct(self, &numhash_type); + if (!SPECIAL_CONST_P(key)) rb_raise(rb_eArgError, "not a special const"); + if (!SPECIAL_CONST_P(data)) rb_raise(rb_eArgError, "not a special const"); + st_insert(tbl, (st_data_t)key, (st_data_t)data); + return self; +} + +static int +numhash_i(st_data_t key, st_data_t value, st_data_t arg, int _) +{ + VALUE ret; + ret = rb_yield_values(3, (VALUE)key, (VALUE)value, (VALUE)arg); + if (ret == Qtrue) return ST_CHECK; + return ST_CONTINUE; +} + +static VALUE +numhash_each(VALUE self) +{ + st_table *table = (st_table *)Check_TypedStruct(self, &numhash_type); + st_data_t data = (st_data_t)self; + return st_foreach_check(table, numhash_i, data, data) ? Qtrue : Qfalse; +} + +static int +update_func(st_data_t *key, st_data_t *value, st_data_t arg, int existing) +{ + VALUE ret = rb_yield_values(existing ? 2 : 1, (VALUE)*key, (VALUE)*value); + switch (ret) { + case Qfalse: + return ST_STOP; + case Qnil: + return ST_DELETE; + default: + *value = ret; + return ST_CONTINUE; + } +} + +static VALUE +numhash_update(VALUE self, VALUE key) +{ + st_table *table = (st_table *)Check_TypedStruct(self, &numhash_type); + if (st_update(table, (st_data_t)key, update_func, 0)) + return Qtrue; + else + return Qfalse; +} + +#if SIZEOF_LONG == SIZEOF_VOIDP +# define ST2NUM(x) ULONG2NUM(x) +#elif SIZEOF_LONG_LONG == SIZEOF_VOIDP +# define ST2NUM(x) ULL2NUM(x) +#endif + +static VALUE +numhash_size(VALUE self) +{ + st_table *table = (st_table *)Check_TypedStruct(self, &numhash_type); + return ST2NUM(table->num_entries); +} + +static VALUE +numhash_delete_safe(VALUE self, VALUE key) +{ + st_table *table = (st_table *)Check_TypedStruct(self, &numhash_type); + st_data_t val, k = (st_data_t)key; + if (st_delete_safe(table, &k, &val, (st_data_t)self)) { + return val; + } + return Qnil; +} + +void +Init_numhash(void) +{ + VALUE st = rb_define_class_under(rb_define_module("Bug"), "StNumHash", rb_cObject); + rb_define_alloc_func(st, numhash_alloc); + rb_define_method(st, "initialize", numhash_init, 0); + rb_define_method(st, "[]", numhash_aref, 1); + rb_define_method(st, "[]=", numhash_aset, 2); + rb_define_method(st, "each", numhash_each, 0); + rb_define_method(st, "update", numhash_update, 1); + rb_define_method(st, "size", numhash_size, 0); + rb_define_method(st, "delete_safe", numhash_delete_safe, 1); +} |
