11 xDim = yDim = zDim = 256
14 def to_vtk(item, xDim, yDim, zDim, nframes, filename):
15 outfile = open(filename,
"w")
16 outfile.write(
"# vtk DataFile Version 3.0\n")
17 outfile.write(
"vtkfile\n")
18 outfile.write(
"ASCII\n")
19 outfile.write(
"DATASET STRUCTURED_POINTS\n")
20 outfile.write(
"DIMENSIONS "+str(xDim)+
" "+str(yDim)+
" "+str(zDim)+
"\n")
21 outfile.write(
"ORIGIN 0 0 0\n")
22 outfile.write(
"SPACING 1 1 1\n")
23 outfile.write(
"POINT_DATA " + str(xDim*yDim*zDim) +
"\n")
24 outfile.write(
"SCALARS scalars float 1\n")
25 outfile.write(
"LOOKUP_TABLE default\n")
29 outfile.write(str(item[i][j][k]) +
" ")
36 if data_dir[0] !=
"/":
37 data_dir =
"../" + data_dir
38 data_real = data_dir +
"/wfc_0_const_%s" % i
39 data_im = data_dir +
"/wfc_0_consti_%s" % i
40 if (pltval ==
"wfc_ev"):
41 data_real = data_dir +
"/wfc_ev_%s" % i
42 data_im = data_dir +
"/wfc_evi_%s" % i
43 elif (pltval ==
"wfc_ramp"):
44 data_real = data_dir +
"/wfc_0_ramp_%s" % i
45 data_im = data_dir +
"/wfc_0_rampi_%s" % i
46 lines_real = np.loadtxt(data_real)
47 lines_im = np.loadtxt(data_im)
48 print(len(lines_real))
49 wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
50 wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
51 wfc = abs(wfc_real + 1j * wfc_im)
53 wfc = np.reshape(wfc,(xDim*yDim*zDim))
56 wfc = np.reshape(wfc,(xDim,yDim,zDim))
59 def wfc_phase(xDim, yDim, zDim, data_dir, pltval, i):
61 if data_dir[0] !=
"/":
62 data_dir =
"../" + data_dir
63 data_real = data_dir +
"/wfc_0_const_%s" % i
64 data_im = data_dir +
"/wfc_0_consti_%s" % i
65 if (pltval ==
"wfc_ev"):
66 data_real = data_dir +
"/wfc_ev_%s" % i
67 data_im = data_dir +
"/wfc_evi_%s" % i
68 elif (pltval ==
"wfc_ramp"):
69 data_real = data_dir +
"/wfc_0_ramp_%s" % i
70 data_im = data_dir +
"/wfc_0_rampi_%s" % i
71 lines_real = np.loadtxt(data_real)
72 lines_im = np.loadtxt(data_im)
73 wfc_real = np.reshape(lines_real, (xDim,yDim, zDim));
74 wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
75 wfc = (wfc_real + 1j * wfc_im)
77 wfc = np.reshape(wfc,(xDim*yDim*zDim))
80 for i
in range(0,len(wfc)):
81 wfc[i] = (wfc[i] - minimum) / (maximum - minimum)
82 wfc = np.reshape(wfc,(xDim,yDim,zDim))
86 if data_dir[0] !=
"/":
87 data_dir =
"../" + data_dir
88 filename = data_dir +
"/wfc_ph_%s" %i
90 data_real = data_dir +
"/wfc_0_const_%s" % i
91 data_im = data_dir +
"/wfc_0_consti_%s" % i
92 if (pltval ==
"wfc_ev"):
93 data_real = data_dir +
"/wfc_ev_%s" % i
94 data_im = data_dir +
"/wfc_evi_%s" % i
95 elif (pltval ==
"wfc_ramp"):
96 data_real = data_dir +
"/wfc_0_ramp_%s" % i
97 data_im = data_dir +
"/wfc_0_rampi_%s" % i
98 lines_real = np.loadtxt(data_real)
99 lines_im = np.loadtxt(data_im)
100 print(len(lines_real))
101 wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
102 wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
103 wfc = (wfc_real + 1j * wfc_im)
105 file = open(filename,
'w')
106 for k
in range(0,xDim):
107 for j
in range(0,yDim):
108 file.write(str(wfc[j][k][zDim/2]) +
'\n')
112 def var(xDim, yDim, zDim, data_dir, pltval):
113 if data_dir[0] !=
"/":
114 data_dir =
"../" + data_dir
115 data = data_dir +
"/" + pltval
116 lines = np.loadtxt(data)
119 for i
in range(0,len(lines)):
120 lines[i] = (lines[i] - minimum) / (maximum - minimum)
121 val = np.reshape(lines, (xDim,yDim,zDim));
125 if data_dir[0] !=
"/":
126 data_dir =
"../" + data_dir
127 data_ax = data_dir +
"/Ax_0" 128 data_ay = data_dir +
"/Ay_0" 130 lines_ax = np.loadtxt(data_ax)
131 lines_ay = np.loadtxt(data_ay)
133 lines_out = [0
for i
in range(len(lines_ax))]
134 for i
in range(0,len(lines_ax)):
135 lines_out[i] = np.sqrt(lines_ay[i]**2 + lines_ax[i]**2)
137 maximum = max(lines_out)
138 minimum = min(lines_out)
139 for i
in range(0,len(lines_out)):
140 lines_out[i] = (lines_out[i] - minimum) / (maximum - minimum)
141 val = np.reshape(lines_out, (xDim,yDim,zDim));
145 proj_var2d(xdim, yDim, zDim, data_dir, pltval,
"var")
147 def proj_var2d(xdim, yDim, zDim, data_dir, pltval, file_string):
148 if data_dir[0] !=
"/":
149 data_dir =
"../" + data_dir
151 filename = data_dir +
"/" + file_string
152 file = open(filename,
"w")
153 data = data_dir +
"/" + pltval
154 lines = np.loadtxt(data)
155 var_data = np.reshape(lines, (xDim, yDim, zDim))
156 for k
in range(0,xDim):
157 for j
in range(0,yDim):
158 file.write(str(var_data[k][j][xDim/2])+
'\n')
161 def proj_var1d(xdim, yDim, zDim, data_dir, pltval, file_string):
162 if data_dir[0] !=
"/":
163 data_dir =
"../" + data_dir
165 filename = data_dir +
"/" + file_string
166 file = open(filename,
"w")
167 data = data_dir +
"/" + pltval
168 lines = np.loadtxt(data)
169 var_data = np.reshape(lines, (xDim, yDim, zDim))
170 for i
in range(0,xDim):
171 file.write(str(var_data[zDim/2][yDim/2][i])+
'\n')
176 def proj_2d(xDim, yDim, zDim, data_dir, pltval, i):
177 if data_dir[0] !=
"/":
178 data_dir =
"../" + data_dir
180 filename = data_dir +
"/Pwfc_%s" %i
182 data_real = data_dir +
"/wfc_0_const_%s" % i
183 data_im = data_dir +
"/wfc_0_consti_%s" % i
184 if (pltval ==
"wfc_ev"):
185 data_real = data_dir +
"/wfc_ev_%s" % i
186 data_im = data_dir +
"/wfc_evi_%s" % i
187 elif (pltval ==
"wfc_ramp"):
188 data_real = data_dir +
"/wfc_0_ramp_%s" % i
189 data_im = data_dir +
"/wfc_0_rampi_%s" % i
190 lines_real = np.loadtxt(data_real)
191 lines_im = np.loadtxt(data_im)
192 print(len(lines_real))
193 wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
194 wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
195 wfc = abs(wfc_real + 1j * wfc_im)
197 file = open(filename,
'w')
198 for k
in range(0,xDim):
199 for j
in range(0,yDim):
200 file.write(str(wfc[k][j][zDim/2]) +
'\n')
203 def proj_k2d(xDim, yDim, zDim, data_dir, pltval, i):
204 if data_dir[0] !=
"/":
205 data_dir =
"../" + data_dir
207 filename = data_dir +
"/wfc_1" 209 data_real = data_dir +
"/wfc_0_const_%s" % i
210 data_im = data_dir +
"/wfc_0_consti_%s" % i
211 lines_real = np.loadtxt(data_real)
212 lines_im = np.loadtxt(data_im)
213 print(len(lines_real))
214 wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
215 wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
216 wfc = np.fft.fftshift(np.fft.fftn(wfc_real + 1j * wfc_im))
217 wfc = abs(wfc) * abs(wfc)
218 file = open(filename,
'w')
219 for k
in range(0,xDim):
220 for j
in range(0,yDim):
221 file.write(str(wfc[j][k][zDim/2]) +
'\n')
225 def to_bvox(item, xDim, yDim, zDim, nframes, filename):
226 header = np.array([xDim, yDim, zDim, nframes])
227 binfile = open(filename,
"wb")
228 header.astype(
'<i4').tofile(binfile)
229 item.astype(
'<f4').tofile(binfile)
233 def wfc_com(xDim, yDim, zDim, data_dir, pltval, i):
234 if data_dir[0] !=
"/":
235 data_dir =
"../" + data_dir
237 filename = data_dir +
"/wfc_%s" %i
239 data_real = data_dir +
"/wfc_0_const_%s" % i
240 data_im = data_dir +
"/wfc_0_consti_%s" % i
241 if (pltval ==
"wfc_ev"):
242 data_real = data_dir +
"/wfc_ev_%s" % i
243 data_im = data_dir +
"/wfc_evi_%s" % i
244 elif (pltval ==
"wfc_ramp"):
245 data_real = data_dir +
"/wfc_0_ramp_%s" % i
246 data_im = data_dir +
"/wfc_0_rampi_%s" % i
247 lines_real = np.loadtxt(data_real)
248 lines_im = np.loadtxt(data_im)
249 print(len(lines_real))
250 wfc_real = np.reshape(lines_real, (xDim,yDim,zDim));
251 wfc_im = np.reshape(lines_im, (xDim,yDim, zDim));
252 wfc = abs(wfc_real + 1j * wfc_im)
259 for i
in range(xDim/2,xDim):
260 for j
in range(0,zDim):
261 comx += wfc[j][i][zDim/2]*i
262 comy += wfc[j][i][zDim/2]*j
263 sum += wfc[j][i][zDim/2]
def proj_phase_2d(xDim, yDim, zDim, data_dir, pltval, i)
def wfc_com(xDim, yDim, zDim, data_dir, pltval, i)
def to_vtk(item, xDim, yDim, zDim, nframes, filename)
def wfc_phase(xDim, yDim, zDim, data_dir, pltval, i)
def var_r2(xDim, yDim, zDim, data_dir)
def wfc_density(xDim, yDim, zDim, data_dir, pltval, i)
def proj_2d(xDim, yDim, zDim, data_dir, pltval, i)
def proj_var1d(xdim, yDim, zDim, data_dir, pltval, file_string)
def proj_k2d(xDim, yDim, zDim, data_dir, pltval, i)
Latin Modern latex print('-dpng','-r300', ['./Comp_CBAR_', int2str(iii),'.png'])
def proj_var2d(xdim, yDim, zDim, data_dir, pltval)
def var(xDim, yDim, zDim, data_dir, pltval)
def to_bvox(item, xDim, yDim, zDim, nframes, filename)