prefixstorage 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 08dea8d443691e75e4c8b082a5f19a07bec791df
+  data.tar.gz: 254f3635a73b7aa7f7bb871be1187f9d1003e779
+SHA512:
+  metadata.gz: aadc261a2a65cbf9931042c9d3b306e058f695740f51252a407ecf629104de86a91969c0b293abb2a013d717ade9f340213975e067cdd319050c8e5bf4eb4183
+  data.tar.gz: cebbf6981365b038a5f83ffdebe69338ef4a99b516be9f0fc38652db0821c1bc108d5e0f85cd3f4644f856a1a59747243bfff0fb960de918c5b4fd97591c95ab

data/lib/storage.rb

@@ -0,0 +1,156 @@
+require "rubygems"
+require "zip/zip"
+
+class Storage
+
+  require_relative 'tree_node'
+
+  def initialize
+    @root = TreeNode.new
+  end
+
+  #adds new string to storage
+  def add(str)
+
+    words = (str.split ",").map! { |word| word.strip }
+
+    words.each do |word|
+      node = @root
+      word.split(//).each do |ch|
+        node.children[ch]=TreeNode.new if node.children[ch].nil?
+
+        node = node.children[ch]
+
+      end
+      node.is_leaf=true
+    end
+    @root.children
+  end
+
+
+  #checks if storage contains string
+  def contains(str)
+    node = @root
+    str.each_char do |ch|
+
+      return false if node.children[ch].nil?
+
+      node = node.children[ch]
+    end
+    node.is_leaf?
+
+  end
+
+
+  #returns all strings that start with str
+  def find(str)
+    node = @root
+    str.each_char do |ch|
+
+      return [] if node.children[ch].nil?
+
+      node = node.children[ch]
+    end
+    words = []
+    get_all_words node, str, words
+
+    words
+  end
+
+
+  def load_from_file(file_name)
+    file_to_read = File.open(file_name, "r")
+    read_from_file file_to_read
+    file_to_read.close
+  end
+
+  def save_to_file(file_name)
+    file_to_write = File.new(file_name, "w+")
+    write_to_file file_to_write
+    file_to_write.close
+
+  end
+
+
+  def load_from_zip (zip_name)
+    zip_name = zip_name+".zip" unless zip_name.end_with? ".zip"
+    Zip::ZipFile.foreach(zip_name) do |file|
+      read_from_file file.get_input_stream
+    end
+
+  end
+
+  def save_to_zip (zip_name, result_file_name)
+    zip_name = zip_name+".zip" unless zip_name.end_with? ".zip"
+    Zip::ZipFile.open(zip_name, Zip::ZipFile::CREATE) do |zipfile|
+      zipfile.get_output_stream(result_file_name)do |f|
+        write_to_file f
+      end
+    end
+
+  end
+
+
+  private
+  def get_all_words(node, ch, words)
+    str = String.new(ch)
+    if node.is_leaf?
+      words.push str
+    end
+    if !node.children.empty?
+
+
+      node.children.each do |key, value|
+        get_all_words(value, str+key, words)
+
+
+      end
+
+
+    end
+
+
+  end
+
+  def read_from_file(file_to_read)
+    lines = file_to_read.readlines
+    lines.each {|line| add(line)}
+
+
+  end
+
+  def write_to_file(file_to_write)
+    words = find("")
+    if !words.empty?
+      last_word  =
+      words.each {|word| file_to_write.write(word+", ")}
+      file_to_write.write last_word
+    end
+
+  end
+
+end
+#ant
+#  chor
+#   ient
+
+=begin
+storage = Storage.new
+puts storage.add ""
+puts storage.add "te, spec ,tetra, post, rock, postcard, trunk, pork"
+
+puts storage.contains "te"
+
+storage.load_from_file "../test_file"
+puts storage.find ""
+
+storage.save_to_file "../result.txt"
+puts File.readline "../result.txt"
+#storage.save_to_zip "result", "result.txt"
+
+#storage.load_from_zip "result"
+
+
+
+=end
+

data/lib/tree_node.rb

@@ -0,0 +1,25 @@
+class TreeNode
+
+  def initialize
+    @children = {}
+    @is_leaf = false
+  end
+
+  def is_leaf?
+    @is_leaf
+  end
+
+  def is_leaf=(is_leaf)
+    @is_leaf = is_leaf
+
+  end
+
+  def children
+    @children
+  end
+
+  def to_s
+    children.to_s
+  end
+
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: prefixstorage
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Artem Petrik
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-04-01 00:00:00.000000000 Z
+dependencies: []
+description: Gem for storing strings in prefix tree (Trie)
+email: earsonheart@gmeil.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/storage.rb
+- lib/tree_node.rb
+homepage: ''
+licenses: []
+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.0.3
+signing_key: 
+specification_version: 4
+summary: Prefix tree strings storage
+test_files: []