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
|
# Authors: The MNE-Python contributors.
# License: BSD-3-Clause
# Copyright the MNE-Python contributors.
import numpy as np
try:
from scipy.io.matlab import MatlabFunction, MatlabOpaque
except ImportError: # scipy < 1.8
from scipy.io.matlab.mio5 import MatlabFunction
from scipy.io.matlab.mio5_params import MatlabOpaque
from scipy.io import loadmat
from ...utils import _import_pymatreader_funcs
def _todict_from_np_struct(data): # taken from pymatreader.utils
data_dict = {}
for cur_field_name in data.dtype.names:
try:
n_items = len(data[cur_field_name])
cur_list = []
for idx in np.arange(n_items):
cur_value = data[cur_field_name].item(idx)
cur_value = _check_for_scipy_mat_struct(cur_value)
cur_list.append(cur_value)
data_dict[cur_field_name] = cur_list
except TypeError:
cur_value = data[cur_field_name].item(0)
cur_value = _check_for_scipy_mat_struct(cur_value)
data_dict[cur_field_name] = cur_value
return data_dict
def _handle_scipy_ndarray(data): # taken from pymatreader.utils
if data.dtype == np.dtype("object") and not isinstance(data, MatlabFunction):
as_list = []
for element in data:
as_list.append(_check_for_scipy_mat_struct(element))
data = as_list
elif isinstance(data.dtype.names, tuple):
data = _todict_from_np_struct(data)
data = _check_for_scipy_mat_struct(data)
if isinstance(data, np.ndarray):
data = np.array(data)
return data
def _check_for_scipy_mat_struct(data): # taken from pymatreader.utils
"""Convert all scipy.io.matlab.mio5_params.mat_struct elements."""
if isinstance(data, dict):
for key in data:
data[key] = _check_for_scipy_mat_struct(data[key])
if isinstance(data, MatlabOpaque):
try:
if data[0][2] == b"string":
return None
except IndexError:
pass
if isinstance(data, np.ndarray):
data = _handle_scipy_ndarray(data)
return data
def _readmat(fname, uint16_codec=None):
try:
read_mat = _import_pymatreader_funcs("EEGLAB I/O")
except RuntimeError: # pymatreader not installed
eeg = loadmat(
fname, squeeze_me=True, mat_dtype=False, uint16_codec=uint16_codec
)
return _check_for_scipy_mat_struct(eeg)
else:
return read_mat(fname, uint16_codec=uint16_codec)
|
