mirror of https://gitlab.com/scemama/qmcchem.git
96 lines
2.2 KiB
Python
Executable File
96 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
|
|
import string
|
|
import os
|
|
|
|
properties = []
|
|
dims = {}
|
|
|
|
files = filter(lambda x: x.startswith("PROPERTIES") and \
|
|
x.endswith("irp.f"), os.listdir(os.getcwd()))
|
|
|
|
files = map(lambda x: 'PROPERTIES/'+x, filter(lambda x: x.endswith("irp.f"), os.listdir(os.getcwd()+'/PROPERTIES')))
|
|
|
|
|
|
#files = filter(lambda x: x.endswith("irp.f"), os.listdir(os.getcwd()))
|
|
|
|
for filename in files:
|
|
lines = []
|
|
check_dims = False
|
|
file = open(filename,'r')
|
|
lines += file.readlines()
|
|
file.close()
|
|
for i,line in enumerate(lines):
|
|
if line.startswith("! PROPERTIES"):
|
|
lines = lines[i:]
|
|
break
|
|
|
|
for line in map(lambda x: x.lower(),lines):
|
|
if line.lstrip().startswith('begin_provider'):
|
|
check_dims = False
|
|
buffer = line
|
|
buffer = buffer.split('[')[1]
|
|
buffer = buffer.split(']')[0]
|
|
buffer = buffer.split(',')
|
|
if (len(buffer) == 2):
|
|
buffer.append("")
|
|
else:
|
|
buffer = [ buffer[0], buffer[1], ','.join(buffer[2:]) ]
|
|
check_dims = True
|
|
buffer = map(string.strip,buffer)
|
|
properties.append(buffer)
|
|
current_prop = buffer[1]
|
|
elif check_dims:
|
|
if 'dimensions :' in line:
|
|
dims[current_prop] = line.split(':')[1].strip()
|
|
|
|
|
|
|
|
def sq(item):
|
|
return [item[0], item[1]+"_2", item[2]]
|
|
properties_with_square = properties + map(sq,properties)
|
|
|
|
for p in [ properties, properties_with_square ]:
|
|
def c(x,y):
|
|
if x[1] > y[1]: return 1
|
|
if x[1] == y[1]: return 0
|
|
if x[1] < y[1]: return -1
|
|
p.sort(c)
|
|
|
|
def namelist():
|
|
buffer = ""
|
|
for p in properties:
|
|
buffer += "calc_"+p[1]+", &\n"
|
|
buffer = buffer[:-4]
|
|
result = " namelist /properties/"+buffer
|
|
return result
|
|
|
|
def touch_all():
|
|
print "TOUCH",
|
|
for p in properties:
|
|
print "calc_"+p[1],
|
|
print ""
|
|
|
|
#file = open('../scripts/properties.py','w')
|
|
#print >>file,'properties = ',properties
|
|
#file.close()
|
|
|
|
def make_dims():
|
|
template = """
|
|
BEGIN_PROVIDER [ integer, size_%(p)s ]
|
|
implicit none
|
|
BEGIN_DOC
|
|
! Size of %(p)s
|
|
END_DOC
|
|
if (calc_%(p)s) then
|
|
size_%(p)s = %(d)s
|
|
else
|
|
size_%(p)s = 1
|
|
endif
|
|
END_PROVIDER
|
|
"""
|
|
for p in dims:
|
|
print template%{'p': p, 'd': dims[p]}
|
|
|
|
|