scons/swig: refactor some of the scons/SWIG code
- Move the random bits of SWIG code generation out of src/SConscript file and into methods on the objects being wrapped. - Cleaned up some variable naming and added some comments to make the process a little clearer. - Did a little generated file/module renaming: - vptype_Foo now Foo_vector - init_Foo is now Foo_init This makes it easier to see all the Foo-related files in a sorted directory listing. - Made cxx_predecls and swig_predecls normal SimObject classmethods. - Got rid of swig_objdecls hook, even though this breaks the System objects get/setMemoryMode method exports. Will be fixing this in a future changeset.
This commit is contained in:
Steve Reinhardt
@@ -97,37 +97,6 @@ allClasses = {}
|
||||
# dict to look up SimObjects based on path
|
||||
instanceDict = {}
|
||||
|
||||
def default_cxx_predecls(cls, code):
|
||||
code('#include "params/$cls.hh"')
|
||||
|
||||
def default_swig_predecls(cls, code):
|
||||
code('%import "python/m5/internal/param_$cls.i"')
|
||||
|
||||
def default_swig_objdecls(cls, code):
|
||||
class_path = cls.cxx_class.split('::')
|
||||
classname = class_path[-1]
|
||||
namespaces = class_path[:-1]
|
||||
|
||||
for ns in namespaces:
|
||||
code('namespace $ns {')
|
||||
|
||||
if namespaces:
|
||||
code('// avoid name conflicts')
|
||||
sep_string = '_COLONS_'
|
||||
flat_name = sep_string.join(class_path)
|
||||
code('%rename($flat_name) $classname;')
|
||||
|
||||
code()
|
||||
code('// stop swig from creating/wrapping default ctor/dtor')
|
||||
code('%nodefault $classname;')
|
||||
code('class $classname')
|
||||
if cls._base:
|
||||
code(' : public ${{cls._base.cxx_class}}')
|
||||
code('{};')
|
||||
|
||||
for ns in reversed(namespaces):
|
||||
code('} // namespace $ns')
|
||||
|
||||
def public_value(key, value):
|
||||
return key.startswith('_') or \
|
||||
isinstance(value, (FunctionType, MethodType, ModuleType,
|
||||
@@ -142,9 +111,6 @@ class MetaSimObject(type):
|
||||
init_keywords = { 'abstract' : bool,
|
||||
'cxx_class' : str,
|
||||
'cxx_type' : str,
|
||||
'cxx_predecls' : MethodType,
|
||||
'swig_objdecls' : MethodType,
|
||||
'swig_predecls' : MethodType,
|
||||
'type' : str }
|
||||
# Attributes that can be set any time
|
||||
keywords = { 'check' : FunctionType }
|
||||
@@ -223,18 +189,6 @@ class MetaSimObject(type):
|
||||
cls._value_dict['cxx_class'] = cls._value_dict['type']
|
||||
|
||||
cls._value_dict['cxx_type'] = '%s *' % cls._value_dict['cxx_class']
|
||||
|
||||
if 'cxx_predecls' not in cls.__dict__:
|
||||
m = MethodType(default_cxx_predecls, cls, MetaSimObject)
|
||||
setattr(cls, 'cxx_predecls', m)
|
||||
|
||||
if 'swig_predecls' not in cls.__dict__:
|
||||
m = MethodType(default_swig_predecls, cls, MetaSimObject)
|
||||
setattr(cls, 'swig_predecls', m)
|
||||
|
||||
if 'swig_objdecls' not in cls.__dict__:
|
||||
m = MethodType(default_swig_objdecls, cls, MetaSimObject)
|
||||
setattr(cls, 'swig_objdecls', m)
|
||||
|
||||
# Now process the _value_dict items. They could be defining
|
||||
# new (or overriding existing) parameters or ports, setting
|
||||
@@ -378,8 +332,76 @@ class MetaSimObject(type):
|
||||
def __str__(cls):
|
||||
return cls.__name__
|
||||
|
||||
def cxx_decl(cls, code):
|
||||
# The 'dict' attribute restricts us to the params declared in
|
||||
# See ParamValue.cxx_predecls for description.
|
||||
def cxx_predecls(cls, code):
|
||||
code('#include "params/$cls.hh"')
|
||||
|
||||
# See ParamValue.swig_predecls for description.
|
||||
def swig_predecls(cls, code):
|
||||
code('%import "python/m5/internal/param_$cls.i"')
|
||||
|
||||
# Generate the declaration for this object for wrapping with SWIG.
|
||||
# Generates code that goes into a SWIG .i file. Called from
|
||||
# src/SConscript.
|
||||
def swig_decl(cls, code):
|
||||
class_path = cls.cxx_class.split('::')
|
||||
classname = class_path[-1]
|
||||
namespaces = class_path[:-1]
|
||||
|
||||
# The 'local' attribute restricts us to the params declared in
|
||||
# the object itself, not including inherited params (which
|
||||
# will also be inherited from the base class's param struct
|
||||
# here).
|
||||
params = cls._params.local.values()
|
||||
|
||||
code('%module(package="m5.internal") param_$cls')
|
||||
code()
|
||||
code('%{')
|
||||
code('#include "params/$cls.hh"')
|
||||
for param in params:
|
||||
param.cxx_predecls(code)
|
||||
code('%}')
|
||||
code()
|
||||
|
||||
for param in params:
|
||||
param.swig_predecls(code)
|
||||
|
||||
code()
|
||||
if cls._base:
|
||||
code('%import "python/m5/internal/param_${{cls._base}}.i"')
|
||||
code()
|
||||
|
||||
for ns in namespaces:
|
||||
code('namespace $ns {')
|
||||
|
||||
if namespaces:
|
||||
code('// avoid name conflicts')
|
||||
sep_string = '_COLONS_'
|
||||
flat_name = sep_string.join(class_path)
|
||||
code('%rename($flat_name) $classname;')
|
||||
|
||||
if cls == SimObject:
|
||||
code('%include "python/swig/sim_object.i"')
|
||||
else:
|
||||
code()
|
||||
code('// stop swig from creating/wrapping default ctor/dtor')
|
||||
code('%nodefault $classname;')
|
||||
code('class $classname')
|
||||
if cls._base:
|
||||
code(' : public ${{cls._base.cxx_class}}')
|
||||
code('{};')
|
||||
|
||||
for ns in reversed(namespaces):
|
||||
code('} // namespace $ns')
|
||||
|
||||
code()
|
||||
code('%include "params/$cls.hh"')
|
||||
|
||||
|
||||
# Generate the C++ declaration (.hh file) for this SimObject's
|
||||
# param struct. Called from src/SConscript.
|
||||
def cxx_param_decl(cls, code):
|
||||
# The 'local' attribute restricts us to the params declared in
|
||||
# the object itself, not including inherited params (which
|
||||
# will also be inherited from the base class's param struct
|
||||
# here).
|
||||
@@ -421,65 +443,29 @@ class MetaSimObject(type):
|
||||
code('#include "enums/${{ptype.__name__}}.hh"')
|
||||
code()
|
||||
|
||||
cls.cxx_struct(code, cls._base, params)
|
||||
# now generate the actual param struct
|
||||
if cls == SimObject:
|
||||
code('#include "sim/sim_object_params.hh"')
|
||||
else:
|
||||
code("struct ${cls}Params")
|
||||
if cls._base:
|
||||
code(" : public ${{cls._base.type}}Params")
|
||||
code("{")
|
||||
if not hasattr(cls, 'abstract') or not cls.abstract:
|
||||
if 'type' in cls.__dict__:
|
||||
code(" ${{cls.cxx_type}} create();")
|
||||
|
||||
code.indent()
|
||||
for param in params:
|
||||
param.cxx_decl(code)
|
||||
code.dedent()
|
||||
code('};')
|
||||
|
||||
code()
|
||||
code('#endif // __PARAMS__${cls}__')
|
||||
return code
|
||||
|
||||
def cxx_struct(cls, code, base, params):
|
||||
if cls == SimObject:
|
||||
code('#include "sim/sim_object_params.hh"')
|
||||
return
|
||||
|
||||
# now generate the actual param struct
|
||||
code("struct ${cls}Params")
|
||||
if base:
|
||||
code(" : public ${{base.type}}Params")
|
||||
code("{")
|
||||
if not hasattr(cls, 'abstract') or not cls.abstract:
|
||||
if 'type' in cls.__dict__:
|
||||
code(" ${{cls.cxx_type}} create();")
|
||||
|
||||
code.indent()
|
||||
for param in params:
|
||||
param.cxx_decl(code)
|
||||
code.dedent()
|
||||
code('};')
|
||||
|
||||
def swig_decl(cls, code):
|
||||
code('''\
|
||||
%module $cls
|
||||
|
||||
%{
|
||||
#include "params/$cls.hh"
|
||||
%}
|
||||
|
||||
''')
|
||||
|
||||
# The 'dict' attribute restricts us to the params declared in
|
||||
# the object itself, not including inherited params (which
|
||||
# will also be inherited from the base class's param struct
|
||||
# here).
|
||||
params = cls._params.local.values()
|
||||
ptypes = [p.ptype for p in params]
|
||||
|
||||
# get all predeclarations
|
||||
for param in params:
|
||||
param.swig_predecls(code)
|
||||
code()
|
||||
|
||||
if cls._base:
|
||||
code('%import "python/m5/internal/param_${{cls._base.type}}.i"')
|
||||
code()
|
||||
|
||||
for ptype in ptypes:
|
||||
if issubclass(ptype, Enum):
|
||||
code('%import "enums/${{ptype.__name__}}.hh"')
|
||||
code()
|
||||
|
||||
code('%import "params/${cls}_type.hh"')
|
||||
code('%include "params/${cls}.hh"')
|
||||
|
||||
# The SimObject class is the root of the special hierarchy. Most of
|
||||
# the code in this class deals with the configuration hierarchy itself
|
||||
@@ -491,10 +477,6 @@ class SimObject(object):
|
||||
type = 'SimObject'
|
||||
abstract = True
|
||||
|
||||
@classmethod
|
||||
def swig_objdecls(cls, code):
|
||||
code('%include "python/swig/sim_object.i"')
|
||||
|
||||
# Initialize new instance. For objects with SimObject-valued
|
||||
# children, we need to recursively clone the classes represented
|
||||
# by those param values as well in a consistent "deep copy"-style
|
||||
|
||||
@@ -81,10 +81,17 @@ class MetaParamValue(type):
|
||||
class ParamValue(object):
|
||||
__metaclass__ = MetaParamValue
|
||||
|
||||
|
||||
# Generate the code needed as a prerequisite for declaring a C++
|
||||
# object of this type. Typically generates one or more #include
|
||||
# statements. Used when declaring parameters of this type.
|
||||
@classmethod
|
||||
def cxx_predecls(cls, code):
|
||||
pass
|
||||
|
||||
# Generate the code needed as a prerequisite for including a
|
||||
# reference to a C++ object of this type in a SWIG .i file.
|
||||
# Typically generates one or more %import or %include statements.
|
||||
@classmethod
|
||||
def swig_predecls(cls, code):
|
||||
pass
|
||||
@@ -101,8 +108,6 @@ class ParamValue(object):
|
||||
|
||||
# Regular parameter description.
|
||||
class ParamDesc(object):
|
||||
file_ext = 'ptype'
|
||||
|
||||
def __init__(self, ptype_str, ptype, *args, **kwargs):
|
||||
self.ptype_str = ptype_str
|
||||
# remember ptype only if it is provided
|
||||
@@ -223,8 +228,6 @@ class SimObjectVector(VectorParamValue):
|
||||
yield obj
|
||||
|
||||
class VectorParamDesc(ParamDesc):
|
||||
file_ext = 'vptype'
|
||||
|
||||
# Convert assigned value to appropriate type. If the RHS is not a
|
||||
# list or tuple, it generates a single-element list.
|
||||
def convert(self, value):
|
||||
@@ -240,10 +243,14 @@ class VectorParamDesc(ParamDesc):
|
||||
else:
|
||||
return VectorParamValue(tmp_list)
|
||||
|
||||
def swig_module_name(self):
|
||||
return "%s_vector" % self.ptype_str
|
||||
|
||||
def swig_predecls(self, code):
|
||||
code('%import "vptype_${{self.ptype_str}}.i"')
|
||||
code('%import "${{self.swig_module_name()}}.i"')
|
||||
|
||||
def swig_decl(self, code):
|
||||
code('%module(package="m5.internal") ${{self.swig_module_name()}}')
|
||||
code('%{')
|
||||
self.ptype.cxx_predecls(code)
|
||||
code('%}')
|
||||
@@ -1043,6 +1050,19 @@ namespace Enums {
|
||||
} // namespace Enums
|
||||
''')
|
||||
|
||||
def swig_decl(cls, code):
|
||||
name = cls.__name__
|
||||
code('''\
|
||||
%module(package="m5.internal") enum_$name
|
||||
|
||||
%{
|
||||
#include "enums/$name.hh"
|
||||
%}
|
||||
|
||||
%include "enums/$name.hh"
|
||||
''')
|
||||
|
||||
|
||||
# Base class for enum types.
|
||||
class Enum(ParamValue):
|
||||
__metaclass__ = MetaEnum
|
||||
|
||||
Reference in New Issue
Block a user