import copy import string from generic import create_output_string_form, create_output_string_deprel, create_output_string_lemma, \ create_output_string_upos, create_output_string_xpos, create_output_string_feats, generate_key class ResultTree(object): def __init__(self, node, children, filters): self.node = node self.children = children self.filters = filters self.key = None self.order_key = None self.order = None self.array = None def __repr__(self): return self.get_key() def set_children(self, children): self.children = children def reset_params(self): self.key = None self.order_key = None self.order = None self.array = None def get_key(self): if self.key: return self.key key = '' write_self_node_to_result = False if self.children: children = self.children for child in children: if self.filters['node_order'] and child.node.location < self.node.location: if self.filters['dependency_type']: # separator = ' <' + deprel[i_child][i_answer] + ' ' separator = ' <' + child.node.deprel + ' ' else: separator = ' < ' key += child.get_key() + separator else: if not write_self_node_to_result: write_self_node_to_result = True key += self.node.name if self.filters['dependency_type']: separator = ' >' + child.node.deprel + ' ' else: separator = ' > ' key += separator + child.get_key() if not write_self_node_to_result: key += self.node.name self.key = '(' + key + ')' else: self.key = self.node.name return self.key def get_key_sorted(self): key = '' write_self_node_to_result = False if self.children: children = sorted(self.children, key=lambda x: x.node.name) for child in children: if not write_self_node_to_result: write_self_node_to_result = True key += self.node.name if self.filters['dependency_type']: separator = ' >' + child.node.deprel + ' ' else: separator = ' > ' key += separator + child.get_key_sorted() if not write_self_node_to_result: key += self.node.name key = '(' + key + ')' else: key = self.node.name return key def get_order_key(self): if self.order_key: return self.order_key order_key = '' write_self_node_to_result = False if self.children: for child in self.children: if self.filters['node_order'] and child.node.location < self.node.location: if self.filters['dependency_type']: # separator = ' <' + deprel[i_child][i_answer] + ' ' separator = ' <' + child.node.deprel + ' ' else: separator = ' < ' order_key += child.get_order_key() + separator else: if not write_self_node_to_result: write_self_node_to_result = True order_key += str(self.node.location) if self.filters['dependency_type']: separator = ' >' + child.node.deprel + ' ' else: separator = ' > ' order_key += separator + child.get_order_key() if not write_self_node_to_result: order_key += str(self.node.location) self.order_key = '(' + order_key + ')' else: self.order_key = str(self.node.location) return self.order_key def get_order(self): if self.order: return self.order order = [] write_self_node_to_result = False if self.children: for child in self.children: if self.filters['node_order'] and child.node.location < self.node.location: order += child.get_order() else: if not write_self_node_to_result: write_self_node_to_result = True order += [self.node.location] order += child.get_order() if not write_self_node_to_result: order += [self.node.location] self.order = order else: self.order = [self.node.location] return self.order def get_array(self): if self.array: return self.array array = [] write_self_node_to_result = False if self.children: for child in self.children: if self.filters['node_order'] and child.node.location < self.node.location: array += child.get_array() else: if not write_self_node_to_result: write_self_node_to_result = True array += [self.node.name_parts] array += child.get_array() if not write_self_node_to_result: array += [self.node.name_parts] self.array = array else: self.array = [self.node.name_parts] return self.array # def add(self, string, architecture_order, separator, is_left): # if is_left: # self.array = [string] + self.array # self.order = [architecture_order] + self.order # # self.order = [architecture_order] + self.order # self.separators = [separator] + self.separators # self.key = string + ' ' + separator + ' ' + self.key # self.order_key = architecture_order + ' ' + separator + ' ' + self.order_key # # else: # self.array += [string] # self.order += [architecture_order] # # self.order += [architecture_order] # self.separators += [separator] # # self.key += ' ' + separator + ' ' + string # self.order_key += ' ' + separator + ' ' + architecture_order # def add_separator(self, separator, left=True): # self_copy = copy.copy(self) # if left: # self_copy.separators += [separator] # self_copy.key += separator # self_copy.order_key += separator # else: # self_copy.separators = [separator] + self_copy.separators # self_copy.key = separator + self_copy.key # self_copy.order_key = separator + self_copy.order_key # return self_copy # def merge_results2(self): # def merge_results(self, right_t, separator, left=True): # left_tree = copy.copy(self) # right_tree = copy.copy(right_t) # # if separator: # if left: # # merged_results.append(left_part + right_part + separator) # left_tree.key = left_tree.key + right_tree.key + separator # left_tree.order_key = left_tree.order_key + right_tree.order_key + separator # left_tree.array = left_tree.array + right_tree.array # left_tree.order = left_tree.order + right_tree.order # # left_tree.order = str([architecture_order]) # left_tree.separators = left_tree.separators + right_tree.separators + [separator] # else: # # merged_results.append(left_part + separator + right_part) # left_tree.key = left_tree.key + separator + right_tree.key # left_tree.order_key = left_tree.order_key + separator + right_tree.order_key # left_tree.array = left_tree.array + right_tree.array # left_tree.order = left_tree.order + right_tree.order # # left_tree.order = str([architecture_order]) # left_tree.separators = left_tree.separators + [separator] + right_tree.separators # else: # # merged_results.append(left_part + right_part) # left_tree.key = left_tree.key + right_tree.key # left_tree.order_key = left_tree.order_key + right_tree.order_key # left_tree.array = left_tree.array + right_tree.array # left_tree.order = left_tree.order + right_tree.order # # left_tree.order = str([architecture_order]) # left_tree.separators = left_tree.separators + right_tree.separators # # return left_tree # def extend_answer(self, other_answer, separator): # self.array.extend(other_answer.array) # self.order.extend(other_answer.order) # self.key += separator + other_answer.key # self.order_key += separator + other_answer.order_key # self.separators.extend(separator) # def put_in_bracelets(self, inplace=False): # if inplace: # self.key = ('(' + self.key + ')') # self.order_key = ('(' + self.order_key + ')') # return # result = copy.copy(self) # result.key = ('(' + result.key + ')') # result.order_key = ('(' + result.order_key + ')') # return result def finalize_result(self): result = copy.copy(self) result.reset_params() # result.key = result.get_key() # result.set_root() # create order letters order = result.get_order() order_letters = [''] * len(result.order) for i in range(len(order)): ind = order.index(min(order)) order[ind] = 10000 order_letters[ind] = string.ascii_uppercase[i] result.order = ''.join(order_letters) # result.order_key = result.order_key[1:-1] # TODO When tree is finalized create relative word order (alphabet)! return result # def set_root(self): # if len(self.array[0]) > 1: # self.root = '&'.join(self.array[0]) # else: # # output_string = create_output_strings[0](node) # self.root = self.array[0][0]