[GR-11498] Improve performance when calling magic methods of native objects.

PullRequest: graalpython/180

Author: fangerer

graalpython/benchmarks/src/micro/c-list-iterating-obj.py

@@ -0,0 +1,200 @@
+# Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
+# DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+#
+# The Universal Permissive License (UPL), Version 1.0
+#
+# Subject to the condition set forth below, permission is hereby granted to any
+# person obtaining a copy of this software, associated documentation and/or
+# data (collectively the "Software"), free of charge and under any and all
+# copyright rights in the Software, and any and all patent rights owned or
+# freely licensable by each licensor hereunder covering either (i) the
+# unmodified Software as contributed to or provided by such licensor, or (ii)
+# the Larger Works (as defined below), to deal in both
+#
+# (a) the Software, and
+#
+# (b) any piece of software and/or hardware listed in the lrgrwrks.txt file if
+# one is included with the Software each a "Larger Work" to which the Software
+# is contributed by such licensors),
+#
+# without restriction, including without limitation the rights to copy, create
+# derivative works of, display, perform, and distribute the Software and make,
+# use, sell, offer for sale, import, export, have made, and have sold the
+# Software and the Larger Work(s), and to sublicense the foregoing rights on
+# either these or other terms.
+#
+# This license is subject to the following condition:
+#
+# The above copyright notice and either this complete permission notice or at a
+# minimum a reference to the UPL must be included in all copies or substantial
+# portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+code = """
+#include "Python.h"
+
+typedef struct {
+    PyObject_HEAD;
+    int bState;
+    uint64_t cnt;
+} NativeMagicMethodsObject;
+
+
+PyObject* nmm_index(PyObject* self) {
+    return PyLong_FromLong((((NativeMagicMethodsObject*)self)->cnt)++);
+}
+
+int nmm_bool(PyObject* self) {
+    int curState = ((NativeMagicMethodsObject*)self)->bState;
+    ((NativeMagicMethodsObject*)self)->bState = !curState;
+    return curState;
+}
+
+PyObject* nmm_iter(PyObject* self) {
+    return NULL;
+}
+
+
+static PyNumberMethods NativeMagicMethods_number_methods = {
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    nmm_bool,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    nmm_index,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    nmm_index,
+
+    0,
+    0,
+
+};
+
+static struct PyMethodDef NativeMagicMethods_methods[] = {
+    {NULL, NULL, 0, NULL},
+    {NULL, NULL, 0, NULL}
+};
+
+static PyTypeObject NativeMagicMethodsType = {
+    PyVarObject_HEAD_INIT(NULL, 0)
+        "NativeMagicMethods.NativeMagicMethods",
+    sizeof(NativeMagicMethodsObject),       /* tp_basicsize */
+    0,                          /* tp_itemsize */
+    0,                          /* tp_dealloc */
+    0,
+    0,
+    0,
+    0,                          /* tp_reserved */
+    0,
+    &NativeMagicMethods_number_methods,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    0,
+    Py_TPFLAGS_DEFAULT,
+    "",
+    0,              /* tp_traverse */
+    0,                 /* tp_clear */
+    0,           /* tp_richcompare */
+    0,                          /* tp_weaklistoffset */
+    nmm_iter,                  /* tp_iter */
+    0,              /* tp_iternext */
+    NativeMagicMethods_methods,             /* tp_methods */
+    NULL,                       /* tp_members */
+    0,                          /* tp_getset */
+    0,                          /* tp_base */
+    0,                  /* tp_dict */
+    0,                          /* tp_descr_get */
+    0,                          /* tp_descr_set */
+    0,                          /* tp_dictoffset */
+    0,                  /* tp_init */
+    PyType_GenericAlloc,        /* tp_alloc */
+    PyType_GenericNew,          /* tp_new */
+    PyObject_Del,               /* tp_free */
+};
+
+static PyModuleDef NativeMagicMethodsmodule = {
+    PyModuleDef_HEAD_INIT,
+    "c_list_iterating_obj_module",
+    "",
+    -1,
+    NULL, NULL, NULL, NULL, NULL
+};
+
+PyMODINIT_FUNC
+PyInit_c_list_iterating_obj_module(void)
+{
+    PyObject* m;
+
+
+    if (PyType_Ready(&NativeMagicMethodsType) < 0)
+        return NULL;
+
+
+    m = PyModule_Create(&NativeMagicMethodsmodule);
+    if (m == NULL)
+        return NULL;
+
+    Py_INCREF(&NativeMagicMethodsType);
+    PyModule_AddObject(m, "NativeMagicMethods", (PyObject *)&NativeMagicMethodsType);
+    return m;
+}
+
+"""
+
+
+ccompile("c_list_iterating_obj_module", code)
+import c_list_iterating_obj_module
+
+def iterate_list(ll, num):
+    idxObj = c_list_iterating_obj_module.NativeMagicMethods()
+    for t in range(num):
+        item = ll[idxObj]
+    return item
+
+
+def measure(num):
+    last_item = iterate_list(list(range(num)), num)
+    print("Last item ", last_item)
+
+
+def __benchmark__(num=1000000):
+    measure(num)
+