Mike Owen
37dcfcb514
building yet 'cause I don't understand waf. Why does every project I look at use a different build system!
49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
import os
|
|
from pybindgen import *
|
|
|
|
#-------------------------------------------------------------------------------
|
|
# This is the implementation of the multi-section factory that allows us to
|
|
# write multiple source files for compiling.
|
|
#
|
|
# This is directly cribbed from Gustavo's example from the ns3 package at
|
|
# http://code.nsnam.org/ns-3-dev/file/e15adc7172f1/bindings/python/ns3modulegen.py
|
|
#-------------------------------------------------------------------------------
|
|
class ModuleGenerationFactory(module.MultiSectionFactory):
|
|
|
|
def __init__(self, main_file_name, modules):
|
|
print "main file name: ", main_file_name
|
|
print "modules: ", modules
|
|
self.basename, ext = os.path.splitext(main_file_name)
|
|
super(ModuleGenerationFactory, self).__init__()
|
|
self.main_file_name = main_file_name
|
|
self.main_sink = FileCodeSink(open(main_file_name, "wt"))
|
|
self.header_name = self.basename + ".hh"
|
|
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 = '%s_%s' % (self.basename, module.replace('-', '_'))
|
|
file_name = os.path.join(os.path.dirname(self.main_file_name), "%s.C" % 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()
|