ruby_trie 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 7b99b117ecaf12bcdde89702426168365cb64e6b
+  data.tar.gz: 7eafccb776fda1922070eb15535ee320219973d3
+SHA512:
+  metadata.gz: 77c32f34f931a65daa2f8c250d66ac4df34944bf97e983e264ffcffd91ce56a217bbbf1551d4faaae042b271aebb4a0961401ebae93c6b0a9c8839d8270c7b18
+  data.tar.gz: 09a2c2c20396b48df18fe6f2ae21719793448db8043ce116ce1b13fb371a25df45f415476723e0cab843141f0fd42a4332249712f8912457949b7de36dc74dc2

data/lib/ruby_trie.rb

@@ -0,0 +1,178 @@
+# ruby_trie
+# a pure Ruby trie implementation
+# because fast-trie is broken and segfaults
+
+# Stephen Gooberman-Hill@amey.co.uk
+# (c) Amey Nov 2016
+# licenced uner the MIT licence
+
+
+gem 'tree'
+require 'tree'
+
+# RubyTrie is a pure Ruby implentation of a Trie based on the 
+# Tree module
+module RubyTrie
+   # the content of a TrieNode
+   class TrieContent
+      # symbol at the node, and the entire string to this node are held
+      attr_reader :symbol, :string
+      
+      # value can be overwritten
+      attr_accessor :value
+      
+      # Initialize with the full string (not just the symbol)
+      # and an optional value
+      # You should not need to use this class unless you start
+      # walking the Trie 
+      def initialize(str, value=nil)
+         @string = str
+         @symbol = str[-1]
+         @value = value
+      end
+      
+      # turn the content into a [symbol, string, value] array
+      def to_a
+         [@symbol, @string, @value]
+      end
+   end
+
+   # A TrieNode is a node in the Trie
+   # derives from the TreeNode class
+   # You should only need this class if you use Tree methods
+   # to walk the tree and such-like
+   class TrieNode < Tree::TreeNode
+      # Specialized version of TreeNode
+      
+      # adds adds a node at a particular point
+      # if the node already exists, then it will
+      # overwrite the value
+      # :call-seq:
+      # add_node(string, value) -> TrieContent
+      def add_node(string, value)
+         current_node = self.root
+         current_string = ''
+         chars = string.split('')
+         
+         chars.each do |c|
+            current_string << c
+            child = current_node[c]
+            unless child 
+               child = TrieNode.new(c)
+               current_node.add(child)
+            end
+            current_node = child
+         end
+         
+         current_node.content = TrieContent.new(string, value)   
+      end
+      
+      # return the TreeNode corresponding to a given string
+      # :call-seq:
+      # get_node(string) -> TrieNode
+
+      def get_node(string)
+         current_node = self.root
+         current_string = ''
+         chars = string.split('')
+         
+         chars.each do |c|
+            current_string << c
+            current_node = current_node[c]
+            break unless current_node
+         end
+         
+         current_node
+      end
+      
+      # gets the value at a node
+      # returns nil if the node does not exist
+      # :call-seq:
+      # get_node_value(string) -> Object
+      def get_node_value(string)
+         node = get_node(string)
+         node&.content&.value
+      end
+   end
+
+   # Trie implements a Trie structure via an
+   # underlying implementation using Tree
+   # Trie has a single attribute (root) which is the root
+   # TrieNode
+   class Trie
+      #The Trie just holds the root node
+      attr_reader :root
+      
+      # create an empty Trie
+      def initialize
+         @root = TrieNode.new('')
+      end
+      
+      # add a string with optional value to the Trie
+      # Note - will overwrite the value if the node
+      # already exists
+      # :call-seq:
+      # add(string, value) -> TrieContent
+
+      def add(string, value=true)
+         @root.add_node(string, value)
+      end
+      
+      # get the value at a node
+      # returns nil if the node does not exist
+      # :call-seq:
+      # get(string) -> Object
+      def get(string)
+         @root.get_node_value(string)
+      end
+      
+      # get all the children of a given prefix
+      # including the prefix, if it exists itself
+      # :call-seq:
+      # children(string) -> Array
+      def children(string)
+         parent = @root.get_node(string)
+         return nil unless parent
+         
+         parent.inject([]) do |a,n| 
+            a << n.content.string if n.content
+            a
+         end
+      end
+      
+      # get all the children of a given prefix
+      # with thier values (as a [key,value] pair)
+      # including the prefix, if it exists itself
+      # :call-seq:
+      # children_with_values(string) -> Array
+      def children_with_values(string)
+         parent = @root.get_node(string)
+         return nil unless parent
+         
+         parent.inject([]) do |a,n| 
+            a << [n.content.string, n.content.value] if n.content
+            a
+         end
+      end
+      
+      # get the content of all children of a given prefix
+      # including the prefix, if it exists itself
+      # :call-seq:
+      # children_content(string) -> Array
+      def children_content(string)
+         parent = @root.get_node(string)
+         return nil unless parent
+         
+         parent.inject([]) do |a,n| 
+            a << n.content.to_a if n.content
+            a
+         end
+      end
+
+
+         
+   end
+end     
+
+
+

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: ruby_trie
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Stephen Gooberman-Hill
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-11-15 00:00:00.000000000 Z
+dependencies: []
+description: A trie is an efficient string storage and search structure
+email: stephen.gooberman-hill@amey.co.uk
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/ruby_trie.rb
+homepage: http://rubygems.org/gems/ruby_trie
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Pure Ruby Trie implemnattion
+test_files: []