ref: 290c921d1d01119b7d4550770da06c491508508f
dir: /sys/src/cmd/python/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(); } }