Class: Markov

Inherits:
Object
  • Object
show all
Defined in:
lib/markov-reloaded.rb

Overview

Main class for generating Markov Chains

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(data, order) ⇒ Markov

data is the hash to generate by, order is the order of the hash



56
57
58
59
 
# File 'lib/markov-reloaded.rb', line 56

def initialize (data, order)
	@data = data
	@order = order
end

Instance Attribute Details

#data ⇒ Object

Returns the value of attribute data.



60
61
62
 
# File 'lib/markov-reloaded.rb', line 60

def data
  @data
end

Class Method Details

.analyze(array, order) ⇒ Object

Analyze array and create Markov object from it



66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/markov-reloaded.rb', line 66

def self.analyze(array, order)
	data = Hash.new(nil)
	for i in 0..array.length-1-order
		el = array[i..i+order-1]
		if data.include? el
			data[el].push array[i+order]# unless i+order+1 == array.length
		else
			data[el] = [array[i+order]]# unless i+order+1 == array.length
		end
	end
	Markov.new(data, order)
end

Instance Method Details

#+(mark) ⇒ Object

Add the data of another markov object to the current one



79
80
81
82
83
84
85
86
87
88
 
# File 'lib/markov-reloaded.rb', line 79

def +(mark)
	hash = mark.data
	hash.each do |key, value|
		if @data.include? key
			@data[key] += value
		else
			@data[key] = value
		end
	end
end

#add(array) ⇒ Object

Analyze an array and add it to the current object’s hash



62
63
64
 
# File 'lib/markov-reloaded.rb', line 62

def add(array)
	self.+(array.toMarkov @order)
end

#generate(maxlength, first = [], seed = (Time.now.nsec), terminator = ".") ⇒ Object

terminate(defaults to “.”). Generates and outputs an array corresponding to a markov chain.



105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/markov-reloaded.rb', line 105

def generate(maxlength, first= [], seed=(Time.now.nsec), terminator = ".")
	r = Random.new seed
	if first.class == Method
		first = first.call(@data, r.rand(10000))
	else
		first = @data.keys.pickRand(r) if first == [] || first == ""
	end
	out = first.dup
	pick = out.dup
	for i in 0..(maxlength-@order)
		element = pickElement(pick, r)
		if out.class == Array
			out.push element
		elsif out.class == String
			out += element
		end
		if pick.class == Array
			pick = pick.drop 1
		elsif pick.class == String
			pick = pick[1..pick.length-1]
		end
		if pick.class == Array
			pick.push element
		elsif pick.class == String
			pick += element
		end
		break if element == terminator
	end
	return out
end

#pickElement(element, random = (Random.new(Time.now.nsec))) ⇒ Object

Given a list of elements with length order(a key for the hash), pick the next element based on the random number generator



90
91
92
93
94
95
96
97
98
99
 
# File 'lib/markov-reloaded.rb', line 90

def pickElement(element, random=(Random.new(Time.now.nsec)))
	arr = @data[element]
	if arr == nil
		print element
		return nil
	else
		i = random.rand(arr.length)
		arr[i]
	end
end