SConstruct

import os
import platform
import sys

import SCons

# ============================ banner ==========================================

banner = []
banner += [""]
banner += [" ___                 _ _ _  ___  _ _ "]
banner += ["| . | ___  ___ ._ _ | | | |/ __>| \ |"]
banner += ["| | || . \/ ._>| ' || | | |\__ \|   |"]
banner += ["`___'|  _/\___.|_|_||__/_/ <___/|_\_|"]
banner += ["     |_|                  openwsn.org"]
banner += [""]
banner = '\n'.join(banner)
print banner

# ===== help text

Help('''
Usage:
    scons [<variable>=<value> ...] <project>
    scons docs
    scons [help-option]

project:
    A project is represented by a subdirectory of the projects directory, for
    a particular board. For example, the 'oos_openwsn' project may be built for
    telosb. To specify a project, exclude the leading digits in the directory
    name, like '03' for oos_openwsn.

    variable=value pairs
    These pairs qualify how the project is built, and are organized here into
    functional groups. Below each variable's description are the valid 
    options, with the default value listed first.
    
    board          Board to build for. 'python' is for software simulation.
                   telosb, wsn430v14, wsn430v13b, gina, z1, python,
                   iot-lab_M3, iot-lab_A8-M3, nrf52840

    version        Board version
        
    toolchain      Toolchain implementation. The 'python' board requires gcc
                   (MinGW on Windows build host).
                   mspgcc, iar, iar-proj, gcc

    Software modules/apps to include and stack/board configuration:
    modules        A comma, separated list of modules to include in the build.
    apps           A comma, separated list of apps to include in the build.
    stackcfg       A comma, separated list of stack configuration options.
    boardopt       A comma, separated list of board options.

    Connected hardware variables:
    bootload       Location of the board to bootload the binary on. 
                   COMx for Windows, /dev entries for Linux
                   Supports parallel operation with a comma-separated list,
                   for example 'COM5,COM6,COM7'.
    jtag           Location of the board to JTAG the binary to.
                   COMx for Windows, /dev entry for Linux
    fet_version    Firmware version running on the MSP-FET430uif for jtag.
                   2, 3
    
    These simulation variables are for a cross-platform build, and are valid
    only from an amd64-linux build host.
    simhost        Host platform and OS for simulation. Default selection is
                   the current platform/OS, which of course is not a cross-
                   build. '-windows' cross-builds require MinGW-w64 toolchain.
                   amd64-linux, x86-linux, amd64-windows, x86-windows
    simhostpy      Home directory for simhost cross-build Python headers and 
                   shared library.
    
    Common variables:
    oflag          Change the compiler optimization level. Default is -O0.

    verbose        Print each complete compile/link command.
                   0 (off), 1 (on)
    
docs:
    Generate source documentation in build{0}docs{0}html directory

help-option:
    --help       Display help text. Also display when no parameters to the
                 scons scommand.
'''.format(os.sep))

# ============================ options =========================================

# first value is default
command_line_options = {
    'board': [
        # MSP430
        'telosb',
        'gina',
        'wsn430v13b',
        'wsn430v14',
        'z1',
        # Cortex-M3
        'openmote-cc2538',
        'openmote-b',
        'openmote-b-24ghz',
        'openmote-b-subghz',
        'silabs-ezr32wg',
        'openmotestm',
        'iot-lab_M3',
        'iot-lab_A8-M3',
        'agilefox',
        'samr21_xpro',
        # Cortex-M0
        'scum',
        # Nordic nRF52840 (Cortex-M4), use revision=DK or revision=DONGLE
        'nrf52840',
        # misc.
        'python',
    ],
    'toolchain': [
        'mspgcc',
        'iar',
        'iar-proj',
        'armgcc',
        'gcc',
    ],
    'logging': [str(l) for l in range(6)],
    'apps': ['c6t', 'cexample', 'cinfo', 'cinfrared', 'cled', 'csensors', 'cstorm', 'cwellknown', 'rrt', 'uecho',
             'uexpiration', 'uexp-monitor', 'uinject', 'userialbridge', 'cjoin', ''],
    'modules': ['coap', 'udp', 'fragmentation', 'icmpv6echo', 'l2-security', ''],
    'stackcfg': ['adaptive-msf', 'dagroot', 'channel', 'pktqueue', 'panid', ''],
    'boardopt' : ['hw-crypto', 'printf', 'fastsim', ''],
    'fet_version': ['2', '3'],
    'verbose': ['0', '1'],
    'simhost': ['amd64-linux', 'x86-linux', 'amd64-windows', 'x86-windows'],
    'simhostpy': [''],  # No reasonable default
    'atmel_24ghz': ['0', '1'],
    'oflag': ['-O0', '-O1', '-O2', '-O3', '-Os'],
    'revision': ['']
}


def validate_option(key, value, env):
    if key not in command_line_options:
        print "Unknown switch {0}.".format(key)
        Exit(-1)

    if key == 'modules' or key == 'apps':
        values = value.split(',')
    elif key == 'stackcfg':
        values = value.split(',')
    else:
        values = [value]

    for v in values:
        if ':' in v:
            v = v.split(':')[0]
        if v not in command_line_options[key]:
            print "Unknown {0} \"{1}\". Options are: {2}.\n\n".format(key, v, ', '.join(
                command_line_options[key]))
            Exit(-1)


# Define default value for simhost option
if os.name == 'nt':
    defaultHost = 2
else:
    defaultHost = 0 if platform.architecture()[0] == '64bit' else 1

command_line_vars = Variables()
command_line_vars.AddVariables(
    (
        'board',  # key
        '',  # help
        command_line_options['board'][0],  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'toolchain',  # key
        '',  # help
        command_line_options['toolchain'][0],  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'logging',  # key
        '',  # help
        command_line_options['logging'][5],  # default
        validate_option, # validator
        None,  # converter
    ),
    (
        'oflag',  # key
        '',  # help
        command_line_options['oflag'][0],  # default
        validate_option, # validator
        None,  # converter
    ),
    (
        'jtag',  # key
        '',  # help
        '',  # default
        None,  # validator
        None,  # converter
    ),
    (
        'fet_version',  # key
        '',  # help
        command_line_options['fet_version'][0],  # default
        validate_option,  # validator
        int,  # converter
    ),
    (
        'bootload',  # key
        '',  # help
        '',  # default
        None,  # validator
        None,  # converter
    ),
    (
        'verbose',  # key
        '',  # help
        command_line_options['verbose'][0],  # default
        validate_option,  # validator
        int,  # converter
    ),
    (
        'modules',  # key
        '',  # help
        '',  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'apps',  # key
        '',  # help
        '',  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'stackcfg',  # key
        '',  # help
        '',  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'boardopt',  # key
        '',  # help
        '',  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'simhost',  # key
        '',  # help
        command_line_options['simhost'][defaultHost],  # default
        validate_option,  # validator
        None,  # converter
    ),
    (
        'simhostpy',  # key
        '',  # help
        command_line_options['simhostpy'][0],  # default
        None,  # validator
        None,  # converter
    ),
    (
        'atmel_24ghz',  # key
        '',  # help
        command_line_options['atmel_24ghz'][0],  # default
        validate_option,  # validator
        int,  # converter
    ),
    (
        'revision',  # key
        'board revision',  # help
        command_line_options['revision'][0],  # default
        None,  # validator
        None,  # converter
    ),
)

if os.name == 'nt':
    env = Environment(
        tools=['mingw'],
        variables=command_line_vars
    )
else:
    simhost = ARGUMENTS.get('simhost', 'none')
    board = ARGUMENTS.get('board', 'none')
    if board == 'python' and simhost.endswith('-windows'):
        # Cross-compile for simulation on Windows
        env = Environment(
            # crossMingw64 requires 'mingw_prefer_amd64' key
            tools=['crossMingw64'],
            mingw_prefer_amd64=simhost.startswith('amd64-'),
            variables=command_line_vars
        )
    else:
        env = Environment(variables=command_line_vars)


def default(env, target, source):
    print SCons.Script.help_text


Default(env.Command('default', None, default))

# ============================ verbose =========================================

if not env['verbose']:
    env['CCCOMSTR'] = "Compiling          $TARGET"
    env['SHCCCOMSTR'] = "Compiling (shared) $TARGET"
    env['ARCOMSTR'] = "Archiving          $TARGET"
    env['RANLIBCOMSTR'] = "Indexing           $TARGET"
    env['LINKCOMSTR'] = "Linking            $TARGET"
    env['SHLINKCOMSTR'] = "Linking (shared)   $TARGET"

# ============================ load SConscript's ===============================

# Initialize targets
env['targets'] = {'all': [], 'all_std': [], 'all_bsp': [], 'all_drv': [], 'all_oos': []}

# Include docs SConscript
env.SConscript(os.path.join('docs', 'SConscript'), exports=['env'])

# Include main SConscript which will discover targets for this board/toolchain
env.SConscript('SConscript', exports=['env'])

# declare target group alias
for k, v in env['targets'].items():
    Alias(k, v)

# ============================ admin targets ===================================

# ===== list

def list_function(env, target, source):
    output = []
    output += ['\n']
    output += ['Available targets for board={0} toolchain={1}'.format(env['board'], env['toolchain'])]
    output += ['\n']
    for k, v in env['targets'].items():
        output += [' - {0}'.format(k)]
        for t in v:
            output += ['    - {0}'.format(t)]
    output = '\n'.join(output)
    print output


list_output = env.Command('list', None, list_function)
AlwaysBuild(list_output)
Alias('list', list_output)


# ===== env

def env_function(env, target, source):
    print env.Dump()

env_command = env.Command('env', None, env_function)
AlwaysBuild(env_command)
Alias('env', env_command)