1
2
3
4
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
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
# Authors: Travis Oliphant, Matthew Brett
"""
Module for reading and writing matlab (TM) .mat files
"""
import os
import sys
import warnings
from miobase import get_matfile_version, docfiller
from mio4 import MatFile4Reader, MatFile4Writer
from mio5 import MatFile5Reader, MatFile5Writer
__all__ = ['find_mat_file', 'mat_reader_factory', 'loadmat', 'savemat']
@docfiller
def find_mat_file(file_name, appendmat=True):
''' Try to find .mat file on system path
file_name : string
file name for mat file
%(append_arg)s
'''
warnings.warn('Searching for mat files on python system path will be ' +
'removed in future versions of scipy',
FutureWarning, stacklevel=2)
if appendmat and file_name.endswith(".mat"):
file_name = file_name[:-4]
if os.sep in file_name:
full_name = file_name
if appendmat:
full_name = file_name + ".mat"
else:
full_name = None
junk, file_name = os.path.split(file_name)
for path in [os.curdir] + list(sys.path):
test_name = os.path.join(path, file_name)
if appendmat:
test_name += ".mat"
try:
fid = open(test_name,'rb')
fid.close()
full_name = test_name
break
except IOError:
pass
return full_name
@docfiller
def mat_reader_factory(file_name, appendmat=True, **kwargs):
"""Create reader for matlab .mat format files
%(file_arg)s
%(append_arg)s
%(basename_arg)s
%(load_args)s
%(struct_arg)s
"""
if isinstance(file_name, basestring):
try:
byte_stream = open(file_name, 'rb')
except IOError:
full_name = find_mat_file(file_name, appendmat)
if full_name is None:
raise IOError, "%s not found on the path." % file_name
byte_stream = open(full_name, 'rb')
else:
try:
file_name.read(0)
except AttributeError:
raise IOError, 'Reader needs file name or open file-like object'
byte_stream = file_name
# Deal with deprecations
if kwargs.has_key('basename'):
warnings.warn(
'basename argument will be removed in future scipy versions',
DeprecationWarning, stacklevel=2)
del kwargs['basename']
mjv, mnv = get_matfile_version(byte_stream)
if mjv == 0:
return MatFile4Reader(byte_stream, **kwargs)
elif mjv == 1:
return MatFile5Reader(byte_stream, **kwargs)
elif mjv == 2:
raise NotImplementedError('Please use HDF reader for matlab v7.3 files')
else:
raise TypeError('Did not recognize version %s' % mjv)
@docfiller
def loadmat(file_name, mdict=None, appendmat=True, **kwargs):
''' Load Matlab(tm) file
%(file_arg)s
m_dict : dict, optional
dictionary in which to insert matfile variables
%(append_arg)s
%(basename_arg)s
%(load_args)s
%(struct_arg)s
Notes
-----
v4 (Level 1.0), v6 and v7 to 7.2 matfiles are supported.
You will need an HDF5 python library to read matlab 7.3 format mat
files. Because scipy does not supply one, we do not implement the
HDF5 / 7.3 interface here.
'''
MR = mat_reader_factory(file_name, appendmat, **kwargs)
matfile_dict = MR.get_variables()
if mdict is not None:
mdict.update(matfile_dict)
else:
mdict = matfile_dict
return mdict
@docfiller
def savemat(file_name, mdict,
appendmat=True,
format='5',
long_field_names=False,
do_compression=False,
oned_as=None):
"""Save a dictionary of names and arrays into the MATLAB-style .mat file.
This saves the arrayobjects in the given dictionary to a matlab
style .mat file.
file_name : {string, file-like object}
Name of the mat file (do not need .mat extension if
appendmat==True) Can also pass open file-like object
m_dict : dict
dictionary from which to save matfile variables
%(append_arg)s
format : {'5', '4'} string, optional
'5' for matlab 5 (up to matlab 7.2)
'4' for matlab 4 mat files
%(long_fields)s
%(do_compression)s
%(oned_as)s
"""
file_is_string = isinstance(file_name, basestring)
if file_is_string:
if appendmat and file_name[-4:] != ".mat":
file_name = file_name + ".mat"
file_stream = open(file_name, 'wb')
else:
try:
file_name.write('')
except AttributeError:
raise IOError, 'Writer needs file name or writeable '\
'file-like object'
file_stream = file_name
if format == '4':
if long_field_names:
raise ValueError("Long field names are not available for version 4 files")
MW = MatFile4Writer(file_stream, oned_as)
elif format == '5':
MW = MatFile5Writer(file_stream,
do_compression=do_compression,
unicode_strings=True,
long_field_names=long_field_names,
oned_as=oned_as)
else:
raise ValueError("Format should be '4' or '5'")
MW.put_variables(mdict)
if file_is_string:
file_stream.close()
|
