building.py 25.9 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
G
goprife@gmail.com 已提交
33 34 35 36 37 38 39 40

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

class Win32Spawn:
    def spawn(self, sh, escape, cmd, args, env):
G
Grissiom 已提交
41 42 43 44 45 46 47 48 49 50 51
        # 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 已提交
52 53 54 55
        import subprocess

        newargs = string.join(args[1:], ' ')
        cmdline = cmd + " " + newargs
G
Grissiom 已提交
56 57

        # Make sure the env is constructed by strings
58
        _e = dict([(k, str(v)) for k, v in env.items()])
G
Grissiom 已提交
59 60 61 62 63 64

        # 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']
65

G
Grissiom 已提交
66
        try:
67
            proc = subprocess.Popen(cmdline, env=_e, shell=False)
G
Grissiom 已提交
68
        except Exception as e:
69 70 71
            print 'Error in calling:\n%s' % cmdline
            print 'Exception: %s: %s' % (e, os.strerror(e.errno))
            return e.errno
G
Grissiom 已提交
72 73
        finally:
            os.environ['PATH'] = old_path
74

G
Grissiom 已提交
75
        return proc.wait()
G
goprife@gmail.com 已提交
76

77
def PrepareBuilding(env, root_directory, has_libcpu=False, remove_components = []):
G
goprife@gmail.com 已提交
78 79 80 81 82 83 84 85 86 87 88
    import SCons.cpp
    import rtconfig

    global BuildOptions
    global Projects
    global Env
    global Rtt_Root

    Env = env
    Rtt_Root = root_directory

89 90 91 92 93
    # 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 已提交
94
                Env['LINKFLAGS']=Env['LINKFLAGS'].replace('RV31', 'armcc')
95

B
Bright Pan 已提交
96
        # reset AR command flags
B
bernard 已提交
97 98
        env['ARCOM'] = '$AR --create $TARGET $SOURCES'
        env['LIBPREFIX']   = ''
B
bernard 已提交
99
        env['LIBSUFFIX']   = '.lib'
B
bernard 已提交
100
        env['LIBLINKPREFIX'] = ''
B
bernard 已提交
101
        env['LIBLINKSUFFIX']   = '.lib'
B
bernard 已提交
102
        env['LIBDIRPREFIX'] = '--userlibpath '
B
bernard 已提交
103

G
goprife@gmail.com 已提交
104
    # patch for win32 spawn
G
Grissiom 已提交
105
    if env['PLATFORM'] == 'win32':
G
goprife@gmail.com 已提交
106 107 108
        win32_spawn = Win32Spawn()
        win32_spawn.env = env
        env['SPAWN'] = win32_spawn.spawn
G
Grissiom 已提交
109

110
    if env['PLATFORM'] == 'win32':
111 112 113
        os.environ['PATH'] = rtconfig.EXEC_PATH + ";" + os.environ['PATH']
    else:
        os.environ['PATH'] = rtconfig.EXEC_PATH + ":" + os.environ['PATH']
G
goprife@gmail.com 已提交
114 115 116

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

B
bernard 已提交
120 121 122 123 124
    # 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 已提交
125 126 127 128 129 130 131 132
    # parse rtconfig.h to get used component
    PreProcessor = SCons.cpp.PreProcessor()
    f = file('rtconfig.h', 'r')
    contents = f.read()
    f.close()
    PreProcessor.process_contents(contents)
    BuildOptions = PreProcessor.cpp_namespace

B
Bright Pan 已提交
133
    # add copy option
134 135 136 137 138
    AddOption('--copy',
                      dest='copy',
                      action='store_true',
                      default=False,
                      help='copy rt-thread directory to local.')
139 140 141 142 143
    AddOption('--copy-header',
                      dest='copy-header',
                      action='store_true',
                      default=False,
                      help='copy header of rt-thread directory to local.')
144 145 146 147 148
    AddOption('--cscope',
                      dest='cscope',
                      action='store_true',
                      default=False,
                      help='Build Cscope cross reference database. Requires cscope installed.')
149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168
    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.
169 170 171 172 173
        # 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'])
174 175 176
        # remove the POST_ACTION as it will cause meaningless errors(file not
        # found or something like that).
        rtconfig.POST_ACTION = ''
177

178
    # add build library option
B
Bright Pan 已提交
179 180
    AddOption('--buildlib',
                      dest='buildlib',
181 182
                      type='string',
                      help='building library of a component')
B
Bright Pan 已提交
183 184
    AddOption('--cleanlib',
                      dest='cleanlib',
B
bernard 已提交
185 186 187
                      action='store_true',
                      default=False,
                      help='clean up the library by --buildlib')
188

G
goprife@gmail.com 已提交
189 190 191 192
    # add target option
    AddOption('--target',
                      dest='target',
                      type='string',
193
                      help='set target project: mdk/mdk4/iar/vs/ua')
G
goprife@gmail.com 已提交
194 195 196 197

    #{target_name:(CROSS_TOOL, PLATFORM)}
    tgt_dict = {'mdk':('keil', 'armcc'),
                'mdk4':('keil', 'armcc'),
B
Bright Pan 已提交
198
                'mdk5':('keil', 'armcc'),
199
                'iar':('iar', 'iar'),
wuyangyong's avatar
wuyangyong 已提交
200
                'vs':('msvc', 'cl'),
201
                'vs2012':('msvc', 'cl'),
B
bernard 已提交
202
                'cb':('keil', 'armcc'),
B
Bernard Xiong 已提交
203
                'ua':('gcc', 'gcc')}
G
goprife@gmail.com 已提交
204 205
    tgt_name = GetOption('target')
    if tgt_name:
206 207 208 209 210 211
        # --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 已提交
212 213 214 215 216 217 218 219
        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) \
220
        and rtconfig.PLATFORM == 'gcc':
G
goprife@gmail.com 已提交
221 222
        AddDepend('RT_USING_MINILIBC')

223
    # add comstr option
224 225
    AddOption('--verbose',
                dest='verbose',
226 227
                action='store_true',
                default=False,
228
                help='print verbose information during build')
229

230 231
    if not GetOption('verbose'):
        # override the default verbose command string
232
        env.Replace(
R
Rogerz Zhang 已提交
233
            ARCOMSTR = 'AR $TARGET',
234
            ASCOMSTR = 'AS $TARGET',
R
Rogerz Zhang 已提交
235
            ASPPCOMSTR = 'AS $TARGET',
236 237 238 239
            CCCOMSTR = 'CC $TARGET',
            CXXCOMSTR = 'CXX $TARGET',
            LINKCOMSTR = 'LINK $TARGET'
        )
G
goprife@gmail.com 已提交
240

241 242 243 244 245 246 247
    # 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.
    bsp_vdir = 'build/bsp'
    kernel_vdir = 'build/kernel'
    # board build script
    objs = SConscript('SConscript', variant_dir=bsp_vdir, duplicate=0)
G
goprife@gmail.com 已提交
248
    # include kernel
249
    objs.extend(SConscript(Rtt_Root + '/src/SConscript', variant_dir=kernel_vdir + '/src', duplicate=0))
G
goprife@gmail.com 已提交
250 251
    # include libcpu
    if not has_libcpu:
252 253
        objs.extend(SConscript(Rtt_Root + '/libcpu/SConscript',
                    variant_dir=kernel_vdir + '/libcpu', duplicate=0))
254

G
goprife@gmail.com 已提交
255
    # include components
256
    objs.extend(SConscript(Rtt_Root + '/components/SConscript',
257
                           variant_dir=kernel_vdir + '/components',
258 259
                           duplicate=0,
                           exports='remove_components'))
G
goprife@gmail.com 已提交
260 261 262

    return objs

B
Bernard Xiong 已提交
263
def PrepareModuleBuilding(env, root_directory, bsp_directory):
G
goprife@gmail.com 已提交
264 265
    import rtconfig

266
    global BuildOptions
G
goprife@gmail.com 已提交
267 268 269 270 271 272
    global Env
    global Rtt_Root

    Env = env
    Rtt_Root = root_directory

B
Bernard Xiong 已提交
273 274 275 276 277 278 279 280
    # parse bsp rtconfig.h to get used component
    PreProcessor = SCons.cpp.PreProcessor()
    f = file(bsp_directory + '/rtconfig.h', 'r')
    contents = f.read()
    f.close()
    PreProcessor.process_contents(contents)
    BuildOptions = PreProcessor.cpp_namespace

B
Bright Pan 已提交
281 282 283
    # add build/clean library option for library checking
    AddOption('--buildlib',
              dest='buildlib',
B
bernard 已提交
284 285
              type='string',
              help='building library of a component')
B
Bright Pan 已提交
286 287
    AddOption('--cleanlib',
              dest='cleanlib',
B
bernard 已提交
288 289 290 291
              action='store_true',
              default=False,
              help='clean up the library by --buildlib')

G
goprife@gmail.com 已提交
292 293 294
    # add program path
    env.PrependENVPath('PATH', rtconfig.EXEC_PATH)

wuyangyong's avatar
wuyangyong 已提交
295 296 297 298 299 300 301
def GetConfigValue(name):
    assert type(name) == str, 'GetConfigValue: only string parameter is valid'
    try:
        return BuildOptions[name]
    except:
        return ''

G
goprife@gmail.com 已提交
302 303 304 305 306 307 308
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 已提交
309

G
goprife@gmail.com 已提交
310 311 312 313 314 315 316 317 318 319 320 321 322
        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

def AddDepend(option):
    BuildOptions[option] = 1

323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339
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']
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357

    # 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']

358 359 360 361 362
    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']
363 364 365
    if group.has_key('LIBS'):
        if src_group.has_key('LIBS'):
            src_group['LIBS'] = src_group['LIBS'] + group['LIBS']
366
        else:
367 368 369 370 371 372
            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']
373

G
goprife@gmail.com 已提交
374 375 376 377 378
def DefineGroup(name, src, depend, **parameters):
    global Env
    if not GetDepend(depend):
        return []

379 380 381 382 383 384 385 386
    # 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 已提交
387 388
    group = parameters
    group['name'] = name
389
    group['path'] = group_path
B
bernard 已提交
390
    if type(src) == type(['src1']):
G
goprife@gmail.com 已提交
391 392 393 394 395
        group['src'] = File(src)
    else:
        group['src'] = src

    if group.has_key('CCFLAGS'):
396
        Env.AppendUnique(CCFLAGS = group['CCFLAGS'])
G
goprife@gmail.com 已提交
397
    if group.has_key('CPPPATH'):
398
        Env.AppendUnique(CPPPATH = group['CPPPATH'])
G
goprife@gmail.com 已提交
399
    if group.has_key('CPPDEFINES'):
400
        Env.AppendUnique(CPPDEFINES = group['CPPDEFINES'])
G
goprife@gmail.com 已提交
401
    if group.has_key('LINKFLAGS'):
402
        Env.AppendUnique(LINKFLAGS = group['LINKFLAGS'])
B
bernard 已提交
403

B
Bright Pan 已提交
404
    # check whether to clean up library
B
bernard 已提交
405
    if GetOption('cleanlib') and os.path.exists(os.path.join(group['path'], GroupLibFullName(name, Env))):
B
bernard 已提交
406
        if group['src'] != []:
B
bernard 已提交
407 408
            print 'Remove library:', GroupLibFullName(name, Env)
            do_rm_file(os.path.join(group['path'], GroupLibFullName(name, Env)))
B
bernard 已提交
409 410

    # check whether exist group library
B
bernard 已提交
411
    if not GetOption('buildlib') and os.path.exists(os.path.join(group['path'], GroupLibFullName(name, Env))):
B
bernard 已提交
412
        group['src'] = []
413 414 415 416
        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 已提交
417

418
    if group.has_key('LIBS'):
419
        Env.AppendUnique(LIBS = group['LIBS'])
420
    if group.has_key('LIBPATH'):
421
        Env.AppendUnique(LIBPATH = group['LIBPATH'])
G
goprife@gmail.com 已提交
422

423
    # check whether to build group library
G
goprife@gmail.com 已提交
424
    if group.has_key('LIBRARY'):
425 426
        objs = Env.Library(name, group['src'])
    else:
427
        # only add source
428
        objs = group['src']
G
goprife@gmail.com 已提交
429

B
Bright Pan 已提交
430
    # merge group
431 432 433 434 435 436
    for g in Projects:
        if g['name'] == name:
            # merge to this group
            MergeGroup(g, group)
            return objs

B
Bright Pan 已提交
437
    # add a new group
438 439
    Projects.append(group)

G
goprife@gmail.com 已提交
440 441 442 443 444 445 446 447 448
    return objs

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

449 450 451 452 453 454 455 456 457 458 459
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 已提交
460
def GroupLibName(name, env):
B
bernard 已提交
461 462 463 464 465 466 467 468 469 470
    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 已提交
471 472 473 474 475

def BuildLibInstallAction(target, source, env):
    lib_name = GetOption('buildlib')
    for Group in Projects:
        if Group['name'] == lib_name:
B
bernard 已提交
476
            lib_name = GroupLibFullName(Group['name'], env)
B
bernard 已提交
477 478 479 480 481
            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

482
def DoBuilding(target, objects):
483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502
    # 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:
        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']:
                objects += Env.Object(source, CCFLAGS = CCFLAGS,
                    CPPPATH = CPPPATH,
                    CPPDEFINES = CPPDEFINES)

503 504 505 506 507 508 509
    program = None
    # check whether special buildlib option
    lib_name = GetOption('buildlib')
    if lib_name:
        # build library with special component
        for Group in Projects:
            if Group['name'] == lib_name:
B
bernard 已提交
510
                lib_name = GroupLibName(Group['name'], Env)
511 512
                objects = Env.Object(Group['src'])
                program = Env.Library(lib_name, objects)
B
bernard 已提交
513 514 515 516

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

517 518 519 520 521 522
                break
    else:
        program = Env.Program(target, objects)

    EndBuilding(target, program)

523
def EndBuilding(target, program = None):
G
goprife@gmail.com 已提交
524
    import rtconfig
525

G
goprife@gmail.com 已提交
526 527 528
    Env.AddPostAction(target, rtconfig.POST_ACTION)

    if GetOption('target') == 'mdk':
B
bernard 已提交
529 530
        from keil import MDKProject
        from keil import MDK4Project
B
Bright Pan 已提交
531
        from keil import MDK5Project
B
bernard 已提交
532

G
goprife@gmail.com 已提交
533 534 535 536 537 538 539 540
        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 已提交
541 542 543 544 545 546
                template = os.path.isfile('template.uvprojx')
                if template:
                    MDK5Project('project.uvprojx', Projects)
                else:
                    print 'No template project file found.'

G
goprife@gmail.com 已提交
547
    if GetOption('target') == 'mdk4':
B
bernard 已提交
548
        from keil import MDK4Project
G
goprife@gmail.com 已提交
549 550
        MDK4Project('project.uvproj', Projects)

B
Bright Pan 已提交
551 552 553 554
    if GetOption('target') == 'mdk5':
        from keil import MDK5Project
        MDK5Project('project.uvprojx', Projects)

G
goprife@gmail.com 已提交
555
    if GetOption('target') == 'iar':
B
bernard 已提交
556
        from iar import IARProject
B
Bright Pan 已提交
557
        IARProject('project.ewp', Projects)
558

559
    if GetOption('target') == 'vs':
B
bernard 已提交
560
        from vs import VSProject
561
        VSProject('project.vcproj', Projects, program)
562

563
    if GetOption('target') == 'vs2012':
B
bernard 已提交
564
        from vs2012 import VS2012Project
565 566
        VS2012Project('project.vcxproj', Projects, program)

wuyangyong's avatar
wuyangyong 已提交
567
    if GetOption('target') == 'cb':
B
bernard 已提交
568
        from codeblocks import CBProject
wuyangyong's avatar
wuyangyong 已提交
569 570
        CBProject('project.cbp', Projects, program)

B
bernard 已提交
571 572 573
    if GetOption('target') == 'ua':
        from ua import PrepareUA
        PrepareUA(Projects, Rtt_Root, str(Dir('#')))
B
Bright Pan 已提交
574

575 576
    if GetOption('copy') and program != None:
        MakeCopy(program)
577 578
    if GetOption('copy-header') and program != None:
        MakeCopyHeader(program)
579

580 581 582 583
    if GetOption('cscope'):
        from cscope import CscopeDatabase
        CscopeDatabase(Projects)

G
goprife@gmail.com 已提交
584
def SrcRemove(src, remove):
G
Grissiom 已提交
585 586 587
    if not src:
        return

588 589 590 591 592 593 594 595 596
    if type(src[0]) == type('str'):
        for item in src:
            if os.path.basename(item) in remove:
                src.remove(item)
        return

    for item in src:
        if os.path.basename(item.rstr()) in remove:
            src.remove(item)
597 598 599 600 601 602 603

def GetVersion():
    import SCons.cpp
    import string

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

B
Bright Pan 已提交
604
    # parse rtdef.h to get RT-Thread version
605 606 607 608 609 610 611 612 613 614 615 616 617 618 619
    prepcessor = SCons.cpp.PreProcessor()
    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)
620

621 622 623 624
def GlobSubDir(sub_dir, ext_name):
    import os
    import glob

625 626 627 628 629 630 631 632 633 634 635 636 637 638 639
    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
640

641 642 643 644 645
def PackageSConscript(package):
    from package import BuildPackage

    return BuildPackage(package)

646 647 648
def file_path_exist(path, *args):
    return os.path.exists(os.path.join(path, *args))

B
bernard 已提交
649 650 651 652
def do_rm_file(src):
    if os.path.exists(src):
       os.unlink(src)

653
def do_copy_file(src, dst):
654
    import shutil
B
Bright Pan 已提交
655
    # check source file
656
    if not os.path.exists(src):
B
Bright Pan 已提交
657
        return
658 659 660 661 662 663 664 665 666 667

    path = os.path.dirname(dst)
    # mkdir if path not exist
    if not os.path.exists(path):
        os.makedirs(path)

    shutil.copy2(src, dst)

def do_copy_folder(src_dir, dst_dir):
    import shutil
B
Bright Pan 已提交
668
    # check source directory
669 670
    if not os.path.exists(src_dir):
        return
B
Bright Pan 已提交
671

672 673
    if os.path.exists(dst_dir):
        shutil.rmtree(dst_dir)
B
Bright Pan 已提交
674

675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700
    shutil.copytree(src_dir, dst_dir)

source_ext = ["c", "h", "s", "S", "cpp", "xpm"]
source_list = []

def walk_children(child):
    global source_list
    global source_ext

    # print child
    full_path = child.rfile().abspath
    file_type  = full_path.rsplit('.',1)[1]
    #print file_type
    if file_type in source_ext:
        if full_path not in source_list:
            source_list.append(full_path)

    children = child.all_children()
    if children != []:
        for item in children:
            walk_children(item)

def MakeCopy(program):
    global source_list
    global Rtt_Root
    global Env
B
Bright Pan 已提交
701

702
    target_path = os.path.join(Dir('#').abspath, 'rt-thread')
B
Bright Pan 已提交
703

704 705 706 707
    if Env['PLATFORM'] == 'win32':
        RTT_ROOT = Rtt_Root.lower()
    else:
        RTT_ROOT = Rtt_Root
B
Bright Pan 已提交
708

709 710
    if target_path.startswith(RTT_ROOT):
        return
711

712 713
    for item in program:
        walk_children(item)
B
Bright Pan 已提交
714

715
    source_list.sort()
B
Bright Pan 已提交
716

717 718 719 720 721 722 723 724 725 726
    # filte source file in RT-Thread
    target_list = []
    for src in source_list:
        if Env['PLATFORM'] == 'win32':
            src = src.lower()

        if src.startswith(RTT_ROOT):
            target_list.append(src)

    source_list = target_list
B
Bright Pan 已提交
727
    # get source path
728 729 730 731 732 733 734 735 736 737 738
    src_dir = []
    for src in source_list:
        src = src.replace(RTT_ROOT, '')
        if src[0] == os.sep or src[0] == '/':
            src = src[1:]

        path = os.path.dirname(src)
        sub_path = path.split(os.sep)
        full_path = RTT_ROOT
        for item in sub_path:
            full_path = os.path.join(full_path, item)
B
Bright Pan 已提交
739
            if full_path not in src_dir:
740 741
                src_dir.append(full_path)

B
Bright Pan 已提交
742
    for item in src_dir:
743 744 745 746 747 748 749 750 751 752
        source_list.append(os.path.join(item, 'SConscript'))

    for src in source_list:
        dst = src.replace(RTT_ROOT, '')
        if dst[0] == os.sep or dst[0] == '/':
            dst = dst[1:]
        print '=> ', dst
        dst = os.path.join(target_path, dst)
        do_copy_file(src, dst)

B
Bright Pan 已提交
753
    # copy tools directory
754 755 756 757
    print "=>  tools"
    do_copy_folder(os.path.join(RTT_ROOT, "tools"), os.path.join(target_path, "tools"))
    do_copy_file(os.path.join(RTT_ROOT, 'AUTHORS'), os.path.join(target_path, 'AUTHORS'))
    do_copy_file(os.path.join(RTT_ROOT, 'COPYING'), os.path.join(target_path, 'COPYING'))
758 759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800

def MakeCopyHeader(program):
    global source_ext
    source_ext = []
    source_ext = ["h", "xpm"]
    global source_list
    global Rtt_Root
    global Env

    target_path = os.path.join(Dir('#').abspath, 'rt-thread')

    if Env['PLATFORM'] == 'win32':
        RTT_ROOT = Rtt_Root.lower()
    else:
        RTT_ROOT = Rtt_Root

    if target_path.startswith(RTT_ROOT):
        return

    for item in program:
        walk_children(item)

    source_list.sort()

    # filte source file in RT-Thread
    target_list = []
    for src in source_list:
        if Env['PLATFORM'] == 'win32':
            src = src.lower()

        if src.startswith(RTT_ROOT):
            target_list.append(src)

    source_list = target_list

    for src in source_list:
        dst = src.replace(RTT_ROOT, '')
        if dst[0] == os.sep or dst[0] == '/':
            dst = dst[1:]
        print '=> ', dst
        dst = os.path.join(target_path, dst)
        do_copy_file(src, dst)

B
Bright Pan 已提交
801
    # copy tools directory
802 803 804 805
    print "=>  tools"
    do_copy_folder(os.path.join(RTT_ROOT, "tools"), os.path.join(target_path, "tools"))
    do_copy_file(os.path.join(RTT_ROOT, 'AUTHORS'), os.path.join(target_path, 'AUTHORS'))
    do_copy_file(os.path.join(RTT_ROOT, 'COPYING'), os.path.join(target_path, 'COPYING'))