Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
185 lines (159 sloc) 4.77 KB
import os
import platform
import re
import subprocess
def subprocess_output(cmdline):
p = subprocess.Popen(cmdline.split(" "), stdout=subprocess.PIPE)
stdout, stderr = p.communicate()
stdout = stdout.decode('utf-8')
return stdout.rstrip()
def read_whole_file(path):
f = open(path)
r = f.read().rstrip()
f.close()
return r
def describe_revision():
if os.path.isdir('.git'):
#
# I'd like to use Dulwich or Git-Python to do this, but
# I concluded that it's highly unlikely that people would
# have the git repo checked out, but not have git.
#
hash = subprocess_output("git rev-list HEAD~1..HEAD")
tag = subprocess_output("git describe --abbrev=0")
revs = subprocess_output("git rev-list %s..HEAD" % (tag,)).split('\n')
if len(revs) and revs[0] == '':
revs = []
n = len(revs)
return ('git', tag, n, hash)
elif os.path.isdir('.hg'):
#
# I considered using the Mercurial API directly, but the
# Mercurial wiki highly discourages it.
#
hash = subprocess_output("hg parent --template {node}")
tag = subprocess_output("hg parent --template {latesttag}")
n = int(subprocess_output("hg parent --template {latesttagdistance}"))
return ('hg', tag, n, hash)
else:
tag = read_whole_file('tools/release_ver')
n = 0
hash = 'no-hash'
return ('rel', tag, n, hash)
def generate_build_header(**kwargs):
env = kwargs['env']
targets = kwargs['target']
source = kwargs['source']
scm, tag, n, hash = describe_revision()
extra = tag.split('-')
version = list(map(int, extra[0].split('.')))
while len(version) < 3:
version.append(0)
if n < 1:
long = "%s" % (tag)
else:
long = "%s+%d-%s-%s" % (tag, n, scm, hash[0:8])
lines = [
"#ifndef __included_cpuid_build_h\n",
"#define __included_cpuid_build_h\n",
"\n",
"#define CPUID_VERSION_MAJOR %d\n" % version[0],
"#define CPUID_VERSION_MINOR %d\n" % version[1],
"#define CPUID_VERSION_REVISION %d\n" % version[2],
"#define CPUID_VERSION_BUILD %d\n" % n,
"#define CPUID_VERSION_TAG \"%s\"\n" % tag,
"#define CPUID_VERSION_LONG \"%s\"\n" % long,
"\n",
"#define CPUID_RESOURCE_VERSION %d,%d,%d,%d\n" % (version[0], version[1], version[2], n),
"#define CPUID_RESOURCE_VERSION_STRING \"%d, %d, %d, %d\"\n" % (version[0], version[1], version[2], n),
"\n",
"#endif\n"
]
for target in targets:
f = open(str(target), "wt")
f.writelines(lines)
f.close()
return 0
def generate_license_header(**kwargs):
env = kwargs['env']
targets = kwargs['target']
source = kwargs['source']
source = str(source[0])
source = open(source, 'rb')
license = source.read().decode('utf-8')
source.close()
license = re.sub('\t', r'\\t', license)
license = re.sub('\n', r'\\n', license)
license = re.sub('"', r'\\"', license)
lines = [
"#ifndef __included_cpuid_license_h\n",
"#define __included_cpuid_license_h\n",
"\n",
"#define CPUID_LICENSE \"%s\"\n" % (license),
"\n",
"#endif\n"
]
for target in targets:
f = open(str(target), 'wt')
f.writelines(lines)
f.close()
env = Environment(
tools=['default'],
ENV=os.environ,
**ARGUMENTS)
env.Decider('MD5-timestamp')
debug = ARGUMENTS.get('debug', 0)
if env['CC'] == 'gcc' or env['CC'] == 'clang':
# Optimization levels
if int(debug):
env.Append(CFLAGS='-O0 -ggdb')
else:
env.Append(CFLAGS='-Os')
# Basic CFLAGS for correctness
env.Append(CFLAGS='-std=gnu89 -fno-strict-aliasing')
# Standard search path
env.Append(CFLAGS='-I.')
# Warning flags
env.Append(CFLAGS='-Wall -Wextra -Wdeclaration-after-statement -Wimplicit-function-declaration -Wmissing-declarations -Wmissing-prototypes -Wno-long-long -Wno-overlength-strings -Wold-style-definition -Wstrict-prototypes')
# pthreads
env.Append(CFLAGS='-pthread')
env.Append(LINKFLAGS='-pthread')
# needed for sqrt
env.Append(LIBS=['-lm'])
if platform.system() == 'Linux':
# needed on older glibc for clock_gettime
env.Append(LIBS=['-lrt'])
elif env['CC'] == 'cl':
# Optimization levels
if int(debug):
env.Append(CFLAGS=['/Od', '/MTd'])
else:
env.Append(CFLAGS=['/O2', '/MT'])
# Standard search path
env.Append(CFLAGS=['/I.', '/Imsvc'])
# For high resolution timers
env.Append(LINKFLAGS='winmm.lib')
sources = [
'cache.c',
'clock.c',
'cpuid.c',
'feature.c',
'handlers.c',
'main.c',
'sanity.c',
'threads.c',
'util.c',
'version.c',
]
if not env.GetOption('clean'):
conf = Configure(env)
if not conf.CheckHeader('getopt.h') or not conf.CheckFunc('getopt_long'):
conf.env.Append(CFLAGS='-Igetopt')
sources.append('getopt/getopt_long.c')
env = conf.Finish()
else:
sources.append('getopt/getopt_long.c')
env.Command('build.h', [Value(describe_revision())], env.Action(generate_build_header, 'Generating build.h'))
env.Command('license.h', '#COPYING', env.Action(generate_license_header, 'Generating license.h'))
env.Program(target='cpuid', source=sources)
# vim: set ts=4 sts=4 sw=4 noet:
You can’t perform that action at this time.