ensenar 0.0.3 → 0.1.0

data/lib/ensenar.rb

@@ -1,57 +1,71 @@
-class Ensenar
-    def initialize
-        @words = []
+class String
+    def sentences
+        ((split(/\.|\?|\!/).delete_if{|x| x=~ /^\s+/}).delete_if {|x| x =~ /\s+^/ }).delete_if {|x| x == []}
     end
     
-    def reset
-        @words = []
+    def words
+        split(/\s+|\W\s+/).delete_if {|x| x == ""}
     end
-    
-    def add(sentence)
-        @words.push sentence.split(/\W+/)
+end
+
+module Ensenar
+    class FrequencyTable
+        
+        attr_accessor :table
+        
+        def initialize()
+            @table = {}
+        end
+        
+        def add(string)
+            string = string.split(/\s+|\.$/)
+            while(word = string.shift)
+                next if string == []
+                @table[word.to_sym]||={}
+                @table[word.to_sym][string[0].to_sym]||=0
+                @table[word.to_sym][string[0].to_sym]+=1
+            end
+            return @table
+        end
+        
+        def reset()
+            @table = {}
+        end
     end
     
-    def generate(length, trigger=nil)
-        text  = []
-        if trigger
-            has = false
-            @words.each do |sentence|
-                if sentence.include? trigger
-                    has = true
-                    break
-                end
-            end
-            if has
-                scope = trigger
-                text << scope
-            else
-                return nil
-            end
-        else
-            x     = rand(@words.length)
-            y     = rand(@words[x].length)
-            scope = @words[x][y]
-            text << @words[x][y] 
+    class Markov
+        
+        def initialize()
+            @frequency_table = FrequencyTable.new
         end
-        while(text.length < length)
-            l = text.length
-            @words.rotate(rand(20)).each do |sentence|
-                if sentence.include? text.last
-                   pos = sentence.find_index text.last
-                   text << sentence[pos + 1]
-                end
-            end
-            if l == text.length
-                x     = rand(@words.length)
-                y     = rand(@words[x].length)
-                scope = @words[x][y]
-                text << @words[x][y] 
-            else
-                next
+        
+        def add(string)
+            @frequency_table.add string
+        end
+        
+        def reset
+            @frequency_table.reset
+        end
+        
+        def generate(length=rand(20),about=nil)
+            text = []
+            word = @frequency_table.table.keys[rand(@frequency_table.table.keys.length)]
+            text << word.to_s
+            until text.length == length
+                (@frequency_table.table[word]) ? word = (@frequency_table.table[word].keys.zip(@frequency_table.table[word].values).sort_by {|x| x[1] })[0][0].to_s : break
+                text << word.to_s
+                word = word.to_sym
+                return text.delete_if {|x| x == ""} if text.length == 20
+                break if @frequency_table.table[word] == {}
             end
+            return text.delete_if {|x| x == ""}
         end
-        text.delete_if {|x| x.nil? }
-        return text
     end
-    
-end
+end
+                
+#
+# Each word
+#  Find sentences that have that word
+#  Each sentence that includes it, add to the entry for it
+#  
+#

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ensenar
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors: