xref: /libCEED/python/build_ceed_cffi.py (revision 37fad103dc80356d29612d88537158f0d6682656)
1# Copyright (c) 2017, Lawrence Livermore National Security, LLC. Produced at
2# the Lawrence Livermore National Laboratory. LLNL-CODE-734707. All Rights
3# reserved. See files LICENSE and NOTICE for details.
4#
5# This file is part of CEED, a collection of benchmarks, miniapps, software
6# libraries and APIs for efficient high-order finite element and spectral
7# element discretizations for exascale applications. For more information and
8# source code availability see http://github.com/ceed.
9#
10# The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11# a collaborative effort of two U.S. Department of Energy organizations (Office
12# of Science and the National Nuclear Security Administration) responsible for
13# the planning and preparation of a capable exascale ecosystem, including
14# software, applications, hardware, advanced system engineering and early
15# testbed platforms, in support of the nation's exascale computing imperative.
16
17import os
18import re
19from cffi import FFI
20ffibuilder = FFI()
21
22# ------------------------------------------------------------------------------
23# Provide C definitions to CFFI
24# ------------------------------------------------------------------------------
25with open(os.path.abspath("include/ceed.h")) as f:
26    lines = [line.strip() for line in f if
27             not line.startswith("#") and
28             not line.startswith("  static") and
29             "CeedErrorImpl" not in line and
30             "const char *, ...);" not in line and
31             not line.startswith("CEED_EXTERN const char *const")]
32    lines = [line.replace("CEED_EXTERN", "extern") for line in lines]
33    header = '\n'.join(lines)
34    header = header.split("static inline CeedInt CeedIntPow", 1)[0]
35    header += '\nextern int CeedVectorGetState(CeedVector, uint64_t*);'
36    # Note: cffi cannot handle vargs
37    header = re.sub("va_list", "const char *", header)
38ffibuilder.cdef(header)
39
40ffibuilder.set_source("_ceed_cffi",
41                      """
42  #define va_list const char *
43  #include <ceed.h>   // the C header of the library
44  """,
45                      include_dirs=[
46                          os.path.abspath("include")],  # include path
47                      libraries=["ceed"],   # library name, for the linker
48                      library_dirs=['./lib'],  # library path, for the linker
49                      # use libceed.so as installed
50                      runtime_library_dirs=['$ORIGIN/libceed/lib']
51                      )
52
53# ------------------------------------------------------------------------------
54# Builder
55# ------------------------------------------------------------------------------
56if __name__ == "__main__":
57    ffibuilder.compile(verbose=True)
58
59# ------------------------------------------------------------------------------
60