compiler: fix printing of nd pointers

Author: mloubout

devito/ir/cgen/printer.py

@@ -2,6 +2,7 @@
 Utilities to turn SymPy objects into C strings.
 """
 from contextlib import suppress
+from ctypes import _Pointer
 
 import numpy as np
 import sympy
@@ -110,11 +111,17 @@ def parenthesize(self, item, level, strict=False):
         return super().parenthesize(item, level, strict=strict)
 
     def _print_PyCPointerType(self, expr):
-        ctype = f'{self._print_type(expr._type_)}'
+        base_type, nstart = expr, 0
+        while issubclass(base_type, _Pointer):
+            base_type = base_type._type_
+            nstart += 1
+
+        ctype = f'{self._print_type(base_type)}'
+        stars = '*' * nstart
         if ctype.endswith('*'):
-            return f'{ctype}*'
+            return f'{ctype}{stars}'
         else:
-            return f'{ctype} *'
+            return f'{ctype} {stars}'
 
     def _print_type(self, expr):
         with suppress(TypeError):

tests/test_iet.py

@@ -1,4 +1,4 @@
-from ctypes import c_void_p
+from ctypes import POINTER, c_void_p
 
 import cgen
 import numpy as np
@@ -20,6 +20,7 @@
     FLOAT, Byref, Class, FieldFromComposite, InlineIf, ListInitializer, Macro, SizeOf,
     String
 )
+from devito.symbolics.extended_dtypes import c_complex
 from devito.tools import CustomDtype, as_tuple, dtype_to_ctype
 from devito.types import Array, CustomDimension, LocalObject, Pointer, Symbol
 from devito.types.misc import FunctionMap
@@ -528,6 +529,16 @@ def test_codegen_quality0():
     assert foo1.parameters[0] is a
 
 
+def test_complex_array():
+    grid = Grid(shape=(4, 4, 4))
+    _, y, z = grid.dimensions
+
+    a = Array(name='a', dimensions=grid.dimensions, dtype=POINTER(c_complex))
+
+    assert str(Definition(a)) == \
+        "float _Complex **restrict a_vec __attribute__ ((aligned (64)));"
+
+
 def test_special_array_definition():
 
     class MyArray(Array):