Commit 3c30fe5b authored by Vincent POLLET's avatar Vincent POLLET
Browse files

Merge branch 'accept_file_handle' into 'sliced_stream'

Accept file handle

See merge request !10
parents a0a92257 0358e8f5
Pipeline #42748 failed with stage
in 11 minutes and 31 seconds
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment