markov_chain 0.1.0

data/lib/markov_chain.rb

@@ -0,0 +1,124 @@
+require 'singleton'
+
+class MarkovChain
+  include Singleton
+
+  class << self
+    def instance
+      @__instance__ ||= new
+    end
+
+    def generate seed, options = {}
+      raise ArgumentError, 'Seed must be at least two characters' unless seed.length > 1
+
+      max_word_length = seed.length + 8
+
+      defaults = {
+        :max_size => 100,
+        :max_word_length => max_word_length,
+        :max_retries => 1000
+      }
+
+      options = defaults.merge options
+
+      load_dictionary! unless dictionary_loaded?
+
+      words = [seed]
+      old_words = []
+      retries = 0
+
+      mc = MarkovChain.instance
+
+      while words.length < options[:max_size] && retries < options[:max_retries] do
+        old_words = words.dup
+        word = seed.dup
+        old_word = ''
+
+        while word.length < options[:max_word_length] && word != old_word
+          old_word = word.dup
+          word << mc.get(word.slice(word.length - 1, 1))
+          words.push word.dup
+        end
+
+        words = words.map { |w| w.gsub(/[^a-z]/, '') }.uniq
+
+        retries += 1 if words == old_words
+      end
+
+      words = words[0..(options[:max_size] - 1)]
+
+      words.sort { |a, b| a.length <=> b.length }
+    end
+
+    def load_dictionary! file = :american_english
+      if file.class == Symbol
+        file = File.join File.dirname(__FILE__), 'dictionaries', file.to_s
+      end
+
+      dictionary = File.read file
+
+      if file
+        file.each do |line|
+          s = line.strip
+          instance.add_str s
+        end
+
+        @dictionary_loaded = true
+        @dictionary = File.expand_path file
+      end
+    end
+
+    def dictionary_loaded?
+      !!@dictionary_loaded
+    end
+
+    def dictionary
+      @dictionary
+    end
+  end
+
+  def initialize
+    @chars = Hash.new
+  end
+
+  def add_str str
+    index = 0
+    each_char(str) do |char|
+      add(char, str[index + 1]) if index <= str.size - 2
+      index += 1
+    end
+  end
+
+  def add char, next_char
+    @chars[char] = Hash.new(0) if !@chars[char]
+    @chars[char][next_char] += 1
+  end
+
+  def get char
+    return '' if !@chars[char]
+
+    followers = @chars[char]
+    sum = followers.inject(0) { |sum,kv| sum += kv[1] }
+    random = rand(sum) + 1
+    partial_sum = 0
+
+    next_char = followers.find do |char, count|
+      partial_sum += count
+      partial_sum >= random
+    end.first
+
+    next_char
+  end
+
+  private
+
+  def each_char str
+    if block_given?
+      str.scan(/./m) do |x|
+        yield x
+      end
+    else
+      str.scan /./m
+    end
+  end
+end

data/test/markov_chain_test.rb

@@ -0,0 +1,13 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class MarkovChainTest < Test::Unit::TestCase
+  should 'load default dictionary' do
+    MarkovChain.load_dictionary!
+    assert MarkovChain.dictionary_loaded?
+    assert_equal 'american_english', File.basename(MarkovChain.dictionary)
+  end
+
+  should 'generate 100 words by default' do
+    assert_equal 100, MarkovChain.generate('foo').length
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,9 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'markov_chain'
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,103 @@
+--- !ruby/object:Gem::Specification 
+name: markov_chain
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Alex Rabarts
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-10-31 00:00:00 +00:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: jeweler
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+description: Markov chain generator
+email: alexrabarts@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+files: 
+- Gemfile
+- Gemfile.lock
+- History.txt
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/dictionaries/american_english
+- lib/markov_chain.rb
+- test/markov_chain_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: http://github.com/alexrabarts/markov_chain
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.4.2
+signing_key: 
+specification_version: 3
+summary: A simple Markov chain generator.
+test_files: []
+