def txtToMarkovChain(path, encoding=None):
if encoding:
file = io.open(path, 'r', encoding=encoding)
else:
file = open(path, 'r')
builder = MarkovChain.builder()
states = []
prec = None
transitions = {}
for i, line in enumerate(file):
for j in range(len(line)):
s = _state(i, j, line)
if not s in transitions:
transitions[s] = {}
builder.addState(s)
if i == j == 0:
builder.setInitial(s, 1)
if prec:
if s not in transitions[prec]:
transitions[prec][s] = 1
else:
transitions[prec][s] += 1
prec = s
for s1 in transitions:
cases = 0
for s2 in transitions[s1]:
cases += transitions[s1][s2]
for s2 in transitions[s1]:
builder.addTransition(s1, s2, Fraction(transitions[s1][s2], cases))
return builder.build()
