llvm.py revision ddb081e3edaadfefe8fdbefe6f572e1632e55942
1"""llvm 2 3Tool-specific initialization for LLVM 4 5""" 6 7# 8# Copyright (c) 2009 VMware, Inc. 9# 10# Permission is hereby granted, free of charge, to any person obtaining 11# a copy of this software and associated documentation files (the 12# "Software"), to deal in the Software without restriction, including 13# without limitation the rights to use, copy, modify, merge, publish, 14# distribute, sublicense, and/or sell copies of the Software, and to 15# permit persons to whom the Software is furnished to do so, subject to 16# the following conditions: 17# 18# The above copyright notice and this permission notice shall be included 19# in all copies or substantial portions of the Software. 20# 21# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY 22# KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE 23# WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 24# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE 25# LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 26# OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 27# WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 28# 29 30import os 31import os.path 32import re 33import sys 34import distutils.version 35 36import SCons.Errors 37import SCons.Util 38 39 40def generate(env): 41 try: 42 llvm_dir = os.environ['LLVM'] 43 except KeyError: 44 # Do nothing -- use the system headers/libs 45 llvm_dir = None 46 else: 47 if not os.path.isdir(llvm_dir): 48 raise SCons.Errors.InternalError, "Specified LLVM directory not found" 49 50 if env['debug']: 51 llvm_subdir = 'Debug' 52 else: 53 llvm_subdir = 'Release' 54 55 llvm_bin_dir = os.path.join(llvm_dir, llvm_subdir, 'bin') 56 if not os.path.isdir(llvm_bin_dir): 57 llvm_bin_dir = os.path.join(llvm_dir, 'bin') 58 if not os.path.isdir(llvm_bin_dir): 59 raise SCons.Errors.InternalError, "LLVM binary directory not found" 60 61 env.PrependENVPath('PATH', llvm_bin_dir) 62 63 if env['platform'] == 'windows': 64 # XXX: There is no llvm-config on Windows, so assume a standard layout 65 if llvm_dir is None: 66 return 67 68 # Try to determine the LLVM version from llvm/Config/config.h 69 llvm_config = os.path.join(llvm_dir, 'include/llvm/Config/config.h') 70 if not os.path.exists(llvm_config): 71 print 'scons: could not find %s' % llvm_config 72 return 73 llvm_version_re = re.compile(r'^#define PACKAGE_VERSION "([^"]*)"') 74 llvm_version = None 75 for line in open(llvm_config, 'rt'): 76 mo = llvm_version_re.match(line) 77 if mo: 78 llvm_version = mo.group(1) 79 llvm_version = distutils.version.LooseVersion(llvm_version) 80 break 81 if llvm_version is None: 82 print 'scons: could not determine the LLVM version from %s' % llvm_config 83 return 84 85 env.Prepend(CPPPATH = [os.path.join(llvm_dir, 'include')]) 86 env.AppendUnique(CPPDEFINES = [ 87 '__STDC_LIMIT_MACROS', 88 '__STDC_CONSTANT_MACROS', 89 'HAVE_STDINT_H', 90 ]) 91 env.Prepend(LIBPATH = [os.path.join(llvm_dir, 'lib')]) 92 if llvm_version >= distutils.version.LooseVersion('2.7'): 93 # 2.7 94 env.Prepend(LIBS = [ 95 'LLVMLinker', 'LLVMipo', 'LLVMInterpreter', 96 'LLVMInstrumentation', 'LLVMJIT', 'LLVMExecutionEngine', 97 'LLVMBitWriter', 'LLVMX86Disassembler', 'LLVMX86AsmParser', 98 'LLVMMCParser', 'LLVMX86AsmPrinter', 'LLVMX86CodeGen', 99 'LLVMSelectionDAG', 'LLVMX86Info', 'LLVMAsmPrinter', 100 'LLVMCodeGen', 'LLVMScalarOpts', 'LLVMInstCombine', 101 'LLVMTransformUtils', 'LLVMipa', 'LLVMAsmParser', 102 'LLVMArchive', 'LLVMBitReader', 'LLVMAnalysis', 'LLVMTarget', 103 'LLVMMC', 'LLVMCore', 'LLVMSupport', 'LLVMSystem', 104 ]) 105 else: 106 # 2.6 107 env.Prepend(LIBS = [ 108 'LLVMX86AsmParser', 'LLVMX86AsmPrinter', 'LLVMX86CodeGen', 109 'LLVMX86Info', 'LLVMLinker', 'LLVMipo', 'LLVMInterpreter', 110 'LLVMInstrumentation', 'LLVMJIT', 'LLVMExecutionEngine', 111 'LLVMDebugger', 'LLVMBitWriter', 'LLVMAsmParser', 112 'LLVMArchive', 'LLVMBitReader', 'LLVMSelectionDAG', 113 'LLVMAsmPrinter', 'LLVMCodeGen', 'LLVMScalarOpts', 114 'LLVMTransformUtils', 'LLVMipa', 'LLVMAnalysis', 115 'LLVMTarget', 'LLVMMC', 'LLVMCore', 'LLVMSupport', 116 'LLVMSystem', 117 ]) 118 env.Append(LIBS = [ 119 'imagehlp', 120 'psapi', 121 ]) 122 if env['msvc']: 123 # Some of the LLVM C headers use the inline keyword without 124 # defining it. 125 env.Append(CPPDEFINES = [('inline', '__inline')]) 126 if env['debug']: 127 # LLVM libraries are static, build with /MT, and they 128 # automatically link agains LIBCMT. When we're doing a 129 # debug build we'll be linking against LIBCMTD, so disable 130 # that. 131 env.Append(LINKFLAGS = ['/nodefaultlib:LIBCMT']) 132 elif env.Detect('llvm-config'): 133 llvm_version = env.backtick('llvm-config --version').rstrip() 134 llvm_version = distutils.version.LooseVersion(llvm_version) 135 136 try: 137 env.ParseConfig('llvm-config --cppflags') 138 env.ParseConfig('llvm-config --libs jit interpreter nativecodegen bitwriter') 139 env.ParseConfig('llvm-config --ldflags') 140 except OSError: 141 print 'llvm-config version %s failed' % llvm_version 142 else: 143 env['LINK'] = env['CXX'] 144 else: 145 return 146 147 assert llvm_version is not None 148 149 print 'scons: Found LLVM version %s' % llvm_version 150 env['LLVM_VERSION'] = llvm_version 151 152 # Define HAVE_LLVM macro with the major/minor version number (e.g., 0x0206 for 2.6) 153 llvm_version_major = int(llvm_version.version[0]) 154 llvm_version_minor = int(llvm_version.version[1]) 155 llvm_version_hex = '0x%02x%02x' % (llvm_version_major, llvm_version_minor) 156 env.Prepend(CPPDEFINES = [('HAVE_LLVM', llvm_version_hex)]) 157 158def exists(env): 159 return True 160 161# vim:set ts=4 sw=4 et: 162