SConscript revision 13655
1# Copyright 2018 Google, Inc. 2# 3# Redistribution and use in source and binary forms, with or without 4# modification, are permitted provided that the following conditions are 5# met: redistributions of source code must retain the above copyright 6# notice, this list of conditions and the following disclaimer; 7# redistributions in binary form must reproduce the above copyright 8# notice, this list of conditions and the following disclaimer in the 9# documentation and/or other materials provided with the distribution; 10# neither the name of the copyright holders nor the names of its 11# contributors may be used to endorse or promote products derived from 12# this software without specific prior written permission. 13# 14# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 15# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 16# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 17# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 18# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 19# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 20# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 21# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 22# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 23# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 24# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 25# 26# Authors: Gabe Black 27 28from __future__ import print_function 29 30Import('*') 31 32if env['USE_SYSTEMC']: 33 34 from gem5_scons import Transform 35 36 import os.path 37 import json 38 39 src = str(Dir('.').srcdir) 40 41 class SystemCTest(object): 42 def __init__(self, dirname, name): 43 self.name = name 44 self.reldir = os.path.relpath(dirname, src) 45 self.target = os.path.join(self.reldir, name) 46 self.sources = [] 47 self.deps = [] 48 49 self.compile_only = False 50 51 def add_source(self, source): 52 self.sources.append(os.path.join(self.reldir, source)) 53 54 def add_sources(self, sources): 55 for source in sources: 56 self.sources.append(os.path.join(self.reldir, '..', source)) 57 58 def properties(self): 59 return { 60 'name' : self.name, 61 'path' : self.reldir, 62 'compile_only' : self.compile_only, 63 'deps' : self.deps 64 } 65 66 test_dir = Dir('.') 67 class SystemCTestBin(Executable): 68 def __init__(self, test): 69 super(SystemCTestBin, self).__init__(test.target, *test.sources) 70 self.reldir = test.reldir 71 self.test_deps = test.deps 72 73 @classmethod 74 def declare_all(cls, env): 75 env = env.Clone() 76 77 # Turn off extra warnings and Werror for the tests. 78 to_remove = ['-Wall', '-Wundef', '-Wextra', '-Werror'] 79 env['CCFLAGS'] = \ 80 filter(lambda f: f not in to_remove, env['CCFLAGS']) 81 82 env.Append(CPPPATH=test_dir.Dir('include')) 83 84 shared_lib_path = env['SHARED_LIB'][0].abspath 85 sl_dir, sl_base = os.path.split(shared_lib_path) 86 env.Append(LIBPATH=[sl_dir], LIBS=[sl_base]) 87 88 super(SystemCTestBin, cls).declare_all(env) 89 90 def declare(self, env): 91 env = env.Clone() 92 sources = list(self.sources) 93 for f in self.filters: 94 sources = Source.all.apply_filter(f) 95 objs = self.srcs_to_objs(env, sources) 96 objs = objs + env['MAIN_OBJS'] 97 relpath = os.path.relpath( 98 env['SHARED_LIB'][0].dir.abspath, 99 self.path(env).dir.abspath) 100 env.Append(LINKFLAGS=Split('-z origin')) 101 env.Append(RPATH=[ 102 env.Literal(os.path.join('\\$$ORIGIN', relpath))]) 103 test_bin = super(SystemCTestBin, self).declare(env, objs) 104 test_dir = self.dir.Dir(self.reldir) 105 for dep in self.test_deps: 106 env.Depends(test_bin, test_dir.File(dep)) 107 return test_bin 108 109 tests = [] 110 def new_test(dirname, name): 111 test = SystemCTest(dirname, name) 112 tests.append(test) 113 return test 114 115 116 def scan_dir_for_tests(subdir): 117 def visitor(arg, dirname, names): 118 # If there's a 'DONTRUN' file in this directory, skip it and any 119 # child directories. 120 if 'DONTRUN' in names: 121 del names[:] 122 return 123 124 endswith = lambda sfx: filter(lambda n: n.endswith(sfx), names) 125 126 cpps = endswith('.cpp') 127 if not cpps: 128 return 129 130 def get_entries(fname): 131 with open(os.path.join(dirname, fname)) as content: 132 lines = content.readlines 133 # Get rid of leading and trailing whitespace. 134 lines = map(lambda x: x.strip(), content.readlines()) 135 # Get rid of blank lines. 136 lines = filter(lambda x: x, lines) 137 return lines 138 139 # If there's only one source file, then that files name is the test 140 # name, and it's the source for that test. 141 if len(cpps) == 1: 142 cpp = cpps[0] 143 144 test = new_test(dirname, os.path.splitext(cpp)[0]) 145 test.add_source(cpp) 146 147 # Otherwise, expect there to be a file that ends in .f. That files 148 # name is the test name, and it will list the source files with 149 # one preceeding path component. 150 else: 151 fs = endswith('.f') 152 if len(fs) != 1: 153 print("In %s, expected 1 *.f file, but found %d.", 154 dirname, len(fs)) 155 for f in fs: 156 print(os.path.join(dirname, f)) 157 return 158 f = fs[0] 159 160 test = new_test(dirname, os.path.splitext(f)[0]) 161 # Add all the sources to this test. 162 test.add_sources(get_entries(f)) 163 164 if 'COMPILE' in names: 165 test.compile_only = True 166 167 if 'DEPS' in names: 168 test.deps = get_entries('DEPS') 169 170 subdir_src = Dir('.').srcdir.Dir(subdir) 171 os.path.walk(str(subdir_src), visitor, None) 172 173 scan_dir_for_tests('systemc') 174 scan_dir_for_tests('tlm') 175 176 177 def build_tests_json(target, source, env): 178 data = { test.target : test.properties() for test in tests } 179 with open(str(target[0]), "w") as tests_json: 180 json.dump(data, tests_json) 181 182 AlwaysBuild(env.Command(File('tests.json'), None, 183 MakeAction(build_tests_json, Transform("TESTJSON")))) 184 185 186 for test in tests: 187 SystemCTestBin(test) 188