Add support for the Shapely extension

andreittr · andreittr · commit 517cf61bf268 · 2023-08-11T19:57:57.000+02:00
This change only adds the necessary module registrations and Kconfig
options. Shapely itself is provided by its own unikraft port.

Signed-off-by: Andrei Tatar &lt;andrei@unikraft.io&gt;
diff --git a/Config.uk b/Config.uk
@@ -88,5 +88,11 @@ config LIBPYTHON3_EXTENSION_NUMPY
     default y
 endif
 
+if LIBPYTHON_SHAPELY
+config LIBPYTHON3_EXTENSION_SHAPELY
+    bool "Shapely"
+    default y
+endif
+
 endif
 endif
diff --git a/modules_config.c b/modules_config.c
@@ -186,6 +186,14 @@ extern PyObject* PyInit__umath_tests(void);
 /* Numpy end */
 #endif
 
+#if CONFIG_LIBPYTHON3_EXTENSION_SHAPELY
+/* Shapely */
+extern PyObject* PyInit_lib(void);
+extern PyObject* PyInit__geometry_helpers(void);
+extern PyObject* PyInit__geos(void);
+/* Shapely end */
+#endif
+
 struct _inittab _PyImport_Inittab[] = {
 
     {"posix", PyInit_posix},
@@ -384,6 +392,12 @@ struct _inittab _PyImport_Inittab[] = {
     {"numpy_core__umath_tests", PyInit__umath_tests},
 #endif
 
+#if CONFIG_LIBPYTHON3_EXTENSION_SHAPELY
+    {"shapely_lib", PyInit_lib},
+    {"shapely__geometry_helpers", PyInit__geometry_helpers},
+    {"shapely__geos", PyInit__geos},
+#endif
+
     /* Sentinel */
     {0, 0}
 };
