Make msgpack package instead of module. and using Cython in setup script.

1 files changed, 179 insertions, 0 deletions

diff --git a/python/msgpack/unpack.h b/python/msgpack/unpack.h
new file mode 100644
index 0000000..694e816
--- /dev/null
+++ b/python/msgpack/unpack.h
@@ -0,0 +1,179 @@
+/*
+ * MessagePack for Python unpacking routine
+ *
+ * Copyright (C) 2009 Naoki INADA
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+
+#include <map>
+#include <string>
+
+#define MSGPACK_MAX_STACK_SIZE  (1024)
+#include "msgpack/unpack_define.h"
+
+using namespace std;
+
+typedef  struct unpack_user {
+    struct array_stack_type{unsigned int size, last;};
+    array_stack_type array_stack[MSGPACK_MAX_STACK_SIZE];
+    int array_current;
+
+    map<string, PyObject*> str_cache;
+
+    ~unpack_user() {
+        map<string, PyObject*>::iterator it, itend;
+        itend = str_cache.end();
+        for (it = str_cache.begin(); it != itend; ++it) {
+            Py_DECREF(it->second);
+        }
+    }
+} unpack_user;
+
+
+#define msgpack_unpack_struct(name) \
+	struct template ## name
+
+#define msgpack_unpack_func(ret, name) \
+	static inline ret template ## name
+
+#define msgpack_unpack_callback(name) \
+	template_callback ## name
+
+#define msgpack_unpack_object PyObject*
+
+#define msgpack_unpack_user unpack_user
+
+
+struct template_context;
+typedef struct template_context template_context;
+
+static inline msgpack_unpack_object template_callback_root(unpack_user* u)
+{
+    u->array_current = -1;
+    return NULL;
+}
+
+static inline int template_callback_uint8(unpack_user* u, uint8_t d, msgpack_unpack_object* o)
+{ *o = PyInt_FromLong((long)d); return 0; }
+
+static inline int template_callback_uint16(unpack_user* u, uint16_t d, msgpack_unpack_object* o)
+{ *o = PyInt_FromLong((long)d); return 0; }
+
+static inline int template_callback_uint32(unpack_user* u, uint32_t d, msgpack_unpack_object* o)
+{
+    if (d > LONG_MAX) {
+        *o = PyLong_FromUnsignedLong((unsigned long)d);
+    } else {
+        *o = PyInt_FromLong((long)d);
+    }
+    return 0;
+}
+
+static inline int template_callback_uint64(unpack_user* u, uint64_t d, msgpack_unpack_object* o)
+{ *o = PyLong_FromUnsignedLongLong(d); return 0; }
+
+static inline int template_callback_int8(unpack_user* u, int8_t d, msgpack_unpack_object* o)
+{ *o = PyInt_FromLong(d); return 0; }
+
+static inline int template_callback_int16(unpack_user* u, int16_t d, msgpack_unpack_object* o)
+{ *o = PyInt_FromLong(d); return 0; }
+
+static inline int template_callback_int32(unpack_user* u, int32_t d, msgpack_unpack_object* o)
+{ *o = PyInt_FromLong(d); return 0; }
+
+static inline int template_callback_int64(unpack_user* u, int64_t d, msgpack_unpack_object* o)
+{ *o = PyLong_FromLongLong(d); return 0; }
+
+static inline int template_callback_float(unpack_user* u, float d, msgpack_unpack_object* o)
+{ *o = PyFloat_FromDouble((double)d); return 0; }
+
+static inline int template_callback_double(unpack_user* u, double d, msgpack_unpack_object* o)
+{ *o = PyFloat_FromDouble(d); return 0; }
+
+static inline int template_callback_nil(unpack_user* u, msgpack_unpack_object* o)
+{ Py_INCREF(Py_None); *o = Py_None; return 0; }
+
+static inline int template_callback_true(unpack_user* u, msgpack_unpack_object* o)
+{ Py_INCREF(Py_True); *o = Py_True; return 0; }
+
+static inline int template_callback_false(unpack_user* u, msgpack_unpack_object* o)
+{ Py_INCREF(Py_False); *o = Py_False; return 0; }
+
+static inline int template_callback_array(unpack_user* u, unsigned int n, msgpack_unpack_object* o)
+{
+    if (n > 0) {
+        int cur = ++u->array_current;
+        u->array_stack[cur].size = n;
+        u->array_stack[cur].last = 0;
+        *o = PyList_New(n);
+    }
+    else {
+        *o = PyList_New(0);
+    }
+    return 0;
+}
+
+static inline int template_callback_array_item(unpack_user* u, msgpack_unpack_object* c, msgpack_unpack_object o)
+{
+    int cur = u->array_current;
+    int n = u->array_stack[cur].size;
+    int last = u->array_stack[cur].last;
+
+    PyList_SetItem(*c, last, o);
+    last++;
+    if (last >= n) {
+        u->array_current--;
+    }
+    else {
+        u->array_stack[cur].last = last;
+    }
+    return 0;
+}
+
+static inline int template_callback_map(unpack_user* u, unsigned int n, msgpack_unpack_object* o)
+{
+    *o = PyDict_New();
+    return 0;
+}
+
+static inline int template_callback_map_item(unpack_user* u, msgpack_unpack_object* c, msgpack_unpack_object k, msgpack_unpack_object v)
+{
+    PyDict_SetItem(*c, k, v);
+    Py_DECREF(k);
+    Py_DECREF(v);
+    return 0;
+}
+
+static inline int template_callback_raw(unpack_user* u, const char* b, const char* p, unsigned int l, msgpack_unpack_object* o)
+{
+    if (l < 16) {
+        string s(p, l);
+        map<string,PyObject*>::iterator it = u->str_cache.find(s);
+        if (it != u->str_cache.end()) {
+            *o = it->second;
+            Py_INCREF(*o);
+        }
+        else {
+            *o = PyString_FromStringAndSize(p, l);
+            Py_INCREF(*o);
+            u->str_cache[s] = *o;
+        }
+    }
+    else {
+        *o = PyString_FromStringAndSize(p, l);
+    }
+    return 0;
+}
+
+#include "msgpack/unpack_template.h"