Adding Array.__repr__

Author: pavanky

arrayfire/array.py

@@ -1018,17 +1018,18 @@ def to_list(self, row_major=False):
 
     def __repr__(self):
         """
-        Displays the meta data of the arrayfire array.
+        Displays the meta data and contents of  the arrayfire array.
 
         Note
         ----
-        Use arrayfire.display(a) to display the contents of the array.
+        You can also use af.display(a, pres) to display the contents of the array with better precision.
         """
-        # Having __repr__ directly print things is a bad idea
-        # Placeholder for when af_array_to_string is available
-        # safe_call(backend.get().af_array_to_string...
-        return 'Type: arrayfire.Array()\nShape: %s\nType char: %s' % \
-            (self.dims(), to_typecode[self.type()])
+
+        arr_str = ct.c_char_p(0)
+        safe_call(backend.get().af_array_to_string(ct.pointer(arr_str), "", self.arr, 4, True))
+
+        return 'Type: arrayfire.Array()\nType: %s\n' % \
+            (to_typename[self.type()]) + to_str(arr_str)
 
     def __array__(self):
         """

arrayfire/util.py

@@ -115,3 +115,14 @@ def get_version():
              Dtype.u64.value : ct.c_ulonglong,
              Dtype.c32.value : ct.c_float * 2,
              Dtype.c64.value : ct.c_double * 2}
+
+to_typename = {Dtype.f32.value : 'float',
+               Dtype.f64.value : 'double',
+               Dtype.b8.value : 'bool',
+               Dtype.u8.value : 'unsigned char',
+               Dtype.s32.value : 'int',
+               Dtype.u32.value : 'unsigned int',
+               Dtype.s64.value : 'long int',
+               Dtype.u64.value : 'unsigned long int',
+               Dtype.c32.value : 'float complex',
+               Dtype.c64.value : 'double complex'}