pyFTS/tree.py

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