def write_bin(shape, file_out, data, bintype=None, header_data=None):
dtype = data.dtype
if dtype.kind != 'i':
if bintype is not None:
if header_data is None:
header_data = BinaryHeader.create(bintype=bintype)
if header_data is not None:
header_data.tofile(file_out)
data.tofile(file_out)
return
def load_bin(shape, file_in, dtype, bintype=None):
nrow, ncol = shape
if bintype is not None:
if dtype not in [np.int]:
header_dtype = BinaryHeader.set_dtype(bintype=bintype)
header_data = np.fromfile(file_in, dtype=header_dtype, count=1)
else:
header_data = None
data = np.fromfile(file_in, dtype=dtype, count=nrow * ncol)
data.resize(nrow, ncol)
return [header_data, data]