310 lines · 6.0 KB · C
RawDownload
 /* 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. */ intPyArg_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;} intPyArg_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;} intPyArg_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 intextractdouble(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 intextractfloat(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();} intPyArg_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;} intPyArg_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;} intPyArg_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;} intPyArg_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();} intPyArg_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. */ intPyArg_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();	}} intPyArg_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();	}} intPyArg_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();	}} intPyArg_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();	}}