building.py 30.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
#
# File      : building.py
# This file is part of RT-Thread RTOS
# COPYRIGHT (C) 2006 - 2015, RT-Thread Development Team
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License along
#  with this program; if not, write to the Free Software Foundation, Inc.,
#  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
# Change Logs:
# Date           Author       Notes
# 2015-01-20     Bernard      Add copyright information
23
# 2015-07-25     Bernard      Add LOCAL_CCFLAGS/LOCAL_CPPPATH/LOCAL_CPPDEFINES for
24
#                             group definition.
25 26
#

G
goprife@gmail.com 已提交
27 28 29
import os
import sys
import string
30
import utils
G
goprife@gmail.com 已提交
31 32

from SCons.Script import *
33
from utils import _make_path_relative
34
from mkdist import do_copy_file
G
goprife@gmail.com 已提交
35 36 37 38 39 40

BuildOptions = {}
Projects = []
Rtt_Root = ''
Env = None

41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
# SCons PreProcessor patch
def start_handling_includes(self, t=None):
    """
    Causes the PreProcessor object to start processing #import,
    #include and #include_next lines.

    This method will be called when a #if, #ifdef, #ifndef or #elif
    evaluates True, or when we reach the #else in a #if, #ifdef,
    #ifndef or #elif block where a condition already evaluated
    False.

    """
    d = self.dispatch_table
    p = self.stack[-1] if self.stack else self.default_table

    for k in ('import', 'include', 'include_next', 'define'):
        d[k] = p[k]

def stop_handling_includes(self, t=None):
    """
    Causes the PreProcessor object to stop processing #import,
    #include and #include_next lines.

    This method will be called when a #if, #ifdef, #ifndef or #elif
    evaluates False, or when we reach the #else in a #if, #ifdef,
    #ifndef or #elif block where a condition already evaluated True.
    """
    d = self.dispatch_table
    d['import'] = self.do_nothing
    d['include'] =  self.do_nothing
    d['include_next'] =  self.do_nothing
    d['define'] =  self.do_nothing
73

74 75 76 77
PatchedPreProcessor = SCons.cpp.PreProcessor
PatchedPreProcessor.start_handling_includes = start_handling_includes
PatchedPreProcessor.stop_handling_includes = stop_handling_includes

G
goprife@gmail.com 已提交
78 79
class Win32Spawn:
    def spawn(self, sh, escape, cmd, args, env):
G
Grissiom 已提交
80 81 82 83 84 85 86
        # deal with the cmd build-in commands which cannot be used in
        # subprocess.Popen
        if cmd == 'del':
            for f in args[1:]:
                try:
                    os.remove(f)
                except Exception as e:
87
                    print ('Error removing file: ' + e)
G
Grissiom 已提交
88 89 90
                    return -1
            return 0

G
goprife@gmail.com 已提交
91 92
        import subprocess

93
        newargs = ' '.join(args[1:])
G
goprife@gmail.com 已提交
94
        cmdline = cmd + " " + newargs
G
Grissiom 已提交
95 96

        # Make sure the env is constructed by strings
97
        _e = dict([(k, str(v)) for k, v in env.items()])
G
Grissiom 已提交
98 99 100 101 102 103

        # Windows(tm) CreateProcess does not use the env passed to it to find
        # the executables. So we have to modify our own PATH to make Popen
        # work.
        old_path = os.environ['PATH']
        os.environ['PATH'] = _e['PATH']
104

G
Grissiom 已提交
105
        try:
106
            proc = subprocess.Popen(cmdline, env=_e, shell=False)
G
Grissiom 已提交
107
        except Exception as e:
108 109
            print ('Error in calling:\n' + cmdline)
            print ('Exception: ' + e + ': ' + os.strerror(e.errno))
110
            return e.errno
G
Grissiom 已提交
111 112
        finally:
            os.environ['PATH'] = old_path
113

G
Grissiom 已提交
114
        return proc.wait()
G
goprife@gmail.com 已提交
115

116 117
# generate cconfig.h file
def GenCconfigFile(env, BuildOptions):
118
    import rtconfig
119

120 121 122 123 124
    if rtconfig.PLATFORM == 'gcc':
        contents = ''
        if not os.path.isfile('cconfig.h'):
            import gcc
            gcc.GenerateGCCConfig(rtconfig)
125

126 127
        # try again
        if os.path.isfile('cconfig.h'):
128
            f = open('cconfig.h', 'r')
129 130 131
            if f:
                contents = f.read()
                f.close();
132

133 134 135
                prep = PatchedPreProcessor()
                prep.process_contents(contents)
                options = prep.cpp_namespace
136

137 138 139 140
                BuildOptions.update(options)

                # add HAVE_CCONFIG_H definition
                env.AppendUnique(CPPDEFINES = ['HAVE_CCONFIG_H'])
141

142
def PrepareBuilding(env, root_directory, has_libcpu=False, remove_components = []):
G
goprife@gmail.com 已提交
143 144 145 146 147 148 149
    import rtconfig

    global BuildOptions
    global Projects
    global Env
    global Rtt_Root

150 151 152 153 154 155
    # ===== Add option to SCons =====
    AddOption('--dist',
                      dest = 'make-dist',
                      action = 'store_true',
                      default = False,
                      help = 'make distribution')
156 157 158 159 160
    AddOption('--dist-strip',
                      dest = 'make-dist-strip',
                      action = 'store_true',
                      default = False,
                      help = 'make distribution and strip useless files')
161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186
    AddOption('--cscope',
                      dest = 'cscope',
                      action = 'store_true',
                      default = False,
                      help = 'Build Cscope cross reference database. Requires cscope installed.')
    AddOption('--clang-analyzer',
                      dest = 'clang-analyzer',
                      action = 'store_true',
                      default = False,
                      help = 'Perform static analyze with Clang-analyzer. ' + \
                           'Requires Clang installed.\n' + \
                           'It is recommended to use with scan-build like this:\n' + \
                           '`scan-build scons --clang-analyzer`\n' + \
                           'If things goes well, scan-build will instruct you to invoke scan-view.')
    AddOption('--buildlib',
                      dest = 'buildlib',
                      type = 'string',
                      help = 'building library of a component')
    AddOption('--cleanlib',
                      dest = 'cleanlib',
                      action = 'store_true',
                      default = False,
                      help = 'clean up the library by --buildlib')
    AddOption('--target',
                      dest = 'target',
                      type = 'string',
187
                      help = 'set target project: mdk/mdk4/mdk5/iar/vs/vsc/ua/cdk/ses')
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202
    AddOption('--genconfig',
                dest = 'genconfig',
                action = 'store_true',
                default = False,
                help = 'Generate .config from rtconfig.h')
    AddOption('--useconfig',
                dest = 'useconfig',
                type = 'string',
                help = 'make rtconfig.h from config file.')
    AddOption('--verbose',
                dest = 'verbose',
                action = 'store_true',
                default = False,
                help = 'print verbose information during build')

G
goprife@gmail.com 已提交
203
    Env = env
204
    Rtt_Root = os.path.abspath(root_directory)
B
Bernard Xiong 已提交
205 206 207 208 209

    # make an absolute root directory
    RTT_ROOT = Rtt_Root
    Export('RTT_ROOT')

210 211 212 213 214
    # set RTT_ROOT in ENV
    Env['RTT_ROOT'] = Rtt_Root
    # set BSP_ROOT in ENV
    Env['BSP_ROOT'] = Dir('#').abspath

215
    sys.path = sys.path + [os.path.join(Rtt_Root, 'tools')]
G
goprife@gmail.com 已提交
216

217 218 219 220 221 222 223 224 225 226
    # {target_name:(CROSS_TOOL, PLATFORM)}
    tgt_dict = {'mdk':('keil', 'armcc'),
                'mdk4':('keil', 'armcc'),
                'mdk5':('keil', 'armcc'),
                'iar':('iar', 'iar'),
                'vs':('msvc', 'cl'),
                'vs2012':('msvc', 'cl'),
                'vsc' : ('gcc', 'gcc'),
                'cb':('keil', 'armcc'),
                'ua':('gcc', 'gcc'),
227 228
                'cdk':('gcc', 'gcc'),
                'ses' : ('gcc', 'gcc')}
229 230 231 232 233 234
    tgt_name = GetOption('target')

    if tgt_name:
        # --target will change the toolchain settings which clang-analyzer is
        # depend on
        if GetOption('clang-analyzer'):
235
            print ('--clang-analyzer cannot be used with --target')
236 237 238 239 240 241 242
            sys.exit(1)

        SetOption('no_exec', 1)
        try:
            rtconfig.CROSS_TOOL, rtconfig.PLATFORM = tgt_dict[tgt_name]
            # replace the 'RTT_CC' to 'CROSS_TOOL'
            os.environ['RTT_CC'] = rtconfig.CROSS_TOOL
243
            utils.ReloadModule(rtconfig)
244
        except KeyError:
245
            print ('Unknow target: '+ tgt_name+'. Avaible targets: ' +', '.join(tgt_dict.keys()))
246 247 248 249 250 251 252
            sys.exit(1)
    elif (GetDepend('RT_USING_NEWLIB') == False and GetDepend('RT_USING_NOLIBC') == False) \
        and rtconfig.PLATFORM == 'gcc':
        AddDepend('RT_USING_MINILIBC')

    # auto change the 'RTT_EXEC_PATH' when 'rtconfig.EXEC_PATH' get failed
    if not os.path.exists(rtconfig.EXEC_PATH):
253
        if 'RTT_EXEC_PATH' in os.environ:
254 255
            # del the 'RTT_EXEC_PATH' and using the 'EXEC_PATH' setting on rtconfig.py
            del os.environ['RTT_EXEC_PATH']
256
            utils.ReloadModule(rtconfig)
257

258 259 260 261 262
    # add compability with Keil MDK 4.6 which changes the directory of armcc.exe
    if rtconfig.PLATFORM == 'armcc':
        if not os.path.isfile(os.path.join(rtconfig.EXEC_PATH, 'armcc.exe')):
            if rtconfig.EXEC_PATH.find('bin40') > 0:
                rtconfig.EXEC_PATH = rtconfig.EXEC_PATH.replace('bin40', 'armcc/bin')
263
                Env['LINKFLAGS'] = Env['LINKFLAGS'].replace('RV31', 'armcc')
264

B
Bright Pan 已提交
265
        # reset AR command flags
B
bernard 已提交
266
        env['ARCOM'] = '$AR --create $TARGET $SOURCES'
267 268
        env['LIBPREFIX'] = ''
        env['LIBSUFFIX'] = '.lib'
B
bernard 已提交
269
        env['LIBLINKPREFIX'] = ''
270
        env['LIBLINKSUFFIX'] = '.lib'
B
bernard 已提交
271
        env['LIBDIRPREFIX'] = '--userlibpath '
B
bernard 已提交
272

G
goprife@gmail.com 已提交
273
    # patch for win32 spawn
G
Grissiom 已提交
274
    if env['PLATFORM'] == 'win32':
G
goprife@gmail.com 已提交
275 276 277
        win32_spawn = Win32Spawn()
        win32_spawn.env = env
        env['SPAWN'] = win32_spawn.spawn
G
Grissiom 已提交
278

279
    if env['PLATFORM'] == 'win32':
280 281 282
        os.environ['PATH'] = rtconfig.EXEC_PATH + ";" + os.environ['PATH']
    else:
        os.environ['PATH'] = rtconfig.EXEC_PATH + ":" + os.environ['PATH']
G
goprife@gmail.com 已提交
283 284 285

    # add program path
    env.PrependENVPath('PATH', rtconfig.EXEC_PATH)
286 287
    # add rtconfig.h path
    env.Append(CPPPATH = [str(Dir('#').abspath)])
G
goprife@gmail.com 已提交
288

B
bernard 已提交
289 290 291 292 293
    # add library build action
    act = SCons.Action.Action(BuildLibInstallAction, 'Install compiled library... $TARGET')
    bld = Builder(action = act)
    Env.Append(BUILDERS = {'BuildLib': bld})

G
goprife@gmail.com 已提交
294
    # parse rtconfig.h to get used component
295
    PreProcessor = PatchedPreProcessor()
296
    f = open('rtconfig.h', 'r')
G
goprife@gmail.com 已提交
297 298 299 300 301
    contents = f.read()
    f.close()
    PreProcessor.process_contents(contents)
    BuildOptions = PreProcessor.cpp_namespace

302 303 304 305 306 307 308 309 310 311
    if GetOption('clang-analyzer'):
        # perform what scan-build does
        env.Replace(
                CC   = 'ccc-analyzer',
                CXX  = 'c++-analyzer',
                # skip as and link
                LINK = 'true',
                AS   = 'true',)
        env["ENV"].update(x for x in os.environ.items() if x[0].startswith("CCC_"))
        # only check, don't compile. ccc-analyzer use CCC_CC as the CC.
312 313 314 315 316
        # fsyntax-only will give us some additional warning messages
        env['ENV']['CCC_CC']  = 'clang'
        env.Append(CFLAGS=['-fsyntax-only', '-Wall', '-Wno-invalid-source-encoding'])
        env['ENV']['CCC_CXX'] = 'clang++'
        env.Append(CXXFLAGS=['-fsyntax-only', '-Wall', '-Wno-invalid-source-encoding'])
317 318 319
        # remove the POST_ACTION as it will cause meaningless errors(file not
        # found or something like that).
        rtconfig.POST_ACTION = ''
320

321 322
    # generate cconfig.h file
    GenCconfigFile(env, BuildOptions)
G
goprife@gmail.com 已提交
323

324 325 326
    # auto append '_REENT_SMALL' when using newlib 'nano.specs' option
    if rtconfig.PLATFORM == 'gcc' and str(env['LINKFLAGS']).find('nano.specs') != -1:
        env.AppendUnique(CPPDEFINES = ['_REENT_SMALL'])
327

B
bernard 已提交
328 329 330 331 332
    if GetOption('genconfig'):
        from genconf import genconfig
        genconfig()
        exit(0)

B
Bernard Xiong 已提交
333
    if env['PLATFORM'] != 'win32':
334
        AddOption('--menuconfig',
B
Bernard Xiong 已提交
335 336 337 338 339 340 341 342 343
                    dest = 'menuconfig',
                    action = 'store_true',
                    default = False,
                    help = 'make menuconfig for RT-Thread BSP')
        if GetOption('menuconfig'):
            from menuconfig import menuconfig
            menuconfig(Rtt_Root)
            exit(0)

344 345 346 347 348 349 350 351 352 353
    AddOption('--pyconfig',
                dest = 'pyconfig',
                action = 'store_true',
                default = False,
                help = 'make menuconfig for RT-Thread BSP')
    if GetOption('pyconfig'):
        from menuconfig import pyconfig
        pyconfig(Rtt_Root)
        exit(0)

B
bernard 已提交
354 355 356 357 358 359
    configfn = GetOption('useconfig')
    if configfn:
        from menuconfig import mk_rtconfig
        mk_rtconfig(configfn)
        exit(0)

360

361 362
    if not GetOption('verbose'):
        # override the default verbose command string
363
        env.Replace(
R
Rogerz Zhang 已提交
364
            ARCOMSTR = 'AR $TARGET',
365
            ASCOMSTR = 'AS $TARGET',
R
Rogerz Zhang 已提交
366
            ASPPCOMSTR = 'AS $TARGET',
367 368 369 370
            CCCOMSTR = 'CC $TARGET',
            CXXCOMSTR = 'CXX $TARGET',
            LINKCOMSTR = 'LINK $TARGET'
        )
G
goprife@gmail.com 已提交
371

372 373 374 375 376
    # fix the linker for C++
    if GetDepend('RT_USING_CPLUSPLUS'):
        if env['LINK'].find('gcc') != -1:
            env['LINK'] = env['LINK'].replace('gcc', 'g++')

377 378 379
    # we need to seperate the variant_dir for BSPs and the kernels. BSPs could
    # have their own components etc. If they point to the same folder, SCons
    # would find the wrong source code to compile.
B
Bernard Xiong 已提交
380
    bsp_vdir = 'build'
381 382 383
    kernel_vdir = 'build/kernel'
    # board build script
    objs = SConscript('SConscript', variant_dir=bsp_vdir, duplicate=0)
G
goprife@gmail.com 已提交
384
    # include kernel
385
    objs.extend(SConscript(Rtt_Root + '/src/SConscript', variant_dir=kernel_vdir + '/src', duplicate=0))
G
goprife@gmail.com 已提交
386 387
    # include libcpu
    if not has_libcpu:
388 389
        objs.extend(SConscript(Rtt_Root + '/libcpu/SConscript',
                    variant_dir=kernel_vdir + '/libcpu', duplicate=0))
390

G
goprife@gmail.com 已提交
391
    # include components
392
    objs.extend(SConscript(Rtt_Root + '/components/SConscript',
393
                           variant_dir=kernel_vdir + '/components',
394 395
                           duplicate=0,
                           exports='remove_components'))
G
goprife@gmail.com 已提交
396 397 398

    return objs

B
Bernard Xiong 已提交
399
def PrepareModuleBuilding(env, root_directory, bsp_directory):
G
goprife@gmail.com 已提交
400 401
    import rtconfig

402
    global BuildOptions
G
goprife@gmail.com 已提交
403 404 405
    global Env
    global Rtt_Root

406 407 408 409 410 411
    # patch for win32 spawn
    if env['PLATFORM'] == 'win32':
        win32_spawn = Win32Spawn()
        win32_spawn.env = env
        env['SPAWN'] = win32_spawn.spawn

G
goprife@gmail.com 已提交
412 413 414
    Env = env
    Rtt_Root = root_directory

B
Bernard Xiong 已提交
415
    # parse bsp rtconfig.h to get used component
416
    PreProcessor = PatchedPreProcessor()
417
    f = open(bsp_directory + '/rtconfig.h', 'r')
B
Bernard Xiong 已提交
418 419 420 421 422
    contents = f.read()
    f.close()
    PreProcessor.process_contents(contents)
    BuildOptions = PreProcessor.cpp_namespace

B
Bright Pan 已提交
423 424 425
    # add build/clean library option for library checking
    AddOption('--buildlib',
              dest='buildlib',
B
bernard 已提交
426 427
              type='string',
              help='building library of a component')
B
Bright Pan 已提交
428 429
    AddOption('--cleanlib',
              dest='cleanlib',
B
bernard 已提交
430 431 432 433
              action='store_true',
              default=False,
              help='clean up the library by --buildlib')

G
goprife@gmail.com 已提交
434 435 436
    # add program path
    env.PrependENVPath('PATH', rtconfig.EXEC_PATH)

wuyangyong's avatar
wuyangyong 已提交
437 438 439 440 441 442 443
def GetConfigValue(name):
    assert type(name) == str, 'GetConfigValue: only string parameter is valid'
    try:
        return BuildOptions[name]
    except:
        return ''

G
goprife@gmail.com 已提交
444 445 446
def GetDepend(depend):
    building = True
    if type(depend) == type('str'):
447
        if not depend in BuildOptions or BuildOptions[depend] == 0:
G
goprife@gmail.com 已提交
448 449 450
            building = False
        elif BuildOptions[depend] != '':
            return BuildOptions[depend]
B
Bright Pan 已提交
451

G
goprife@gmail.com 已提交
452 453 454 455 456
        return building

    # for list type depend
    for item in depend:
        if item != '':
457
            if not item in BuildOptions or BuildOptions[item] == 0:
G
goprife@gmail.com 已提交
458 459 460 461
                building = False

    return building

B
Bernard Xiong 已提交
462 463 464 465 466 467
def LocalOptions(config_filename):
    from SCons.Script import SCons

    # parse wiced_config.h to get used component
    PreProcessor = SCons.cpp.PreProcessor()

468
    f = open(config_filename, 'r')
B
Bernard Xiong 已提交
469 470 471 472 473 474 475 476 477 478 479
    contents = f.read()
    f.close()

    PreProcessor.process_contents(contents)
    local_options = PreProcessor.cpp_namespace

    return local_options

def GetLocalDepend(options, depend):
    building = True
    if type(depend) == type('str'):
480
        if not depend in options or options[depend] == 0:
B
Bernard Xiong 已提交
481 482 483 484 485 486 487 488 489
            building = False
        elif options[depend] != '':
            return options[depend]

        return building

    # for list type depend
    for item in depend:
        if item != '':
490
            if not item in options or options[item] == 0:
B
Bernard Xiong 已提交
491 492 493 494
                building = False

    return building

G
goprife@gmail.com 已提交
495 496 497
def AddDepend(option):
    BuildOptions[option] = 1

498 499
def MergeGroup(src_group, group):
    src_group['src'] = src_group['src'] + group['src']
500 501
    if 'CCFLAGS' in group:
        if 'CCFLAGS' in src_group:
502 503 504
            src_group['CCFLAGS'] = src_group['CCFLAGS'] + group['CCFLAGS']
        else:
            src_group['CCFLAGS'] = group['CCFLAGS']
505 506
    if 'CPPPATH' in group:
        if 'CPPPATH' in src_group:
507 508 509
            src_group['CPPPATH'] = src_group['CPPPATH'] + group['CPPPATH']
        else:
            src_group['CPPPATH'] = group['CPPPATH']
510 511
    if 'CPPDEFINES' in group:
        if 'CPPDEFINES' in src_group:
512 513 514
            src_group['CPPDEFINES'] = src_group['CPPDEFINES'] + group['CPPDEFINES']
        else:
            src_group['CPPDEFINES'] = group['CPPDEFINES']
515 516
    if 'ASFLAGS' in group:
        if 'ASFLAGS' in src_group:
517 518 519
            src_group['ASFLAGS'] = src_group['ASFLAGS'] + group['ASFLAGS']
        else:
            src_group['ASFLAGS'] = group['ASFLAGS']
520 521

    # for local CCFLAGS/CPPPATH/CPPDEFINES
522 523
    if 'LOCAL_CCFLAGS' in group:
        if 'LOCAL_CCFLAGS' in src_group:
524 525 526
            src_group['LOCAL_CCFLAGS'] = src_group['LOCAL_CCFLAGS'] + group['LOCAL_CCFLAGS']
        else:
            src_group['LOCAL_CCFLAGS'] = group['LOCAL_CCFLAGS']
527 528
    if 'LOCAL_CPPPATH' in group:
        if 'LOCAL_CPPPATH' in src_group:
529 530 531
            src_group['LOCAL_CPPPATH'] = src_group['LOCAL_CPPPATH'] + group['LOCAL_CPPPATH']
        else:
            src_group['LOCAL_CPPPATH'] = group['LOCAL_CPPPATH']
532 533
    if 'LOCAL_CPPDEFINES' in group:
        if 'LOCAL_CPPDEFINES' in src_group:
534 535 536 537
            src_group['LOCAL_CPPDEFINES'] = src_group['LOCAL_CPPDEFINES'] + group['LOCAL_CPPDEFINES']
        else:
            src_group['LOCAL_CPPDEFINES'] = group['LOCAL_CPPDEFINES']

538 539
    if 'LINKFLAGS' in group:
        if 'LINKFLAGS' in src_group:
540 541 542
            src_group['LINKFLAGS'] = src_group['LINKFLAGS'] + group['LINKFLAGS']
        else:
            src_group['LINKFLAGS'] = group['LINKFLAGS']
543 544
    if 'LIBS' in group:
        if 'LIBS' in src_group:
545
            src_group['LIBS'] = src_group['LIBS'] + group['LIBS']
546
        else:
547
            src_group['LIBS'] = group['LIBS']
548 549
    if 'LIBPATH' in group:
        if 'LIBPATH' in src_group:
550 551 552
            src_group['LIBPATH'] = src_group['LIBPATH'] + group['LIBPATH']
        else:
            src_group['LIBPATH'] = group['LIBPATH']
553 554
    if 'LOCAL_ASFLAGS' in group:
        if 'LOCAL_ASFLAGS' in src_group:
555 556 557
            src_group['LOCAL_ASFLAGS'] = src_group['LOCAL_ASFLAGS'] + group['LOCAL_ASFLAGS']
        else:
            src_group['LOCAL_ASFLAGS'] = group['LOCAL_ASFLAGS']
558

G
goprife@gmail.com 已提交
559 560 561 562 563
def DefineGroup(name, src, depend, **parameters):
    global Env
    if not GetDepend(depend):
        return []

564 565 566 567 568 569 570 571
    # find exist group and get path of group
    group_path = ''
    for g in Projects:
        if g['name'] == name:
            group_path = g['path']
    if group_path == '':
        group_path = GetCurrentDir()

G
goprife@gmail.com 已提交
572 573
    group = parameters
    group['name'] = name
574
    group['path'] = group_path
575
    if type(src) == type([]):
G
goprife@gmail.com 已提交
576 577 578 579
        group['src'] = File(src)
    else:
        group['src'] = src

580
    if 'CCFLAGS' in group:
581
        Env.AppendUnique(CCFLAGS = group['CCFLAGS'])
582
    if 'CPPPATH' in group:
583 584 585 586
        paths = []
        for item in group['CPPPATH']:
            paths.append(os.path.abspath(item))
        group['CPPPATH'] = paths
587
        Env.AppendUnique(CPPPATH = group['CPPPATH'])
588
    if 'CPPDEFINES' in group:
589
        Env.AppendUnique(CPPDEFINES = group['CPPDEFINES'])
590
    if 'LINKFLAGS' in group:
591
        Env.AppendUnique(LINKFLAGS = group['LINKFLAGS'])
592
    if 'ASFLAGS' in group:
593
        Env.AppendUnique(ASFLAGS = group['ASFLAGS'])
594 595 596 597 598 599 600 601 602 603 604 605
    if 'LOCAL_CPPPATH' in group:
        paths = []
        for item in group['LOCAL_CPPPATH']:
            paths.append(os.path.abspath(item))
        group['LOCAL_CPPPATH'] = paths

    import rtconfig
    if rtconfig.PLATFORM == 'gcc':
        if 'CCFLAGS' in group:
            group['CCFLAGS'] = utils.GCCC99Patch(group['CCFLAGS'])
        if 'LOCAL_CCFLAGS' in group:
            group['LOCAL_CCFLAGS'] = utils.GCCC99Patch(group['LOCAL_CCFLAGS'])
B
bernard 已提交
606

B
Bright Pan 已提交
607
    # check whether to clean up library
B
bernard 已提交
608
    if GetOption('cleanlib') and os.path.exists(os.path.join(group['path'], GroupLibFullName(name, Env))):
B
bernard 已提交
609
        if group['src'] != []:
610
            print ('Remove library:'+ GroupLibFullName(name, Env))
B
bernard 已提交
611 612 613
            fn = os.path.join(group['path'], GroupLibFullName(name, Env))
            if os.path.exists(fn):
                os.unlink(fn)
B
bernard 已提交
614

615
    if 'LIBS' in group:
616
        Env.AppendUnique(LIBS = group['LIBS'])
617
    if 'LIBPATH' in group:
618
        Env.AppendUnique(LIBPATH = group['LIBPATH'])
G
goprife@gmail.com 已提交
619

620
    # check whether to build group library
621
    if 'LIBRARY' in group:
622 623
        objs = Env.Library(name, group['src'])
    else:
624
        # only add source
625
        objs = group['src']
G
goprife@gmail.com 已提交
626

B
Bright Pan 已提交
627
    # merge group
628 629 630 631 632 633
    for g in Projects:
        if g['name'] == name:
            # merge to this group
            MergeGroup(g, group)
            return objs

B
Bright Pan 已提交
634
    # add a new group
635 636
    Projects.append(group)

G
goprife@gmail.com 已提交
637 638 639 640 641 642 643 644 645
    return objs

def GetCurrentDir():
    conscript = File('SConscript')
    fn = conscript.rfile()
    name = fn.name
    path = os.path.dirname(fn.abspath)
    return path

646 647 648 649 650 651 652 653 654 655 656
PREBUILDING = []
def RegisterPreBuildingAction(act):
    global PREBUILDING
    assert callable(act), 'Could only register callable objects. %s received' % repr(act)
    PREBUILDING.append(act)

def PreBuilding():
    global PREBUILDING
    for a in PREBUILDING:
        a()

B
bernard 已提交
657
def GroupLibName(name, env):
B
bernard 已提交
658 659 660 661 662 663 664 665 666 667
    import rtconfig
    if rtconfig.PLATFORM == 'armcc':
        return name + '_rvds'
    elif rtconfig.PLATFORM == 'gcc':
        return name + '_gcc'

    return name

def GroupLibFullName(name, env):
    return env['LIBPREFIX'] + GroupLibName(name, env) + env['LIBSUFFIX']
B
bernard 已提交
668 669 670 671 672

def BuildLibInstallAction(target, source, env):
    lib_name = GetOption('buildlib')
    for Group in Projects:
        if Group['name'] == lib_name:
B
bernard 已提交
673
            lib_name = GroupLibFullName(Group['name'], env)
B
bernard 已提交
674
            dst_name = os.path.join(Group['path'], lib_name)
675
            print ('Copy '+lib_name+' => ' +dst_name)
B
bernard 已提交
676 677 678
            do_copy_file(lib_name, dst_name)
            break

679
def DoBuilding(target, objects):
680 681 682 683 684 685 686 687 688 689 690

    # merge all objects into one list
    def one_list(l):
        lst = []
        for item in l:
            if type(item) == type([]):
                lst += one_list(item)
            else:
                lst.append(item)
        return lst

691 692
    # handle local group
    def local_group(group, objects):
693
        if 'LOCAL_CCFLAGS' in group or 'LOCAL_CPPPATH' in group or 'LOCAL_CPPDEFINES' in group or 'LOCAL_ASFLAGS' in group:
694 695 696
            CCFLAGS = Env.get('CCFLAGS', '') + group.get('LOCAL_CCFLAGS', '')
            CPPPATH = Env.get('CPPPATH', ['']) + group.get('LOCAL_CPPPATH', [''])
            CPPDEFINES = Env.get('CPPDEFINES', ['']) + group.get('LOCAL_CPPDEFINES', [''])
697
            ASFLAGS = Env.get('ASFLAGS', '') + group.get('LOCAL_ASFLAGS', '')
698 699

            for source in group['src']:
700
                objects.append(Env.Object(source, CCFLAGS = CCFLAGS, ASFLAGS = ASFLAGS,
701 702 703 704 705 706 707
                    CPPPATH = CPPPATH, CPPDEFINES = CPPDEFINES))

            return True

        return False

    objects = one_list(objects)
708

709 710 711 712
    program = None
    # check whether special buildlib option
    lib_name = GetOption('buildlib')
    if lib_name:
713
        objects = [] # remove all of objects
714 715 716
        # build library with special component
        for Group in Projects:
            if Group['name'] == lib_name:
B
bernard 已提交
717
                lib_name = GroupLibName(Group['name'], Env)
718 719 720
                if not local_group(Group, objects):
                    objects = Env.Object(Group['src'])

721
                program = Env.Library(lib_name, objects)
B
bernard 已提交
722 723 724 725

                # add library copy action
                Env.BuildLib(lib_name, program)

726 727
                break
    else:
728 729
        # remove source files with local flags setting
        for group in Projects:
730
            if 'LOCAL_CCFLAGS' in group or 'LOCAL_CPPPATH' in group or 'LOCAL_CPPDEFINES' in group:
731 732 733 734 735 736 737 738 739
                for source in group['src']:
                    for obj in objects:
                        if source.abspath == obj.abspath or (len(obj.sources) > 0 and source.abspath == obj.sources[0].abspath):
                            objects.remove(obj)

        # re-add the source files to the objects
        for group in Projects:
            local_group(group, objects)

740 741 742
        program = Env.Program(target, objects)

    EndBuilding(target, program)
743

744
def GenTargetProject(program = None):
G
goprife@gmail.com 已提交
745 746

    if GetOption('target') == 'mdk':
B
bernard 已提交
747 748
        from keil import MDKProject
        from keil import MDK4Project
B
Bright Pan 已提交
749
        from keil import MDK5Project
B
bernard 已提交
750

G
goprife@gmail.com 已提交
751 752 753 754 755 756 757 758
        template = os.path.isfile('template.Uv2')
        if template:
            MDKProject('project.Uv2', Projects)
        else:
            template = os.path.isfile('template.uvproj')
            if template:
                MDK4Project('project.uvproj', Projects)
            else:
B
Bright Pan 已提交
759 760 761 762
                template = os.path.isfile('template.uvprojx')
                if template:
                    MDK5Project('project.uvprojx', Projects)
                else:
763
                    print ('No template project file found.')
B
Bright Pan 已提交
764

G
goprife@gmail.com 已提交
765
    if GetOption('target') == 'mdk4':
B
bernard 已提交
766
        from keil import MDK4Project
G
goprife@gmail.com 已提交
767 768
        MDK4Project('project.uvproj', Projects)

B
Bright Pan 已提交
769 770 771 772
    if GetOption('target') == 'mdk5':
        from keil import MDK5Project
        MDK5Project('project.uvprojx', Projects)

G
goprife@gmail.com 已提交
773
    if GetOption('target') == 'iar':
B
bernard 已提交
774
        from iar import IARProject
B
Bright Pan 已提交
775
        IARProject('project.ewp', Projects)
776

777
    if GetOption('target') == 'vs':
B
bernard 已提交
778
        from vs import VSProject
779
        VSProject('project.vcproj', Projects, program)
780

781
    if GetOption('target') == 'vs2012':
B
bernard 已提交
782
        from vs2012 import VS2012Project
783 784
        VS2012Project('project.vcxproj', Projects, program)

wuyangyong's avatar
wuyangyong 已提交
785
    if GetOption('target') == 'cb':
B
bernard 已提交
786
        from codeblocks import CBProject
wuyangyong's avatar
wuyangyong 已提交
787 788
        CBProject('project.cbp', Projects, program)

B
bernard 已提交
789 790 791
    if GetOption('target') == 'ua':
        from ua import PrepareUA
        PrepareUA(Projects, Rtt_Root, str(Dir('#')))
B
Bright Pan 已提交
792

793 794 795
    if GetOption('target') == 'vsc':
        from vsc import GenerateVSCode
        GenerateVSCode(Env)
796

797 798 799
    if GetOption('target') == 'cdk':
        from cdk import CDKProject
        CDKProject('project.cdkproj', Projects)
800

801 802 803 804
    if GetOption('target') == 'ses':
        from ses import SESProject
        SESProject(Env)

805 806 807 808 809 810 811 812
def EndBuilding(target, program = None):
    import rtconfig

    need_exit = False

    Env['target']  = program
    Env['project'] = Projects

813 814 815
    if hasattr(rtconfig, 'BSP_LIBRARY_TYPE'):
        Env['bsp_lib_type'] = rtconfig.BSP_LIBRARY_TYPE

816 817 818 819 820 821 822 823 824
    Env.AddPostAction(target, rtconfig.POST_ACTION)
    # Add addition clean files
    Clean(target, 'cconfig.h')
    Clean(target, 'rtua.py')
    Clean(target, 'rtua.pyc')

    if GetOption('target'):
        GenTargetProject(program)

B
bernard 已提交
825 826 827 828
    BSP_ROOT = Dir('#').abspath
    if GetOption('make-dist') and program != None:
        from mkdist import MkDist
        MkDist(program, BSP_ROOT, Rtt_Root, Env)
829 830 831
    if GetOption('make-dist-strip') and program != None:
        from mkdist import MkDist_Strip
        MkDist_Strip(program, BSP_ROOT, Rtt_Root, Env)
832
        need_exit = True
833 834 835 836
    if GetOption('cscope'):
        from cscope import CscopeDatabase
        CscopeDatabase(Projects)

837 838
    if not GetOption('help') and not GetOption('target'):
        if not os.path.exists(rtconfig.EXEC_PATH):
839
            print ("Error: the toolchain path (" + rtconfig.EXEC_PATH + ") is not exist, please check 'EXEC_PATH' in path or rtconfig.py.")
840 841 842 843
            need_exit = True

    if need_exit:
        exit(0)
844

G
goprife@gmail.com 已提交
845
def SrcRemove(src, remove):
G
Grissiom 已提交
846 847 848
    if not src:
        return

X
xieyangrun 已提交
849
    src_bak = src[:]
850 851 852 853 854 855

    if type(remove) == type('str'):
        if os.path.isabs(remove):
            remove = os.path.relpath(remove, GetCurrentDir())
        remove = os.path.normpath(remove)

X
xieyangrun 已提交
856
        for item in src_bak:
857 858 859 860
            if type(item) == type('str'):
                item_str = item
            else:
                item_str = item.rstr()
861

862 863 864
            if os.path.isabs(item_str):
                item_str = os.path.relpath(item_str, GetCurrentDir())
            item_str = os.path.normpath(item_str)
865 866

            if item_str == remove:
X
xieyangrun 已提交
867
                src.remove(item)
868 869 870 871 872 873 874
    else:
        for remove_item in remove:
            remove_str = str(remove_item)
            if os.path.isabs(remove_str):
                remove_str = os.path.relpath(remove_str, GetCurrentDir())
            remove_str = os.path.normpath(remove_str)

X
xieyangrun 已提交
875
            for item in src_bak:
876 877 878 879 880 881 882 883 884 885
                if type(item) == type('str'):
                    item_str = item
                else:
                    item_str = item.rstr()

                if os.path.isabs(item_str):
                    item_str = os.path.relpath(item_str, GetCurrentDir())
                item_str = os.path.normpath(item_str)

                if item_str == remove_str:
X
xieyangrun 已提交
886
                    src.remove(item)
887 888 889 890 891 892 893

def GetVersion():
    import SCons.cpp
    import string

    rtdef = os.path.join(Rtt_Root, 'include', 'rtdef.h')

B
Bright Pan 已提交
894
    # parse rtdef.h to get RT-Thread version
895
    prepcessor = PatchedPreProcessor()
896
    f = open(rtdef, 'r')
897 898 899 900 901 902 903 904
    contents = f.read()
    f.close()
    prepcessor.process_contents(contents)
    def_ns = prepcessor.cpp_namespace

    version = int(filter(lambda ch: ch in '0123456789.', def_ns['RT_VERSION']))
    subversion = int(filter(lambda ch: ch in '0123456789.', def_ns['RT_SUBVERSION']))

905
    if 'RT_REVISION' in def_ns:
906 907 908 909
        revision = int(filter(lambda ch: ch in '0123456789.', def_ns['RT_REVISION']))
        return '%d.%d.%d' % (version, subversion, revision)

    return '0.%d.%d' % (version, subversion)
910

911 912 913 914
def GlobSubDir(sub_dir, ext_name):
    import os
    import glob

915 916 917 918 919 920 921 922 923 924 925 926 927 928 929
    def glob_source(sub_dir, ext_name):
        list = os.listdir(sub_dir)
        src = glob.glob(os.path.join(sub_dir, ext_name))

        for item in list:
            full_subdir = os.path.join(sub_dir, item)
            if os.path.isdir(full_subdir):
                src += glob_source(full_subdir, ext_name)
        return src

    dst = []
    src = glob_source(sub_dir, ext_name)
    for item in src:
        dst.append(os.path.relpath(item, sub_dir))
    return dst
930

931 932 933 934
def PackageSConscript(package):
    from package import BuildPackage

    return BuildPackage(package)