annoy-rb 0.7.0 → 0.7.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 26a73345b4bb8bb29cbfc836599215b38dfa2e61f53bcf8183978802dbafbadc
-  data.tar.gz: fdf69a4b34bd4c1a06cab5a68c54de13ac7a521c079ff9526981c8b13056373b
+  metadata.gz: d1f1b7326f096ef01be709b819e6fcac709776e5561d931b8b8a5b022aed1dc2
+  data.tar.gz: e767849b0970773b1c01307b830d6bbf15c250d86a4716d809867dc0cf8f211d
 SHA512:
-  metadata.gz: 7999995d341e22f10ce89c645b5eaa2a7911812a03e3c04291c28e120bfcec27397930815b8491854a2f65577804a8b7fb50ac6bb7eaba2d7cc1d780c90df711
-  data.tar.gz: b58d1820ce6a48d7bae526afccaefe16770091c9b84b00679517ffeabfde16df5cd70ababc751037b54c3f04a064131614edb852d9c8298a756efa6fc09592e3
+  metadata.gz: da4627d3414ed6ba062a3fb5d9ba546a32dced3e11d42ab4756badc219c29813b9d5389cbcb2159c027ad9a3b43c58f23f0f19660836c557d12c2c1b2a37330f
+  data.tar.gz: f195b96f9010a67ccc01088006a09e1ec5f857e7226f3d6b7c58a9ab5205757365061bbd1eb571715bfcd24a8f47c00e6aea7fcc1d7024f8cb6e511b7cb37255

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.7.1
+
+- Fix bug that item elements are converted to unsingned integers when dtype is set to float32 ([#3](https://github.com/yoshoku/annoy-rb/issues/3)).
+
 ## 0.7.0
 
 - Update bundled Annoy version to 1.17.1.

data/ext/annoy/annoyext.hpp

@@ -111,7 +111,7 @@ private:
 
     F* vec = (F*)ruby_xmalloc(n_dims * sizeof(F));
     for (int i = 0; i < n_dims; i++) {
-      vec[i] = typeid(F) == typeid(double) ? NUM2DBL(rb_ary_entry(arr, i)) : NUM2UINT(rb_ary_entry(arr, i));
+      vec[i] = (F)(typeid(F) == typeid(uint64_t) ? NUM2UINT(rb_ary_entry(arr, i)) : NUM2DBL(rb_ary_entry(arr, i)));
     }
 
     char* error;
@@ -195,7 +195,7 @@ private:
       const int sz_distances = distances.size();
       VALUE distances_arr = rb_ary_new2(sz_distances);
       for (int i = 0; i < sz_distances; i++) {
-        rb_ary_store(distances_arr, i, typeid(F) == typeid(double) ? DBL2NUM(distances[i]) : UINT2NUM(distances[i]));
+        rb_ary_store(distances_arr, i, typeid(F) == typeid(uint64_t) ? UINT2NUM(distances[i]) : DBL2NUM(distances[i]));
       }
       VALUE res = rb_ary_new2(2);
       rb_ary_store(res, 0, neighbors_arr);
@@ -222,7 +222,7 @@ private:
 
     F* vec = (F*)ruby_xmalloc(n_dims * sizeof(F));
     for (int i = 0; i < n_dims; i++) {
-      vec[i] = typeid(F) == typeid(double) ? NUM2DBL(rb_ary_entry(_vec, i)) : NUM2UINT(rb_ary_entry(_vec, i));
+      vec[i] = (F)(typeid(F) == typeid(uint64_t) ? NUM2UINT(rb_ary_entry(_vec, i)) : NUM2DBL(rb_ary_entry(_vec, i)));
     }
 
     const int n_neighbors = NUM2INT(_n_neighbors);
@@ -246,7 +246,7 @@ private:
       const int sz_distances = distances.size();
       VALUE distances_arr = rb_ary_new2(sz_distances);
       for (int i = 0; i < sz_distances; i++) {
-        rb_ary_store(distances_arr, i, typeid(F) == typeid(double) ? DBL2NUM(distances[i]) : UINT2NUM(distances[i]));
+        rb_ary_store(distances_arr, i, typeid(F) == typeid(uint64_t) ? UINT2NUM(distances[i]) : DBL2NUM(distances[i]));
       }
       VALUE res = rb_ary_new2(2);
       rb_ary_store(res, 0, neighbors_arr);
@@ -266,7 +266,7 @@ private:
     get_annoy_index(self)->get_item(idx, vec);
 
     for (int i = 0; i < n_dims; i++) {
-      rb_ary_store(arr, i, typeid(F) == typeid(double) ? DBL2NUM(vec[i]) : UINT2NUM(vec[i]));
+      rb_ary_store(arr, i, typeid(F) == typeid(uint64_t) ? UINT2NUM(vec[i]) : DBL2NUM(vec[i]));
     }
 
     ruby_xfree(vec);
@@ -277,7 +277,7 @@ private:
     const int32_t i = (int32_t)NUM2INT(_i);
     const int32_t j = (int32_t)NUM2INT(_j);
     const F dist = get_annoy_index(self)->get_distance(i, j);
-    return typeid(F) == typeid(double) ? DBL2NUM(dist) : UINT2NUM(dist);
+    return typeid(F) == typeid(uint64_t) ? UINT2NUM(dist) : DBL2NUM(dist);
   };
 
   static VALUE _annoy_index_get_n_items(VALUE self) {

data/lib/annoy/version.rb

@@ -3,7 +3,7 @@
 # Annoy.rb is a Ruby wrapper for Annoy (Approximate Nearest Neighbors Oh Yeah).
 module Annoy
   # The version of Annoy.rb you are using.
-  VERSION = '0.7.0'
+  VERSION = '0.7.1'
 
   # The version of Annoy included with gem.
   ANNOY_VERSION = '1.17.1'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: annoy-rb
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.7.1
 platform: ruby
 authors:
 - yoshoku
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-08-13 00:00:00.000000000 Z
+date: 2022-09-30 00:00:00.000000000 Z
 dependencies: []
 description: Annoy.rb provides Ruby bindings for the Annoy (Approximate Nearest Neighbors
   Oh Yeah).