## -*- Mode: python; py-indent-offset: 4; indent-tabs-mode: nil; coding: utf-8; -*-

# python lib modules

import sys

import shutil

import types

import optparse

import os.path

import re

import shlex

import textwrap

# WAF modules

import pproc as subprocess

import Options

import Logs

import TaskGen

import Constants

import ccroot

ccroot.USE_TOP_LEVEL = True

import Task

Task.algotype = Constants.JOBCONTROL # so that Task.maxjobs=1 takes effect

import Utils

import Build

import Configure

import Scripting

from utils import read_config_file

# By default, all modules will be enabled, examples will be disabled,

# and tests will be disabled.

modules_enabled  = ['all_modules']

examples_enabled = False

tests_enabled    = False

# Get the information out of the NS-3 configuration file.

config_file_exists = False

(config_file_exists, modules_enabled, examples_enabled, tests_enabled) = read_config_file()

sys.path.insert(0, os.path.abspath('waf-tools'))

try:

    import cflags # override the build profiles from waf

finally:

    sys.path.pop(0)

cflags.profiles = {

	# profile name: [optimization_level, warnings_level, debug_level]

	'debug':     [0, 2, 3],

	'optimized': [3, 2, 1],

	'release':   [3, 2, 0],

	}

cflags.default_profile = 'debug'

# local modules

import wutils

Configure.autoconfig = 1

# the following two variables are used by the target "waf dist"

VERSION = file("VERSION", "rt").read().strip()

APPNAME = 'ns'

wutils.VERSION = VERSION

wutils.APPNAME = APPNAME

# these variables are mandatory ('/' are converted automatically)

srcdir = '.'

blddir = 'build'

def load_env():

    bld_cls = getattr(Utils.g_module, 'build_context', Utils.Context)

    bld_ctx = bld_cls()

    bld_ctx.load_dirs(os.path.abspath(os.path.join (srcdir,'..')),

                      os.path.abspath(os.path.join (srcdir,'..', blddir)))

    bld_ctx.load_envs()

    env = bld_ctx.get_env()

    return env

def get_files(base_dir):

    retval = []

    reference=os.path.dirname(base_dir)

    for root, dirs, files in os.walk(base_dir):

        if root.find('.hg') != -1:

            continue

        for file in files:

            if file.find('.hg') != -1:

                continue

            fullname = os.path.join(root,file)

            # we can't use os.path.relpath because it's new in python 2.6

            relname = fullname.replace(reference + '/','')

            retval.append([fullname,relname])

    return retval

def dist_hook():

    import tarfile

    shutil.rmtree("doc/html", True)

    shutil.rmtree("doc/latex", True)

    shutil.rmtree("nsc", True)

def set_options(opt):

    # options provided by the modules

    opt.tool_options('compiler_cc')

    opt.tool_options('compiler_cxx')

    opt.tool_options('cflags')

    opt.add_option('--cwd',

                   help=('Set the working directory for a program.'),

                   action="store", type="string", default=None,

                   dest='cwd_launch')

    opt.add_option('--enable-gcov',

                   help=('Enable code coverage analysis.'

                         ' WARNING: this option only has effect '

                         'with the configure command.'),

                   action="store_true", default=False,

                   dest='enable_gcov')

    opt.add_option('--no-task-lines',

                   help=("Don't print task lines, i.e. messages saying which tasks are being executed by WAF."

                         "  Coupled with a single -v will cause WAF to output only the executed commands,"

                         " just like 'make' does by default."),

                   action="store_true", default=False,

                   dest='no_task_lines')

    opt.add_option('--lcov-report',

                   help=('Generate a code coverage report '

                         '(use this option at build time, not in configure)'),

                   action="store_true", default=False,

                   dest='lcov_report')

    opt.add_option('--run',

                   help=('Run a locally built program; argument can be a program name,'

                         ' or a command starting with the program name.'),

                   type="string", default='', dest='run')

    opt.add_option('--visualize',

                   help=('Modify --run arguments to enable the visualizer'),

                   action="store_true", default=False, dest='visualize')

    opt.add_option('--command-template',

                   help=('Template of the command used to run the program given by --run;'

                         ' It should be a shell command string containing %s inside,'

                         ' which will be replaced by the actual program.'),

                   type="string", default=None, dest='command_template')

    opt.add_option('--pyrun',

                   help=('Run a python program using locally built ns3 python module;'

                         ' argument is the path to the python program, optionally followed'

                         ' by command-line options that are passed to the program.'),

                   type="string", default='', dest='pyrun')

    opt.add_option('--valgrind',

                   help=('Change the default command template to run programs and unit tests with valgrind'),

                   action="store_true", default=False,

                   dest='valgrind')

    opt.add_option('--shell',

                   help=('DEPRECATED (run ./waf shell)'),

                   action="store_true", default=False,

                   dest='shell')

    opt.add_option('--enable-sudo',

                   help=('Use sudo to setup suid bits on ns3 executables.'),

                   dest='enable_sudo', action='store_true',

                   default=False)

    opt.add_option('--enable-tests',

                   help=('Build the ns-3 tests.'),

                   dest='enable_tests', action='store_true',

                   default=False)

    opt.add_option('--disable-tests',

                   help=('Do not build the ns-3 tests.'),

                   dest='disable_tests', action='store_true',

                   default=False)

    opt.add_option('--enable-examples',

                   help=('Build the ns-3 examples.'),

                   dest='enable_examples', action='store_true',

                   default=False)

    opt.add_option('--disable-examples',

                   help=('Do not build the ns-3 examples.'),

                   dest='disable_examples', action='store_true',

                   default=False)

    opt.add_option('--check',

                   help=('DEPRECATED (run ./test.py)'),

                   default=False, dest='check', action="store_true")

    opt.add_option('--enable-static',

                   help=('Compile NS-3 statically: works only on linux, without python'),

                   dest='enable_static', action='store_true',

                   default=False)

    opt.add_option('--enable-mpi',

                   help=('Compile NS-3 with MPI and distributed simulation support'),

                   dest='enable_mpi', action='store_true',

                   default=False)

    opt.add_option('--doxygen-no-build',

                   help=('Run doxygen to generate html documentation from source comments, '

                         'but do not wait for ns-3 to finish the full build.'),

                   action="store_true", default=False,

                   dest='doxygen_no_build')

    # options provided in subdirectories

    opt.sub_options('src')

    opt.sub_options('bindings/python')

    opt.sub_options('src/internet')

def _check_compilation_flag(conf, flag, mode='cxx'):

    """

    Checks if the C++ compiler accepts a certain compilation flag or flags

    flag: can be a string or a list of strings

    """

    env = conf.env.copy()

    if mode == 'cxx':

        fname = 'test.cc'

        env.append_value('CXXFLAGS', flag)

    else:

        fname = 'test.c'

        env.append_value('CCFLAGS', flag)

    try:

        retval = conf.run_c_code(code='#include <stdio.h>\nint main() { return 0; }\n',

                                 env=env, compile_filename=fname,

                                 compile_mode=mode, type='cprogram', execute=False)

    except Configure.ConfigurationError:

        ok = False

    else:

        ok = (retval == 0)

    conf.check_message_custom(flag, 'support', (ok and 'yes' or 'no'))

    return ok

def report_optional_feature(conf, name, caption, was_enabled, reason_not_enabled):

    conf.env.append_value('NS3_OPTIONAL_FEATURES', (name, caption, was_enabled, reason_not_enabled))

def configure(conf):

    # attach some extra methods

    conf.check_compilation_flag = types.MethodType(_check_compilation_flag, conf)

    conf.report_optional_feature = types.MethodType(report_optional_feature, conf)

    conf.env['NS3_OPTIONAL_FEATURES'] = []

    conf.env['NS3_BUILDDIR'] = conf.blddir

    conf.check_tool('compiler_cc')

    conf.check_tool('compiler_cxx')

    conf.check_tool('cflags', ['waf-tools'])

    try:

        conf.check_tool('pkgconfig', ['waf-tools'])

    except Configure.ConfigurationError:

        pass

    conf.check_tool('command', ['waf-tools'])

    # create the second environment, set the variant and set its name

    variant_env = conf.env.copy()

    variant_name = Options.options.build_profile

    if Options.options.enable_gcov:

        variant_name += '-gcov'

        variant_env.append_value('CCFLAGS', '-fprofile-arcs')

        variant_env.append_value('CCFLAGS', '-ftest-coverage')

        variant_env.append_value('CXXFLAGS', '-fprofile-arcs')

        variant_env.append_value('CXXFLAGS', '-ftest-coverage')

        variant_env.append_value('LINKFLAGS', '-fprofile-arcs')

    conf.env['NS3_ACTIVE_VARIANT'] = variant_name

    variant_env['NS3_ACTIVE_VARIANT'] = variant_name

    variant_env.set_variant(variant_name)

    conf.set_env_name(variant_name, variant_env)

    conf.setenv(variant_name)

    env = variant_env

    if Options.options.build_profile == 'debug':

        env.append_value('CXXDEFINES', 'NS3_ASSERT_ENABLE')

        env.append_value('CXXDEFINES', 'NS3_LOG_ENABLE')

    env['PLATFORM'] = sys.platform

    if conf.env['CXX_NAME'] in ['gcc', 'icc']:

        if Options.options.build_profile == 'release': 

            env.append_value('CXXFLAGS', '-fomit-frame-pointer') 

            if conf.check_compilation_flag('-march=native'):

                env.append_value('CXXFLAGS', '-march=native') 

        if sys.platform == 'win32':

            env.append_value("LINKFLAGS", "-Wl,--enable-runtime-pseudo-reloc")

        elif sys.platform == 'cygwin':

            env.append_value("LINKFLAGS", "-Wl,--enable-auto-import")

        cxx, = env['CXX']

        p = subprocess.Popen([cxx, '-print-file-name=libstdc++.so'], stdout=subprocess.PIPE)

        libstdcxx_location = os.path.dirname(p.stdout.read().strip())

        p.wait()

        if libstdcxx_location:

            conf.env.append_value('NS3_MODULE_PATH', libstdcxx_location)

        if Options.platform in ['linux']:

            if conf.check_compilation_flag('-Wl,--soname=foo'):

                env['WL_SONAME_SUPPORTED'] = True

    conf.sub_config('src')

    # Set the list of enabled modules.

    if Options.options.enable_modules:

        # Use the modules explicitly enabled. 

        conf.env['NS3_ENABLED_MODULES'] = ['ns3-'+mod for mod in

                                           Options.options.enable_modules.split(',')]

    else:

        # Use the enabled modules list from the ns3 configuration file.

        if modules_enabled[0] == 'all_modules':

            # Enable all modules if requested.

            conf.env['NS3_ENABLED_MODULES'] = conf.env['NS3_MODULES']

        else:

            # Enable the modules from the list.

            conf.env['NS3_ENABLED_MODULES'] = ['ns3-'+mod for mod in

                                               modules_enabled]

    conf.sub_config('bindings/python')

    conf.sub_config('src/mpi')

    # for suid bits

    conf.find_program('sudo', var='SUDO')

    why_not_sudo = "because we like it"

    if Options.options.enable_sudo and conf.env['SUDO']:

        env['ENABLE_SUDO'] = True

    else:

        env['ENABLE_SUDO'] = False

        if Options.options.enable_sudo:

            why_not_sudo = "program sudo not found"

        else:

            why_not_sudo = "option --enable-sudo not selected"

    conf.report_optional_feature("ENABLE_SUDO", "Use sudo to set suid bit", env['ENABLE_SUDO'], why_not_sudo)

    # Decide if tests will be built or not.

    if Options.options.enable_tests:

        # Tests were explicitly enabled. 

        env['ENABLE_TESTS'] = True

        why_not_tests = "option --enable-tests selected"

    elif Options.options.disable_tests:

        # Tests were explicitly disabled. 

        env['ENABLE_TESTS'] = False

        why_not_tests = "option --disable-tests selected"

    else:

        # Enable tests based on the ns3 configuration file.

        env['ENABLE_TESTS'] = tests_enabled

        if config_file_exists:

            why_not_tests = "based on configuration file"

        elif tests_enabled:

            why_not_tests = "defaults to enabled"

        else:

            why_not_tests = "defaults to disabled"

    conf.report_optional_feature("ENABLE_TESTS", "Build tests", env['ENABLE_TESTS'], why_not_tests)

    # Decide if examples will be built or not.

    if Options.options.enable_examples:

        # Examples were explicitly enabled. 

        env['ENABLE_EXAMPLES'] = True

        why_not_examples = "option --enable-examples selected"

    elif Options.options.disable_examples:

        # Examples were explicitly disabled. 

        env['ENABLE_EXAMPLES'] = False

        why_not_examples = "option --disable-examples selected"

    else:

        # Enable examples based on the ns3 configuration file.

        env['ENABLE_EXAMPLES'] = examples_enabled

        if config_file_exists:

            why_not_examples = "based on configuration file"

        elif examples_enabled:

            why_not_examples = "defaults to enabled"

        else:

            why_not_examples = "defaults to disabled"

    env['EXAMPLE_DIRECTORIES'] = []

    for dir in os.listdir('examples'):

        if dir.startswith('.') or dir == 'CVS':

            continue

        if os.path.isdir(os.path.join('examples', dir)):

            env['EXAMPLE_DIRECTORIES'].append(dir)

    conf.report_optional_feature("ENABLE_EXAMPLES", "Build examples", env['ENABLE_EXAMPLES'], 

                                 why_not_examples)

    conf.find_program('valgrind', var='VALGRIND')

    # These flags are used for the implicitly dependent modules.

    if env['ENABLE_STATIC_NS3']:

        if sys.platform == 'darwin':

            env.STATICLIB_MARKER = '-Wl,-all_load'

        else:

            env.STATICLIB_MARKER = '-Wl,--whole-archive,-Bstatic'

            env.SHLIB_MARKER = '-Wl,-Bdynamic,--no-whole-archive'

    have_gsl = conf.pkg_check_modules('GSL', 'gsl', mandatory=False)

    conf.env['ENABLE_GSL'] = have_gsl

    conf.report_optional_feature("GSL", "GNU Scientific Library (GSL)",

                                 conf.env['ENABLE_GSL'],

                                 "GSL not found")

    if have_gsl:

        conf.env.append_value('CXXDEFINES', "ENABLE_GSL")

        conf.env.append_value('CCDEFINES', "ENABLE_GSL")

    # for compiling C code, copy over the CXX* flags

    conf.env.append_value('CCFLAGS', conf.env['CXXFLAGS'])

    conf.env.append_value('CCDEFINES', conf.env['CXXDEFINES'])

    def add_gcc_flag(flag):

        if env['COMPILER_CXX'] == 'g++' and 'CXXFLAGS' not in os.environ:

            if conf.check_compilation_flag(flag, mode='cxx'):

                env.append_value('CXXFLAGS', flag)

        if env['COMPILER_CC'] == 'gcc' and 'CCFLAGS' not in os.environ:

            if conf.check_compilation_flag(flag, mode='cc'):

                env.append_value('CCFLAGS', flag)

    add_gcc_flag('-Wno-error=deprecated-declarations')

    add_gcc_flag('-fstrict-aliasing')

    add_gcc_flag('-Wstrict-aliasing')