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
|
# Copyright 2013-2015. The Regents of the University of California.
# Copyright 2021. Uecker Lab. University Center Göttingen.
# All rights reserved. Use of this source code is governed by
# a BSD-style license which can be found in the LICENSE file.
#
# Authors:
# 2013 Martin Uecker <uecker@eecs.berkeley.edu>
# 2015 Jonathan Tamir <jtamir@eecs.berkeley.edu>
from __future__ import print_function
from __future__ import with_statement
import numpy as np
import mmap
import os
def readcfl(name):
# get dims from .hdr
with open(name + ".hdr", "rt") as h:
h.readline() # skip
l = h.readline()
dims = [int(i) for i in l.split()]
# remove singleton dimensions from the end
n = np.prod(dims)
dims_prod = np.cumprod(dims)
dims = dims[:np.searchsorted(dims_prod, n)+1]
# load data and reshape into dims
with open(name + ".cfl", "rb") as d:
a = np.fromfile(d, dtype=np.complex64, count=n);
return a.reshape(dims, order='F') # column-major
def readmulticfl(name):
# get dims from .hdr
with open(name + ".hdr", "rt") as h:
lines = h.read().splitlines()
index_dim = 1 + lines.index('# Dimensions')
total_size = int(lines[index_dim])
index_sizes = 1 + lines.index('# SizesDimensions')
sizes = [int(i) for i in lines[index_sizes].split()]
index_dims = 1 + lines.index('# MultiDimensions')
with open(name + ".cfl", "rb") as d:
a = np.fromfile(d, dtype=np.complex64, count=total_size)
offset = 0
result = []
for i in range(len(sizes)):
dims = ([int(i) for i in lines[index_dims + i].split()])
n = np.prod(dims)
result.append(a[offset:offset+n].reshape(dims, order='F'))
offset += n
if total_size != offset:
print("Error")
return result
def writecfl(name, array):
with open(name + ".hdr", "wt") as h:
h.write('# Dimensions\n')
for i in (array.shape):
h.write("%d " % i)
h.write('\n')
size = np.prod(array.shape) * np.dtype(np.complex64).itemsize
with open(name + ".cfl", "a+b") as d:
os.ftruncate(d.fileno(), size)
mm = mmap.mmap(d.fileno(), size, flags=mmap.MAP_SHARED, prot=mmap.PROT_WRITE)
if array.dtype != np.complex64:
array = array.astype(np.complex64)
mm.write(np.ascontiguousarray(array.T))
mm.close()
#with mmap.mmap(d.fileno(), size, flags=mmap.MAP_SHARED, prot=mmap.PROT_WRITE) as mm:
# mm.write(array.astype(np.complex64).tobytes(order='F'))
def writemulticfl(name, arrays):
size = 0
dims = []
for array in arrays:
size += array.size
dims.append(array.shape)
with open(name + ".hdr", "wt") as h:
h.write('# Dimensions\n')
h.write("%d\n" % size)
h.write('# SizesDimensions\n')
for dim in dims:
h.write("%d " % len(dim))
h.write('\n')
h.write('# MultiDimensions\n')
for dim in dims:
for i in dim:
h.write("%d " % i)
h.write('\n')
size = size * np.dtype(np.complex64).itemsize
with open(name + ".cfl", "a+b") as d:
os.ftruncate(d.fileno(), size)
mm = mmap.mmap(d.fileno(), size, flags=mmap.MAP_SHARED, prot=mmap.PROT_WRITE)
for array in arrays:
if array.dtype != np.complex64:
array = array.astype(np.complex64)
mm.write(np.ascontiguousarray(array.T))
mm.close()
|
