Merge branch 'accept_file_handle' into 'sliced_stream'

Accept file handle

See merge request !10

bob/io/stream/stream_file.py

@@ -8,9 +8,11 @@ from bob.ip.stereo import load_camera_config
 from .config import load_data_config
 
 class StreamFile:
-    def __init__(self, hdf5_file_path=None, data_format_config_file_path=None, camera_config_file_path=None, mode="r"):
-        if hdf5_file_path is not None:
-            self.hdf5_file = HDF5File(hdf5_file_path, mode)
+    def __init__(self, hdf5_file=None, data_format_config_file_path=None, camera_config_file_path=None, mode="r"):
+        if isinstance(hdf5_file, str):  # case string: it is a path, use bob.io.base.HDF5File
+            self.hdf5_file = HDF5File(hdf5_file, mode)
+        elif hdf5_file is not None:  # otherwise expect an opened file object (h5py.File, etc...)
+            self.hdf5_file = hdf5_file
         else:
             self.hdf5_file = None
         if data_format_config_file_path is not None:
@@ -26,16 +28,16 @@ class StreamFile:
         pass
 
     def __exit__(self, exc_type, exc_val, exc_tb):
-        if self.hdf5_file is not None:
+        if isinstance(self.hdf5_file, HDF5File):
             self.hdf5_file.close()
 
     # set source
     def set_source(
-        self, hdf5_file_path=None, data_format_config_file_path=None, camera_config_file_path=None, mode="r"
+        self, hdf5_file=None, data_format_config_file_path=None, camera_config_file_path=None, mode="r"
     ):
 
         self.__init__(
-            hdf5_file_path=hdf5_file_path,
+            hdf5_file=hdf5_file,
             data_format_config_file_path=data_format_config_file_path,
             camera_config_file_path=camera_config_file_path,
             mode=mode,