Operators like __add__

1 files changed, 242 insertions, 0 deletions

diff --git a/Modules/operator.c b/Modules/operator.c
new file mode 100644
index 0000000..6e0db3d
--- /dev/null
+++ b/Modules/operator.c
@@ -0,0 +1,242 @@
+static char operator_doc[] = "\
+Operator interface.\n\
+\n\
+This module exports a set of functions implemented in C corresponding\n\
+to the intrinsic operators of Python.  For example, operator.add(x, y)\n\
+is equivalent to the expression x+y.  The function names are those\n\
+used for special class methods; variants without leading and trailing\n\
+'__' are also provided for convenience.\n\
+";
+
+/*
+
+     Copyright 
+
+       Copyright 1996 Digital Creations, L.C., 910 Princess Anne
+       Street, Suite 300, Fredericksburg, Virginia 22401 U.S.A. All
+       rights reserved.  Copyright in this software is owned by DCLC,
+       unless otherwise indicated. Permission to use, copy and
+       distribute this software is hereby granted, provided that the
+       above copyright notice appear in all copies and that both that
+       copyright notice and this permission notice appear. Note that
+       any product, process or technology described in this software
+       may be the subject of other Intellectual Property rights
+       reserved by Digital Creations, L.C. and are not licensed
+       hereunder.
+
+     Trademarks 
+
+       Digital Creations & DCLC, are trademarks of Digital Creations, L.C..
+       All other trademarks are owned by their respective companies. 
+
+     No Warranty 
+
+       The software is provided "as is" without warranty of any kind,
+       either express or implied, including, but not limited to, the
+       implied warranties of merchantability, fitness for a particular
+       purpose, or non-infringement. This software could include
+       technical inaccuracies or typographical errors. Changes are
+       periodically made to the software; these changes will be
+       incorporated in new editions of the software. DCLC may make
+       improvements and/or changes in this software at any time
+       without notice.
+
+     Limitation Of Liability 
+
+       In no event will DCLC be liable for direct, indirect, special,
+       incidental, economic, cover, or consequential damages arising
+       out of the use of or inability to use this software even if
+       advised of the possibility of such damages. Some states do not
+       allow the exclusion or limitation of implied warranties or
+       limitation of liability for incidental or consequential
+       damages, so the above limitation or exclusion may not apply to
+       you.
+
+
+    If you have questions regarding this software,
+    contact:
+   
+      Jim Fulton, jim@digicool.com
+      Digital Creations L.C.  
+   
+      (540) 371-6909
+
+    Modifications
+    
+      Renamed and slightly rearranged by Guido van Rossum
+
+*/
+
+#include "Python.h"
+
+#define spam1(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1; \
+  if(! PyArg_ParseTuple(a,"O",&a1)) return NULL; \
+  return AOP(a1); }
+
+#define spam2(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1, *a2; \
+  if(! PyArg_ParseTuple(a,"OO",&a1,&a2)) return NULL; \
+  return AOP(a1,a2); }
+
+#define spamoi(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1; int a2; \
+  if(! PyArg_ParseTuple(a,"Oi",&a1,&a2)) return NULL; \
+  return AOP(a1,a2); }
+
+#define spam2n(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1, *a2; \
+  if(! PyArg_ParseTuple(a,"OO",&a1,&a2)) return NULL; \
+  if(-1 == AOP(a1,a2)) return NULL; \
+  Py_INCREF(Py_None); \
+  return Py_None; }
+
+#define spam3n(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1, *a2, *a3; \
+  if(! PyArg_ParseTuple(a,"OOO",&a1,&a2,&a3)) return NULL; \
+  if(-1 == AOP(a1,a2,a3)) return NULL; \
+  Py_INCREF(Py_None); \
+  return Py_None; }
+
+#define spami(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1; long r; \
+  if(! PyArg_ParseTuple(a,"O",&a1)) return NULL; \
+  if(-1 == (r=AOP(a1))) return NULL; \
+  return PyInt_FromLong(r); }
+
+#define spami2(OP,AOP) static PyObject *OP(s,a) PyObject *s, *a; { \
+  PyObject *a1, *a2; long r; \
+  if(! PyArg_ParseTuple(a,"OO",&a1,&a2)) return NULL; \
+  if(-1 == (r=AOP(a1,a2))) return NULL; \
+  return PyInt_FromLong(r); }
+
+spami(isCallable       , PyCallable_Check)
+spami(isNumberType     , PyNumber_Check)
+spami(truth            , PyObject_IsTrue)
+spam2(__add          , PyNumber_Add)
+spam2(__sub          , PyNumber_Subtract)
+spam2(__mul          , PyNumber_Multiply)
+spam2(__div          , PyNumber_Divide)
+spam2(__mod          , PyNumber_Remainder)
+spam1(__neg          , PyNumber_Negative)
+spam1(__pos          , PyNumber_Positive)
+spam1(__abs          , PyNumber_Absolute)
+spam1(__inv          , PyNumber_Invert)
+spam2(__lshift       , PyNumber_Lshift)
+spam2(__rshift       , PyNumber_Rshift)
+spam2(__and          , PyNumber_And)
+spam2(__xor          , PyNumber_Xor)
+spam2(__or           , PyNumber_Or)
+spami(isSequenceType   , PySequence_Check)
+spam2(__concat       , PySequence_Concat)
+spamoi(__repeat       , PySequence_Repeat)
+spami2(sequenceIncludes, PySequence_In)
+spami2(indexOf         , PySequence_Index)
+spami2(countOf         , PySequence_Count)
+spami(isMappingType    , PyMapping_Check)
+spam2(__getitem      , PyObject_GetItem)
+spam3n(__setitem     , PyObject_SetItem)
+
+static PyObject*
+__getslice(s,a)
+     PyObject *s, *a;
+{
+  PyObject *a1;
+  long a2,a3;
+
+  if(! PyArg_ParseTuple(a,"Oii",&a1,&a2,&a3)) return NULL;
+
+  return PySequence_GetSlice(a1,a2,a3);
+}
+
+static PyObject*
+__setslice(s,a)
+     PyObject *s, *a;
+{
+  PyObject *a1, *a4;
+  long a2,a3;
+
+  if(! PyArg_ParseTuple(a,"OiiO",&a1,&a2,&a3,&a4)) return NULL;
+
+  if(-1 == PySequence_SetSlice(a1,a2,a3,a4)) return NULL;
+
+  Py_INCREF(Py_None);
+  return Py_None;
+}
+
+#undef spam1
+#undef spam2
+#define spam1(OP,DOC) {#OP, OP, 1, #OP "(o) -- " DOC},
+#define spam2(OP,DOC) {#OP, OP, 1, #OP "(a,b) -- " DOC},
+#define spam3(OP,DOC) {#OP, OP, 1, #OP "(a,b,c) -- " DOC},
+
+static struct PyMethodDef operator_methods[] = {
+spam1(isCallable      , "Return 1 if o is callable, and zero otherwise.")
+spam1(isNumberType    , "Return 1 if o has a numeric type, and zero otherwise.")
+spam1(isSequenceType  , "Return 1 if o has a sequence type, and zero otherwise.")
+spam1(truth           , "Return 1 if o is true, and 0 otherwise.")
+spam2(sequenceIncludes, "Return 1 is a includes b, and 0 otherwise.")
+spam2(indexOf         , "Return the index of b in a.")
+spam2(countOf         , "Return the number of times b occurs in a.")
+spam1(isMappingType   , "Return 1 if o has a mapping type, and zero otherwise.")
+
+#undef spam1
+#undef spam2
+#undef spam3
+#define spam1(OP,DOC) {#OP, __ ## OP, 1, #OP "(o) -- " DOC}, \
+            {"__" #OP "__", __ ## OP, 1, #OP "(o) -- " DOC}, 
+#define spam2(OP,DOC) {#OP, __ ## OP, 1, #OP "(a,b) -- " DOC}, \
+            {"__" #OP "__", __ ## OP, 1, #OP "(a,b) -- " DOC}, 
+#define spam3(OP,DOC) {#OP, __ ## OP, 1, #OP "(a,b,c) -- " DOC}, \
+            {"__" #OP "__", __ ## OP, 1, #OP "(a,b,c) -- " DOC}, 
+#define spam4(OP,DOC) {#OP, __ ## OP, 1, #OP "(a,b,c,v) -- " DOC}, \
+            {"__" #OP "__", __ ## OP, 1, #OP "(a,b,c,v) -- " DOC}, 
+
+
+spam2(add         , "Return a + b, for a and b numbers.")
+spam2(sub         , "Return a - b.")
+spam2(mul         , "Return a * b, for a and b numbers.")
+spam2(div         , "Return a / b.")
+spam2(mod         , "Return a % b.")
+spam1(neg         , "Return o negated.")
+spam1(pos         , "Return o positive.")
+spam1(abs         , "Return the absolute value of o.")
+spam1(inv         , "Return the inverse of o.")
+spam2(lshift      , "Return a shifted left by b.")
+spam2(rshift      , "Return a shifted right by b.")
+spam2(and         , "Return the bitwise and of a and b.")
+spam2(xor         , "Return the bitwise exclusive-or of a and b.")
+spam2(or          , "Return the bitwise or of a and b.")
+spam2(concat      , "Return a + b, for a and b sequences.")
+spam2(repeat      , "Return a + b, where a is a sequence, and b is an "
+                        "integer.")
+spam2(getitem     , "Return the value of a at index b.")
+spam3(setitem     , "Set the value of a at b to c.")
+spam2(getslice    , "Return the slice of a from b to c-1.")
+spam4(setslice    , "Set the slice of a from b to c-1 to the sequence, v.")
+	{NULL,		NULL}		/* sentinel */
+
+
+};
+
+
+/* Initialization function for the module (*must* be called initoperator) */
+
+void
+initoperator()
+{
+  PyObject *m, *d;
+  
+  /* Create the module and add the functions */
+  m = Py_InitModule4("operator", operator_methods,
+		     operator_doc,
+		     (PyObject*)NULL,PYTHON_API_VERSION);
+
+  /* Add some symbolic constants to the module */
+  d = PyModule_GetDict(m);
+  PyDict_SetItemString(d, "__version__",PyString_FromString("$Rev$"));
+  
+  /* Check for errors */
+  if (PyErr_Occurred())
+    Py_FatalError("can't initialize module operator");
+}