add hg and python

1 files changed, 310 insertions, 0 deletions

diff --git a/sys/src/cmd/python/Modules/cgensupport.c b/sys/src/cmd/python/Modules/cgensupport.c
new file mode 100644
index 000000000..7e7d0ff9f
--- /dev/null
+++ b/sys/src/cmd/python/Modules/cgensupport.c
@@ -0,0 +1,310 @@
+
+/* 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();
+	}
+}