68 lines
1.4 KiB
Python
68 lines
1.4 KiB
Python
class CFG_Node:
|
|
__counter = 1
|
|
|
|
def __init__(self, ast_node = None):
|
|
self.ast_node = ast_node
|
|
self.children = set()
|
|
self.parents = set()
|
|
|
|
self.id = CFG_Node.__counter
|
|
CFG_Node.__counter += 1
|
|
|
|
def get_children(self):
|
|
return self.children
|
|
|
|
def get_parents(self):
|
|
return self.parents
|
|
|
|
def add_child(self, child: CFG_Node, propagate = True):
|
|
if propagate:
|
|
child.parents.add(self)
|
|
self.children.add(child)
|
|
|
|
def add_parent(self, parent: CFG_Node, propagate = True):
|
|
if propagate:
|
|
parent.add_child(self)
|
|
self.parents.add(parent)
|
|
|
|
def remove_child(self, child: CFG_Node, propagate = True):
|
|
if propagate:
|
|
child.parents.remove(self)
|
|
self.children.remove(child)
|
|
|
|
def remove_parent(self, parent: CFG_Node, propagate = True):
|
|
if propagate:
|
|
parent.children.remove(self)
|
|
self.parents.remove(parent)
|
|
|
|
|
|
class CFG_START(CFG_Node):
|
|
def dot_shape(self):
|
|
return "box"
|
|
|
|
def dot_label(self):
|
|
return "START"
|
|
|
|
|
|
class CFG_END(CFG_Node):
|
|
def dot_shape(self):
|
|
return "box"
|
|
|
|
def dot_label(self):
|
|
return "END"
|
|
|
|
|
|
class CFG_DIAMOND(CFG_Node):
|
|
def dot_shape(self):
|
|
return "diamond"
|
|
|
|
|
|
class CFG_CALL(CFG_Node):
|
|
def dot_shape(self):
|
|
return "box"
|
|
|
|
|
|
class CFG_RETURN(CFG_Node):
|
|
def dot_shape(self):
|
|
return "box"
|