blob: 3fef9e0900d59b2cd57a719cbd5d69aa0571683c [file] [log] [blame]
José Fonseca836a9f02009-09-01 12:22:52 +01001"""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
José Fonseca8edc6b02010-03-05 15:07:50 +000032import re
José Fonsecabf484472009-10-22 19:11:48 +010033import sys
José Fonseca8edc6b02010-03-05 15:07:50 +000034import distutils.version
José Fonseca836a9f02009-09-01 12:22:52 +010035
José Fonseca836a9f02009-09-01 12:22:52 +010036import SCons.Errors
37import SCons.Util
38
39
40def generate(env):
José Fonseca601498a2010-11-01 13:30:22 +000041 env['llvm'] = False
42
José Fonseca836a9f02009-09-01 12:22:52 +010043 try:
44 llvm_dir = os.environ['LLVM']
45 except KeyError:
46 # Do nothing -- use the system headers/libs
José Fonsecabf484472009-10-22 19:11:48 +010047 llvm_dir = None
José Fonseca836a9f02009-09-01 12:22:52 +010048 else:
49 if not os.path.isdir(llvm_dir):
50 raise SCons.Errors.InternalError, "Specified LLVM directory not found"
51
52 if env['debug']:
53 llvm_subdir = 'Debug'
54 else:
55 llvm_subdir = 'Release'
56
57 llvm_bin_dir = os.path.join(llvm_dir, llvm_subdir, 'bin')
58 if not os.path.isdir(llvm_bin_dir):
José Fonseca459ea002009-09-16 10:39:06 +010059 llvm_bin_dir = os.path.join(llvm_dir, 'bin')
60 if not os.path.isdir(llvm_bin_dir):
61 raise SCons.Errors.InternalError, "LLVM binary directory not found"
José Fonseca836a9f02009-09-01 12:22:52 +010062
63 env.PrependENVPath('PATH', llvm_bin_dir)
64
José Fonseca9e3728c2009-11-25 18:06:12 +000065 if env['platform'] == 'windows':
José Fonsecabf484472009-10-22 19:11:48 +010066 # XXX: There is no llvm-config on Windows, so assume a standard layout
José Fonseca8edc6b02010-03-05 15:07:50 +000067 if llvm_dir is None:
José Fonsecaea532f02010-04-10 02:41:39 +010068 print 'scons: LLVM environment variable must be specified when building for windows'
José Fonseca601498a2010-11-01 13:30:22 +000069 return
José Fonseca8edc6b02010-03-05 15:07:50 +000070
71 # Try to determine the LLVM version from llvm/Config/config.h
72 llvm_config = os.path.join(llvm_dir, 'include/llvm/Config/config.h')
73 if not os.path.exists(llvm_config):
74 print 'scons: could not find %s' % llvm_config
José Fonseca601498a2010-11-01 13:30:22 +000075 return
José Fonseca8edc6b02010-03-05 15:07:50 +000076 llvm_version_re = re.compile(r'^#define PACKAGE_VERSION "([^"]*)"')
77 llvm_version = None
78 for line in open(llvm_config, 'rt'):
79 mo = llvm_version_re.match(line)
80 if mo:
81 llvm_version = mo.group(1)
José Fonseca19a63332010-03-06 09:18:15 +000082 llvm_version = distutils.version.LooseVersion(llvm_version)
José Fonseca8edc6b02010-03-05 15:07:50 +000083 break
84 if llvm_version is None:
85 print 'scons: could not determine the LLVM version from %s' % llvm_config
José Fonseca601498a2010-11-01 13:30:22 +000086 return
José Fonseca8edc6b02010-03-05 15:07:50 +000087
88 env.Prepend(CPPPATH = [os.path.join(llvm_dir, 'include')])
89 env.AppendUnique(CPPDEFINES = [
90 '__STDC_LIMIT_MACROS',
91 '__STDC_CONSTANT_MACROS',
92 'HAVE_STDINT_H',
93 ])
94 env.Prepend(LIBPATH = [os.path.join(llvm_dir, 'lib')])
95 if llvm_version >= distutils.version.LooseVersion('2.7'):
96 # 2.7
José Fonsecabf484472009-10-22 19:11:48 +010097 env.Prepend(LIBS = [
José Fonseca8edc6b02010-03-05 15:07:50 +000098 'LLVMLinker', 'LLVMipo', 'LLVMInterpreter',
99 'LLVMInstrumentation', 'LLVMJIT', 'LLVMExecutionEngine',
100 'LLVMBitWriter', 'LLVMX86Disassembler', 'LLVMX86AsmParser',
101 'LLVMMCParser', 'LLVMX86AsmPrinter', 'LLVMX86CodeGen',
102 'LLVMSelectionDAG', 'LLVMX86Info', 'LLVMAsmPrinter',
103 'LLVMCodeGen', 'LLVMScalarOpts', 'LLVMInstCombine',
104 'LLVMTransformUtils', 'LLVMipa', 'LLVMAsmParser',
105 'LLVMArchive', 'LLVMBitReader', 'LLVMAnalysis', 'LLVMTarget',
106 'LLVMMC', 'LLVMCore', 'LLVMSupport', 'LLVMSystem',
José Fonsecabf484472009-10-22 19:11:48 +0100107 ])
José Fonseca8edc6b02010-03-05 15:07:50 +0000108 else:
109 # 2.6
110 env.Prepend(LIBS = [
111 'LLVMX86AsmParser', 'LLVMX86AsmPrinter', 'LLVMX86CodeGen',
112 'LLVMX86Info', 'LLVMLinker', 'LLVMipo', 'LLVMInterpreter',
113 'LLVMInstrumentation', 'LLVMJIT', 'LLVMExecutionEngine',
114 'LLVMDebugger', 'LLVMBitWriter', 'LLVMAsmParser',
115 'LLVMArchive', 'LLVMBitReader', 'LLVMSelectionDAG',
116 'LLVMAsmPrinter', 'LLVMCodeGen', 'LLVMScalarOpts',
117 'LLVMTransformUtils', 'LLVMipa', 'LLVMAnalysis',
118 'LLVMTarget', 'LLVMMC', 'LLVMCore', 'LLVMSupport',
119 'LLVMSystem',
120 ])
121 env.Append(LIBS = [
122 'imagehlp',
123 'psapi',
124 ])
125 if env['msvc']:
126 # Some of the LLVM C headers use the inline keyword without
127 # defining it.
128 env.Append(CPPDEFINES = [('inline', '__inline')])
José Fonsecae3a3a532010-09-29 14:24:52 +0100129 if env['build'] in ('debug', 'checked'):
José Fonseca8edc6b02010-03-05 15:07:50 +0000130 # LLVM libraries are static, build with /MT, and they
131 # automatically link agains LIBCMT. When we're doing a
132 # debug build we'll be linking against LIBCMTD, so disable
133 # that.
134 env.Append(LINKFLAGS = ['/nodefaultlib:LIBCMT'])
José Fonsecaea532f02010-04-10 02:41:39 +0100135 else:
136 if not env.Detect('llvm-config'):
137 print 'scons: llvm-config script not found' % llvm_version
José Fonseca601498a2010-11-01 13:30:22 +0000138 return
José Fonsecaea532f02010-04-10 02:41:39 +0100139
José Fonseca8edc6b02010-03-05 15:07:50 +0000140 llvm_version = env.backtick('llvm-config --version').rstrip()
José Fonseca19a63332010-03-06 09:18:15 +0000141 llvm_version = distutils.version.LooseVersion(llvm_version)
José Fonseca836a9f02009-09-01 12:22:52 +0100142
Vinson Lee79f48c92009-09-07 15:16:25 +0100143 try:
144 env.ParseConfig('llvm-config --cppflags')
145 env.ParseConfig('llvm-config --libs jit interpreter nativecodegen bitwriter')
146 env.ParseConfig('llvm-config --ldflags')
147 except OSError:
José Fonsecaea532f02010-04-10 02:41:39 +0100148 print 'scons: llvm-config version %s failed' % llvm_version
José Fonseca601498a2010-11-01 13:30:22 +0000149 return
José Fonseca8edc6b02010-03-05 15:07:50 +0000150
151 assert llvm_version is not None
José Fonseca601498a2010-11-01 13:30:22 +0000152 env['llvm'] = True
José Fonseca8edc6b02010-03-05 15:07:50 +0000153
154 print 'scons: Found LLVM version %s' % llvm_version
José Fonseca8edc6b02010-03-05 15:07:50 +0000155 env['LLVM_VERSION'] = llvm_version
156
157 # Define HAVE_LLVM macro with the major/minor version number (e.g., 0x0206 for 2.6)
158 llvm_version_major = int(llvm_version.version[0])
159 llvm_version_minor = int(llvm_version.version[1])
160 llvm_version_hex = '0x%02x%02x' % (llvm_version_major, llvm_version_minor)
161 env.Prepend(CPPDEFINES = [('HAVE_LLVM', llvm_version_hex)])
José Fonseca836a9f02009-09-01 12:22:52 +0100162
163def exists(env):
164 return True
165
166# vim:set ts=4 sw=4 et: