derek/gem5
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

python: Teach PyBindMethod how to set return_value_policy.

Browse Source 
Change-Id: Ia208e43672672556b36f905e8f71dce44b978d22
Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/17033
Reviewed-by: Andreas Sandberg <andreas.sandberg@arm.com>
Maintainer: Andreas Sandberg <andreas.sandberg@arm.com>
This commit is contained in:
Gabe Black
2019-03-07 00:27:52 -08:00
parent 8cedf1df11
commit 1a2a7c76a0
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/python/m5/util/pybind.py
View File

@@ -58,10 +58,12 @@ class PyBindProperty(PyBindExport):
code('.${export}("${{self.name}}", &${cname}::${{self.cxx_name}})')
class PyBindMethod(PyBindExport):
def __init__(self, name, cxx_name=None, args=None):
def __init__(self, name, cxx_name=None, args=None,
return_value_policy=None):
self.name = name
self.cxx_name = cxx_name if cxx_name else name
self.args = args
self.return_value_policy = return_value_policy
def _conv_arg(self, value):
if isinstance(value, bool):
@@ -72,6 +74,10 @@ class PyBindMethod(PyBindExport):
raise TypeError("Unsupported PyBind default value type")
def export(self, code, cname):
arguments = [ '"${{self.name}}"', '&${cname}::${{self.cxx_name}}' ]
if self.return_value_policy:
arguments.append('pybind11::return_value_policy::'
'${{self.return_value_policy}}')
if self.args:
def get_arg_decl(arg):
if isinstance(arg, tuple):
@@ -81,8 +87,5 @@ class PyBindMethod(PyBindExport):
else:
return 'py::arg("%s")' % arg
code('.def("${{self.name}}", &${cname}::${{self.name}}, ')
code(' ' + \
', '.join([ get_arg_decl(a) for a in self.args ]) + ')')
else:
code('.def("${{self.name}}", &${cname}::${{self.cxx_name}})')
arguments.extend(list([ get_arg_decl(a) for a in self.args ]))
code('.def(' + ', '.join(arguments) + ')')