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