Add static data

Author: cdce8p

mypyc/lib-rt/init.c

@@ -1,5 +1,6 @@
 #include <Python.h>
 #include "CPy.h"
+#include "static_data.c"
 
 struct ExcDummyStruct _CPy_ExcDummyStruct = { PyObject_HEAD_INIT(NULL) };
 PyObject *_CPy_ExcDummy = (PyObject *)&_CPy_ExcDummyStruct;
@@ -12,6 +13,7 @@ PyObject * __mypyc_empty_tuple__ = NULL;
 // other dynamic libraries. This means we need to initialize
 // things at load time.
 void CPy_Init(void) {
+    intern_strings();
     _CPy_ExcDummyStruct.ob_base.ob_type = &PyBaseObject_Type;
 
     // Initialize system-wide empty tuple constant

mypyc/lib-rt/pythonsupport.h

@@ -11,6 +11,7 @@
 #include "pythoncapi_compat.h"
 #include <frameobject.h>
 #include <assert.h>
+#include "static_data.h"
 #include "mypyc_util.h"
 
 #if CPY_3_13_FEATURES

mypyc/lib-rt/static_data.c

@@ -0,0 +1,58 @@
+#ifndef STATIC_DATA
+#define STATIC_DATA
+
+#include "static_data.h"
+
+// Adopted from numpy 2.4.0: numpy/_core/src/multiarry/npy_static_data.c
+
+mypyc_interned_str_struct mypyc_interned_str;
+
+#define INTERN_STRING(struct_member, string) \
+    assert(mypyc_interned_str.struct_member == NULL); \
+    mypyc_interned_str.struct_member = PyUnicode_InternFromString(string); \
+    if (mypyc_interned_str.struct_member == NULL) { \
+        return -1; \
+    }
+
+int
+intern_strings(void) {
+    INTERN_STRING(__init_subclass__, "__init_subclass__");
+    INTERN_STRING(__mro_entries__, "__mro_entries__");
+    INTERN_STRING(__name__, "__name__");
+    INTERN_STRING(__radd__, "__radd__");
+    INTERN_STRING(__rsub__, "__rsub__");
+    INTERN_STRING(__rmul__, "__rmul__");
+    INTERN_STRING(__rtruediv__, "__rtruediv__");
+    INTERN_STRING(__rmod__, "__rmod__");
+    INTERN_STRING(__rdivmod__, "__rdivmod__");
+    INTERN_STRING(__rfloordiv__, "__rfloordiv__");
+    INTERN_STRING(__rpow__, "__rpow__");
+    INTERN_STRING(__rmatmul__, "__rmatmul__");
+    INTERN_STRING(__rand__, "__rand__");
+    INTERN_STRING(__ror__, "__ror__");
+    INTERN_STRING(__rxor__, "__rxor__");
+    INTERN_STRING(__rlshift__, "__rlshift__");
+    INTERN_STRING(__rrshift__, "__rrshift__");
+    INTERN_STRING(__eq__, "__eq__");
+    INTERN_STRING(__ne__, "__ne__");
+    INTERN_STRING(__gt__, "__gt__");
+    INTERN_STRING(__le__, "__le__");
+    INTERN_STRING(__lt__, "__lt__");
+    INTERN_STRING(__ge__, "__ge__");
+    INTERN_STRING(clear, "clear");
+    INTERN_STRING(close_, "close");
+    INTERN_STRING(copy, "copy");
+    INTERN_STRING(keys, "keys");
+    INTERN_STRING(items, "items");
+    INTERN_STRING(join, "join");
+    INTERN_STRING(send, "send");
+    INTERN_STRING(setdefault, "setdefault");
+    INTERN_STRING(startswith, "startswith");
+    INTERN_STRING(throw_, "throw");
+    INTERN_STRING(translate, "translate");
+    INTERN_STRING(update, "update");
+    INTERN_STRING(values, "values");
+    return 0;
+}
+
+#endif

mypyc/lib-rt/static_data.h

@@ -0,0 +1,59 @@
+#ifndef STATIC_DATA_H
+#define STATIC_DATA_H
+
+#include <Python.h>
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+// Adopted from numpy 2.4.0: numpy/_core/src/multiarry/npy_static_data.h
+
+int intern_strings(void);
+
+typedef struct mypyc_interned_str_struct {
+    PyObject *__init_subclass__;
+    PyObject *__mro_entries__;
+    PyObject *__name__;
+    PyObject *__radd__;
+    PyObject *__rsub__;
+    PyObject *__rmul__;
+    PyObject *__rtruediv__;
+    PyObject *__rmod__;
+    PyObject *__rdivmod__;
+    PyObject *__rfloordiv__;
+    PyObject *__rpow__;
+    PyObject *__rmatmul__;
+    PyObject *__rand__;
+    PyObject *__ror__;
+    PyObject *__rxor__;
+    PyObject *__rlshift__;
+    PyObject *__rrshift__;
+    PyObject *__eq__;
+    PyObject *__ne__;
+    PyObject *__gt__;
+    PyObject *__le__;
+    PyObject *__lt__;
+    PyObject *__ge__;
+    PyObject *clear;
+    PyObject *close_;
+    PyObject *copy;
+    PyObject *keys;
+    PyObject *items;
+    PyObject *join;
+    PyObject *send;
+    PyObject *setdefault;
+    PyObject *startswith;
+    PyObject *throw_;
+    PyObject *translate;
+    PyObject *update;
+    PyObject *values;
+} mypyc_interned_str_struct;
+
+extern mypyc_interned_str_struct mypyc_interned_str;
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif