84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
|
import numpy as np
|
||
|
|
import math
|
||
|
|
|
||
|
|
class Node:
|
||
|
|
def __init__(self, tuple, word, words, infertile):
|
||
|
|
self.tuple = tuple
|
||
|
|
self.children = []
|
||
|
|
self.word = word
|
||
|
|
self.infertile = infertile
|
||
|
|
words[str(word)] = self
|
||
|
|
|
||
|
|
def dc_not_too_big(self, g, generator, dual_curvature):
|
||
|
|
temp = np.matmul(generator, self.tuple)
|
||
|
|
if g == 0:
|
||
|
|
temp = [temp[i] for i in (0,1,8)]
|
||
|
|
elif g == 1:
|
||
|
|
temp = [temp[i] for i in (0,1,5)]
|
||
|
|
elif g == 2:
|
||
|
|
temp = [temp[i] for i in (1,4,10)]
|
||
|
|
elif g == 3:
|
||
|
|
temp = [temp[i] for i in (3,7,9)]
|
||
|
|
elif g == 4:
|
||
|
|
temp = [temp[i] for i in (4,5,11)]
|
||
|
|
elif g == 5:
|
||
|
|
temp = [temp[i] for i in (6,8,10)]
|
||
|
|
elif g == 6:
|
||
|
|
temp = [temp[i] for i in (7,9,11)]
|
||
|
|
elif g == 7:
|
||
|
|
temp = [temp[i] for i in (3,6,7)]
|
||
|
|
elif g == 8:
|
||
|
|
temp = [temp[i] for i in (2,4,11)]
|
||
|
|
elif g == 9:
|
||
|
|
temp = [temp[i] for i in (2,6,10)]
|
||
|
|
elif g == 10:
|
||
|
|
temp = [temp[i] for i in (2,7,11)]
|
||
|
|
elif g == 11:
|
||
|
|
temp = [temp[i] for i in (2,6,7)]
|
||
|
|
elif g == 12:
|
||
|
|
temp = [temp[i] for i in (0,3,9)]
|
||
|
|
elif g == 13:
|
||
|
|
temp = [temp[i] for i in (1,4,5)]
|
||
|
|
elif g == 14:
|
||
|
|
temp = [temp[i] for i in (1,8,10)]
|
||
|
|
elif g == 15:
|
||
|
|
temp = [temp[i] for i in (2,4,10)]
|
||
|
|
elif g == 16:
|
||
|
|
temp = [temp[i] for i in (0,5,9)]
|
||
|
|
elif g == 17:
|
||
|
|
temp = [temp[i] for i in (0,3,8)]
|
||
|
|
elif g == 18:
|
||
|
|
temp = [temp[i] for i in (5,9,11)]
|
||
|
|
elif g == 19:
|
||
|
|
temp = [temp[i] for i in (3,6,8)]
|
||
|
|
return math.sqrt(temp[0]*temp[1] + temp[0]*temp[2] + temp[1]*temp[2]) < dual_curvature
|
||
|
|
|
||
|
|
def next_generation(self, words, dual_curvature, generators):
|
||
|
|
if self.infertile:
|
||
|
|
return [self]
|
||
|
|
if self.children == []:
|
||
|
|
for g,generator in enumerate(generators):
|
||
|
|
if len(self.word) == 0 or g != self.word[-1]:
|
||
|
|
self.children.append(Node(np.matmul(generator,self.tuple), self.word[:] + [g], words, not self.dc_not_too_big(g, generator, dual_curvature)))
|
||
|
|
if self.children == []:
|
||
|
|
return [self]
|
||
|
|
else:
|
||
|
|
return self.children
|
||
|
|
|
||
|
|
def next(self):
|
||
|
|
if self.children == []:
|
||
|
|
return [self]
|
||
|
|
else:
|
||
|
|
return self.children
|
||
|
|
|
||
|
|
def leaves(self):
|
||
|
|
current_leaves = [self]
|
||
|
|
while True:
|
||
|
|
new_leaves = []
|
||
|
|
for leaf in current_leaves:
|
||
|
|
new_leaves += leaf.next()
|
||
|
|
if current_leaves == new_leaves:
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
current_leaves = new_leaves
|
||
|
|
return current_leaves
|