Skip to content
Snippets Groups Projects

Add own implementation for printing an unknown function

Merged Stephan Seitz requested to merge seitz/pystencils:function-call-printing into master
2 files
+ 3
4
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -385,7 +385,8 @@ class CustomSympyPrinter(CCodePrinter):
elif expr.func == int_div:
return "((%s) / (%s))" % (self._print(expr.args[0]), self._print(expr.args[1]))
else:
return super(CustomSympyPrinter, self)._print_Function(expr)
arg_str = ', '.join(self._print(a) for a in expr.args)
return f'{expr.name}({arg_str})'
def _typed_number(self, number, dtype):
res = self._print(number)
Loading