Newer
Older
from ._library import File, VideoReader, VideoWriter, HDF5File
from . import version
from .version import module as __version__
from .version import api as __api_version__
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
import os
def create_directories_save(directory, dryrun=False):
"""Creates a directory if it does not exists, with concurrent access support.
This function will also create any parent directories that might be required.
If the dryrun option is selected, it does not actually create the directory,
but just writes the (Linux) command that would have been executed.
Parameters:
directory
The directory that you want to create.
dryrun
Only write the command, but do not execute it.
"""
try:
if dryrun:
print("[dry-run] mkdir -p '%s'" % directory)
else:
if directory and not os.path.exists(directory): os.makedirs(directory)
except OSError as exc: # Python >2.5
import errno
if exc.errno != errno.EEXIST:
raise
def load(inputs):
"""Loads the contents of a file, an iterable of files, or an iterable of
:py:class:`bob.io.File`'s into a :py:class:`numpy.ndarray`.
Parameters:
inputs
This might represent several different entities:
1. The name of a file (full path) from where to load the data. In this
case, this assumes that the file contains an array and returns a loaded
numpy ndarray.
2. An iterable of filenames to be loaded in memory. In this case, this
would assume that each file contains a single 1D sample or a set of 1D
samples, load them in memory and concatenate them into a single and
returned 2D numpy ndarray.
3. An iterable of :py:class:`bob.io.File`. In this case, this would assume
that each :py:class:`bob.io.File` contains a single 1D sample or a set
of 1D samples, load them in memory if required and concatenate them into
a single and returned 2D numpy ndarray.
4. An iterable with mixed filenames and :py:class:`bob.io.File`. In this
case, this would returned a 2D :py:class:`numpy.ndarray`, as described
by points 2 and 3 above.
"""
from collections import Iterable
import numpy
from .utils import is_string
if is_string(inputs):
return File(inputs, 'r').read()
elif isinstance(inputs, Iterable):
retval = []
for obj in inputs:
if is_string(obj):
retval.append(load(obj))
elif isinstance(obj, File):
retval.append(obj.read())
else:
raise TypeError("Iterable contains an object which is not a filename nor a bob.io.File.")
return numpy.vstack(retval)
else:
raise TypeError("Unexpected input object. This function is expecting a filename, or an iterable of filenames and/or bob.io.File's")
def merge(filenames):
"""Converts an iterable of filenames into an iterable over read-only
bob.io.File's.
Parameters:
filenames
This might represent:
1. A single filename. In this case, an iterable with a single
:py:class:`bob.io.File` is returned.
2. An iterable of filenames to be converted into an iterable of
:py:class:`bob.io.File`'s.
"""
from collections import Iterable
from .utils import is_string
if is_string(filenames):
return [File(filenames, 'r')]
elif isinstance(filenames, Iterable):
return [File(k, 'r') for k in filenames]
else:
raise TypeError("Unexpected input object. This function is expecting an iterable of filenames.")
def save(array, filename, create_directories = False):
"""Saves the contents of an array-like object to file.
Effectively, this is the same as creating a :py:class:`bob.io.File` object
with the mode flag set to `w` (write with truncation) and calling
:py:meth:`bob.io.File.write` passing `array` as parameter.
Parameters:
array
The array-like object to be saved on the file
filename
The name of the file where you need the contents saved to
create_directories
Automatically generate the directories if required
"""
# create directory if not existent yet
if create_directories:
create_directories_save(os.path.dirname(filename))
return File(filename, 'w').write(array)
# Just to make it homogenous with the C++ API
write = save
def append(array, filename):
"""Appends the contents of an array-like object to file.
Effectively, this is the same as creating a :py:class:`bob.io.File` object
with the mode flag set to `a` (append) and calling
:py:meth:`bob.io.File.append` passing `array` as parameter.
Parameters:
array
The array-like object to be saved on the file
filename
The name of the file where you need the contents saved to
"""
return File(filename, 'a').append(array)
def peek(filename):
"""Returns the type of array (frame or sample) saved in the given file.
Effectively, this is the same as creating a :py:class:`bob.io.File` object
with the mode flag set to `r` (read-only) and returning
:py:attr:`bob.io.File.describe()`.
Parameters:
filename
The name of the file to peek information from
"""
return File(filename, 'r').describe()
def peek_all(filename):
"""Returns the type of array (for full readouts) saved in the given file.
Effectively, this is the same as creating a :py:class:`bob.io.File` object
with the mode flag set to `r` (read-only) and returning
:py:attr:`bob.io.File.describe(all=True)`.
Parameters:
filename
The name of the file to peek information from
"""
return File(filename, 'r').describe(all=True)
# Keeps compatibility with the previously existing API
open = File
def get_include():
"""Returns the directory containing the C/C++ API include directives"""
return __import__('pkg_resources').resource_filename(__name__, 'include')
# gets sphinx autodoc done right - don't remove it
__all__ = [_ for _ in dir() if not _.startswith('_')]