1*3d8e8822SJeremy L Thompson# Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors 2*3d8e8822SJeremy L Thompson# All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 37b8bbb42Svaleriabarra# 4*3d8e8822SJeremy L Thompson# SPDX-License-Identifier: BSD-2-Clause 57b8bbb42Svaleriabarra# 6*3d8e8822SJeremy L Thompson# This file is part of CEED: http://github.com/ceed 77b8bbb42Svaleriabarra 87b8bbb42Svaleriabarraimport os 9477729cfSJeremy L Thompsonimport re 107b8bbb42Svaleriabarrafrom cffi import FFI 117b8bbb42Svaleriabarraffibuilder = FFI() 127b8bbb42Svaleriabarra 13b3bcc043SJed Brownceed_version_ge = re.compile(r'\s+\(!?CEED_VERSION.*') 14bd4df462SJed Brown 157b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 167b8bbb42Svaleriabarra# Provide C definitions to CFFI 177b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 18ec3da8bcSJed Brownwith open(os.path.abspath("include/ceed/ceed.h")) as f: 197b8bbb42Svaleriabarra lines = [line.strip() for line in f if 2080a9ef05SNatalie Beams not (line.startswith("#") and not line.startswith("#include")) and 217b8bbb42Svaleriabarra not line.startswith(" static") and 227b8bbb42Svaleriabarra "CeedErrorImpl" not in line and 237b8bbb42Svaleriabarra "const char *, ...);" not in line and 24bd4df462SJed Brown not line.startswith("CEED_EXTERN const char *const") and 25bd4df462SJed Brown not ceed_version_ge.match(line)] 267b8bbb42Svaleriabarra lines = [line.replace("CEED_EXTERN", "extern") for line in lines] 2780a9ef05SNatalie Beams # Find scalar type inclusion line and insert definitions 2880a9ef05SNatalie Beams for line in lines: 2980a9ef05SNatalie Beams if re.search("ceed-f32.h", line) is not None: 3080a9ef05SNatalie Beams insert_index = lines.index(line) + 1 3180a9ef05SNatalie Beams extra_lines = ['typedef float CeedScalar;'] 3280a9ef05SNatalie Beams extra_lines.append('static const int CEED_SCALAR_TYPE;') 3380a9ef05SNatalie Beams extra_lines.append('static const double CEED_EPSILON;') 3480a9ef05SNatalie Beams elif re.search("ceed-f64.h", line) is not None: 3580a9ef05SNatalie Beams insert_index = lines.index(line) + 1 3680a9ef05SNatalie Beams extra_lines = ['typedef double CeedScalar;'] 3780a9ef05SNatalie Beams extra_lines.append('static const int CEED_SCALAR_TYPE;') 3880a9ef05SNatalie Beams extra_lines.append('static const double CEED_EPSILON;') 3980a9ef05SNatalie Beams lines[insert_index: insert_index] = extra_lines 4080a9ef05SNatalie Beams # Remove all include statements now that scalar type has been dealt with 4180a9ef05SNatalie Beams lines = [line for line in lines if not line.startswith("#include")] 4280a9ef05SNatalie Beams # Build header from lines 431d83af80SJed Brown header = '\n'.join(lines) 447b8bbb42Svaleriabarra header = header.split("static inline CeedInt CeedIntPow", 1)[0] 4519798369SJed Brown header += '\nextern int CeedVectorGetState(CeedVector, uint64_t*);' 46c3a5a609SJeremy L Thompson header += '\nextern int CeedElemRestrictionGetELayout(CeedElemRestriction, CeedInt *layout);' 47477729cfSJeremy L Thompson # Note: cffi cannot handle vargs 48477729cfSJeremy L Thompson header = re.sub("va_list", "const char *", header) 497b8bbb42Svaleriabarraffibuilder.cdef(header) 507b8bbb42Svaleriabarra 517b8bbb42Svaleriabarraffibuilder.set_source("_ceed_cffi", 527b8bbb42Svaleriabarra """ 53477729cfSJeremy L Thompson #define va_list const char * 5480a9ef05SNatalie Beams #include <ceed/ceed.h> // the C header of the library 557b8bbb42Svaleriabarra """, 56450bb777Svaleriabarra include_dirs=[ 57450bb777Svaleriabarra os.path.abspath("include")], # include path 587b8bbb42Svaleriabarra libraries=["ceed"], # library name, for the linker 5937c134eaSJed Brown library_dirs=['./lib'], # library path, for the linker 607a7b0fa3SJed Brown # use libceed.so as installed 617a7b0fa3SJed Brown runtime_library_dirs=['$ORIGIN/libceed/lib'] 627b8bbb42Svaleriabarra ) 637b8bbb42Svaleriabarra 647b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 657b8bbb42Svaleriabarra# Builder 667b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 677b8bbb42Svaleriabarraif __name__ == "__main__": 687b8bbb42Svaleriabarra ffibuilder.compile(verbose=True) 697b8bbb42Svaleriabarra 707b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 71