diff options
author | Nathan Lasseter <nathan@4574.co.uk> | 2014-01-30 20:10:56 +0000 |
---|---|---|
committer | Nathan Lasseter <nathan@4574.co.uk> | 2014-01-30 20:10:56 +0000 |
commit | 69ee9398256a7e15e64aa73428e7291f6f50a11c (patch) | |
tree | e0f6d35a62c2d1abf2589815a3e7b09d47b32841 /markov-generate.rb | |
parent | 46804facd4b74e6c5796b3427ccbb1a273528395 (diff) |
Diffstat (limited to 'markov-generate.rb')
-rw-r--r-- | markov-generate.rb | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/markov-generate.rb b/markov-generate.rb index 36d80a1..4213a58 100644 --- a/markov-generate.rb +++ b/markov-generate.rb @@ -1,8 +1,12 @@ module Markov class Generator - def initialize(inputfile) + def initialize(input, type=:var) @prng = Random.new(Time.now.to_i) - @stats = Marshal::load(File.open(inputfile).read) + if type == :var then + @stats = input + elsif type == :file then + @stats = Marshal::load(File.open(input).read) + end @chunklength = @stats["__SETTINGS__"]["__CHUNKLENGTH__"] end |