summaryrefslogtreecommitdiff
path: root/ext/fiddle/memory_view.c
diff options
context:
space:
mode:
authorHiroshi SHIBATA <[email protected]>2025-01-24 16:23:16 +0900
committerHiroshi SHIBATA <[email protected]>2025-02-05 07:36:59 +0900
commit908529b7e0cf39bb1204588bd7f86c6bd40b8a00 (patch)
treeb1bcb3d15b22f04cbce25f0658935b5c5d218a17 /ext/fiddle/memory_view.c
parent91a10c07579f282a94e4b5830feaeca87f9a7dd3 (diff)
Migrate fiddle as bundled gems
Notes
Notes: Merged: https://github.com/ruby/ruby/pull/12616
Diffstat (limited to 'ext/fiddle/memory_view.c')
-rw-r--r--ext/fiddle/memory_view.c326
1 files changed, 0 insertions, 326 deletions
diff --git a/ext/fiddle/memory_view.c b/ext/fiddle/memory_view.c
deleted file mode 100644
index 5fd7b9b408..0000000000
--- a/ext/fiddle/memory_view.c
+++ /dev/null
@@ -1,326 +0,0 @@
-#include <fiddle.h>
-
-#ifdef HAVE_RUBY_MEMORY_VIEW_H
-
-#include <stdbool.h>
-#include <ruby/ruby.h>
-#include <ruby/encoding.h>
-#include <ruby/memory_view.h>
-
-#if SIZEOF_INTPTR_T == SIZEOF_LONG_LONG
-# define INTPTR2NUM LL2NUM
-# define UINTPTR2NUM ULL2NUM
-#elif SIZEOF_INTPTR_T == SIZEOF_LONG
-# define INTPTR2NUM LONG2NUM
-# define UINTPTR2NUM ULONG2NUM
-#else
-# define INTPTR2NUM INT2NUM
-# define UINTPTR2NUM UINT2NUM
-#endif
-
-VALUE rb_cMemoryView = Qnil;
-
-struct memview_data {
- rb_memory_view_t view;
- rb_memory_view_item_component_t *members;
- size_t n_members;
-};
-
-static void
-fiddle_memview_mark(void *ptr)
-{
- const struct memview_data *data = ptr;
- rb_gc_mark(data->view.obj);
-}
-
-static void
-fiddle_memview_release(struct memview_data *data)
-{
- if (NIL_P(data->view.obj)) return;
-
- rb_memory_view_release(&data->view);
- data->view.obj = Qnil;
- data->view.byte_size = 0;
- if (data->members) {
- xfree(data->members);
- data->members = NULL;
- data->n_members = 0;
- }
-}
-
-static void
-fiddle_memview_free(void *ptr)
-{
- struct memview_data *data = ptr;
- fiddle_memview_release(data);
- xfree(ptr);
-}
-
-static size_t
-fiddle_memview_memsize(const void *ptr)
-{
- const struct memview_data *data = ptr;
- return sizeof(*data) + sizeof(rb_memory_view_item_component_t)*data->n_members + (size_t)data->view.byte_size;
-}
-
-static const rb_data_type_t fiddle_memview_data_type = {
- .wrap_struct_name = "fiddle/memory_view",
- .function = {
- .dmark = fiddle_memview_mark,
- .dfree = fiddle_memview_free,
- .dsize = fiddle_memview_memsize,
- },
- .flags = FIDDLE_DEFAULT_TYPED_DATA_FLAGS,
-};
-
-static VALUE
-rb_fiddle_memview_s_allocate(VALUE klass)
-{
- struct memview_data *data;
- VALUE obj = TypedData_Make_Struct(klass, struct memview_data, &fiddle_memview_data_type, data);
- data->view.obj = Qnil;
- data->view.byte_size = 0;
- data->members = NULL;
- data->n_members = 0;
- return obj;
-}
-
-static VALUE
-rb_fiddle_memview_release(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- fiddle_memview_release(data);
- return Qnil;
-}
-
-static VALUE
-rb_fiddle_memview_s_export(VALUE klass, VALUE target)
-{
- ID id_new;
- CONST_ID(id_new, "new");
- VALUE memview = rb_funcall(klass, id_new, 1, target);
- return rb_ensure(rb_yield, memview, rb_fiddle_memview_release, memview);
-}
-
-static VALUE
-rb_fiddle_memview_initialize(VALUE obj, VALUE target)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (!rb_memory_view_get(target, &data->view, 0)) {
- data->view.obj = Qnil;
- rb_raise(rb_eArgError, "Unable to get a memory view from %+"PRIsVALUE, target);
- }
-
- return Qnil;
-}
-
-static VALUE
-rb_fiddle_memview_get_obj(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- return data->view.obj;
-}
-
-static VALUE
-rb_fiddle_memview_get_byte_size(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- return SSIZET2NUM(data->view.byte_size);
-}
-
-static VALUE
-rb_fiddle_memview_get_readonly(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- return data->view.readonly ? Qtrue : Qfalse;
-}
-
-static VALUE
-rb_fiddle_memview_get_format(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- return data->view.format == NULL ? Qnil : rb_str_new_cstr(data->view.format);
-}
-
-static VALUE
-rb_fiddle_memview_get_item_size(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- return SSIZET2NUM(data->view.item_size);
-}
-
-static VALUE
-rb_fiddle_memview_get_ndim(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- return SSIZET2NUM(data->view.ndim);
-}
-
-static VALUE
-rb_fiddle_memview_get_shape(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- if (data->view.shape == NULL) return Qnil;
-
- const ssize_t ndim = data->view.ndim;
- VALUE shape = rb_ary_new_capa(ndim);
- ssize_t i;
- for (i = 0; i < ndim; ++i) {
- rb_ary_push(shape, SSIZET2NUM(data->view.shape[i]));
- }
- return shape;
-}
-
-static VALUE
-rb_fiddle_memview_get_strides(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- if (data->view.strides == NULL) return Qnil;
-
- const ssize_t ndim = data->view.ndim;
- VALUE strides = rb_ary_new_capa(ndim);
- ssize_t i;
- for (i = 0; i < ndim; ++i) {
- rb_ary_push(strides, SSIZET2NUM(data->view.strides[i]));
- }
- return strides;
-}
-
-static VALUE
-rb_fiddle_memview_get_sub_offsets(VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
- if (data->view.sub_offsets == NULL) return Qnil;
-
- const ssize_t ndim = data->view.ndim;
- VALUE sub_offsets = rb_ary_new_capa(ndim);
- ssize_t i;
- for (i = 0; i < ndim; ++i) {
- rb_ary_push(sub_offsets, SSIZET2NUM(data->view.sub_offsets[i]));
- }
- return sub_offsets;
-}
-
-static VALUE
-rb_fiddle_memview_aref(int argc, VALUE *argv, VALUE obj)
-{
- struct memview_data *data;
- TypedData_Get_Struct(obj, struct memview_data, &fiddle_memview_data_type, data);
-
- if (NIL_P(data->view.obj)) return Qnil;
-
- const ssize_t ndim = data->view.ndim;
- if (argc != ndim) {
- rb_raise(rb_eIndexError, "wrong number of index (%d for %"PRIdSIZE")", argc, ndim);
- }
-
- VALUE indices_v = 0;
- ssize_t *indices = ALLOCV_N(ssize_t, indices_v, ndim);
-
- ssize_t i;
- for (i = 0; i < ndim; ++i) {
- ssize_t x = NUM2SSIZET(argv[i]);
- indices[i] = x;
- }
-
- uint8_t *ptr = rb_memory_view_get_item_pointer(&data->view, indices);
- ALLOCV_END(indices_v);
-
- if (data->view.format == NULL) {
- return INT2FIX(*ptr);
- }
-
- if (!data->members) {
- const char *err;
- if (rb_memory_view_parse_item_format(data->view.format, &data->members, &data->n_members, &err) < 0) {
- rb_raise(rb_eRuntimeError, "Unable to recognize item format at %"PRIdSIZE" in \"%s\"",
- err - data->view.format, data->view.format);
- }
- }
-
- return rb_memory_view_extract_item_members(ptr, data->members, data->n_members);
-}
-
-static VALUE
-rb_fiddle_memview_to_s(VALUE self)
-{
- struct memview_data *data;
- const char *raw_data;
- long byte_size;
- VALUE string;
-
- TypedData_Get_Struct(self,
- struct memview_data,
- &fiddle_memview_data_type,
- data);
-
- if (NIL_P(data->view.obj)) {
- raw_data = NULL;
- byte_size = 0;
- } else {
- raw_data = data->view.data;
- byte_size = data->view.byte_size;
- }
-
- string = rb_enc_str_new_static(raw_data, byte_size, rb_ascii8bit_encoding());
- {
- ID id_memory_view;
- CONST_ID(id_memory_view, "memory_view");
- rb_ivar_set(string, id_memory_view, self);
- }
- return rb_obj_freeze(string);
-}
-
-void
-Init_fiddle_memory_view(void)
-{
- rb_cMemoryView = rb_define_class_under(mFiddle, "MemoryView", rb_cObject);
- rb_define_alloc_func(rb_cMemoryView, rb_fiddle_memview_s_allocate);
- rb_define_singleton_method(rb_cMemoryView, "export", rb_fiddle_memview_s_export, 1);
- rb_define_method(rb_cMemoryView, "initialize", rb_fiddle_memview_initialize, 1);
- rb_define_method(rb_cMemoryView, "release", rb_fiddle_memview_release, 0);
- rb_define_method(rb_cMemoryView, "obj", rb_fiddle_memview_get_obj, 0);
- rb_define_method(rb_cMemoryView, "byte_size", rb_fiddle_memview_get_byte_size, 0);
- rb_define_method(rb_cMemoryView, "readonly?", rb_fiddle_memview_get_readonly, 0);
- rb_define_method(rb_cMemoryView, "format", rb_fiddle_memview_get_format, 0);
- rb_define_method(rb_cMemoryView, "item_size", rb_fiddle_memview_get_item_size, 0);
- rb_define_method(rb_cMemoryView, "ndim", rb_fiddle_memview_get_ndim, 0);
- rb_define_method(rb_cMemoryView, "shape", rb_fiddle_memview_get_shape, 0);
- rb_define_method(rb_cMemoryView, "strides", rb_fiddle_memview_get_strides, 0);
- rb_define_method(rb_cMemoryView, "sub_offsets", rb_fiddle_memview_get_sub_offsets, 0);
- rb_define_method(rb_cMemoryView, "[]", rb_fiddle_memview_aref, -1);
- rb_define_method(rb_cMemoryView, "to_s", rb_fiddle_memview_to_s, 0);
-}
-
-#endif /* HAVE_RUBY_MEMORY_VIEW_H */