1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
import subprocess
import os
import sys
import re
indent = " "
def escape_special_symbols(strval):
retval = ""
for c in strval:
if c in ("\\", "\""):
retval += "\\" + c
elif ord(c) < 0x20:
retval += c.encode("string_escape")
else:
retval += c
return retval
def escape_line_feed(strval):
return re.sub(r'\\n', r'\\n"\\\n' + indent + '"', strval)
def escaped_define(strkey, strval):
return "#define " + strkey + " \"" + escape_line_feed(escape_special_symbols(strval)) + "\""
def get_build_info(compiler, flags):
build_info = "Build info:\n"
build_info += indent + "Compiler: " + compiler + "\n"
build_info += indent + "Compiler version: \n" + get_compiler_info(compiler) + "\n"
build_info += indent + "Compile flags: " + (flags if flags else "no flags info")
return build_info
def get_compiler_info(compiler):
compiler_binary = os.path.basename(compiler).lower()
if len(compiler.split(' ')) > 1 or compiler_binary == "ymake" or compiler_binary == "ymake.exe":
compiler_ver_out = "Build by wrapper. No useful info here."
else:
compiler_ver_cmd = [compiler]
if compiler_binary not in ("cl", "cl.exe"):
compiler_ver_cmd.append('--version')
compiler_ver_out = subprocess.Popen(compiler_ver_cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT).stdout.read()
return "\n".join(['{}{}'.format(indent * 2, line.strip()) for line in compiler_ver_out.splitlines() if line.strip()])
def main():
if len(sys.argv) != 4:
print >>sys.stderr, "Usage: build_info_gen.py <output file> <CXX compiler> <CXX flags>"
sys.exit(1)
cxx_compiler = sys.argv[2]
cxx_flags = sys.argv[3]
with open(sys.argv[1], 'w') as result:
print >> result, "#pragma once\n"
print >> result, escaped_define("BUILD_INFO", get_build_info(cxx_compiler, cxx_flags))
print >> result, escaped_define("BUILD_COMPILER", cxx_compiler)
print >> result, escaped_define("BUILD_COMPILER_VERSION", get_compiler_info(cxx_compiler))
print >> result, escaped_define("BUILD_COMPILER_FLAGS", cxx_flags)
if __name__ == "__main__":
main()
|
