pybind.py revision 11988
111988Sandreas.sandberg@arm.com# Copyright (c) 2017 ARM Limited 211988Sandreas.sandberg@arm.com# All rights reserved. 311988Sandreas.sandberg@arm.com# 411988Sandreas.sandberg@arm.com# The license below extends only to copyright in the software and shall 511988Sandreas.sandberg@arm.com# not be construed as granting a license to any other intellectual 611988Sandreas.sandberg@arm.com# property including but not limited to intellectual property relating 711988Sandreas.sandberg@arm.com# to a hardware implementation of the functionality of the software 811988Sandreas.sandberg@arm.com# licensed hereunder. You may use the software subject to the license 911988Sandreas.sandberg@arm.com# terms below provided that you ensure that this notice is replicated 1011988Sandreas.sandberg@arm.com# unmodified and in its entirety in all distributions of the software, 1111988Sandreas.sandberg@arm.com# modified or unmodified, in source code or in binary form. 1211988Sandreas.sandberg@arm.com# 1311988Sandreas.sandberg@arm.com# Redistribution and use in source and binary forms, with or without 1411988Sandreas.sandberg@arm.com# modification, are permitted provided that the following conditions are 1511988Sandreas.sandberg@arm.com# met: redistributions of source code must retain the above copyright 1611988Sandreas.sandberg@arm.com# notice, this list of conditions and the following disclaimer; 1711988Sandreas.sandberg@arm.com# redistributions in binary form must reproduce the above copyright 1811988Sandreas.sandberg@arm.com# notice, this list of conditions and the following disclaimer in the 1911988Sandreas.sandberg@arm.com# documentation and/or other materials provided with the distribution; 2011988Sandreas.sandberg@arm.com# neither the name of the copyright holders nor the names of its 2111988Sandreas.sandberg@arm.com# contributors may be used to endorse or promote products derived from 2211988Sandreas.sandberg@arm.com# this software without specific prior written permission. 2311988Sandreas.sandberg@arm.com# 2411988Sandreas.sandberg@arm.com# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 2511988Sandreas.sandberg@arm.com# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 2611988Sandreas.sandberg@arm.com# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 2711988Sandreas.sandberg@arm.com# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2811988Sandreas.sandberg@arm.com# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2911988Sandreas.sandberg@arm.com# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 3011988Sandreas.sandberg@arm.com# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 3111988Sandreas.sandberg@arm.com# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 3211988Sandreas.sandberg@arm.com# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 3311988Sandreas.sandberg@arm.com# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 3411988Sandreas.sandberg@arm.com# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 3511988Sandreas.sandberg@arm.com# 3611988Sandreas.sandberg@arm.com# Authors: Andreas Sandberg 3711988Sandreas.sandberg@arm.com 3811988Sandreas.sandberg@arm.comfrom abc import * 3911988Sandreas.sandberg@arm.com 4011988Sandreas.sandberg@arm.comclass PyBindExport(object): 4111988Sandreas.sandberg@arm.com __metaclass__ = ABCMeta 4211988Sandreas.sandberg@arm.com 4311988Sandreas.sandberg@arm.com @abstractmethod 4411988Sandreas.sandberg@arm.com def export(self, code, cname): 4511988Sandreas.sandberg@arm.com pass 4611988Sandreas.sandberg@arm.com 4711988Sandreas.sandberg@arm.comclass PyBindProperty(PyBindExport): 4811988Sandreas.sandberg@arm.com def __init__(self, name, cxx_name=None, writable=True): 4911988Sandreas.sandberg@arm.com self.name = name 5011988Sandreas.sandberg@arm.com self.cxx_name = cxx_name if cxx_name else name 5111988Sandreas.sandberg@arm.com self.writable = writable 5211988Sandreas.sandberg@arm.com 5311988Sandreas.sandberg@arm.com def export(self, code, cname): 5411988Sandreas.sandberg@arm.com export = "def_readwrite" if self.writable else "def_readonly" 5511988Sandreas.sandberg@arm.com code('.${export}("${{self.name}}", &${cname}::${{self.cxx_name}})') 5611988Sandreas.sandberg@arm.com 5711988Sandreas.sandberg@arm.comclass PyBindMethod(PyBindExport): 5811988Sandreas.sandberg@arm.com def __init__(self, name, cxx_name=None, args=None): 5911988Sandreas.sandberg@arm.com self.name = name 6011988Sandreas.sandberg@arm.com self.cxx_name = cxx_name if cxx_name else name 6111988Sandreas.sandberg@arm.com self.args = args 6211988Sandreas.sandberg@arm.com 6311988Sandreas.sandberg@arm.com def _conv_arg(self, value): 6411988Sandreas.sandberg@arm.com if isinstance(value, bool): 6511988Sandreas.sandberg@arm.com return "true" if value else "false" 6611988Sandreas.sandberg@arm.com elif isinstance(value, float, int): 6711988Sandreas.sandberg@arm.com return repr(value) 6811988Sandreas.sandberg@arm.com else: 6911988Sandreas.sandberg@arm.com raise TypeError("Unsupported PyBind default value type") 7011988Sandreas.sandberg@arm.com 7111988Sandreas.sandberg@arm.com def export(self, code, cname): 7211988Sandreas.sandberg@arm.com if self.args: 7311988Sandreas.sandberg@arm.com def get_arg_decl(arg): 7411988Sandreas.sandberg@arm.com if isinstance(arg, tuple): 7511988Sandreas.sandberg@arm.com name, default = arg 7611988Sandreas.sandberg@arm.com return 'py::arg("%s") = %s' % ( 7711988Sandreas.sandberg@arm.com name, self._conv_arg(default)) 7811988Sandreas.sandberg@arm.com else: 7911988Sandreas.sandberg@arm.com return 'py::arg("%s")' % arg 8011988Sandreas.sandberg@arm.com 8111988Sandreas.sandberg@arm.com code('.def("${{self.name}}", &${cname}::${{self.name}}, ') 8211988Sandreas.sandberg@arm.com code(' ' + \ 8311988Sandreas.sandberg@arm.com ', '.join([ get_arg_decl(a) for a in self.args ]) + ')') 8411988Sandreas.sandberg@arm.com else: 8511988Sandreas.sandberg@arm.com code('.def("${{self.name}}", &${cname}::${{self.cxx_name}})') 86