LOCAL_MODULES = [

     #'my_extra_api_definitions',

     ]

 import sys

 import os

 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

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

 class ErrorHandler(pybindgen.settings.ErrorHandler):

     def handle_error(self, wrapper, exception, 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 == 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[2:])

     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)

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

             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])

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

     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()