mirror of
https://github.com/TREX-CoE/trexio.git
synced 2024-12-23 04:43:57 +01:00
generator for text backend before test
This commit is contained in:
parent
0da57b6b6c
commit
aa67877948
187
src/generator.py
187
src/generator.py
@ -1,6 +1,6 @@
|
|||||||
import json
|
import json
|
||||||
|
|
||||||
from os import listdir
|
from os import listdir, scandir, remove
|
||||||
from os.path import isfile, join, dirname, abspath
|
from os.path import isfile, join, dirname, abspath
|
||||||
|
|
||||||
fileDir = dirname(abspath(__file__))
|
fileDir = dirname(abspath(__file__))
|
||||||
@ -15,7 +15,8 @@ del config0['metadata']
|
|||||||
|
|
||||||
config = {}
|
config = {}
|
||||||
for k,v in config0.items():
|
for k,v in config0.items():
|
||||||
if k == 'nucleus' or k == 'ecp':
|
#if k == 'nucleus' or k == 'mo':
|
||||||
|
if k == 'nucleus': #or k == 'electron':
|
||||||
config[k] = v
|
config[k] = v
|
||||||
|
|
||||||
#print(config)
|
#print(config)
|
||||||
@ -74,52 +75,152 @@ for k,v in datasets.items():
|
|||||||
datasets_nostr[k] = tmp_dict
|
datasets_nostr[k] = tmp_dict
|
||||||
|
|
||||||
print(datasets_nostr['nucleus_coord'])
|
print(datasets_nostr['nucleus_coord'])
|
||||||
#print(numbers)
|
|
||||||
|
|
||||||
|
#put also dimensioning variables in numbers
|
||||||
|
numbers.update(dim_variables)
|
||||||
|
print(numbers)
|
||||||
|
|
||||||
|
templ_path_text = join(fileDir,'templates_text')
|
||||||
templ_path_hdf5 = join(fileDir,'templates_hdf5')
|
templ_path_hdf5 = join(fileDir,'templates_hdf5')
|
||||||
templ_path_front = join(fileDir,'templates_front')
|
templ_path_front = join(fileDir,'templates_front')
|
||||||
|
|
||||||
|
#clean the populated/ directory
|
||||||
|
for popdir in [templ_path_front, templ_path_front, templ_path_text]:
|
||||||
|
cleandir = join(popdir, 'populated')
|
||||||
|
for f in scandir(cleandir):
|
||||||
|
remove(f.path)
|
||||||
|
|
||||||
files_exclude = ['prefix_hdf5.c', 'prefix_hdf5.h', 'suffix_hdf5.h',
|
files_exclude = ['prefix_hdf5.c', 'prefix_hdf5.h', 'suffix_hdf5.h',
|
||||||
|
'prefix_text.c', 'prefix_text.h', 'suffix_text.h',
|
||||||
'prefix_front.c', 'prefix_front.h', 'suffix_front.h',
|
'prefix_front.c', 'prefix_front.h', 'suffix_front.h',
|
||||||
'prefix_s_front.h', 'suffix_s_front.h',
|
'prefix_s_front.h', 'suffix_s_front.h',
|
||||||
'templator_front.org', 'templator_hdf5.org']
|
'templator_front.org', 'templator_hdf5.org', 'templator_text.org']
|
||||||
|
|
||||||
|
files_text = [f for f in listdir(templ_path_text) if isfile(join(templ_path_text, 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_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_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 = files_text + 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 'flush_' in f or 'free_' 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]
|
||||||
files_funcs_nums = [f for f in files_funcs if 'num' in f]
|
files_funcs_nums = [f for f in files_funcs if 'num' in f]
|
||||||
|
files_funcs_groups = [f for f in files_funcs if 'group' in f]
|
||||||
|
|
||||||
|
files_funcs_groups.append('struct_text_group_dset.h')
|
||||||
|
print(files_funcs_groups)
|
||||||
|
|
||||||
files_auxil = [f for f in files if not ('read_' in f or 'write_' in f or 'rw_' in f)]
|
files_auxil = [f for f in files if not ('read_' in f or 'write_' in f or 'rw_' in f)]
|
||||||
|
|
||||||
# build files with functions
|
# build files with functions for text groups
|
||||||
for fname in files_funcs_nums:
|
|
||||||
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():
|
|
||||||
|
|
||||||
grname = dim.split('_')[0]
|
for fname in files_funcs_groups:
|
||||||
|
fname_new = join('populated',f'pop_{fname}')
|
||||||
|
if '_text' in fname:
|
||||||
|
templ_path = templ_path_text
|
||||||
|
|
||||||
|
groups_done = []
|
||||||
|
for group in config.keys():
|
||||||
|
|
||||||
|
grname = group.split('_')[0]
|
||||||
|
if grname in groups_done:
|
||||||
|
continue
|
||||||
|
else:
|
||||||
|
groups_done.append(grname)
|
||||||
|
|
||||||
|
subloop = False
|
||||||
|
do_dset = False
|
||||||
|
do_num = False
|
||||||
|
loop_body = ''
|
||||||
with open(join(templ_path,fname), 'r') as f_in :
|
with open(join(templ_path,fname), 'r') as f_in :
|
||||||
with open(join(templ_path,fname_new), 'a') as f_out :
|
with open(join(templ_path,fname_new), 'a') as f_out :
|
||||||
for line in f_in :
|
for line in f_in :
|
||||||
if '$' in line:
|
|
||||||
templine1 = line.replace('$GROUP_NUM$', dim.upper())
|
if 'END REPEAT' in line:
|
||||||
templine2 = templine1.replace('$group_num$', dim)
|
|
||||||
|
|
||||||
templine1 = templine2.replace('$group$', grname)
|
if do_dset:
|
||||||
templine2 = templine1.replace('$GROUP$', grname.upper())
|
for dset,params in datasets_nostr.items():
|
||||||
|
dset_grname = dset.split('_')[0]
|
||||||
|
if dset_grname != grname:
|
||||||
|
continue
|
||||||
|
|
||||||
|
templine1 = loop_body.replace('$group_dset$', dset)
|
||||||
|
templine2 = templine1.replace('$group$', grname)
|
||||||
|
f_out.write(templine2)
|
||||||
|
elif do_num:
|
||||||
|
#for dim in dim_variables.keys():
|
||||||
|
for dim in numbers.keys():
|
||||||
|
|
||||||
|
num_grname = dim.split('_')[0]
|
||||||
|
if num_grname != grname:
|
||||||
|
continue
|
||||||
|
|
||||||
|
templine1 = loop_body.replace('$group_num$', dim)
|
||||||
|
templine2 = templine1.replace('$group$', grname)
|
||||||
|
f_out.write(templine2)
|
||||||
|
else:
|
||||||
|
print('fishy')
|
||||||
|
|
||||||
|
#print(loop_body)
|
||||||
|
loop_body = ''
|
||||||
|
subloop = False
|
||||||
|
do_dset = False
|
||||||
|
do_num = False
|
||||||
|
continue
|
||||||
|
|
||||||
|
if subloop:
|
||||||
|
loop_body += line
|
||||||
|
|
||||||
|
if 'START REPEAT' in line:
|
||||||
|
if 'GROUP_DSET' in line:
|
||||||
|
do_dset = True
|
||||||
|
if 'GROUP_NUM' in line:
|
||||||
|
do_num = True
|
||||||
|
subloop = True
|
||||||
|
|
||||||
|
if '$group_dset' in line and not subloop:
|
||||||
|
for dset,params in datasets_nostr.items():
|
||||||
|
|
||||||
f_out.write(templine2)
|
dset_grname = dset.split('_')[0]
|
||||||
else:
|
if dset_grname != grname:
|
||||||
|
continue
|
||||||
|
|
||||||
|
templine1 = line.replace('$group_dset$', dset)
|
||||||
|
templine2 = templine1
|
||||||
|
|
||||||
|
templine1 = templine2.replace('$group_dset_dtype$', params['dtype'])
|
||||||
|
templine2 = templine1
|
||||||
|
|
||||||
|
templine1 = templine2.replace('$group$', grname)
|
||||||
|
templine2 = templine1.replace('$GROUP$', grname.upper())
|
||||||
|
|
||||||
|
f_out.write(templine2)
|
||||||
|
elif '$group_num' in line and not subloop:
|
||||||
|
#for dim in dim_variables.keys():
|
||||||
|
for dim in numbers.keys():
|
||||||
|
num_grname = dim.split('_')[0]
|
||||||
|
if num_grname != grname:
|
||||||
|
continue
|
||||||
|
|
||||||
|
templine1 = line.replace('$GROUP_NUM$', dim.upper())
|
||||||
|
templine2 = templine1.replace('$group_num$', dim)
|
||||||
|
|
||||||
|
templine1 = templine2.replace('$group$', grname)
|
||||||
|
templine2 = templine1.replace('$GROUP$', grname.upper())
|
||||||
|
|
||||||
|
f_out.write(templine2)
|
||||||
|
|
||||||
|
elif '$group$' in line and not subloop:
|
||||||
|
|
||||||
|
templine1 = line.replace('$group$', grname)
|
||||||
|
templine2 = templine1.replace('$GROUP$', grname.upper())
|
||||||
|
f_out.write(templine2)
|
||||||
|
|
||||||
|
elif not subloop:
|
||||||
f_out.write(line)
|
f_out.write(line)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
# 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}')
|
||||||
@ -127,6 +228,9 @@ for fname in files_funcs_dsets:
|
|||||||
templ_path = templ_path_hdf5
|
templ_path = templ_path_hdf5
|
||||||
if '_front' in fname:
|
if '_front' in fname:
|
||||||
templ_path = templ_path_front
|
templ_path = templ_path_front
|
||||||
|
if '_text' in fname:
|
||||||
|
templ_path = templ_path_text
|
||||||
|
|
||||||
for dset,params in datasets_nostr.items():
|
for dset,params in datasets_nostr.items():
|
||||||
|
|
||||||
grname = dset.split('_')[0]
|
grname = dset.split('_')[0]
|
||||||
@ -178,13 +282,43 @@ for fname in files_funcs_dsets:
|
|||||||
else:
|
else:
|
||||||
f_out.write(line)
|
f_out.write(line)
|
||||||
|
|
||||||
# build files with $group$ and $group$-based
|
# build files with functions
|
||||||
for fname in ['def_hdf5.c', 'basic_hdf5.c', 'struct_hdf5.h'] :
|
for fname in files_funcs_nums:
|
||||||
fname_new = join('populated',f'pop_{fname}')
|
fname_new = join('populated',f'pop_{fname}')
|
||||||
if '_hdf5' in fname:
|
if '_hdf5' in fname:
|
||||||
templ_path = templ_path_hdf5
|
templ_path = templ_path_hdf5
|
||||||
if '_front' in fname:
|
if '_front' in fname:
|
||||||
templ_path = templ_path_front
|
templ_path = templ_path_front
|
||||||
|
if '_text' in fname:
|
||||||
|
templ_path = templ_path_text
|
||||||
|
|
||||||
|
#for dim in dim_variables.keys():
|
||||||
|
for dim in numbers.keys():
|
||||||
|
grname = dim.split('_')[0]
|
||||||
|
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 :
|
||||||
|
if '$' in line:
|
||||||
|
templine1 = line.replace('$GROUP_NUM$', dim.upper())
|
||||||
|
templine2 = templine1.replace('$group_num$', dim)
|
||||||
|
|
||||||
|
templine1 = templine2.replace('$group$', grname)
|
||||||
|
templine2 = templine1.replace('$GROUP$', grname.upper())
|
||||||
|
|
||||||
|
f_out.write(templine2)
|
||||||
|
else:
|
||||||
|
f_out.write(line)
|
||||||
|
|
||||||
|
# build files with $group$ and $group$-based
|
||||||
|
for fname in ['def_hdf5.c', 'basic_hdf5.c', 'basic_text_group.c',
|
||||||
|
'struct_hdf5.h', 'struct_text_group.h'] :
|
||||||
|
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
|
||||||
|
if '_text' in fname:
|
||||||
|
templ_path = templ_path_text
|
||||||
|
|
||||||
with open(join(templ_path,fname), 'r') as f_in :
|
with open(join(templ_path,fname), 'r') as f_in :
|
||||||
with open(join(templ_path,fname_new), 'a') as f_out :
|
with open(join(templ_path,fname_new), 'a') as f_out :
|
||||||
@ -195,7 +329,8 @@ for fname in ['def_hdf5.c', 'basic_hdf5.c', 'struct_hdf5.h'] :
|
|||||||
templine2 = templine1.replace('$group_dset$', dset)
|
templine2 = templine1.replace('$group_dset$', dset)
|
||||||
f_out.write(templine2)
|
f_out.write(templine2)
|
||||||
elif '$group_num$' in line or '$GROUP_NUM$' in line :
|
elif '$group_num$' in line or '$GROUP_NUM$' in line :
|
||||||
for num in dim_variables.keys():
|
#for num in dim_variables.keys():
|
||||||
|
for num in numbers.keys():
|
||||||
templine1 = line.replace('$GROUP_NUM$', num.upper())
|
templine1 = line.replace('$GROUP_NUM$', num.upper())
|
||||||
templine2 = templine1.replace('$group_num$', num)
|
templine2 = templine1.replace('$group_num$', num)
|
||||||
f_out.write(templine2)
|
f_out.write(templine2)
|
||||||
|
31
src/templates_text/build.sh
Normal file
31
src/templates_text/build.sh
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
#!/bin/sh
|
||||||
|
|
||||||
|
cat prefix_text.c > trexio_text.c
|
||||||
|
cat prefix_text.h > trexio_text.h
|
||||||
|
|
||||||
|
cat basic_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_basic_text_group.c >> trexio_text.c
|
||||||
|
|
||||||
|
cat populated/pop_struct_text_group_dset.h >> trexio_text.h
|
||||||
|
cat populated/pop_struct_text_group.h >> trexio_text.h
|
||||||
|
cat basic_text.h >> trexio_text.h
|
||||||
|
|
||||||
|
cat populated/pop_free_group_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_read_group_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_flush_group_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_free_group_text.h >> trexio_text.h
|
||||||
|
cat populated/pop_read_group_text.h >> trexio_text.h
|
||||||
|
cat populated/pop_flush_group_text.h >> trexio_text.h
|
||||||
|
|
||||||
|
cat populated/pop_read_dset_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_read_num_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_write_dset_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_write_num_text.c >> trexio_text.c
|
||||||
|
cat populated/pop_rw_num_text.h >> trexio_text.h
|
||||||
|
cat populated/pop_rw_dset_text.h >> trexio_text.h
|
||||||
|
|
||||||
|
cat rdm_text.c >> trexio_text.c
|
||||||
|
cat rdm_text.h >> trexio_text.h
|
||||||
|
|
||||||
|
cat suffix_text.h >> trexio_text.h
|
||||||
|
|
@ -1,6 +1,7 @@
|
|||||||
#+Title: TEXT back end of the TREX Input/Ouput library (TREXIO)
|
|
||||||
|
|
||||||
* File prefixes :noxport:
|
#+Title: Templator for TEXT backend
|
||||||
|
|
||||||
|
* Constant file prefixes (not used by generator) :noxport:
|
||||||
|
|
||||||
#+begin_src emacs-lisp
|
#+begin_src emacs-lisp
|
||||||
(setq-local org-babel-default-header-args:c '((:comments . "both")))
|
(setq-local org-babel-default-header-args:c '((:comments . "both")))
|
||||||
@ -26,7 +27,7 @@
|
|||||||
|
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h :noweb yes
|
#+begin_src c :tangle prefix_text.h :noweb yes
|
||||||
<<header>>
|
<<header>>
|
||||||
#ifndef _TREXIO_TEXT_H
|
#ifndef _TREXIO_TEXT_H
|
||||||
#define _TREXIO_TEXT_H
|
#define _TREXIO_TEXT_H
|
||||||
@ -46,17 +47,20 @@
|
|||||||
|
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c :noweb yes
|
#+begin_src c :tangle prefix_text.c :noweb yes
|
||||||
/* This file was generated from the trexio.org org-mode file.
|
/* This file was generated from the trexio.org org-mode file.
|
||||||
To generate it, open trexio.org in Emacs and execute
|
To generate it, open trexio.org in Emacs and execute
|
||||||
M-x org-babel-tangle
|
M-x org-babel-tangle
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#include "trexio_text.h"
|
#include "trexio_text.h"
|
||||||
|
|
||||||
|
#define DEBUG printf("%s : line %d\n", __FILE__, __LINE__);
|
||||||
|
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
|
|
||||||
* TEXT Back end
|
* TEXT Back end
|
||||||
|
|
||||||
The "file" produced by the text back end is a directory with one
|
The "file" produced by the text back end is a directory with one
|
||||||
@ -71,26 +75,24 @@
|
|||||||
|
|
||||||
The file is written when closed, or when the flush function is called.
|
The file is written when closed, or when the flush function is called.
|
||||||
|
|
||||||
*** Structs for blocks
|
*** Template for group-related structures in text back end
|
||||||
#+begin_src c :tangle trexio_text.h
|
|
||||||
typedef struct nucleus_s {
|
|
||||||
FILE* file;
|
|
||||||
double* coord;
|
|
||||||
double* charge;
|
|
||||||
uint64_t num;
|
|
||||||
uint64_t dims_charge[16];
|
|
||||||
uint64_t dims_coord[16];
|
|
||||||
uint32_t rank_charge;
|
|
||||||
uint32_t rank_coord;
|
|
||||||
int to_flush;
|
|
||||||
} nucleus_t;
|
|
||||||
|
|
||||||
typedef struct electron_s {
|
#+begin_src c :tangle struct_text_group_dset.h
|
||||||
|
|
||||||
|
typedef struct $group$_s {
|
||||||
FILE* file;
|
FILE* file;
|
||||||
uint64_t alpha_num;
|
$group_dset_dtype$* $group_dset$;
|
||||||
uint64_t beta_num;
|
uint64_t $group_num$;
|
||||||
int to_flush;
|
uint64_t dims_$group_dset$[16];
|
||||||
} electron_t;
|
uint32_t rank_$group_dset$;
|
||||||
|
int to_flush;
|
||||||
|
} $group$_t;
|
||||||
|
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
*** Template for general structure in text back end
|
||||||
|
|
||||||
|
#+begin_src c :tangle struct_text_group.h
|
||||||
|
|
||||||
typedef struct rdm_s {
|
typedef struct rdm_s {
|
||||||
FILE* file;
|
FILE* file;
|
||||||
@ -99,28 +101,23 @@ typedef struct rdm_s {
|
|||||||
char* two_e_file_name;
|
char* two_e_file_name;
|
||||||
int to_flush;
|
int to_flush;
|
||||||
} rdm_t;
|
} rdm_t;
|
||||||
#+end_src
|
|
||||||
*** Structs for the text back end
|
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
|
||||||
typedef struct trexio_text_s {
|
typedef struct trexio_text_s {
|
||||||
trexio_t parent ;
|
trexio_t parent ;
|
||||||
int lock_file;
|
int lock_file;
|
||||||
|
$group$_t* $group$;
|
||||||
nucleus_t* nucleus;
|
|
||||||
electron_t* electron;
|
|
||||||
rdm_t* rdm;
|
rdm_t* rdm;
|
||||||
} trexio_text_t;
|
} trexio_text_t;
|
||||||
|
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
*** Init/deinit functions
|
*** Init/deinit functions (constant part)
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle basic_text.h
|
||||||
trexio_exit_code trexio_text_init(trexio_t* file);
|
trexio_exit_code trexio_text_init(trexio_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle basic_text.c
|
||||||
trexio_exit_code trexio_text_init(trexio_t* file) {
|
trexio_exit_code trexio_text_init(trexio_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
@ -165,11 +162,11 @@ trexio_exit_code trexio_text_init(trexio_t* file) {
|
|||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle basic_text.h
|
||||||
trexio_exit_code trexio_text_lock(trexio_t* file);
|
trexio_exit_code trexio_text_lock(trexio_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle basic_text.c
|
||||||
trexio_exit_code trexio_text_lock(trexio_t* file) {
|
trexio_exit_code trexio_text_lock(trexio_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
@ -191,30 +188,15 @@ trexio_exit_code trexio_text_lock(trexio_t* file) {
|
|||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle basic_text.h
|
||||||
trexio_exit_code trexio_text_finalize(trexio_t* file);
|
trexio_exit_code trexio_text_finalize(trexio_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
|
||||||
trexio_exit_code trexio_text_finalize(trexio_t* file) {
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
|
||||||
|
|
||||||
trexio_exit_code rc;
|
#+begin_src c :tangle basic_text.h
|
||||||
rc = trexio_text_free_nucleus( (trexio_text_t*) file);
|
|
||||||
assert (rc == TREXIO_SUCCESS);
|
|
||||||
|
|
||||||
rc = trexio_text_free_rdm( (trexio_text_t*) file);
|
|
||||||
assert (rc == TREXIO_SUCCESS);
|
|
||||||
|
|
||||||
return TREXIO_SUCCESS;
|
|
||||||
}
|
|
||||||
#+end_src
|
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
|
||||||
trexio_exit_code trexio_text_unlock(trexio_t* file);
|
trexio_exit_code trexio_text_unlock(trexio_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle basic_text.c
|
||||||
trexio_exit_code trexio_text_unlock(trexio_t* file) {
|
trexio_exit_code trexio_text_unlock(trexio_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
@ -235,45 +217,62 @@ trexio_exit_code trexio_text_unlock(trexio_t* file) {
|
|||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
*** Nucleus struct
|
|
||||||
|
*** Init/deinit functions (templated part)
|
||||||
|
|
||||||
|
#+begin_src c :tangle basic_text_group.c
|
||||||
|
trexio_exit_code trexio_text_finalize(trexio_t* file) {
|
||||||
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
**** Read the struct
|
trexio_exit_code rc;
|
||||||
|
/*rc = trexio_text_free_$group$( (trexio_text_t*) file);
|
||||||
|
assert (rc == TREXIO_SUCCESS);*/
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
assert (trexio_text_free_$group$( (trexio_text_t*) file) == TREXIO_SUCCESS);
|
||||||
nucleus_t* trexio_text_read_nucleus(trexio_text_t* file);
|
|
||||||
|
rc = trexio_text_free_rdm( (trexio_text_t*) file);
|
||||||
|
assert (rc == TREXIO_SUCCESS);
|
||||||
|
|
||||||
|
return TREXIO_SUCCESS;
|
||||||
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
*** Template for text read struct
|
||||||
#define DEBUG printf("%s : line %d\n", __FILE__, __LINE__);
|
|
||||||
|
|
||||||
nucleus_t* trexio_text_read_nucleus(trexio_text_t* file) {
|
#+begin_src c :tangle read_group_text.h
|
||||||
|
$group$_t* trexio_text_read_$group$(trexio_text_t* file);
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
#+begin_src c :tangle read_group_text.c
|
||||||
|
|
||||||
|
$group$_t* trexio_text_read_$group$(trexio_text_t* file) {
|
||||||
if (file == NULL) return NULL;
|
if (file == NULL) return NULL;
|
||||||
|
|
||||||
/* If the data structure exists, return it */
|
/* If the data structure exists, return it */
|
||||||
if (file->nucleus != NULL) {
|
if (file->$group$ != NULL) {
|
||||||
return file->nucleus;
|
return file->$group$;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Allocate the data structure */
|
/* Allocate the data structure */
|
||||||
nucleus_t* nucleus = MALLOC(nucleus_t);
|
$group$_t* $group$ = MALLOC($group$_t);
|
||||||
if (nucleus == NULL) return NULL;
|
if ($group$ == NULL) return NULL;
|
||||||
|
|
||||||
memset(nucleus,0,sizeof(nucleus_t));
|
memset($group$,0,sizeof($group$_t));
|
||||||
|
|
||||||
/* Build the file name */
|
/* Build the file name */
|
||||||
const char* nucleus_file_name = "/nucleus.txt";
|
const char* $group$_file_name = "/$group$.txt";
|
||||||
char * file_name = (char*)
|
char * file_name = (char*)
|
||||||
calloc( strlen(file->parent.file_name) + strlen(nucleus_file_name) + 1,
|
calloc( strlen(file->parent.file_name) + strlen($group$_file_name) + 1,
|
||||||
sizeof(char));
|
sizeof(char));
|
||||||
|
|
||||||
if (file_name == NULL) {
|
if (file_name == NULL) {
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
strcpy (file_name, file->parent.file_name);
|
strcpy (file_name, file->parent.file_name);
|
||||||
strcat (file_name, nucleus_file_name);
|
strcat (file_name, $group$_file_name);
|
||||||
|
|
||||||
/* If the file exists, read it */
|
/* If the file exists, read it */
|
||||||
FILE* f = fopen(file_name,"r");
|
FILE* f = fopen(file_name,"r");
|
||||||
@ -287,7 +286,7 @@ DEBUG
|
|||||||
char* buffer = CALLOC(sz,char);
|
char* buffer = CALLOC(sz,char);
|
||||||
if (buffer == NULL) {
|
if (buffer == NULL) {
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
@ -295,485 +294,359 @@ DEBUG
|
|||||||
/* Read the dimensioning variables */
|
/* Read the dimensioning variables */
|
||||||
int rc;
|
int rc;
|
||||||
|
|
||||||
|
// START REPEAT GROUP_DSET
|
||||||
|
|
||||||
rc = fscanf(f, "%s", buffer);
|
rc = fscanf(f, "%s", buffer);
|
||||||
if ((rc != 1) || (strcmp(buffer, "rank_charge") != 0)) {
|
if ((rc != 1) || (strcmp(buffer, "rank_$group_dset$") != 0)) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
rc = fscanf(f, "%u", &(nucleus->rank_charge));
|
rc = fscanf(f, "%u", &($group$->rank_$group_dset$));
|
||||||
if (rc != 1) {
|
if (rc != 1) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint64_t size_charge = 1;
|
uint64_t size_$group_dset$ = 1;
|
||||||
for (unsigned int i=0; i<nucleus->rank_charge; i++){
|
for (unsigned int i=0; i<$group$->rank_$group_dset$; i++){
|
||||||
|
|
||||||
unsigned int j=-1;
|
unsigned int j=-1;
|
||||||
rc = fscanf(f, "%s %u", buffer, &j);
|
rc = fscanf(f, "%s %u", buffer, &j);
|
||||||
if ((rc != 2) || (strcmp(buffer, "dims_charge") != 0) || (j!=i)) {
|
if ((rc != 2) || (strcmp(buffer, "dims_$group_dset$") != 0) || (j!=i)) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
rc = fscanf(f, "%lu\n", &(nucleus->dims_charge[i]));
|
rc = fscanf(f, "%lu\n", &($group$->dims_$group_dset$[i]));
|
||||||
assert(!(rc != 1));
|
assert(!(rc != 1));
|
||||||
if (rc != 1) {
|
if (rc != 1) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
size_charge *= nucleus->dims_charge[i];
|
size_$group_dset$ *= $group$->dims_$group_dset$[i];
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// END REPEAT GROUP_DSET
|
||||||
|
|
||||||
|
// START REPEAT GROUP_NUM
|
||||||
|
|
||||||
|
/* Read data */
|
||||||
rc = fscanf(f, "%s", buffer);
|
rc = fscanf(f, "%s", buffer);
|
||||||
if ((rc != 1) || (strcmp(buffer, "rank_coord") != 0)) {
|
assert(!((rc != 1) || (strcmp(buffer, "$group_num$") != 0)));
|
||||||
|
if ((rc != 1) || (strcmp(buffer, "$group_num$") != 0)) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus);
|
FREE($group$);
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
rc = fscanf(f, "%u", &(nucleus->rank_coord));
|
|
||||||
assert(!(rc != 1));
|
|
||||||
if (rc != 1) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint64_t size_coord = 1;
|
rc = fscanf(f, "%lu", &($group$->$group_num$));
|
||||||
for (unsigned int i=0; i<nucleus->rank_coord; i++){
|
|
||||||
|
|
||||||
unsigned int j=-1;
|
|
||||||
rc = fscanf(f, "%s %u", buffer, &j);
|
|
||||||
if ((rc != 2) || (strcmp(buffer, "dims_coord") != 0) || (j!=i)) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
rc = fscanf(f, "%lu", &(nucleus->dims_coord[i]));
|
|
||||||
assert(!(rc != 1));
|
|
||||||
if (rc != 1) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
size_coord *= nucleus->dims_coord[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Allocate arrays */
|
|
||||||
nucleus->charge = (double*) calloc(size_charge, sizeof(double));
|
|
||||||
assert (!(nucleus->charge == NULL));
|
|
||||||
if (nucleus->charge == NULL) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
nucleus->coord = (double*) calloc(size_coord, sizeof(double));
|
|
||||||
assert (!(nucleus->coord == NULL));
|
|
||||||
if (nucleus->coord == NULL) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus->charge);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Read data */
|
|
||||||
rc = fscanf(f, "%s", buffer);
|
|
||||||
assert(!((rc != 1) || (strcmp(buffer, "num") != 0)));
|
|
||||||
if ((rc != 1) || (strcmp(buffer, "num") != 0)) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus->charge);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
rc = fscanf(f, "%lu", &(nucleus->num));
|
|
||||||
assert(!(rc != 1));
|
assert(!(rc != 1));
|
||||||
if (rc != 1) {
|
if (rc != 1) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus->charge);
|
FREE($group$);
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// END REPEAT GROUP_NUM
|
||||||
|
|
||||||
rc = fscanf(f, "%s", buffer);
|
// START REPEAT GROUP_DSET
|
||||||
assert(!((rc != 1) || (strcmp(buffer, "charge") != 0)));
|
|
||||||
if ((rc != 1) || (strcmp(buffer, "charge") != 0)) {
|
/* Allocate arrays */
|
||||||
|
$group$->$group_dset$ = (double*) calloc(size_$group_dset$, sizeof(double));
|
||||||
|
assert (!($group$->$group_dset$ == NULL));
|
||||||
|
if ($group$->$group_dset$ == NULL) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus->charge);
|
FREE($group$);
|
||||||
FREE(nucleus);
|
DEBUG
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
rc = fscanf(f, "%s", buffer);
|
||||||
|
assert(!((rc != 1) || (strcmp(buffer, "$group_dset$") != 0)));
|
||||||
|
if ((rc != 1) || (strcmp(buffer, "$group_dset$") != 0)) {
|
||||||
|
FREE(buffer);
|
||||||
|
FREE(file_name);
|
||||||
|
// TODO: free all dsets
|
||||||
|
FREE($group$->$group_dset$);
|
||||||
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (uint64_t i=0 ; i<size_charge ; i++) {
|
for (uint64_t i=0 ; i<size_$group_dset$ ; i++) {
|
||||||
rc = fscanf(f, "%lf", &(nucleus->charge[i]));
|
rc = fscanf(f, "%lf", &($group$->$group_dset$[i]));
|
||||||
assert(!(rc != 1));
|
assert(!(rc != 1));
|
||||||
if (rc != 1) {
|
if (rc != 1) {
|
||||||
FREE(buffer);
|
FREE(buffer);
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
FREE(nucleus->charge);
|
// TODO: free all dsets
|
||||||
FREE(nucleus);
|
FREE($group$->$group_dset$);
|
||||||
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// END REPEAT GROUP_DSET
|
||||||
|
|
||||||
rc = fscanf(f, "%s", buffer);
|
|
||||||
assert(!((rc != 1) || (strcmp(buffer, "coord") != 0)));
|
|
||||||
if ((rc != 1) || (strcmp(buffer, "coord") != 0)) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus->charge);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (uint64_t i=0 ; i<size_coord ; i++) {
|
|
||||||
rc = fscanf(f, "%lf", &(nucleus->coord[i]));
|
|
||||||
assert(!(rc != 1));
|
|
||||||
if (rc != 1) {
|
|
||||||
FREE(buffer);
|
|
||||||
FREE(file_name);
|
|
||||||
FREE(nucleus->charge);
|
|
||||||
FREE(nucleus);
|
|
||||||
DEBUG
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
FREE(buffer);
|
|
||||||
fclose(f);
|
|
||||||
f = NULL;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (file->parent.mode == 'w') {
|
if (file->parent.mode == 'w') {
|
||||||
nucleus->file = fopen(file_name,"a");
|
$group$->file = fopen(file_name,"a");
|
||||||
} else {
|
} else {
|
||||||
nucleus->file = fopen(file_name,"r");
|
$group$->file = fopen(file_name,"r");
|
||||||
}
|
}
|
||||||
FREE(file_name);
|
FREE(file_name);
|
||||||
assert (!(nucleus->file == NULL));
|
assert (!($group$->file == NULL));
|
||||||
if (nucleus->file == NULL) {
|
if ($group$->file == NULL) {
|
||||||
FREE(nucleus->charge);
|
// TODO: free all dsets
|
||||||
FREE(nucleus);
|
FREE($group$->$group_dset$);
|
||||||
|
FREE($group$);
|
||||||
DEBUG
|
DEBUG
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
fseek(nucleus->file, 0L, SEEK_SET);
|
fseek($group$->file, 0L, SEEK_SET);
|
||||||
file->nucleus = nucleus;
|
file->$group$ = $group$;
|
||||||
return nucleus;
|
return $group$;
|
||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
|
*** Template for text flush struct
|
||||||
|
|
||||||
**** Flush the struct
|
#+begin_src c :tangle flush_group_text.h
|
||||||
|
trexio_exit_code trexio_text_flush_$group$(const trexio_text_t* file);
|
||||||
#+begin_src c :tangle trexio_text.h
|
|
||||||
trexio_exit_code trexio_text_flush_nucleus(const trexio_text_t* file);
|
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle flush_group_text.c
|
||||||
trexio_exit_code trexio_text_flush_nucleus(const trexio_text_t* file) {
|
trexio_exit_code trexio_text_flush_$group$(const trexio_text_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
if (file->parent.mode == 'r') return TREXIO_READONLY;
|
if (file->parent.mode == 'r') return TREXIO_READONLY;
|
||||||
|
|
||||||
nucleus_t* nucleus = file->nucleus;
|
$group$_t* $group$ = file->$group$;
|
||||||
|
|
||||||
if (nucleus == NULL) return TREXIO_SUCCESS;
|
if ($group$ == NULL) return TREXIO_SUCCESS;
|
||||||
|
|
||||||
if (nucleus->to_flush == 0) return TREXIO_SUCCESS;
|
if ($group$->to_flush == 0) return TREXIO_SUCCESS;
|
||||||
|
|
||||||
FILE* f = nucleus->file;
|
FILE* f = $group$->file;
|
||||||
assert (f != NULL);
|
assert (f != NULL);
|
||||||
rewind(f);
|
rewind(f);
|
||||||
|
|
||||||
/* Write the dimensioning variables */
|
/* Write the dimensioning variables */
|
||||||
fprintf(f, "rank_charge %d\n", nucleus->rank_charge);
|
|
||||||
|
// START REPEAT GROUP_DSET
|
||||||
|
|
||||||
|
fprintf(f, "rank_$group_dset$ %d\n", $group$->rank_$group_dset$);
|
||||||
|
|
||||||
uint64_t size_charge = 1;
|
uint64_t size_$group_dset$ = 1;
|
||||||
for (unsigned int i=0; i<nucleus->rank_charge; i++){
|
for (unsigned int i=0; i<$group$->rank_$group_dset$; i++){
|
||||||
fprintf(f, "dims_charge %d %ld\n", i, nucleus->dims_charge[i]);
|
fprintf(f, "dims_$group_dset$ %d %ld\n", i, $group$->dims_$group_dset$[i]);
|
||||||
size_charge *= nucleus->dims_charge[i];
|
size_$group_dset$ *= $group$->dims_$group_dset$[i];
|
||||||
}
|
}
|
||||||
|
|
||||||
fprintf(f, "rank_coord %d\n", nucleus->rank_coord);
|
// END REPEAT GROUP_DSET
|
||||||
|
|
||||||
uint64_t size_coord = 1;
|
// START REPEAT GROUP_NUM
|
||||||
for (unsigned int i=0; i<nucleus->rank_coord; i++){
|
|
||||||
fprintf(f, "dims_coord %d %ld\n", i, nucleus->dims_coord[i]);
|
|
||||||
size_coord *= nucleus->dims_coord[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
|
fprintf(f, "$group_num$ %ld\n", $group$->$group_num$);
|
||||||
|
|
||||||
|
// END REPEAT GROUP_NUM
|
||||||
|
|
||||||
|
// START REPEAT GROUP_DSET
|
||||||
|
|
||||||
/* Write arrays */
|
/* Write arrays */
|
||||||
fprintf(f, "num %ld\n", nucleus->num);
|
|
||||||
fprintf(f, "charge\n");
|
fprintf(f, "$group_dset$\n");
|
||||||
for (uint64_t i=0 ; i<size_charge ; i++) {
|
for (uint64_t i=0 ; i<size_$group_dset$ ; i++) {
|
||||||
fprintf(f, "%lf\n", nucleus->charge[i]);
|
fprintf(f, "%lf\n", $group$->$group_dset$[i]);
|
||||||
}
|
}
|
||||||
|
|
||||||
fprintf(f, "coord\n");
|
// END REPEAT GROUP_DSET
|
||||||
for (uint64_t i=0 ; i<size_coord ; i++) {
|
|
||||||
fprintf(f, "%lf\n", nucleus->coord[i]);
|
|
||||||
}
|
|
||||||
fflush(f);
|
fflush(f);
|
||||||
file->nucleus->to_flush = 0;
|
file->$group$->to_flush = 0;
|
||||||
return TREXIO_SUCCESS;
|
return TREXIO_SUCCESS;
|
||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
**** Free memory
|
*** Template for text free memory
|
||||||
|
|
||||||
Memory is allocated when reading. The following function frees memory.
|
Memory is allocated when reading. The following function frees memory.
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle free_group_text.h
|
||||||
trexio_exit_code trexio_text_free_nucleus(trexio_text_t* file);
|
trexio_exit_code trexio_text_free_$group$(trexio_text_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle free_group_text.c
|
||||||
trexio_exit_code trexio_text_free_nucleus(trexio_text_t* file) {
|
trexio_exit_code trexio_text_free_$group$(trexio_text_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
trexio_exit_code rc;
|
trexio_exit_code rc;
|
||||||
|
|
||||||
if (file->parent.mode != 'r') {
|
if (file->parent.mode != 'r') {
|
||||||
rc = trexio_text_flush_nucleus(file);
|
rc = trexio_text_flush_$group$(file);
|
||||||
if (rc != TREXIO_SUCCESS) return TREXIO_FAILURE;
|
if (rc != TREXIO_SUCCESS) return TREXIO_FAILURE;
|
||||||
}
|
}
|
||||||
|
|
||||||
nucleus_t* nucleus = file->nucleus;
|
$group$_t* $group$ = file->$group$;
|
||||||
if (nucleus == NULL) return TREXIO_SUCCESS;
|
if ($group$ == NULL) return TREXIO_SUCCESS;
|
||||||
|
|
||||||
if (nucleus->file != NULL) {
|
if ($group$->file != NULL) {
|
||||||
fclose(nucleus->file);
|
fclose($group$->file);
|
||||||
nucleus->file = NULL;
|
$group$->file = NULL;
|
||||||
}
|
|
||||||
|
|
||||||
if (nucleus->coord != NULL) {
|
|
||||||
FREE (nucleus->coord);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (nucleus->charge != NULL) {
|
// START REPEAT GROUP_DSET
|
||||||
FREE (nucleus->charge);
|
|
||||||
|
if ($group$->$group_dset$ != NULL) {
|
||||||
|
FREE ($group$->$group_dset$);
|
||||||
}
|
}
|
||||||
|
|
||||||
FREE (nucleus);
|
// END REPEAT GROUP_DSET
|
||||||
|
|
||||||
|
FREE ($group$);
|
||||||
return TREXIO_SUCCESS;
|
return TREXIO_SUCCESS;
|
||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
**** Read/Write the num attribute
|
*** Template for read/write the $group_num$ attribute
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rw_num_text.h
|
||||||
trexio_exit_code trexio_text_read_nucleus_num(const trexio_t* file, uint64_t* num);
|
trexio_exit_code trexio_text_read_$group_num$(const trexio_t* file, uint64_t* num);
|
||||||
trexio_exit_code trexio_text_write_nucleus_num(const trexio_t* file, const uint64_t num);
|
trexio_exit_code trexio_text_write_$group_num$(const trexio_t* file, const uint64_t num);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle read_num_text.c
|
||||||
trexio_exit_code trexio_text_read_nucleus_num(const trexio_t* file, uint64_t* num) {
|
trexio_exit_code trexio_text_read_$group_num$(const trexio_t* file, uint64_t* num) {
|
||||||
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
if (num == NULL) return TREXIO_INVALID_ARG_2;
|
if (num == NULL) return TREXIO_INVALID_ARG_2;
|
||||||
|
|
||||||
nucleus_t* nucleus = trexio_text_read_nucleus((trexio_text_t*) file);
|
$group$_t* $group$ = trexio_text_read_$group$((trexio_text_t*) file);
|
||||||
if (nucleus == NULL) return TREXIO_FAILURE;
|
if ($group$ == NULL) return TREXIO_FAILURE;
|
||||||
|
|
||||||
/**/ *num = nucleus->num;
|
/**/ *num = $group$->$group_num$;
|
||||||
|
|
||||||
return TREXIO_SUCCESS;
|
return TREXIO_SUCCESS;
|
||||||
}
|
}
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
#+begin_src c :tangle write_num_text.c
|
||||||
|
|
||||||
trexio_exit_code trexio_text_write_nucleus_num(const trexio_t* file, const uint64_t num) {
|
trexio_exit_code trexio_text_write_$group_num$(const trexio_t* file, const uint64_t num) {
|
||||||
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
if (file->mode == 'r') return TREXIO_READONLY;
|
if (file->mode == 'r') return TREXIO_READONLY;
|
||||||
|
|
||||||
nucleus_t* nucleus = trexio_text_read_nucleus((trexio_text_t*) file);
|
$group$_t* $group$ = trexio_text_read_$group$((trexio_text_t*) file);
|
||||||
if (nucleus == NULL) return TREXIO_FAILURE;
|
if ($group$ == NULL) return TREXIO_FAILURE;
|
||||||
|
|
||||||
nucleus->num = num;
|
$group$->$group_num$ = num;
|
||||||
nucleus->to_flush = 1;
|
$group$->to_flush = 1;
|
||||||
|
|
||||||
return TREXIO_SUCCESS;
|
return TREXIO_SUCCESS;
|
||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
**** Read/Write the coord attribute
|
*** Template for read/write the $group_dset$ dataset
|
||||||
|
|
||||||
The ~coord~ array is assumed allocated with the appropriate size.
|
The ~dset~ array is assumed allocated with the appropriate size.
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rw_dset_text.h
|
||||||
trexio_exit_code trexio_text_read_nucleus_coord(const trexio_t* file, double* coord, const uint32_t rank, const uint64_t* dims);
|
trexio_exit_code trexio_text_read_$group_dset$(const trexio_t* file, $group_dset_dtype$* $group_dset$, const uint32_t rank, const uint64_t* dims);
|
||||||
trexio_exit_code trexio_text_write_nucleus_coord(const trexio_t* file, const double* coord, const uint32_t rank, const uint64_t* dims);
|
trexio_exit_code trexio_text_write_$group_dset$(const trexio_t* file, const $group_dset_dtype$* $group_dset$, const uint32_t rank, const uint64_t* dims);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle read_dset_text.c
|
||||||
trexio_exit_code trexio_text_read_nucleus_coord(const trexio_t* file, double* coord, const uint32_t rank, const uint64_t* dims) {
|
trexio_exit_code trexio_text_read_$group_dset$(const trexio_t* file, $group_dset_dtype$* $group_dset$, const uint32_t rank, const uint64_t* dims) {
|
||||||
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
if (coord == NULL) return TREXIO_INVALID_ARG_2;
|
if (coord == NULL) return TREXIO_INVALID_ARG_2;
|
||||||
|
|
||||||
nucleus_t* nucleus = trexio_text_read_nucleus((trexio_text_t*) file);
|
$group$_t* $group$ = trexio_text_read_$group$((trexio_text_t*) file);
|
||||||
if (nucleus == NULL) return TREXIO_FAILURE;
|
if ($group$ == NULL) return TREXIO_FAILURE;
|
||||||
|
|
||||||
if (rank != nucleus->rank_coord) return TREXIO_INVALID_ARG_3;
|
if (rank != $group$->rank_$group_dset$) return TREXIO_INVALID_ARG_3;
|
||||||
|
|
||||||
uint64_t dim_size = 1;
|
uint64_t dim_size = 1;
|
||||||
for (unsigned int i=0; i<rank; i++){
|
for (unsigned int i=0; i<rank; i++){
|
||||||
if (dims[i] != nucleus->dims_coord[i]) return TREXIO_INVALID_ARG_4;
|
if (dims[i] != $group$->dims_$group_dset$[i]) return TREXIO_INVALID_ARG_4;
|
||||||
dim_size *= dims[i];
|
dim_size *= dims[i];
|
||||||
}
|
}
|
||||||
|
|
||||||
for (uint64_t i=0 ; i<dim_size ; i++) {
|
for (uint64_t i=0 ; i<dim_size ; i++) {
|
||||||
coord[i] = nucleus->coord[i];
|
$group_dset$[i] = $group$->$group_dset$[i];
|
||||||
}
|
}
|
||||||
|
|
||||||
return TREXIO_SUCCESS;
|
return TREXIO_SUCCESS;
|
||||||
}
|
}
|
||||||
|
#+end_src
|
||||||
|
|
||||||
|
#+begin_src c :tangle write_dset_text.c
|
||||||
|
|
||||||
trexio_exit_code trexio_text_write_nucleus_coord(const trexio_t* file, const double* coord, const uint32_t rank, const uint64_t* dims) {
|
trexio_exit_code trexio_text_write_$group_dset$(const trexio_t* file, const $group_dset_dtype$* $group_dset$, const uint32_t rank, const uint64_t* dims) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
if (coord == NULL) return TREXIO_INVALID_ARG_2;
|
if (coord == NULL) return TREXIO_INVALID_ARG_2;
|
||||||
|
|
||||||
if (file->mode == 'r') return TREXIO_READONLY;
|
if (file->mode == 'r') return TREXIO_READONLY;
|
||||||
|
|
||||||
nucleus_t* nucleus = trexio_text_read_nucleus((trexio_text_t*) file);
|
$group$_t* $group$ = trexio_text_read_$group$((trexio_text_t*) file);
|
||||||
if (nucleus == NULL) return TREXIO_FAILURE;
|
if ($group$ == NULL) return TREXIO_FAILURE;
|
||||||
|
|
||||||
if (nucleus->coord != NULL) {
|
if ($group$->$group_dset$ != NULL) {
|
||||||
FREE(nucleus->coord);
|
FREE($group$->$group_dset$);
|
||||||
}
|
}
|
||||||
|
|
||||||
nucleus->rank_coord = rank;
|
$group$->rank_$group_dset$ = rank;
|
||||||
|
|
||||||
uint64_t dim_size = 1;
|
uint64_t dim_size = 1;
|
||||||
for (unsigned int i=0; i<nucleus->rank_coord; i++){
|
for (unsigned int i=0; i<$group$->rank_$group_dset$; i++){
|
||||||
nucleus->dims_coord[i] = dims[i];
|
$group$->dims_$group_dset$[i] = dims[i];
|
||||||
dim_size *= dims[i];
|
dim_size *= dims[i];
|
||||||
}
|
}
|
||||||
|
|
||||||
nucleus->coord = (double*) calloc(dim_size, sizeof(double));
|
$group$->$group_dset$ = (double*) calloc(dim_size, sizeof(double));
|
||||||
|
|
||||||
for (uint64_t i=0 ; i<dim_size ; i++) {
|
for (uint64_t i=0 ; i<dim_size ; i++) {
|
||||||
nucleus->coord[i] = coord[i];
|
$group$->$group_dset$[i] = $group_dset$[i];
|
||||||
}
|
}
|
||||||
|
|
||||||
nucleus->to_flush = 1;
|
$group$->to_flush = 1;
|
||||||
return TREXIO_SUCCESS;
|
|
||||||
}
|
|
||||||
#+end_src
|
|
||||||
**** Read/Write the charge attribute
|
|
||||||
|
|
||||||
The ~charge~ array is assumed allocated with the appropriate size.
|
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
|
||||||
trexio_exit_code trexio_text_read_nucleus_charge(const trexio_t* file, double* charge, const uint32_t rank, const uint64_t* dims);
|
|
||||||
trexio_exit_code trexio_text_write_nucleus_charge(const trexio_t* file, const double* charge, const uint32_t rank, const uint64_t* dims);
|
|
||||||
#+end_src
|
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
|
||||||
trexio_exit_code trexio_text_read_nucleus_charge(const trexio_t* file, double* charge, const uint32_t rank, const uint64_t* dims) {
|
|
||||||
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
|
||||||
if (charge == NULL) return TREXIO_INVALID_ARG_2;
|
|
||||||
|
|
||||||
nucleus_t* nucleus = trexio_text_read_nucleus((trexio_text_t*) file);
|
|
||||||
if (nucleus == NULL) return TREXIO_FAILURE;
|
|
||||||
|
|
||||||
if (rank != nucleus->rank_charge) return TREXIO_INVALID_ARG_3;
|
|
||||||
|
|
||||||
uint64_t dim_size = 1;
|
|
||||||
for (unsigned int i=0; i<rank; i++){
|
|
||||||
if (dims[i] != nucleus->dims_charge[i]) return TREXIO_INVALID_ARG_4;
|
|
||||||
dim_size *= dims[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
for (uint64_t i=0 ; i<dim_size ; i++) {
|
|
||||||
charge[i] = nucleus->charge[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
return TREXIO_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
trexio_exit_code trexio_text_write_nucleus_charge(const trexio_t* file, const double* charge, const uint32_t rank, const uint64_t* dims) {
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
|
||||||
if (charge == NULL) return TREXIO_INVALID_ARG_2;
|
|
||||||
|
|
||||||
if (file->mode == 'r') return TREXIO_READONLY;
|
|
||||||
|
|
||||||
nucleus_t* nucleus = trexio_text_read_nucleus((trexio_text_t*) file);
|
|
||||||
if (nucleus == NULL) return TREXIO_FAILURE;
|
|
||||||
|
|
||||||
if (nucleus->charge != NULL) {
|
|
||||||
FREE(nucleus->charge);
|
|
||||||
}
|
|
||||||
|
|
||||||
nucleus->rank_charge = rank;
|
|
||||||
|
|
||||||
uint64_t dim_size = 1;
|
|
||||||
for (unsigned int i=0; i<nucleus->rank_charge; i++){
|
|
||||||
nucleus->dims_charge[i] = dims[i];
|
|
||||||
dim_size *= dims[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
nucleus->charge = (double*) calloc(dim_size, sizeof(double));
|
|
||||||
|
|
||||||
for (uint64_t i=0 ; i<dim_size ; i++) {
|
|
||||||
nucleus->charge[i] = charge[i];
|
|
||||||
}
|
|
||||||
|
|
||||||
nucleus->to_flush = 1;
|
|
||||||
return TREXIO_SUCCESS;
|
return TREXIO_SUCCESS;
|
||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
|
|
||||||
*** RDM struct
|
*** RDM struct
|
||||||
**** Read the complete struct
|
**** Read the complete struct
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rdm_text.h
|
||||||
rdm_t* trexio_text_read_rdm(trexio_text_t* file);
|
rdm_t* trexio_text_read_rdm(trexio_text_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle rdm_text.c
|
||||||
rdm_t* trexio_text_read_rdm(trexio_text_t* file) {
|
rdm_t* trexio_text_read_rdm(trexio_text_t* file) {
|
||||||
if (file == NULL) return NULL;
|
if (file == NULL) return NULL;
|
||||||
|
|
||||||
@ -857,11 +730,11 @@ rdm_t* trexio_text_read_rdm(trexio_text_t* file) {
|
|||||||
|
|
||||||
**** Flush the complete struct
|
**** Flush the complete struct
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rdm_text.h
|
||||||
trexio_exit_code trexio_text_flush_rdm(const trexio_text_t* file);
|
trexio_exit_code trexio_text_flush_rdm(const trexio_text_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle rdm_text.c
|
||||||
trexio_exit_code trexio_text_flush_rdm(const trexio_text_t* file) {
|
trexio_exit_code trexio_text_flush_rdm(const trexio_text_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
@ -898,11 +771,11 @@ trexio_exit_code trexio_text_flush_rdm(const trexio_text_t* file) {
|
|||||||
|
|
||||||
Memory is allocated when reading. The followig function frees memory.
|
Memory is allocated when reading. The followig function frees memory.
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rdm_text.h
|
||||||
trexio_exit_code trexio_text_free_rdm(trexio_text_t* file);
|
trexio_exit_code trexio_text_free_rdm(trexio_text_t* file);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle rdm_text.c
|
||||||
trexio_exit_code trexio_text_free_rdm(trexio_text_t* file) {
|
trexio_exit_code trexio_text_free_rdm(trexio_text_t* file) {
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
|
|
||||||
@ -938,12 +811,12 @@ trexio_exit_code trexio_text_free_rdm(trexio_text_t* file) {
|
|||||||
|
|
||||||
The ~one_e~ array is assumed allocated with the appropriate size.
|
The ~one_e~ array is assumed allocated with the appropriate size.
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rdm_text.h
|
||||||
trexio_exit_code trexio_text_read_rdm_one_e(const trexio_t* file, double* one_e, const uint64_t dim_one_e);
|
trexio_exit_code trexio_text_read_rdm_one_e(const trexio_t* file, double* one_e, const uint64_t dim_one_e);
|
||||||
trexio_exit_code trexio_text_write_rdm_one_e(const trexio_t* file, const double* one_e, const uint64_t dim_one_e);
|
trexio_exit_code trexio_text_write_rdm_one_e(const trexio_t* file, const double* one_e, const uint64_t dim_one_e);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle rdm_text.c
|
||||||
trexio_exit_code trexio_text_read_rdm_one_e(const trexio_t* file, double* one_e, const uint64_t dim_one_e) {
|
trexio_exit_code trexio_text_read_rdm_one_e(const trexio_t* file, double* one_e, const uint64_t dim_one_e) {
|
||||||
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
@ -988,12 +861,12 @@ trexio_exit_code trexio_text_write_rdm_one_e(const trexio_t* file, const double*
|
|||||||
In the text back end, the easiest way to do it is to create a
|
In the text back end, the easiest way to do it is to create a
|
||||||
file for each sparse float structure.
|
file for each sparse float structure.
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
#+begin_src c :tangle rdm_text.h
|
||||||
trexio_exit_code trexio_text_buffered_read_rdm_two_e(const trexio_t* file, const uint64_t offset, const uint64_t size, int64_t* index, double* value);
|
trexio_exit_code trexio_text_buffered_read_rdm_two_e(const trexio_t* file, const uint64_t offset, const uint64_t size, int64_t* index, double* value);
|
||||||
trexio_exit_code trexio_text_buffered_write_rdm_two_e(const trexio_t* file, const uint64_t offset, const uint64_t size, const int64_t* index, const double* value);
|
trexio_exit_code trexio_text_buffered_write_rdm_two_e(const trexio_t* file, const uint64_t offset, const uint64_t size, const int64_t* index, const double* value);
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.c
|
#+begin_src c :tangle rdm_text.c
|
||||||
trexio_exit_code trexio_text_buffered_read_rdm_two_e(const trexio_t* file, const uint64_t offset, const uint64_t size, int64_t* index, double* value) {
|
trexio_exit_code trexio_text_buffered_read_rdm_two_e(const trexio_t* file, const uint64_t offset, const uint64_t size, int64_t* index, double* value) {
|
||||||
|
|
||||||
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
if (file == NULL) return TREXIO_INVALID_ARG_1;
|
||||||
@ -1060,9 +933,11 @@ trexio_exit_code trexio_text_buffered_write_rdm_two_e(const trexio_t* file, cons
|
|||||||
}
|
}
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
* File suffixes :noxport:
|
:noexport:
|
||||||
|
|
||||||
#+begin_src c :tangle trexio_text.h
|
* Constant file suffixes :noxport:
|
||||||
|
|
||||||
|
#+begin_src c :tangle suffix_text.h
|
||||||
#endif
|
#endif
|
||||||
#+end_src
|
#+end_src
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user