# Copyright (C) 2013 INRIA
#
# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
+# it under the terms of the GNU General Public License version 2 as
+# published by the Free Software Foundation;
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
#
# Author: Alina Quereilhac <alina.quereilhac@inria.fr>
+#
+# Instructions to automatically generate ns-3 ResourceManagers
+#
+# Configure the ns-3 enviorment (e.g.):
+#
+# export PYTHONPATH=~/.nepi/nepi-usr/bin/ns-3/ns-3.20/optimized/build/lib/python/site-packages
+# export LD_LIBRARY_PATH=~/.nepi/nepi-usr/bin/ns-3/ns-3.20/optimized/build/lib
+#
+# Run the RM generator:
+#
+# PYTHONPATH=$PYTHONPATH:~/repos/nepi/src python src/nepi/resources/ns3/resource_manager_generator.py
+#
+
+from __future__ import print_function
+
# Force the load of ns3 libraries
from nepi.resources.ns3.ns3wrapper import load_ns3_module
import os
import re
-def discard(ns3, tid):
+adapted_types = ["ns3::Node",
+ "ns3::Icmpv4L4Protocol",
+ "ns3::ArpL3Protocol",
+ "ns3::Ipv4L3Protocol",
+ "ns3::PropagationLossModel",
+ "ns3::MobilityModel",
+ "ns3::PropagationDelayModel",
+ "ns3::WifiRemoteStationManager",
+ "ns3::WifiNetDevice",
+ "ns3::WifiChannel",
+ "ns3::WifiPhy",
+ "ns3::WifiMac",
+ "ns3::ErrorModel",
+ "ns3::ErrorRateModel",
+ "ns3::Application",
+ "ns3::FdNetDevice",
+ #"ns3::DceApplication",
+ "ns3::NetDevice",
+ "ns3::Channel",
+ "ns3::Queue"]
+
+base_types = ["ns3::IpL4Protocol"]
+
+def select_base_class(ns3, tid):
+ base_class_import = None
+ base_class = None
+
rtype = tid.GetName()
- words = ["variable", "object", "probe", "adaptor", "wrapper",
- "container", "derived", "simple"]
- for word in words:
- if rtype.lower().find(word) > -1:
- return True
-
- bases = ["ns3::Scheduler", "ns3::SimulatorImpl"]
type_id = ns3.TypeId()
- for base in bases:
- tid_base = type_id.LookupByName(base)
- if tid.IsChildOf(tid_base):
- return True
- return False
+ for type_name in adapted_types:
+ tid_base = type_id.LookupByName(type_name)
+ if type_name == rtype or tid.IsChildOf(tid_base):
+ base_class = "NS3Base" + type_name.replace("ns3::", "")
+ base_module = "ns3" + type_name.replace("ns3::", "").lower()
+ base_class_import = "from nepi.resources.ns3.%s import %s " % (
+ base_module, base_class)
+ return (base_class_import, base_class)
-def select_base_class(ns3, tid):
base_class_import = "from nepi.resources.ns3.ns3base import NS3Base"
base_class = "NS3Base"
-
- rtype = tid.GetName()
- if rtype == "ns3::Node":
- base_class_import = "from nepi.resources.ns3.ns3node import NS3BaseNode"
- base_class = "NS3BaseNode"
- elif rtype == "ns3::Ipv4L3Protocol":
- base_class_import = "from nepi.resources.ns3.ns3ipv4l3protocol import NS3BaseIpv4L3Protocol"
- base_class = "NS3BaseIpv4L3Protocol"
- else:
- type_id = ns3.TypeId()
-
- bases = ["ns3::Application",
- "ns3::NetDevice",
- "ns3::Channel",
- "ns3::Queue",
- "ns3::PropagationLossModel",
- "ns3::PropagationDelayModel",
- "ns3::WifiRemoteStationManager",
- "ns3::WifiPhy",
- "ns3::WifiMac",
- "ns3::ErrorModel",
- "ns3::ErrorRateModel"]
-
- for base in bases:
- tid_base = type_id.LookupByName(base)
- if tid.IsChildOf(tid_base):
- base_class = "NS3Base" + base.replace("ns3::", "")
- base_module = "ns3" + base.replace("ns3::", "").lower()
- base_class_import = "from nepi.resources.ns3.%s import %s " % (
- base_module, base_class)
-
- return (base_class_import, base_class)
+ for type_name in base_types:
+ tid_base = type_id.LookupByName(type_name)
+ if type_name == rtype or tid.IsChildOf(tid_base):
+ return (base_class_import, base_class)
+
+ return (None, None)
def create_ns3_rms():
ns3 = load_ns3_module()
for i in xrange(tid_count):
tid = type_id.GetRegistered(i)
- if discard(ns3, tid):
+ (base_class_import, base_class) = select_base_class(ns3, tid)
+ if not base_class:
continue
if tid.MustHideFromDocumentation() or \
attributes = "\n" + attributes if attributes else "pass"
traces = "\n" + traces if traces else "pass"
- (base_class_import, base_class) = select_base_class(ns3, tid)
-
category = tid.GetGroupName()
rtype = tid.GetName()
short_rtype = uncamm_rtype.replace("::","-")
d = os.path.dirname(os.path.realpath(__file__))
- ftemp = open(os.path.join(d, "templates", "resource_manager_template.txt"), "r")
- template = ftemp.read()
- ftemp.close()
+ with open(os.path.join(d, "templates", "resource_manager_template.txt"), "r") as ftemp:
+ template = ftemp.read()
template = template. \
replace("<CLASS_NAME>", classname). \
replace('::', ''). \
replace("-","_").lower() + ".py"
- f = open(os.path.join(d, "classes", fname), "w")
- print os.path.join(d, fname)
- print template
- f.write(template)
- f.close()
+ with open(os.path.join(d, "classes", fname), "w") as f:
+ print(os.path.join(d, fname))
+ print(template)
+ f.write(template)
def template_attributes(ns3, tid):
d = os.path.dirname(os.path.realpath(__file__))
- ftemp = open(os.path.join(d, "templates", "attribute_template.txt"), "r")
- template = ftemp.read()
- ftemp.close()
+ with open(os.path.join(d, "templates", "attribute_template.txt"), "r") as ftemp:
+ template = ftemp.read()
attributes = ""
if not attr_info.accessor.HasGetter():
continue
- attr_flags = "None"
+ attr_flags = "Flags.Reserved"
flags = attr_info.flags
- if (flags & ns3.TypeId.ATTR_SET) != ns3.TypeId.ATTR_SET:
- attr_flags = "Flags.Design"
- elif (flags & ns3.TypeId.ATTR_CONSTRUCT) == ns3.TypeId.ATTR_CONSTRUCT:
- attr_flags = "Flags.Construct"
+ if (flags & ns3.TypeId.ATTR_CONSTRUCT) == ns3.TypeId.ATTR_CONSTRUCT:
+ attr_flags += " | Flags.Construct"
+ else:
+ if (flags & ns3.TypeId.ATTR_GET) != ns3.TypeId.ATTR_GET:
+ attr_flags += " | Flags.NoRead"
+ elif (flags & ns3.TypeId.ATTR_SET) != ns3.TypeId.ATTR_SET:
+ attr_flags += " | Flags.NoWrite"
attr_name = attr_info.name
checker = attr_info.checker
def template_traces(ns3, tid):
d = os.path.dirname(os.path.realpath(__file__))
- ftemp = open(os.path.join(d, "templates", "trace_template.txt"), "r")
- template = ftemp.read()
- ftemp.close()
+ with open(os.path.join(d, "templates", "trace_template.txt"), "r") as ftemp:
+ template = ftemp.read()
traces = ""
git://git.onelab.eu / nepi.git / blobdiff