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
|
"""@package docstring
File IO to load/decode JData-based files to Python data or encode/save Python data to JData files
Copyright (c) 2019 Qianqian Fang <q.fang at neu.edu>
"""
__all__ = ['load','save','show','loadt','savet','loadb','saveb','jext']
##====================================================================================
## dependent libraries
##====================================================================================
import json
import os
import jdata as jd
from collections import OrderedDict
##====================================================================================
## global variables
##====================================================================================
jext={'t':['.json','.jdt','.jdat','.jnii','.jmsh','.jnirs'], 'b':['.ubj','.bjd','.jdb','.jbat','.bnii','.bmsh','.jamm','.bnirs']};
##====================================================================================
## Loading and saving data based on file extensions
##====================================================================================
def load(fname, opt={}, **kwargs):
"""@brief Loading a JData file (binary or text) according to the file extension
@param[in] fname: a JData file name (accept .json,.jdat,.jbat,.jnii,.bnii,.jmsh,.bmsh)
@param[in] opt: options, if opt['decode']=True or 1 (default), call jdata.decode() after loading
"""
spl = os.path.splitext(fname)
ext = spl[1].lower()
if(ext in jext['t']):
return loadt(fname, opt, **kwargs);
elif(ext in jext['b']):
return loadb(fname, opt, **kwargs);
else:
raise Exception('JData', 'file extension is not recognized, accept (.json,.jdat,.jbat,.jnii,.bnii,.jmsh,.bmsh)')
def save(data, fname, opt={}, **kwargs):
"""@brief Saving Python data to file (binary or text) according to the file extension
@param[in] data: data to be saved
@param[in] fname: a JData file name
@param[in] opt: options, if opt['encode']=True or 1 (default), call jdata.encode() before saving
"""
spl = os.path.splitext(fname)
ext = spl[1].lower()
if(ext in jext['t']):
savet(data, fname, opt, **kwargs);
elif(ext in jext['b']):
saveb(data, fname, opt, **kwargs);
else:
raise Exception('JData', 'file extension is not recognized, accept (.json,.jdat,.jbat,.jnii,.bnii,.jmsh,.bmsh)')
##====================================================================================
## Loading and saving text-based JData (i.e. JSON) files
##====================================================================================
def loadt(fname, opt={}, **kwargs):
"""@brief Loading a text-based (JSON) JData file and decode it to native Python data
@param[in] fname: a text JData (JSON based) file name
@param[in] opt: options, if opt['decode']=True or 1 (default), call jdata.decode() after loading
"""
kwargs.setdefault('strict',False);
kwargs.setdefault('object_pairs_hook',OrderedDict);
opt.setdefault('decode',True);
opt['base64']=True;
with open(fname, "r") as fid:
data=json.load(fid, **kwargs);
if(opt['decode']):
data=jd.decode(data,opt);
return data
def savet(data, fname, opt={}, **kwargs):
"""@brief Saving a Python data structure to a text-based JData (JSON) file
@param[in] data: data to be saved
@param[in] fname: a text JData (JSON based) file name
@param[in] opt: options, if opt['encode']=True or 1 (default), call jdata.encode() before saving
"""
kwargs.setdefault('default',jd.jsonfilter);
opt.setdefault('encode',True);
if(opt['encode']):
data=jd.encode(data,opt);
with open(fname, "w") as fid:
json.dump(data, fid, **kwargs);
def show(data, opt={}, **kwargs):
"""@brief Printing a python data as JSON string or return the JSON string (opt['string']=True)
@param[in] data: data to be saved
@param[in] opt: options, if opt['encode']=True or 1 (default), call jdata.encode() before printing
"""
kwargs.setdefault('default',jd.jsonfilter);
opt.setdefault('string',False);
opt.setdefault('encode',True);
if(opt['encode']):
data=jd.encode(data,opt);
str=json.dumps(data, **kwargs);
if(opt['string']):
return str;
else:
print(str);
##====================================================================================
## Loading and saving binary JData (i.e. UBJSON) files
##====================================================================================
def loadb(fname, opt={}, **kwargs):
"""@brief Loading a binary (BJData/UBJSON) JData file and decode it to native Python data
@param[in] fname: a binary (BJData/UBJSON based) JData file name
@param[in] opt: options, if opt['decode']=True or 1 (default), call jdata.decode() before saving
"""
opt.setdefault('decode',True)
opt['base64']=False;
try:
import bjdata
except ImportError:
raise ImportError('To read/write binary JData files, you must install the bjdata module by "pip install bjdata"')
else:
with open(fname, "rb") as fid:
data=bjdata.load(fid,**kwargs);
if(opt['decode']):
data=jd.decode(data,opt);
return data
def saveb(data, fname, opt={}, **kwargs):
"""@brief Saving a Python data structure to a binary JData (BJData/UBJSON) file
@param[in] data: data to be saved
@param[in] fname: a binary (BJData/UBJSON based) JData file name
@param[in] opt: options, if opt['encode']=True or 1 (default), call jdata.encode() before saving
"""
opt.setdefault('encode',True)
try:
import bjdata
except ImportError:
raise ImportError('To read/write binary JData files, you must install the bjdata module by "pip install bjdata"')
else:
if(opt['encode']):
data=jd.encode(data,opt);
with open(fname, "wb") as fid:
bjdata.dump(data, fid,**kwargs);
##====================================================================================
## helper functions
##====================================================================================
|
