File: gen_mpi_prototypes.py

package info (click to toggle)
mpich 4.3.0%2Breally4.2.1-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 419,120 kB
  • sloc: ansic: 1,215,557; cpp: 74,755; javascript: 40,763; f90: 20,649; sh: 18,463; xml: 14,418; python: 14,397; perl: 13,772; makefile: 9,279; fortran: 8,063; java: 4,553; asm: 324; ruby: 176; lisp: 19; php: 8; sed: 4
file content (49 lines) | stat: -rw-r--r-- 1,463 bytes parent folder | download | duplicates (2)
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
##
## Copyright (C) by Argonne National Laboratory
##     See COPYRIGHT in top-level directory
##
#
# This script generates all MPI function prototypes for both
# MPI and PMPI prefixes.
#
# It serves as an example of how to use the python framework.
#

from local_python import MPI_API_Global as G
from local_python.mpi_api import *
from local_python.binding_c import *
from local_python import RE

def main():
    binding_dir = G.get_srcdir_path("src/binding")
    load_mpi_api(binding_dir + "/mpi_standard_api.txt")
    load_mpi_mapping(binding_dir + "/apis_mapping.txt")

    func_list = G.FUNCS.values()

    decls = []
    fndefs = []
    for func in func_list:
        if re.match(r'.*_(function)', func['name']):
            # these are function typedefs
            fndefs.append(get_declare_function(func, False))
            if function_has_POLY_parameters(func):
                fndefs.append(get_declare_function(func, True))
        else:
            decls.append(get_declare_function(func, False))
            if function_has_POLY_parameters(func):
                decls.append(get_declare_function(func, True))
    
    for l in fndefs:
        print("typedef " + l + ";")

    for l in decls:
        print(l + ";")

    for l in decls:
        # assume the first MPI_ is the function name
        print(re.sub(r'\b(MPI_\w+)\(', r'P\1(', l) + ";")

# ---------------------------------------------------------
if __name__ == "__main__":
    main()