remove h5py dependency

Author: Ehsan Totoni

hpat/io/pio.py

@@ -16,7 +16,7 @@
 import hpat.io
 from hpat.io import pio_api, pio_lower
 from hpat.utils import find_str_const, debug_prints
-import h5py
+
 
 
 def remove_h5(rhs, lives, call_list):

hpat/io/pio_api.py

@@ -5,14 +5,14 @@
 from numba.typing.templates import infer_global, AbstractTemplate, AttributeTemplate, bound_function
 from numba.typing import signature
 from llvmlite import ir as lir
-import h5py
 from numba.extending import register_model, models, infer_getattr, infer, intrinsic
 from hpat.str_ext import string_type
 import hpat
 from hpat.utils import unliteral_all
 import hpat.io
 
 if hpat.config._has_h5py:
+    import h5py
     from hpat.io import _hdf5
     import llvmlite.binding as ll
     ll.add_symbol('hpat_h5_read_filter', _hdf5.hpat_h5_read_filter)
@@ -49,11 +49,12 @@ def __init__(self):
 
 h5file_data_type = types.int64
 
-# hid_t is 32bit in 1.8 but 64bit in 1.10
-if h5py.version.hdf5_version_tuple[1] == 8:
-    h5file_data_type = types.int32
-else:
-    assert h5py.version.hdf5_version_tuple[1] == 10
+if hpat.config._has_h5py:
+    # hid_t is 32bit in 1.8 but 64bit in 1.10
+    if h5py.version.hdf5_version_tuple[1] == 8:
+        h5file_data_type = types.int32
+    else:
+        assert h5py.version.hdf5_version_tuple[1] == 10
 
 
 @register_model(H5FileType)
@@ -192,11 +193,12 @@ def generic(self, args, kws):
         return signature(ret_typ, *args)
 
 
-@infer_global(h5py.File)
-class H5File(AbstractTemplate):
-    def generic(self, args, kws):
-        assert not kws
-        return signature(h5file_type, *unliteral_all(args))
+if hpat.config._has_h5py:
+    @infer_global(h5py.File)
+    class H5File(AbstractTemplate):
+        def generic(self, args, kws):
+            assert not kws
+            return signature(h5file_type, *unliteral_all(args))
 
 
 @infer_global(h5size)

hpat/io/pio_lower.py

@@ -8,11 +8,12 @@
 from hpat.io.pio_api import (h5file_type, h5dataset_or_group_type, h5dataset_type,
                           h5group_type)
 from hpat.str_ext import string_type, gen_get_unicode_chars, gen_std_str_to_unicode
-import h5py
+
 from llvmlite import ir as lir
 import llvmlite.binding as ll
 import hpat.io
 if hpat.config._has_h5py:
+    import h5py
     from hpat.io import _hdf5
     ll.add_symbol('hpat_h5_open', _hdf5.hpat_h5_open)
     ll.add_symbol('hpat_h5_open_dset_or_group_obj', _hdf5.hpat_h5_open_dset_or_group_obj)
@@ -29,11 +30,12 @@
 
 h5file_lir_type = lir.IntType(64)
 
-# hid_t is 32bit in 1.8 but 64bit in 1.10
-if h5py.version.hdf5_version_tuple[1] == 8:
-    h5file_lir_type = lir.IntType(32)
-else:
-    assert h5py.version.hdf5_version_tuple[1] == 10
+if hpat.config._has_h5py:
+    # hid_t is 32bit in 1.8 but 64bit in 1.10
+    if h5py.version.hdf5_version_tuple[1] == 8:
+        h5file_lir_type = lir.IntType(32)
+    else:
+        assert h5py.version.hdf5_version_tuple[1] == 10
 
 h5g_close = types.ExternalFunction("h5g_close", types.none(h5group_type))
 
@@ -48,19 +50,20 @@ def h5_open_dset_lower(context, builder, sig, args):
     return builder.call(fn, [fg_id, dset_name])
 
 
-@lower_builtin(h5py.File, string_type, string_type)
-@lower_builtin(h5py.File, string_type, string_type, types.int64)
-def h5_open(context, builder, sig, args):
-    fname = args[0]
-    mode = args[1]
-    fname = gen_get_unicode_chars(context, builder, fname)
-    mode = gen_get_unicode_chars(context, builder, mode)
-
-    is_parallel = context.get_constant(types.int64, 0) if len(args) < 3 else args[2]
-    fnty = lir.FunctionType(h5file_lir_type, [lir.IntType(
-        8).as_pointer(), lir.IntType(8).as_pointer(), lir.IntType(64)])
-    fn = builder.module.get_or_insert_function(fnty, name="hpat_h5_open")
-    return builder.call(fn, [fname, mode, is_parallel])
+if hpat.config._has_h5py:
+    @lower_builtin(h5py.File, string_type, string_type)
+    @lower_builtin(h5py.File, string_type, string_type, types.int64)
+    def h5_open(context, builder, sig, args):
+        fname = args[0]
+        mode = args[1]
+        fname = gen_get_unicode_chars(context, builder, fname)
+        mode = gen_get_unicode_chars(context, builder, mode)
+
+        is_parallel = context.get_constant(types.int64, 0) if len(args) < 3 else args[2]
+        fnty = lir.FunctionType(h5file_lir_type, [lir.IntType(
+            8).as_pointer(), lir.IntType(8).as_pointer(), lir.IntType(64)])
+        fn = builder.module.get_or_insert_function(fnty, name="hpat_h5_open")
+        return builder.call(fn, [fname, mode, is_parallel])
 
 
 @lower_builtin(pio_api.h5size, h5dataset_or_group_type, types.int32)