Simplifying the array constructor using python maps

Author: pavanky

arrayfire/array.py

@@ -105,22 +105,7 @@ def __init__(self, src=None, dims=(0,)):
                 idims = [buf_len, 1, 1, 1]
                 numdims = 1
 
-            if type_char == 'f':
-                dtype = f32
-            elif type_char == 'd':
-                dtype = f64
-            elif type_char == 'b':
-                dtype = b8
-            elif type_char == 'B':
-                dtype = u8
-            elif type_char == 'i':
-                dtype = s32
-            elif type_char == 'I':
-                dtype = u32
-            elif type_char == 'l':
-                dtype = s64
-            elif type_char == 'L':
-                dtype = u64
+            dtype = to_dtype[type_char]
 
             self.arr = create_array(buf, numdims, idims, dtype)
 

arrayfire/util.py

@@ -30,3 +30,13 @@ def safe_call(af_error):
         c_err_len = c_longlong(0)
         clib.af_get_last_error(pointer(c_err_str), pointer(c_err_len))
         raise RuntimeError(to_str(c_err_str), af_error)
+
+
+to_dtype = {'f' : f32,
+            'd' : f64,
+            'b' : b8,
+            'B' : u8,
+            'i' : s32,
+            'I' : u32,
+            'l' : s64,
+            'L' : u64}