Skip to content

Commit 7fb572b

Browse files
author
Joel Bernier
committed
Merge branch 'instrument' of https://github.com/joelvbernier/hexrd into instrument
2 parents 5179077 + 74e1842 commit 7fb572b

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

hexrd/imageseries/save.py

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,13 @@ def __init__(self, ims, fname, **kwargs):
5757
self._fname = fname
5858
self._opts = kwargs
5959

60+
# split filename into components
61+
tmp = os.path.split(fname)
62+
self._fname_dir = tmp[0]
63+
tmp = os.path.splitext(tmp[1])
64+
self._fname_base = tmp[0]
65+
self._fname_suff = tmp[1]
66+
6067
pass # end class
6168

6269
class WriteH5(Writer):
@@ -144,8 +151,18 @@ def _process_meta(self):
144151
d = {}
145152
for k, v in self._meta.items():
146153
if isinstance(v, np.ndarray):
147-
d[k] = '++np.array'
148-
d[k + '-array'] = v.tolist()
154+
# Save as a numpy array file
155+
# if file does not exist (careful about directory)
156+
# create new file
157+
158+
cdir = os.path.dirname(self._cache)
159+
b = self._fname_base
160+
fname = os.path.join(cdir, "%s-%s.npy" % (b,k))
161+
if not os.path.exists(fname):
162+
np.save(fname, v)
163+
164+
# add trigger in yml file
165+
d[k] = "! load-numpy-array %s" % fname
149166
else:
150167
d[k] = v
151168

0 commit comments

Comments
 (0)