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
|
import os, re, sys
types=['INTEGER', 'REAL(DP)', 'COMPLEX(DP)'] # WARNING, only double real and complex are tested
typeconv={'INTEGER': 'INT', 'REAL': 'REAL', 'REAL(DP)': 'DBLE', 'COMPLEX': 'CMPLX', 'COMPLEX(DP)': 'DCMPLX'}
datasize='10'
ranks = {'1': '', \
'v': '(datasize)', \
'm': '(datasize,datasize)', \
't': '(datasize,datasize,datasize)', \
'4': '(datasize,datasize,datasize,datasize)', \
'5': '(datasize,datasize,datasize,datasize,datasize)', \
'6': '(datasize,datasize,datasize,datasize,datasize,datasize)'}
nextr={'1': 'v', 'v': 'm', 'm': 't', 't': '4', '4': '5', '5': '', '6': ''}
compare={'INTEGER': 'equal', 'REAL': 'close', 'REAL(DP)': 'close', 'COMPLEX': 'close', 'COMPLEX(DP)': 'close'}
input_file_names = sys.argv[1:]
input_file_names = [os.path.basename(x) for x in input_file_names]
mkfile = open('./autotest.inc', 'w')
mkfile.write('override SRCS += ')
for file in os.listdir("."):
if file.endswith(".tmpl"):
if input_file_names:
if not (os.path.basename(file) in input_file_names):
continue
with open(file,'r') as f:
data = f.read()
# ! Implemented: i1, iv, rv, iv, rm, im, cv
mtch = re.findall(r'Implemented\s*:\s*((\w\w|\*)(,\s*\w\w)*)',data)
if mtch:
impl_interfaces = mtch[0][0].replace(' ','').split(',')
else:
impl_interfaces = ['*']
for t in types:
for k,s in ranks.items():
tmp = os.path.splitext(os.path.basename(file))[0]
tname = t[0].lower()+k
ofile = 'auto' + tmp + '_' + tname + '.f90'
if not ( ('*' in impl_interfaces) or (tname in impl_interfaces) ):
continue
with open(ofile,'w') as fo:
if k == '1':
allf = ''; sumf = ''
else:
allf = 'ALL'; sumf = 'SUM'
fo.write(data.format( vname=(t[0].lower()+k), \
datasize=datasize, \
type=t, size=s, \
all=s.replace('datasize',':'), \
sizep1=ranks.get(nextr[k],'invalid'), \
allp1=ranks.get(nextr[k],'invalid').replace('datasize',':'), \
allf=allf, sumf=sumf, typeconv=typeconv[t],
compare=compare[t]))
mkfile.write(ofile + '\\\n')
|
