source: vendor/python/2.5/Modules/cgensupport.c

/* Functions used by cgen output */

#include "Python.h"
#include "cgensupport.h"


/* Functions to extract arguments.
   These needs to know the total number of arguments supplied,
   since the argument list is a tuple only of there is more than
   one argument. */

int
PyArg_GetObject(register PyObject *args, int nargs, int i, PyObject **p_arg)
{
        if (nargs != 1) {
                if (args == NULL || !PyTuple_Check(args) ||
                                nargs != PyTuple_Size(args) ||
                                i < 0 || i >= nargs) {
                        return PyErr_BadArgument();
                }
                else {
                        args = PyTuple_GetItem(args, i);
                }
        }
        if (args == NULL) {
                return PyErr_BadArgument();
        }
        *p_arg = args;
        return 1;
}

int
PyArg_GetLong(register PyObject *args, int nargs, int i, long *p_arg)
{
        if (nargs != 1) {
                if (args == NULL || !PyTuple_Check(args) ||
                                nargs != PyTuple_Size(args) ||
                                i < 0 || i >= nargs) {
                        return PyErr_BadArgument();
                }
                args = PyTuple_GetItem(args, i);
        }
        if (args == NULL || !PyInt_Check(args)) {
                return PyErr_BadArgument();
        }
        *p_arg = PyInt_AsLong(args);
        return 1;
}

int
PyArg_GetShort(register PyObject *args, int nargs, int i, short *p_arg)
{
        long x;
        if (!PyArg_GetLong(args, nargs, i, &x))
                return 0;
        *p_arg = (short) x;
        return 1;
}

static int
extractdouble(register PyObject *v, double *p_arg)
{
        if (v == NULL) {
                /* Fall through to error return at end of function */
        }
        else if (PyFloat_Check(v)) {
                *p_arg = PyFloat_AS_DOUBLE((PyFloatObject *)v);
                return 1;
        }
        else if (PyInt_Check(v)) {
                *p_arg = PyInt_AS_LONG((PyIntObject *)v);
                return 1;
        }
        else if (PyLong_Check(v)) {
                *p_arg = PyLong_AsDouble(v);
                return 1;
        }
        return PyErr_BadArgument();
}

static int
extractfloat(register PyObject *v, float *p_arg)
{
        if (v == NULL) {
                /* Fall through to error return at end of function */
        }
        else if (PyFloat_Check(v)) {
                *p_arg = (float) PyFloat_AS_DOUBLE((PyFloatObject *)v);
                return 1;
        }
        else if (PyInt_Check(v)) {
                *p_arg = (float) PyInt_AS_LONG((PyIntObject *)v);
                return 1;
        }
        else if (PyLong_Check(v)) {
                *p_arg = (float) PyLong_AsDouble(v);
                return 1;
        }
        return PyErr_BadArgument();
}

int
PyArg_GetFloat(register PyObject *args, int nargs, int i, float *p_arg)
{
        PyObject *v;
        float x;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (!extractfloat(v, &x))
                return 0;
        *p_arg = x;
        return 1;
}

int
PyArg_GetString(PyObject *args, int nargs, int i, string *p_arg)
{
        PyObject *v;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (!PyString_Check(v)) {
                return PyErr_BadArgument();
        }
        *p_arg = PyString_AsString(v);
        return 1;
}

int
PyArg_GetChar(PyObject *args, int nargs, int i, char *p_arg)
{
        string x;
        if (!PyArg_GetString(args, nargs, i, &x))
                return 0;
        if (x[0] == '\0' || x[1] != '\0') {
                /* Not exactly one char */
                return PyErr_BadArgument();
        }
        *p_arg = x[0];
        return 1;
}

int
PyArg_GetLongArraySize(PyObject *args, int nargs, int i, long *p_arg)
{
        PyObject *v;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (PyTuple_Check(v)) {
                *p_arg = PyTuple_Size(v);
                return 1;
        }
        if (PyList_Check(v)) {
                *p_arg = PyList_Size(v);
                return 1;
        }
        return PyErr_BadArgument();
}

int
PyArg_GetShortArraySize(PyObject *args, int nargs, int i, short *p_arg)
{
        long x;
        if (!PyArg_GetLongArraySize(args, nargs, i, &x))
                return 0;
        *p_arg = (short) x;
        return 1;
}

/* XXX The following four are too similar.  Should share more code. */

int
PyArg_GetLongArray(PyObject *args, int nargs, int i, int n, long *p_arg)
{
        PyObject *v, *w;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (PyTuple_Check(v)) {
                if (PyTuple_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyTuple_GetItem(v, i);
                        if (!PyInt_Check(w)) {
                                return PyErr_BadArgument();
                        }
                        p_arg[i] = PyInt_AsLong(w);
                }
                return 1;
        }
        else if (PyList_Check(v)) {
                if (PyList_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyList_GetItem(v, i);
                        if (!PyInt_Check(w)) {
                                return PyErr_BadArgument();
                        }
                        p_arg[i] = PyInt_AsLong(w);
                }
                return 1;
        }
        else {
                return PyErr_BadArgument();
        }
}

int
PyArg_GetShortArray(PyObject *args, int nargs, int i, int n, short *p_arg)
{
        PyObject *v, *w;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (PyTuple_Check(v)) {
                if (PyTuple_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyTuple_GetItem(v, i);
                        if (!PyInt_Check(w)) {
                                return PyErr_BadArgument();
                        }
                        p_arg[i] = (short) PyInt_AsLong(w);
                }
                return 1;
        }
        else if (PyList_Check(v)) {
                if (PyList_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyList_GetItem(v, i);
                        if (!PyInt_Check(w)) {
                                return PyErr_BadArgument();
                        }
                        p_arg[i] = (short) PyInt_AsLong(w);
                }
                return 1;
        }
        else {
                return PyErr_BadArgument();
        }
}

int
PyArg_GetDoubleArray(PyObject *args, int nargs, int i, int n, double *p_arg)
{
        PyObject *v, *w;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (PyTuple_Check(v)) {
                if (PyTuple_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyTuple_GetItem(v, i);
                        if (!extractdouble(w, &p_arg[i]))
                                return 0;
                }
                return 1;
        }
        else if (PyList_Check(v)) {
                if (PyList_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyList_GetItem(v, i);
                        if (!extractdouble(w, &p_arg[i]))
                                return 0;
                }
                return 1;
        }
        else {
                return PyErr_BadArgument();
        }
}

int
PyArg_GetFloatArray(PyObject *args, int nargs, int i, int n, float *p_arg)
{
        PyObject *v, *w;
        if (!PyArg_GetObject(args, nargs, i, &v))
                return 0;
        if (PyTuple_Check(v)) {
                if (PyTuple_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyTuple_GetItem(v, i);
                        if (!extractfloat(w, &p_arg[i]))
                                return 0;
                }
                return 1;
        }
        else if (PyList_Check(v)) {
                if (PyList_Size(v) != n) {
                        return PyErr_BadArgument();
                }
                for (i = 0; i < n; i++) {
                        w = PyList_GetItem(v, i);
                        if (!extractfloat(w, &p_arg[i]))
                                return 0;
                }
                return 1;
        }
        else {
                return PyErr_BadArgument();
        }
}