Skip to content
Snippets Groups Projects
Commit ec188036 authored by Vincent POLLET's avatar Vincent POLLET
Browse files

Write timestamps to file as dataset attributes along with data

parent 4669efb1
Branches
Tags
1 merge request!19Write timestamps to file as dataset attributes along with data
Pipeline #54317 failed
...@@ -862,7 +862,16 @@ class StreamSave(StreamFilter): ...@@ -862,7 +862,16 @@ class StreamSave(StreamFilter):
raise ValueError("Output file is not a valid StreamFile.") raise ValueError("Output file is not a valid StreamFile.")
def put(self, data, timestamp=None): def put(self, data, timestamp=None):
self.file.put_frame(self.name, data) """Pass data and timestamp to the :obj:`~bob.io.stream.StreamFile` to write to disk.
Parameters
----------
data : :obj:`numpy.ndarray`
data to write to file.
timestamp : int or float
Timestamp of `data`, by default None.
"""
self.file.put_frame(self.name, data, timestamp)
################################################################################ ################################################################################
......
...@@ -200,7 +200,7 @@ class StreamFile: ...@@ -200,7 +200,7 @@ class StreamFile:
# TODO rotate if requested # TODO rotate if requested
return data return data
def put_frame(self, name, data): def put_frame(self, name, data, timestamp=None):
"""Appends `data` (a frame of a stream) to the hdf5 file. """Appends `data` (a frame of a stream) to the hdf5 file.
Parameters Parameters
...@@ -211,3 +211,9 @@ class StreamFile: ...@@ -211,3 +211,9 @@ class StreamFile:
Data frame to append. Data frame to append.
""" """
self.hdf5_file.append(name, data) self.hdf5_file.append(name, data)
if timestamp is not None:
try:
previous_timestamps = self.hdf5_file.get_attribute("timestamps", name)
except RuntimeError: # No previous timestamps
previous_timestamps = []
self.hdf5_file.set_attribute("timestamps", np.append(previous_timestamps, timestamp), name)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment