mirror of https://github.com/exaloop/codon.git
2005 lines
75 KiB
Python
2005 lines
75 KiB
Python
# Copyright (C) 2022-2025 Exaloop Inc. <https://exaloop.io>
|
|
|
|
import os
|
|
|
|
from gc import atomic, alloc_uncollectable
|
|
from internal.dlopen import *
|
|
|
|
# general
|
|
Py_DecRef = Function[[cobj], NoneType](cobj())
|
|
Py_IncRef = Function[[cobj], NoneType](cobj())
|
|
Py_Initialize = Function[[], NoneType](cobj())
|
|
PyImport_AddModule = Function[[cobj], cobj](cobj())
|
|
PyImport_AddModuleObject = Function[[cobj], cobj](cobj())
|
|
PyImport_ImportModule = Function[[cobj], cobj](cobj())
|
|
PyRun_SimpleString = Function[[cobj], NoneType](cobj())
|
|
PyEval_GetGlobals = Function[[], cobj](cobj())
|
|
PyEval_GetBuiltins = Function[[], cobj](cobj())
|
|
PyOS_setsig = Function[[i32, cobj], cobj](cobj())
|
|
|
|
# conversions
|
|
PyLong_AsLong = Function[[cobj], int](cobj())
|
|
PyLong_FromLong = Function[[int], cobj](cobj())
|
|
PyFloat_AsDouble = Function[[cobj], float](cobj())
|
|
PyFloat_FromDouble = Function[[float], cobj](cobj())
|
|
PyBool_FromLong = Function[[int], cobj](cobj())
|
|
PyBytes_AsString = Function[[cobj], cobj](cobj())
|
|
PyBytes_Size = Function[[cobj], int](cobj())
|
|
PyList_New = Function[[int], cobj](cobj())
|
|
PyList_Size = Function[[cobj], int](cobj())
|
|
PyList_GetItem = Function[[cobj, int], cobj](cobj())
|
|
PyList_SetItem = Function[[cobj, int, cobj], cobj](cobj())
|
|
PyDict_New = Function[[], cobj](cobj())
|
|
PyDict_Next = Function[[cobj, Ptr[int], Ptr[cobj], Ptr[cobj]], int](cobj())
|
|
PyDict_GetItem = Function[[cobj, cobj], cobj](cobj())
|
|
PyDict_GetItemString = Function[[cobj, cobj], cobj](cobj())
|
|
PyDict_SetItem = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyDict_Size = Function[[cobj], int](cobj())
|
|
PySet_Add = Function[[cobj, cobj], cobj](cobj())
|
|
PySet_New = Function[[cobj], cobj](cobj())
|
|
PyTuple_New = Function[[int], cobj](cobj())
|
|
PyTuple_Size = Function[[cobj], int](cobj())
|
|
PyTuple_GetItem = Function[[cobj, int], cobj](cobj())
|
|
PyTuple_SetItem = Function[[cobj, int, cobj], NoneType](cobj())
|
|
PyUnicode_AsEncodedString = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyUnicode_DecodeFSDefaultAndSize = Function[[cobj, int], cobj](cobj())
|
|
PyUnicode_FromString = Function[[cobj], cobj](cobj())
|
|
PyComplex_FromDoubles = Function[[float, float], cobj](cobj())
|
|
PyComplex_RealAsDouble = Function[[cobj], float](cobj())
|
|
PyComplex_ImagAsDouble = Function[[cobj], float](cobj())
|
|
PyIter_Next = Function[[cobj], cobj](cobj())
|
|
PySlice_New = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PySlice_Unpack = Function[[cobj, Ptr[int], Ptr[int], Ptr[int]], int](cobj())
|
|
PyCapsule_New = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyCapsule_GetPointer = Function[[cobj, cobj], cobj](cobj())
|
|
|
|
# number
|
|
PyNumber_Add = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Subtract = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Multiply = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_MatrixMultiply = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_FloorDivide = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_TrueDivide = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Remainder = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Divmod = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Power = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyNumber_Negative = Function[[cobj], cobj](cobj())
|
|
PyNumber_Positive = Function[[cobj], cobj](cobj())
|
|
PyNumber_Absolute = Function[[cobj], cobj](cobj())
|
|
PyNumber_Invert = Function[[cobj], cobj](cobj())
|
|
PyNumber_Lshift = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Rshift = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_And = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Xor = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Or = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceAdd = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceSubtract = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceMultiply = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceMatrixMultiply = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceFloorDivide = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceTrueDivide = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceRemainder = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlacePower = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceLshift = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceRshift = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceAnd = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceXor = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_InPlaceOr = Function[[cobj, cobj], cobj](cobj())
|
|
PyNumber_Long = Function[[cobj], cobj](cobj())
|
|
PyNumber_Float = Function[[cobj], cobj](cobj())
|
|
PyNumber_Index = Function[[cobj], cobj](cobj())
|
|
|
|
# object
|
|
PyObject_Call = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyObject_GetAttr = Function[[cobj, cobj], cobj](cobj())
|
|
PyObject_GetAttrString = Function[[cobj, cobj], cobj](cobj())
|
|
PyObject_GetIter = Function[[cobj], cobj](cobj())
|
|
PyObject_HasAttrString = Function[[cobj, cobj], int](cobj())
|
|
PyObject_IsTrue = Function[[cobj], int](cobj())
|
|
PyObject_Length = Function[[cobj], int](cobj())
|
|
PyObject_LengthHint = Function[[cobj, int], int](cobj())
|
|
PyObject_SetAttrString = Function[[cobj, cobj, cobj], cobj](cobj())
|
|
PyObject_Str = Function[[cobj], cobj](cobj())
|
|
PyObject_Repr = Function[[cobj], cobj](cobj())
|
|
PyObject_Hash = Function[[cobj], int](cobj())
|
|
PyObject_GetItem = Function[[cobj, cobj], cobj](cobj())
|
|
PyObject_SetItem = Function[[cobj, cobj, cobj], int](cobj())
|
|
PyObject_DelItem = Function[[cobj, cobj], int](cobj())
|
|
PyObject_RichCompare = Function[[cobj, cobj, i32], cobj](cobj())
|
|
PyObject_IsInstance = Function[[cobj, cobj], i32](cobj())
|
|
|
|
# error handling
|
|
PyErr_Fetch = Function[[Ptr[cobj], Ptr[cobj], Ptr[cobj]], NoneType](cobj())
|
|
PyErr_NormalizeException = Function[[Ptr[cobj], Ptr[cobj], Ptr[cobj]], NoneType](cobj())
|
|
PyErr_SetString = Function[[cobj, cobj], NoneType](cobj())
|
|
|
|
# constants
|
|
Py_None = cobj()
|
|
Py_True = cobj()
|
|
Py_False = cobj()
|
|
Py_Ellipsis = cobj()
|
|
Py_NotImplemented = cobj()
|
|
Py_LT = 0
|
|
Py_LE = 1
|
|
Py_EQ = 2
|
|
Py_NE = 3
|
|
Py_GT = 4
|
|
Py_GE = 5
|
|
|
|
# types
|
|
PyLong_Type = cobj()
|
|
PyFloat_Type = cobj()
|
|
PyBool_Type = cobj()
|
|
PyUnicode_Type = cobj()
|
|
PyBytes_Type = cobj()
|
|
PyComplex_Type = cobj()
|
|
PyList_Type = cobj()
|
|
PyDict_Type = cobj()
|
|
PySet_Type = cobj()
|
|
PyTuple_Type = cobj()
|
|
PySlice_Type = cobj()
|
|
PyCapsule_Type = cobj()
|
|
|
|
# exceptions
|
|
PyExc_BaseException = cobj()
|
|
PyExc_Exception = cobj()
|
|
PyExc_NameError = cobj()
|
|
PyExc_OSError = cobj()
|
|
PyExc_IOError = cobj()
|
|
PyExc_ValueError = cobj()
|
|
PyExc_LookupError = cobj()
|
|
PyExc_IndexError = cobj()
|
|
PyExc_KeyError = cobj()
|
|
PyExc_TypeError = cobj()
|
|
PyExc_ArithmeticError = cobj()
|
|
PyExc_ZeroDivisionError = cobj()
|
|
PyExc_OverflowError = cobj()
|
|
PyExc_AttributeError = cobj()
|
|
PyExc_RuntimeError = cobj()
|
|
PyExc_NotImplementedError = cobj()
|
|
PyExc_StopIteration = cobj()
|
|
PyExc_AssertionError = cobj()
|
|
PyExc_EOFError = cobj()
|
|
PyExc_SystemExit = cobj()
|
|
|
|
_PY_MODULE_CACHE = Dict[str, pyobj]()
|
|
|
|
_PY_INIT = """
|
|
import io
|
|
|
|
clsf = None
|
|
clsa = None
|
|
plt = None
|
|
try:
|
|
import matplotlib.figure
|
|
import matplotlib.pyplot
|
|
plt = matplotlib.pyplot
|
|
clsf = matplotlib.figure.Figure
|
|
clsa = matplotlib.artist.Artist
|
|
except ModuleNotFoundError:
|
|
pass
|
|
|
|
def __codon_repr__(fig):
|
|
if clsf and isinstance(fig, clsf):
|
|
stream = io.StringIO()
|
|
fig.savefig(stream, format="svg")
|
|
return 'image/svg+xml', stream.getvalue()
|
|
elif clsa and isinstance(fig, list) and all(
|
|
isinstance(i, clsa) for i in fig
|
|
):
|
|
stream = io.StringIO()
|
|
plt.gcf().savefig(stream, format="svg")
|
|
return 'image/svg+xml', stream.getvalue()
|
|
elif hasattr(fig, "_repr_html_"):
|
|
return 'text/html', fig._repr_html_()
|
|
else:
|
|
return 'text/plain', fig.__repr__()
|
|
"""
|
|
|
|
_PY_INITIALIZED = False
|
|
|
|
def init_handles_dlopen(py_handle: cobj):
|
|
global Py_DecRef
|
|
global Py_IncRef
|
|
global Py_Initialize
|
|
global PyImport_AddModule
|
|
global PyImport_AddModuleObject
|
|
global PyImport_ImportModule
|
|
global PyRun_SimpleString
|
|
global PyEval_GetGlobals
|
|
global PyEval_GetBuiltins
|
|
global PyOS_setsig
|
|
global PyLong_AsLong
|
|
global PyLong_FromLong
|
|
global PyFloat_AsDouble
|
|
global PyFloat_FromDouble
|
|
global PyBool_FromLong
|
|
global PyBytes_AsString
|
|
global PyBytes_Size
|
|
global PyList_New
|
|
global PyList_Size
|
|
global PyList_GetItem
|
|
global PyList_SetItem
|
|
global PyDict_New
|
|
global PyDict_Next
|
|
global PyDict_GetItem
|
|
global PyDict_GetItemString
|
|
global PyDict_SetItem
|
|
global PyDict_Size
|
|
global PySet_Add
|
|
global PySet_New
|
|
global PyTuple_New
|
|
global PyTuple_Size
|
|
global PyTuple_GetItem
|
|
global PyTuple_SetItem
|
|
global PyUnicode_AsEncodedString
|
|
global PyUnicode_DecodeFSDefaultAndSize
|
|
global PyUnicode_FromString
|
|
global PyComplex_FromDoubles
|
|
global PyComplex_RealAsDouble
|
|
global PyComplex_ImagAsDouble
|
|
global PyIter_Next
|
|
global PySlice_New
|
|
global PySlice_Unpack
|
|
global PyCapsule_New
|
|
global PyCapsule_GetPointer
|
|
global PyNumber_Add
|
|
global PyNumber_Subtract
|
|
global PyNumber_Multiply
|
|
global PyNumber_MatrixMultiply
|
|
global PyNumber_FloorDivide
|
|
global PyNumber_TrueDivide
|
|
global PyNumber_Remainder
|
|
global PyNumber_Divmod
|
|
global PyNumber_Power
|
|
global PyNumber_Negative
|
|
global PyNumber_Positive
|
|
global PyNumber_Absolute
|
|
global PyNumber_Invert
|
|
global PyNumber_Lshift
|
|
global PyNumber_Rshift
|
|
global PyNumber_And
|
|
global PyNumber_Xor
|
|
global PyNumber_Or
|
|
global PyNumber_InPlaceAdd
|
|
global PyNumber_InPlaceSubtract
|
|
global PyNumber_InPlaceMultiply
|
|
global PyNumber_InPlaceMatrixMultiply
|
|
global PyNumber_InPlaceFloorDivide
|
|
global PyNumber_InPlaceTrueDivide
|
|
global PyNumber_InPlaceRemainder
|
|
global PyNumber_InPlacePower
|
|
global PyNumber_InPlaceLshift
|
|
global PyNumber_InPlaceRshift
|
|
global PyNumber_InPlaceAnd
|
|
global PyNumber_InPlaceXor
|
|
global PyNumber_InPlaceOr
|
|
global PyNumber_Long
|
|
global PyNumber_Float
|
|
global PyNumber_Index
|
|
global PyObject_Call
|
|
global PyObject_GetAttr
|
|
global PyObject_GetAttrString
|
|
global PyObject_GetIter
|
|
global PyObject_HasAttrString
|
|
global PyObject_IsTrue
|
|
global PyObject_Length
|
|
global PyObject_LengthHint
|
|
global PyObject_SetAttrString
|
|
global PyObject_Str
|
|
global PyObject_Repr
|
|
global PyObject_Hash
|
|
global PyObject_GetItem
|
|
global PyObject_SetItem
|
|
global PyObject_DelItem
|
|
global PyObject_RichCompare
|
|
global PyObject_IsInstance
|
|
global PyErr_Fetch
|
|
global PyErr_NormalizeException
|
|
global PyErr_SetString
|
|
global Py_None
|
|
global Py_True
|
|
global Py_False
|
|
global Py_Ellipsis
|
|
global Py_NotImplemented
|
|
global PyLong_Type
|
|
global PyFloat_Type
|
|
global PyBool_Type
|
|
global PyUnicode_Type
|
|
global PyBytes_Type
|
|
global PyComplex_Type
|
|
global PyList_Type
|
|
global PyDict_Type
|
|
global PySet_Type
|
|
global PyTuple_Type
|
|
global PySlice_Type
|
|
global PyCapsule_Type
|
|
global PyExc_BaseException
|
|
global PyExc_Exception
|
|
global PyExc_NameError
|
|
global PyExc_OSError
|
|
global PyExc_IOError
|
|
global PyExc_ValueError
|
|
global PyExc_LookupError
|
|
global PyExc_IndexError
|
|
global PyExc_KeyError
|
|
global PyExc_TypeError
|
|
global PyExc_ArithmeticError
|
|
global PyExc_ZeroDivisionError
|
|
global PyExc_OverflowError
|
|
global PyExc_AttributeError
|
|
global PyExc_RuntimeError
|
|
global PyExc_NotImplementedError
|
|
global PyExc_StopIteration
|
|
global PyExc_AssertionError
|
|
global PyExc_EOFError
|
|
global PyExc_SystemExit
|
|
|
|
Py_DecRef = dlsym(py_handle, "Py_DecRef")
|
|
Py_IncRef = dlsym(py_handle, "Py_IncRef")
|
|
Py_Initialize = dlsym(py_handle, "Py_Initialize")
|
|
PyImport_AddModule = dlsym(py_handle, "PyImport_AddModule")
|
|
PyImport_AddModuleObject = dlsym(py_handle, "PyImport_AddModuleObject")
|
|
PyImport_ImportModule = dlsym(py_handle, "PyImport_ImportModule")
|
|
PyRun_SimpleString = dlsym(py_handle, "PyRun_SimpleString")
|
|
PyEval_GetGlobals = dlsym(py_handle, "PyEval_GetGlobals")
|
|
PyEval_GetBuiltins = dlsym(py_handle, "PyEval_GetBuiltins")
|
|
PyOS_setsig = dlsym(py_handle, "PyOS_setsig")
|
|
PyLong_AsLong = dlsym(py_handle, "PyLong_AsLong")
|
|
PyLong_FromLong = dlsym(py_handle, "PyLong_FromLong")
|
|
PyFloat_AsDouble = dlsym(py_handle, "PyFloat_AsDouble")
|
|
PyFloat_FromDouble = dlsym(py_handle, "PyFloat_FromDouble")
|
|
PyBool_FromLong = dlsym(py_handle, "PyBool_FromLong")
|
|
PyBytes_AsString = dlsym(py_handle, "PyBytes_AsString")
|
|
PyBytes_Size = dlsym(py_handle, "PyBytes_Size")
|
|
PyList_New = dlsym(py_handle, "PyList_New")
|
|
PyList_Size = dlsym(py_handle, "PyList_Size")
|
|
PyList_GetItem = dlsym(py_handle, "PyList_GetItem")
|
|
PyList_SetItem = dlsym(py_handle, "PyList_SetItem")
|
|
PyDict_New = dlsym(py_handle, "PyDict_New")
|
|
PyDict_Next = dlsym(py_handle, "PyDict_Next")
|
|
PyDict_GetItem = dlsym(py_handle, "PyDict_GetItem")
|
|
PyDict_GetItemString = dlsym(py_handle, "PyDict_GetItemString")
|
|
PyDict_SetItem = dlsym(py_handle, "PyDict_SetItem")
|
|
PyDict_Size = dlsym(py_handle, "PyDict_Size")
|
|
PySet_Add = dlsym(py_handle, "PySet_Add")
|
|
PySet_New = dlsym(py_handle, "PySet_New")
|
|
PyTuple_New = dlsym(py_handle, "PyTuple_New")
|
|
PyTuple_Size = dlsym(py_handle, "PyTuple_Size")
|
|
PyTuple_GetItem = dlsym(py_handle, "PyTuple_GetItem")
|
|
PyTuple_SetItem = dlsym(py_handle, "PyTuple_SetItem")
|
|
PyUnicode_AsEncodedString = dlsym(py_handle, "PyUnicode_AsEncodedString")
|
|
PyUnicode_DecodeFSDefaultAndSize = dlsym(py_handle, "PyUnicode_DecodeFSDefaultAndSize")
|
|
PyUnicode_FromString = dlsym(py_handle, "PyUnicode_FromString")
|
|
PyComplex_FromDoubles = dlsym(py_handle, "PyComplex_FromDoubles")
|
|
PyComplex_RealAsDouble = dlsym(py_handle, "PyComplex_RealAsDouble")
|
|
PyComplex_ImagAsDouble = dlsym(py_handle, "PyComplex_ImagAsDouble")
|
|
PyIter_Next = dlsym(py_handle, "PyIter_Next")
|
|
PySlice_New = dlsym(py_handle, "PySlice_New")
|
|
PySlice_Unpack = dlsym(py_handle, "PySlice_Unpack")
|
|
PyCapsule_New = dlsym(py_handle, "PyCapsule_New")
|
|
PyCapsule_GetPointer = dlsym(py_handle, "PyCapsule_GetPointer")
|
|
PyNumber_Add = dlsym(py_handle, "PyNumber_Add")
|
|
PyNumber_Subtract = dlsym(py_handle, "PyNumber_Subtract")
|
|
PyNumber_Multiply = dlsym(py_handle, "PyNumber_Multiply")
|
|
PyNumber_MatrixMultiply = dlsym(py_handle, "PyNumber_MatrixMultiply")
|
|
PyNumber_FloorDivide = dlsym(py_handle, "PyNumber_FloorDivide")
|
|
PyNumber_TrueDivide = dlsym(py_handle, "PyNumber_TrueDivide")
|
|
PyNumber_Remainder = dlsym(py_handle, "PyNumber_Remainder")
|
|
PyNumber_Divmod = dlsym(py_handle, "PyNumber_Divmod")
|
|
PyNumber_Power = dlsym(py_handle, "PyNumber_Power")
|
|
PyNumber_Negative = dlsym(py_handle, "PyNumber_Negative")
|
|
PyNumber_Positive = dlsym(py_handle, "PyNumber_Positive")
|
|
PyNumber_Absolute = dlsym(py_handle, "PyNumber_Absolute")
|
|
PyNumber_Invert = dlsym(py_handle, "PyNumber_Invert")
|
|
PyNumber_Lshift = dlsym(py_handle, "PyNumber_Lshift")
|
|
PyNumber_Rshift = dlsym(py_handle, "PyNumber_Rshift")
|
|
PyNumber_And = dlsym(py_handle, "PyNumber_And")
|
|
PyNumber_Xor = dlsym(py_handle, "PyNumber_Xor")
|
|
PyNumber_Or = dlsym(py_handle, "PyNumber_Or")
|
|
PyNumber_InPlaceAdd = dlsym(py_handle, "PyNumber_InPlaceAdd")
|
|
PyNumber_InPlaceSubtract = dlsym(py_handle, "PyNumber_InPlaceSubtract")
|
|
PyNumber_InPlaceMultiply = dlsym(py_handle, "PyNumber_InPlaceMultiply")
|
|
PyNumber_InPlaceMatrixMultiply = dlsym(py_handle, "PyNumber_InPlaceMatrixMultiply")
|
|
PyNumber_InPlaceFloorDivide = dlsym(py_handle, "PyNumber_InPlaceFloorDivide")
|
|
PyNumber_InPlaceTrueDivide = dlsym(py_handle, "PyNumber_InPlaceTrueDivide")
|
|
PyNumber_InPlaceRemainder = dlsym(py_handle, "PyNumber_InPlaceRemainder")
|
|
PyNumber_InPlacePower = dlsym(py_handle, "PyNumber_InPlacePower")
|
|
PyNumber_InPlaceLshift = dlsym(py_handle, "PyNumber_InPlaceLshift")
|
|
PyNumber_InPlaceRshift = dlsym(py_handle, "PyNumber_InPlaceRshift")
|
|
PyNumber_InPlaceAnd = dlsym(py_handle, "PyNumber_InPlaceAnd")
|
|
PyNumber_InPlaceXor = dlsym(py_handle, "PyNumber_InPlaceXor")
|
|
PyNumber_InPlaceOr = dlsym(py_handle, "PyNumber_InPlaceOr")
|
|
PyNumber_Long = dlsym(py_handle, "PyNumber_Long")
|
|
PyNumber_Float = dlsym(py_handle, "PyNumber_Float")
|
|
PyNumber_Index = dlsym(py_handle, "PyNumber_Index")
|
|
PyObject_Call = dlsym(py_handle, "PyObject_Call")
|
|
PyObject_GetAttr = dlsym(py_handle, "PyObject_GetAttr")
|
|
PyObject_GetAttrString = dlsym(py_handle, "PyObject_GetAttrString")
|
|
PyObject_GetIter = dlsym(py_handle, "PyObject_GetIter")
|
|
PyObject_HasAttrString = dlsym(py_handle, "PyObject_HasAttrString")
|
|
PyObject_IsTrue = dlsym(py_handle, "PyObject_IsTrue")
|
|
PyObject_Length = dlsym(py_handle, "PyObject_Length")
|
|
PyObject_LengthHint = dlsym(py_handle, "PyObject_LengthHint")
|
|
PyObject_SetAttrString = dlsym(py_handle, "PyObject_SetAttrString")
|
|
PyObject_Str = dlsym(py_handle, "PyObject_Str")
|
|
PyObject_Repr = dlsym(py_handle, "PyObject_Repr")
|
|
PyObject_Hash = dlsym(py_handle, "PyObject_Hash")
|
|
PyObject_GetItem = dlsym(py_handle, "PyObject_GetItem")
|
|
PyObject_SetItem = dlsym(py_handle, "PyObject_SetItem")
|
|
PyObject_DelItem = dlsym(py_handle, "PyObject_DelItem")
|
|
PyObject_RichCompare = dlsym(py_handle, "PyObject_RichCompare")
|
|
PyObject_IsInstance = dlsym(py_handle, "PyObject_IsInstance")
|
|
PyErr_Fetch = dlsym(py_handle, "PyErr_Fetch")
|
|
PyErr_NormalizeException = dlsym(py_handle, "PyErr_NormalizeException")
|
|
PyErr_SetString = dlsym(py_handle, "PyErr_SetString")
|
|
Py_None = dlsym(py_handle, "_Py_NoneStruct")
|
|
Py_True = dlsym(py_handle, "_Py_TrueStruct")
|
|
Py_False = dlsym(py_handle, "_Py_FalseStruct")
|
|
Py_Ellipsis = dlsym(py_handle, "_Py_EllipsisObject")
|
|
Py_NotImplemented = dlsym(py_handle, "_Py_NotImplementedStruct")
|
|
PyLong_Type = dlsym(py_handle, "PyLong_Type")
|
|
PyFloat_Type = dlsym(py_handle, "PyFloat_Type")
|
|
PyBool_Type = dlsym(py_handle, "PyBool_Type")
|
|
PyUnicode_Type = dlsym(py_handle, "PyUnicode_Type")
|
|
PyBytes_Type = dlsym(py_handle, "PyBytes_Type")
|
|
PyComplex_Type = dlsym(py_handle, "PyComplex_Type")
|
|
PyList_Type = dlsym(py_handle, "PyList_Type")
|
|
PyDict_Type = dlsym(py_handle, "PyDict_Type")
|
|
PySet_Type = dlsym(py_handle, "PySet_Type")
|
|
PyTuple_Type = dlsym(py_handle, "PyTuple_Type")
|
|
PySlice_Type = dlsym(py_handle, "PySlice_Type")
|
|
PyCapsule_Type = dlsym(py_handle, "PyCapsule_Type")
|
|
PyExc_BaseException = Ptr[cobj](dlsym(py_handle, "PyExc_BaseException", cobj))[0]
|
|
PyExc_Exception = Ptr[cobj](dlsym(py_handle, "PyExc_Exception", cobj))[0]
|
|
PyExc_NameError = Ptr[cobj](dlsym(py_handle, "PyExc_NameError", cobj))[0]
|
|
PyExc_OSError = Ptr[cobj](dlsym(py_handle, "PyExc_OSError", cobj))[0]
|
|
PyExc_IOError = Ptr[cobj](dlsym(py_handle, "PyExc_IOError", cobj))[0]
|
|
PyExc_ValueError = Ptr[cobj](dlsym(py_handle, "PyExc_ValueError", cobj))[0]
|
|
PyExc_LookupError = Ptr[cobj](dlsym(py_handle, "PyExc_LookupError", cobj))[0]
|
|
PyExc_IndexError = Ptr[cobj](dlsym(py_handle, "PyExc_IndexError", cobj))[0]
|
|
PyExc_KeyError = Ptr[cobj](dlsym(py_handle, "PyExc_KeyError", cobj))[0]
|
|
PyExc_TypeError = Ptr[cobj](dlsym(py_handle, "PyExc_TypeError", cobj))[0]
|
|
PyExc_ArithmeticError = Ptr[cobj](dlsym(py_handle, "PyExc_ArithmeticError", cobj))[0]
|
|
PyExc_ZeroDivisionError = Ptr[cobj](dlsym(py_handle, "PyExc_ZeroDivisionError", cobj))[0]
|
|
PyExc_OverflowError = Ptr[cobj](dlsym(py_handle, "PyExc_OverflowError", cobj))[0]
|
|
PyExc_AttributeError = Ptr[cobj](dlsym(py_handle, "PyExc_AttributeError", cobj))[0]
|
|
PyExc_RuntimeError = Ptr[cobj](dlsym(py_handle, "PyExc_RuntimeError", cobj))[0]
|
|
PyExc_NotImplementedError = Ptr[cobj](dlsym(py_handle, "PyExc_NotImplementedError", cobj))[0]
|
|
PyExc_StopIteration = Ptr[cobj](dlsym(py_handle, "PyExc_StopIteration", cobj))[0]
|
|
PyExc_AssertionError = Ptr[cobj](dlsym(py_handle, "PyExc_AssertionError", cobj))[0]
|
|
PyExc_EOFError = Ptr[cobj](dlsym(py_handle, "PyExc_EOFError", cobj))[0]
|
|
PyExc_SystemExit = Ptr[cobj](dlsym(py_handle, "PyExc_SystemExit", cobj))[0]
|
|
|
|
def init_handles_static():
|
|
from C import Py_DecRef(cobj) as _Py_DecRef
|
|
from C import Py_IncRef(cobj) as _Py_IncRef
|
|
from C import Py_Initialize() as _Py_Initialize
|
|
from C import PyImport_AddModule(cobj) -> cobj as _PyImport_AddModule
|
|
from C import PyImport_AddModuleObject(cobj) -> cobj as _PyImport_AddModuleObject
|
|
from C import PyImport_ImportModule(cobj) -> cobj as _PyImport_ImportModule
|
|
from C import PyRun_SimpleString(cobj) as _PyRun_SimpleString
|
|
from C import PyEval_GetGlobals() -> cobj as _PyEval_GetGlobals
|
|
from C import PyEval_GetBuiltins() -> cobj as _PyEval_GetBuiltins
|
|
from C import PyOS_setsig(i32, cobj) -> cobj as _PyOS_setsig
|
|
from C import PyLong_AsLong(cobj) -> int as _PyLong_AsLong
|
|
from C import PyLong_FromLong(int) -> cobj as _PyLong_FromLong
|
|
from C import PyFloat_AsDouble(cobj) -> float as _PyFloat_AsDouble
|
|
from C import PyFloat_FromDouble(float) -> cobj as _PyFloat_FromDouble
|
|
from C import PyBool_FromLong(int) -> cobj as _PyBool_FromLong
|
|
from C import PyBytes_AsString(cobj) -> cobj as _PyBytes_AsString
|
|
from C import PyBytes_Size(cobj) -> int as _PyBytes_Size
|
|
from C import PyList_New(int) -> cobj as _PyList_New
|
|
from C import PyList_Size(cobj) -> int as _PyList_Size
|
|
from C import PyList_GetItem(cobj, int) -> cobj as _PyList_GetItem
|
|
from C import PyList_SetItem(cobj, int, cobj) -> cobj as _PyList_SetItem
|
|
from C import PyDict_New() -> cobj as _PyDict_New
|
|
from C import PyDict_Next(cobj, Ptr[int], Ptr[cobj], Ptr[cobj]) -> int as _PyDict_Next
|
|
from C import PyDict_GetItem(cobj, cobj) -> cobj as _PyDict_GetItem
|
|
from C import PyDict_GetItemString(cobj, cobj) -> cobj as _PyDict_GetItemString
|
|
from C import PyDict_SetItem(cobj, cobj, cobj) -> cobj as _PyDict_SetItem
|
|
from C import PyDict_Size(cobj) -> int as _PyDict_Size
|
|
from C import PySet_Add(cobj, cobj) -> cobj as _PySet_Add
|
|
from C import PySet_New(cobj) -> cobj as _PySet_New
|
|
from C import PyTuple_New(int) -> cobj as _PyTuple_New
|
|
from C import PyTuple_Size(cobj) -> int as _PyTuple_Size
|
|
from C import PyTuple_GetItem(cobj, int) -> cobj as _PyTuple_GetItem
|
|
from C import PyTuple_SetItem(cobj, int, cobj) as _PyTuple_SetItem
|
|
from C import PyUnicode_AsEncodedString(cobj, cobj, cobj) -> cobj as _PyUnicode_AsEncodedString
|
|
from C import PyUnicode_DecodeFSDefaultAndSize(cobj, int) -> cobj as _PyUnicode_DecodeFSDefaultAndSize
|
|
from C import PyUnicode_FromString(cobj) -> cobj as _PyUnicode_FromString
|
|
from C import PyComplex_FromDoubles(float, float) -> cobj as _PyComplex_FromDoubles
|
|
from C import PyComplex_RealAsDouble(cobj) -> float as _PyComplex_RealAsDouble
|
|
from C import PyComplex_ImagAsDouble(cobj) -> float as _PyComplex_ImagAsDouble
|
|
from C import PyIter_Next(cobj) -> cobj as _PyIter_Next
|
|
from C import PySlice_New(cobj, cobj, cobj) -> cobj as _PySlice_New
|
|
from C import PySlice_Unpack(cobj, Ptr[int], Ptr[int], Ptr[int]) -> int as _PySlice_Unpack
|
|
from C import PyCapsule_New(cobj, cobj, cobj) -> cobj as _PyCapsule_New
|
|
from C import PyCapsule_GetPointer(cobj, cobj) -> cobj as _PyCapsule_GetPointer
|
|
from C import PyNumber_Add(cobj, cobj) -> cobj as _PyNumber_Add
|
|
from C import PyNumber_Subtract(cobj, cobj) -> cobj as _PyNumber_Subtract
|
|
from C import PyNumber_Multiply(cobj, cobj) -> cobj as _PyNumber_Multiply
|
|
from C import PyNumber_MatrixMultiply(cobj, cobj) -> cobj as _PyNumber_MatrixMultiply
|
|
from C import PyNumber_FloorDivide(cobj, cobj) -> cobj as _PyNumber_FloorDivide
|
|
from C import PyNumber_TrueDivide(cobj, cobj) -> cobj as _PyNumber_TrueDivide
|
|
from C import PyNumber_Remainder(cobj, cobj) -> cobj as _PyNumber_Remainder
|
|
from C import PyNumber_Divmod(cobj, cobj) -> cobj as _PyNumber_Divmod
|
|
from C import PyNumber_Power(cobj, cobj, cobj) -> cobj as _PyNumber_Power
|
|
from C import PyNumber_Negative(cobj) -> cobj as _PyNumber_Negative
|
|
from C import PyNumber_Positive(cobj) -> cobj as _PyNumber_Positive
|
|
from C import PyNumber_Absolute(cobj) -> cobj as _PyNumber_Absolute
|
|
from C import PyNumber_Invert(cobj) -> cobj as _PyNumber_Invert
|
|
from C import PyNumber_Lshift(cobj, cobj) -> cobj as _PyNumber_Lshift
|
|
from C import PyNumber_Rshift(cobj, cobj) -> cobj as _PyNumber_Rshift
|
|
from C import PyNumber_And(cobj, cobj) -> cobj as _PyNumber_And
|
|
from C import PyNumber_Xor(cobj, cobj) -> cobj as _PyNumber_Xor
|
|
from C import PyNumber_Or(cobj, cobj) -> cobj as _PyNumber_Or
|
|
from C import PyNumber_InPlaceAdd(cobj, cobj) -> cobj as _PyNumber_InPlaceAdd
|
|
from C import PyNumber_InPlaceSubtract(cobj, cobj) -> cobj as _PyNumber_InPlaceSubtract
|
|
from C import PyNumber_InPlaceMultiply(cobj, cobj) -> cobj as _PyNumber_InPlaceMultiply
|
|
from C import PyNumber_InPlaceMatrixMultiply(cobj, cobj) -> cobj as _PyNumber_InPlaceMatrixMultiply
|
|
from C import PyNumber_InPlaceFloorDivide(cobj, cobj) -> cobj as _PyNumber_InPlaceFloorDivide
|
|
from C import PyNumber_InPlaceTrueDivide(cobj, cobj) -> cobj as _PyNumber_InPlaceTrueDivide
|
|
from C import PyNumber_InPlaceRemainder(cobj, cobj) -> cobj as _PyNumber_InPlaceRemainder
|
|
from C import PyNumber_InPlacePower(cobj, cobj, cobj) -> cobj as _PyNumber_InPlacePower
|
|
from C import PyNumber_InPlaceLshift(cobj, cobj) -> cobj as _PyNumber_InPlaceLshift
|
|
from C import PyNumber_InPlaceRshift(cobj, cobj) -> cobj as _PyNumber_InPlaceRshift
|
|
from C import PyNumber_InPlaceAnd(cobj, cobj) -> cobj as _PyNumber_InPlaceAnd
|
|
from C import PyNumber_InPlaceXor(cobj, cobj) -> cobj as _PyNumber_InPlaceXor
|
|
from C import PyNumber_InPlaceOr(cobj, cobj) -> cobj as _PyNumber_InPlaceOr
|
|
from C import PyNumber_Long(cobj) -> cobj as _PyNumber_Long
|
|
from C import PyNumber_Float(cobj) -> cobj as _PyNumber_Float
|
|
from C import PyNumber_Index(cobj) -> cobj as _PyNumber_Index
|
|
from C import PyObject_Call(cobj, cobj, cobj) -> cobj as _PyObject_Call
|
|
from C import PyObject_GetAttr(cobj, cobj) -> cobj as _PyObject_GetAttr
|
|
from C import PyObject_GetAttrString(cobj, cobj) -> cobj as _PyObject_GetAttrString
|
|
from C import PyObject_GetIter(cobj) -> cobj as _PyObject_GetIter
|
|
from C import PyObject_HasAttrString(cobj, cobj) -> int as _PyObject_HasAttrString
|
|
from C import PyObject_IsTrue(cobj) -> int as _PyObject_IsTrue
|
|
from C import PyObject_Length(cobj) -> int as _PyObject_Length
|
|
from C import PyObject_LengthHint(cobj, int) -> int as _PyObject_LengthHint
|
|
from C import PyObject_SetAttrString(cobj, cobj, cobj) -> cobj as _PyObject_SetAttrString
|
|
from C import PyObject_Str(cobj) -> cobj as _PyObject_Str
|
|
from C import PyObject_Repr(cobj) -> cobj as _PyObject_Repr
|
|
from C import PyObject_Hash(cobj) -> int as _PyObject_Hash
|
|
from C import PyObject_GetItem(cobj, cobj) -> cobj as _PyObject_GetItem
|
|
from C import PyObject_SetItem(cobj, cobj, cobj) -> int as _PyObject_SetItem
|
|
from C import PyObject_DelItem(cobj, cobj) -> int as _PyObject_DelItem
|
|
from C import PyObject_RichCompare(cobj, cobj, i32) -> cobj as _PyObject_RichCompare
|
|
from C import PyObject_IsInstance(cobj, cobj) -> i32 as _PyObject_IsInstance
|
|
from C import PyErr_Fetch(Ptr[cobj], Ptr[cobj], Ptr[cobj]) as _PyErr_Fetch
|
|
from C import PyErr_NormalizeException(Ptr[cobj], Ptr[cobj], Ptr[cobj]) as _PyErr_NormalizeException
|
|
from C import PyErr_SetString(cobj, cobj) as _PyErr_SetString
|
|
from C import _Py_NoneStruct: cobj
|
|
from C import _Py_TrueStruct: cobj
|
|
from C import _Py_FalseStruct: cobj
|
|
from C import _Py_EllipsisObject: cobj
|
|
from C import _Py_NotImplementedStruct: cobj
|
|
from C import PyLong_Type: cobj as _PyLong_Type
|
|
from C import PyFloat_Type: cobj as _PyFloat_Type
|
|
from C import PyBool_Type: cobj as _PyBool_Type
|
|
from C import PyUnicode_Type: cobj as _PyUnicode_Type
|
|
from C import PyBytes_Type: cobj as _PyBytes_Type
|
|
from C import PyComplex_Type: cobj as _PyComplex_Type
|
|
from C import PyList_Type: cobj as _PyList_Type
|
|
from C import PyDict_Type: cobj as _PyDict_Type
|
|
from C import PySet_Type: cobj as _PySet_Type
|
|
from C import PyTuple_Type: cobj as _PyTuple_Type
|
|
from C import PySlice_Type: cobj as _PySlice_Type
|
|
from C import PyCapsule_Type: cobj as _PyCapsule_Type
|
|
from C import PyExc_BaseException: cobj as _PyExc_BaseException
|
|
from C import PyExc_Exception: cobj as _PyExc_Exception
|
|
from C import PyExc_NameError: cobj as _PyExc_NameError
|
|
from C import PyExc_OSError: cobj as _PyExc_OSError
|
|
from C import PyExc_IOError: cobj as _PyExc_IOError
|
|
from C import PyExc_ValueError: cobj as _PyExc_ValueError
|
|
from C import PyExc_LookupError: cobj as _PyExc_LookupError
|
|
from C import PyExc_IndexError: cobj as _PyExc_IndexError
|
|
from C import PyExc_KeyError: cobj as _PyExc_KeyError
|
|
from C import PyExc_TypeError: cobj as _PyExc_TypeError
|
|
from C import PyExc_ArithmeticError: cobj as _PyExc_ArithmeticError
|
|
from C import PyExc_ZeroDivisionError: cobj as _PyExc_ZeroDivisionError
|
|
from C import PyExc_OverflowError: cobj as _PyExc_OverflowError
|
|
from C import PyExc_AttributeError: cobj as _PyExc_AttributeError
|
|
from C import PyExc_RuntimeError: cobj as _PyExc_RuntimeError
|
|
from C import PyExc_NotImplementedError: cobj as _PyExc_NotImplementedError
|
|
from C import PyExc_StopIteration: cobj as _PyExc_StopIteration
|
|
from C import PyExc_AssertionError: cobj as _PyExc_AssertionError
|
|
from C import PyExc_EOFError: cobj as _PyExc_EOFError
|
|
from C import PyExc_SystemExit: cobj as _PyExc_SystemExit
|
|
|
|
global Py_DecRef
|
|
global Py_IncRef
|
|
global Py_Initialize
|
|
global PyImport_AddModule
|
|
global PyImport_AddModuleObject
|
|
global PyImport_ImportModule
|
|
global PyRun_SimpleString
|
|
global PyEval_GetGlobals
|
|
global PyEval_GetBuiltins
|
|
global PyOS_setsig
|
|
global PyLong_AsLong
|
|
global PyLong_FromLong
|
|
global PyFloat_AsDouble
|
|
global PyFloat_FromDouble
|
|
global PyBool_FromLong
|
|
global PyBytes_AsString
|
|
global PyBytes_Size
|
|
global PyList_New
|
|
global PyList_Size
|
|
global PyList_GetItem
|
|
global PyList_SetItem
|
|
global PyDict_New
|
|
global PyDict_Next
|
|
global PyDict_GetItem
|
|
global PyDict_GetItemString
|
|
global PyDict_SetItem
|
|
global PyDict_Size
|
|
global PySet_Add
|
|
global PySet_New
|
|
global PyTuple_New
|
|
global PyTuple_Size
|
|
global PyTuple_GetItem
|
|
global PyTuple_SetItem
|
|
global PyUnicode_AsEncodedString
|
|
global PyUnicode_DecodeFSDefaultAndSize
|
|
global PyUnicode_FromString
|
|
global PyComplex_FromDoubles
|
|
global PyComplex_RealAsDouble
|
|
global PyComplex_ImagAsDouble
|
|
global PyIter_Next
|
|
global PySlice_New
|
|
global PySlice_Unpack
|
|
global PyCapsule_New
|
|
global PyCapsule_GetPointer
|
|
global PyNumber_Add
|
|
global PyNumber_Subtract
|
|
global PyNumber_Multiply
|
|
global PyNumber_MatrixMultiply
|
|
global PyNumber_FloorDivide
|
|
global PyNumber_TrueDivide
|
|
global PyNumber_Remainder
|
|
global PyNumber_Divmod
|
|
global PyNumber_Power
|
|
global PyNumber_Negative
|
|
global PyNumber_Positive
|
|
global PyNumber_Absolute
|
|
global PyNumber_Invert
|
|
global PyNumber_Lshift
|
|
global PyNumber_Rshift
|
|
global PyNumber_And
|
|
global PyNumber_Xor
|
|
global PyNumber_Or
|
|
global PyNumber_InPlaceAdd
|
|
global PyNumber_InPlaceSubtract
|
|
global PyNumber_InPlaceMultiply
|
|
global PyNumber_InPlaceMatrixMultiply
|
|
global PyNumber_InPlaceFloorDivide
|
|
global PyNumber_InPlaceTrueDivide
|
|
global PyNumber_InPlaceRemainder
|
|
global PyNumber_InPlacePower
|
|
global PyNumber_InPlaceLshift
|
|
global PyNumber_InPlaceRshift
|
|
global PyNumber_InPlaceAnd
|
|
global PyNumber_InPlaceXor
|
|
global PyNumber_InPlaceOr
|
|
global PyNumber_Long
|
|
global PyNumber_Float
|
|
global PyNumber_Index
|
|
global PyObject_Call
|
|
global PyObject_GetAttr
|
|
global PyObject_GetAttrString
|
|
global PyObject_GetIter
|
|
global PyObject_HasAttrString
|
|
global PyObject_IsTrue
|
|
global PyObject_Length
|
|
global PyObject_LengthHint
|
|
global PyObject_SetAttrString
|
|
global PyObject_Str
|
|
global PyObject_Repr
|
|
global PyObject_Hash
|
|
global PyObject_GetItem
|
|
global PyObject_SetItem
|
|
global PyObject_DelItem
|
|
global PyObject_RichCompare
|
|
global PyObject_IsInstance
|
|
global PyErr_Fetch
|
|
global PyErr_NormalizeException
|
|
global PyErr_SetString
|
|
global Py_None
|
|
global Py_True
|
|
global Py_False
|
|
global Py_Ellipsis
|
|
global Py_NotImplemented
|
|
global PyLong_Type
|
|
global PyFloat_Type
|
|
global PyBool_Type
|
|
global PyUnicode_Type
|
|
global PyBytes_Type
|
|
global PyComplex_Type
|
|
global PyList_Type
|
|
global PyDict_Type
|
|
global PySet_Type
|
|
global PyTuple_Type
|
|
global PySlice_Type
|
|
global PyCapsule_Type
|
|
global PyExc_BaseException
|
|
global PyExc_Exception
|
|
global PyExc_NameError
|
|
global PyExc_OSError
|
|
global PyExc_IOError
|
|
global PyExc_ValueError
|
|
global PyExc_LookupError
|
|
global PyExc_IndexError
|
|
global PyExc_KeyError
|
|
global PyExc_TypeError
|
|
global PyExc_ArithmeticError
|
|
global PyExc_ZeroDivisionError
|
|
global PyExc_OverflowError
|
|
global PyExc_AttributeError
|
|
global PyExc_RuntimeError
|
|
global PyExc_NotImplementedError
|
|
global PyExc_StopIteration
|
|
global PyExc_AssertionError
|
|
global PyExc_EOFError
|
|
global PyExc_SystemExit
|
|
|
|
Py_DecRef = _Py_DecRef
|
|
Py_IncRef = _Py_IncRef
|
|
Py_Initialize = _Py_Initialize
|
|
PyImport_AddModule = _PyImport_AddModule
|
|
PyImport_AddModuleObject = _PyImport_AddModuleObject
|
|
PyImport_ImportModule = _PyImport_ImportModule
|
|
PyRun_SimpleString = _PyRun_SimpleString
|
|
PyEval_GetGlobals = _PyEval_GetGlobals
|
|
PyEval_GetBuiltins = _PyEval_GetBuiltins
|
|
PyOS_setsig = _PyOS_setsig
|
|
PyLong_AsLong = _PyLong_AsLong
|
|
PyLong_FromLong = _PyLong_FromLong
|
|
PyFloat_AsDouble = _PyFloat_AsDouble
|
|
PyFloat_FromDouble = _PyFloat_FromDouble
|
|
PyBool_FromLong = _PyBool_FromLong
|
|
PyBytes_AsString = _PyBytes_AsString
|
|
PyBytes_Size = _PyBytes_Size
|
|
PyList_New = _PyList_New
|
|
PyList_Size = _PyList_Size
|
|
PyList_GetItem = _PyList_GetItem
|
|
PyList_SetItem = _PyList_SetItem
|
|
PyDict_New = _PyDict_New
|
|
PyDict_Next = _PyDict_Next
|
|
PyDict_GetItem = _PyDict_GetItem
|
|
PyDict_GetItemString = _PyDict_GetItemString
|
|
PyDict_SetItem = _PyDict_SetItem
|
|
PyDict_Size = _PyDict_Size
|
|
PySet_Add = _PySet_Add
|
|
PySet_New = _PySet_New
|
|
PyTuple_New = _PyTuple_New
|
|
PyTuple_Size = _PyTuple_Size
|
|
PyTuple_GetItem = _PyTuple_GetItem
|
|
PyTuple_SetItem = _PyTuple_SetItem
|
|
PyUnicode_AsEncodedString = _PyUnicode_AsEncodedString
|
|
PyUnicode_DecodeFSDefaultAndSize = _PyUnicode_DecodeFSDefaultAndSize
|
|
PyUnicode_FromString = _PyUnicode_FromString
|
|
PyComplex_FromDoubles = _PyComplex_FromDoubles
|
|
PyComplex_RealAsDouble = _PyComplex_RealAsDouble
|
|
PyComplex_ImagAsDouble = _PyComplex_ImagAsDouble
|
|
PyIter_Next = _PyIter_Next
|
|
PySlice_New = _PySlice_New
|
|
PySlice_Unpack = _PySlice_Unpack
|
|
PyCapsule_New = _PyCapsule_New
|
|
PyCapsule_GetPointer = _PyCapsule_GetPointer
|
|
PyNumber_Add = _PyNumber_Add
|
|
PyNumber_Subtract = _PyNumber_Subtract
|
|
PyNumber_Multiply = _PyNumber_Multiply
|
|
PyNumber_MatrixMultiply = _PyNumber_MatrixMultiply
|
|
PyNumber_FloorDivide = _PyNumber_FloorDivide
|
|
PyNumber_TrueDivide = _PyNumber_TrueDivide
|
|
PyNumber_Remainder = _PyNumber_Remainder
|
|
PyNumber_Divmod = _PyNumber_Divmod
|
|
PyNumber_Power = _PyNumber_Power
|
|
PyNumber_Negative = _PyNumber_Negative
|
|
PyNumber_Positive = _PyNumber_Positive
|
|
PyNumber_Absolute = _PyNumber_Absolute
|
|
PyNumber_Invert = _PyNumber_Invert
|
|
PyNumber_Lshift = _PyNumber_Lshift
|
|
PyNumber_Rshift = _PyNumber_Rshift
|
|
PyNumber_And = _PyNumber_And
|
|
PyNumber_Xor = _PyNumber_Xor
|
|
PyNumber_Or = _PyNumber_Or
|
|
PyNumber_InPlaceAdd = _PyNumber_InPlaceAdd
|
|
PyNumber_InPlaceSubtract = _PyNumber_InPlaceSubtract
|
|
PyNumber_InPlaceMultiply = _PyNumber_InPlaceMultiply
|
|
PyNumber_InPlaceMatrixMultiply = _PyNumber_InPlaceMatrixMultiply
|
|
PyNumber_InPlaceFloorDivide = _PyNumber_InPlaceFloorDivide
|
|
PyNumber_InPlaceTrueDivide = _PyNumber_InPlaceTrueDivide
|
|
PyNumber_InPlaceRemainder = _PyNumber_InPlaceRemainder
|
|
PyNumber_InPlacePower = _PyNumber_InPlacePower
|
|
PyNumber_InPlaceLshift = _PyNumber_InPlaceLshift
|
|
PyNumber_InPlaceRshift = _PyNumber_InPlaceRshift
|
|
PyNumber_InPlaceAnd = _PyNumber_InPlaceAnd
|
|
PyNumber_InPlaceXor = _PyNumber_InPlaceXor
|
|
PyNumber_InPlaceOr = _PyNumber_InPlaceOr
|
|
PyNumber_Long = _PyNumber_Long
|
|
PyNumber_Float = _PyNumber_Float
|
|
PyNumber_Index = _PyNumber_Index
|
|
PyObject_Call = _PyObject_Call
|
|
PyObject_GetAttr = _PyObject_GetAttr
|
|
PyObject_GetAttrString = _PyObject_GetAttrString
|
|
PyObject_GetIter = _PyObject_GetIter
|
|
PyObject_HasAttrString = _PyObject_HasAttrString
|
|
PyObject_IsTrue = _PyObject_IsTrue
|
|
PyObject_Length = _PyObject_Length
|
|
PyObject_LengthHint = _PyObject_LengthHint
|
|
PyObject_SetAttrString = _PyObject_SetAttrString
|
|
PyObject_Str = _PyObject_Str
|
|
PyObject_Repr = _PyObject_Repr
|
|
PyObject_Hash = _PyObject_Hash
|
|
PyObject_GetItem = _PyObject_GetItem
|
|
PyObject_SetItem = _PyObject_SetItem
|
|
PyObject_DelItem = _PyObject_DelItem
|
|
PyObject_RichCompare = _PyObject_RichCompare
|
|
PyObject_IsInstance = _PyObject_IsInstance
|
|
PyErr_Fetch = _PyErr_Fetch
|
|
PyErr_NormalizeException = _PyErr_NormalizeException
|
|
PyErr_SetString = _PyErr_SetString
|
|
Py_None = __ptr__(_Py_NoneStruct).as_byte()
|
|
Py_True = __ptr__(_Py_TrueStruct).as_byte()
|
|
Py_False = __ptr__(_Py_FalseStruct).as_byte()
|
|
Py_Ellipsis = __ptr__(_Py_EllipsisObject).as_byte()
|
|
Py_NotImplemented = __ptr__(_Py_NotImplementedStruct).as_byte()
|
|
PyLong_Type = __ptr__(_PyLong_Type).as_byte()
|
|
PyFloat_Type = __ptr__(_PyFloat_Type).as_byte()
|
|
PyBool_Type = __ptr__(_PyBool_Type).as_byte()
|
|
PyUnicode_Type = __ptr__(_PyUnicode_Type).as_byte()
|
|
PyBytes_Type = __ptr__(_PyBytes_Type).as_byte()
|
|
PyComplex_Type = __ptr__(_PyComplex_Type).as_byte()
|
|
PyList_Type = __ptr__(_PyList_Type).as_byte()
|
|
PyDict_Type = __ptr__(_PyDict_Type).as_byte()
|
|
PySet_Type = __ptr__(_PySet_Type).as_byte()
|
|
PyTuple_Type = __ptr__(_PyTuple_Type).as_byte()
|
|
PySlice_Type = __ptr__(_PySlice_Type).as_byte()
|
|
PyCapsule_Type = __ptr__(_PyCapsule_Type).as_byte()
|
|
PyExc_BaseException = _PyExc_BaseException
|
|
PyExc_Exception = _PyExc_Exception
|
|
PyExc_NameError = _PyExc_NameError
|
|
PyExc_OSError = _PyExc_OSError
|
|
PyExc_IOError = _PyExc_IOError
|
|
PyExc_ValueError = _PyExc_ValueError
|
|
PyExc_LookupError = _PyExc_LookupError
|
|
PyExc_IndexError = _PyExc_IndexError
|
|
PyExc_KeyError = _PyExc_KeyError
|
|
PyExc_TypeError = _PyExc_TypeError
|
|
PyExc_ArithmeticError = _PyExc_ArithmeticError
|
|
PyExc_ZeroDivisionError = _PyExc_ZeroDivisionError
|
|
PyExc_OverflowError = _PyExc_OverflowError
|
|
PyExc_AttributeError = _PyExc_AttributeError
|
|
PyExc_RuntimeError = _PyExc_RuntimeError
|
|
PyExc_NotImplementedError = _PyExc_NotImplementedError
|
|
PyExc_StopIteration = _PyExc_StopIteration
|
|
PyExc_AssertionError = _PyExc_AssertionError
|
|
PyExc_EOFError = _PyExc_EOFError
|
|
PyExc_SystemExit = _PyExc_SystemExit
|
|
|
|
def init_error_py_types():
|
|
BaseException._pytype = PyExc_BaseException
|
|
Exception._pytype = PyExc_Exception
|
|
NameError._pytype = PyExc_NameError
|
|
OSError._pytype = PyExc_OSError
|
|
IOError._pytype = PyExc_IOError
|
|
ValueError._pytype = PyExc_ValueError
|
|
LookupError._pytype = PyExc_LookupError
|
|
IndexError._pytype = PyExc_IndexError
|
|
KeyError._pytype = PyExc_KeyError
|
|
TypeError._pytype = PyExc_TypeError
|
|
ArithmeticError._pytype = PyExc_ArithmeticError
|
|
ZeroDivisionError._pytype = PyExc_ZeroDivisionError
|
|
OverflowError._pytype = PyExc_OverflowError
|
|
AttributeError._pytype = PyExc_AttributeError
|
|
RuntimeError._pytype = PyExc_RuntimeError
|
|
NotImplementedError._pytype = PyExc_NotImplementedError
|
|
StopIteration._pytype = PyExc_StopIteration
|
|
AssertionError._pytype = PyExc_AssertionError
|
|
EOFError._pytype = PyExc_EOFError
|
|
SystemExit._pytype = PyExc_SystemExit
|
|
|
|
def setup_python(python_loaded: bool):
|
|
global _PY_INITIALIZED
|
|
if _PY_INITIALIZED:
|
|
return
|
|
|
|
py_handle = cobj()
|
|
if python_loaded:
|
|
py_handle = dlopen("", RTLD_LOCAL | RTLD_NOW)
|
|
else:
|
|
LD = os.getenv("CODON_PYTHON", default="libpython." + dlext())
|
|
py_handle = dlopen(LD, RTLD_LOCAL | RTLD_NOW)
|
|
|
|
init_handles_dlopen(py_handle)
|
|
init_error_py_types()
|
|
|
|
if not python_loaded:
|
|
Py_Initialize()
|
|
|
|
PyOS_setsig(i32(2), cobj()) # disable CPython's SIGINT handler
|
|
_PY_INITIALIZED = True
|
|
|
|
def ensure_initialized(python_loaded: bool = False):
|
|
if __py_extension__:
|
|
init_handles_static()
|
|
init_error_py_types()
|
|
else:
|
|
setup_python(python_loaded)
|
|
PyRun_SimpleString(_PY_INIT.c_str())
|
|
|
|
def setup_decorator():
|
|
setup_python(True)
|
|
|
|
@tuple
|
|
class _PyArg_Parser:
|
|
initialized: i32
|
|
format: cobj
|
|
keywords: Ptr[cobj]
|
|
fname: cobj
|
|
custom_msg: cobj
|
|
pos: i32
|
|
min: i32
|
|
max: i32
|
|
kwtuple: cobj
|
|
next: cobj
|
|
|
|
def __new__(fname: cobj, keywords: Ptr[cobj], format: cobj):
|
|
z = i32(0)
|
|
o = cobj()
|
|
return _PyArg_Parser(z, format, keywords, fname, o, z, z, z, o, o)
|
|
|
|
@extend
|
|
class pyobj:
|
|
def __new__() -> pyobj:
|
|
return __internal__.class_alloc(pyobj)
|
|
|
|
def __raw__(self) -> Ptr[byte]:
|
|
return __internal__.class_raw(self)
|
|
|
|
def __init__(self, p: Ptr[byte], steal: bool = False):
|
|
self.p = p
|
|
if not steal:
|
|
self.incref()
|
|
|
|
def __del__(self):
|
|
self.decref()
|
|
|
|
def _getattr(self, name: str) -> pyobj:
|
|
return pyobj(pyobj.exc_wrap(PyObject_GetAttrString(self.p, name.c_str())), steal=True)
|
|
|
|
def __add__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Add(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __radd__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Add(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __sub__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Subtract(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rsub__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Subtract(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __mul__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Multiply(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rmul__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Multiply(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __matmul__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_MatrixMultiply(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rmatmul__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_MatrixMultiply(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __floordiv__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_FloorDivide(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rfloordiv__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_FloorDivide(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __truediv__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_TrueDivide(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rtruediv__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_TrueDivide(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __mod__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Remainder(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rmod__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Remainder(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __divmod__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Divmod(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rdivmod__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Divmod(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __pow__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Power(self.p, other.__to_py__(), Py_None)), steal=True)
|
|
|
|
def __rpow__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Power(other.__to_py__(), self.p, Py_None)), steal=True)
|
|
|
|
def __neg__(self):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Negative(self.p)), steal=True)
|
|
|
|
def __pos__(self):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Positive(self.p)), steal=True)
|
|
|
|
def __invert__(self):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Invert(self.p)), steal=True)
|
|
|
|
def __lshift__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Lshift(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rlshift__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Lshift(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __rshift__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Rshift(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rrshift__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Rshift(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __and__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_And(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rand__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_And(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __xor__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Xor(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __rxor__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Xor(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __or__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Or(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __ror__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_Or(other.__to_py__(), self.p)), steal=True)
|
|
|
|
def __iadd__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceAdd(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __isub__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceSubtract(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __imul__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceMultiply(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __imatmul__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceMatrixMultiply(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __ifloordiv__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceFloorDivide(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __itruediv__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceTrueDivide(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __imod__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceRemainder(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __ipow__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlacePower(self.p, other.__to_py__(), Py_None)), steal=True)
|
|
|
|
def __ilshift__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceLshift(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __irshift__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceRshift(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __iand__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceAnd(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __ixor__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceXor(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __ior__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyNumber_InPlaceOr(self.p, other.__to_py__())), steal=True)
|
|
|
|
def __int__(self):
|
|
o = pyobj.exc_wrap(PyNumber_Long(self.p))
|
|
x = int.__from_py__(o)
|
|
pyobj.decref(o)
|
|
return x
|
|
|
|
def __float__(self):
|
|
o = pyobj.exc_wrap(PyNumber_Float(self.p))
|
|
x = float.__from_py__(o)
|
|
pyobj.decref(o)
|
|
return x
|
|
|
|
def __index__(self):
|
|
o = pyobj.exc_wrap(PyNumber_Index(self.p))
|
|
x = int.__from_py__(o)
|
|
pyobj.decref(o)
|
|
return x
|
|
|
|
def __len__(self) -> int:
|
|
return pyobj.exc_wrap(PyObject_Length(self.p))
|
|
|
|
def __length_hint__(self) -> int:
|
|
return pyobj.exc_wrap(PyObject_LengthHint(self.p))
|
|
|
|
def __getitem__(self, key):
|
|
return pyobj(pyobj.exc_wrap(PyObject_GetItem(self.p, key.__to_py__())), steal=True)
|
|
|
|
def __setitem__(self, key, v):
|
|
pyobj.exc_wrap(PyObject_SetItem(self.p, key.__to_py__(), v.__to_py__()))
|
|
|
|
def __delitem__(self, key):
|
|
return pyobj.exc_wrap(PyObject_DelItem(self.p, key.__to_py__()))
|
|
|
|
def __lt__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyObject_RichCompare(self.p, other.__to_py__(), i32(Py_LT))), steal=True)
|
|
|
|
def __le__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyObject_RichCompare(self.p, other.__to_py__(), i32(Py_LE))), steal=True)
|
|
|
|
def __eq__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyObject_RichCompare(self.p, other.__to_py__(), i32(Py_EQ))), steal=True)
|
|
|
|
def __ne__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyObject_RichCompare(self.p, other.__to_py__(), i32(Py_NE))), steal=True)
|
|
|
|
def __gt__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyObject_RichCompare(self.p, other.__to_py__(), i32(Py_GT))), steal=True)
|
|
|
|
def __ge__(self, other):
|
|
return pyobj(pyobj.exc_wrap(PyObject_RichCompare(self.p, other.__to_py__(), i32(Py_GE))), steal=True)
|
|
|
|
def __to_py__(self) -> cobj:
|
|
return self.p
|
|
|
|
def __from_py__(p: cobj) -> pyobj:
|
|
return pyobj(p)
|
|
|
|
def __str__(self) -> str:
|
|
o = pyobj.exc_wrap(PyObject_Str(self.p))
|
|
return pyobj.exc_wrap(str.__from_py__(o))
|
|
|
|
def __repr__(self) -> str:
|
|
o = pyobj.exc_wrap(PyObject_Repr(self.p))
|
|
return pyobj.exc_wrap(str.__from_py__(o))
|
|
|
|
def __hash__(self) -> int:
|
|
return pyobj.exc_wrap(PyObject_Hash(self.p))
|
|
|
|
def __iter__(self) -> Generator[pyobj]:
|
|
it = PyObject_GetIter(self.p)
|
|
if not it:
|
|
raise TypeError("Python object is not iterable")
|
|
try:
|
|
while i := PyIter_Next(it):
|
|
yield pyobj(pyobj.exc_wrap(i), steal=True)
|
|
finally:
|
|
pyobj.decref(it)
|
|
pyobj.exc_check()
|
|
|
|
def to_str(self, errors: str, empty: str = "") -> str:
|
|
return pyobj.to_str(self.p, errors, empty)
|
|
|
|
def to_str(p: cobj, errors: str, empty: str = "") -> str:
|
|
obj = PyUnicode_AsEncodedString(p, "utf-8".ptr, errors.c_str() if errors else "".ptr)
|
|
if obj == cobj():
|
|
return empty
|
|
bts = PyBytes_AsString(obj)
|
|
res = str.from_ptr(bts)
|
|
pyobj.decref(obj)
|
|
return res
|
|
|
|
def exc_check():
|
|
ptype, pvalue, ptraceback = cobj(), cobj(), cobj()
|
|
PyErr_Fetch(__ptr__(ptype), __ptr__(pvalue), __ptr__(ptraceback))
|
|
PyErr_NormalizeException(__ptr__(ptype), __ptr__(pvalue), __ptr__(ptraceback))
|
|
if ptype != cobj():
|
|
py_msg = PyObject_Str(pvalue) if pvalue != cobj() else pvalue
|
|
msg = pyobj.to_str(py_msg, "ignore", "<empty Python message>")
|
|
|
|
pyobj.decref(ptype)
|
|
pyobj.decref(ptraceback)
|
|
pyobj.decref(py_msg)
|
|
|
|
# pyobj.decref(pvalue)
|
|
raise PyError(msg, pyobj(pvalue))
|
|
|
|
def exc_wrap(_retval: T, T: type) -> T:
|
|
pyobj.exc_check()
|
|
return _retval
|
|
|
|
def incref(self):
|
|
Py_IncRef(self.p)
|
|
return self
|
|
|
|
def incref(ptr: Ptr[byte]):
|
|
Py_IncRef(ptr)
|
|
|
|
def decref(self):
|
|
Py_DecRef(self.p)
|
|
return self
|
|
|
|
def decref(ptr: Ptr[byte]):
|
|
Py_DecRef(ptr)
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
args_py = args.__to_py__()
|
|
kws_py = cobj()
|
|
if staticlen(kwargs) > 0:
|
|
names = iter(kwargs.__dict__())
|
|
kws = {next(names): pyobj(i.__to_py__(), steal=True) for i in kwargs}
|
|
kws_py = kws.__to_py__()
|
|
return pyobj(pyobj.exc_wrap(PyObject_Call(self.p, args_py, kws_py)), steal=True)
|
|
|
|
def _tuple_new(length: int):
|
|
return pyobj.exc_wrap(PyTuple_New(length))
|
|
|
|
def _tuple_size(p: cobj):
|
|
return pyobj.exc_wrap(PyTuple_Size(p))
|
|
|
|
def _tuple_set(p: cobj, idx: int, val: cobj):
|
|
PyTuple_SetItem(p, idx, val)
|
|
pyobj.exc_check()
|
|
|
|
def _tuple_get(p: cobj, idx: int) -> cobj:
|
|
return pyobj.exc_wrap(PyTuple_GetItem(p, idx))
|
|
|
|
def _import(name: str) -> pyobj:
|
|
ensure_initialized()
|
|
if name in _PY_MODULE_CACHE:
|
|
return _PY_MODULE_CACHE[name]
|
|
m = pyobj(pyobj.exc_wrap(PyImport_ImportModule(name.c_str())), steal=True)
|
|
_PY_MODULE_CACHE[name] = m
|
|
return m
|
|
|
|
def _exec(code: str):
|
|
ensure_initialized()
|
|
PyRun_SimpleString(code.c_str())
|
|
|
|
def _globals() -> pyobj:
|
|
p = PyEval_GetGlobals()
|
|
if p == cobj():
|
|
Py_IncRef(Py_None)
|
|
return pyobj(Py_None)
|
|
return pyobj(p)
|
|
|
|
def _builtins() -> pyobj:
|
|
return pyobj(PyEval_GetBuiltins())
|
|
|
|
def _get_module(name: str) -> pyobj:
|
|
p = pyobj(pyobj.exc_wrap(PyImport_AddModule(name.c_str())))
|
|
return p
|
|
|
|
def _main_module() -> pyobj:
|
|
return pyobj._get_module("__main__")
|
|
|
|
def _repr_mimebundle_(self, bundle=Set[str]()) -> Dict[str, str]:
|
|
fn = pyobj._main_module()._getattr("__codon_repr__")
|
|
assert fn.p != cobj(), "cannot find python.__codon_repr__"
|
|
mime, txt = Tuple[str, str].__from_py__(fn.__call__(self).p)
|
|
return {mime: txt}
|
|
|
|
def __bool__(self):
|
|
return bool(pyobj.exc_wrap(PyObject_IsTrue(self.p) == 1))
|
|
|
|
def _get_identifier(typ: str) -> pyobj:
|
|
t = pyobj._builtins()[typ]
|
|
if t.p == cobj():
|
|
t = pyobj._main_module()[typ]
|
|
return t
|
|
|
|
def _isinstance(what: pyobj, typ: pyobj) -> bool:
|
|
return bool(pyobj.exc_wrap(PyObject_IsInstance(what.p, typ.p)))
|
|
|
|
@tuple
|
|
class _PyObject_Struct:
|
|
refcnt: int
|
|
pytype: cobj
|
|
|
|
def _conversion_error(name: Static[str]):
|
|
raise PyError("conversion error: Python object did not have type '" + name + "'")
|
|
|
|
def _get_type(o: cobj):
|
|
return Ptr[_PyObject_Struct](o)[0].pytype
|
|
|
|
def _ensure_type(o: cobj, t: cobj, name: Static[str]):
|
|
if _get_type(o) != t:
|
|
_conversion_error(name)
|
|
|
|
|
|
# Type conversions
|
|
|
|
@extend
|
|
class NoneType:
|
|
def __to_py__(self) -> cobj:
|
|
Py_IncRef(Py_None)
|
|
return Py_None
|
|
|
|
def __from_py__(x: cobj) -> None:
|
|
if x != Py_None:
|
|
_conversion_error("NoneType")
|
|
return
|
|
|
|
@extend
|
|
class int:
|
|
def __to_py__(self) -> cobj:
|
|
return pyobj.exc_wrap(PyLong_FromLong(self))
|
|
|
|
def __from_py__(i: cobj) -> int:
|
|
_ensure_type(i, PyLong_Type, "int")
|
|
return PyLong_AsLong(i)
|
|
|
|
@extend
|
|
class float:
|
|
def __to_py__(self) -> cobj:
|
|
return pyobj.exc_wrap(PyFloat_FromDouble(self))
|
|
|
|
def __from_py__(d: cobj) -> float:
|
|
return pyobj.exc_wrap(PyFloat_AsDouble(d))
|
|
|
|
@extend
|
|
class bool:
|
|
def __to_py__(self) -> cobj:
|
|
return pyobj.exc_wrap(PyBool_FromLong(int(self)))
|
|
|
|
def __from_py__(b: cobj) -> bool:
|
|
_ensure_type(b, PyBool_Type, "bool")
|
|
return PyObject_IsTrue(b) != 0
|
|
|
|
@extend
|
|
class byte:
|
|
def __to_py__(self) -> cobj:
|
|
return str.__to_py__(str(__ptr__(self), 1))
|
|
|
|
def __from_py__(c: cobj) -> byte:
|
|
return str.__from_py__(c).ptr[0]
|
|
|
|
@extend
|
|
class str:
|
|
def __to_py__(self) -> cobj:
|
|
return pyobj.exc_wrap(PyUnicode_DecodeFSDefaultAndSize(self.ptr, self.len))
|
|
|
|
def __from_py__(s: cobj) -> str:
|
|
if _get_type(s) == PyBytes_Type:
|
|
n = PyBytes_Size(s)
|
|
p0 = PyBytes_AsString(s)
|
|
p1 = cobj(n)
|
|
str.memcpy(p1, p0, n)
|
|
return str(p1, n)
|
|
else:
|
|
return pyobj.exc_wrap(pyobj.to_str(s, "strict"))
|
|
|
|
@extend
|
|
class complex:
|
|
def __to_py__(self) -> cobj:
|
|
return pyobj.exc_wrap(PyComplex_FromDoubles(self.real, self.imag))
|
|
|
|
def __from_py__(c: cobj) -> complex:
|
|
_ensure_type(c, PyComplex_Type, "complex")
|
|
real = PyComplex_RealAsDouble(c)
|
|
imag = PyComplex_ImagAsDouble(c)
|
|
return complex(real, imag)
|
|
|
|
@extend
|
|
class List:
|
|
def __to_py__(self) -> cobj:
|
|
pylist = PyList_New(len(self))
|
|
pyobj.exc_check()
|
|
idx = 0
|
|
for a in self:
|
|
PyList_SetItem(pylist, idx, a.__to_py__())
|
|
pyobj.exc_check()
|
|
idx += 1
|
|
return pylist
|
|
|
|
def __from_py__(v: cobj) -> List[T]:
|
|
_ensure_type(v, PyList_Type, "list")
|
|
n = PyList_Size(v)
|
|
t = List[T](n)
|
|
for i in range(n):
|
|
elem = PyList_GetItem(v, i)
|
|
t.append(T.__from_py__(elem))
|
|
return t
|
|
|
|
@extend
|
|
class Dict:
|
|
def __to_py__(self) -> cobj:
|
|
pydict = PyDict_New()
|
|
pyobj.exc_check()
|
|
for k, v in self.items():
|
|
PyDict_SetItem(pydict, k.__to_py__(), v.__to_py__())
|
|
pyobj.exc_check()
|
|
return pydict
|
|
|
|
def __from_py__(d: cobj) -> Dict[K, V]:
|
|
_ensure_type(d, PyDict_Type, "dict")
|
|
b = dict[K, V]()
|
|
pos = 0
|
|
k_ptr = cobj()
|
|
v_ptr = cobj()
|
|
while PyDict_Next(d, __ptr__(pos), __ptr__(k_ptr), __ptr__(v_ptr)):
|
|
k = K.__from_py__(k_ptr)
|
|
v = V.__from_py__(v_ptr)
|
|
b[k] = v
|
|
return b
|
|
|
|
@extend
|
|
class Set:
|
|
def __to_py__(self) -> cobj:
|
|
pyset = PySet_New(cobj())
|
|
pyobj.exc_check()
|
|
for a in self:
|
|
PySet_Add(pyset, a.__to_py__())
|
|
pyobj.exc_check()
|
|
return pyset
|
|
|
|
def __from_py__(s: cobj) -> Set[K]:
|
|
_ensure_type(s, PySet_Type, "set")
|
|
b = set[K]()
|
|
s_iter = PyObject_GetIter(s)
|
|
while True:
|
|
k_ptr = pyobj.exc_wrap(PyIter_Next(s_iter))
|
|
if not k_ptr:
|
|
break
|
|
k = K.__from_py__(k_ptr)
|
|
pyobj.decref(k_ptr)
|
|
b.add(k)
|
|
pyobj.decref(s_iter)
|
|
return b
|
|
|
|
@extend
|
|
class DynamicTuple:
|
|
def __to_py__(self) -> cobj:
|
|
pytup = PyTuple_New(len(self))
|
|
i = 0
|
|
for a in self:
|
|
PyTuple_SetItem(pytup, i, a.__to_py__())
|
|
pyobj.exc_check()
|
|
i += 1
|
|
return pytup
|
|
|
|
def __from_py__(t: cobj) -> DynamicTuple[T]:
|
|
_ensure_type(t, PyTuple_Type, "tuple")
|
|
n = PyTuple_Size(t)
|
|
p = Ptr[T](n)
|
|
for i in range(n):
|
|
p[i] = T.__from_py__(PyTuple_GetItem(t, i))
|
|
return DynamicTuple(p, n)
|
|
|
|
@extend
|
|
class Slice:
|
|
def __to_py__(self) -> cobj:
|
|
start = self.start
|
|
stop = self.stop
|
|
step = self.step
|
|
start_py = start.__to_py__() if start is not None else cobj()
|
|
stop_py = stop.__to_py__() if stop is not None else cobj()
|
|
step_py = step.__to_py__() if step is not None else cobj()
|
|
return PySlice_New(start_py, stop_py, step_py)
|
|
|
|
def __from_py__(s: cobj) -> Slice:
|
|
_ensure_type(s, PySlice_Type, "slice")
|
|
start = 0
|
|
stop = 0
|
|
step = 0
|
|
PySlice_Unpack(s, __ptr__(start), __ptr__(stop), __ptr__(step))
|
|
return Slice(Optional(start), Optional(stop), Optional(step))
|
|
|
|
@extend
|
|
class Optional:
|
|
def __to_py__(self) -> cobj:
|
|
if self is None:
|
|
return Py_None
|
|
else:
|
|
return self.__val__().__to_py__()
|
|
|
|
def __from_py__(o: cobj) -> Optional[T]:
|
|
if o == Py_None:
|
|
return Optional[T]()
|
|
else:
|
|
return Optional[T](T.__from_py__(o))
|
|
|
|
@extend
|
|
class ellipsis:
|
|
def __to_py__(self) -> cobj:
|
|
return Py_Ellipsis
|
|
|
|
def __from_py__(e: cobj) -> ellipsis:
|
|
if e == Py_Ellipsis:
|
|
return Ellipsis
|
|
else:
|
|
_conversion_error("ellipsis")
|
|
|
|
__pyenv__: Optional[pyobj] = None
|
|
def _____(): __pyenv__ # make it global!
|
|
|
|
|
|
import internal.static as _S
|
|
|
|
|
|
class _PyWrapError(Static[PyError]):
|
|
def __init__(self, message: str, pytype: pyobj = pyobj(cobj(), steal=True)):
|
|
super().__init__("_PyWrapError", message)
|
|
self.pytype = pytype
|
|
|
|
def __init__(self, e: PyError):
|
|
self.__init__("_PyWrapError", e.message, e.pytype)
|
|
|
|
|
|
class _PyWrap:
|
|
def _dispatch_error(F: Static[str]):
|
|
raise TypeError("could not find callable method '" + F + "' for given arguments")
|
|
|
|
def _wrap(args, T: type, F: Static[str], map):
|
|
for fn in _S.fn_overloads(T, F):
|
|
a = _PyWrap._args_from_py(fn, args)
|
|
if a is None:
|
|
continue
|
|
if _S.fn_can_call(fn, *a):
|
|
try:
|
|
return map(fn, a)
|
|
except PyError as e:
|
|
pass
|
|
_PyWrap._dispatch_error(F)
|
|
|
|
def _wrap_unary(obj: cobj, T: type, F: Static[str]) -> cobj:
|
|
return _PyWrap._wrap(
|
|
(obj,), T=T, F=F,
|
|
map=lambda f, a: f(*a).__to_py__()
|
|
)
|
|
|
|
def _args_from_py(fn, args):
|
|
def err(fail: Ptr[bool], T: type = NoneType) -> T:
|
|
fail[0] = True
|
|
# auto-return zero-initialized T
|
|
|
|
def get_arg(F, p, k, fail: Ptr[bool], i: Static[int]):
|
|
if _S.fn_arg_has_type(F, i):
|
|
return _S.fn_arg_get_type(F, i).__from_py__(p[i]) if p[i] != cobj() else (
|
|
_S.fn_get_default(F, i) if _S.fn_has_default(F, i)
|
|
else err(fail, _S.fn_arg_get_type(F, i))
|
|
)
|
|
else:
|
|
return pyobj(p[i], steal=False) if p[i] != cobj() else (
|
|
_S.fn_get_default(F, i) if _S.fn_has_default(F, i) else err(fail)
|
|
)
|
|
|
|
fail = False
|
|
pargs = Ptr[cobj](__ptr__(args).as_byte())
|
|
try:
|
|
ta = tuple(get_arg(fn, pargs, k, __ptr__(fail), i) for i, k in staticenumerate(_S.fn_args(fn)))
|
|
if fail:
|
|
return None
|
|
return _S.fn_wrap_call_args(fn, *ta)
|
|
except PyError:
|
|
return None
|
|
|
|
def _reorder_args(fn, self: cobj, args: cobj, kwargs: cobj, M: Static[int] = 1):
|
|
nargs = PyTuple_Size(args)
|
|
nkwargs = PyDict_Size(kwargs) if kwargs != cobj() else 0
|
|
|
|
args_ordered = tuple(cobj() for _ in _S.fn_args(fn))
|
|
pargs = Ptr[cobj](__ptr__(args_ordered).as_byte())
|
|
|
|
if nargs + nkwargs + M > len(args_ordered):
|
|
return None
|
|
|
|
if M:
|
|
pargs[0] = self
|
|
|
|
for i in range(nargs):
|
|
pargs[i + M] = PyTuple_GetItem(args, i)
|
|
|
|
kwused = 0
|
|
for i, k in staticenumerate(_S.fn_args(fn)):
|
|
if i < nargs + M:
|
|
continue
|
|
|
|
p = PyDict_GetItemString(kwargs, k.ptr) if nkwargs else cobj()
|
|
if p != cobj():
|
|
pargs[i] = p
|
|
kwused += 1
|
|
|
|
if kwused != nkwargs:
|
|
return None
|
|
|
|
return _PyWrap._args_from_py(fn, args_ordered)
|
|
|
|
def _reorder_args_fastcall(
|
|
fn, self: cobj, args: Ptr[cobj], nargs: int,
|
|
kwds: Ptr[str], nkw: int, M: Static[int] = 1
|
|
):
|
|
args_ordered = tuple(cobj() for _ in _S.fn_args(fn))
|
|
pargs = Ptr[cobj](__ptr__(args_ordered).as_byte())
|
|
|
|
if nargs + M > len(args_ordered):
|
|
return None
|
|
|
|
if M:
|
|
pargs[0] = self
|
|
|
|
for i in range(nargs):
|
|
pargs[i + M] = args[i]
|
|
|
|
for i in range(nargs, nargs + nkw):
|
|
kw = kwds[i - nargs]
|
|
o = args[i]
|
|
|
|
found = False
|
|
j = M
|
|
for i, k in staticenumerate(_S.fn_args(fn)):
|
|
if M and i == 0:
|
|
continue
|
|
if kw == k:
|
|
if not pargs[j]:
|
|
pargs[j] = o
|
|
else:
|
|
return None
|
|
found = True
|
|
break
|
|
j += 1
|
|
if not found:
|
|
return None
|
|
|
|
return _PyWrap._args_from_py(fn, args_ordered)
|
|
|
|
def wrap_magic_abs(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__abs__")
|
|
|
|
def wrap_magic_pos(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__pos__")
|
|
|
|
def wrap_magic_neg(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__neg__")
|
|
|
|
def wrap_magic_invert(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__invert__")
|
|
|
|
def wrap_magic_int(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__int__")
|
|
|
|
def wrap_magic_float(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__float__")
|
|
|
|
def wrap_magic_index(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__index__")
|
|
|
|
def wrap_magic_repr(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__repr__")
|
|
|
|
def wrap_magic_str(obj: cobj, T: type):
|
|
return _PyWrap._wrap_unary(obj, T, "__str__")
|
|
|
|
def _wrap_binary(obj: cobj, obj2: cobj, T: type, F: Static[str]) -> cobj:
|
|
return _PyWrap._wrap(
|
|
(obj, obj2), T=T, F=F,
|
|
map=lambda f, a: f(*a).__to_py__()
|
|
)
|
|
|
|
def wrap_magic_add(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__add__")
|
|
|
|
def wrap_magic_radd(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__radd__")
|
|
|
|
def wrap_magic_iadd(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__iadd__")
|
|
|
|
def wrap_magic_sub(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__sub__")
|
|
|
|
def wrap_magic_rsub(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rsub__")
|
|
|
|
def wrap_magic_isub(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__isub__")
|
|
|
|
def wrap_magic_mul(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__mul__")
|
|
|
|
def wrap_magic_rmul(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rmul__")
|
|
|
|
def wrap_magic_imul(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__imul__")
|
|
|
|
def wrap_magic_mod(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__mod__")
|
|
|
|
def wrap_magic_rmod(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rmod__")
|
|
|
|
def wrap_magic_imod(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__imod__")
|
|
|
|
def wrap_magic_divmod(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__divmod__")
|
|
|
|
def wrap_magic_rdivmod(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rdivmod__")
|
|
|
|
def wrap_magic_lshift(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__lshift__")
|
|
|
|
def wrap_magic_rlshift(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rlshift__")
|
|
|
|
def wrap_magic_ilshift(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__ilshift__")
|
|
|
|
def wrap_magic_rshift(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rshift__")
|
|
|
|
def wrap_magic_rrshift(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rrshift__")
|
|
|
|
def wrap_magic_irshift(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__irshift__")
|
|
|
|
def wrap_magic_and(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__and__")
|
|
|
|
def wrap_magic_rand(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rand__")
|
|
|
|
def wrap_magic_iand(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__iand__")
|
|
|
|
def wrap_magic_xor(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__xor__")
|
|
|
|
def wrap_magic_rxor(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rxor__")
|
|
|
|
def wrap_magic_ixor(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__ixor__")
|
|
|
|
def wrap_magic_or(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__or__")
|
|
|
|
def wrap_magic_ror(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__ror__")
|
|
|
|
def wrap_magic_ior(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__ior__")
|
|
|
|
def wrap_magic_floordiv(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__floordiv__")
|
|
|
|
def wrap_magic_ifloordiv(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__ifloordiv__")
|
|
|
|
def wrap_magic_truediv(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__truediv__")
|
|
|
|
def wrap_magic_itruediv(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__itruediv__")
|
|
|
|
def wrap_magic_matmul(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__matmul__")
|
|
|
|
def wrap_magic_rmatmul(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rmatmul__")
|
|
|
|
def wrap_magic_imatmul(obj: cobj, obj2: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__imatmul__")
|
|
|
|
def wrap_magic_pow(obj: cobj, obj2: cobj, obj3: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__pow__")
|
|
|
|
def wrap_magic_rpow(obj: cobj, obj2: cobj, obj3: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__rpow__")
|
|
|
|
def wrap_magic_ipow(obj: cobj, obj2: cobj, obj3: cobj, T: type):
|
|
return _PyWrap._wrap_binary(obj, obj2, T, "__ipow__")
|
|
|
|
def _wrap_hash(obj: cobj, T: type, F: Static[str]) -> i64:
|
|
return _PyWrap._wrap(
|
|
(obj,), T=T, F=F,
|
|
map=lambda f, a: f(*a)
|
|
)
|
|
|
|
def wrap_magic_len(obj: cobj, T: type):
|
|
return _PyWrap._wrap_hash(obj, T, "__len__")
|
|
|
|
def wrap_magic_hash(obj: cobj, T: type):
|
|
return _PyWrap._wrap_hash(obj, T, "__hash__")
|
|
|
|
def wrap_magic_bool(obj: cobj, T: type) -> i32:
|
|
return _PyWrap._wrap(
|
|
(obj,), T=T, F="__bool__",
|
|
map=lambda f, a: i32(1) if f(*a) else i32(0)
|
|
)
|
|
|
|
def wrap_magic_del(obj: cobj, T: type):
|
|
_PyWrap._wrap(
|
|
(obj,), T=T, F="__del__",
|
|
map=lambda f, a: f(*a)
|
|
)
|
|
|
|
def wrap_magic_contains(obj: cobj, arg: cobj, T: type) -> i32:
|
|
return _PyWrap._wrap(
|
|
(obj, arg,), T=T, F="__contains__",
|
|
map=lambda f, a: i32(1) if f(*a) else i32(0)
|
|
)
|
|
|
|
def wrap_magic_init(obj: cobj, args: cobj, kwargs: cobj, T: type) -> i32:
|
|
if isinstance(T, ByRef):
|
|
F: Static[str] = "__init__"
|
|
for fn in _S.fn_overloads(T, F):
|
|
a = _PyWrap._reorder_args(fn, obj, args, kwargs, M=1)
|
|
if a is not None and _S.fn_can_call(fn, *a):
|
|
fn(*a)
|
|
return i32(0)
|
|
_PyWrap._dispatch_error(F)
|
|
else:
|
|
F: Static[str] = "__new__"
|
|
for fn in _S.fn_overloads(T, F):
|
|
a = _PyWrap._reorder_args(fn, obj, args, kwargs, M=0)
|
|
if a is not None and _S.fn_can_call(fn, *a):
|
|
x = fn(*a)
|
|
p = Ptr[PyObject](obj) + 1
|
|
Ptr[T](p.as_byte())[0] = x
|
|
return i32(0)
|
|
_PyWrap._dispatch_error(F)
|
|
|
|
def wrap_magic_call(obj: cobj, args: cobj, kwargs: cobj, T: type) -> cobj:
|
|
F: Static[str] = "__call__"
|
|
for fn in _S.fn_overloads(T, F):
|
|
a = _PyWrap._reorder_args(fn, obj, args, kwargs, M=1)
|
|
if a is not None and _S.fn_can_call(fn, *a):
|
|
return fn(*a).__to_py__()
|
|
_PyWrap._dispatch_error(F)
|
|
|
|
def _wrap_cmp(obj: cobj, other: cobj, T: type, F: Static[str]) -> cobj:
|
|
return _PyWrap._wrap(
|
|
(obj, other), T=T, F=F,
|
|
map=lambda f, a: f(*a).__to_py__()
|
|
)
|
|
|
|
def wrap_magic_lt(obj: cobj, other: cobj, T: type):
|
|
return _PyWrap._wrap_cmp(obj, other, T, "__lt__")
|
|
|
|
def wrap_magic_le(obj: cobj, other: cobj, T: type):
|
|
return _PyWrap._wrap_cmp(obj, other, T, "__le__")
|
|
|
|
def wrap_magic_eq(obj: cobj, other: cobj, T: type):
|
|
return _PyWrap._wrap_cmp(obj, other, T, "__eq__")
|
|
|
|
def wrap_magic_ne(obj: cobj, other: cobj, T: type):
|
|
return _PyWrap._wrap_cmp(obj, other, T, "__ne__")
|
|
|
|
def wrap_magic_gt(obj: cobj, other: cobj, T: type):
|
|
return _PyWrap._wrap_cmp(obj, other, T, "__gt__")
|
|
|
|
def wrap_magic_ge(obj: cobj, other: cobj, T: type):
|
|
return _PyWrap._wrap_cmp(obj, other, T, "__ge__")
|
|
|
|
def wrap_cmp(obj: cobj, other: cobj, op: i32, C: type) -> cobj:
|
|
if hasattr(C, "__lt__") and op == 0i32:
|
|
return _PyWrap.wrap_magic_lt(obj, other, C)
|
|
elif hasattr(C, "__le__") and op == 1i32:
|
|
return _PyWrap.wrap_magic_le(obj, other, C)
|
|
elif hasattr(C, "__eq__") and op == 2i32:
|
|
return _PyWrap.wrap_magic_eq(obj, other, C)
|
|
elif hasattr(C, "__ne__") and op == 3i32:
|
|
return _PyWrap.wrap_magic_ne(obj, other, C)
|
|
elif hasattr(C, "__gt__") and op == 4i32:
|
|
return _PyWrap.wrap_magic_gt(obj, other, C)
|
|
elif hasattr(C, "__ge__") and op == 5i32:
|
|
return _PyWrap.wrap_magic_ge(obj, other, C)
|
|
else:
|
|
Py_IncRef(Py_NotImplemented)
|
|
return Py_NotImplemented
|
|
|
|
def wrap_magic_getitem(obj: cobj, idx: cobj, T: type):
|
|
return _PyWrap._wrap(
|
|
(obj, idx), T=T, F="__getitem__",
|
|
map=lambda f, a: f(*a).__to_py__()
|
|
)
|
|
|
|
def wrap_magic_setitem(obj: cobj, idx: cobj, val: cobj, T: type) -> i32:
|
|
if val == cobj():
|
|
_PyWrap._wrap(
|
|
(obj, idx), T=T, F="__delitem__",
|
|
map=lambda f, a: f(*a)
|
|
)
|
|
else:
|
|
_PyWrap._wrap(
|
|
(obj, idx, val), T=T, F="__setitem__",
|
|
map=lambda f, a: f(*a)
|
|
)
|
|
return i32(0)
|
|
|
|
class IterWrap:
|
|
_gen: cobj
|
|
T: type
|
|
|
|
def _init(obj: cobj, T: type) -> cobj:
|
|
return _PyWrap.IterWrap(T.__from_py__(obj)).__to_py__()
|
|
|
|
@realize_without_self
|
|
def __init__(self, it: T):
|
|
self._gen = it.__iter__().__raw__()
|
|
|
|
def _iter(obj: cobj) -> cobj:
|
|
T # need separate fn for each instantiation
|
|
p = Ptr[PyObject](obj)
|
|
o = p[0]
|
|
p[0] = PyObject(o.refcnt + 1, o.pytype)
|
|
return obj
|
|
|
|
def _iternext(self: cobj) -> cobj:
|
|
pt = _PyWrap.IterWrap[T].__from_py__(self)
|
|
if pt._gen == cobj():
|
|
return cobj()
|
|
|
|
gt = type(T().__iter__())(pt._gen)
|
|
if gt.done():
|
|
pt._gen = cobj()
|
|
return cobj()
|
|
else:
|
|
return gt.next().__to_py__()
|
|
|
|
def __to_py__(self):
|
|
return _PyWrap.wrap_to_py(self)
|
|
|
|
def __from_py__(obj: cobj):
|
|
return _PyWrap.wrap_from_py(obj, _PyWrap.IterWrap[T])
|
|
|
|
def wrap_magic_iter(obj: cobj, T: type) -> cobj:
|
|
return _PyWrap.IterWrap._init(obj, T)
|
|
|
|
def wrap_multiple(
|
|
obj: cobj, args: Ptr[cobj], nargs: int, _kwds: cobj, T: type, F: Static[str],
|
|
M: Static[int] = 1
|
|
):
|
|
kwds = Ptr[str]()
|
|
nkw = 0
|
|
if _kwds:
|
|
nkw = PyTuple_Size(_kwds)
|
|
kwds = Ptr[str](nkw)
|
|
for i in range(nkw):
|
|
kwds[i] = str.__from_py__(PyTuple_GetItem(_kwds, i))
|
|
|
|
for fn in _S.fn_overloads(T, F):
|
|
a = _PyWrap._reorder_args_fastcall(fn, obj, args, nargs, kwds, nkw, M)
|
|
if a is not None and _S.fn_can_call(fn, *a):
|
|
return fn(*a).__to_py__()
|
|
|
|
_PyWrap._dispatch_error(F)
|
|
|
|
def wrap_get(obj: cobj, closure: cobj, T: type, S: Static[str]):
|
|
return getattr(T.__from_py__(obj), S).__to_py__()
|
|
|
|
def wrap_set(obj: cobj, what: cobj, closure: cobj, T: type, S: Static[str]) -> i32:
|
|
t = T.__from_py__(obj)
|
|
val = type(getattr(t, S)).__from_py__(what)
|
|
setattr(t, S, val)
|
|
return i32(0)
|
|
|
|
def py_type(T: type) -> cobj:
|
|
return cobj()
|
|
|
|
def wrap_to_py(o) -> cobj:
|
|
O = type(o)
|
|
P = PyWrapper[O]
|
|
sz = sizeof(P)
|
|
pytype = _PyWrap.py_type(O)
|
|
mem = alloc_atomic_uncollectable(sz) if atomic(O) else alloc_uncollectable(sz)
|
|
obj = Ptr[P](mem.as_byte())
|
|
obj[0] = PyWrapper(PyObject(1, pytype), o)
|
|
return obj.as_byte()
|
|
|
|
def wrap_from_py(o: cobj, T: type) -> T:
|
|
obj = Ptr[PyWrapper[T]](o)[0]
|
|
pytype = _PyWrap.py_type(T)
|
|
if obj.head.pytype != pytype:
|
|
_conversion_error(T.__name__)
|
|
return obj.data
|