101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
# This file is part of PeachPy package and is licensed under the Simplified BSD license.
|
|
# See license.rst for the full text of the license.
|
|
|
|
|
|
class CodeGenerator(object):
|
|
def __init__(self, use_tabs=True):
|
|
self.indentationLevel = 0
|
|
self.use_tabs = use_tabs
|
|
self.code = []
|
|
|
|
def indent(self):
|
|
self.indentationLevel += 1
|
|
return self
|
|
|
|
def dedent(self):
|
|
self.indentationLevel -= 1
|
|
return self
|
|
|
|
def add_line(self, string='', indent=None):
|
|
if indent == None:
|
|
indent = self.indentationLevel
|
|
elif indent < 0:
|
|
indent += self.indentationLevel
|
|
if string == '':
|
|
self.code.append(string)
|
|
else:
|
|
if self.use_tabs:
|
|
self.code.append("\t" * indent + string)
|
|
else:
|
|
self.code.append(" " * indent + string)
|
|
return self
|
|
|
|
def add_lines(self, lines, indent=None):
|
|
for line in lines:
|
|
self.add_line(line, indent)
|
|
|
|
def add_empty_lines(self, count):
|
|
for i in range(count):
|
|
self.add_line()
|
|
return self
|
|
|
|
def add_c_comment(self, lines, doxygen=False):
|
|
if isinstance(lines, str) and lines.find("\n") != -1:
|
|
lines = lines.split("\n")
|
|
if isinstance(lines, list) and len(lines) > 1:
|
|
if doxygen:
|
|
self.add_line("/**")
|
|
else:
|
|
self.add_line("/*")
|
|
for line in lines:
|
|
self.add_line(" * " + line)
|
|
self.add_line(" */")
|
|
else:
|
|
line = lines[0] if isinstance(lines, list) else str(lines)
|
|
if doxygen:
|
|
self.add_line("/** " + line + "*/")
|
|
else:
|
|
self.add_line("/* " + line + "*/")
|
|
|
|
def add_assembly_comment(self, lines, indent=None):
|
|
for line in lines:
|
|
self.add_line("; " + line, indent)
|
|
|
|
def add_fortran90_comment(self, lines, doxygen=False):
|
|
if isinstance(lines, str) and lines.find("\n") != -1:
|
|
lines = lines.split("\n")
|
|
elif isinstance(lines, str):
|
|
lines = [lines]
|
|
for index, line in enumerate(lines):
|
|
if doxygen:
|
|
if index == 0:
|
|
self.add_line("!> " + line)
|
|
else:
|
|
self.add_line("!! " + line)
|
|
else:
|
|
self.add_line("! " + line)
|
|
|
|
def add_csharp_comment(self, lines, doxygen=False):
|
|
if isinstance(lines, str) and lines.find("\n") != -1:
|
|
lines = lines.split("\n")
|
|
if isinstance(lines, list) and len(lines) > 1:
|
|
if not doxygen:
|
|
self.add_line("/*")
|
|
for line in lines:
|
|
if doxygen:
|
|
self.add_line("/// " + line)
|
|
else:
|
|
self.add_line(" * " + line)
|
|
if not doxygen:
|
|
self.add_line(" */")
|
|
else:
|
|
line = lines[0] if isinstance(lines, list) else str(lines)
|
|
if doxygen:
|
|
self.add_line("/// " + line)
|
|
else:
|
|
self.add_line("// " + line)
|
|
|
|
def get_code(self):
|
|
return "\n".join(self.code)
|
|
|