108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
## -*- Mode: python; py-indent-offset: 4; indent-tabs-mode: nil; coding: utf-8; -*-
|
|
|
|
import os, os.path
|
|
import shutil
|
|
import types
|
|
|
|
import Action
|
|
import Common
|
|
import Object
|
|
import Params
|
|
|
|
|
|
all_modules = (
|
|
'core',
|
|
'common',
|
|
'simulator',
|
|
'node',
|
|
'internet-node',
|
|
'devices/point-to-point',
|
|
'devices/csma',
|
|
'applications',
|
|
'routing/global-routing',
|
|
'mobility',
|
|
)
|
|
|
|
def set_options(opt):
|
|
opt.sub_options('simulator')
|
|
|
|
opt.add_option('--enable-rpath',
|
|
help=("Link programs with rpath"
|
|
" (normally not needed, see "
|
|
" --run and --shell; moreover, only works in some"
|
|
" specific platforms, such as Linux and Solaris)"),
|
|
action="store_true", dest='enable_rpath', default=False)
|
|
|
|
|
|
def configure(conf):
|
|
conf.sub_config('core')
|
|
conf.sub_config('simulator')
|
|
|
|
blddir = os.path.abspath(os.path.join(conf.m_blddir, conf.env.variant()))
|
|
conf.env['NS3_MODULE_PATH'] = [os.path.join(blddir)]
|
|
if Params.g_options.enable_rpath:
|
|
conf.env.append_value('RPATH', '-Wl,-rpath=%s' % (os.path.join(blddir),))
|
|
|
|
## Used to link the 'run-tests' program with all of ns-3 code
|
|
conf.env['NS3_MODULES'] = ['ns3-' + module.split('/')[-1] for module in all_modules]
|
|
|
|
def create_ns3_module(bld, name, dependencies=()):
|
|
module = bld.create_obj('cpp', 'objects')
|
|
module.name = 'ns3-' + name
|
|
module.target = module.name
|
|
module.add_objects = ['ns3-' + dep for dep in dependencies]
|
|
module.env.append_value('CXXFLAGS', module.env['shlib_CXXFLAGS'])
|
|
return module
|
|
|
|
|
|
def build(bld):
|
|
Object.register('ns3header', Ns3Header)
|
|
Action.Action('ns3header', func=_ns3_headers_inst, color='BLUE')
|
|
bld.create_ns3_module = types.MethodType(create_ns3_module, bld)
|
|
|
|
bld.add_subdirs(list(all_modules))
|
|
|
|
|
|
class Ns3Header(Object.genobj):
|
|
"""A set of NS-3 header files"""
|
|
def __init__(self, env=None):
|
|
Object.genobj.__init__(self, 'other')
|
|
self.inst_var = 'INCLUDEDIR'
|
|
self.inst_dir = 'ns3'
|
|
self.env = env
|
|
if not self.env:
|
|
self.env = Params.g_build.m_allenvs['default']
|
|
|
|
def apply(self):
|
|
ns3_dir_node = Params.g_build.m_srcnode.find_dir("ns3")
|
|
for filename in self.to_list(self.source):
|
|
src_node = self.path.find_source(filename)
|
|
if src_node is None:
|
|
Params.fatal("source ns3 header file %s not found" % (filename,))
|
|
dst_node = ns3_dir_node.find_build(os.path.basename(filename))
|
|
assert dst_node is not None
|
|
task = self.create_task('ns3header', self.env, 1)
|
|
task.set_inputs([src_node])
|
|
task.set_outputs([dst_node])
|
|
|
|
def install(self):
|
|
for i in self.m_tasks:
|
|
current = Params.g_build.m_curdirnode
|
|
lst = map(lambda a: a.relpath_gen(current), i.m_outputs)
|
|
Common.install_files(self.inst_var, self.inst_dir, lst)
|
|
|
|
def _ns3_headers_inst(task):
|
|
assert len(task.m_inputs) == len(task.m_outputs)
|
|
inputs = [node.srcpath(task.m_env) for node in task.m_inputs]
|
|
outputs = [node.bldpath(task.m_env) for node in task.m_outputs]
|
|
for src, dst in zip(inputs, outputs):
|
|
try:
|
|
os.chmod(dst, 0600)
|
|
except OSError:
|
|
pass
|
|
shutil.copy2(src, dst)
|
|
## make the headers in builddir read-only, to prevent
|
|
## accidental modification
|
|
os.chmod(dst, 0400)
|
|
return 0
|