from .basicblock import BasicBlock

class Function:
   def __init__(self, name):
      self.name = name
      self.bbs = []
      self.entry = None

   def __repr__(self):
      return 'Function {0}'.format(self.name)

   def addBB(self, bb):
      self.bbs.append(bb)
      bb.parent = self
   addBasicBlock = addBB

   def removeBasicBlock(self, bb):
      self.bbs.remove(bb)
      bb.parent = None

   def getBBs(self):
      return self.bbs
   BasicBlocks = property(getBBs)