LOCAL_MODULES = [

    #'my_extra_api_definitions',

    ]

import sys

import os

sys.path.insert(0, sys.argv[2])

from pybindgen import FileCodeSink, write_preamble

from pybindgen.module import MultiSectionFactory

import pybindgen.settings

from ns3modulegen_generated import module_init, register_types, register_methods, register_functions

import ns3modulegen_core_customizations

import callbacks_list

import traceback

this_script_dir = os.path.dirname(os.path.abspath(sys.argv[0]))

class ErrorHandler(pybindgen.settings.ErrorHandler):

    def handle_error(self, wrapper, exception, traceback_):

        print >> sys.stderr

        print >> sys.stderr, "---- location:"

        traceback.print_stack()

        print >> sys.stderr, "---- error:"

        traceback.print_tb(traceback_)

        try:

            stack = wrapper.stack_where_defined

        except AttributeError:

            print >> sys.stderr, "??:??: %s / %r" % (wrapper, exception)

        else:

            stack = list(stack)

            stack.reverse()

            for (filename, line_number, function_name, text) in stack:

                file_dir = os.path.dirname(os.path.abspath(filename))

                if file_dir.startswith(this_script_dir):

                    print >> sys.stderr, "%s:%i: %r" % (os.path.join("..", "bindings", "python", os.path.basename(filename)),

                                                        line_number, exception)

                    break

        return True

pybindgen.settings.error_handler = ErrorHandler()

pybindgen.settings.wrapper_registry = pybindgen.settings.StdMapWrapperRegistry

class MyMultiSectionFactory(MultiSectionFactory):

    def __init__(self, main_file_name, modules):

        super(MyMultiSectionFactory, self).__init__()

        self.main_file_name = main_file_name

        self.main_sink = FileCodeSink(open(main_file_name, "wt"))

        self.header_name = "ns3module.h"

        header_file_name = os.path.join(os.path.dirname(self.main_file_name), self.header_name)

        self.header_sink = FileCodeSink(open(header_file_name, "wt"))

        self.section_sinks = {'__main__': self.main_sink}

        for module in modules:

            section_name = 'ns3_module_%s' % module.replace('-', '_')

            file_name = os.path.join(os.path.dirname(self.main_file_name), "%s.cc" % section_name)

            sink = FileCodeSink(open(file_name, "wt"))

            self.section_sinks[section_name] = sink            

    def get_section_code_sink(self, section_name):

        return self.section_sinks[section_name]

    def get_main_code_sink(self):

        return self.main_sink

    def get_common_header_code_sink(self):

        return self.header_sink

    def get_common_header_include(self):

        return '"%s"' % self.header_name

    def close(self):

        self.header_sink.file.close()

        self.main_sink.file.close()

        for sink in self.section_sinks.itervalues():

            sink.file.close()

def main():

    out = MyMultiSectionFactory(sys.argv[1], sys.argv[3:])

    root_module = module_init()

    root_module.add_include('"everything.h"')

    register_types(root_module)

    ns3modulegen_core_customizations.Simulator_customizations(root_module)

    ns3modulegen_core_customizations.CommandLine_customizations(root_module)

    ns3modulegen_core_customizations.TypeId_customizations(root_module)

    ns3modulegen_core_customizations.add_std_ofstream(root_module)

    ns3modulegen_core_customizations.add_ipv4_address_tp_hash(root_module)

    for local_module in LOCAL_MODULES:

        mod = __import__(local_module)

        mod.register_types(root_module)

    ns3modulegen_core_customizations.generate_callback_classes(root_module.after_forward_declarations,

                                                               callbacks_list.callback_classes)

    register_methods(root_module)

    for local_module in LOCAL_MODULES:

        mod = __import__(local_module)

        mod.register_methods(root_module)

    ns3modulegen_core_customizations.Object_customizations(root_module)

    ns3modulegen_core_customizations.Attribute_customizations(root_module)

    register_functions(root_module)

    for local_module in LOCAL_MODULES:

        mod = __import__(local_module)

        mod.register_functions(root_module)

    enabled_features = os.environ['NS3_ENABLED_FEATURES'].split(',')

    # if GtkConfigStore support is disabled, disable the class wrapper

    if 'GtkConfigStore' not in enabled_features:

        try:

            root_module.classes.remove(root_module['ns3::GtkConfigStore'])

        except KeyError:

            pass

    # if no sqlite, the class SqliteDataOutput is disabled

    if 'SqliteDataOutput' not in enabled_features:

        try:

            root_module.classes.remove(root_module['ns3::SqliteDataOutput'])

        except KeyError:

            pass

    if 'Threading' not in enabled_features:

        for clsname in ['SystemThread', 'SystemMutex', 'SystemCondition', 'CriticalSection',

                        'SimpleRefCount< ns3::SystemThread, ns3::empty, ns3::DefaultDeleter<ns3::SystemThread> >']:

            root_module.classes.remove(root_module['ns3::%s' % clsname])

    if 'EmuNetDevice' not in enabled_features:

        for clsname in ['EmuNetDevice', 'EmuHelper']:

            root_module.classes.remove(root_module['ns3::%s' % clsname])

        root_module.enums.remove(root_module['ns3::EmuNetDevice::EncapsulationMode'])

    if 'RealTime' not in enabled_features:

        for clsname in ['WallClockSynchronizer', 'RealtimeSimulatorImpl']:

            root_module.classes.remove(root_module['ns3::%s' % clsname])

        root_module.enums.remove(root_module['ns3::RealtimeSimulatorImpl::SynchronizationMode'])

    if 'TapBridge' not in enabled_features:

        for clsname in ['TapBridge', 'TapBridgeHelper']:

            root_module.classes.remove(root_module['ns3::%s' % clsname])

        root_module.enums.remove(root_module['ns3::TapBridge::Mode'])

    root_module.generate(out, '_ns3')

    out.close()

if __name__ == '__main__':

    if 0:

        try:

            import cProfile as profile

        except ImportError:

            main()

        else:

            print >> sys.stderr, "** running under profiler"

            profile.run('main()', 'ns3modulegen.pstat')

    else:

        main()