def save_edf_image_int(data, filename):
"""
saving an image to .edf format (int)
:param data: input image
:param filename: filename
:return: None
"""
data_to_store = data.astype(np.int)
edf.EdfImage(data=data_to_store).write(filename)
def setUp(self):
"""Generate test files and headers"""
edfpath = 'test.edf'
self.header = [('start', {
'time': 1534432676.1674924,
'uid': '4becc62f-8f0b-477f-a772-f1c66d6ca70b'
}),
('descriptor', {
'data_keys': {
'image': {
'source': 'file',
'dtype': 'array',
'shape': (128, 128)
}
},
'time': 1534432676.1675735,
'uid': 'eba6ce45-1f7c-457a-8c63-9d5db3374270',
'start': '4becc62f-8f0b-477f-a772-f1c66d6ca70b'
}),
('event', {
'data': {
'image': np.random.random((128, 128)).tolist()
},
'timestamps': {
'image': 1534432660.8167932
},
'time': 1534432660.8167932,
'uid': '4bff5f16-8183-4bfe-8984-7558d263a259',
'descriptor':
'eba6ce45-1f7c-457a-8c63-9d5db3374270',
'EDF_DataBlockID': '0.Image.Psd',
'EDF_BinarySize': '131072',
'EDF_HeaderSize': '512',
'ByteOrder': 'LowByteFirst',
'DataType': 'DoubleValue',
'Dim_1': '128',
'Dim_2': '128',
'Image': '0',
'HeaderID': 'EH:000000:000000:000000',
'Size': '131072',
'test': '1'
}),
('stop', {
'exit_status': 'success',
'time': 1534432676.1710825,
'uid': '97cf57c1-eb4f-4263-af67-799ddeb10453',
'start': '4becc62f-8f0b-477f-a772-f1c66d6ca70b'
})]
edfimage.EdfImage(data=np.random.random((128, 128)),
header={
'test': 1
}).write(edfpath)
self.paths = [edfpath]
def saveEdf(data, filename):
print(filename)
dataToStore = data.astype(np.float32)
edf.EdfImage(data=dataToStore).write(filename)