def chainBuilder(self):
parsedEpisodes = getEpisodes(self.episodes)
for e in parsedEpisodes:
castList = e.getCast()
for c in castList:
if c not in self.suppCharacters and c not in self.mainCharacters:
self.suppCharacters.add(c)
directChain = MarkovChain(2) #Change direction order here
stageDirections = getAllStageDirs(self.episodes)
for direction in stageDirections:
directChain.addData(direction.getChainableSource())
self.chains["STAGEDIR"] = directChain
for i in self.mainCharacters:
self.characterChain(i)
for m in self.suppCharacters:
self.characterChain(m)
def characterChain(self, character):
tempChain = MarkovChain(2) #Change line order here
characterLines = getAllCharLines(self.episodes, character)
for line in characterLines:
tempChain.addData(line.getChainableSource())
self.chains[character] = tempChain
