core
/
eccodes
mirror of https://github.com/ecmwf/eccodes.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Better for() generation

This commit is contained in:
kevstone 2024-01-18 14:53:47 +00:00
parent 42ad179867
commit ce73909da6
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/clang_convert/code_object/for_statement.py
View File

@ -62,18 +62,19 @@ class ForStatement(code_interface.CodeInterface):
return self._statement
def as_lines(self):
lines = ["for("]
init_string = cond_string = iter_string = ""
if self._init_statement:
lines.extend(self._init_statement.as_lines())
if not lines[-1].endswith(";"):
lines[-1] += ";"
init_string = self._init_statement.as_string()
if not init_string.endswith(";"):
init_string += ";"
if self._condition:
lines.extend(self._condition.as_lines())
if not lines[-1].endswith(";"):
lines[-1] += ";"
cond_string = self._condition.as_string()
if not cond_string.endswith(";"):
cond_string += ";"
if self._iteration_expression:
lines.extend(self._iteration_expression.as_lines())
lines[-1] += ")"
iter_string = self._iteration_expression.as_string()
lines = [f"for({init_string}{cond_string}{iter_string})"]
if self._statement:
lines.extend(self._statement.as_lines())