building.py 26.1 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 24
# 2015-07-25     Bernard      Add LOCAL_CCFLAGS/LOCAL_CPPPATH/LOCAL_CPPDEFINES for
#                             group definition. 
25 26
#

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

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

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

40 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 73 74 75 76 77
# 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
    
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 87 88 89 90
        # 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:
                    print 'Error removing file: %s' % e
                    return -1
            return 0

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

        newargs = string.join(args[1:], ' ')
        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 110
            print 'Error in calling:\n%s' % cmdline
            print 'Exception: %s: %s' % (e, os.strerror(e.errno))
            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
def PrepareBuilding(env, root_directory, has_libcpu=False, remove_components = []):
G
goprife@gmail.com 已提交
117 118 119 120 121 122 123 124 125
    import SCons.cpp
    import rtconfig

    global BuildOptions
    global Projects
    global Env
    global Rtt_Root

    Env = env
126
    Rtt_Root = os.path.abspath(root_directory)
G
goprife@gmail.com 已提交
127

128 129 130 131 132
    # 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')
P
prife 已提交
133
                Env['LINKFLAGS']=Env['LINKFLAGS'].replace('RV31', 'armcc')
134

B
Bright Pan 已提交
135
        # reset AR command flags
B
bernard 已提交
136 137
        env['ARCOM'] = '$AR --create $TARGET $SOURCES'
        env['LIBPREFIX']   = ''
B
bernard 已提交
138
        env['LIBSUFFIX']   = '.lib'
B
bernard 已提交
139
        env['LIBLINKPREFIX'] = ''
B
bernard 已提交
140
        env['LIBLINKSUFFIX']   = '.lib'
B
bernard 已提交
141
        env['LIBDIRPREFIX'] = '--userlibpath '
B
bernard 已提交
142

G
goprife@gmail.com 已提交
143
    # patch for win32 spawn
G
Grissiom 已提交
144
    if env['PLATFORM'] == 'win32':
G
goprife@gmail.com 已提交
145 146 147
        win32_spawn = Win32Spawn()
        win32_spawn.env = env
        env['SPAWN'] = win32_spawn.spawn
G
Grissiom 已提交
148

149
    if env['PLATFORM'] == 'win32':
150 151 152
        os.environ['PATH'] = rtconfig.EXEC_PATH + ";" + os.environ['PATH']
    else:
        os.environ['PATH'] = rtconfig.EXEC_PATH + ":" + os.environ['PATH']
G
goprife@gmail.com 已提交
153 154 155

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

B
bernard 已提交
159 160 161 162 163
    # 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 已提交
164
    # parse rtconfig.h to get used component
165
    PreProcessor = PatchedPreProcessor()
G
goprife@gmail.com 已提交
166 167 168 169 170 171
    f = file('rtconfig.h', 'r')
    contents = f.read()
    f.close()
    PreProcessor.process_contents(contents)
    BuildOptions = PreProcessor.cpp_namespace

B
Bright Pan 已提交
172
    # add copy option
173 174 175 176 177
    AddOption('--copy',
                      dest='copy',
                      action='store_true',
                      default=False,
                      help='copy rt-thread directory to local.')
178 179 180 181 182
    AddOption('--copy-header',
                      dest='copy-header',
                      action='store_true',
                      default=False,
                      help='copy header of rt-thread directory to local.')
B
bernard 已提交
183 184 185 186 187
    AddOption('--dist',
                      dest = 'make-dist',
                      action = 'store_true',
                      default=False,
                      help = 'make distribution')
188 189 190 191 192
    AddOption('--cscope',
                      dest='cscope',
                      action='store_true',
                      default=False,
                      help='Build Cscope cross reference database. Requires cscope installed.')
193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212
    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.')

    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.
213 214 215 216 217
        # 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'])
218 219 220
        # remove the POST_ACTION as it will cause meaningless errors(file not
        # found or something like that).
        rtconfig.POST_ACTION = ''
221

222
    # add build library option
B
Bright Pan 已提交
223 224
    AddOption('--buildlib',
                      dest='buildlib',
225 226
                      type='string',
                      help='building library of a component')
B
Bright Pan 已提交
227 228
    AddOption('--cleanlib',
                      dest='cleanlib',
B
bernard 已提交
229 230 231
                      action='store_true',
                      default=False,
                      help='clean up the library by --buildlib')
232

G
goprife@gmail.com 已提交
233 234 235 236
    # add target option
    AddOption('--target',
                      dest='target',
                      type='string',
237
                      help='set target project: mdk/mdk4/iar/vs/ua')
G
goprife@gmail.com 已提交
238 239 240 241

    #{target_name:(CROSS_TOOL, PLATFORM)}
    tgt_dict = {'mdk':('keil', 'armcc'),
                'mdk4':('keil', 'armcc'),
B
Bright Pan 已提交
242
                'mdk5':('keil', 'armcc'),
243
                'iar':('iar', 'iar'),
wuyangyong's avatar
wuyangyong 已提交
244
                'vs':('msvc', 'cl'),
245
                'vs2012':('msvc', 'cl'),
B
bernard 已提交
246
                'cb':('keil', 'armcc'),
B
Bernard Xiong 已提交
247
                'ua':('gcc', 'gcc')}
G
goprife@gmail.com 已提交
248 249
    tgt_name = GetOption('target')
    if tgt_name:
250 251 252 253 254 255
        # --target will change the toolchain settings which clang-analyzer is
        # depend on
        if GetOption('clang-analyzer'):
            print '--clang-analyzer cannot be used with --target'
            sys.exit(1)

G
goprife@gmail.com 已提交
256 257 258 259 260 261 262 263
        SetOption('no_exec', 1)
        try:
            rtconfig.CROSS_TOOL, rtconfig.PLATFORM = tgt_dict[tgt_name]
        except KeyError:
            print 'Unknow target: %s. Avaible targets: %s' % \
                    (tgt_name, ', '.join(tgt_dict.keys()))
            sys.exit(1)
    elif (GetDepend('RT_USING_NEWLIB') == False and GetDepend('RT_USING_NOLIBC') == False) \
264
        and rtconfig.PLATFORM == 'gcc':
G
goprife@gmail.com 已提交
265 266
        AddDepend('RT_USING_MINILIBC')

B
bernard 已提交
267 268 269 270 271 272 273 274 275 276
    AddOption('--genconfig', 
                dest = 'genconfig',
                action = 'store_true',
                default = False, 
                help = 'Generate .config from rtconfig.h')
    if GetOption('genconfig'):
        from genconf import genconfig
        genconfig()
        exit(0)

B
Bernard Xiong 已提交
277 278 279 280 281 282 283 284 285 286 287
    if env['PLATFORM'] != 'win32':
        AddOption('--menuconfig', 
                    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)

288
    # add comstr option
289 290
    AddOption('--verbose',
                dest='verbose',
291
                action='store_true',
B
Bernard Xiong 已提交
292
                default=False,
293
                help='print verbose information during build')
294

295 296
    if not GetOption('verbose'):
        # override the default verbose command string
297
        env.Replace(
R
Rogerz Zhang 已提交
298
            ARCOMSTR = 'AR $TARGET',
299
            ASCOMSTR = 'AS $TARGET',
R
Rogerz Zhang 已提交
300
            ASPPCOMSTR = 'AS $TARGET',
301 302 303 304
            CCCOMSTR = 'CC $TARGET',
            CXXCOMSTR = 'CXX $TARGET',
            LINKCOMSTR = 'LINK $TARGET'
        )
G
goprife@gmail.com 已提交
305

306 307 308 309 310
    # fix the linker for C++
    if GetDepend('RT_USING_CPLUSPLUS'):
        if env['LINK'].find('gcc') != -1:
            env['LINK'] = env['LINK'].replace('gcc', 'g++')

311 312 313
    # 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 已提交
314
    bsp_vdir = 'build'
315 316 317
    kernel_vdir = 'build/kernel'
    # board build script
    objs = SConscript('SConscript', variant_dir=bsp_vdir, duplicate=0)
G
goprife@gmail.com 已提交
318
    # include kernel
319
    objs.extend(SConscript(Rtt_Root + '/src/SConscript', variant_dir=kernel_vdir + '/src', duplicate=0))
G
goprife@gmail.com 已提交
320 321
    # include libcpu
    if not has_libcpu:
322 323
        objs.extend(SConscript(Rtt_Root + '/libcpu/SConscript',
                    variant_dir=kernel_vdir + '/libcpu', duplicate=0))
324

G
goprife@gmail.com 已提交
325
    # include components
326
    objs.extend(SConscript(Rtt_Root + '/components/SConscript',
327
                           variant_dir=kernel_vdir + '/components',
328 329
                           duplicate=0,
                           exports='remove_components'))
G
goprife@gmail.com 已提交
330 331 332

    return objs

B
Bernard Xiong 已提交
333
def PrepareModuleBuilding(env, root_directory, bsp_directory):
G
goprife@gmail.com 已提交
334 335
    import rtconfig

336
    global BuildOptions
G
goprife@gmail.com 已提交
337 338 339
    global Env
    global Rtt_Root

340 341 342 343 344 345
    # patch for win32 spawn
    if env['PLATFORM'] == 'win32':
        win32_spawn = Win32Spawn()
        win32_spawn.env = env
        env['SPAWN'] = win32_spawn.spawn

G
goprife@gmail.com 已提交
346 347 348
    Env = env
    Rtt_Root = root_directory

B
Bernard Xiong 已提交
349
    # parse bsp rtconfig.h to get used component
350
    PreProcessor = PatchedPreProcessor()
B
Bernard Xiong 已提交
351 352 353 354 355 356
    f = file(bsp_directory + '/rtconfig.h', 'r')
    contents = f.read()
    f.close()
    PreProcessor.process_contents(contents)
    BuildOptions = PreProcessor.cpp_namespace

B
Bright Pan 已提交
357 358 359
    # add build/clean library option for library checking
    AddOption('--buildlib',
              dest='buildlib',
B
bernard 已提交
360 361
              type='string',
              help='building library of a component')
B
Bright Pan 已提交
362 363
    AddOption('--cleanlib',
              dest='cleanlib',
B
bernard 已提交
364 365 366 367
              action='store_true',
              default=False,
              help='clean up the library by --buildlib')

G
goprife@gmail.com 已提交
368 369 370
    # add program path
    env.PrependENVPath('PATH', rtconfig.EXEC_PATH)

wuyangyong's avatar
wuyangyong 已提交
371 372 373 374 375 376 377
def GetConfigValue(name):
    assert type(name) == str, 'GetConfigValue: only string parameter is valid'
    try:
        return BuildOptions[name]
    except:
        return ''

G
goprife@gmail.com 已提交
378 379 380 381 382 383 384
def GetDepend(depend):
    building = True
    if type(depend) == type('str'):
        if not BuildOptions.has_key(depend) or BuildOptions[depend] == 0:
            building = False
        elif BuildOptions[depend] != '':
            return BuildOptions[depend]
B
Bright Pan 已提交
385

G
goprife@gmail.com 已提交
386 387 388 389 390 391 392 393 394 395
        return building

    # for list type depend
    for item in depend:
        if item != '':
            if not BuildOptions.has_key(item) or BuildOptions[item] == 0:
                building = False

    return building

B
Bernard Xiong 已提交
396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428
def LocalOptions(config_filename):
    from SCons.Script import SCons

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

    f = file(config_filename, 'r')
    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'):
        if not options.has_key(depend) or options[depend] == 0:
            building = False
        elif options[depend] != '':
            return options[depend]

        return building

    # for list type depend
    for item in depend:
        if item != '':
            if not options.has_key(item) or options[item] == 0:
                building = False

    return building

G
goprife@gmail.com 已提交
429 430 431
def AddDepend(option):
    BuildOptions[option] = 1

432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448
def MergeGroup(src_group, group):
    src_group['src'] = src_group['src'] + group['src']
    if group.has_key('CCFLAGS'):
        if src_group.has_key('CCFLAGS'):
            src_group['CCFLAGS'] = src_group['CCFLAGS'] + group['CCFLAGS']
        else:
            src_group['CCFLAGS'] = group['CCFLAGS']
    if group.has_key('CPPPATH'):
        if src_group.has_key('CPPPATH'):
            src_group['CPPPATH'] = src_group['CPPPATH'] + group['CPPPATH']
        else:
            src_group['CPPPATH'] = group['CPPPATH']
    if group.has_key('CPPDEFINES'):
        if src_group.has_key('CPPDEFINES'):
            src_group['CPPDEFINES'] = src_group['CPPDEFINES'] + group['CPPDEFINES']
        else:
            src_group['CPPDEFINES'] = group['CPPDEFINES']
449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466

    # for local CCFLAGS/CPPPATH/CPPDEFINES
    if group.has_key('LOCAL_CCFLAGS'):
        if src_group.has_key('LOCAL_CCFLAGS'):
            src_group['LOCAL_CCFLAGS'] = src_group['LOCAL_CCFLAGS'] + group['LOCAL_CCFLAGS']
        else:
            src_group['LOCAL_CCFLAGS'] = group['LOCAL_CCFLAGS']
    if group.has_key('LOCAL_CPPPATH'):
        if src_group.has_key('LOCAL_CPPPATH'):
            src_group['LOCAL_CPPPATH'] = src_group['LOCAL_CPPPATH'] + group['LOCAL_CPPPATH']
        else:
            src_group['LOCAL_CPPPATH'] = group['LOCAL_CPPPATH']
    if group.has_key('LOCAL_CPPDEFINES'):
        if src_group.has_key('LOCAL_CPPDEFINES'):
            src_group['LOCAL_CPPDEFINES'] = src_group['LOCAL_CPPDEFINES'] + group['LOCAL_CPPDEFINES']
        else:
            src_group['LOCAL_CPPDEFINES'] = group['LOCAL_CPPDEFINES']

467 468 469 470 471
    if group.has_key('LINKFLAGS'):
        if src_group.has_key('LINKFLAGS'):
            src_group['LINKFLAGS'] = src_group['LINKFLAGS'] + group['LINKFLAGS']
        else:
            src_group['LINKFLAGS'] = group['LINKFLAGS']
472 473 474
    if group.has_key('LIBS'):
        if src_group.has_key('LIBS'):
            src_group['LIBS'] = src_group['LIBS'] + group['LIBS']
475
        else:
476 477 478 479 480 481
            src_group['LIBS'] = group['LIBS']
    if group.has_key('LIBPATH'):
        if src_group.has_key('LIBPATH'):
            src_group['LIBPATH'] = src_group['LIBPATH'] + group['LIBPATH']
        else:
            src_group['LIBPATH'] = group['LIBPATH']
482

G
goprife@gmail.com 已提交
483 484 485 486 487
def DefineGroup(name, src, depend, **parameters):
    global Env
    if not GetDepend(depend):
        return []

488 489 490 491 492 493 494 495
    # 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 已提交
496 497
    group = parameters
    group['name'] = name
498
    group['path'] = group_path
B
bernard 已提交
499
    if type(src) == type(['src1']):
G
goprife@gmail.com 已提交
500 501 502 503 504
        group['src'] = File(src)
    else:
        group['src'] = src

    if group.has_key('CCFLAGS'):
505
        Env.AppendUnique(CCFLAGS = group['CCFLAGS'])
G
goprife@gmail.com 已提交
506
    if group.has_key('CPPPATH'):
507
        Env.AppendUnique(CPPPATH = group['CPPPATH'])
G
goprife@gmail.com 已提交
508
    if group.has_key('CPPDEFINES'):
509
        Env.AppendUnique(CPPDEFINES = group['CPPDEFINES'])
G
goprife@gmail.com 已提交
510
    if group.has_key('LINKFLAGS'):
511
        Env.AppendUnique(LINKFLAGS = group['LINKFLAGS'])
B
bernard 已提交
512

B
Bright Pan 已提交
513
    # check whether to clean up library
B
bernard 已提交
514
    if GetOption('cleanlib') and os.path.exists(os.path.join(group['path'], GroupLibFullName(name, Env))):
B
bernard 已提交
515
        if group['src'] != []:
B
bernard 已提交
516
            print 'Remove library:', GroupLibFullName(name, Env)
B
bernard 已提交
517 518 519
            fn = os.path.join(group['path'], GroupLibFullName(name, Env))
            if os.path.exists(fn):
                os.unlink(fn)
B
bernard 已提交
520 521

    # check whether exist group library
B
bernard 已提交
522
    if not GetOption('buildlib') and os.path.exists(os.path.join(group['path'], GroupLibFullName(name, Env))):
B
bernard 已提交
523
        group['src'] = []
524 525 526 527
        if group.has_key('LIBS'): group['LIBS'] = group['LIBS'] + [GroupLibName(name, Env)]
        else : group['LIBS'] = [GroupLibName(name, Env)]
        if group.has_key('LIBPATH'): group['LIBPATH'] = group['LIBPATH'] + [GetCurrentDir()]
        else : group['LIBPATH'] = [GetCurrentDir()]
B
bernard 已提交
528

529
    if group.has_key('LIBS'):
530
        Env.AppendUnique(LIBS = group['LIBS'])
531
    if group.has_key('LIBPATH'):
532
        Env.AppendUnique(LIBPATH = group['LIBPATH'])
G
goprife@gmail.com 已提交
533

534
    # check whether to build group library
G
goprife@gmail.com 已提交
535
    if group.has_key('LIBRARY'):
536 537
        objs = Env.Library(name, group['src'])
    else:
538
        # only add source
539
        objs = group['src']
G
goprife@gmail.com 已提交
540

B
Bright Pan 已提交
541
    # merge group
542 543 544 545 546 547
    for g in Projects:
        if g['name'] == name:
            # merge to this group
            MergeGroup(g, group)
            return objs

B
Bright Pan 已提交
548
    # add a new group
549 550
    Projects.append(group)

G
goprife@gmail.com 已提交
551 552 553 554 555 556 557 558 559
    return objs

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

560 561 562 563 564 565 566 567 568 569 570
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 已提交
571
def GroupLibName(name, env):
B
bernard 已提交
572 573 574 575 576 577 578 579 580 581
    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 已提交
582 583 584 585 586

def BuildLibInstallAction(target, source, env):
    lib_name = GetOption('buildlib')
    for Group in Projects:
        if Group['name'] == lib_name:
B
bernard 已提交
587
            lib_name = GroupLibFullName(Group['name'], env)
B
bernard 已提交
588 589 590 591 592
            dst_name = os.path.join(Group['path'], lib_name)
            print 'Copy %s => %s' % (lib_name, dst_name)
            do_copy_file(lib_name, dst_name)
            break

593
def DoBuilding(target, objects):
594 595 596 597 598 599 600 601 602 603 604

    # 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

605 606
    # handle local group
    def local_group(group, objects):
607 608 609 610 611 612
        if group.has_key('LOCAL_CCFLAGS') or group.has_key('LOCAL_CPPPATH') or group.has_key('LOCAL_CPPDEFINES'):
            CCFLAGS = Env.get('CCFLAGS', '') + group.get('LOCAL_CCFLAGS', '')
            CPPPATH = Env.get('CPPPATH', ['']) + group.get('LOCAL_CPPPATH', [''])
            CPPDEFINES = Env.get('CPPDEFINES', ['']) + group.get('LOCAL_CPPDEFINES', [''])

            for source in group['src']:
613 614 615 616 617 618 619 620
                objects.append(Env.Object(source, CCFLAGS = CCFLAGS,
                    CPPPATH = CPPPATH, CPPDEFINES = CPPDEFINES))

            return True

        return False

    objects = one_list(objects)
621

622 623 624 625
    program = None
    # check whether special buildlib option
    lib_name = GetOption('buildlib')
    if lib_name:
626
        objects = [] # remove all of objects
627 628 629
        # build library with special component
        for Group in Projects:
            if Group['name'] == lib_name:
B
bernard 已提交
630
                lib_name = GroupLibName(Group['name'], Env)
631 632 633
                if not local_group(Group, objects):
                    objects = Env.Object(Group['src'])

634
                program = Env.Library(lib_name, objects)
B
bernard 已提交
635 636 637 638

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

639 640
                break
    else:
641 642 643 644 645 646 647 648 649 650 651 652
        # remove source files with local flags setting
        for group in Projects:
            if group.has_key('LOCAL_CCFLAGS') or group.has_key('LOCAL_CPPPATH') or group.has_key('LOCAL_CPPDEFINES'):
                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)

653 654 655 656
        program = Env.Program(target, objects)

    EndBuilding(target, program)

657
def EndBuilding(target, program = None):
G
goprife@gmail.com 已提交
658
    import rtconfig
659

G
goprife@gmail.com 已提交
660 661 662
    Env.AddPostAction(target, rtconfig.POST_ACTION)

    if GetOption('target') == 'mdk':
B
bernard 已提交
663 664
        from keil import MDKProject
        from keil import MDK4Project
B
Bright Pan 已提交
665
        from keil import MDK5Project
B
bernard 已提交
666

G
goprife@gmail.com 已提交
667 668 669 670 671 672 673 674
        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 已提交
675 676 677 678 679 680
                template = os.path.isfile('template.uvprojx')
                if template:
                    MDK5Project('project.uvprojx', Projects)
                else:
                    print 'No template project file found.'

G
goprife@gmail.com 已提交
681
    if GetOption('target') == 'mdk4':
B
bernard 已提交
682
        from keil import MDK4Project
G
goprife@gmail.com 已提交
683 684
        MDK4Project('project.uvproj', Projects)

B
Bright Pan 已提交
685 686 687 688
    if GetOption('target') == 'mdk5':
        from keil import MDK5Project
        MDK5Project('project.uvprojx', Projects)

G
goprife@gmail.com 已提交
689
    if GetOption('target') == 'iar':
B
bernard 已提交
690
        from iar import IARProject
B
Bright Pan 已提交
691
        IARProject('project.ewp', Projects)
692

693
    if GetOption('target') == 'vs':
B
bernard 已提交
694
        from vs import VSProject
695
        VSProject('project.vcproj', Projects, program)
696

697
    if GetOption('target') == 'vs2012':
B
bernard 已提交
698
        from vs2012 import VS2012Project
699 700
        VS2012Project('project.vcxproj', Projects, program)

wuyangyong's avatar
wuyangyong 已提交
701
    if GetOption('target') == 'cb':
B
bernard 已提交
702
        from codeblocks import CBProject
wuyangyong's avatar
wuyangyong 已提交
703 704
        CBProject('project.cbp', Projects, program)

B
bernard 已提交
705 706 707
    if GetOption('target') == 'ua':
        from ua import PrepareUA
        PrepareUA(Projects, Rtt_Root, str(Dir('#')))
B
Bright Pan 已提交
708

B
bernard 已提交
709
    BSP_ROOT = Dir('#').abspath
710
    if GetOption('copy') and program != None:
B
bernard 已提交
711 712 713
        from mkdist import MakeCopy
        MakeCopy(program, BSP_ROOT, Rtt_Root, Env)
        exit(0)
714
    if GetOption('copy-header') and program != None:
B
bernard 已提交
715 716 717 718 719 720 721
        from mkdist import MakeCopyHeader
        MakeCopyHeader(program, BSP_ROOT, Rtt_Root, Env)
        exit(0)
    if GetOption('make-dist') and program != None:
        from mkdist import MkDist
        MkDist(program, BSP_ROOT, Rtt_Root, Env)
        exit(0)
722 723 724 725
    if GetOption('cscope'):
        from cscope import CscopeDatabase
        CscopeDatabase(Projects)

G
goprife@gmail.com 已提交
726
def SrcRemove(src, remove):
G
Grissiom 已提交
727 728 729
    if not src:
        return

730 731
    for item in src:
        if type(item) == type('str'):
732 733
            if os.path.basename(item) in remove:
                src.remove(item)
734 735 736
        else:
            if os.path.basename(item.rstr()) in remove:
                src.remove(item)
737 738 739 740 741 742 743

def GetVersion():
    import SCons.cpp
    import string

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

B
Bright Pan 已提交
744
    # parse rtdef.h to get RT-Thread version
745
    prepcessor = PatchedPreProcessor()
746 747 748 749 750 751 752 753 754 755 756 757 758 759
    f = file(rtdef, 'r')
    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']))

    if def_ns.has_key('RT_REVISION'):
        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)
760

761 762 763 764
def GlobSubDir(sub_dir, ext_name):
    import os
    import glob

765 766 767 768 769 770 771 772 773 774 775 776 777 778 779
    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
780

781 782 783 784 785
def PackageSConscript(package):
    from package import BuildPackage

    return BuildPackage(package)

786