NSIS/SCons/utils.py

116 lines
2.4 KiB
Python

def AddAvailableLibs(env, libs):
"""
Scans through a list of libraries and adds
available libraries to the environment.
"""
conf = env.Configure()
for lib in libs:
conf.CheckLib(lib)
conf.Finish()
def AddZLib(env, platform, alias='install-utils'):
"""
Checks for platform specific zlib and adds the
appropriate compiler and linker options to the environment
"""
zlib = 'z'
if platform == 'win32':
if 'ZLIB_W32' in env:
# Add include and library path of zlib for Win32
env.Append(CPPPATH = env['ZLIB_W32_INC'])
env.Append(LIBPATH = env['ZLIB_W32_LIB'])
zlib = ['zdll', 'z']
if 'ZLIB_W32_DLL' in env:
env.DistributeW32Bin(env['ZLIB_W32_DLL'], alias=alias)
else:
print 'Please specify folder of zlib for Win32 via ZLIB_W32'
Exit(1)
conf = env.Configure()
if not conf.CheckLibWithHeader(zlib, 'zlib.h', 'c'):
print 'zlib (%s) is missing!' % (platform)
Exit(1)
env = conf.Finish()
def GetAvailableLibs(env, libs):
"""
Scans through a list list of libraries and adds
available libraries to the environment.
"""
conf = env.Configure()
avail_libs = []
for lib in libs:
if conf.CheckLib(lib):
avail_libs.append(lib)
conf.Finish()
return avail_libs
def check_compile_flag(ctx, flag):
"""
Checks if a compiler flag is valid.
"""
ctx.Message('Checking for compiler flag %s... ' % flag)
old_flags = ctx.env['CCFLAGS']
ctx.env.Append(CCFLAGS = [flag])
test = """
int main() {
return 0;
}
"""
result = ctx.TryCompile(test, '.c')
ctx.Result(result)
if not result:
ctx.env.Replace(CCFLAGS = [old_flags])
return result
def check_link_flag(ctx, flag, run = 0, extension = '.c', code = None):
"""
Checks if a linker flag is valid.
"""
ctx.Message('Checking for linker flag %s... ' % flag)
old_flags = ctx.env['LINKFLAGS']
ctx.env.Append(LINKFLAGS = [flag])
if code:
test = code
else:
test = """
int main() {
return 0;
}
"""
result = ctx.TryLink(test, extension)
if run:
result = result and ctx.TryRun(test, extension)[0]
ctx.Result(result)
if not result:
ctx.env.Replace(LINKFLAGS = [old_flags])
return result
def FlagsConfigure(env):
"""
Wrapper for env.Configure which adds two new tests:
CheckCompileFlag - checks for a compiler flag
CheckLinkFlag - checks for a linker flag
"""
return env.Configure(custom_tests = { 'CheckCompileFlag' : check_compile_flag, 'CheckLinkFlag': check_link_flag })
Export('AddAvailableLibs AddZLib FlagsConfigure GetAvailableLibs')