compiler: Add more bit-level operations

Author: FabioLuporini

devito/ir/cgen/printer.py

@@ -281,10 +281,9 @@ def _print_BitwiseNot(self, expr):
         arg = expr.args[0]
         return f'~{self._print(arg)}'
 
-    def _print_BitwiseXor(self, expr):
-        # Binary function
+    def _print_BitwiseBinaryOp(self, expr):
         arg0, arg1 = expr.args
-        return f'{self._print(arg0)} ^ {self._print(arg1)}'
+        return f'{self._print(arg0)} {expr.op} {self._print(arg1)}'
 
     def _print_Add(self, expr, order=None):
         """"

devito/symbolics/extended_sympy.py

@@ -17,13 +17,14 @@
 from devito.types import Symbol
 from devito.types.basic import Basic
 
-__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']
+__all__ = ['CondEq', 'CondNe', 'BitwiseNot', 'BitwiseXor', 'BitwiseAnd',  # noqa
+           'LeftShift', 'RightShift', 'IntDiv', '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):
@@ -65,16 +66,33 @@ def negated(self):
 
 
 class BitwiseNot(BooleanFunction):
-    pass
+    op = '~'
 
 
-class BitwiseXor(BooleanFunction):
+class BitwiseBinaryOp(BooleanFunction):
+    op = ''
 
     # Enforce two args
     def __new__(cls, arg0, arg1, **kwargs):
         return super().__new__(cls, arg0, arg1, **kwargs)
 
 
+class BitwiseXor(BitwiseBinaryOp):
+    op = '^'
+
+
+class BitwiseAnd(BitwiseBinaryOp):
+    op = '&'
+
+
+class LeftShift(BitwiseBinaryOp):
+    op = '<<'
+
+
+class RightShift(BitwiseBinaryOp):
+    op = '>>'
+
+
 class IntDiv(sympy.Expr):
 
     """