dartsclone 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ffd79764c75c8fd8c2b7dc55b5f80c3875a0641420f05a9d1e07d701ffc15a17
-  data.tar.gz: 70799ddcdb885b78fa338a36b28063df1ae41a0f87d3be6a1aad8faec8b939a7
+  metadata.gz: dce86b5df78cc6b62be497a33f5203a352def9f83767191840e2cd5f2bcae460
+  data.tar.gz: da2b74859ccc9834583b33ac4c9628b976a6e329b9a8d77348b4947767503794
 SHA512:
-  metadata.gz: af070c15bc7b53a58d06200a2a8da7ce931161c470248f43a80365af3323122db200fa467bce0fe738d5ecfbd1dde8865733c37a27dcf7d00f5c8947341feeef
-  data.tar.gz: d4bad89bbbb9b2a7c8aa0defb663e56db7cbdae2a934308ebb86c8ad4ddc24a8d659722186b1ce9de4ce1f4faeaefb17821f1564ec54865e668813ca06617994
+  metadata.gz: 074b992e52c7604970772c30976cbbe4cf8efd7f7cb6d26dcf62e8f3eb2998b4cb08dd615ab224809786cd1f65fc4759f80fe3d8ba53e4674a17afe25188b43d
+  data.tar.gz: 0d136b219466c196dbb9616b02f01ab98b3225fef93831e6e80aefa049561aab621073fa6ac6589e3132b71e803eb65c0f0f1474a9047be62093e58ce426594a

data/CHANGELOG.md

@@ -1,2 +1,5 @@
+## 0.2.0
+- Add get_array and set_array methods to dump and load array data.
+
 ## 0.1.0
 - First release.

data/ext/dartsclone/dartscloneext.hpp

@@ -30,6 +30,8 @@ class RbDoubleArray
       rb_define_method(rb_cDoubleArray, "build", RUBY_METHOD_FUNC(_double_array_build), -1);
       rb_define_method(rb_cDoubleArray, "open", RUBY_METHOD_FUNC(_double_array_open), -1);
       rb_define_method(rb_cDoubleArray, "save", RUBY_METHOD_FUNC(_double_array_save), -1);
+      rb_define_method(rb_cDoubleArray, "get_array", RUBY_METHOD_FUNC(_double_array_get_array), 0);
+      rb_define_method(rb_cDoubleArray, "set_array", RUBY_METHOD_FUNC(_double_array_set_array), 1);
       rb_define_method(rb_cDoubleArray, "exact_match_search", RUBY_METHOD_FUNC(_double_array_exact_match_search), -1);
       rb_define_method(rb_cDoubleArray, "common_prefix_search", RUBY_METHOD_FUNC(_double_array_common_prefix_search), -1);
       rb_define_method(rb_cDoubleArray, "traverse", RUBY_METHOD_FUNC(_double_array_traverse), -1);
@@ -90,8 +92,8 @@ class RbDoubleArray
 
       const char* filename = StringValueCStr(_filename);
       const char* mode = kwvalues[0] == Qundef ? "rb" : StringValueCStr(kwvalues[0]);
-      const size_t offset = kwvalues[1] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[1]);
-      const size_t size = kwvalues[2] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[2]);
+      const size_t offset = kwvalues[1] == Qundef ? 0 : NUM2SIZET(kwvalues[1]);
+      const size_t size = kwvalues[2] == Qundef ? 0 : NUM2SIZET(kwvalues[2]);
 
       if (get_double_array(self)->open(filename, mode, offset, size) != 0) {
         return Qfalse;
@@ -110,7 +112,7 @@ class RbDoubleArray
 
       const char* filename = StringValueCStr(_filename);
       const char* mode = kwvalues[0] == Qundef ? "wb" : StringValueCStr(kwvalues[0]);
-      const size_t offset = kwvalues[1] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[1]);
+      const size_t offset = kwvalues[1] == Qundef ? 0 : NUM2SIZET(kwvalues[1]);
 
       if (get_double_array(self)->save(filename, mode, offset) != 0) {
         return Qfalse;
@@ -118,6 +120,20 @@ class RbDoubleArray
       return Qtrue;
     };
 
+    static VALUE _double_array_get_array(VALUE self) {
+      const char* arr = (char*)get_double_array(self)->array();
+      const size_t sz = get_double_array(self)->total_size();
+      return rb_str_new(arr, sz);
+    };
+
+    static VALUE _double_array_set_array(VALUE self, VALUE bytes) {
+      const size_t sz = NUM2SIZET(rb_funcall(bytes, rb_intern("size"), 0));
+      const size_t total_sz = sz / get_double_array(self)->unit_size();
+      char* arr = StringValuePtr(bytes);
+      get_double_array(self)->set_array(arr, total_sz);
+      return Qnil;
+    };
+
     static VALUE _double_array_exact_match_search(int argc, VALUE* argv, VALUE self) {
       VALUE _key = Qnil;
       VALUE kwargs = Qnil;
@@ -133,8 +149,8 @@ class RbDoubleArray
       }
 
       const char* key = StringValueCStr(_key);
-      const size_t length = kwvalues[0] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[0]);
-      const size_t node_pos = kwvalues[1] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[1]);
+      const size_t length = kwvalues[0] == Qundef ? 0 : NUM2SIZET(kwvalues[0]);
+      const size_t node_pos = kwvalues[1] == Qundef ? 0 : NUM2SIZET(kwvalues[1]);
 
       Darts::DoubleArray::value_type value;
       get_double_array(self)->exactMatchSearch(key, value, length, node_pos);
@@ -156,9 +172,9 @@ class RbDoubleArray
       }
 
       const char* key = StringValueCStr(_key);
-      const size_t max_num_results = kwvalues[0] == Qundef ? (size_t)NUM2INT(rb_funcall(_key, rb_intern("size"), 0)) : (size_t)NUM2INT(kwvalues[0]);
-      const size_t length = kwvalues[1] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[1]);
-      const size_t node_pos = kwvalues[2] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[2]);
+      const size_t max_num_results = kwvalues[0] == Qundef ? NUM2SIZET(rb_funcall(_key, rb_intern("size"), 0)) : NUM2SIZET(kwvalues[0]);
+      const size_t length = kwvalues[1] == Qundef ? 0 : NUM2SIZET(kwvalues[1]);
+      const size_t node_pos = kwvalues[2] == Qundef ? 0 : NUM2SIZET(kwvalues[2]);
 
       Darts::DoubleArray::result_pair_type* results =
         (Darts::DoubleArray::result_pair_type*)ruby_xmalloc(max_num_results * sizeof(Darts::DoubleArray::result_pair_type));
@@ -191,9 +207,9 @@ class RbDoubleArray
       rb_get_kwargs(kwargs, kwtable, 2, 1, kwvalues);
 
       const char* key = StringValueCStr(_key);
-      size_t node_pos = (size_t)NUM2INT(kwvalues[0]);
-      size_t key_pos = (size_t)NUM2INT(kwvalues[1]);
-      const size_t length = kwvalues[2] == Qundef ? 0 : (size_t)NUM2INT(kwvalues[2]);
+      size_t node_pos = NUM2SIZET(kwvalues[0]);
+      size_t key_pos = NUM2SIZET(kwvalues[1]);
+      const size_t length = kwvalues[2] == Qundef ? 0 : NUM2SIZET(kwvalues[2]);
 
       Darts::DoubleArray::value_type value = get_double_array(self)->traverse(key, node_pos, key_pos, length);
 

data/lib/dartsclone/version.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
 module DartsClone
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
   DARTS_CLONE_VERSION = '0.32'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dartsclone
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - yoshoku
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-10-04 00:00:00.000000000 Z
+date: 2020-10-10 00:00:00.000000000 Z
 dependencies: []
 description: Darts-clone.rb is a Ruby binding for the Darts-clone.
 email: