13d8e8822SJeremy L Thompson# Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors 23d8e8822SJeremy L Thompson# All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 37b8bbb42Svaleriabarra# 43d8e8822SJeremy L Thompson# SPDX-License-Identifier: BSD-2-Clause 57b8bbb42Svaleriabarra# 63d8e8822SJeremy 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# ------------------------------------------------------------------------------ 18c9c2c079SJeremy L Thompsonlines = [] 19c9c2c079SJeremy L Thompsonfor header_path in ["include/ceed/types.h", "include/ceed/ceed.h"]: 20c9c2c079SJeremy L Thompson with open(os.path.abspath(header_path)) as f: 21c9c2c079SJeremy L Thompson lines += [line.strip() for line in f if 2280a9ef05SNatalie Beams not (line.startswith("#") and not line.startswith("#include")) and 237b8bbb42Svaleriabarra not line.startswith(" static") and 24*228d9efbSJames Wright not line.startswith(" CEED_QFUNCTION_ATTR") and 257b8bbb42Svaleriabarra "CeedErrorImpl" not in line and 267b8bbb42Svaleriabarra "const char *, ...);" not in line and 27bd4df462SJed Brown not line.startswith("CEED_EXTERN const char *const") and 28bd4df462SJed Brown not ceed_version_ge.match(line)] 297b8bbb42Svaleriabarralines = [line.replace("CEED_EXTERN", "extern") for line in lines] 30c9c2c079SJeremy L Thompson 3180a9ef05SNatalie Beams# Find scalar type inclusion line and insert definitions 3280a9ef05SNatalie Beamsfor line in lines: 3380a9ef05SNatalie Beams if re.search("ceed-f32.h", line) is not None: 3480a9ef05SNatalie Beams insert_index = lines.index(line) + 1 3580a9ef05SNatalie Beams extra_lines = ['typedef float CeedScalar;'] 3680a9ef05SNatalie Beams extra_lines.append('static const int CEED_SCALAR_TYPE;') 3780a9ef05SNatalie Beams extra_lines.append('static const double CEED_EPSILON;') 3880a9ef05SNatalie Beams elif re.search("ceed-f64.h", line) is not None: 3980a9ef05SNatalie Beams insert_index = lines.index(line) + 1 4080a9ef05SNatalie Beams extra_lines = ['typedef double CeedScalar;'] 4180a9ef05SNatalie Beams extra_lines.append('static const int CEED_SCALAR_TYPE;') 4280a9ef05SNatalie Beams extra_lines.append('static const double CEED_EPSILON;') 4380a9ef05SNatalie Beamslines[insert_index: insert_index] = extra_lines 44c9c2c079SJeremy L Thompson 4580a9ef05SNatalie Beams# Remove all include statements now that scalar type has been dealt with 4680a9ef05SNatalie Beamslines = [line for line in lines if not line.startswith("#include")] 47c9c2c079SJeremy L Thompson 4880a9ef05SNatalie Beams# Build header from lines 491d83af80SJed Brownheader = '\n'.join(lines) 507b8bbb42Svaleriabarraheader = header.split("static inline CeedInt CeedIntPow", 1)[0] 5119798369SJed Brownheader += '\nextern int CeedVectorGetState(CeedVector, uint64_t*);' 52c3a5a609SJeremy L Thompsonheader += '\nextern int CeedElemRestrictionGetELayout(CeedElemRestriction, CeedInt *layout);' 53c9c2c079SJeremy L Thompson 54477729cfSJeremy L Thompson# Note: cffi cannot handle vargs 55477729cfSJeremy L Thompsonheader = re.sub("va_list", "const char *", header) 56c9c2c079SJeremy L Thompson 577b8bbb42Svaleriabarraffibuilder.cdef(header) 587b8bbb42Svaleriabarra 597b8bbb42Svaleriabarraffibuilder.set_source("_ceed_cffi", 607b8bbb42Svaleriabarra """ 61477729cfSJeremy L Thompson #define va_list const char * 6280a9ef05SNatalie Beams #include <ceed/ceed.h> // the C header of the library 637b8bbb42Svaleriabarra """, 64450bb777Svaleriabarra include_dirs=[ 65450bb777Svaleriabarra os.path.abspath("include")], # include path 667b8bbb42Svaleriabarra libraries=["ceed"], # library name, for the linker 6737c134eaSJed Brown library_dirs=['./lib'], # library path, for the linker 687a7b0fa3SJed Brown # use libceed.so as installed 697a7b0fa3SJed Brown runtime_library_dirs=['$ORIGIN/libceed/lib'] 707b8bbb42Svaleriabarra ) 717b8bbb42Svaleriabarra 727b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 737b8bbb42Svaleriabarra# Builder 747b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 757b8bbb42Svaleriabarraif __name__ == "__main__": 767b8bbb42Svaleriabarra ffibuilder.compile(verbose=True) 777b8bbb42Svaleriabarra 787b8bbb42Svaleriabarra# ------------------------------------------------------------------------------ 79