srDFT_G2/Big_data/Large_core/vtz/produce_data_DFT.py
2019-04-12 12:24:44 +02:00

83 lines
2.5 KiB
Python
Executable File

#!/usr/bin/env python2
import sys
basis = sys.argv[1]
filecc = "data_CC_"+basis+"_g09"
fileHF = "data_HF_"+basis+"_g09"
fileHF5 = "data_HF_v5z_g09"
fileLDA = "data_LDA_"+basis
filePBE = "data_PBE_"+basis
cc = []
LDA_ful = []
LDA_val = []
PBE_ful = []
PBE_val = []
HF = []
HF5 = []
system = []
with open(filecc, "r") as fp:
for line in fp:
a=line.split()
system.append(a[0])
b = a[1].replace("D","E")
cc.append(float(b))
with open(fileHF, "r") as fp:
for line in fp:
a=line.split()
b = a[1].replace("D","E")
HF.append(float(b))
with open(fileHF5, "r") as fp:
for line in fp:
a=line.split()
b = a[1].replace("D","E")
HF5.append(float(b))
with open(fileLDA, "r") as fp:
for line in fp:
a=line.split()
LDA_ful.append(float(a[1]))
LDA_val.append(float(a[2]))
with open(filePBE, "r") as fp:
for line in fp:
a=line.split()
PBE_ful.append(float(a[1]))
PBE_val.append(float(a[2]))
file_output_cc = open("data_CC_"+basis,"w+")
file_output_ecor = open("data_ec_"+basis,"w+")
file_output_LDA_ful = open("data_CC+LDA_ful_"+basis,"w+")
file_output_PBE_ful = open("data_CC+PBE_ful_"+basis,"w+")
file_output_LDA_val = open("data_CC+LDA_val_"+basis,"w+")
file_output_PBE_val = open("data_CC+PBE_val_"+basis,"w+")
file_output_ecor_LDA_val = open("data_ec+LDA_val_"+basis,"w+")
file_output_ecor_PBE_val = open("data_ec+PBE_val_"+basis,"w+")
file_output_ecor_LDA_val_HF_5z = open("data_HF_5z+ec+LDA_val_"+basis,"w+")
file_output_ecor_PBE_val_HF_5z = open("data_HF_5z+ec+PBE_val_"+basis,"w+")
file_output_ecor_HF_5z = open("data_HF_5z+ec_"+basis,"w+")
icount = 0
for e in cc:
file_output_cc.write(system[icount]+' '+str(e) +'\n')
ec = e - HF[icount]
file_output_ecor.write(system[icount]+' '+str(ec) +'\n')
file_output_ecor_LDA_val.write(system[icount]+' '+str(ec + LDA_val[icount]) +'\n')
file_output_ecor_PBE_val.write(system[icount]+' '+str(ec + PBE_val[icount]) +'\n')
file_output_ecor_HF_5z.write(system[icount]+' '+str(HF5[icount] + ec) +'\n')
file_output_ecor_LDA_val_HF_5z.write(system[icount]+' '+str(HF5[icount] + ec + LDA_val[icount]) +'\n')
file_output_ecor_PBE_val_HF_5z.write(system[icount]+' '+str(HF5[icount] + ec + PBE_val[icount]) +'\n')
file_output_LDA_ful.write(system[icount]+' '+str(e + LDA_ful[icount]) +'\n')
file_output_PBE_ful.write(system[icount]+' '+str(e + PBE_ful[icount]) +'\n')
file_output_LDA_val.write(system[icount]+' '+str(e + LDA_val[icount]) +'\n')
file_output_PBE_val.write(system[icount]+' '+str(e + PBE_val[icount]) +'\n')
icount += 1