aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xmarkov-run.rb26
-rwxr-xr-xmarkov-serialise.rb4
2 files changed, 17 insertions, 13 deletions
diff --git a/markov-run.rb b/markov-run.rb
index ef81315..991c5e7 100755
--- a/markov-run.rb
+++ b/markov-run.rb
@@ -2,7 +2,11 @@
PRNG = Random.new(Time.now.to_i)
-stats = Marshal::load(File.open(ARGV[0]).read)
+STATS = Marshal::load(File.open(ARGV[0]).read)
+
+CHUNK = STATS["__SETTINGS__"]["__CHUNK__"]
+
+NUM = (ARGV[2] or "1").to_i
def cfd(hash)
tot = 0
@@ -19,17 +23,17 @@ def sel(hash)
}
end
-current = stats.keys.select{|k|k.first.match(/^[A-Z]/)}.sample
-
-print "#{current.join(" ")} "
-
-def achunk(chunk, hash)
+def aword(chunk, hash)
sel(cfd(hash[chunk]))
end
-(ARGV[1].to_i - 1).times do
- current = achunk(current, stats)
- print "#{current.join(" ")} "
-end
+NUM.times do
+ current = STATS.keys.select{|k| k != "__SETTINGS__" }.sample
+
+ (ARGV[1].to_i - CHUNK).times do
+ current << aword(current[-CHUNK..-1], STATS)
+ end
-puts
+ puts current.join(" ")
+ puts
+end
diff --git a/markov-serialise.rb b/markov-serialise.rb
index 00e459e..1199453 100755
--- a/markov-serialise.rb
+++ b/markov-serialise.rb
@@ -2,7 +2,7 @@
CHUNK = ARGV[1].to_i
-stats = Hash.new
+stats = {"__SETTINGS__" => {"__CHUNK__" => CHUNK}}
words = File.readlines(ARGV[0])
.map(&:split)
@@ -10,7 +10,7 @@ words = File.readlines(ARGV[0])
(0 .. (words.length - CHUNK - CHUNK)).each do |i|
k = words[i...(i+CHUNK)]
- v = words[(i+CHUNK)...(i+CHUNK+CHUNK)]
+ v = words[(i+CHUNK)]
if stats.include?(k) then
t = stats[k]
if t.include?(v) then