diff options
author | nkozlovskiy <nmk@ydb.tech> | 2023-09-29 12:24:06 +0300 |
---|---|---|
committer | nkozlovskiy <nmk@ydb.tech> | 2023-09-29 12:41:34 +0300 |
commit | e0e3e1717e3d33762ce61950504f9637a6e669ed (patch) | |
tree | bca3ff6939b10ed60c3d5c12439963a1146b9711 /contrib/tools/python3/src/Modules/_opcode.c | |
parent | 38f2c5852db84c7b4d83adfcb009eb61541d1ccd (diff) | |
download | ydb-e0e3e1717e3d33762ce61950504f9637a6e669ed.tar.gz |
add ydb deps
Diffstat (limited to 'contrib/tools/python3/src/Modules/_opcode.c')
-rw-r--r-- | contrib/tools/python3/src/Modules/_opcode.c | 114 |
1 files changed, 114 insertions, 0 deletions
diff --git a/contrib/tools/python3/src/Modules/_opcode.c b/contrib/tools/python3/src/Modules/_opcode.c new file mode 100644 index 0000000000..4812716c67 --- /dev/null +++ b/contrib/tools/python3/src/Modules/_opcode.c @@ -0,0 +1,114 @@ +#include "Python.h" +#include "opcode.h" +#include "internal/pycore_code.h" + +/*[clinic input] +module _opcode +[clinic start generated code]*/ +/*[clinic end generated code: output=da39a3ee5e6b4b0d input=117442e66eb376e6]*/ + +#include "clinic/_opcode.c.h" + +/*[clinic input] + +_opcode.stack_effect -> int + + opcode: int + oparg: object = None + / + * + jump: object = None + +Compute the stack effect of the opcode. +[clinic start generated code]*/ + +static int +_opcode_stack_effect_impl(PyObject *module, int opcode, PyObject *oparg, + PyObject *jump) +/*[clinic end generated code: output=64a18f2ead954dbb input=461c9d4a44851898]*/ +{ + int effect; + int oparg_int = 0; + int jump_int; + if (HAS_ARG(opcode)) { + if (oparg == Py_None) { + PyErr_SetString(PyExc_ValueError, + "stack_effect: opcode requires oparg but oparg was not specified"); + return -1; + } + oparg_int = (int)PyLong_AsLong(oparg); + if ((oparg_int == -1) && PyErr_Occurred()) { + return -1; + } + } + else if (oparg != Py_None) { + PyErr_SetString(PyExc_ValueError, + "stack_effect: opcode does not permit oparg but oparg was specified"); + return -1; + } + if (jump == Py_None) { + jump_int = -1; + } + else if (jump == Py_True) { + jump_int = 1; + } + else if (jump == Py_False) { + jump_int = 0; + } + else { + PyErr_SetString(PyExc_ValueError, + "stack_effect: jump must be False, True or None"); + return -1; + } + if (IS_ARTIFICIAL(opcode)) { + effect = PY_INVALID_STACK_EFFECT; + } + else { + effect = PyCompile_OpcodeStackEffectWithJump(opcode, oparg_int, jump_int); + } + if (effect == PY_INVALID_STACK_EFFECT) { + PyErr_SetString(PyExc_ValueError, + "invalid opcode or oparg"); + return -1; + } + return effect; +} + +/*[clinic input] + +_opcode.get_specialization_stats + +Return the specialization stats +[clinic start generated code]*/ + +static PyObject * +_opcode_get_specialization_stats_impl(PyObject *module) +/*[clinic end generated code: output=fcbc32fdfbec5c17 input=e1f60db68d8ce5f6]*/ +{ +#ifdef Py_STATS + return _Py_GetSpecializationStats(); +#else + Py_RETURN_NONE; +#endif +} + +static PyMethodDef +opcode_functions[] = { + _OPCODE_STACK_EFFECT_METHODDEF + _OPCODE_GET_SPECIALIZATION_STATS_METHODDEF + {NULL, NULL, 0, NULL} +}; + +static struct PyModuleDef opcodemodule = { + PyModuleDef_HEAD_INIT, + .m_name = "_opcode", + .m_doc = "Opcode support module.", + .m_size = 0, + .m_methods = opcode_functions +}; + +PyMODINIT_FUNC +PyInit__opcode(void) +{ + return PyModuleDef_Init(&opcodemodule); +} |