53 lines
1.2 KiB
Python
53 lines
1.2 KiB
Python
from pyFTS import *
|
|
from functools import reduce
|
|
import numpy as np
|
|
|
|
|
|
class FLRGTreeNode:
|
|
def __init__(self, value):
|
|
self.isRoot = False
|
|
self.children = []
|
|
self.value = value
|
|
|
|
def appendChild(self, child):
|
|
self.children.append(child)
|
|
|
|
def getChildren(self):
|
|
for child in self.children:
|
|
yield child
|
|
|
|
def paths(self, acc=[]):
|
|
if len(self.children) == 0:
|
|
yield [self.value] + acc
|
|
|
|
for child in self.children:
|
|
for leaf_path in child.paths([self.value] + acc): # these two
|
|
yield leaf_path
|
|
|
|
def getStr(self, k):
|
|
if self.isRoot:
|
|
tmp = str(self.value)
|
|
else:
|
|
tmp = "\\" + ("-" * k) + str(self.value)
|
|
for child in self.getChildren():
|
|
tmp = tmp + "\n" + child.getStr(k + 1)
|
|
return tmp
|
|
|
|
def __str__(self):
|
|
return self.getStr(0)
|
|
|
|
|
|
class FLRGTree:
|
|
def __init__(self):
|
|
self.root = FLRGTreeNode(None)
|
|
|
|
|
|
def flat(dados):
|
|
for inst in dados:
|
|
if isinstance(inst, (list, tuple)):
|
|
x = flat(inst)
|
|
for k in x:
|
|
yield k
|
|
else:
|
|
yield inst
|