Python C ++ API-在函数重载中返回不同类型

时间:2019-05-17 15:55:02

标签: python c++ overloading python-c-api

我有一个C ++类,其中包含两个重载方法 sayHi(),每个方法具有不同的返回类型:

class Box {
   public:

      Box();

      void sayHi(char *name);

      int sayHi(int number);

};

当我尝试包装此函数时,我遵循了ShadowRanger的建议,然后 “编写一个包装器函数,该函数对提供的参数进行类型检查,并将其分派到正确的'real'函数”。

这里是:

static PyObject *pyBox_sayHi_char(PyBox *self, char *Name)
{
    self->bx->sayHi(Name);

    Py_RETURN_NONE;
}

static int *pyBox_sayHi_int(PyBox *self,  int number)
{
    int answer;

    answer = self->bx->sayHi(number);

    return answer;
}

static PyObject *Hi_overload_switch(PyBox *self,  PyObject *args)
{
    PyObject *x;

    if (!PyArg_ParseTuple(args, "O", &x))
        return NULL;

    if (PyUnicode_Check(x))
    {
        const char* s = PyBytes_AsString(PyUnicode_AsUTF8String(x)); // convert PyObject to char*

        Py_DECREF(x);

        return pyBox_sayHi_char(self, s);
    }
    if (PyLong_Check(x))
    {   
        return pyBox_sayHi_int( self, PyLong_AsLong(x) );
    }
    Py_RETURN_NOTIMPLEMENTED;
}

和以下方法表:

static PyMethodDef pyBox_methods[] = {
    {"Hi", (PyCFunction)Hi_overload_switch, METH_VARARGS, "Hi"},
    {NULL, NULL, 0, NULL}
};

但是,我遇到了错误:

error: cannot convert ‘int*’ to ‘PyObject*’ {aka ‘_object*’} in return
         return pyBox_sayHi_int( self, PyLong_AsLong(x) );

AFAIK,我不想将返回类型转换为PyObjects,因为重载的方法可能需要返回自定义类型。因此,在我的示例中, int void 仅用于说明目的-我可能已经写了 PyBanana PyOrange 数据类型。

我的问题是,如何使函数 Hi_overload_switch 返回不同的类型?

谢谢!

1 个答案:

答案 0 :(得分:1)

您需要返回Python int。在C / C ++源代码中,这是PyLongObject(这是PyObject的子类型,因此可以将PyLongObject*强制转换为PyObject*并返回)。要将C ++ int转换为PyObject* Python int,请使用PyObject* PyLong_FromLong(long)

static PyObject *pyBox_sayHi_int(PyBox *self,  int number)
{
    int answer;

    answer = self->bx->sayHi(number);

    return PyLong_FromLong(answer);
}