zelkova 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 86b06cdb9578edcaa80c633a1b2205d9a01fd231aaecbc3c8721482ab960da36
-  data.tar.gz: b94f3080271dc65a3c4f21c7e7bf95d1ea25f3b9509825278aeea819b13bd6f4
+  metadata.gz: 99845c2d87f2d9f7e4c66c96912a9aac723677f4069a205bdb5e0979c3be035c
+  data.tar.gz: 6c1d6d6e4cb6d5c53bd56dec2319cbb1d43756d11bcd955afd9a7a9b03ddafac
 SHA512:
-  metadata.gz: a2e6e47acd4da9b54db90a18039f6af0d7635cc54e630570bbbbf3e7f9da5ebc71c5fe7acbe7aa5f2eea40ae8a0ad2c53c1534ee77932649c03f3e04e355d4bd
-  data.tar.gz: f9f6a520545eeaf470729a6ad1332884b6753857b8fd7535c8c3088086b8fb308460dc742a1d791b813176a8ab5247945dd07ab09068603622b95a59f9378dab
+  metadata.gz: 75ac38244fb0a07394dc799bda375a930431ec670fe706ce4ec826504cdd22acb6777f73f8a2b5f4834dd670a103eea7ac2b0bf82aefdb190fb08b738a68d53b
+  data.tar.gz: 4622edc7572b76f1f0b938e1cfa0e9f38896842c1debfe937c63acc7862a9e37576727b083c305fb1665f7597f4b9a5990d9e96c60a5ff36de6d4ae63a51dbf7

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    zelkova (0.1.0)
+    zelkova (0.1.1)
       eikon
 
 GEM

data/lib/zelkova/graph.rb

@@ -18,16 +18,28 @@ class Zelkova::Graph
   sig { returns(Integer) }
   attr_reader :depth
 
-  sig { void }
-  def initialize
+  sig { returns(Integer) }
+  attr_accessor :radius
+
+  sig { params(radius: T.nilable(Integer)).void }
+  def initialize(radius = 2)
     @root_node = T.let(nil, T.nilable(Zelkova::Node))
     @nodes = T.let([], T::Array[Zelkova::Node])
     @depth = T.let(0, Integer)
+
+    @radius = T.let(T.must(radius), Integer)
+  end
+
+  sig { void }
+  def reset
+    @root_node = nil
+    @nodes = []
+    @depth = 0
   end
 
-  sig { params(word: String).returns(Zelkova::Node) }
-  def add_node(word)
-    node = Zelkova::Node.new(word, self)
+  sig { params(word: String, metadata: T.nilable(T::Hash[T.untyped, T.untyped])).returns(Zelkova::Node) }
+  def add_node(word, metadata = {})
+    node = Zelkova::Node.new(word, self, metadata)
     nodes << node
 
     if @root_node.nil?
@@ -60,9 +72,9 @@ class Zelkova::Graph
     node
   end
 
-  sig { params(query: String).returns(T::Array[String]) }
+  sig { params(query: String).returns(T::Array[T::Hash[Zelkova::Node, Float]]) }
   def search(query)
-    radius = 2
+    radius = @radius
     candidate_nodes = []
     candidate_edges = []
 
@@ -85,20 +97,6 @@ class Zelkova::Graph
     end
 
     candidate_nodes.sort_by! { |candidate_node| candidate_node[:distance] }
-    candidate_nodes.map { |candidate_node| candidate_node[:node].word }
+    candidate_nodes
   end
 end
-
-# class Comparator
-#   extend T::Sig
-
-#   sig { params(dhash_1: String, dhash_2: String).returns(Integer) }
-#   def self.compare(dhash_1, dhash_2)
-#     hamming_distance = 0
-#     dhash_1.chars.each_with_index do |character, index|
-#       hamming_distance += 1 if character != dhash_2.chars[index]
-#     end
-
-#     hamming_distance + (dhash_1.length - dhash_2.length).abs
-#   end
-# end

data/lib/zelkova/node.rb

@@ -11,18 +11,20 @@ class Zelkova::Node
   attr_reader :edges
   sig { returns(Zelkova::Graph) }
   attr_reader :graph
+  sig { returns(T::Hash[T.untyped, T.untyped]) }
+  attr_reader :metadata
 
-  sig { params(word: String, graph: Zelkova::Graph).void }
-  def initialize(word, graph)
+  sig { params(word: String, graph: Zelkova::Graph, metadata: T.nilable(T::Hash[T.untyped, T.untyped])).void }
+  def initialize(word, graph, metadata = {})
     @word = T.let(word, String)
     @graph = T.let(graph, Zelkova::Graph)
-
     @edges = T.let([], T::Array[Zelkova::Edge])
+    @metadata = T.let(T.must(metadata), T::Hash[T.untyped, T.untyped])
   end
 
   sig { returns(String) }
   def inspect
-    "Word: #{self.word}, Number of Edges: #{self.edges.count}"
+    "Word: #{self.word}, Number of Edges: #{self.edges.count}, Metadata: #{self.metadata}"
   end
 
   sig { params(node: Zelkova::Node, weight: Integer).void }

data/lib/zelkova/version.rb

@@ -1,4 +1,4 @@
 # typed: strict
 module Zelkova
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/zelkova.rb

@@ -7,4 +7,16 @@ require_relative "zelkova/graph"
 
 module Zelkova
   class Error < StandardError; end
+
+  @@graph = T.let(Zelkova::Graph.new, Zelkova::Graph)
+
+  class << self
+    extend T::Sig
+
+    sig { returns(Zelkova::Graph) }
+    def graph
+      # TODO: make this lazy when in development mode for speed
+      @@graph
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: zelkova
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Christopher Guess
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-04-27 00:00:00.000000000 Z
+date: 2022-05-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: eikon