compiler: Add BitwiseNot and BitwiseXor

Author: FabioLuporini

devito/ir/cgen/printer.py

@@ -276,6 +276,16 @@ def _print_Abs(self, expr):
             return f"fabs({self._print(arg)})"
         return self._print_fmath_func('abs', expr)
 
+    def _print_BitwiseNot(self, expr):
+        # Unary function, single argument
+        arg = expr.args[0]
+        return f'~{self._print(arg)}'
+
+    def _print_BitwiseXor(self, expr):
+        # Binary function
+        arg0, arg1 = expr.args
+        return f'{self._print(arg0)} ^ {self._print(arg1)}'
+
     def _print_Add(self, expr, order=None):
         """"
         Print an addition.

devito/symbolics/extended_sympy.py

@@ -7,6 +7,7 @@
 import sympy
 from sympy import Expr, Function, Number, Tuple, cacheit, sympify
 from sympy.core.decorators import call_highest_priority
+from sympy.logic.boolalg import BooleanFunction
 
 from devito.finite_differences.elementary import Min, Max
 from devito.tools import (Pickable, Bunch, as_tuple, is_integer, float2,  # noqa
@@ -16,13 +17,13 @@
 from devito.types import Symbol
 from devito.types.basic import Basic
 
-__all__ = ['CondEq', 'CondNe', 'IntDiv', 'CallFromPointer',  # noqa
-           'CallFromComposite', 'FieldFromPointer', 'FieldFromComposite',
-           'ListInitializer', 'Byref', 'IndexedPointer', 'Cast', 'DefFunction',
-           'MathFunction', 'InlineIf', 'ReservedWord', 'Keyword', 'String',
-           'Macro', 'Class', 'MacroArgument', 'Deref', 'Namespace',
-           'Rvalue', 'Null', 'SizeOf', 'rfunc', 'BasicWrapperMixin', 'ValueLimit',
-           'VectorAccess']
+__all__ = ['CondEq', 'CondNe', 'BitwiseNot', 'BitwiseXor', 'IntDiv',  # noqa
+           'CallFromPointer', 'CallFromComposite', 'FieldFromPointer',
+           'FieldFromComposite', 'ListInitializer', 'Byref', 'IndexedPointer',
+           'Cast', 'DefFunction', 'MathFunction', 'InlineIf', 'ReservedWord',
+           'Keyword', 'String', 'Macro', 'Class', 'MacroArgument', 'Deref',
+           'Namespace', 'Rvalue', 'Null', 'SizeOf', 'rfunc', 'BasicWrapperMixin',
+           'ValueLimit', 'VectorAccess']
 
 
 class CondEq(sympy.Eq):
@@ -63,6 +64,17 @@ def negated(self):
         return CondEq(*self.args, evaluate=False)
 
 
+class BitwiseNot(BooleanFunction):
+    pass
+
+
+class BitwiseXor(BooleanFunction):
+
+    # Enforce two args
+    def __new__(cls, arg0, arg1, **kwargs):
+        return super().__new__(cls, arg0, arg1, **kwargs)
+
+
 class IntDiv(sympy.Expr):
 
     """