-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcalcvibron.py
executable file
·104 lines (87 loc) · 2.94 KB
/
calcvibron.py
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
#!/Users/student/Anaconda3/bin/python
import matplotlib as mpl
import sys
mpl.use('Agg')
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
import scipy
import scipy.integrate
from scipy.interpolate import griddata
import datetime
import subprocess
from spektras import spektras
cdict3 = {'red': ((0.0, 0.0, 0.0),
(0.25, 0.0, 0.0),
(0.5, 0.8, 1.0),
(0.75, 1.0, 1.0),
(1.0, 0.4, 1.0)),
'green': ((0.0, 0.0, 0.0),
(0.25, 0.0, 0.0),
(0.5, 0.9, 0.9),
(0.75, 0.0, 0.0),
(1.0, 0.0, 0.0)),
'blue': ((0.0, 0.0, 0.4),
(0.25, 1.0, 1.0),
(0.5, 1.0, 0.8),
(0.75, 0.0, 0.0),
(1.0, 0.0, 0.0))
}
plt.register_cmap(name='BlueRed3', data=cdict3)
#s,om,j,T,virpnum2=[float(i) for i in (sys.argv[1:])]
s=float(sys.argv[1])
om=float(sys.argv[2])
j=float(sys.argv[3])
T=float(sys.argv[4])
virpnum2=int(sys.argv[5])
fig, ax=plt.subplots()
vardas='1td_'+str(s)+'_'+str(om)+'_'+str(j)+'_'+str(T)+'K_'+str(virpnum2)
spektras(ax,s,om,j,T,virpnum=virpnum2,nam="BRC/"+vardas)
ax.set_xlim([11000,14000])
fig.savefig('BRCspekt/'+vardas+'.png')
cmd = ['cp outr.txt spekt/'+vardas+'_redfieldrez.txt']
subprocess.Popen(cmd, shell=True).wait()
cmd = ['cp outasr.txt spekt/'+vardas+'_redfieldout.txt']
subprocess.Popen(cmd, shell=True).wait()
# print(datetime.datetime.now())
# cmd = ['../uqcfp/bin/tba.calculator_3rd_levels input_level2d outl2d.txt > outasl2d.txt']
# subprocess.Popen(cmd,shell=True).wait()
# print(datetime.datetime.now())
# file2=open("outl2d.txt")
# l1 = []
# l2=[]
# n=0
# for line in file2:
# if n<1:
# n+=1
# continue
# else:
# l1.append(line.split())
# file2.close()
# l1=np.array(l1).astype(np.float)
# l1=l1.transpose()
# l1[1]=l1[1]*-1
# x=np.arange(11000,14000,30)
# y=np.arange(11000,14000,30)
# X,Y=np.meshgrid(x,y)
# zi = griddata((l1[1], l1[0]),l1[3]*-1,(X, Y),method='linear')
# vardas='2d_'+str(s)+'_'+str(om)+'_'+str(j)+'_'+str(T)+'K_'+str(virpnum2)
# cmd = ['cp input_level2d spekt/'+vardas+'_input.txt']
# subprocess.Popen(cmd,shell=True).wait()
# cmd = ['cp outl2d.txt spekt/'+vardas+'_rezult.txt']
# subprocess.Popen(cmd,shell=True).wait()
# cmd = ['cp outasl2d.txt spekt/'+vardas+'_output.txt']
# subprocess.Popen(cmd,shell=True).wait()
# skk=40
# pj=1
# #ii=int(10000000/860)#12450
# #ii=np.arange(11000,14000,10)[np.diag(zi/(1*zi.max()))>0.65][0]
# plt.figure()
# plt.contourf(X,Y,zi/(1*zi.max()),np.arange(-pj,pj+2*pj/skk,2*pj/skk), cmap='BlueRed3')
# plt.colorbar()
# skk=20
# plt.contour(X,Y,zi/(1*zi.max()),np.arange(-pj,pj+2*pj/skk,2*pj/skk),colors=('k',))
# plt.plot(x,y,'k-',linewidth=1.5)
# plt.axes().set_aspect('equal','box-forced')
# #vardas='2d_'+str(s)+'_'+str(om)+'_'+str(j)+'_'+str(T)+'K_'+str(virpnum2)
# plt.savefig('spekt/'+vardas+'.png')