mirror of https://github.com/python/cpython
107 lines
2.7 KiB
C
107 lines
2.7 KiB
C
#include "parts.h"
|
|
#include "util.h"
|
|
|
|
|
|
static PyObject *
|
|
complex_fromccomplex(PyObject *Py_UNUSED(module), PyObject *obj)
|
|
{
|
|
Py_complex complex;
|
|
|
|
if (!PyArg_Parse(obj, "D", &complex)) {
|
|
return NULL;
|
|
}
|
|
|
|
return PyComplex_FromCComplex(complex);
|
|
}
|
|
|
|
static PyObject *
|
|
complex_asccomplex(PyObject *Py_UNUSED(module), PyObject *obj)
|
|
{
|
|
Py_complex complex;
|
|
|
|
NULLABLE(obj);
|
|
complex = PyComplex_AsCComplex(obj);
|
|
|
|
if (complex.real == -1. && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
|
|
return PyComplex_FromCComplex(complex);
|
|
}
|
|
|
|
static PyObject*
|
|
_py_c_neg(PyObject *Py_UNUSED(module), PyObject *num)
|
|
{
|
|
Py_complex complex;
|
|
|
|
complex = PyComplex_AsCComplex(num);
|
|
if (complex.real == -1. && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
|
|
return PyComplex_FromCComplex(_Py_c_neg(complex));
|
|
}
|
|
|
|
#define _PY_C_FUNC2(suffix) \
|
|
static PyObject * \
|
|
_py_c_##suffix(PyObject *Py_UNUSED(module), PyObject *args) \
|
|
{ \
|
|
Py_complex num, exp, res; \
|
|
\
|
|
if (!PyArg_ParseTuple(args, "DD", &num, &exp)) { \
|
|
return NULL; \
|
|
} \
|
|
\
|
|
errno = 0; \
|
|
res = _Py_c_##suffix(num, exp); \
|
|
return Py_BuildValue("Di", &res, errno); \
|
|
};
|
|
|
|
_PY_C_FUNC2(sum)
|
|
_PY_C_FUNC2(diff)
|
|
_PY_C_FUNC2(prod)
|
|
_PY_C_FUNC2(quot)
|
|
_PY_C_FUNC2(pow)
|
|
|
|
static PyObject*
|
|
_py_c_abs(PyObject *Py_UNUSED(module), PyObject* obj)
|
|
{
|
|
Py_complex complex;
|
|
double res;
|
|
|
|
NULLABLE(obj);
|
|
complex = PyComplex_AsCComplex(obj);
|
|
|
|
if (complex.real == -1. && PyErr_Occurred()) {
|
|
return NULL;
|
|
}
|
|
|
|
errno = 0;
|
|
res = _Py_c_abs(complex);
|
|
return Py_BuildValue("di", res, errno);
|
|
}
|
|
|
|
|
|
static PyMethodDef test_methods[] = {
|
|
{"complex_fromccomplex", complex_fromccomplex, METH_O},
|
|
{"complex_asccomplex", complex_asccomplex, METH_O},
|
|
{"_py_c_sum", _py_c_sum, METH_VARARGS},
|
|
{"_py_c_diff", _py_c_diff, METH_VARARGS},
|
|
{"_py_c_neg", _py_c_neg, METH_O},
|
|
{"_py_c_prod", _py_c_prod, METH_VARARGS},
|
|
{"_py_c_quot", _py_c_quot, METH_VARARGS},
|
|
{"_py_c_pow", _py_c_pow, METH_VARARGS},
|
|
{"_py_c_abs", _py_c_abs, METH_O},
|
|
{NULL},
|
|
};
|
|
|
|
int
|
|
_PyTestCapi_Init_Complex(PyObject *mod)
|
|
{
|
|
if (PyModule_AddFunctions(mod, test_methods) < 0) {
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|