diff --git a/chart.py b/chart.py index bef527b..c845148 100644 --- a/chart.py +++ b/chart.py @@ -13,10 +13,10 @@ def form_values(gammme, depth, fragmentation_percentage, should_defragment, custom_branching): - lst, ratios, kick_depth = new_parser.parse_games(gammme, depth, custom_branching) #whether or not to implement custom branching """Create parent, id, labels, and values """ + lst, ratios, kick_depth = new_parser.parse_games(gammme, depth, custom_branching) #whether or not to implement custom branching firstx = [ - lst[i][:depth] for i in range(len(lst)) + lst[i][:depth+kick_depth] for i in range(len(lst)) ] # probably unneeded but for safety's sake... all_level_moves, exclude_first_moves = [], [] # for parent/labels later @@ -26,7 +26,7 @@ def form_values(gammme, depth, fragmentation_percentage, should_defragment, cust percentage_holder, firstmove = [], [] - while counter < depth: + while counter < depth + kick_depth: counter += 1 othermove = list( Counter([tuple(firstx[i][kick_depth:counter]) for i in range(len(lst))]).items() diff --git a/new_parser.py b/new_parser.py index 5971075..35ca4a2 100644 --- a/new_parser.py +++ b/new_parser.py @@ -1,5 +1,4 @@ import pgn -import sys def parse_games(database, depth, custom_branching):