blob: 940aa6e15e253eedd6ab7ebbec0c326eb937e8bf [file] [log] [blame]
Alexander Afanasyev381dea02011-11-03 08:33:26 -07001from pybindgen import Module, FileCodeSink, param, retval, cppclass, typehandlers
2
Alexander Afanasyev5713e7a2015-01-02 01:08:18 -08003from pybindgen.typehandlers.smart_ptr import StdSharedPtr
4
5from ns3_ptr import Ns3PtrMemoryPolicy
6
Alexander Afanasyev381dea02011-11-03 08:33:26 -07007import pybindgen.settings
8import warnings
9
Alexander Afanasyev381dea02011-11-03 08:33:26 -070010import sys
11
12def module_init():
Alexander Afanasyev6d98ac32012-06-06 13:01:48 -070013 root_module = Module('ns.ndnSIM', cpp_namespace='::ns3')
Alexander Afanasyev381dea02011-11-03 08:33:26 -070014 return root_module
15
16def register_types(module):
Alexander Afanasyev5713e7a2015-01-02 01:08:18 -080017 module.add_class('ObjectBase', allow_subclassing=True, import_from_module='ns.core')
18 module.add_class('SimpleRefCount', automatic_type_narrowing=True, import_from_module='ns.core',
19 template_parameters=['ns3::Object', 'ns3::ObjectBase', 'ns3::ObjectDeleter'],
20 parent=module['ns3::ObjectBase'],
21 memory_policy=cppclass.ReferenceCountingMethodsPolicy(incref_method='Ref', decref_method='Unref', peekref_method='GetReferenceCount'))
22 module.add_class('Object', import_from_module='ns.core', parent=module['ns3::SimpleRefCount< ns3::Object, ns3::ObjectBase, ns3::ObjectDeleter >'])
23
24 module.add_class('TypeId', import_from_module='ns.core')
25 module.add_class('AttributeValue', import_from_module='ns.core')
26
27 module.add_class('NodeContainer', import_from_module='ns.network')
28 module.add_class('Node', import_from_module='ns.network', parent=module['ns3::Object'])
29 module.add_class('ApplicationContainer', import_from_module='ns.network')
30
31 def reg_ndn(module):
32 module.add_class('StackHelper')
33 module.add_class('FibHelper')
34 module.add_class('StrategyChoiceHelper')
35 module.add_class('AppHelper')
36 module.add_class('GlobalRoutingHelper')
37
38 module.add_class('L3Protocol', parent=module.get_root()['ns3::Object'])
39
40 module.add_class('Name')
41 module.add_class('Interest')
42 module.add_class('Data')
43 module.add_class('Face', memory_policy=StdSharedPtr('ns3::ndn::Face'))
44 module.add_class('FaceContainer', memory_policy=Ns3PtrMemoryPolicy('::ns3::ndn::FaceContainer'))
45
46 def reg_name(module):
47 module.add_class('Component')
48 reg_name(module.add_cpp_namespace('name'))
49
50 def reg_nfd(module):
51 module.add_class('Forwarder', memory_policy=StdSharedPtr('::ns3::ndn::nfd::Forwarder'), is_singleton=True)
52 module.add_class('Fib')
53 module.add_class('Pit')
54 module.add_class('Cs')
55
56 def reg_fib(module):
57 module.add_class('Entry')#, memory_policy=StdSharedPtr('ns3::ndn::nfd::fib::Entry'))
58 module.add_class('NextHop')
59 module.add_class('NextHopList')
60 reg_fib(module.add_cpp_namespace('fib'))
61
62 def reg_pit(module):
63 module.add_class('Entry')#, memory_policy=StdSharedPtr('ns3::ndn::nfd::pit::Entry'))
64 reg_pit(module.add_cpp_namespace('pit'))
65
66 def reg_cs(module):
67 module.add_class('Entry')#, memory_policy=StdSharedPtr('ns3::ndn::nfd::cs::Entry'))
68 reg_cs(module.add_cpp_namespace('cs'))
69
70 reg_nfd(module.add_cpp_namespace('nfd'))
71 reg_ndn(module.add_cpp_namespace('ndn'))
Alexander Afanasyeva6cc9102013-07-15 18:44:24 -070072
Alexander Afanasyev381dea02011-11-03 08:33:26 -070073def register_methods(root_module):
Alexander Afanasyev5713e7a2015-01-02 01:08:18 -080074 reg_other_modules(root_module)
75
76 def reg_stackhelper(cls):
77 cls.add_constructor([])
78
79 cls.add_method('Install', 'ns3::Ptr<ns3::ndn::FaceContainer>', [param('ns3::Ptr<ns3::Node>', 'node')], is_const=True)
80 cls.add_method('Install', 'ns3::Ptr<ns3::ndn::FaceContainer>', [param('std::string const&', 'nodeName')], is_const=True)
81 cls.add_method('Install', 'ns3::Ptr<ns3::ndn::FaceContainer>', [param('const ns3::NodeContainer&', 'c')], is_const=True)
82 cls.add_method('InstallAll', 'ns3::Ptr<ns3::ndn::FaceContainer>', [], is_const=True)
83
84 cls.add_method('SetDefaultRoutes', retval('void'), [param('bool', 'isEnabled', default_value='true')], is_const=True)
85 cls.add_method('SetStackAttributes',
86 retval('void'),
87 [param('const std::string&', 'attr1', default_value='""'), param('const std::string&', 'value1', default_value='""'),
88 param('const std::string&', 'attr2', default_value='""'), param('const std::string&', 'value2', default_value='""'),
89 param('const std::string&', 'attr3', default_value='""'), param('const std::string&', 'value3', default_value='""'),
90 param('const std::string&', 'attr4', default_value='""'), param('const std::string&', 'value4', default_value='""')])
91
92 cls.add_method('setCsSize', retval('void'), [param('size_t', 'maxSize')])
93 cls.add_method('SetOldContentStore',
94 retval('void'),
95 [param('const std::string&', 'contentStoreClass'),
96 param('const std::string&', 'attr1', default_value='""'), param('const std::string&', 'value1', default_value='""'),
97 param('const std::string&', 'attr2', default_value='""'), param('const std::string&', 'value2', default_value='""'),
98 param('const std::string&', 'attr3', default_value='""'), param('const std::string&', 'value3', default_value='""'),
99 param('const std::string&', 'attr4', default_value='""'), param('const std::string&', 'value4', default_value='""')])
100 reg_stackhelper(root_module['ns3::ndn::StackHelper'])
101
102 def reg_fibhelper(cls):
103 cls.add_method('AddRoute', retval('void'), [
104 param('const std::string&', 'nodeName'), param('const std::string&', 'prefix'),
105 param('uint32_t', 'faceId'), param('int32_t', 'metric'),
106 ], is_const=True, is_static=True)
107 cls.add_method('AddRoute', retval('void'), [
108 param('ns3::Ptr<ns3::Node>', 'node'), param('const ns3::ndn::Name&', 'prefix'),
109 param('uint32_t', 'faceId'), param('int32_t', 'metric')
110 ], is_const=True, is_static=True)
111 cls.add_method('AddRoute', retval('void'), [
112 param('ns3::Ptr<ns3::Node>', 'node'), param('const ns3::ndn::Name&', 'prefix'),
113 param('std::shared_ptr<ns3::ndn::Face>', 'face'),
114 param('int32_t', 'metric'),
115 ], is_const=True, is_static=True)
116 cls.add_method('AddRoute', retval('void'), [
117 param('ns3::Ptr<ns3::Node>', 'node'), param('const ns3::ndn::Name&', 'prefix'),
118 param('ns3::Ptr<ns3::Node>', 'otherNode'),
119 param('int32_t', 'metric'),
120 ], is_const=True, is_static=True)
121 cls.add_method('AddRoute', retval('void'), [
122 param('const std::string&', 'nodeName'), param('const std::string&', 'prefix'),
123 param('const std::string&', 'otherNodeName'),
124 param('int32_t', 'metric'),
125 ], is_const=True, is_static=True)
126 reg_fibhelper(root_module['ns3::ndn::FibHelper'])
127
128 def reg_strategychoicehelper(cls):
129 cls.add_method('Install', retval('void'), [param('ns3::Ptr<ns3::Node>', 'node'),
130 param('const const std::string&', 'name'),
131 param('const const std::string&', 'strategy')], is_const=True, is_static=True)
132 cls.add_method('Install', retval('void'), [param('const ns3::NodeContainer&', 'c'),
133 param('const const std::string&', 'name'),
134 param('const const std::string&', 'strategy')], is_const=True, is_static=True)
135 cls.add_method('InstallAll', retval('void'), [param('const std::string&', 'name'),
136 param('const std::string&', 'strategy')], is_const=True, is_static=True)
137 reg_strategychoicehelper(root_module['ns3::ndn::StrategyChoiceHelper'])
138
139 def reg_apphelper(cls):
140 cls.add_constructor([param('const std::string&', 'prefix')])
141 cls.add_method('SetPrefix', 'void', [param('const std::string&', 'prefix')])
142 cls.add_method('SetAttribute', 'void', [param('std::string', 'name'), param('const ns3::AttributeValue&', 'value')])
143 cls.add_method('Install', 'ns3::ApplicationContainer', [param('ns3::NodeContainer', 'c')])
144 cls.add_method('Install', 'ns3::ApplicationContainer', [param('ns3::Ptr<ns3::Node>', 'node')])
145 cls.add_method('Install', 'ns3::ApplicationContainer', [param('std::string', 'nodeName')])
146 reg_apphelper(root_module['ns3::ndn::AppHelper'])
147
148 def reg_GlobalRoutingHelper(cls):
149 cls.add_constructor([])
150 cls.add_method('Install', 'void', [param('ns3::Ptr<ns3::Node>', 'node')])
151 cls.add_method('Install', 'void', [param('const ns3::NodeContainer&', 'nodes')])
152 cls.add_method('InstallAll', 'void', [])
153 cls.add_method('AddOrigin', 'void', [param('const std::string&', 'prefix'), param('ns3::Ptr<ns3::Node>', 'node')])
154 cls.add_method('AddOrigin', 'void', [param('const std::string&', 'prefix'), param('const std::string&', 'nodeName')])
155 cls.add_method('AddOrigins', 'void', [param('const std::string&', 'prefix'), param('const ns3::NodeContainer&', 'nodes')])
156 cls.add_method('AddOriginsForAll', 'void', [])
Alexander Afanasyev8e60bcd2015-01-15 20:55:40 +0000157 cls.add_method('CalculateRoutes', 'void', [])
158 cls.add_method('CalculateAllPossibleRoutes', 'void', [])
Alexander Afanasyev5713e7a2015-01-02 01:08:18 -0800159 reg_GlobalRoutingHelper(root_module['ns3::ndn::GlobalRoutingHelper'])
160
161 def reg_Name(root_module, cls):
162 cls.implicitly_converts_to(root_module['ns3::ndn::Interest'])
163 cls.add_output_stream_operator()
164 for op in ['==', '!=', '<', '<=', '>', '>=']:
165 cls.add_binary_comparison_operator(op)
166 cls.add_container_traits(retval('const ns3::ndn::name::Component&'),
167 begin_method='begin', end_method='end', iterator_type='const_iterator')
168
169 cls.add_constructor([])
170 cls.add_constructor([param('const ns3::ndn::Name&', 'other')])
171 cls.add_constructor([param('const std::string&', 'url')])
172 cls.add_method('append', 'ns3::ndn::Name &', [param('const ns3::ndn::name::Component&', 'comp')])
173 cls.add_method('get', 'const ns3::ndn::name::Component&', [param('int', 'index')], is_const=True)
174 cls.add_method('getPrefix', 'ns3::ndn::Name', [param('size_t', 'len')], is_const=True)
175 cls.add_method('size', 'size_t', [], is_const=True)
176 cls.add_method('toUri', retval('std::string'), [], is_const=True)
177 reg_Name(root_module, root_module['ns3::ndn::Name'])
178
179 def reg_NameComponent(cls):
180 cls.add_output_stream_operator()
181 for op in ['==', '!=', '<', '<=', '>', '>=']:
182 cls.add_binary_comparison_operator(op)
183
184 cls.add_constructor([param('const ns3::ndn::name::Component&', 'arg0')])
185 cls.add_constructor([])
186 cls.add_method('fromNumber', 'ns3::ndn::name::Component', [param('uint64_t', 'number')])
187 cls.add_method('fromNumberWithMarker', 'ns3::ndn::name::Component', [param('uint64_t', 'number'), param('unsigned char', 'marker')])
188 cls.add_method('fromEscapedString', 'ns3::ndn::name::Component', [param('const std::string&', 'uri')])
189 reg_NameComponent(root_module['ns3::ndn::name::Component'])
190
191 def reg_Interest(cls):
192 cls.add_output_stream_operator()
193
194 cls.add_constructor([param('const ns3::ndn::Interest&', 'interest')])
195 cls.add_constructor([])
196 reg_Interest(root_module['ns3::ndn::Interest'])
197
198 def reg_Data(cls):
199 cls.add_output_stream_operator()
200
201 cls.add_constructor([param('const ns3::ndn::Data&', 'data')])
202 cls.add_constructor([])
203 reg_Data(root_module['ns3::ndn::Data'])
204
205 #########################################################################################
206 ## Interface to NFD
207 #########################################################################################
208
209 def register_L3Protocol(cls):
210 cls.add_method('getL3Protocol', 'ns3::Ptr<ns3::ndn::L3Protocol>', [param('ns3::Ptr<ns3::Object>', 'node')], is_static=True)
211 cls.add_method('getForwarder', 'std::shared_ptr<ns3::ndn::nfd::Forwarder>', [])
212 register_L3Protocol(root_module['ns3::ndn::L3Protocol'])
213
214 def reg_Face(cls):
215 cls.add_output_stream_operator()
216 cls.add_method('getId', retval('int64_t'), [], is_const=True)
217 reg_Face(root_module['ns3::ndn::Face'])
218
219 def reg_NfdForwarder(cls):
220 cls.add_method('getFib', retval('const ns3::ndn::nfd::Fib&', caller_manages_return=False), [], is_const=True)
221 cls.add_method('getPit', retval('const ns3::ndn::nfd::Pit&', caller_manages_return=False), [], is_const=True)
222 cls.add_method('getCs', retval('const ns3::ndn::nfd::Cs&', caller_manages_return=False), [], is_const=True)
223 reg_NfdForwarder(root_module['ns3::ndn::nfd::Forwarder'])
224
225 #############
226 #### FIB ####
227 def reg_NfdFib(root_module, cls):
228 cls.add_method('size', retval('size_t'), [], is_const=True)
229 cls.add_container_traits(retval('const ns3::ndn::nfd::fib::Entry&', caller_manages_return=False),
230 begin_method='begin', end_method='end', iterator_type='const_iterator')
231
232 # The following is not supported
233 # cls.add_method('findLongestPrefixMatch', retval('std::shared_ptr<ns3::ndn::nfd::fib::Entry>'),
234 # [param('const ns3::ndn::Name&', 'prefix')], is_const=True)
235 # cls.add_method('findExactMatch', retval('std::shared_ptr<ns3::ndn::nfd::fib::Entry>'),
236 # [param('const ns3::ndn::Name&', 'prefix')], is_const=True)
237 # cls.add_method('findLongestPrefixMatch', retval('shared_ptr<ns3::ndn::nfd::fib::Entry>'),
238 # [param('const pit::Entry&', 'pitEntry')], is_const=True)
239 # cls.add_method('findLongestPrefixMatch', retval('shared_ptr<ns3::ndn::nfd::fib::Entry>'),
240 # [param('const measurements::Entry&', 'measurementsEntry')], is_const=True)
241
242 # cls.add_method('insert', retval('std::pair<std::shared_ptr<ns3::ndn::nfd::fib::Entry>, bool>'), [param('const ns3::ndn::Name&', 'prefix')])
243 cls.add_method('erase', retval('void'), [param('const ns3::ndn::Name&', 'prefix')])
244 cls.add_method('erase', retval('void'), [param('const ns3::ndn::nfd::fib::Entry&', 'entry')])
245 cls.add_method('removeNextHopFromAllEntries', retval('void'), [param('std::shared_ptr<ns3::ndn::Face>', 'face')])
246
247 def reg_Entry(cls):
248 cls.add_method('getPrefix', 'const ns3::ndn::Name&', [], is_const=True)
249 cls.add_method('getNextHops', retval('const ns3::ndn::nfd::fib::NextHopList&', caller_manages_return=False), [], is_const=True)
250 cls.add_method('hasNextHops', 'bool', [], is_const=True)
251 reg_Entry(root_module['ns3::ndn::nfd::fib::Entry'])
252
253 def reg_NextHop(cls):
254 cls.add_constructor([param('std::shared_ptr<ns3::ndn::Face>', 'face')])
255
256 cls.add_method('getFace', 'std::shared_ptr<ns3::ndn::Face>', [], is_const=True)
257 cls.add_method('setCost', 'void', [param('uint64_t', 'cost')])
258 cls.add_method('getCost', 'uint64_t', [], is_const=True)
259 reg_NextHop(root_module['ns3::ndn::nfd::fib::NextHop'])
260
261 def reg_NextHopList(cls):
262 cls.add_method('size', retval('size_t'), [], is_const=True)
263 cls.add_container_traits(retval('const ns3::ndn::nfd::fib::NextHop&', caller_manages_return=False),
264 begin_method='begin', end_method='end', iterator_type='const_iterator')
265 reg_NextHopList(root_module['ns3::ndn::nfd::fib::NextHopList'])
266 reg_NfdFib(root_module, root_module['ns3::ndn::nfd::Fib'])
267 #### FIB ####
268 #############
269
270 #############
271 #### PIT ####
272 def reg_NfdPit(root_module, cls):
273 cls.add_method('size', retval('size_t'), [], is_const=True)
274 cls.add_container_traits(retval('const ns3::ndn::nfd::pit::Entry&', caller_manages_return=False),
275 begin_method='begin', end_method='end', iterator_type='const_iterator')
276
277 def reg_Entry(cls):
278 cls.add_method('getInterest', retval('const ns3::ndn::Interest&'), [], is_const=True)
279 cls.add_method('getName', retval('const ns3::ndn::Name&'), [], is_const=True)
280 reg_Entry(root_module['ns3::ndn::nfd::pit::Entry'])
281 reg_NfdPit(root_module, root_module['ns3::ndn::nfd::Pit'])
282 #### PIT ####
283 #############
284
285 #############
286 #### CS ####
287 def reg_NfdCs(root_module, cls):
288 cls.add_method('size', retval('size_t'), [], is_const=True)
289 cls.add_container_traits(retval('const ns3::ndn::nfd::cs::Entry&', caller_manages_return=False),
290 begin_method='begin', end_method='end', iterator_type='const_iterator')
291
292 def reg_Entry(cls):
293 cls.add_method('getName', retval('const ns3::ndn::Name&'), [], is_const=True)
294 reg_Entry(root_module['ns3::ndn::nfd::cs::Entry'])
295 reg_NfdCs(root_module, root_module['ns3::ndn::nfd::Cs'])
296 #### CS ####
297 #############
298
299def reg_other_modules(root_module):
300 def reg_ApplicationContainer(cls):
301 cls.add_constructor([])
302 cls.add_constructor([param('ns3::ApplicationContainer', 'container')])
303 reg_ApplicationContainer(root_module['ns3::ApplicationContainer'])
Alexander Afanasyevcf6dc922012-08-10 16:55:27 -0700304
Alexander Afanasyev381dea02011-11-03 08:33:26 -0700305def register_functions(root_module):
Alexander Afanasyeva6cc9102013-07-15 18:44:24 -0700306 return
307
Alexander Afanasyev381dea02011-11-03 08:33:26 -0700308def main():
309 out = FileCodeSink(sys.stdout)
310 root_module = module_init()
311 register_types(root_module)
312 register_methods(root_module)
313 register_functions(root_module)
314 root_module.generate(out)
315
316if __name__ == '__main__':
317 main()