Как я могу написать функцию C, которая принимает либо int, либо float?

Question

Я хочу создать функцию в C, которая расширяет Python и может принимать входные данные либо типа float, либо типа int. В общем, я хочу f(5) а также f(5.5) быть приемлемым входом.

Я не думаю, что я могу использовать if (!PyArg_ParseTuple(args, "i", $value)) потому что он принимает только int или только float.

Как я могу сделать так, чтобы моя функция позволяла вводить данные как целые числа или числа с плавающей точкой?

Мне интересно, должен ли я просто взять входные данные и поместить их в PyObject и каким-то образом взять тип PyObject - это правильный подход?

Answer 1

Если вы объявите функцию C для принятия значений с плавающей точкой, компилятор не будет жаловаться, если вы передадите ей int. Например, эта программа выдает ответ 2.000000:

#include <stdio.h>

float f(float x) {
  return x+1;
}

int main() {
  int i=1;
  printf ("%f", f(i));
}

Версия модуля Python, iorf.c:

#include <Python.h>

static PyObject *IorFError;

float f(float x) {
  return x+1;
}


static PyObject *
fwrap(PyObject *self, PyObject *args) {
  float in=0.0;
  if (!PyArg_ParseTuple(args, "f", &in))
    return NULL;
  return Py_BuildValue("f", f(in));
}

static PyMethodDef IorFMethods[] = {
    {"fup",  fwrap, METH_VARARGS,
     "Arg + 1"},
    {NULL, NULL, 0, NULL}        /* Sentinel */
};


PyMODINIT_FUNC
initiorf(void)
{
  PyObject *m;

  m = Py_InitModule("iorf", IorFMethods);
  if (m == NULL)
    return;

  IorFError = PyErr_NewException("iorf.error", NULL, NULL);
  Py_INCREF(IorFError);
  PyModule_AddObject(m, "error", IorFError);
}

Setup.py:

from distutils.core import setup, Extension

module1 = Extension('iorf',
                    sources = ['iorf.c'])

setup (name = 'iorf',
       version = '0.1',
       description = 'This is a test package',
       ext_modules = [module1])

Пример:

03:21 $ python
Python 2.7.10 (default, Jul 30 2016, 18:31:42)
[GCC 4.2.1 Compatible Apple LLVM 8.0.0 (clang-800.0.34)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import iorf
>>> print iorf.fup(2)
3.0
>>> print iorf.fup(2.5)
3.5

Answer 2

Поплавки (обычно) передаются через регистры, а целые (обычно) передаются через стек. Это означает, что вы буквально не можете внутри функции проверить, является ли аргумент float или int.

Единственный обходной путь - использовать переменные аргументы, причем первый аргумент указывает тип как int или double (не с плавающей точкой).

func_int_or_double (uint8_t type, ...) {
va_list ap;
va_start (ap, type);
int intarg;
double doublearg;
if (type==1) {
   intarg = va_arg (ap, int);
}
if (type==2) {
   doublearg = va_arg (ap, double);
}
va_end (ap);
// Your code goes here
}

Хотя я не совсем уверен, что Python может справиться с вызовом функций с переменным числом, так что YMMV. В качестве последнего усилия вы всегда можете спринтфить значение в буфер и позволить вашей функции sscanf float/int из буфера.

Answer 3

Вы можете проверить тип входного значения следующим образом:

    PyObject* check_type(PyObject*self, PyObject*args) {
    PyObject*any;

    if (!PyArg_ParseTuple(args, "O", &any)) {
        PyErr_SetString(PyExc_TypeError, "Nope.");
        return NULL;
    }

    if (PyFloat_Check(any)) {
        printf("indeed float");
    }
    else {
        printf("\nint\n");
    }

    Py_INCREF(Py_None);

    return Py_None;
}

Вы можете извлечь значение с плавающей точкой из объекта, используя:

double result=PyFloat_AsDouble(any);

Но в этой конкретной ситуации, вероятно, нет необходимости делать это, независимо от того, что вы анализируете int или float, вы можете взять его как float и проверить округлость:

    float target;
    if (!PyArg_ParseTuple(args, "f", &target)) {
                PyErr_SetString(PyExc_TypeError, "Nope.");
                return NULL;
    }

    if (target - (int)target) {
        printf("\n input is float \n");
    }
    else {
        printf("\n input is int \n");
    }