From c514ad9b097ef73b14abbf4f5af3617c54dbb154 Mon Sep 17 00:00:00 2001 From: Nathan Binkert Date: Thu, 9 Sep 2010 14:15:41 -0700 Subject: code_formatter: make it easier to insert whitespace a newline by just doing "code()". indent() and dedent() now take a "count" parameter to indent/dedent multiple levels. --- src/python/m5/util/code_formatter.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) (limited to 'src/python/m5') diff --git a/src/python/m5/util/code_formatter.py b/src/python/m5/util/code_formatter.py index 47106e0d9..023e189cd 100644 --- a/src/python/m5/util/code_formatter.py +++ b/src/python/m5/util/code_formatter.py @@ -131,12 +131,12 @@ class code_formatter(object): if args: self.__call__(args) - def indent(self): - self._indent_level += self._indent_spaces + def indent(self, count=1): + self._indent_level += self._indent_spaces * count - def dedent(self): - assert self._indent_level >= self._indent_spaces - self._indent_level -= self._indent_spaces + def dedent(self, count=1): + assert self._indent_level >= (self._indent_spaces * count) + self._indent_level -= self._indent_spaces * count def fix(self, status): previous = self._fix_newlines @@ -200,10 +200,14 @@ class code_formatter(object): initial_newline = False - def insert_newline(self): - self._data.append('\n') + def __call__(self, *args, **kwargs): + if not args: + self._data.append('\n') + return + + format = args[0] + args = args[1:] - def __call__(self, format, *args, **kwargs): frame = inspect.currentframe().f_back l = lookup(self, frame, *args, **kwargs) -- cgit v1.2.3