TREX/trexio
1
0
Fork 0
mirror of https://github.com/TREX-CoE/trexio.git synced 2025-01-09 20:48:53 +01:00
Code Issues Projects Releases Wiki Activity

adapt generator for frontend

Browse Source
This commit is contained in:
q-posev 2021-03-10 14:50:51 +01:00
parent 8f295d4d12
commit c0b319a751
1 changed files with 44 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
src/generator.py
View File

@ -64,17 +64,30 @@ for k,v in datasets.items():
datatype = 'int64_t' datatype = 'int64_t'
tmp_dict['dtype'] = datatype tmp_dict['dtype'] = datatype
tmp_dict['dims'] = [dim.replace('.','_') for dim in v[1]] tmp_dict['dims'] = [dim.replace('.','_') for dim in v[1]]
tmp_dict['rank'] = len(v[1])
dim_str = tmp_dict['dims'][0]
if tmp_dict['rank'] > 1:
for i in range(1, tmp_dict['rank']):
dim_toadd = tmp_dict['dims'][i]
dim_str += f', {dim_toadd}'
tmp_dict['dim_list'] = dim_str
datasets_nostr[k] = tmp_dict datasets_nostr[k] = tmp_dict
#print(datasets_nostr) print(datasets_nostr['nucleus_coord'])
#print(numbers) #print(numbers)
temp_path = join(fileDir,'templates_hdf5') templ_path_hdf5 = join(fileDir,'templates_hdf5')
templ_path_front = join(fileDir,'templates_front')
files_exclude = ['prefix_hdf5.c', 'prefix_hdf5.h', 'suffix_hdf5.h', 'templator_hdf5.org'] files_exclude = ['prefix_hdf5.c', 'prefix_hdf5.h', 'suffix_hdf5.h',
'prefix_front.c', 'prefix_front.h', 'suffix_front.h',
'prefix_s_front.h', 'suffix_s_front.h',
'templator_front.org', 'templator_hdf5.org']
files = [f for f in listdir(temp_path) if isfile(join(temp_path, f)) and f not in files_exclude] files_hdf5 = [f for f in listdir(templ_path_hdf5) if isfile(join(templ_path_hdf5, f)) and f not in files_exclude]
files_front = [f for f in listdir(templ_path_front) if isfile(join(templ_path_front, f)) and f not in files_exclude]
files = files_hdf5 + files_front
files_funcs = [f for f in files if 'read_' in f or 'write_' in f or 'rw_' in f ] files_funcs = [f for f in files if 'read_' in f or 'write_' in f or 'rw_' in f ]
files_funcs_dsets = [f for f in files_funcs if 'dset' in f] files_funcs_dsets = [f for f in files_funcs if 'dset' in f]
@ -85,12 +98,16 @@ files_auxil = [f for f in files if not ('read_' in f or 'write_' in f or 'rw_' i
# build files with functions # build files with functions
for fname in files_funcs_nums: for fname in files_funcs_nums:
fname_new = join('populated',f'pop_{fname}') fname_new = join('populated',f'pop_{fname}')
if '_hdf5' in fname:
templ_path = templ_path_hdf5
if '_front' in fname:
templ_path = templ_path_front
for dim in dim_variables.keys(): for dim in dim_variables.keys():
grname = dim.split('_')[0] grname = dim.split('_')[0]
with open(join(templ_path,fname), 'r') as f_in :
with open(join(temp_path,fname), 'r') as f_in : with open(join(templ_path,fname_new), 'a') as f_out :
with open(join(temp_path,fname_new), 'a') as f_out :
for line in f_in : for line in f_in :
if '$' in line: if '$' in line:
templine1 = line.replace('$GROUP_NUM$', dim.upper()) templine1 = line.replace('$GROUP_NUM$', dim.upper())
@ -106,12 +123,15 @@ for fname in files_funcs_nums:
# build files with functions # build files with functions
for fname in files_funcs_dsets: for fname in files_funcs_dsets:
fname_new = join('populated',f'pop_{fname}') fname_new = join('populated',f'pop_{fname}')
if '_hdf5' in fname:
templ_path = templ_path_hdf5
if '_front' in fname:
templ_path = templ_path_front
for dset,params in datasets_nostr.items(): for dset,params in datasets_nostr.items():
grname = dset.split('_')[0] grname = dset.split('_')[0]
with open(join(templ_path,fname), 'r') as f_in :
with open(join(temp_path,fname), 'r') as f_in : with open(join(templ_path,fname_new), 'a') as f_out :
with open(join(temp_path,fname_new), 'a') as f_out :
for line in f_in : for line in f_in :
if '$' in line: if '$' in line:
@ -145,6 +165,12 @@ for fname in files_funcs_dsets:
templine1 = templine2.replace('$group_dset_h5_dtype$', h5_dtype) templine1 = templine2.replace('$group_dset_h5_dtype$', h5_dtype)
templine2 = templine1.replace('$group_dset_h5_dtype$'.upper(), h5_dtype.upper()) templine2 = templine1.replace('$group_dset_h5_dtype$'.upper(), h5_dtype.upper())
templine1 = templine2.replace('$group_dset_rank$', str(params['rank']))
templine2 = templine1
templine1 = templine2.replace('$group_dset_dim_list$', params['dim_list'])
templine2 = templine1
templine1 = templine2.replace('$group$', grname) templine1 = templine2.replace('$group$', grname)
templine2 = templine1.replace('$GROUP$', grname.upper()) templine2 = templine1.replace('$GROUP$', grname.upper())
@ -155,8 +181,13 @@ for fname in files_funcs_dsets:
# build files with $group$ and $group$-based # build files with $group$ and $group$-based
for fname in ['def_hdf5.c', 'basic_hdf5.c', 'struct_hdf5.h'] : for fname in ['def_hdf5.c', 'basic_hdf5.c', 'struct_hdf5.h'] :
fname_new = join('populated',f'pop_{fname}') fname_new = join('populated',f'pop_{fname}')
with open(join(temp_path,fname), 'r') as f_in : if '_hdf5' in fname:
with open(join(temp_path,fname_new), 'a') as f_out : templ_path = templ_path_hdf5
if '_front' in fname:
templ_path = templ_path_front
with open(join(templ_path,fname), 'r') as f_in :
with open(join(templ_path,fname_new), 'a') as f_out :
for line in f_in : for line in f_in :
if '$group_dset$' in line or '$GROUP_DSET$' in line : if '$group_dset$' in line or '$GROUP_DSET$' in line :
for dset in datasets_nostr.keys(): for dset in datasets_nostr.keys():