From da54b3ec67316bd9ab2fe9facf51adb7508f75c5 Mon Sep 17 00:00:00 2001
From: Nils Kohl <nils.kohl@fau.de>
Date: Wed, 13 Nov 2019 15:47:48 +0100
Subject: [PATCH] Added integer function that does a post increment
---
pystencils/backends/cbackend.py | 4 +++-
pystencils/integer_functions.py | 1 +
2 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/pystencils/backends/cbackend.py b/pystencils/backends/cbackend.py
index f9bf0b12..e22c21c7 100644
--- a/pystencils/backends/cbackend.py
+++ b/pystencils/backends/cbackend.py
@@ -13,7 +13,7 @@ except ImportError:
from sympy.printing.ccode import CCodePrinter # for sympy versions < 1.1
from pystencils.integer_functions import bitwise_xor, bit_shift_right, bit_shift_left, bitwise_and, \
- bitwise_or, modulo_ceil, int_div, int_power_of_2, int_mod
+ bitwise_or, modulo_ceil, int_div, int_power_of_2, int_mod, inc_post
from pystencils.astnodes import Node, KernelFunction
from pystencils.data_types import create_type, PointerType, get_type_of_expression, VectorType, cast_func, \
vector_memory_access, reinterpret_cast_func
@@ -311,6 +311,8 @@ class CustomSympyPrinter(CCodePrinter):
return "((%s) / (%s))" % (self._print(expr.args[0]), self._print(expr.args[1]))
elif expr.func == int_mod:
return "((%s) %% (%s))" % (self._print(expr.args[0]), self._print(expr.args[1]))
+ elif expr.func == inc_post:
+ return "(%s++)" % (self._print(expr.args[0]))
else:
return super(CustomSympyPrinter, self)._print_Function(expr)
diff --git a/pystencils/integer_functions.py b/pystencils/integer_functions.py
index 5b243622..eb60dde7 100644
--- a/pystencils/integer_functions.py
+++ b/pystencils/integer_functions.py
@@ -11,6 +11,7 @@ bitwise_or = sp.Function("bitwise_or")
int_div = sp.Function("int_div")
int_power_of_2 = sp.Function("int_power_of_2")
int_mod = sp.Function("int_mod")
+inc_post = sp.Function("inc_post")
# noinspection PyPep8Naming
--
GitLab