Esempio n. 1
0
 def addScheduleTree(self, FlatTree):
     length = len(self._nodes)
     if isinstance(FlatTree, str):
         FlatTree = StringIO(FlatTree)
         FlatTree = ResultTrees.ImTreeUtil.read_csv(FlatTree)
     self.FlatTree = self.FlatTree.append(FlatTree.iloc[1:]).reset_index(drop=True)
     it = FlatTree.iterrows()
     i = 0
     data = ImNodeData(it.__next__()) # Throw Away root
     data = ImNodeData(it.__next__())
     self.create_node(tag = str(data), identifier=length, data=data, parent=0)
     self.get_node(0).data.ExposureAmount += data.ExposureAmount
     self.get_node(0).tag=str(self.get_node(0).data)
     for row in it:
         data=ImNodeData(row)
         i += 1
         node = Node(tag=str(data), identifier=i+length, data=data)
         parent_id = self.identify_parent_id(node)
         self.add_node(node, parent_id)
Esempio n. 2
0
 def __init__(self, FlatTree):
     self.hasEulerAllocation = False
     self.hasStandaloneAllocation = False
     #if it isn't a String it should be a pandas Dataframe
     if isinstance(FlatTree, str):
         FlatTree = StringIO(FlatTree)
         FlatTree = ResultTrees.ImTreeUtil.read_csv(FlatTree)
     self.FlatTree = FlatTree
     super(ImTree, self).__init__()
     it = FlatTree.iterrows()
     i = 0
     data = ImNodeData(it.__next__())
     self.create_node(tag=str(data), identifier=i, data=data)
     while i<(FlatTree.shape[0] - 1):
         i = i + 1
         data = ImNodeData(it.__next__())
         node = Node(tag=str(data), identifier=i, data=data)
         parent_id = self.identify_parent_id(node)
         self.create_node(tag = str(data), identifier=i, data=data, parent=parent_id)