Skip to content

Commit 79d3269

Browse files
leiosleios
committed
adding smallscale modifications to python scripts.
1 parent a3360f0 commit 79d3269

8 files changed

Lines changed: 208 additions & 7 deletions

File tree

‎py/find_scissors_angle.py‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
from gen_data import *
2+
3+
xDim = 256
4+
yDim = 256
5+
zDim = 256
6+
7+
f = open("angle_out.dat", "w")
8+
#data_dir = "/media/james/ExtraDrive1/GPUE/data_elong"
9+
data_dir = "/media/james/ExtraDrive1/GPUE/data_elong_w0.5n"
10+
11+
for i in range(0,500):
12+
index = i*100
13+
thresh = find_thresh(xDim, yDim, data_dir,
14+
"Pwfc_", index, 1)
15+
print(thresh)
16+
com = find_com(xDim, yDim, data_dir, "Pwfc_",0)
17+
print(com)
18+
angle = find_angle(xDim, yDim, data_dir,
19+
"Pwfc_", index, thresh, com)
20+
print("angle is: ", angle)
21+
f.write(str(angle) + "\n")
22+
23+
f.close()

‎py/gen_data.py‎

Lines changed: 82 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,11 @@
1111
xDim = yDim = zDim = 256
1212

1313
# Function to create plot with vtk
14-
def to_vtk(item, xDim, yDim, zDim, nframes, filename):
15-
outfile = open(filename, "w")
14+
def to_vtk(item, xDim, yDim, zDim, data_dir, filename):
15+
data_loc = filename
16+
if data_dir != "":
17+
data_loc = data_dir + "/" + filename
18+
outfile = open(data_loc, "w")
1619
outfile.write("# vtk DataFile Version 3.0\n")
1720
outfile.write("vtkfile\n")
1821
outfile.write("ASCII\n")
@@ -267,3 +270,80 @@ def wfc_com(xDim, yDim, zDim, data_dir, pltval, i):
267270

268271
return comx, comy
269272

273+
def find_thresh(xDim, yDim, data_dir, pltval, i, thresh_percent):
274+
if data_dir[0] != "/":
275+
data_dir = "../" + data_dir
276+
277+
filename = data_dir + "/" + pltval + "%s" %i
278+
print(i)
279+
280+
lines = np.loadtxt(filename)
281+
'''
282+
sum = 0
283+
for i in range(xDim*yDim):
284+
sum += lines[i]
285+
sum /= (xDim*yDim)
286+
sum *= thresh_percent
287+
return sum
288+
'''
289+
max = 0
290+
for i in range(xDim*yDim):
291+
if lines[i] > max:
292+
max = lines[i]
293+
return max*thresh_percent
294+
295+
def find_com(xDim, yDim, data_dir, pltval, i):
296+
if data_dir[0] != "/":
297+
data_dir = "../" + data_dir
298+
299+
filename = data_dir + "/" + pltval + "%s" %i
300+
print(i)
301+
302+
lines = np.loadtxt(filename)
303+
wfc_2d = np.reshape(lines, (xDim, yDim))
304+
sum = 0
305+
com_x = 0
306+
com_y = 0
307+
for i in range(xDim/2, xDim):
308+
for j in range(yDim):
309+
sum += wfc_2d[j,i]
310+
com_y += j*wfc_2d[j,i]
311+
com_x += i*wfc_2d[j,i]
312+
313+
com_y /= sum
314+
com_x /= sum
315+
316+
return (com_x, com_y)
317+
318+
319+
def find_angle(xDim, yDim, data_dir, pltval, i, thresh, com_tot):
320+
if data_dir[0] != "/":
321+
data_dir = "../" + data_dir
322+
323+
filename = data_dir + "/" + pltval + "%s" %i
324+
print(i)
325+
326+
lines = np.loadtxt(filename)
327+
wfc_2d = np.reshape(lines, (xDim, yDim))
328+
329+
# find CoM and angle for each y element
330+
angle = 0
331+
count = 0
332+
for j in range(yDim):
333+
sum = 0
334+
com = 0
335+
for i in range(xDim/2, xDim):
336+
com += i*wfc_2d[j,i]
337+
sum += wfc_2d[j,i]
338+
339+
if (sum >= thresh):
340+
com /= sum
341+
#print(com)
342+
343+
#angle += math.atan(((com_tot[1]) - j)/((com)-(com_tot[0])))
344+
angle += math.atan2((yDim/2) - j,(com)-(xDim*0.25))
345+
#angle += math.atan2((com)-(xDim*0.25),(yDim/2) - j)
346+
count += 1
347+
348+
angle /= count
349+
return angle

‎py/scissors.py‎

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
from gen_data import *
2+
3+
step = 100
4+
for i in range(0,500):
5+
print(i)
6+
#item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong","wfc_ev", (i*step))
7+
item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5n","wfc_ev", (i*step))
8+
9+
if (i % 10 == 0):
10+
#item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong","Edges_%s" %(i*step))
11+
#to_vtk(item, xDim, yDim, zDim, "/media/james/ExtraDrive1/GPUE/data_elong", "Edges_%s.vtk" % (i*step))
12+
13+
item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5n","Edges_%s" %(i*step))
14+
to_vtk(item, xDim, yDim, zDim, "/media/james/ExtraDrive1/GPUE/data_elong_w0.5n", "Edges_%s.vtk" % (i*step))
15+
16+
#item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5","wfc_ev", (i*10000))
17+
#item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w1","wfc_ev", (i*10000))
18+
#item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w1.5","wfc_ev", (i*10000))
19+
20+

‎py/scissors_0.py‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
from gen_data import *
2+
3+
for i in range(0,5):
4+
print(i)
5+
item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong","wfc_ev", (i*10000))
6+
7+

‎py/scissors_check.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
from gen_data import *
2+
3+
item = proj_2d(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong","wfc", (i*10000))

‎py/script1.py‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
from gen_data import *
2+
3+
f = open("w40_output.dat", "w")
4+
for i in range(0,1):
5+
comx, comy = wfc_com(256, 256, 256, "data",
6+
"wfc", i*1000)
7+
f.write("%s\t%s" % (comx, comy))
8+
f.close()

‎py/vis_scripts.py‎

Lines changed: 63 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,66 @@
141141
#print(comy)
142142
'''
143143

144-
item = var(xDim, yDim, zDim,"data_lin","Edges_0")
145-
to_vtk(item, xDim, yDim, zDim, 1, "Edges.vtk")
146-
#proj_var1d(xDim, yDim, zDim, "data", "Az_0", "Az_1d.dat")
144+
'''
145+
for i in range(0,3):
146+
if (i == 2):
147+
data_dir = "data_HE21lin2_w50"
148+
if (i == 1):
149+
data_dir = "data_lin2_w50_real_time"
150+
if (i == 0):
151+
data_dir = "data_broad2_w50_realtime1"
152+
153+
item = proj_var2d(xDim, yDim, zDim, data_dir, "Bx_0", "BX2D.dat")
154+
item = proj_var2d(xDim, yDim, zDim, data_dir, "By_0", "BY2D.dat")
155+
item = proj_var2d(xDim, yDim, zDim, data_dir, "Bz_0", "BZ2D.dat")
156+
item = proj_var2d(xDim, yDim, zDim, data_dir, "Br_0", "BR2D.dat")
157+
item = proj_var2d(xDim, yDim, zDim, data_dir, "Bphi_0", "BPHI2D.dat")
158+
print(data_dir + " is done")
159+
'''
160+
#item = proj_var1d(xDim, yDim, zDim, "data_check", "Br_0", "BR1D.dat")
161+
#item = proj_2d(xDim, yDim, zDim, "data_check", "wfc", 0)
162+
#item = var(xDim, yDim, zDim,"data_broad2_w50_realtime1","Edges_0")
163+
'''
164+
for i in range(0,10):
165+
print(i)
166+
item = proj_var2d(xDim, yDim, zDim,"data_check","Edges_%s"%(i*100), "Edges_%s.dat"%(i*100))
167+
#to_vtk(item, xDim, yDim, zDim, 1, "test_Edges%s.vtk"%(i*100))
168+
'''
169+
170+
'''
171+
xDim = 128
172+
yDim = 128
173+
zDim = 128
174+
175+
item = var(xDim, yDim, zDim,"data_lin","Az_0")
176+
to_vtk(item, xDim, yDim, zDim, 1, "test_Az.vtk")
177+
178+
item = var(xDim, yDim, zDim,"data_lin","Ax_0")
179+
to_vtk(item, xDim, yDim, zDim, 1, "test_Ax.vtk")
180+
181+
item = var(xDim, yDim, zDim,"data_lin","Ay_0")
182+
to_vtk(item, xDim, yDim, zDim, 1, "test_Ay.vtk")
183+
'''
184+
185+
#item = var_r2(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_lin")
186+
#to_vtk(item, xDim, yDim, zDim, 1, "test_Ar.vtk")
187+
188+
#item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5n",
189+
# "Edges_0")
190+
#to_vtk(item, xDim, yDim, zDim, 1, "test_Edges.vtk")
191+
#item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_lin","Edges_0")
192+
193+
item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5n","Edges_0")
194+
to_vtk(item, xDim, yDim, zDim, 1, "test_Edges.vtk")
195+
item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5n","V_0")
196+
to_vtk(item, xDim, yDim, zDim, 1, "test_V.vtk")
197+
item = var(xDim, yDim, zDim,"/media/james/ExtraDrive1/GPUE/data_elong_w0.5n","Az_0")
198+
to_vtk(item, xDim, yDim, zDim, 1, "test_Az.vtk")
199+
#to_bvox(item, xDim, yDim, zDim, 1, "test_Edges.bvox")
200+
201+
#thresh = find_thresh(xDim, yDim, "/media/james/ExtraDrive1/GPUE/data_elong",
202+
# "Pwfc_", 0, 0.5)
203+
#print(thresh)
204+
#angle = find_angle(xDim, yDim, "/media/james/ExtraDrive1/GPUE/data_elong",
205+
# "Pwfc_", 0, thresh)
206+
#print(angle)

‎py/visualize_3d.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def create_new_material (passedName,passedcolor):
4747
tempMat.specular_shader = 'COOKTORR'
4848
tempMat.specular_intensity = 0.5
4949
tempMat.use_transparency=False
50-
tempMat.alpha = 0.5
50+
tempMat.alpha = 0.01
5151
tempMat.ambient = 0.2
5252
tempMat.emit = 0.9
5353
tempMat.keyframe_insert(data_path="diffuse_color", frame=1, index=-1)
@@ -176,7 +176,7 @@ def add_fiber():
176176
ob.scale[2] = 5
177177
ob.name = "fiber"
178178
me = ob.data
179-
color = (1, 1, 1)
179+
color = (0.7, 0.7, 0.7)
180180
mat = create_new_material(ob.name, color)
181181
me.materials.append(mat)
182182

0 commit comments

Comments
 (0)