Source code
Revision control
Copy as Markdown
Other Tools
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
import sys
from ipdl.ast import Visitor
class CodePrinter:
def __init__(self, outf=sys.stdout, indentCols=4):
self.outf = outf
self.col = 0
self.indentCols = indentCols
def write(self, str):
self.outf.write(str)
def printdent(self, str=""):
self.write((" " * self.col) + str)
def println(self, str=""):
self.write(str + "\n")
def printdentln(self, str):
self.write((" " * self.col) + str + "\n")
def indent(self):
self.col += self.indentCols
def dedent(self):
self.col -= self.indentCols
# -----------------------------------------------------------------------------
class IPDLCodeGen(CodePrinter, Visitor):
"""Spits back out equivalent IPDL to the code that generated this.
Also known as pretty-printing."""
def __init__(self, outf=sys.stdout, indentCols=4, printed=set()):
CodePrinter.__init__(self, outf, indentCols)
self.printed = printed
def visitTranslationUnit(self, tu):
self.printed.add(tu.filename)
self.println("//\n// Automatically generated by ipdlc\n//")
Visitor.visitTranslationUnit(self, tu) # NOQA: F821
def visitCxxInclude(self, inc):
self.println('include "' + inc.file + '";')
def visitProtocolInclude(self, inc):
self.println('include protocol "' + inc.file + '";')
if inc.tu.filename not in self.printed:
self.println("/* Included file:")
IPDLCodeGen(
outf=self.outf, indentCols=self.indentCols, printed=self.printed
).visitTranslationUnit(inc.tu)
self.println("*/")
def visitProtocol(self, p):
self.println()
for namespace in p.namespaces:
namespace.accept(self)
self.println("%s protocol %s\n{" % (p.sendSemantics, p.name))
self.indent()
for mgs in p.managesStmts:
mgs.accept(self)
if len(p.managesStmts):
self.println()
for msgDecl in p.messageDecls:
msgDecl.accept(self)
self.println()
self.dedent()
self.println("}")
self.write("}\n" * len(p.namespaces))
def visitManagerStmt(self, mgr):
self.printdentln("manager " + mgr.name + ";")
def visitManagesStmt(self, mgs):
self.printdentln("manages " + mgs.name + ";")
def visitMessageDecl(self, msg):
self.printdent("%s %s %s(" % (msg.sendSemantics, msg.direction, msg.name))
for i, inp in enumerate(msg.inParams):
inp.accept(self)
if i != (len(msg.inParams) - 1):
self.write(", ")
self.write(")")
if 0 == len(msg.outParams):
self.println(";")
return
self.println()
self.indent()
self.printdent("returns (")
for i, outp in enumerate(msg.outParams):
outp.accept(self)
if i != (len(msg.outParams) - 1):
self.write(", ")
self.println(");")
self.dedent()