100 lines
2.6 KiB
Python
100 lines
2.6 KiB
Python
![]() |
#
|
||
|
# preprocessor.py
|
||
|
#
|
||
|
import subprocess,os,re
|
||
|
|
||
|
verbose = 0
|
||
|
|
||
|
def blab(str):
|
||
|
if verbose:
|
||
|
print(str)
|
||
|
|
||
|
################################################################################
|
||
|
#
|
||
|
# Invoke GCC to run the preprocessor and extract enabled features
|
||
|
#
|
||
|
preprocessor_cache = {}
|
||
|
def run_preprocessor(env, fn=None):
|
||
|
filename = fn or 'buildroot/share/PlatformIO/scripts/common-dependencies.h'
|
||
|
if filename in preprocessor_cache:
|
||
|
return preprocessor_cache[filename]
|
||
|
|
||
|
# Process defines
|
||
|
build_flags = env.get('BUILD_FLAGS')
|
||
|
build_flags = env.ParseFlagsExtended(build_flags)
|
||
|
|
||
|
cxx = search_compiler(env)
|
||
|
cmd = ['"' + cxx + '"']
|
||
|
|
||
|
# Build flags from board.json
|
||
|
#if 'BOARD' in env:
|
||
|
# cmd += [env.BoardConfig().get("build.extra_flags")]
|
||
|
for s in build_flags['CPPDEFINES']:
|
||
|
if isinstance(s, tuple):
|
||
|
cmd += ['-D' + s[0] + '=' + str(s[1])]
|
||
|
else:
|
||
|
cmd += ['-D' + s]
|
||
|
|
||
|
cmd += ['-D__MARLIN_DEPS__ -w -dM -E -x c++']
|
||
|
depcmd = cmd + [ filename ]
|
||
|
cmd = ' '.join(depcmd)
|
||
|
blab(cmd)
|
||
|
define_list = subprocess.check_output(cmd, shell=True).splitlines()
|
||
|
preprocessor_cache[filename] = define_list
|
||
|
return define_list
|
||
|
|
||
|
|
||
|
################################################################################
|
||
|
#
|
||
|
# Find a compiler, considering the OS
|
||
|
#
|
||
|
def search_compiler(env):
|
||
|
|
||
|
ENV_BUILD_PATH = os.path.join(env.Dictionary('PROJECT_BUILD_DIR'), env['PIOENV'])
|
||
|
GCC_PATH_CACHE = os.path.join(ENV_BUILD_PATH, ".gcc_path")
|
||
|
|
||
|
try:
|
||
|
filepath = env.GetProjectOption('custom_gcc')
|
||
|
blab("Getting compiler from env")
|
||
|
return filepath
|
||
|
except:
|
||
|
pass
|
||
|
|
||
|
if os.path.exists(GCC_PATH_CACHE):
|
||
|
blab("Getting g++ path from cache")
|
||
|
with open(GCC_PATH_CACHE, 'r') as f:
|
||
|
return f.read()
|
||
|
|
||
|
# Find the current platform compiler by searching the $PATH
|
||
|
# which will be in a platformio toolchain bin folder
|
||
|
path_regex = re.escape(env['PROJECT_PACKAGES_DIR'])
|
||
|
gcc = "g++"
|
||
|
if env['PLATFORM'] == 'win32':
|
||
|
path_separator = ';'
|
||
|
path_regex += r'.*\\bin'
|
||
|
gcc += ".exe"
|
||
|
else:
|
||
|
path_separator = ':'
|
||
|
path_regex += r'/.+/bin'
|
||
|
|
||
|
# Search for the compiler
|
||
|
for pathdir in env['ENV']['PATH'].split(path_separator):
|
||
|
if not re.search(path_regex, pathdir, re.IGNORECASE):
|
||
|
continue
|
||
|
for filepath in os.listdir(pathdir):
|
||
|
if not filepath.endswith(gcc):
|
||
|
continue
|
||
|
# Use entire path to not rely on env PATH
|
||
|
filepath = os.path.sep.join([pathdir, filepath])
|
||
|
# Cache the g++ path to no search always
|
||
|
if os.path.exists(ENV_BUILD_PATH):
|
||
|
blab("Caching g++ for current env")
|
||
|
with open(GCC_PATH_CACHE, 'w+') as f:
|
||
|
f.write(filepath)
|
||
|
|
||
|
return filepath
|
||
|
|
||
|
filepath = env.get('CXX')
|
||
|
blab("Couldn't find a compiler! Fallback to %s" % filepath)
|
||
|
return filepath
|