123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- import string
- from primitives_visitor import PrimitivesVisitor, Action
- class BootstrapVisitor(PrimitivesVisitor):
- def __init__(self, args):
- PrimitivesVisitor.__init__(self, args)
- for f in args:
- if "--prepend:" in f:
- self.prepend_name = f.split(":", 1)[1]
- break
- else:
- self.prepend_name = "auto"
- def dump(self):
- link_id = 0
- call = self.value(Action("call"))
- access = self.value(Action("access"))
- resolve = self.value(Action("resolve"))
- main = self.value("__main")
- self.dict(resolve, "var", main)
- self.dict(access, "var", resolve)
- self.dict(call, "func", access)
- self.dict(self.last_instruction, "next", call)
- output = []
- for t, data in self.output:
- if t == "N":
- output.append("Node %s_%s()\n" % (self.prepend_name, data))
- elif t == "V":
- name, value = data
- name = name if self.first != name else "initial_IP"
- output.append("Node %s_%s(%s)\n" % (self.prepend_name, name, value))
- elif t == "D":
- source, value, target = data
- source = source if self.first != source else "auto_initial_IP"
- target = target if self.first != target else "auto_initial_IP"
- source = "%s_%s" % (self.prepend_name, source if isinstance(source, int) else source)
- target = "%s_%s" % (self.prepend_name, target if isinstance(target, int) else target)
- # output.append("D %s,%s,%s\n" % (source, value, target))
- linkname = "%s_%s_%s" % (source, link_id, target)
- link_id += 1
- output.append("Edge _%s_0(%s, %s)\n" % (linkname, source, target))
- output.append("Node _%s_2(%s)\n" % (linkname, value))
- output.append("Edge _%s_1(_%s_0, _%s_2)\n" % (linkname, linkname, linkname))
- elif t == "E":
- name, source, target = data
- source = source if self.first != source else "auto_initial_IP"
- target = target if self.first != target else "auto_initial_IP"
- name = "%s_%s" % (self.prepend_name, name if isinstance(name, int) else name)
- source = "%s_%s" % (self.prepend_name, source if isinstance(source, int) else source)
- target = "%s_%s" % (self.prepend_name, target if isinstance(target, int) else target)
- output.append("Edge _%s(_%s, _%s)\n" % (name, source, target))
- return ''.join(output)
|